@ariacompute/engine-rn 1.22.0 → 2.0.0

package/README.md

@@ -1,32 +1,8 @@
 # @ariacompute/engine-rn
 
 ```js
-import { AriaEngine } from '@ariacompute/engine-rn';
-const eng = new AriaEngine('/path/to/bundle');
-console.log(await eng.complete([{ role: 'user', content: 'hi' }]));
+const { AriaEngine } = require('@ariacompute/engine-rn');
+const eng = new AriaEngine(checkpoint, 'decoder');
 ```
 
-npm publish name: `@ariacompute/engine-rn`. Device-farm CI: `.github/workflows/bindings-mobile.yml`.
-
-`AriaEngine.open` installs `libaria_ffi` into `~/.ariacompute/lib/` when it is not already on `ARIA_FFI_LIB` or in that cache (same Releases asset as `aria-engine upgrade`).
-
-## 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 regional
-public hub (same as `aria-engine download`: `.com` → Hugging Face, `.cn` →
-ModelScope; `site` defaults to `https://ariacompute.com`) into
-`~/.ariacompute/models/{model}` and then loads it. Dashboard is not used. A
-Dashboard `sk-` / `sk-bf-` token is ignored for hub auth. Token is optional for
-public models. Gated files: pass `hfToken` / `modelscopeApiToken` (same as
-`aria-engine setup`); if omitted, reads `~/.ariacompute/engine.yml`. Instance
-`setup` is in-memory only (does not write that file). A valid cached bundle is
-reused without re-downloading.
-
-```js
-const eng = await AriaEngine.open("gemma-4-e2b-it_q4");
-const gated = new AriaEngine();
-gated.setup({ hf_token: "hf_..." });
-await gated.open("gemma-4-e2b-it_q4");
-```
+Installs `libaria-engine_ffi` into `~/.ariacompute/lib/` (same Releases asset as `aria-engine upgrade`).

package/package.json

@@ -1,13 +1,11 @@
 {
   "name": "@ariacompute/engine-rn",
-  "version": "1.22.0",
-  "description": "React Native binding for Aria Engine",
+  "version": "2.0.0",
+  "description": "React Native binding for AFM-D System One",
   "main": "src/index.js",
   "license": "MIT",
-  "peerDependencies": {
-    "react-native": ">=0.72.0"
-  },
-  "scripts": {
-    "test": "node --test test/setup.test.cjs"
-  }
+  "files": [
+    "src",
+    "README.md"
+  ]
 }

package/src/index.js

@@ -1,487 +1,24 @@
+'use strict';
+
 /**
- * React Native surface — native module bridges to libaria-engine_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 regional public hub (Hugging Face / ModelScope — same as
- * `aria-engine download`; Dashboard is not used), writes files to
- * `~/.ariacompute/models/{model}`, then loads via native init.
+ * React Native AFM-D engine binding.
+ * Native module should link libaria-engine_ffi and expose systemone(JSON).
+ * This JS layer mirrors the TypeScript API shape for host tests.
  */
-import { NativeModules } from 'react-native';
-import {
-  defaultSetupConfig,
-  applySetup,
-} from './setup.js';
-
-const AriaEngineModule = NativeModules.AriaEngine;
-
-const DEFAULT_SITE = 'https://ariacompute.com';
-const DEFAULT_SDK = 'v1.0';
-const HUB_REQUIRED = ['config.json', 'weight.bin'];
-const HUB_OPTIONAL = [
-  'tokenizer.json',
-  'tokenizer.model',
-  'tokenizer_config.json',
-  'special_tokens_map.json',
-  'vocab.json',
-  'merges.txt',
-];
 
