├── .editorconfig
├── .gitattributes
├── .github
├── actions
│ └── setup
│ │ └── action.yml
└── workflows
│ ├── check-android.yml
│ ├── check-clang.yml
│ └── ci.yml
├── .gitignore
├── .gitmodules
├── .nvmrc
├── .watchmanconfig
├── .yarn
├── plugins
│ └── @yarnpkg
│ │ ├── plugin-interactive-tools.cjs
│ │ └── plugin-workspace-tools.cjs
└── releases
│ └── yarn-3.6.1.cjs
├── .yarnrc.yml
├── CODE_OF_CONDUCT.md
├── CONTRIBUTING.md
├── LICENSE
├── README.md
├── android
├── .editorconfig
├── build.gradle
├── gradle.properties
└── src
│ ├── main
│ ├── AndroidManifest.xml
│ ├── AndroidManifestNew.xml
│ └── java
│ │ └── com
│ │ └── ai
│ │ ├── AiModule.kt
│ │ ├── AiPackage.kt
│ │ ├── ChatState.kt
│ │ └── ModelState.kt
│ ├── newarch
│ └── AiSpec.kt
│ └── oldarch
│ └── AiSpec.kt
├── babel.config.js
├── example
├── .watchmanconfig
├── android
│ ├── app
│ │ ├── build.gradle
│ │ ├── debug.keystore
│ │ ├── proguard-rules.pro
│ │ └── src
│ │ │ ├── debug
│ │ │ └── AndroidManifest.xml
│ │ │ └── main
│ │ │ ├── AndroidManifest.xml
│ │ │ ├── java
│ │ │ └── com
│ │ │ │ └── aiexample
│ │ │ │ ├── MainActivity.kt
│ │ │ │ └── MainApplication.kt
│ │ │ └── res
│ │ │ ├── drawable
│ │ │ └── rn_edit_text_material.xml
│ │ │ ├── mipmap-hdpi
│ │ │ ├── ic_launcher.png
│ │ │ └── ic_launcher_round.png
│ │ │ ├── mipmap-mdpi
│ │ │ ├── ic_launcher.png
│ │ │ └── ic_launcher_round.png
│ │ │ ├── mipmap-xhdpi
│ │ │ ├── ic_launcher.png
│ │ │ └── ic_launcher_round.png
│ │ │ ├── mipmap-xxhdpi
│ │ │ ├── ic_launcher.png
│ │ │ └── ic_launcher_round.png
│ │ │ ├── mipmap-xxxhdpi
│ │ │ ├── ic_launcher.png
│ │ │ └── ic_launcher_round.png
│ │ │ └── values
│ │ │ ├── strings.xml
│ │ │ └── styles.xml
│ ├── build.gradle
│ ├── gradle.properties
│ ├── gradle
│ │ └── wrapper
│ │ │ ├── gradle-wrapper.jar
│ │ │ └── gradle-wrapper.properties
│ ├── gradlew
│ ├── gradlew.bat
│ ├── mlc-package-config.json
│ └── settings.gradle
├── app.json
├── assets
│ └── avatar.png
├── babel.config.js
├── index.js
├── ios
│ ├── .xcode.env
│ ├── AiExample-Bridging-Header.h
│ ├── AiExample.xcodeproj
│ │ ├── project.pbxproj
│ │ └── xcshareddata
│ │ │ └── xcschemes
│ │ │ └── AiExample.xcscheme
│ ├── AiExample.xcworkspace
│ │ ├── contents.xcworkspacedata
│ │ └── xcshareddata
│ │ │ └── IDEWorkspaceChecks.plist
│ ├── AiExample
│ │ ├── AiExample.entitlements
│ │ ├── AppDelegate.h
│ │ ├── AppDelegate.mm
│ │ ├── Images.xcassets
│ │ │ ├── AppIcon.appiconset
│ │ │ │ └── Contents.json
│ │ │ └── Contents.json
│ │ ├── Info.plist
│ │ ├── LaunchScreen.storyboard
│ │ ├── PrivacyInfo.xcprivacy
│ │ └── main.m
│ ├── AiExampleTests
│ │ ├── AiExampleTests.m
│ │ └── Info.plist
│ ├── File.swift
│ ├── Podfile
│ ├── Podfile.lock
│ ├── README.md
│ └── mlc-package-config.json
├── metro.config.js
├── mlc-config.json
├── package.json
├── polyfills.js
├── react-native.config.js
└── src
│ ├── App.tsx
│ ├── ModelSelection.tsx
│ └── NetworkInfo.tsx
├── ios
├── .clang-format
├── Ai.h
├── Ai.mm
├── BackgroundWorker.h
├── BackgroundWorker.mm
├── EngineState.h
├── EngineState.mm
├── LLMEngine.h
├── LLMEngine.mm
├── MLCEngine.h
└── MLCEngine.mm
├── lefthook.yml
├── package.json
├── react-native-ai.podspec
├── scripts
├── format-android.sh
├── format-ios.sh
└── mlc-prepare.js
├── src
├── NativeAi.ts
├── __tests__
│ └── index.test.tsx
├── index.tsx
└── polyfills.ts
├── tsconfig.build.json
├── tsconfig.json
├── turbo.json
└── yarn.lock
/.editorconfig:
--------------------------------------------------------------------------------
1 | # EditorConfig helps developers define and maintain consistent
2 | # coding styles between different editors and IDEs
3 | # editorconfig.org
4 |
5 | root = true
6 |
7 | [*]
8 |
9 | indent_style = space
10 | indent_size = 2
11 |
12 | end_of_line = lf
13 | charset = utf-8
14 | trim_trailing_whitespace = true
15 | insert_final_newline = true
16 |
--------------------------------------------------------------------------------
/.gitattributes:
--------------------------------------------------------------------------------
1 | *.pbxproj -text
2 | # specific for windows script files
3 | *.bat text eol=crlf
--------------------------------------------------------------------------------
/.github/actions/setup/action.yml:
--------------------------------------------------------------------------------
1 | name: Setup
2 | description: Setup Node.js and install dependencies
3 |
4 | runs:
5 | using: composite
6 | steps:
7 | - name: Setup Node.js
8 | uses: actions/setup-node@v3
9 | with:
10 | node-version-file: .nvmrc
11 |
12 | - name: Cache dependencies
13 | id: yarn-cache
14 | uses: actions/cache@v3
15 | with:
16 | path: |
17 | **/node_modules
18 | .yarn/install-state.gz
19 | key: ${{ runner.os }}-yarn-${{ hashFiles('yarn.lock') }}-${{ hashFiles('**/package.json', '!node_modules/**') }}
20 | restore-keys: |
21 | ${{ runner.os }}-yarn-${{ hashFiles('yarn.lock') }}
22 | ${{ runner.os }}-yarn-
23 |
24 | - name: Install dependencies
25 | if: steps.yarn-cache.outputs.cache-hit != 'true'
26 | run: yarn install --immutable
27 | shell: bash
28 |
--------------------------------------------------------------------------------
/.github/workflows/check-android.yml:
--------------------------------------------------------------------------------
1 | name: Check Android
2 |
3 | on:
4 | push:
5 | branches:
6 | - main
7 | paths:
8 | - '.github/workflows/check-android.yml'
9 | - 'android/**'
10 | pull_request:
11 | paths:
12 | - '.github/workflows/check-android.yml'
13 | - 'android/**'
14 |
15 | jobs:
16 | Kotlin-Lint:
17 | runs-on: ubuntu-latest
18 | steps:
19 | - uses: actions/checkout@v4
20 | - run: |
21 | curl -sSLO https://github.com/pinterest/ktlint/releases/download/1.0.1/ktlint && chmod a+x ktlint && sudo mv ktlint /usr/local/bin/
22 | - name: run ktlint
23 | run: |
24 | ./scripts/format-android.sh
25 | continue-on-error: true
26 | - uses: yutailang0119/action-ktlint@v3
27 | with:
28 | report-path: ./android/build/*.xml
29 | continue-on-error: false
30 | - uses: actions/upload-artifact@v4
31 | with:
32 | name: ktlint-report
33 | path: ./android/build/*.xml
34 |
--------------------------------------------------------------------------------
/.github/workflows/check-clang.yml:
--------------------------------------------------------------------------------
1 | name: Check CLang
2 |
3 | on:
4 | push:
5 | branches:
6 | - main
7 | paths:
8 | - '.github/workflows/check-clang.yml'
9 | - 'ios/**'
10 | pull_request:
11 | branches:
12 | - main
13 | paths:
14 | - '.github/workflows/check-clang.yml'
15 | - 'ios/**'
16 |
17 | jobs:
18 | CLang-Format:
19 | runs-on: macos-latest
20 | steps:
21 | - uses: actions/checkout@v4
22 | - name: Install clang-format
23 | run: brew install clang-format
24 | - name: Check ios clang formatting
25 | run: |
26 | ./scripts/format-ios.sh
27 | - name: Check for changes
28 | run: git diff --exit-code HEAD
29 |
--------------------------------------------------------------------------------
/.github/workflows/ci.yml:
--------------------------------------------------------------------------------
1 | name: CI
2 | on:
3 | push:
4 | branches:
5 | - main
6 | pull_request:
7 | branches:
8 | - main
9 |
10 | jobs:
11 | lint:
12 | runs-on: ubuntu-latest
13 | steps:
14 | - name: Checkout
15 | uses: actions/checkout@v3
16 |
17 | - name: Setup
18 | uses: ./.github/actions/setup
19 |
20 | - name: Lint files
21 | run: yarn lint
22 |
23 | - name: Typecheck files
24 | run: yarn typecheck
25 |
26 | test:
27 | runs-on: ubuntu-latest
28 | steps:
29 | - name: Checkout
30 | uses: actions/checkout@v3
31 |
32 | - name: Setup
33 | uses: ./.github/actions/setup
34 |
35 | - name: Run unit tests
36 | run: yarn test --maxWorkers=2 --coverage
37 |
38 | build-library:
39 | runs-on: ubuntu-latest
40 | steps:
41 | - name: Checkout
42 | uses: actions/checkout@v3
43 |
44 | - name: Setup
45 | uses: ./.github/actions/setup
46 |
47 | - name: Build package
48 | run: yarn prepare
49 | # TODO: Bring it back with all the following steps with setting up the environment variables related to MLC
50 | # build-android:
51 | # runs-on: ubuntu-latest
52 | # env:
53 | # TURBO_CACHE_DIR: .turbo/android
54 | # steps:
55 | # - name: Checkout
56 | # uses: actions/checkout@v4
57 |
58 | # - name: Setup
59 | # uses: ./.github/actions/setup
60 |
61 | # - name: Cache turborepo for Android
62 | # uses: actions/cache@v4
63 | # with:
64 | # path: ${{ env.TURBO_CACHE_DIR }}
65 | # key: ${{ runner.os }}-turborepo-android-${{ hashFiles('yarn.lock') }}
66 | # restore-keys: |
67 | # ${{ runner.os }}-turborepo-android-
68 |
69 | # - name: Check turborepo cache for Android
70 | # run: |
71 | # TURBO_CACHE_STATUS=$(node -p "($(yarn turbo run build:android --cache-dir="${{ env.TURBO_CACHE_DIR }}" --dry=json)).tasks.find(t => t.task === 'build:android').cache.status")
72 |
73 | # if [[ $TURBO_CACHE_STATUS == "HIT" ]]; then
74 | # echo "turbo_cache_hit=1" >> $GITHUB_ENV
75 | # fi
76 |
77 | # - name: Install JDK
78 | # if: env.turbo_cache_hit != 1
79 | # uses: actions/setup-java@v4
80 | # with:
81 | # distribution: 'zulu'
82 | # java-version: '17'
83 |
84 | # - name: Finalize Android SDK
85 | # if: env.turbo_cache_hit != 1
86 | # run: |
87 | # /bin/bash -c "yes | $ANDROID_HOME/cmdline-tools/latest/bin/sdkmanager --licenses > /dev/null"
88 |
89 | # - name: Create local.properties
90 | # run: echo "sdk.dir=$ANDROID_HOME" > example/android/local.properties
91 |
92 | # - name: Cache Gradle
93 | # if: env.turbo_cache_hit != 1
94 | # uses: actions/cache@v4
95 | # with:
96 | # path: |
97 | # ~/.gradle/wrapper
98 | # ~/.gradle/caches
99 | # key: ${{ runner.os }}-gradle-${{ hashFiles('example/android/gradle/wrapper/gradle-wrapper.properties') }}
100 | # restore-keys: |
101 | # ${{ runner.os }}-gradle-
102 |
103 | # - name: Build example for Android
104 | # env:
105 | # JAVA_OPTS: '-XX:MaxHeapSize=6g'
106 | # run: |
107 | # yarn turbo run build:android --cache-dir="${{ env.TURBO_CACHE_DIR }}"
108 |
109 | # build-ios:
110 | # runs-on: macos-14
111 | # env:
112 | # TURBO_CACHE_DIR: .turbo/ios
113 | # steps:
114 | # - name: Checkout
115 | # uses: actions/checkout@v3
116 | # with:
117 | # submodules: recursive
118 |
119 | # - name: Setup
120 | # uses: ./.github/actions/setup
121 |
122 | # - name: Cache turborepo for iOS
123 | # uses: actions/cache@v3
124 | # with:
125 | # path: ${{ env.TURBO_CACHE_DIR }}
126 | # key: ${{ runner.os }}-turborepo-ios-${{ hashFiles('yarn.lock') }}
127 | # restore-keys: |
128 | # ${{ runner.os }}-turborepo-ios-
129 |
130 | # - name: Check turborepo cache for iOS
131 | # run: |
132 | # TURBO_CACHE_STATUS=$(node -p "($(yarn turbo run build:ios --cache-dir="${{ env.TURBO_CACHE_DIR }}" --dry=json)).tasks.find(t => t.task === 'build:ios').cache.status")
133 |
134 | # if [[ $TURBO_CACHE_STATUS == "HIT" ]]; then
135 | # echo "turbo_cache_hit=1" >> $GITHUB_ENV
136 | # fi
137 |
138 | # - name: Cache cocoapods
139 | # if: env.turbo_cache_hit != 1
140 | # id: cocoapods-cache
141 | # uses: actions/cache@v3
142 | # with:
143 | # path: |
144 | # **/ios/Pods
145 | # key: ${{ runner.os }}-cocoapods-${{ hashFiles('example/ios/Podfile.lock') }}
146 | # restore-keys: |
147 | # ${{ runner.os }}-cocoapods-
148 |
149 | # - name: Install cocoapods
150 | # if: env.turbo_cache_hit != 1 && steps.cocoapods-cache.outputs.cache-hit != 'true'
151 | # run: |
152 | # cd example/ios
153 | # pod install
154 | # env:
155 | # NO_FLIPPER: 1
156 |
157 | # - name: Build example for iOS
158 | # run: |
159 | # yarn turbo run build:ios --cache-dir="${{ env.TURBO_CACHE_DIR }}"
160 |
--------------------------------------------------------------------------------
/.gitignore:
--------------------------------------------------------------------------------
1 | # OSX
2 | #
3 | .DS_Store
4 |
5 | # XDE
6 | .expo/
7 |
8 | # VSCode
9 | .vscode/
10 | jsconfig.json
11 |
12 | # Xcode
13 | #
14 | build/
15 | *.pbxuser
16 | !default.pbxuser
17 | *.mode1v3
18 | !default.mode1v3
19 | *.mode2v3
20 | !default.mode2v3
21 | *.perspectivev3
22 | !default.perspectivev3
23 | xcuserdata
24 | *.xccheckout
25 | *.moved-aside
26 | DerivedData
27 | *.hmap
28 | *.ipa
29 | *.xcuserstate
30 | project.xcworkspace
31 | .xcode.env.local
32 |
33 | # Android/IJ
34 | #
35 | .classpath
36 | .cxx
37 | .gradle
38 | .idea
39 | .project
40 | .settings
41 | local.properties
42 | android.iml
43 |
44 | # Cocoapods
45 | #
46 | example/ios/Pods
47 |
48 | # Ruby
49 | example/vendor/
50 |
51 | # node.js
52 | #
53 | node_modules/
54 | npm-debug.log
55 | yarn-debug.log
56 | yarn-error.log
57 |
58 | # BUCK
59 | buck-out/
60 | \.buckd/
61 | android/app/libs
62 | android/keystores/debug.keystore
63 |
64 | # Yarn
65 | .yarn/*
66 | !.yarn/patches
67 | !.yarn/plugins
68 | !.yarn/releases
69 | !.yarn/sdks
70 | !.yarn/versions
71 |
72 | # Expo
73 | .expo/
74 |
75 | # Turborepo
76 | .turbo/
77 |
78 | # generated by bob
79 | lib/
80 |
81 | # other
82 |
83 | dist/
--------------------------------------------------------------------------------
/.gitmodules:
--------------------------------------------------------------------------------
1 | [submodule "3rdparty/tvm"]
2 | path = 3rdparty/tvm
3 | url = https://github.com/mlc-ai/relax.git
4 |
--------------------------------------------------------------------------------
/.nvmrc:
--------------------------------------------------------------------------------
1 | v18
2 |
--------------------------------------------------------------------------------
/.watchmanconfig:
--------------------------------------------------------------------------------
1 | {}
--------------------------------------------------------------------------------
/.yarnrc.yml:
--------------------------------------------------------------------------------
1 | nodeLinker: node-modules
2 | nmHoistingLimits: workspaces
3 |
4 | plugins:
5 | - path: .yarn/plugins/@yarnpkg/plugin-interactive-tools.cjs
6 | spec: "@yarnpkg/plugin-interactive-tools"
7 | - path: .yarn/plugins/@yarnpkg/plugin-workspace-tools.cjs
8 | spec: "@yarnpkg/plugin-workspace-tools"
9 |
10 | yarnPath: .yarn/releases/yarn-3.6.1.cjs
11 |
--------------------------------------------------------------------------------
/CODE_OF_CONDUCT.md:
--------------------------------------------------------------------------------
1 |
2 | # Contributor Covenant Code of Conduct
3 |
4 | ## Our Pledge
5 |
6 | We as members, contributors, and leaders pledge to make participation in our
7 | community a harassment-free experience for everyone, regardless of age, body
8 | size, visible or invisible disability, ethnicity, sex characteristics, gender
9 | identity and expression, level of experience, education, socio-economic status,
10 | nationality, personal appearance, race, caste, color, religion, or sexual
11 | identity and orientation.
12 |
13 | We pledge to act and interact in ways that contribute to an open, welcoming,
14 | diverse, inclusive, and healthy community.
15 |
16 | ## Our Standards
17 |
18 | Examples of behavior that contributes to a positive environment for our
19 | community include:
20 |
21 | * Demonstrating empathy and kindness toward other people
22 | * Being respectful of differing opinions, viewpoints, and experiences
23 | * Giving and gracefully accepting constructive feedback
24 | * Accepting responsibility and apologizing to those affected by our mistakes,
25 | and learning from the experience
26 | * Focusing on what is best not just for us as individuals, but for the overall
27 | community
28 |
29 | Examples of unacceptable behavior include:
30 |
31 | * The use of sexualized language or imagery, and sexual attention or advances of
32 | any kind
33 | * Trolling, insulting or derogatory comments, and personal or political attacks
34 | * Public or private harassment
35 | * Publishing others' private information, such as a physical or email address,
36 | without their explicit permission
37 | * Other conduct which could reasonably be considered inappropriate in a
38 | professional setting
39 |
40 | ## Enforcement Responsibilities
41 |
42 | Community leaders are responsible for clarifying and enforcing our standards of
43 | acceptable behavior and will take appropriate and fair corrective action in
44 | response to any behavior that they deem inappropriate, threatening, offensive,
45 | or harmful.
46 |
47 | Community leaders have the right and responsibility to remove, edit, or reject
48 | comments, commits, code, wiki edits, issues, and other contributions that are
49 | not aligned to this Code of Conduct, and will communicate reasons for moderation
50 | decisions when appropriate.
51 |
52 | ## Scope
53 |
54 | This Code of Conduct applies within all community spaces, and also applies when
55 | an individual is officially representing the community in public spaces.
56 | Examples of representing our community include using an official e-mail address,
57 | posting via an official social media account, or acting as an appointed
58 | representative at an online or offline event.
59 |
60 | ## Enforcement
61 |
62 | Instances of abusive, harassing, or otherwise unacceptable behavior may be
63 | reported to the community leaders responsible for enforcement at
64 | [INSERT CONTACT METHOD].
65 | All complaints will be reviewed and investigated promptly and fairly.
66 |
67 | All community leaders are obligated to respect the privacy and security of the
68 | reporter of any incident.
69 |
70 | ## Enforcement Guidelines
71 |
72 | Community leaders will follow these Community Impact Guidelines in determining
73 | the consequences for any action they deem in violation of this Code of Conduct:
74 |
75 | ### 1. Correction
76 |
77 | **Community Impact**: Use of inappropriate language or other behavior deemed
78 | unprofessional or unwelcome in the community.
79 |
80 | **Consequence**: A private, written warning from community leaders, providing
81 | clarity around the nature of the violation and an explanation of why the
82 | behavior was inappropriate. A public apology may be requested.
83 |
84 | ### 2. Warning
85 |
86 | **Community Impact**: A violation through a single incident or series of
87 | actions.
88 |
89 | **Consequence**: A warning with consequences for continued behavior. No
90 | interaction with the people involved, including unsolicited interaction with
91 | those enforcing the Code of Conduct, for a specified period of time. This
92 | includes avoiding interactions in community spaces as well as external channels
93 | like social media. Violating these terms may lead to a temporary or permanent
94 | ban.
95 |
96 | ### 3. Temporary Ban
97 |
98 | **Community Impact**: A serious violation of community standards, including
99 | sustained inappropriate behavior.
100 |
101 | **Consequence**: A temporary ban from any sort of interaction or public
102 | communication with the community for a specified period of time. No public or
103 | private interaction with the people involved, including unsolicited interaction
104 | with those enforcing the Code of Conduct, is allowed during this period.
105 | Violating these terms may lead to a permanent ban.
106 |
107 | ### 4. Permanent Ban
108 |
109 | **Community Impact**: Demonstrating a pattern of violation of community
110 | standards, including sustained inappropriate behavior, harassment of an
111 | individual, or aggression toward or disparagement of classes of individuals.
112 |
113 | **Consequence**: A permanent ban from any sort of public interaction within the
114 | community.
115 |
116 | ## Attribution
117 |
118 | This Code of Conduct is adapted from the [Contributor Covenant][homepage],
119 | version 2.1, available at
120 | [https://www.contributor-covenant.org/version/2/1/code_of_conduct.html][v2.1].
121 |
122 | Community Impact Guidelines were inspired by
123 | [Mozilla's code of conduct enforcement ladder][Mozilla CoC].
124 |
125 | For answers to common questions about this code of conduct, see the FAQ at
126 | [https://www.contributor-covenant.org/faq][FAQ]. Translations are available at
127 | [https://www.contributor-covenant.org/translations][translations].
128 |
129 | [homepage]: https://www.contributor-covenant.org
130 | [v2.1]: https://www.contributor-covenant.org/version/2/1/code_of_conduct.html
131 | [Mozilla CoC]: https://github.com/mozilla/diversity
132 | [FAQ]: https://www.contributor-covenant.org/faq
133 | [translations]: https://www.contributor-covenant.org/translations
134 |
--------------------------------------------------------------------------------
/CONTRIBUTING.md:
--------------------------------------------------------------------------------
1 | # Contributing
2 |
3 | Contributions are always welcome, no matter how large or small!
4 |
5 | We want this community to be friendly and respectful to each other. Please follow it in all your interactions with the project. Before contributing, please read the [code of conduct](./CODE_OF_CONDUCT.md).
6 |
7 | ## Development workflow
8 |
9 | This project is a monorepo managed using [Yarn workspaces](https://yarnpkg.com/features/workspaces). It contains the following packages:
10 |
11 | - The library package in the root directory.
12 | - An example app in the `example/` directory.
13 |
14 | To get started with the project, run `yarn` in the root directory to install the required dependencies for each package:
15 |
16 | ```sh
17 | yarn
18 | ```
19 |
20 | > Since the project relies on Yarn workspaces, you cannot use [`npm`](https://github.com/npm/cli) for development.
21 |
22 | The [example app](/example/) demonstrates usage of the library. You need to run it to test any changes you make.
23 |
24 | It is configured to use the local version of the library, so any changes you make to the library's source code will be reflected in the example app. Changes to the library's JavaScript code will be reflected in the example app without a rebuild, but native code changes will require a rebuild of the example app.
25 |
26 | If you want to use Android Studio or XCode to edit the native code, you can open the `example/android` or `example/ios` directories respectively in those editors. To edit the Objective-C or Swift files, open `example/ios/AiExample.xcworkspace` in XCode and find the source files at `Pods > Development Pods > react-native-ai`.
27 |
28 | To edit the the Kotlin files, open `example/android` in Android studio and find the source files at `react-native-ai` under `Android`.
29 |
30 | You can use various commands from the root directory to work with the project.
31 |
32 | To start the packager:
33 |
34 | ```sh
35 | yarn example start
36 | ```
37 |
38 | To run the example app on Android:
39 |
40 | ```sh
41 | yarn example android
42 | ```
43 |
44 | To run the example app on iOS:
45 |
46 | ```sh
47 | yarn example ios
48 | ```
49 |
50 | By default, the example is configured to build with the old architecture. To run the example with the new architecture, you can do the following:
51 |
52 | 1. For Android, run:
53 |
54 | ```sh
55 | ORG_GRADLE_PROJECT_newArchEnabled=true yarn example android
56 | ```
57 |
58 | 2. For iOS, run:
59 |
60 | ```sh
61 | cd example/ios
62 | RCT_NEW_ARCH_ENABLED=1 pod install
63 | cd -
64 | yarn example ios
65 | ```
66 |
67 | If you are building for a different architecture than your previous build, make sure to remove the build folders first. You can run the following command to cleanup all build folders:
68 |
69 | ```sh
70 | yarn clean
71 | ```
72 |
73 | To confirm that the app is running with the new architecture, you can check the Metro logs for a message like this:
74 |
75 | ```sh
76 | Running "AiExample" with {"fabric":true,"initialProps":{"concurrentRoot":true},"rootTag":1}
77 | ```
78 |
79 | Note the `"fabric":true` and `"concurrentRoot":true` properties.
80 |
81 | Make sure your code passes TypeScript and ESLint. Run the following to verify:
82 |
83 | ```sh
84 | yarn typecheck
85 | yarn lint
86 | ```
87 |
88 | To fix formatting errors, run the following:
89 |
90 | ```sh
91 | yarn lint --fix
92 | ```
93 |
94 | Remember to add tests for your change if possible. Run the unit tests by:
95 |
96 | ```sh
97 | yarn test
98 | ```
99 |
100 | ### Commit message convention
101 |
102 | We follow the [conventional commits specification](https://www.conventionalcommits.org/en) for our commit messages:
103 |
104 | - `fix`: bug fixes, e.g. fix crash due to deprecated method.
105 | - `feat`: new features, e.g. add new method to the module.
106 | - `refactor`: code refactor, e.g. migrate from class components to hooks.
107 | - `docs`: changes into documentation, e.g. add usage example for the module..
108 | - `test`: adding or updating tests, e.g. add integration tests using detox.
109 | - `chore`: tooling changes, e.g. change CI config.
110 |
111 | Our pre-commit hooks verify that your commit message matches this format when committing.
112 |
113 | ### Linting and tests
114 |
115 | [ESLint](https://eslint.org/), [Prettier](https://prettier.io/), [TypeScript](https://www.typescriptlang.org/)
116 |
117 | We use [TypeScript](https://www.typescriptlang.org/) for type checking, [ESLint](https://eslint.org/) with [Prettier](https://prettier.io/) for linting and formatting the code, and [Jest](https://jestjs.io/) for testing.
118 |
119 | Our pre-commit hooks verify that the linter and tests pass when committing.
120 |
121 | ### Publishing to npm
122 |
123 | We use [release-it](https://github.com/release-it/release-it) to make it easier to publish new versions. It handles common tasks like bumping version based on semver, creating tags and releases etc.
124 |
125 | To publish new versions, run the following:
126 |
127 | ```sh
128 | yarn release
129 | ```
130 |
131 | ### Scripts
132 |
133 | The `package.json` file contains various scripts for common tasks:
134 |
135 | - `yarn`: setup project by installing dependencies.
136 | - `yarn typecheck`: type-check files with TypeScript.
137 | - `yarn lint`: lint files with ESLint.
138 | - `yarn test`: run unit tests with Jest.
139 | - `yarn example start`: start the Metro server for the example app.
140 | - `yarn example android`: run the example app on Android.
141 | - `yarn example ios`: run the example app on iOS.
142 |
143 | ### Sending a pull request
144 |
145 | > **Working on your first pull request?** You can learn how from this _free_ series: [How to Contribute to an Open Source Project on GitHub](https://app.egghead.io/playlists/how-to-contribute-to-an-open-source-project-on-github).
146 |
147 | When you're sending a pull request:
148 |
149 | - Prefer small pull requests focused on one change.
150 | - Verify that linters and tests are passing.
151 | - Review the documentation to make sure it looks good.
152 | - Follow the pull request template when opening a pull request.
153 | - For pull requests that change the API or implementation, discuss with maintainers first by opening an issue.
154 |
--------------------------------------------------------------------------------
/LICENSE:
--------------------------------------------------------------------------------
1 | MIT License
2 |
3 | Copyright (c) 2024 szymonrybczak
4 | Permission is hereby granted, free of charge, to any person obtaining a copy
5 | of this software and associated documentation files (the "Software"), to deal
6 | in the Software without restriction, including without limitation the rights
7 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8 | copies of the Software, and to permit persons to whom the Software is
9 | furnished to do so, subject to the following conditions:
10 |
11 | The above copyright notice and this permission notice shall be included in all
12 | copies or substantial portions of the Software.
13 |
14 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
20 | SOFTWARE.
21 |
--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
1 | react-native-ai
2 |
3 |
7 |
8 |
9 |
10 |
11 | [](https://github.com/callstackincubator/ai/blob/main/LICENSE)
12 | [](https://www.npmjs.org/package/react-native-ai)
13 | [](https://www.npmjs.org/package/react-native-ai)
14 | [](https://www.npmjs.org/package/react-native-ai)
15 |
16 |
17 |
18 |
19 |
20 |
21 | ## Installation
22 |
23 | #### 1. Install the package
24 |
25 | ```
26 | npm install react-native-ai
27 | ```
28 |
29 | #### 2. Clone MLC LLM Engine repository and set environment variable.
30 |
31 | ```
32 | git clone https://github.com/mlc-ai/mlc-llm
33 | cd mlc-llm
34 | git submodule update --init --recursive
35 | MLC_LLM_SOURCE_DIR=$(pwd) // Add this to your environment variables e.g. in .zshrc
36 | ```
37 |
38 | > [!IMPORTANT]
39 | > Ensure that `mlc_llm` works and `MLC_LLM_SOURCE_DIR` is set in your environment variables.
40 |
41 | #### 3. Install `mlc_llm` CLI:
42 |
43 | To install `mlc_llm` CLI, please follow steps described [in the official guide](https://llm.mlc.ai/docs/install/mlc_llm.html).
44 |
45 | To ensure that CLI is installed correctly, run the following command:
46 |
47 | ```
48 | mlc_llm
49 | ```
50 |
51 | If you see any output then it means that CLI is installed correctly.
52 |
53 | #### 4. Add `mlc-config.json` with models and other properties to root directory of your project:
54 |
55 | ```js
56 | {
57 | "iphone": [
58 | {
59 | "model": "MODEL_NAME",
60 | "model_id": "MODEL_ID",
61 | // "estimated_vram_bytes": 3043000000
62 | }
63 | ],
64 | "android": [
65 | {
66 | "model": "MODEL_NAME",
67 | "model_id": "MODEL_ID",
68 | // "estimated_vram_bytes": 3043000000
69 | }
70 | ]
71 | }
72 | ```
73 |
74 | Read more about configuration for [Android](https://llm.mlc.ai/docs/deploy/android.html#customize-the-app) and for [iOS](https://llm.mlc.ai/docs/deploy/ios.html#customize-the-app).
75 |
76 | You can also check out [example config](https://github.com/callstackincubator/ai/blob/main/example/mlc-config.json) in the repository.
77 |
78 | #### 4. **[Android only]**
79 |
80 | If you want to execute models also on Android you need to set `ANDROID_NDK` and `TVM_NDK_CC` environment variables. Everything is described in [MLC LLM docs](https://llm.mlc.ai/docs/deploy/android.html#id2).
81 |
82 | #### 5. **[iOS only]** If you want to execute models also on iOS you need to:
83 |
84 | - Add "Increased Memory Limit" capability inside your Xcode project inside Signing & Capabilities tab.
85 | 
86 | - Install Cocoapods:
87 | ```
88 | cd ios && pod install
89 | ```
90 |
91 | #### 6. Run the following command to prepare binaries and static libraries for the project
92 |
93 | This command prepares the necessary binaries and static libraries for your React Native project by:
94 | - Validating required dependencies
95 | - Setting up platform-specific configurations (Android/iOS)
96 | - Running `mlc_llm package` for each platform to prepare model binaries
97 | - Handling environment variables and platform-specific requirements
98 |
99 | ```
100 | npx react-native-ai mlc-prepare
101 | ```
102 |
103 | #### 7. Add missing polyfills
104 |
105 | To make the Vercel AI SDK work in your project, you should include polyfills by first installing these pacakges:
106 |
107 | ```
108 | npm install @azure/core-asynciterator-polyfill @ungap/structured-clone web-streams-polyfill text-encoding
109 | ```
110 |
111 | and creating `polyfills.ts` file which will contain following imports:
112 |
113 | ```js
114 | import '@azure/core-asynciterator-polyfill';
115 | import structuredClone from '@ungap/structured-clone';
116 | import { polyfillGlobal } from 'react-native/Libraries/Utilities/PolyfillFunctions';
117 |
118 | const webStreamPolyfills = require('web-streams-polyfill/ponyfill/es6');
119 |
120 | polyfillGlobal('TextEncoder', () => require('text-encoding').TextEncoder);
121 | polyfillGlobal('ReadableStream', () => webStreamPolyfills.ReadableStream);
122 | polyfillGlobal('TransformStream', () => webStreamPolyfills.TransformStream);
123 | polyfillGlobal('WritableStream', () => webStreamPolyfills.WritableStream);
124 | polyfillGlobal('TextEncoderStream', () => webStreamPolyfills.TextEncoderStream);
125 | polyfillGlobal('structuredClone', () => structuredClone);
126 | ```
127 |
128 | Make sure to include them inside `index.js` before registering the root component.
129 |
130 | #### 8. Build the project! 🚀
131 |
132 | ## API
133 |
134 | This library provides first-class compatibility with the [Vercel AI SDK](https://sdk.vercel.ai/docs), allowing you to use familiar functions like `streamText` and `generateText` with locally run models.
135 |
136 | ### Key Functions
137 |
138 | * **`getModels(): Promise`**
139 | Retrieves a list of available models configured in your `mlc-config.json`.
140 |
141 | ```typescript
142 | import { getModels } from 'react-native-ai';
143 |
144 | async function listModels() {
145 | const models = await getModels();
146 | console.log('Available models:', models);
147 | }
148 | ```
149 |
150 | * **`downloadModel(modelId: string, callbacks?: DownloadCallbacks): Promise`**
151 | Downloads the specified model files. It accepts optional callbacks to track the download progress.
152 |
153 | ```typescript
154 | import { downloadModel, type DownloadProgress } from 'react-native-ai';
155 |
156 | await downloadModel('Mistral-7B-Instruct-v0.2-q3f16_1', {
157 | onStart: () => console.log('Download started...'),
158 | onProgress: (progress: DownloadProgress) => {
159 | console.log(`Downloading: ${progress.percentage.toFixed(2)}%`);
160 | },
161 | onComplete: () => console.log('Download complete!'),
162 | onError: (error: Error) => console.error('Download error:', error),
163 | });
164 | ```
165 |
166 | * **`prepareModel(modelId: string): Promise`**
167 | Prepares the downloaded model for use by loading it into memory, if the model is not on the device it'll fetch it. However we recommend using `downloadModel` before calling `prepareModel`.
168 |
169 | ```typescript
170 | import { prepareModel } from 'react-native-ai';
171 |
172 | await prepareModel('Mistral-7B-Instruct-v0.2-q3f16_1');
173 | console.log('Model is ready!');
174 | ```
175 |
176 | * **`getModel(modelId: string): LanguageModelV1`**
177 | Returns a model instance compatible with the Vercel AI SDK (`LanguageModelV1` interface). You can pass this instance directly to Vercel AI SDK functions.
178 |
179 | ### Usage with Vercel AI SDK
180 |
181 | Once a model is downloaded and prepared, you can use it with the Vercel AI SDK functions.
182 |
183 | ```typescript
184 | import { getModel, prepareModel } from 'react-native-ai';
185 | import { streamText, type CoreMessage } from 'ai';
186 |
187 | async function runInference(modelId: string, messages: CoreMessage[]) {
188 | await prepareModel(modelId);
189 | const llm = getModel(modelId);
190 |
191 | const { textStream } = streamText({
192 | model: llm,
193 | messages: messages,
194 | });
195 |
196 | for await (const textPart of textStream) {
197 | console.log(textPart);
198 | }
199 | }
200 |
201 | const exampleMessages: CoreMessage[] = [
202 | { role: 'user', content: 'Hello! Tell me a short story.' },
203 | ];
204 | runInference('Mistral-7B-Instruct-v0.2-q3f16_1', exampleMessages);
205 | ```
206 |
207 | This setup allows you to leverage the power of the Vercel AI SDK's unified API while running models directly on the user's device.
208 |
209 | ## Contributing
210 |
211 | Read the [contribution guidelines](/CONTRIBUTING.md) before contributing.
212 |
213 | ## Made with ❤️ at Callstack
214 |
215 | react-native-ai is an open source project and will always remain free to use. If you think it's cool, please star it 🌟.
216 |
217 | [Callstack][callstack-readme-with-love] is a group of React and React Native geeks, contact us at [hello@callstack.com](mailto:hello@callstack.com) if you need any help with these or just want to say hi!
218 |
219 | ---
220 |
221 | Made with [create-react-native-library](https://github.com/callstack/react-native-builder-bob)
222 |
223 | [callstack-readme-with-love]: https://callstack.com/?utm_source=github.com&utm_medium=referral&utm_campaign=react-native-ai&utm_term=readme-with-love
224 |
--------------------------------------------------------------------------------
/android/.editorconfig:
--------------------------------------------------------------------------------
1 | [*.{kt,kts}]
2 | indent_style=space
3 | indent_size=2
4 | continuation_indent_size=2
5 | insert_final_newline=true
6 | max_line_length=160
7 | ktlint_code_style=android_studio
8 | ktlint_standard=enabled
9 | ktlint_experimental=enabled
10 | ktlint_standard_filename=disabled # dont require PascalCase filenames
11 | ktlint_standard_no-wildcard-imports=disabled # allow .* imports
12 | ktlint_function_signature_body_expression_wrapping=multiline
13 | ij_kotlin_allow_trailing_comma_on_call_site=false
14 | ij_kotlin_allow_trailing_comma=false
15 |
--------------------------------------------------------------------------------
/android/build.gradle:
--------------------------------------------------------------------------------
1 | buildscript {
2 | // Buildscript is evaluated before everything else so we can't use getExtOrDefault
3 | def kotlin_version = rootProject.ext.has("kotlinVersion") ? rootProject.ext.get("kotlinVersion") : project.properties["Ai_kotlinVersion"]
4 |
5 | repositories {
6 | google()
7 | mavenCentral()
8 | }
9 |
10 | dependencies {
11 | classpath "com.android.tools.build:gradle:7.2.1"
12 | // noinspection DifferentKotlinGradleVersion
13 | classpath "org.jetbrains.kotlin:kotlin-gradle-plugin:$kotlin_version"
14 | }
15 | }
16 |
17 | def reactNativeArchitectures() {
18 | def value = rootProject.getProperties().get("reactNativeArchitectures")
19 | return value ? value.split(",") : ["armeabi-v7a", "x86", "x86_64", "arm64-v8a"]
20 | }
21 |
22 | def isNewArchitectureEnabled() {
23 | return rootProject.hasProperty("newArchEnabled") && rootProject.getProperty("newArchEnabled") == "true"
24 | }
25 |
26 | apply plugin: "com.android.library"
27 | apply plugin: "kotlin-android"
28 |
29 | if (isNewArchitectureEnabled()) {
30 | apply plugin: "com.facebook.react"
31 | }
32 |
33 | def getExtOrDefault(name) {
34 | return rootProject.ext.has(name) ? rootProject.ext.get(name) : project.properties["Ai_" + name]
35 | }
36 |
37 | def getExtOrIntegerDefault(name) {
38 | return rootProject.ext.has(name) ? rootProject.ext.get(name) : (project.properties["Ai_" + name]).toInteger()
39 | }
40 |
41 | def supportsNamespace() {
42 | def parsed = com.android.Version.ANDROID_GRADLE_PLUGIN_VERSION.tokenize('.')
43 | def major = parsed[0].toInteger()
44 | def minor = parsed[1].toInteger()
45 |
46 | // Namespace support was added in 7.3.0
47 | return (major == 7 && minor >= 3) || major >= 8
48 | }
49 |
50 | android {
51 | if (supportsNamespace()) {
52 | namespace "com.ai"
53 |
54 | sourceSets {
55 | main {
56 | manifest.srcFile "src/main/AndroidManifestNew.xml"
57 | }
58 | }
59 | }
60 |
61 | compileSdkVersion getExtOrIntegerDefault("compileSdkVersion")
62 |
63 | defaultConfig {
64 | minSdkVersion getExtOrIntegerDefault("minSdkVersion")
65 | targetSdkVersion getExtOrIntegerDefault("targetSdkVersion")
66 | buildConfigField "boolean", "IS_NEW_ARCHITECTURE_ENABLED", isNewArchitectureEnabled().toString()
67 |
68 | }
69 |
70 | buildFeatures {
71 | buildConfig true
72 | }
73 |
74 | buildTypes {
75 | release {
76 | minifyEnabled false
77 | }
78 | }
79 |
80 | lintOptions {
81 | disable "GradleCompatible"
82 | }
83 |
84 | compileOptions {
85 | sourceCompatibility JavaVersion.VERSION_1_8
86 | targetCompatibility JavaVersion.VERSION_1_8
87 | }
88 |
89 | sourceSets {
90 | main {
91 | if (isNewArchitectureEnabled()) {
92 | java.srcDirs += [
93 | "src/newarch",
94 | // This is needed to build Kotlin project with NewArch enabled
95 | "${project.buildDir}/generated/source/codegen/java"
96 | ]
97 | } else {
98 | java.srcDirs += ["src/oldarch"]
99 | }
100 | }
101 | }
102 | }
103 |
104 | repositories {
105 | mavenCentral()
106 | google()
107 | }
108 |
109 | def kotlin_version = getExtOrDefault("kotlinVersion")
110 |
111 | dependencies {
112 | // For < 0.71, this will be from the local maven repo
113 | // For > 0.71, this will be replaced by `com.facebook.react:react-android:$version` by react gradle plugin
114 | //noinspection GradleDynamicVersion
115 | implementation "com.facebook.react:react-native:+"
116 | implementation "org.jetbrains.kotlin:kotlin-stdlib:$kotlin_version"
117 | implementation 'com.google.code.gson:gson:2.10.1'
118 | implementation project(":mlc4j")
119 | implementation 'androidx.core:core-ktx:1.10.1'
120 | implementation 'androidx.lifecycle:lifecycle-runtime-ktx:2.6.1'
121 | implementation 'com.github.jeziellago:compose-markdown:0.5.2'
122 | implementation 'androidx.activity:activity-compose:1.7.1'
123 | implementation platform('androidx.compose:compose-bom:2022.10.00')
124 | implementation 'androidx.lifecycle:lifecycle-viewmodel-compose:2.6.1'
125 | implementation 'androidx.compose.material3:material3:1.1.0'
126 | implementation 'androidx.compose.material:material-icons-extended'
127 | implementation 'androidx.appcompat:appcompat:1.6.1'
128 | implementation 'androidx.navigation:navigation-compose:2.5.3'
129 | implementation 'com.google.code.gson:gson:2.10.1'
130 | implementation fileTree(dir: 'src/main/libs', include: ['*.aar', '*.jar'], exclude: [])
131 | testImplementation 'junit:junit:4.13.2'
132 | androidTestImplementation 'androidx.test.ext:junit:1.1.5'
133 | androidTestImplementation 'androidx.test.espresso:espresso-core:3.5.1'
134 | androidTestImplementation platform('androidx.compose:compose-bom:2022.10.00')
135 | }
136 |
137 | if (isNewArchitectureEnabled()) {
138 | react {
139 | jsRootDir = file("../src/")
140 | libraryName = "Ai"
141 | codegenJavaPackageName = "com.ai"
142 | }
143 | }
144 |
--------------------------------------------------------------------------------
/android/gradle.properties:
--------------------------------------------------------------------------------
1 | Ai_kotlinVersion=1.7.0
2 | Ai_minSdkVersion=21
3 | Ai_targetSdkVersion=31
4 | Ai_compileSdkVersion=31
5 | Ai_ndkversion=21.4.7075529
6 |
--------------------------------------------------------------------------------
/android/src/main/AndroidManifest.xml:
--------------------------------------------------------------------------------
1 |
3 |
4 |
--------------------------------------------------------------------------------
/android/src/main/AndroidManifestNew.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
6 |
7 |
10 |
11 |
12 |
--------------------------------------------------------------------------------
/android/src/main/java/com/ai/AiModule.kt:
--------------------------------------------------------------------------------
1 | package com.ai
2 |
3 | import ai.mlc.mlcllm.OpenAIProtocol
4 | import ai.mlc.mlcllm.OpenAIProtocol.ChatCompletionMessage
5 | import android.os.Environment
6 | import android.util.Log
7 | import com.facebook.react.bridge.*
8 | import com.facebook.react.bridge.ReactContext.RCTDeviceEventEmitter
9 | import com.facebook.react.module.annotations.ReactModule
10 | import com.facebook.react.turbomodule.core.interfaces.TurboModule
11 | import com.google.gson.Gson
12 | import com.google.gson.annotations.SerializedName
13 | import java.io.File
14 | import java.io.FileOutputStream
15 | import java.net.URL
16 | import java.nio.channels.Channels
17 | import java.util.UUID
18 | import kotlinx.coroutines.CoroutineScope
19 | import kotlinx.coroutines.Dispatchers
20 | import kotlinx.coroutines.launch
21 | import kotlinx.coroutines.withContext
22 |
23 | @ReactModule(name = AiModule.NAME)
24 | class AiModule(reactContext: ReactApplicationContext) :
25 | ReactContextBaseJavaModule(reactContext),
26 | TurboModule {
27 |
28 | override fun getName(): String = NAME
29 |
30 | companion object {
31 | const val NAME = "Ai"
32 |
33 | const val APP_CONFIG_FILENAME = "mlc-app-config.json"
34 | const val MODEL_CONFIG_FILENAME = "mlc-chat-config.json"
35 | const val PARAMS_CONFIG_FILENAME = "ndarray-cache.json"
36 | const val MODEL_URL_SUFFIX = "/resolve/main/"
37 | }
38 |
39 | private var appConfig = AppConfig(
40 | emptyList().toMutableList(),
41 | emptyList().toMutableList()
42 | )
43 | private val gson = Gson()
44 | private lateinit var chat: Chat
45 |
46 | private fun getAppConfig(): AppConfig {
47 | val appConfigFile = File(reactApplicationContext.applicationContext.getExternalFilesDir(""), APP_CONFIG_FILENAME)
48 |
49 | val jsonString: String = if (appConfigFile.exists()) {
50 | appConfigFile.readText()
51 | } else {
52 | reactApplicationContext.applicationContext.assets.open(APP_CONFIG_FILENAME).bufferedReader().use { it.readText() }
53 | }
54 |
55 | return gson.fromJson(jsonString, AppConfig::class.java)
56 | }
57 |
58 | private suspend fun getModelConfig(modelRecord: ModelRecord): ModelConfig {
59 | downloadModelConfig(modelRecord)
60 |
61 | val modelDirFile = File(reactApplicationContext.getExternalFilesDir(""), modelRecord.modelId)
62 | val modelConfigFile = File(modelDirFile, MODEL_CONFIG_FILENAME)
63 |
64 | val jsonString: String = if (modelConfigFile.exists()) {
65 | modelConfigFile.readText()
66 | } else {
67 | throw Error("Requested model config not found")
68 | }
69 |
70 | val modelConfig = gson.fromJson(jsonString, ModelConfig::class.java)
71 |
72 | modelConfig.apply {
73 | modelId = modelRecord.modelId
74 | modelUrl = modelRecord.modelUrl
75 | modelLib = modelRecord.modelLib
76 | }
77 |
78 | return modelConfig
79 | }
80 |
81 | @ReactMethod
82 | fun getModel(name: String, promise: Promise) {
83 | appConfig = getAppConfig()
84 |
85 | val modelConfig = appConfig.modelList.find { modelRecord -> modelRecord.modelId == name }
86 |
87 | if (modelConfig == null) {
88 | promise.reject("Model not found", "Didn't find the model")
89 | return
90 | }
91 |
92 | // Return a JSON object with details
93 | val modelConfigInstance = Arguments.createMap().apply {
94 | putString("modelId", modelConfig.modelId)
95 | putString("modelLib", modelConfig.modelLib) // Add more fields if needed
96 | }
97 |
98 | promise.resolve(modelConfigInstance)
99 | }
100 |
101 | @ReactMethod
102 | fun getModels(promise: Promise) {
103 | try {
104 | appConfig = getAppConfig()
105 | appConfig.modelLibs = emptyList().toMutableList()
106 |
107 | val modelsArray = Arguments.createArray().apply {
108 | for (modelRecord in appConfig.modelList) {
109 | pushMap(Arguments.createMap().apply { putString("model_id", modelRecord.modelId) })
110 | }
111 | }
112 |
113 | promise.resolve(modelsArray)
114 | } catch (e: Exception) {
115 | promise.reject("JSON_ERROR", "Error creating JSON array", e)
116 | }
117 | }
118 |
119 | @ReactMethod
120 | fun doGenerate(instanceId: String, messages: ReadableArray, promise: Promise) {
121 | val messageList = mutableListOf()
122 |
123 | for (i in 0 until messages.size()) {
124 | val messageMap = messages.getMap(i) // Extract ReadableMap
125 |
126 | val role = if (messageMap.getString("role") == "user") OpenAIProtocol.ChatCompletionRole.user else OpenAIProtocol.ChatCompletionRole.assistant
127 | val content = messageMap.getString("content") ?: ""
128 |
129 | messageList.add(ChatCompletionMessage(role, content))
130 | }
131 |
132 | CoroutineScope(Dispatchers.Main).launch {
133 | try {
134 | chat.generateResponse(
135 | messageList,
136 | callback = object : Chat.GenerateCallback {
137 | override fun onMessageReceived(message: String) {
138 | promise.resolve(message)
139 | }
140 | }
141 | )
142 | } catch (e: Exception) {
143 | Log.e("AI", "Error generating response", e)
144 | }
145 | }
146 | }
147 |
148 | @ReactMethod
149 | fun doStream(instanceId: String, messages: ReadableArray, promise: Promise) {
150 | val messageList = mutableListOf()
151 |
152 | for (i in 0 until messages.size()) {
153 | val messageMap = messages.getMap(i) // Extract ReadableMap
154 |
155 | val role = if (messageMap.getString("role") == "user") OpenAIProtocol.ChatCompletionRole.user else OpenAIProtocol.ChatCompletionRole.assistant
156 | val content = messageMap.getString("content") ?: ""
157 |
158 | messageList.add(ChatCompletionMessage(role, content))
159 | }
160 | CoroutineScope(Dispatchers.Main).launch {
161 | chat.streamResponse(
162 | messageList,
163 | callback = object : Chat.StreamCallback {
164 | override fun onUpdate(message: String) {
165 | val event: WritableMap = Arguments.createMap().apply {
166 | putString("content", message)
167 | }
168 | sendEvent("onChatUpdate", event)
169 | }
170 |
171 | override fun onFinished(message: String) {
172 | val event: WritableMap = Arguments.createMap().apply {
173 | putString("content", message)
174 | }
175 | sendEvent("onChatComplete", event)
176 | }
177 | }
178 | )
179 | }
180 | promise.resolve(null)
181 | }
182 |
183 | @ReactMethod
184 | fun downloadModel(instanceId: String, promise: Promise) {
185 | CoroutineScope(Dispatchers.IO).launch {
186 | try {
187 | val appConfig = getAppConfig()
188 | val modelRecord = appConfig.modelList.find { modelRecord -> modelRecord.modelId == instanceId }
189 | if (modelRecord == null) {
190 | throw Error("There's no record for requested model")
191 | }
192 |
193 | val modelConfig = getModelConfig(modelRecord)
194 |
195 | val modelDir = File(reactApplicationContext.getExternalFilesDir(""), modelConfig.modelId)
196 |
197 | val modelState = ModelState(modelConfig, modelDir)
198 |
199 | modelState.initialize()
200 |
201 | sendEvent("onDownloadStart", null)
202 |
203 | CoroutineScope(Dispatchers.IO).launch {
204 | modelState.progress.collect { newValue ->
205 | val event: WritableMap = Arguments.createMap().apply {
206 | putDouble("percentage", (newValue.toDouble() / modelState.total.intValue) * 100)
207 | }
208 | sendEvent("onDownloadProgress", event)
209 | }
210 | }
211 |
212 | modelState.download()
213 |
214 | sendEvent("onDownloadComplete", null)
215 |
216 | withContext(Dispatchers.Main) { promise.resolve("Model downloaded: $instanceId") }
217 | } catch (e: Exception) {
218 | sendEvent("onDownloadError", e.message ?: "Unknown error")
219 | withContext(Dispatchers.Main) { promise.reject("MODEL_ERROR", "Error downloading model", e) }
220 | }
221 | }
222 | }
223 |
224 | private fun sendEvent(eventName: String, data: Any?) {
225 | reactApplicationContext.getJSModule(RCTDeviceEventEmitter::class.java)?.emit(eventName, data)
226 | }
227 |
228 | @ReactMethod
229 | fun prepareModel(instanceId: String, promise: Promise) {
230 | CoroutineScope(Dispatchers.IO).launch {
231 | try {
232 | val appConfig = getAppConfig()
233 |
234 | val modelRecord = appConfig.modelList.find { modelRecord -> modelRecord.modelId == instanceId }
235 |
236 | if (modelRecord == null) {
237 | throw Error("There's no record for requested model")
238 | }
239 | val modelConfig = getModelConfig(modelRecord)
240 |
241 | val modelDir = File(reactApplicationContext.getExternalFilesDir(""), modelConfig.modelId)
242 |
243 | val modelState = ModelState(modelConfig, modelDir)
244 | modelState.initialize()
245 | modelState.download()
246 |
247 | chat = Chat(modelConfig, modelDir)
248 |
249 | withContext(Dispatchers.Main) { promise.resolve("Model prepared: $instanceId") }
250 | } catch (e: Exception) {
251 | withContext(Dispatchers.Main) { promise.reject("MODEL_ERROR", "Error preparing model", e) }
252 | }
253 | }
254 | }
255 |
256 | private suspend fun downloadModelConfig(modelRecord: ModelRecord) {
257 | withContext(Dispatchers.IO) {
258 | // Don't download if config is downloaded already
259 | val modelFile = File(reactApplicationContext.getExternalFilesDir(""), modelRecord.modelId)
260 | if (modelFile.exists()) {
261 | return@withContext
262 | }
263 |
264 | // Prepare temp file for streaming
265 | val url = URL("${modelRecord.modelUrl}${MODEL_URL_SUFFIX}$MODEL_CONFIG_FILENAME")
266 | val tempId = UUID.randomUUID().toString()
267 | val tempFile = File(
268 | reactApplicationContext.getExternalFilesDir(Environment.DIRECTORY_DOWNLOADS),
269 | tempId
270 | )
271 |
272 | // Download
273 | url.openStream().use {
274 | Channels.newChannel(it).use { src ->
275 | FileOutputStream(tempFile).use { fileOutputStream ->
276 | fileOutputStream.channel.transferFrom(src, 0, Long.MAX_VALUE)
277 | }
278 | }
279 | }
280 | require(tempFile.exists())
281 |
282 | // Create object form config
283 | val modelConfigString = tempFile.readText()
284 | val modelConfig = gson.fromJson(modelConfigString, ModelConfig::class.java).apply {
285 | modelId = modelRecord.modelId
286 | modelLib = modelRecord.modelLib
287 | estimatedVramBytes = modelRecord.estimatedVramBytes
288 | }
289 |
290 | // Copy to config location and remove temp file
291 | val modelDirFile = File(reactApplicationContext.getExternalFilesDir(""), modelConfig.modelId)
292 | val modelConfigFile = File(modelDirFile, MODEL_CONFIG_FILENAME)
293 | tempFile.copyTo(modelConfigFile, overwrite = true)
294 | tempFile.delete()
295 |
296 | return@withContext
297 | }
298 | }
299 | }
300 |
301 | enum class ModelChatState {
302 | Generating,
303 | Resetting,
304 | Reloading,
305 | Terminating,
306 | Ready,
307 | Failed
308 | }
309 |
310 | data class MessageData(val role: String, val text: String, val id: UUID = UUID.randomUUID())
311 |
312 | data class ModelConfig(
313 | @SerializedName("model_lib") var modelLib: String,
314 | @SerializedName("model_id") var modelId: String,
315 | @SerializedName("model_url") var modelUrl: String,
316 | @SerializedName("estimated_vram_bytes") var estimatedVramBytes: Long?,
317 | @SerializedName("tokenizer_files") val tokenizerFiles: List,
318 | @SerializedName("context_window_size") val contextWindowSize: Int,
319 | @SerializedName("prefill_chunk_size") val prefillChunkSize: Int
320 | )
321 |
322 | data class AppConfig(@SerializedName("model_libs") var modelLibs: MutableList, @SerializedName("model_list") val modelList: MutableList)
323 |
324 | data class ModelRecord(
325 | @SerializedName("model_url") val modelUrl: String,
326 | @SerializedName("model_id") val modelId: String,
327 | @SerializedName("estimated_vram_bytes") val estimatedVramBytes: Long?,
328 | @SerializedName("model_lib") val modelLib: String
329 | )
330 |
331 | data class DownloadTask(val url: URL, val file: File)
332 |
333 | data class ParamsConfig(@SerializedName("records") val paramsRecords: List)
334 |
335 | data class ParamsRecord(@SerializedName("dataPath") val dataPath: String)
336 |
--------------------------------------------------------------------------------
/android/src/main/java/com/ai/AiPackage.kt:
--------------------------------------------------------------------------------
1 | package com.ai
2 |
3 | import com.facebook.react.TurboReactPackage
4 | import com.facebook.react.bridge.NativeModule
5 | import com.facebook.react.bridge.ReactApplicationContext
6 | import com.facebook.react.module.model.ReactModuleInfo
7 | import com.facebook.react.module.model.ReactModuleInfoProvider
8 | import java.util.HashMap
9 |
10 | class AiPackage : TurboReactPackage() {
11 | override fun getModule(name: String, reactContext: ReactApplicationContext): NativeModule? =
12 | if (name == AiModule.NAME) {
13 | AiModule(reactContext)
14 | } else {
15 | null
16 | }
17 |
18 | override fun getReactModuleInfoProvider(): ReactModuleInfoProvider =
19 | ReactModuleInfoProvider {
20 | val moduleInfos: MutableMap = HashMap()
21 | val isTurboModule: Boolean = BuildConfig.IS_NEW_ARCHITECTURE_ENABLED
22 | moduleInfos[AiModule.NAME] = ReactModuleInfo(
23 | AiModule.NAME,
24 | AiModule.NAME,
25 | // canOverrideExistingModule
26 | false,
27 | // needsEagerInit
28 | false,
29 | // hasConstants
30 | true,
31 | // isCxxModule
32 | false,
33 | // isTurboModule
34 | isTurboModule
35 | )
36 |
37 | moduleInfos
38 | }
39 | }
40 |
--------------------------------------------------------------------------------
/android/src/main/java/com/ai/ChatState.kt:
--------------------------------------------------------------------------------
1 | package com.ai
2 |
3 | import ai.mlc.mlcllm.MLCEngine
4 | import ai.mlc.mlcllm.OpenAIProtocol
5 | import ai.mlc.mlcllm.OpenAIProtocol.ChatCompletionMessage
6 | import java.io.File
7 | import java.util.concurrent.Executors
8 | import kotlinx.coroutines.CoroutineScope
9 | import kotlinx.coroutines.Dispatchers
10 | import kotlinx.coroutines.Job
11 | import kotlinx.coroutines.channels.toList
12 | import kotlinx.coroutines.launch
13 |
14 | class Chat(modelConfig: ModelConfig, modelDir: File) {
15 | private val engine = MLCEngine()
16 | private val executorService = Executors.newSingleThreadExecutor()
17 | private val viewModelScope = CoroutineScope(Dispatchers.Main + Job())
18 |
19 | init {
20 | engine.unload()
21 | engine.reload(modelDir.path, modelConfig.modelLib)
22 | }
23 |
24 | fun generateResponse(messages: MutableList, callback: GenerateCallback) {
25 | executorService.submit {
26 | viewModelScope.launch {
27 | val chatResponse = engine.chat.completions.create(messages = messages)
28 | val response = chatResponse.toList().joinToString("") { it.choices.joinToString("") { it.delta.content?.text ?: "" } }
29 | callback.onMessageReceived(response)
30 | }
31 | }
32 | }
33 |
34 | fun streamResponse(messages: MutableList, callback: StreamCallback) {
35 | executorService.submit {
36 | viewModelScope.launch {
37 | val chatResponse = engine.chat.completions.create(messages = messages, stream_options = OpenAIProtocol.StreamOptions(include_usage = true))
38 |
39 | var finishReasonLength = false
40 | var streamingText = ""
41 |
42 | for (res in chatResponse) {
43 | for (choice in res.choices) {
44 | choice.delta.content?.let { content ->
45 | streamingText = content.asText()
46 | }
47 | choice.finish_reason?.let { finishReason ->
48 | if (finishReason == "length") {
49 | finishReasonLength = true
50 | }
51 | }
52 | }
53 |
54 | callback.onUpdate(streamingText)
55 | if (finishReasonLength) {
56 | streamingText = " [output truncated due to context length limit...]"
57 | callback.onUpdate(streamingText)
58 | }
59 | }
60 | callback.onFinished(streamingText)
61 | }
62 | }
63 | }
64 |
65 | interface GenerateCallback {
66 | fun onMessageReceived(message: String)
67 | }
68 |
69 | interface StreamCallback {
70 | fun onUpdate(message: String)
71 | fun onFinished(message: String)
72 | }
73 | }
74 |
--------------------------------------------------------------------------------
/android/src/main/java/com/ai/ModelState.kt:
--------------------------------------------------------------------------------
1 | package com.ai
2 |
3 | import androidx.compose.runtime.mutableIntStateOf
4 | import com.ai.AiModule.Companion.MODEL_CONFIG_FILENAME
5 | import com.ai.AiModule.Companion.MODEL_URL_SUFFIX
6 | import com.ai.AiModule.Companion.PARAMS_CONFIG_FILENAME
7 | import com.google.gson.Gson
8 | import java.io.File
9 | import java.io.FileOutputStream
10 | import java.net.URL
11 | import java.nio.channels.Channels
12 | import java.util.UUID
13 | import kotlinx.coroutines.Dispatchers
14 | import kotlinx.coroutines.flow.MutableStateFlow
15 | import kotlinx.coroutines.withContext
16 |
17 | class ModelState(private val modelConfig: ModelConfig, private val modelDir: File) {
18 | private var paramsConfig = ParamsConfig(emptyList())
19 | val progress = MutableStateFlow(0)
20 | val total = mutableIntStateOf(1)
21 | val id: UUID = UUID.randomUUID()
22 | private val remainingTasks = emptySet().toMutableSet()
23 | private val downloadingTasks = emptySet().toMutableSet()
24 | private val maxDownloadTasks = 3
25 | private val gson = Gson()
26 |
27 | suspend fun initialize() {
28 | val paramsConfigFile = File(modelDir, PARAMS_CONFIG_FILENAME)
29 | if (!paramsConfigFile.exists()) {
30 | downloadParamsConfig()
31 | }
32 |
33 | loadParamsConfig()
34 | indexModel()
35 | }
36 |
37 | private fun loadParamsConfig() {
38 | val paramsConfigFile = File(modelDir, PARAMS_CONFIG_FILENAME)
39 | require(paramsConfigFile.exists())
40 | val jsonString = paramsConfigFile.readText()
41 | paramsConfig = gson.fromJson(jsonString, ParamsConfig::class.java)
42 | }
43 |
44 | private suspend fun downloadParamsConfig() {
45 | withContext(Dispatchers.IO) {
46 | val url = URL("${modelConfig.modelUrl}$MODEL_URL_SUFFIX$PARAMS_CONFIG_FILENAME")
47 | val tempId = UUID.randomUUID().toString()
48 | val tempFile = File(modelDir, tempId)
49 | url.openStream().use {
50 | Channels.newChannel(it).use { src ->
51 | FileOutputStream(tempFile).use { fileOutputStream ->
52 | fileOutputStream.channel.transferFrom(src, 0, Long.MAX_VALUE)
53 | }
54 | }
55 | }
56 | require(tempFile.exists())
57 | val paramsConfigFile = File(modelDir, PARAMS_CONFIG_FILENAME)
58 | tempFile.renameTo(paramsConfigFile)
59 | require(paramsConfigFile.exists())
60 | }
61 | }
62 |
63 | suspend fun download() {
64 | while (remainingTasks.isNotEmpty() && downloadingTasks.size < maxDownloadTasks) {
65 | val downloadTask = remainingTasks.first()
66 | remainingTasks.remove(downloadTask)
67 | handleNewDownload(downloadTask)
68 | }
69 | }
70 |
71 | private suspend fun handleNewDownload(downloadTask: DownloadTask) {
72 | require(!downloadingTasks.contains(downloadTask))
73 | downloadingTasks.add(downloadTask)
74 |
75 | withContext(Dispatchers.IO) {
76 | val tempId = UUID.randomUUID().toString()
77 | val tempFile = File(modelDir, tempId)
78 |
79 | downloadTask.url.openStream().use {
80 | Channels.newChannel(it).use { src ->
81 | FileOutputStream(tempFile).use { fileOutputStream ->
82 | fileOutputStream.channel.transferFrom(src, 0, Long.MAX_VALUE)
83 | }
84 | }
85 | }
86 | require(tempFile.exists())
87 | tempFile.renameTo(downloadTask.file)
88 | require(downloadTask.file.exists())
89 |
90 | handleFinishDownload(downloadTask)
91 | }
92 | }
93 |
94 | private fun handleFinishDownload(downloadTask: DownloadTask) {
95 | remainingTasks.remove(downloadTask)
96 | downloadingTasks.remove(downloadTask)
97 | ++progress.value
98 | }
99 |
100 | private fun clear() {
101 | val files = modelDir.listFiles { dir, name ->
102 | !(dir == modelDir && name == MODEL_CONFIG_FILENAME)
103 | }
104 | require(files != null)
105 | for (file in files) {
106 | file.deleteRecursively()
107 | require(!file.exists())
108 | }
109 | val modelConfigFile = File(modelDir, MODEL_CONFIG_FILENAME)
110 | require(modelConfigFile.exists())
111 | indexModel()
112 | }
113 |
114 | private fun indexModel() {
115 | progress.value = 0
116 | total.intValue = modelConfig.tokenizerFiles.size + paramsConfig.paramsRecords.size
117 |
118 | // Adding Tokenizer to download tasks
119 | for (tokenizerFilename in modelConfig.tokenizerFiles) {
120 | val file = File(modelDir, tokenizerFilename)
121 | if (file.exists()) {
122 | ++progress.value
123 | } else {
124 | remainingTasks.add(
125 | DownloadTask(
126 | URL("${modelConfig.modelUrl}$MODEL_URL_SUFFIX$tokenizerFilename"),
127 | file
128 | )
129 | )
130 | }
131 | }
132 |
133 | // Adding params to download tasks
134 | for (paramsRecord in paramsConfig.paramsRecords) {
135 | val file = File(modelDir, paramsRecord.dataPath)
136 | if (file.exists()) {
137 | ++progress.value
138 | } else {
139 | remainingTasks.add(
140 | DownloadTask(
141 | URL("${modelConfig.modelUrl}$MODEL_URL_SUFFIX${paramsRecord.dataPath}"),
142 | file
143 | )
144 | )
145 | }
146 | }
147 | }
148 | }
149 |
--------------------------------------------------------------------------------
/android/src/newarch/AiSpec.kt:
--------------------------------------------------------------------------------
1 | package com.ai
2 |
3 | import com.facebook.react.bridge.ReactApplicationContext
4 |
5 | abstract class AiSpec internal constructor(context: ReactApplicationContext) : NativeAiSpec(context)
6 |
--------------------------------------------------------------------------------
/android/src/oldarch/AiSpec.kt:
--------------------------------------------------------------------------------
1 | package com.ai
2 |
3 | import com.facebook.react.bridge.Promise
4 | import com.facebook.react.bridge.ReactApplicationContext
5 | import com.facebook.react.bridge.ReactContextBaseJavaModule
6 |
7 | abstract class AiSpec internal constructor(context: ReactApplicationContext) : ReactContextBaseJavaModule(context) {
8 |
9 | abstract fun getModel(name: String, promise: Promise)
10 | abstract fun getModels(promise: Promise)
11 | abstract fun doGenerate(instanceId: String, text: String, promise: Promise)
12 | abstract fun doStream(instanceId: String, text: String, promise: Promise)
13 | }
14 |
--------------------------------------------------------------------------------
/babel.config.js:
--------------------------------------------------------------------------------
1 | module.exports = {
2 | presets: ['module:@react-native/babel-preset'],
3 | };
4 |
--------------------------------------------------------------------------------
/example/.watchmanconfig:
--------------------------------------------------------------------------------
1 | {}
2 |
--------------------------------------------------------------------------------
/example/android/app/build.gradle:
--------------------------------------------------------------------------------
1 | plugins {
2 | id("com.android.application")
3 | id("org.jetbrains.kotlin.android")
4 | id("com.facebook.react")
5 | id("org.jetbrains.kotlin.plugin.serialization")
6 | }
7 |
8 | /**
9 | * This is the configuration block to customize your React Native Android app.
10 | * By default you don't need to apply any configuration, just uncomment the lines you need.
11 | */
12 | react {
13 | /* Folders */
14 | // The root of your project, i.e. where "package.json" lives. Default is '..'
15 | // root = file("../")
16 | // The folder where the react-native NPM package is. Default is ../node_modules/react-native
17 | // reactNativeDir = file("../node_modules/react-native")
18 | // The folder where the react-native Codegen package is. Default is ../node_modules/@react-native/codegen
19 | // codegenDir = file("../node_modules/@react-native/codegen")
20 | // The cli.js file which is the React Native CLI entrypoint. Default is ../node_modules/react-native/cli.js
21 | // cliFile = file("../node_modules/react-native/cli.js")
22 |
23 | /* Variants */
24 | // The list of variants to that are debuggable. For those we're going to
25 | // skip the bundling of the JS bundle and the assets. By default is just 'debug'.
26 | // If you add flavors like lite, prod, etc. you'll have to list your debuggableVariants.
27 | // debuggableVariants = ["liteDebug", "prodDebug"]
28 |
29 | /* Bundling */
30 | // A list containing the node command and its flags. Default is just 'node'.
31 | // nodeExecutableAndArgs = ["node"]
32 | //
33 | // The command to run when bundling. By default is 'bundle'
34 | // bundleCommand = "ram-bundle"
35 | //
36 | // The path to the CLI configuration file. Default is empty.
37 | // bundleConfig = file(../rn-cli.config.js)
38 | //
39 | // The name of the generated asset file containing your JS bundle
40 | // bundleAssetName = "MyApplication.android.bundle"
41 | //
42 | // The entry file for bundle generation. Default is 'index.android.js' or 'index.js'
43 | // entryFile = file("../js/MyApplication.android.js")
44 | //
45 | // A list of extra flags to pass to the 'bundle' commands.
46 | // See https://github.com/react-native-community/cli/blob/main/docs/commands.md#bundle
47 | // extraPackagerArgs = []
48 |
49 | /* Hermes Commands */
50 | // The hermes compiler command to run. By default it is 'hermesc'
51 | // hermesCommand = "$rootDir/my-custom-hermesc/bin/hermesc"
52 | //
53 | // The list of flags to pass to the Hermes compiler. By default is "-O", "-output-source-map"
54 | // hermesFlags = ["-O", "-output-source-map"]
55 | }
56 |
57 | /**
58 | * Set this to true to Run Proguard on Release builds to minify the Java bytecode.
59 | */
60 | def enableProguardInReleaseBuilds = false
61 |
62 | /**
63 | * The preferred build flavor of JavaScriptCore (JSC)
64 | *
65 | * For example, to use the international variant, you can use:
66 | * `def jscFlavor = 'org.webkit:android-jsc-intl:+'`
67 | *
68 | * The international variant includes ICU i18n library and necessary data
69 | * allowing to use e.g. `Date.toLocaleString` and `String.localeCompare` that
70 | * give correct results when using with locales other than en-US. Note that
71 | * this variant is about 6MiB larger per architecture than default.
72 | */
73 | def jscFlavor = 'org.webkit:android-jsc:+'
74 |
75 | android {
76 | ndkVersion rootProject.ext.ndkVersion
77 | buildToolsVersion rootProject.ext.buildToolsVersion
78 | compileSdk rootProject.ext.compileSdkVersion
79 |
80 | namespace "com.aiexample"
81 | defaultConfig {
82 | applicationId "com.aiexample"
83 | minSdkVersion rootProject.ext.minSdkVersion
84 | targetSdkVersion rootProject.ext.targetSdkVersion
85 | versionCode 1
86 | versionName "1.0"
87 | }
88 | signingConfigs {
89 | debug {
90 | storeFile file('debug.keystore')
91 | storePassword 'android'
92 | keyAlias 'androiddebugkey'
93 | keyPassword 'android'
94 | }
95 | }
96 | buildTypes {
97 | debug {
98 | signingConfig signingConfigs.debug
99 | }
100 | release {
101 | // Caution! In production, you need to generate your own keystore file.
102 | // see https://reactnative.dev/docs/signed-apk-android.
103 | signingConfig signingConfigs.debug
104 | minifyEnabled enableProguardInReleaseBuilds
105 | proguardFiles getDefaultProguardFile("proguard-android.txt"), "proguard-rules.pro"
106 | }
107 | }
108 | }
109 |
110 | dependencies {
111 | // The version of react-native is set by the React Native Gradle Plugin
112 | implementation("com.facebook.react:react-android")
113 | implementation("org.jetbrains.kotlinx:kotlinx-serialization-json:1.6.3")
114 |
115 | if (hermesEnabled.toBoolean()) {
116 | implementation("com.facebook.react:hermes-android")
117 | } else {
118 | implementation jscFlavor
119 | }
120 | }
121 |
122 | apply from: file("../../node_modules/@react-native-community/cli-platform-android/native_modules.gradle"); applyNativeModulesAppBuildGradle(project)
123 |
--------------------------------------------------------------------------------
/example/android/app/debug.keystore:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/callstackincubator/ai/fdfdfafd834f42124650b499df05fcb859b51792/example/android/app/debug.keystore
--------------------------------------------------------------------------------
/example/android/app/proguard-rules.pro:
--------------------------------------------------------------------------------
1 | # Add project specific ProGuard rules here.
2 | # By default, the flags in this file are appended to flags specified
3 | # in /usr/local/Cellar/android-sdk/24.3.3/tools/proguard/proguard-android.txt
4 | # You can edit the include path and order by changing the proguardFiles
5 | # directive in build.gradle.
6 | #
7 | # For more details, see
8 | # http://developer.android.com/guide/developing/tools/proguard.html
9 |
10 | # Add any project specific keep options here:
11 |
--------------------------------------------------------------------------------
/example/android/app/src/debug/AndroidManifest.xml:
--------------------------------------------------------------------------------
1 |
2 |
4 |
5 |
9 |
10 |
--------------------------------------------------------------------------------
/example/android/app/src/main/AndroidManifest.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 |
12 |
19 |
20 |
21 |
22 |
23 |
24 |
25 |
26 |
--------------------------------------------------------------------------------
/example/android/app/src/main/java/com/aiexample/MainActivity.kt:
--------------------------------------------------------------------------------
1 | package com.aiexample
2 |
3 | import com.facebook.react.ReactActivity
4 | import com.facebook.react.ReactActivityDelegate
5 | import com.facebook.react.defaults.DefaultNewArchitectureEntryPoint.fabricEnabled
6 | import com.facebook.react.defaults.DefaultReactActivityDelegate
7 |
8 | class MainActivity : ReactActivity() {
9 | /**
10 | * Returns the name of the main component registered from JavaScript. This is used to schedule
11 | * rendering of the component.
12 | */
13 | override fun getMainComponentName(): String = "AiExample"
14 |
15 | /**
16 | * Returns the instance of the [ReactActivityDelegate]. We use [DefaultReactActivityDelegate]
17 | * which allows you to enable New Architecture with a single boolean flags [fabricEnabled]
18 | */
19 | override fun createReactActivityDelegate(): ReactActivityDelegate = DefaultReactActivityDelegate(this, mainComponentName, fabricEnabled)
20 | }
21 |
--------------------------------------------------------------------------------
/example/android/app/src/main/java/com/aiexample/MainApplication.kt:
--------------------------------------------------------------------------------
1 | package com.aiexample
2 |
3 | import android.app.Application
4 | import com.facebook.react.PackageList
5 | import com.facebook.react.ReactApplication
6 | import com.facebook.react.ReactHost
7 | import com.facebook.react.ReactNativeHost
8 | import com.facebook.react.ReactPackage
9 | import com.facebook.react.defaults.DefaultNewArchitectureEntryPoint.load
10 | import com.facebook.react.defaults.DefaultReactHost.getDefaultReactHost
11 | import com.facebook.react.defaults.DefaultReactNativeHost
12 | import com.facebook.soloader.SoLoader
13 |
14 | class MainApplication :
15 | Application(),
16 | ReactApplication {
17 | override val reactNativeHost: ReactNativeHost =
18 | object : DefaultReactNativeHost(this) {
19 | override fun getPackages(): List =
20 | PackageList(this).packages.apply {
21 | // Packages that cannot be autolinked yet can be added manually here, for example:
22 | // add(MyReactNativePackage())
23 | }
24 |
25 | override fun getJSMainModuleName(): String = "index"
26 |
27 | override fun getUseDeveloperSupport(): Boolean = BuildConfig.DEBUG
28 |
29 | override val isNewArchEnabled: Boolean = BuildConfig.IS_NEW_ARCHITECTURE_ENABLED
30 | override val isHermesEnabled: Boolean = BuildConfig.IS_HERMES_ENABLED
31 | }
32 |
33 | override val reactHost: ReactHost
34 | get() = getDefaultReactHost(applicationContext, reactNativeHost)
35 |
36 | override fun onCreate() {
37 | super.onCreate()
38 | SoLoader.init(this, false)
39 | if (BuildConfig.IS_NEW_ARCHITECTURE_ENABLED) {
40 | // If you opted-in for the New Architecture, we load the native entry point for this app.
41 | load()
42 | }
43 | }
44 | }
45 |
--------------------------------------------------------------------------------
/example/android/app/src/main/res/drawable/rn_edit_text_material.xml:
--------------------------------------------------------------------------------
1 |
2 |
16 |
22 |
23 |
24 |
33 |