@ariacompute/engine-rn 1.5.1 → 1.6.1

package/README.md

@@ -7,3 +7,17 @@ console.log(await eng.complete([{ role: 'user', content: 'hi' }]));
 ```
 
 npm publish name: `@ariacompute/engine-rn`. Device-farm CI: `.github/workflows/bindings-mobile.yml`.
+
+## Auto-download by model name
+
+`AriaEngine.open(modelRef, { token, site })` accepts a local bundle path **or**
+an Aria model name (e.g. `gemma-4-e2b-it_q4`). A value containing `/` or already
+on disk is loaded directly; otherwise the SDK downloads it from the Dashboard
+private source (requires `token`; `site` defaults to `https://ariacompute.com`)
+into `~/.ariacompute/models/{model}` and then loads it. A valid cached bundle is
+reused without re-downloading. The zip streaming + extraction is delegated to the
+native `AriaEngineModule` bridge.
+
+```js
+const eng = await AriaEngine.open("gemma-4-e2b-it_q4", { token: "API_TOKEN" });
+```

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ariacompute/engine-rn",
-  "version": "1.5.1",
+  "version": "1.6.1",
   "description": "React Native binding for Aria Engine",
   "main": "src/index.js",
   "license": "MIT",

package/src/index.js

@@ -1,20 +1,98 @@
 /**
  * React Native surface — native module bridges to libaria_ffi (Turbo Module / JSI).
  * JS API mirrors @ariacompute/engine-ts for docs samples.
+ *
+ * Auto-download: `AriaEngine.open(modelRef, { token, site })` resolves a model
+ * name via the Dashboard private source (requires `token`), downloads the zip,
+ * extracts it through the native `AriaEngineModule` (which bridges unzip and
+ * writes to `~/.ariacompute/models/{model}`), then loads via the native init.
  */
+import { NativeModules, Platform } from 'react-native';
+
+const AriaEngineModule = NativeModules.AriaEngine;
+
+const DEFAULT_SITE = 'https://ariacompute.com';
+
+function parseBundleName(model) {
+  if (!model || model.includes('/') || model.includes('\\')) {
+    throw new Error(`invalid model name: ${model}`);
+  }
+  const idx = model.lastIndexOf('_q');
+  if (idx !== -1) {
+    const slug = model.slice(0, idx);
+    const suffix = model.slice(idx + 2);
+    const quant =
+      suffix === '4'
+        ? 'int4'
+        : suffix === '8'
+        ? 'int8'
+        : suffix === '326' || suffix === '3.26'
+        ? 'int326'
+        : (() => {
+            throw new Error(`unknown quant suffix _q${suffix}`);
+          })();
+    if (!slug) throw new Error(`invalid model name: ${model}`);
+    return { slug, quant };
+  }
+  return { slug: model, quant: 'int4' };
+}
+
+function ariaHome() {
+  if (process.env.ARIA_COMPUTE_HOME) return process.env.ARIA_COMPUTE_HOME;
+  return `${require('os').homedir()}/.ariacompute`;
+}
+
+function cacheDir(model) {
+  return `${ariaHome()}/models/${model}`;
+}
+
+function isLocalRef(ref) {
+  return ref.includes('/') || ref.includes('\\') || require('fs').existsSync(ref);
+}
+
+async function downloadModel(model, token, site = DEFAULT_SITE) {
+  if (!token) throw new Error('api token is required to download a model');
+  const { slug, quant } = parseBundleName(model);
+  const cache = cacheDir(model);
+  if (require('fs').existsSync(cache) && (await AriaEngineModule?.isValidBundle?.(cache))) {
+    return cache;
+  }
+  if (!AriaEngineModule?.downloadModel) {
+    throw new Error('native AriaEngineModule.downloadModel bridge is not linked');
+  }
+  const metaUrl = `${site.replace(/\/$/, '')}/api/dashboard/models/${encodeURIComponent(
+    slug,
+  )}/download?quant=${encodeURIComponent(quant)}&sdk=v1.0&format=json`;
+  const metaResp = await fetch(metaUrl, { headers: { Authorization: `Bearer ${token}` } });
+  if (!metaResp.ok) throw new Error(`dashboard request failed: ${metaResp.status}`);
+  const meta = await metaResp.json();
+  if (!meta.url) throw new Error('dashboard meta returned empty url');
+  // Delegates streaming + zip extraction (and bundle validation) to native.
+  return AriaEngineModule.downloadModel(model, meta.url, token, cache);
+}
+
 export class AriaEngine {
   constructor(bundlePath) {
     this.bundlePath = bundlePath;
     // NativeModules.AriaEngine.init(bundlePath) when linked.
   }
 
+  static async open(modelRef, opts = {}) {
+    if (isLocalRef(modelRef)) return new AriaEngine(modelRef);
+    if (!opts.token) {
+      throw new Error(`model name '${modelRef}' requires an api token to download`);
+    }
+    const bundle = await downloadModel(modelRef, opts.token, opts.site ?? DEFAULT_SITE);
+    return new AriaEngine(bundle);
+  }
+
   async complete(messages, options = { max_tokens: 16 }, tools = []) {
     // return NativeModules.AriaEngine.complete(...)
     return {
       success: true,
-      response: "",
+      response: '',
       function_calls: [],
-      note: "Link native module to libaria_ffi",
+      note: 'Link native module to libaria_ffi',
     };
   }