-function parseBundleName(model) {
-  if (!model || model.includes('/') || model.includes('\\')) {
-    throw new Error(`invalid model name: ${model}`);
+class AriaEngine {
+  constructor(checkpoint, track = 'encoder') {
+    this.checkpoint = checkpoint;
+    this.track = track;
   }
-  const idx = model.lastIndexOf('_q');
-  if (idx !== -1) {
-    const slug = model.slice(0, idx);
-    let suffix = model.slice(idx + 2);
-    if (suffix.endsWith('_channel') || suffix.endsWith('_group')) {
-      suffix = suffix.slice(0, suffix.lastIndexOf('_'));
-    }
-    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);
-}
-
-function preferredPublicHub(site) {
-  if (site && String(site).toLowerCase().includes('ariacompute.cn')) return 'modelscope';
-  return 'huggingface';
-}
-
-function hubBearer(token) {
-  const t = (token || '').trim();
-  if (!t) return undefined;
-  const low = t.toLowerCase();
-  if (low.startsWith('sk-')) return undefined;
-  return t;
-}
-
-function unquoteYaml(v) {
-  const t = (v || '').trim();
-  if (t.length >= 2 && ((t.startsWith('"') && t.endsWith('"')) || (t.startsWith("'") && t.endsWith("'")))) {
-    return t.slice(1, -1);
+  async systemone(request) {
+    throw new Error(
+      'React Native native module not linked; embed libaria-engine_ffi and bridge aria_systemone'
+    );
   }
-  return t;
-}
 
-function configYmlScalar(key) {
-  const fs = require('fs');
-  const path = require('path');
-  const home = ariaHome();
-  for (const name of ['engine.yml', 'config.yml']) {
-    try {
-      const raw = fs.readFileSync(path.join(home, name), 'utf8');
-      for (const line of raw.split('\n')) {
-        if (line.startsWith(' ') || line.startsWith('\t')) continue;
-        const s = line.trim();
-        if (!s || s.startsWith('#') || !s.includes(':')) continue;
-        const idx = s.indexOf(':');
-        if (s.slice(0, idx).trim() !== key) continue;
-        const v = unquoteYaml(s.slice(idx + 1));
-        if (v) return v;
-      }
-    } catch {
-      continue;
-    }
-  }
-  return undefined;
+  destroy() {}
 }
 
-function resolveHubToken(source, opts = {}) {
-  const named = source === 'modelscope' ? opts.modelscopeApiToken : opts.hfToken;
-  const field = source === 'modelscope' ? 'modelscope_api_token' : 'hf_token';
-  for (const cand of [named, opts.token, configYmlScalar(field)]) {
-    const b = hubBearer(cand);
-    if (b) return b;
-  }
-  return undefined;
-}
-
-function hubPathNames(model) {
-  const names = [model];
-  let lower = model.toLowerCase();
-  let core = model;
-  for (const suf of ['_channel', '_group']) {
-    if (lower.endsWith(suf)) {
-      core = model.slice(0, -suf.length);
-      lower = core.toLowerCase();
-      break;
-    }
-  }
-  const stems = [core];
-  if (lower.endsWith('_q326')) stems.push(`${core.slice(0, -5)}q3.26`);
-  else if (lower.endsWith('_q3.26')) stems.push(`${core.slice(0, -6)}q326`);
-  for (const stem of stems) {
-    for (const share of ['', '_channel', '_group']) {
-      const cand = `${stem}${share}`;
-      if (!names.includes(cand)) names.push(cand);
-    }
-  }
-  return names;
-}
-
-function hubFileUrls(source, model, file) {
-  const urls = [];
-  for (const name of hubPathNames(model)) {
-    if (source === 'modelscope') {
-      for (const repo of [`AriaCompute/${name}`, 'AriaCompute/model']) {
-        urls.push(`https://www.modelscope.cn/models/${repo}/resolve/master/${DEFAULT_SDK}/${name}/${file}`);
-        urls.push(`https://modelscope.cn/models/${repo}/resolve/master/${DEFAULT_SDK}/${name}/${file}`);
-      }
-    } else {
-      for (const repo of [`ariacompute/${name}`, 'ariacompute/model']) {
-        urls.push(`https://huggingface.co/${repo}/resolve/main/${DEFAULT_SDK}/${name}/${file}`);
-      }
-    }
-  }
-  return urls;
-}
-
-function isValidBundleSync(dir) {
-  const fs = require('fs');
-  const path = require('path');
-  try {
-    if (!fs.statSync(path.join(dir, 'weight.bin')).isFile()) return false;
-    const meta = JSON.parse(fs.readFileSync(path.join(dir, 'config.json'), 'utf8'));
-    return meta.format === 'aria-quant-bundle';
-  } catch {
-    return false;
-  }
-}
-
-async function isValidBundle(dir) {
-  if (AriaEngineModule?.isValidBundle) {
-    try {
-      return await AriaEngineModule.isValidBundle(dir);
-    } catch {
-      /* fall through */
-    }
-  }
-  return isValidBundleSync(dir);
-}
-
-async function fetchUrlToFile(url, dest, token) {
-  const headers = {};
-  if (token) headers.Authorization = `Bearer ${token}`;
-  const resp = await fetch(url, { headers });
-  if (resp.status === 401 || resp.status === 403) {
-    const err = new Error(`HTTP ${resp.status}`);
-    err.status = resp.status;
-    throw err;
-  }
-  if (!resp.ok) {
-    const err = new Error(`HTTP ${resp.status}`);
-    err.status = resp.status;
-    throw err;
-  }
-  const fs = require('fs');
-  const path = require('path');
-  fs.mkdirSync(path.dirname(dest), { recursive: true });
-  const buf = Buffer.from(await resp.arrayBuffer());
-  fs.writeFileSync(dest, buf);
-}
-
-async function fetchHubFile(source, model, file, dest, token, required) {
-  let last;
-  for (const url of hubFileUrls(source, model, file)) {
-    try {
-      await fetchUrlToFile(url, dest, token);
-      return true;
-    } catch (e) {
-      last = e;
-      if (e && (e.status === 401 || e.status === 403)) {
-        const field = source === 'modelscope' ? 'modelscope_api_token' : 'hf_token';
-        throw new Error(
-          `auth failed HTTP ${e.status}; set ${field} via aria-engine setup (do not pass a Dashboard sk-/sk-bf- key as the hub token)`,
-        );
-      }
-    }
-  }
-  if (required) throw new Error(`${source}: missing ${file}${last ? `: ${last}` : ''}`);
-  return false;
-}
-
-async function downloadModel(model, tokenOrOpts, site = DEFAULT_SITE) {
-  const opts =
-    tokenOrOpts && typeof tokenOrOpts === 'object'
-      ? tokenOrOpts
-      : { token: tokenOrOpts, site };
-  parseBundleName(model);
-  const fs = require('fs');
-  const path = require('path');
-  const resolvedSite = opts.site ?? site ?? DEFAULT_SITE;
-  const source = preferredPublicHub(resolvedSite);
-  const hubToken = resolveHubToken(source, opts);
-  const cache = cacheDir(model);
-  if (fs.existsSync(cache) && (await isValidBundle(cache))) {
-    return cache;
-  }
-  const staging = path.join(ariaHome(), 'models', `.${model}.partial`);
-  try {
-    if (fs.existsSync(staging)) fs.rmSync(staging, { recursive: true, force: true });
-    fs.mkdirSync(staging, { recursive: true });
-    for (const file of HUB_REQUIRED) {
-      await fetchHubFile(source, model, file, path.join(staging, file), hubToken, true);
-    }
-    for (const extra of HUB_OPTIONAL) {
-      try {
-        await fetchHubFile(source, model, extra, path.join(staging, extra), hubToken, false);
-      } catch {
-        /* optional tokenizer sidecar */
-      }
-    }
-    if (!(await isValidBundle(staging))) {
-      throw new Error(
-        `${source} fetch completed but bundle invalid (need weight.bin + aria-quant-bundle config.json)`,
-      );
-    }
-    if (fs.existsSync(cache)) fs.rmSync(cache, { recursive: true, force: true });
-    fs.renameSync(staging, cache);
-    return cache;
-  } catch (e) {
-    if (fs.existsSync(staging)) fs.rmSync(staging, { recursive: true, force: true });
-    throw e;
-  }
-}
-
-const SDK_UA = 'aria-engine-sdk/0.1.0';
-const zlib = require('zlib');
-
-function ffiLibName(platform = process.platform) {
-  if (platform === 'win32' || String(platform).toLowerCase().startsWith('win')) return 'aria-engine_ffi.dll';
-  if (platform === 'darwin') return 'libaria-engine_ffi.dylib';
-  return 'libaria-engine_ffi.so';
-}
-
-function libDir() {
-  const path = require('path');
-  return path.join(ariaHome(), 'lib');
-}
-
-function cachedFfiPath(platform = process.platform) {
-  const path = require('path');
-  const fs = require('fs');
-  const candidate = path.join(libDir(), ffiLibName(platform));
-  return fs.existsSync(candidate) ? candidate : undefined;
-}
-
-function ffiAssetOs(platform = process.platform, arch = process.arch) {
-  const p = String(platform).toLowerCase();
-  const a = String(arch).toLowerCase();
-  if (p === 'linux' && (a === 'x64' || a === 'x86_64' || a === 'amd64')) return 'linux_x86_64';
-  if (p === 'linux' && (a === 'arm64' || a === 'aarch64')) return 'linux_arm64';
-  if (p === 'darwin' || p === 'macos') return 'macos';
-  if ((p === 'win32' || p.startsWith('win')) && (a === 'x64' || a === 'x86_64' || a === 'amd64')) {
-    return 'windows_x86_64';
-  }
-  throw new Error(`unsupported platform ${platform}/${arch} for libaria-engine_ffi`);
-}
-
-function stripV(tag) {
-  const t = String(tag).trim();
-  return t.startsWith('v') || t.startsWith('V') ? t.slice(1) : t;
-}
-
-function selectLatestStable(releases) {
-  let bestTag;
-  let bestKey = [-1, -1, -1];
-  for (const rel of releases) {
-    if (rel.draft || rel.prerelease) continue;
-    const tag = String(rel.tag_name || rel.tag || '');
-    const core = stripV(tag).split('-')[0].split('+')[0];
-    const parts = core.split('.');
-    if (!parts.length || !/^\d+$/.test(parts[0])) continue;
-    const parsed = [
-      Number(parts[0]),
-      parts[1] && /^\d+$/.test(parts[1]) ? Number(parts[1]) : 0,
-      parts[2] && /^\d+$/.test(parts[2]) ? Number(parts[2]) : 0,
-    ];
-    if (
-      parsed[0] > bestKey[0] ||
-      (parsed[0] === bestKey[0] && parsed[1] > bestKey[1]) ||
-      (parsed[0] === bestKey[0] && parsed[1] === bestKey[1] && parsed[2] > bestKey[2])
-    ) {
-      bestKey = parsed;
-      bestTag = tag;
-    }
-  }
-  if (!bestTag) throw new Error('no stable release found for libaria-engine_ffi');
-  return stripV(bestTag);
-}
-
-function upgradeOrg(site) {
-  const cfg = configYmlScalar('upgrade_url');
-  if (cfg) return cfg.replace(/\/$/, '');
-  const hint = String(site || configYmlScalar('site_url') || DEFAULT_SITE).toLowerCase();
-  if (hint.includes('ariacompute.cn') || hint.includes('gitee.com')) {
-    return 'https://gitee.com/ariacompute';
-  }
-  return 'https://github.com/ariacompute';
-}
-
-function releasesApiUrl(org) {
-  const owner = org.replace(/\/$/, '').split('/').pop() || 'ariacompute';
-  if (org.toLowerCase().includes('gitee.com')) {
-    return `https://gitee.com/api/v5/repos/${owner}/engine/releases?per_page=30`;
-  }
-  return `https://api.github.com/repos/${owner}/engine/releases?per_page=30`;
-}
-
-function extractFfiArchive(archive, destDir, want = ffiLibName()) {
-  const fs = require('fs');
-  const path = require('path');
-  const tar = zlib.gunzipSync(fs.readFileSync(archive));
-  let offset = 0;
-  while (offset + 512 <= tar.length) {
-    const header = tar.subarray(offset, offset + 512);
-    if (header.every((b) => b === 0)) break;
-    const name = header.subarray(0, 100).toString('utf8').replace(/\0.*$/, '');
-    const sizeOctal = header.subarray(124, 136).toString('utf8').replace(/\0/g, '').trim();
-    const size = Number.parseInt(sizeOctal, 8) || 0;
-    const typeFlag = header[156];
-    offset += 512;
-    const isFile = typeFlag === 0 || typeFlag === 48;
-    if (isFile && path.basename(name) === want) {
-      fs.mkdirSync(destDir, { recursive: true });
-      const dest = path.join(destDir, want);
-      fs.writeFileSync(dest, tar.subarray(offset, offset + size));
-      try {
-        fs.chmodSync(dest, 0o755);
-      } catch {
-        /* windows */
-      }
-      return dest;
-    }
-    offset += Math.ceil(size / 512) * 512;
-  }
-  throw new Error(`${want} not found in ${archive}`);
-}
-
-async function httpGetBytes(url) {
-  const resp = await fetch(url, { headers: { 'User-Agent': SDK_UA }, redirect: 'follow' });
-  if (!resp.ok) throw new Error(`HTTP ${resp.status} ${url}`);
-  return Buffer.from(await resp.arrayBuffer());
-}
-
-async function ensureFfiLib(site) {
-  const fs = require('fs');
-  const path = require('path');
-  const env = process.env.ARIA_FFI_LIB;
-  if (env && fs.existsSync(env)) return env;
-  const cached = cachedFfiPath();
-  if (cached) return cached;
-  const org = upgradeOrg(site);
-  const raw = await httpGetBytes(releasesApiUrl(org));
-  const releases = JSON.parse(raw.toString('utf8'));
-  if (!Array.isArray(releases)) throw new Error('unexpected releases payload');
-  const ver = selectLatestStable(releases);
-  const assetName = `libaria-engine_ffi_${ver}_${ffiAssetOs()}.tar.gz`;
-  let url;
-  for (const rel of releases) {
-    const tag = String(rel.tag_name || rel.tag || '');
-    if (stripV(tag) !== ver) continue;
-    for (const asset of rel.assets || []) {
-      if (asset.name === assetName) {
-        url = asset.browser_download_url || asset.direct_asset_url;
-        break;
-      }
-    }
-    if (url) break;
-  }
-  if (!url) throw new Error(`release asset not found: ${assetName}`);
-  const staging = path.join(ariaHome(), 'tmp', `ffi-${ver}`);
-  fs.rmSync(staging, { recursive: true, force: true });
-  fs.mkdirSync(staging, { recursive: true });
-  const archive = path.join(staging, assetName);
-  try {
-    fs.writeFileSync(archive, await httpGetBytes(url));
-    return extractFfiArchive(archive, libDir(), ffiLibName());
-  } finally {
-    fs.rmSync(staging, { recursive: true, force: true });
-  }
-}
-
-export class AriaEngine {
-  constructor(bundlePath) {
-    this.bundlePath = bundlePath;
-    this.cfg = defaultSetupConfig();
-    this.opts = {};
-    // NativeModules.AriaEngine.init(bundlePath) when linked.
-  }
-
-  setup(updates) {
-    this.cfg = applySetup(this.cfg, updates);
-    return this;
-  }
-
-  setupStatus() {
-    return { ...this.cfg };
-  }
-
-  setupClear() {
-    this.cfg = defaultSetupConfig();
-    return this;
-  }
-
-  async open(modelRef) {
-    const opts = {
-      ...this.opts,
-      site: this.cfg.site_url || this.opts.site,
-      hfToken: this.cfg.hf_token || this.opts.hfToken,
-      modelscopeApiToken: this.cfg.modelscope_api_token || this.opts.modelscopeApiToken,
-    };
-    await ensureFfiLib(opts.site);
-    this.bundlePath = isLocalRef(modelRef) ? modelRef : await downloadModel(modelRef, opts);
-    return this;
-  }
-
-  static async open(modelRef, opts = {}) {
-    const eng = new AriaEngine();
-    eng.opts = opts;
-    if (opts.site || opts.hfToken || opts.modelscopeApiToken) {
-      eng.setup({
-        site_url: opts.site,
-        hf_token: opts.hfToken,
-        modelscope_api_token: opts.modelscopeApiToken,
-      });
-    }
-    await eng.open(modelRef);
-    return eng;
-  }
-
-  async complete(messages, options = { max_tokens: 16 }, tools = []) {
-    // return NativeModules.AriaEngine.complete(...)
-    return {
-      success: true,
-      response: '',
-      function_calls: [],
-      note: 'Link native module to libaria-engine_ffi',
-    };
-  }
-
-  async close() {}
-}
+module.exports = { AriaEngine };

package/src/setup.js

@@ -1,69 +0,0 @@
-/** Instance-level Engine setup (in-memory; does not write engine.yml). */
-
-const INTL_SITE = 'https://ariacompute.com';
-const INTL_UPGRADE = 'https://github.com/ariacompute';
-const CN_SITE = 'https://ariacompute.cn';
-const CN_UPGRADE = 'https://gitee.com/ariacompute';
-
-function defaultSetupConfig() {
-  return {
-    router: '',
-    router_api_key: '',
-    site_url: '',
-    upgrade_url: '',
-    compute: 'auto',
-    hf_token: '',
-    modelscope_api_token: '',
-  };
-}
-
-function gatewayRegion(url) {
-  const lower = (url || '').toLowerCase();
-  if (lower.includes('ariacompute.cn') || lower.includes('gitee.com/ariacompute')) return 'cn';
-  if (lower.includes('ariacompute.com') || lower.includes('github.com/ariacompute')) return 'intl';
-  return undefined;
-}
-
-function pairUrls(region) {
-  return region === 'cn' ? [CN_SITE, CN_UPGRADE] : [INTL_SITE, INTL_UPGRADE];
-}
-
-function fillSetupUrls(cfg) {
-  const out = { ...cfg };
-  const region = gatewayRegion(out.site_url) || gatewayRegion(out.upgrade_url);
-  if (!region) return out;
-  const [site, upgrade] = pairUrls(region);
-  if (!out.site_url) out.site_url = site;
-  if (!out.upgrade_url) out.upgrade_url = upgrade;
-  return out;
-}
-
-function validateRouterApiKey(key) {
-  const t = (key || '').trim();
-  if (!t) return;
-  if (t.startsWith('sk-aria_') || t.startsWith('sk-bf-')) return;
-  throw new Error('router_api_key must start with sk-aria_ or sk-bf-');
-}
-
-function applySetup(existing, updates) {
-  const out = { ...existing };
-  for (const [k, v] of Object.entries(updates || {})) {
-    if (v === undefined) continue;
-    out[k] = v;
-  }
-  if (!['auto', 'cpu', 'cuda'].includes(out.compute)) {
-    throw new Error(`invalid compute: ${out.compute}`);
-  }
-  validateRouterApiKey(out.router_api_key);
-  return fillSetupUrls(out);
-}
-
-module.exports = {
-  INTL_SITE,
-  INTL_UPGRADE,
-  CN_SITE,
-  CN_UPGRADE,
-  defaultSetupConfig,
-  fillSetupUrls,
-  applySetup,
-};

package/test/setup.test.cjs

@@ -1,77 +0,0 @@
-const test = require('node:test');
-const assert = require('node:assert/strict');
-const fs = require('node:fs');
-const os = require('node:os');
-const path = require('node:path');
-const {
-  applySetup,
-  defaultSetupConfig,
-  fillSetupUrls,
-  CN_SITE,
-  CN_UPGRADE,
-} = require('../src/setup.js');
-
-test('setup defaults', () => {
-  const cfg = defaultSetupConfig();
-  assert.equal(cfg.compute, 'auto');
-  assert.equal(cfg.router, '');
-});
-
-test('setup invalid enum', () => {
-  assert.throws(() => applySetup(defaultSetupConfig(), { compute: 'gpu' }));
-});
-
-test('setup fills urls from cn site', () => {
-  const got = fillSetupUrls({ site_url: CN_SITE, upgrade_url: '' });
-  assert.equal(got.upgrade_url, CN_UPGRADE);
-});
-
-test('setup instance all fields via applySetup', () => {
-  const st = applySetup(defaultSetupConfig(), {
-    router: 'http://127.0.0.1:8080',
-    site_url: CN_SITE,
-    upgrade_url: CN_UPGRADE,
-    compute: 'cpu',
-    hf_token: 'hf_abc',
-    modelscope_api_token: 'ms_xyz',
-  });
-  assert.equal(st.router, 'http://127.0.0.1:8080');
-  assert.equal(st.hf_token, 'hf_abc');
-});
-
-test('setup partial merge', () => {
-  const once = applySetup(defaultSetupConfig(), { hf_token: 'hf_one', router: 'http://127.0.0.1:1' });
-  const st = applySetup(once, { compute: 'cuda' });
-  assert.equal(st.hf_token, 'hf_one');
-  assert.equal(st.router, 'http://127.0.0.1:1');
-  assert.equal(st.compute, 'cuda');
-});
-
-test('setup invalid enum leaves state', () => {
-  const once = applySetup(defaultSetupConfig(), { compute: 'cpu' });
-  assert.throws(() => applySetup(once, { compute: 'gpu' }));
-  assert.equal(once.compute, 'cpu');
-});
-
-test('setup fills urls from site tld', () => {
-  const st = applySetup(defaultSetupConfig(), { site_url: 'https://ariacompute.cn' });
-  assert.equal(st.upgrade_url, CN_UPGRADE);
-});
-
-test('setup does not write engine.yml', () => {
-  const prev = process.env.ARIA_COMPUTE_HOME;
-  const home = fs.mkdtempSync(path.join(os.tmpdir(), 'aria-rn-setup-'));
-  process.env.ARIA_COMPUTE_HOME = home;
-  try {
-    applySetup(defaultSetupConfig(), {
-      router: 'http://127.0.0.1:8080',
-      site_url: 'https://ariacompute.com',
-      hf_token: 'hf_x',
-    });
-    assert.equal(fs.existsSync(path.join(home, 'engine.yml')), false);
-    assert.equal(fs.existsSync(path.join(home, 'config.yml')), false);
-  } finally {
-    if (prev === undefined) delete process.env.ARIA_COMPUTE_HOME;
-    else process.env.ARIA_COMPUTE_HOME = prev;
-  }
-});