mslxdff 0.1.66 → 0.1.68

package/src/cli/bootstrap.js

@@ -1,408 +1 @@
-import { join, dirname } from "node:path";
-import { fileURLToPath } from "node:url";
-import { startServer } from "../server.js";
-import { defaultStateFile } from "../state.js";
-import { createRouter } from "../routes.js";
-import { createUpstreamClient } from "../upstream.js";
-import { createModelsService } from "../models.js";
-import { loadToken } from "../state.js";
-import { createAutoSelector } from "../auto.js";
-import { createPeersService } from "../peers.js";
-import { createEventBus } from "../events.js";
-import { createGroupsService, createBansService, refreshGroupMembers, syncPeersFromMembers } from "../groups.js";
-import { logDir, appendCall, appendError, appendEvent } from "../logs.js";
-import { loadPlugins, runHook, resolvePluginDirs } from "../plugins.js";
-import { createOpenCodeProvider } from "../providers/opencode.js";
-import { loadProviderKeys, loadProviderAuths, loadProviderConfigs } from "../state.js";
-import { effectiveHost, refreshIntervalMs, modelCooldownMs, slowCooldownMs, peerCooldownMs, peerHeatMs, maxHopsValue, groupSyncIntervalMs, autoUpdateIntervalMs, banWindowMs, banThreshold } from "./policy.js";
-import { fmtEvent } from "./format.js";
-import { errMsg, npmCmd, run } from "./util.js";
-import { loadGroupsJoined } from "../state.js";
-import { writePid } from "../daemon.js";
-import { resolvePort } from "../server.js";
-
-const logs = { appendCall, appendError, appendEvent };
-
-export async function startDaemonMain(VERSION) {
-  const { token, created } = await loadToken();
-  const pkgRoot = dirname(dirname(fileURLToPath(import.meta.url)));
-  // src/cli/bootstrap.js -> 2 levels up to src, then 1 to root
-  const pkgRoot2 = join(dirname(fileURLToPath(import.meta.url)), "..", "..");
-  const pluginDirs = resolvePluginDirs({ pkgRoot: pkgRoot2 });
-  const { plugins: loadedPlugins, errors: pluginErrors } = await loadPlugins({ dirs: pluginDirs });
-  for (const e of pluginErrors) {
-    console.log(`plugin load failed: ${e.file} — ${e.error}`);
-    appendEvent({ ts: Date.now(), type: "plugin-load-error", file: e.file, error: e.error });
-  }
-  if (loadedPlugins.length) {
-    console.log(`plugins loaded (${loadedPlugins.length}): ${loadedPlugins.map((p) => `${p.name}${p.version ? `@${p.version}` : ""}`).join(", ")}`);
-    appendEvent({ ts: Date.now(), type: "plugins-loaded", plugins: loadedPlugins.map((p) => ({ name: p.name, version: p.version })) });
-  }
-  const upstreamHooks = loadedPlugins.length
-    ? (name, ctx) => runHook(loadedPlugins, name, ctx)
-    : null;
-  const providerPlugin = loadedPlugins.find((p) => typeof p.createUpstream === "function");
-  let upstream;
-  const baseUrl = process.env.UPSTREAM_BASE_URL || "https://opencode.ai";
-  let providers = [];
-  if (providerPlugin) {
-    if (!upstream) {
-      try {
-        upstream = await providerPlugin.createUpstream({ baseUrl, authToken: process.env.UPSTREAM_AUTH_TOKEN || "public", env: process.env });
-        console.log(`upstream provider replaced by plugin: ${providerPlugin.name}`);
-        appendEvent({ ts: Date.now(), type: "plugin-upstream-active", plugin: providerPlugin.name });
-      } catch (err) {
-        console.log(`plugin upstream (${providerPlugin.name}) failed: ${errMsg(err)} — falling back to default`);
-        appendEvent({ ts: Date.now(), type: "plugin-upstream-error", plugin: providerPlugin.name, error: errMsg(err) });
-      }
-    }
-    if (!upstream) upstream = createUpstreamClient({ hooks: upstreamHooks });
-  } else {
-    const opencodeClient = createUpstreamClient({ hooks: upstreamHooks });
-    const opencodeModels = createModelsService({
-      baseUrl,
-      headers: opencodeClient.headers,
-      refreshMs: refreshIntervalMs(),
-      cacheFile: join(logDir(), "models.json"),
-    });
-    providers.push(createOpenCodeProvider({ upstream: opencodeClient, modelsService: opencodeModels }));
-    const orKeys = loadProviderKeys("openrouter");
-    if (orKeys.length) {
-      const { createOpenRouterProvider } = await import("../providers/openrouter.js");
-      providers.push(createOpenRouterProvider({ apiKeys: orKeys }));
-      console.log(`provider enabled: openrouter (${orKeys.length} key${orKeys.length > 1 ? "s" : ""})`);
-      appendEvent({ ts: Date.now(), type: "provider-enabled", provider: "openrouter", keys: orKeys.length });
-    }
-    const genericConfigs = loadProviderConfigs();
-    for (const [gid, cfg] of Object.entries(genericConfigs)) {
-      if (gid === "opencode" || gid === "openrouter") continue;
-      if (gid === "workbuddy") {
-        const base = String(cfg?.baseUrl || "").trim() || "https://copilot.tencent.com";
-        const keys = Array.isArray(cfg?.keys) ? cfg.keys.filter((k) => typeof k === "string" && k.trim()) : [];
-        const auths = Array.isArray(cfg?.auths) ? cfg.auths : [];
-        if (!keys.length) continue;
-        try {
-          const { createWorkbuddyProvider } = await import("../providers/workbuddy.js");
-          providers.push(createWorkbuddyProvider({ baseUrl: base, apiKeys: keys, auths }));
-          console.log(`provider enabled: workbuddy (${keys.length} key${keys.length > 1 ? "s" : ""}) baseUrl=${base}`);
-          appendEvent({ ts: Date.now(), type: "provider-enabled", provider: gid, keys: keys.length, baseUrl: base });
-        } catch (err) {
-          console.log(`provider ${gid} failed: ${err?.message || err}`);
-          appendEvent({ ts: Date.now(), type: "provider-error", provider: gid, error: String(err?.message || err) });
-        }
-        continue;
-      }
-      const base = String(cfg?.baseUrl || "").trim();
-      const keys = Array.isArray(cfg?.keys) ? cfg.keys.filter((k) => typeof k === "string" && k.trim()) : [];
-      if (!base || !keys.length) continue;
-      try {
-        const { createGenericProvider } = await import("../providers/generic.js");
-        providers.push(createGenericProvider({ id: gid, baseUrl: base, apiKeys: keys }));
-        console.log(`provider enabled: ${gid} (${keys.length} key${keys.length > 1 ? "s" : ""}) baseUrl=${base}`);
-        appendEvent({ ts: Date.now(), type: "provider-enabled", provider: gid, keys: keys.length, baseUrl: base });
-      } catch (err) {
-        console.log(`provider ${gid} failed: ${err?.message || err}`);
-        appendEvent({ ts: Date.now(), type: "provider-error", provider: gid, error: String(err?.message || err) });
-      }
-    }
-    if (!genericConfigs["workbuddy"]) {
-      const wbKeys = loadProviderKeys("workbuddy");
-      if (wbKeys.length) {
-        try {
-          const { createWorkbuddyProvider } = await import("../providers/workbuddy.js");
-          const wbAuths = loadProviderAuths("workbuddy");
-          providers.push(createWorkbuddyProvider({ apiKeys: wbKeys, auths: wbAuths }));
-          console.log(`provider enabled: workbuddy (${wbKeys.length} key${wbKeys.length > 1 ? "s" : ""}) baseUrl=https://copilot.tencent.com (env)`);
-          appendEvent({ ts: Date.now(), type: "provider-enabled", provider: "workbuddy", keys: wbKeys.length, baseUrl: "https://copilot.tencent.com" });
-        } catch (err) {
-          console.log(`provider workbuddy failed: ${err?.message || err}`);
-        }
-      }
-    }
-    const { createProviderDispatcher } = await import("../providers/dispatcher.js");
-    upstream = createProviderDispatcher(providers);
-    appendEvent({ ts: Date.now(), type: "providers", providers: providers.map((p) => p.id) });
-  }
-  const opencodeProvider = providers.find((p) => p.id === "opencode");
-  const models = createModelsService({
-    providers: providers.length > 1 ? providers : undefined,
-    baseUrl,
-    headers: upstream.headers || opencodeProvider?.upstream?.headers,
-    refreshMs: refreshIntervalMs(),
-    cacheFile: join(logDir(), "models.json"),
-  });
-  const auto = createAutoSelector({
-    cooldownMs: modelCooldownMs(),
-    slowCooldownMs: slowCooldownMs(),
-    file: defaultStateFile(),
-    loadCandidates: async () => {
-      try {
-        return (await models.get()).data.map((m) => m.id);
-      } catch {
-        return null;
-      }
-    },
-  });
-  const peers = createPeersService({ cooldownMs: peerCooldownMs(), heatMs: peerHeatMs() });
-  const groups = createGroupsService({});
-  const bans = createBansService({ windowMs: banWindowMs(), threshold: banThreshold() });
-
-  const isDebug = process.env.MSLXDFF_DEBUG === "1";
-  const bus = createEventBus();
-
-  try {
-    const { loadModelAliases } = await import("../providers/model-id.js");
-    loadModelAliases();
-  } catch {}
-
-  const router = createRouter({ token, upstream, models, auto, logs, peers, maxHops: maxHopsValue(), groups, bans, bus, plugins: loadedPlugins });
-  const listenHost = effectiveHost();
-  const srv = startServer({
-    router,
-    signals: !isDebug,
-    host: listenHost,
-    onBeforeClose: loadedPlugins.length
-      ? () => runHook(loadedPlugins, "server:stop", { version: VERSION }).then(() => {})
-      : undefined,
-  });
-
-  if (isDebug) {
-    bus.subscribe((e) => {
-      try {
-        console.log(fmtEvent(e));
-      } catch {}
-    });
-    const { startDaemon: sd } = await import("../daemon.js");
-    const restore2 = () => {
-      console.log("\n[debug] restoring background daemon...");
-      try {
-        const restoredPid = sd([]);
-        console.log(`[debug] daemon restored (pid ${restoredPid})`);
-      } catch (err) {
-        console.error(`[debug] could not restore daemon: ${err.message}`);
-      }
-      setTimeout(() => process.exit(0), 300);
-    };
-    process.on("SIGINT", restore2);
-    process.on("SIGTERM", restore2);
-  }
-
-  await srv.ready();
-
-  if (loadedPlugins.length) {
-    runHook(loadedPlugins, "server:start", { port: srv.server.address()?.port, host: listenHost, version: VERSION }).catch(() => {});
-    const eventPlugins = loadedPlugins.filter((p) => typeof p.onEvent === "function");
-    if (eventPlugins.length) {
-      bus.subscribe((e) => {
-        for (const p of eventPlugins) {
-          try { p.onEvent(e); } catch {}
-        }
-      });
-    }
-  }
-
-  setTimeout(() => {
-    upstream.preheat().then((r) => {
-      const entry = { ts: Date.now(), type: "upstream-preheat", ...r, baseUrl };
-      try { bus.emit(entry); } catch {}
-      try { logs.appendEvent(entry); } catch {}
-      if (r.skipped) console.log(`[preheat] skipped (MSLXDFF_PREHEAT disabled)`);
-      else if (r.ok) console.log(`[preheat] opencode models ok ${r.status} ${r.ms}ms`);
-      else console.log(`[preheat] opencode models failed ${r.error || r.status || ""} ${r.ms || 0}ms`);
-    }).catch(() => {});
-  }, 100).unref?.();
-
-  models.startAutoRefresh();
-  if (process.env.MSLXDFF_DAEMON) {
-    writePid(process.pid, VERSION);
-  }
-  const addr = srv.server.address();
-  const host = addr.address === "0.0.0.0" || addr.address === "::" ? "localhost" : addr.address;
-  console.log(`mslxdff v${VERSION} listening on http://${host}:${addr.port}`);
-  if (created) {
-    console.log(`auth token: ${token}`);
-  }
-  console.log(`endpoint:   http://${host}:${addr.port}/v1`);
-  try {
-    const { hedgeDelayMs } = await import("../routes/hedge.js");
-    const hd = hedgeDelayMs();
-    console.log(`hedge:      ${hd ? `${hd}ms` : "off"} (MSLXDFF_HEDGE_DELAY_MS)`);
-  } catch {}
-
-  // group sync
-  const { syncAllJoinedGroups } = await import("./group-helpers.js");
-  syncAllJoinedGroups({ peers, groups })
-    .then((results) => {
-      for (const r of results) {
-        if (r.error) console.log(`group sync ${r.name}: failed — ${r.error}`);
-        else console.log(`group sync ${r.name}: ${r.total} member(s), ${r.added} peer(s)`);
-      }
-    })
-    .catch((err) => console.log(`group sync: ${errMsg(err)}`));
-  const groupSyncTimer = setInterval(() => {
-    syncAllJoinedGroups({ peers, groups })
-      .then((results) => {
-        for (const r of results) {
-          if (r.error) console.log(`group sync ${r.name}: failed — ${r.error}`);
-          else if (r.added) console.log(`group sync ${r.name}: ${r.total} member(s), ${r.added} peer(s)`);
-        }
-      })
-      .catch((err) => console.log(`group sync: ${errMsg(err)}`));
-  }, groupSyncIntervalMs());
-  groupSyncTimer.unref();
-
-  const broadbandGroups = () => loadGroupsJoined().filter((g) => g.kind === "broadband" && g.leaderUrl);
-  if (broadbandGroups().length) {
-    const doHeartbeat = async () => {
-      for (const g of broadbandGroups()) {
-        try {
-          const res = await fetch(`${g.leaderUrl}/v1/groups/relay/heartbeat`, {
-            method: "POST",
-            headers: { "Content-Type": "application/json", "Authorization": `Bearer ${token}` },
-            body: JSON.stringify({ name: g.name, group: g.name }),
-            signal: AbortSignal.timeout(5000),
-          });
-          if (!res.ok) {
-            const txt = await res.text().catch(() => "");
-            console.log(`broadband heartbeat ${g.name}: ${res.status} ${txt.slice(0, 100)}`);
-          }
-        } catch (err) {
-          console.log(`broadband heartbeat ${g.name}: failed — ${errMsg(err)}`);
-        }
-      }
-    };
-    const doPoll = async () => {
-      for (const g of broadbandGroups()) {
-        try {
-          const pollRes = await fetch(`${g.leaderUrl}/v1/groups/relay/poll`, {
-            method: "POST",
-            headers: { "Content-Type": "application/json", "Authorization": `Bearer ${token}` },
-            body: JSON.stringify({ name: g.name, group: g.name }),
-            signal: AbortSignal.timeout(8000),
-          });
-          if (!pollRes.ok) continue;
-          const data = await pollRes.json().catch(() => ({}));
-          const items = data.data || [];
-          for (const item of items) {
-            const { reqId, body } = item;
-            let result;
-            try {
-              const upRes = await upstream.chat(body);
-              const ct = upRes.headers.get("content-type") || "";
-              const isStream = Boolean(body?.stream) || ct.includes("text/event-stream");
-              if (isStream && upRes.body) {
-                let collected = "";
-                for await (const chunk of upRes.body) {
-                  collected += Buffer.isBuffer(chunk) ? chunk.toString("utf8") : String(chunk);
-                }
-                result = { status: upRes.status, headers: { "Content-Type": "text/event-stream" }, body: collected };
-              } else {
-                const txt = await upRes.text();
-                let parsed;
-                try { parsed = JSON.parse(txt); } catch { parsed = txt; }
-                result = { status: upRes.status, headers: { "Content-Type": upRes.headers.get("content-type") || "application/json" }, body: typeof parsed === "string" ? parsed : JSON.stringify(parsed) };
-              }
-            } catch (err) {
-              result = { status: 502, headers: { "Content-Type": "application/json" }, body: JSON.stringify({ error: errMsg(err) }) };
-            }
-            try {
-              await fetch(`${g.leaderUrl}/v1/groups/relay/result`, {
-                method: "POST",
-                headers: { "Content-Type": "application/json", "Authorization": `Bearer ${token}` },
-                body: JSON.stringify({ name: g.name, group: g.name, reqId, result }),
-                signal: AbortSignal.timeout(5000),
-              });
-            } catch {}
-          }
-        } catch {}
-      }
-    };
-    doHeartbeat().catch(() => {});
-    const hbTimer = setInterval(doHeartbeat, 30_000);
-    hbTimer.unref();
-    const pollTimer = setInterval(doPoll, 1000);
-    pollTimer.unref();
-    console.log(`broadband relay: heartbeat 30s + poll 1s for ${broadbandGroups().length} group(s)`);
-  }
-
-  const autoUpdateMs = autoUpdateIntervalMs();
-  function emitAutoUpdate(type, data = {}) {
-    const entry = { ts: Date.now(), type, ...data };
-    try { bus?.emit(entry); } catch {}
-    try { logs?.appendEvent?.(entry); } catch {}
-    const line = `[auto-update] ${type} ${JSON.stringify(data)}`;
-    console.log(line);
-  }
-  if (autoUpdateMs) {
-    console.log(`auto-update enabled: checking every ${Math.round(autoUpdateMs / 60000)}m`);
-    emitAutoUpdate("auto-update-enabled", { intervalMs: autoUpdateMs, current: VERSION });
-    setTimeout(() => {
-      emitAutoUpdate("auto-update-check", { current: VERSION });
-      checkAndAutoUpdate().catch((err) => {
-        console.log(`auto-update check failed: ${errMsg(err)}`);
-        emitAutoUpdate("auto-update-failed", { error: errMsg(err) });
-      });
-    }, 30_000).unref?.();
-    const autoUpdateTimer = setInterval(() => {
-      emitAutoUpdate("auto-update-check", { current: VERSION });
-      checkAndAutoUpdate().catch((err) => {
-        console.log(`auto-update check failed: ${errMsg(err)}`);
-        emitAutoUpdate("auto-update-failed", { error: errMsg(err) });
-      });
-    }, autoUpdateMs);
-    autoUpdateTimer.unref();
-  } else {
-    console.log(`auto-update disabled (set MSLXDFF_AUTO_UPDATE=1 to enable hourly)`);
-    emitAutoUpdate("auto-update-disabled", { current: VERSION });
-  }
-
-  async function checkAndAutoUpdate() {
-    emitAutoUpdate("auto-update-query", { current: VERSION });
-    const info = await run(npmCmd(), ["view", "mslxdff", "dist-tags.latest", "--json"]);
-    if (info.err) {
-      emitAutoUpdate("auto-update-query-failed", { error: info.err.message || String(info.stderr || "").slice(0, 500) });
-      throw new Error(info.err.message || String(info.stderr || "").slice(0, 500));
-    }
-    let latest = "";
-    try {
-      latest = JSON.parse(String(info.stdout || "").trim());
-      if (Array.isArray(latest)) latest = latest[latest.length - 1];
-      latest = String(latest || "").replace(/^v/, "").trim();
-    } catch {
-      const raw = String(info.stdout || "").trim();
-      const m = raw.match(/(\d+\.\d+\.\d+[^\s'"]*)/);
-      latest = m ? m[1] : raw.split(/\s+/).pop()?.replace(/['"]/g, "") || "";
-    }
-    latest = latest.replace(/['"]/g, "").trim();
-    emitAutoUpdate("auto-update-queried", { current: VERSION, latest, stdout: String(info.stdout || "").trim().slice(0, 200) });
-    if (!latest || latest === VERSION) {
-      emitAutoUpdate("auto-update-noop", { current: VERSION, latest });
-      return;
-    }
-    const { compareSemver } = await import("./policy.js");
-    if (compareSemver(latest, VERSION) <= 0) {
-      emitAutoUpdate("auto-update-noop", { current: VERSION, latest, reason: "not newer" });
-      return;
-    }
-    emitAutoUpdate("auto-update-found", { current: VERSION, latest });
-    console.log(`auto-update: v${VERSION} -> v${latest}, installing...`);
-    emitAutoUpdate("auto-update-installing", { current: VERSION, latest });
-    const up = await run(npmCmd(), ["install", "-g", `mslxdff@${latest}`]);
-    if (up.err) {
-      emitAutoUpdate("auto-update-install-failed", { current: VERSION, latest, error: up.err.message || String(up.stderr || "").slice(0, 500) });
-      throw new Error(up.err.message || String(up.stderr || "").slice(0, 500));
-    }
-    emitAutoUpdate("auto-update-installed", { current: VERSION, latest, stdout: String(up.stdout || "").slice(0, 500) });
-    console.log(`auto-update: installed v${latest}, restarting daemon...`);
-    emitAutoUpdate("auto-update-restarting", { current: VERSION, latest });
-    const { stopDaemon, startDaemon } = await import("../daemon.js");
-    try { stopDaemon(); } catch (e) { emitAutoUpdate("auto-update-stop-failed", { error: errMsg(e) }); }
-    const { waitForHealth } = await import("./policy.js");
-    const newPid = startDaemon([]);
-    await waitForHealth(resolvePort(), 8000);
-    console.log(`auto-update: restarted as v${latest} (pid ${newPid})`);
-    emitAutoUpdate("auto-update-restarted", { current: VERSION, latest, newPid });
-  }
-}
-
-
+export * from "../runtime/bootstrap.js";

package/src/cli/commands/provider/bench.js

@@ -0,0 +1,140 @@
+import { probeModels } from "../../../bench/probe.js";
+import { runOne } from "../../../bench/runner.js";
+import { formatReport } from "../../../bench/report.js";
+import { defaultModelsPath, defaultChatPath } from "../../../state/provider-config.js";
+
+function parseBenchArgs(rest) {
+  const opts = { json: false, prompt: "hi", maxTokens: 32, timeoutMs: 30000 };
+  for (let i = 0; i < rest.length; i++) {
+    const a = rest[i];
+    if (a === "--json" || a === "-json") opts.json = true;
+    else if (a === "--prompt" && rest[i + 1]) opts.prompt = rest[++i];
+    else if (a.startsWith("--prompt=")) opts.prompt = a.slice(9);
+    else if (a === "--max-tokens" && rest[i + 1]) opts.maxTokens = Number(rest[++i]) || 32;
+    else if (a.startsWith("--max-tokens=")) opts.maxTokens = Number(a.split("=")[1]) || 32;
+    else if (a === "--timeout" && rest[i + 1]) opts.timeoutMs = Number(rest[++i]) || 30000;
+    else if (a.startsWith("--timeout=")) opts.timeoutMs = Number(a.split("=")[1]) || 30000;
+  }
+  return opts;
+}
+
+function buildHeadersForProvider(providerId, apiKey, auth) {
+  const h = {};
+  if (String(providerId).toLowerCase() === "workbuddy") {
+    h["Content-Type"] = "application/json";
+    h["Accept"] = "text/event-stream";
+    h["User-Agent"] = "CLI/2.115.0 WorkBuddy/2.115.0";
+    h["Origin"] = "https://www.codebuddy.cn";
+    h["Referer"] = "https://www.codebuddy.cn/";
+    h["X-Product"] = "SaaS";
+    if (apiKey) h["Authorization"] = `Bearer ${apiKey}`;
+    if (auth?.uid) h["X-User-Id"] = auth.uid;
+    h["X-Domain"] = auth?.domain || "www.codebuddy.cn";
+    if (auth?.enterpriseId) {
+      h["X-Enterprise-Id"] = auth.enterpriseId;
+      h["X-Tenant-Id"] = auth.enterpriseId;
+    }
+    return h;
+  }
+  if (apiKey) h["Authorization"] = `Bearer ${apiKey}`;
+  return h;
+}
+
+export async function handleProviderBench(id, sub, rest, args, deps = {}) {
+  const isBench = sub === "bench" || sub === "benchmark" || sub === "eval" || sub === "test";
+  if (!isBench) return false;
+  const fetchImpl = deps.fetchImpl || globalThis.fetch;
+  const loadConfigs = deps.loadProviderConfigs || (await import("../../../state.js")).loadProviderConfigs;
+  const loadKeys = deps.loadProviderKeys || (await import("../../../state.js")).loadProviderKeys;
+  const loadAllowed = deps.loadProviderAllowedModels || (await import("../../../state.js")).loadProviderAllowedModels;
+  const loadAllowAny = deps.loadProviderAllowAnyModels || (await import("../../../state.js")).loadProviderAllowAnyModels;
+  const loadBaseUrl = deps.loadProviderBaseUrl || (await import("../../../state.js")).loadProviderBaseUrl;
+  const loadAuths = deps.loadProviderAuths || (async () => { try { const m = await import("../../../state.js"); return m.loadProviderAuths ? m.loadProviderAuths(id) : []; } catch { return []; } }) && (await import("../../../state.js")).loadProviderAuths;
+
+  const providerId = String(id || "").trim();
+  if (!providerId) {
+    console.error("usage: mslxdff -provider <id> bench [--json] [--prompt hi] [--max-tokens 32]");
+    process.exit(1);
+  }
+  const opts = parseBenchArgs(rest || []);
+  const configs = loadConfigs();
+  const cfg = configs[providerId] || {};
+  const keys = loadKeys(providerId) || [];
+  const allowed = loadAllowed(providerId) || [];
+  const allowAny = loadAllowAny(providerId);
+  const baseUrl = (loadBaseUrl(providerId) || cfg.baseUrl || "").trim();
+  let auths = [];
+  try { const m = await import("../../../state.js"); auths = m.loadProviderAuths ? m.loadProviderAuths(providerId) : []; } catch {}
+  if (!baseUrl) {
+    console.error(`provider ${providerId}: missing baseUrl — 先设置: mslxdff -provider ${providerId} set-url https://api.example.com/v1`);
+    process.exit(1);
+  }
+  if (!keys.length) {
+    console.error(`provider ${providerId}: 未配置 Key — 先设置: mslxdff -provider ${providerId} <key>`);
+    process.exit(1);
+  }
+  // 空勾选 → 不发 chat，仅探活模型列表并提示
+  if (!allowed.length) {
+    const modelsPath = cfg.modelsPath || defaultModelsPath(providerId);
+    console.log(`provider ${providerId}: 未设置 allowlist（allowAny=${allowAny ? "ON" : "OFF"}），不发起测速，仅探活模型列表...`);
+    console.log(`尝试：GET ${baseUrl}${modelsPath} → GET ${baseUrl}/v1/models → GET ${baseUrl}/models`);
+    const headers = buildHeadersForProvider(providerId, keys[0], auths[0]);
+    const probed = await probeModels({ baseUrl, modelsPath, headers, fetchImpl, timeoutMs: 8000 });
+    if (!probed.ok) {
+      console.error(`探活失败：${probed.error}`);
+      console.error(`已尝试：${probed.tried.join(", ")}`);
+      console.error(`请手动设置 allowlist：mslxdff -provider ${providerId} allowlist set <model>`);
+      if (opts.json) console.log(JSON.stringify({ ok: false, error: probed.error, tried: probed.tried }, null, 2));
+      process.exit(1);
+    }
+    const list = probed.data || [];
+    if (!list.length) {
+      console.log("探活成功但返回空列表，请确认上游是否暴露 /v1/models");
+      if (opts.json) console.log(JSON.stringify({ ok: true, data: [] }, null, 2));
+      process.exit(0);
+    }
+    console.log(`\n发现 ${list.length} 个模型：`);
+    for (const m of list.slice(0, 30)) {
+      console.log(`  - ${m.id || m.model || m.name || JSON.stringify(m).slice(0, 80)}`);
+    }
+    if (list.length > 30) console.log(`  ... 还有 ${list.length - 30} 个未展示`);
+    console.log(`\n下一步：勾选后再测（只测勾选，避免扣费）`);
+    console.log(`  mslxdff -provider ${providerId} allowlist set ${list.slice(0, 2).map((m) => m.id).join(" ")}`);
+    console.log(`  mslxdff -provider ${providerId} bench${opts.json ? " --json" : ""}`);
+    if (opts.json) console.log(JSON.stringify({ ok: true, data: list, hint: `pick then bench` }, null, 2));
+    process.exit(0);
+  }
+
+  // 有勾选 → 逐个测
+  console.log(`bench ${providerId}: 共 ${allowed.length} 个已勾选模型，逐个测速（串行，${opts.timeoutMs}ms 超时）...`);
+  if (!opts.json) console.log(`prompt="${opts.prompt}" maxTokens=${opts.maxTokens}\n`);
+  const chatPath = cfg.chatPath || defaultChatPath(providerId);
+  const results = [];
+  for (let i = 0; i < allowed.length; i++) {
+    const raw = allowed[i];
+    const model = String(raw || "").trim();
+    if (!opts.json) process.stdout.write(`  [${i + 1}/${allowed.length}] ${model} ... `);
+    // 轮询 key/auth：按索引取，超长循环
+    const kIdx = i % keys.length;
+    const aIdx = Math.min(kIdx, auths.length - 1);
+    const key = keys[kIdx];
+    const auth = auths[aIdx] || auths[0] || null;
+    const headers = buildHeadersForProvider(providerId, key, auth);
+    const r = await runOne({ baseUrl, chatPath, model, providerId, apiKey: key, headers, prompt: opts.prompt, maxTokens: opts.maxTokens, timeoutMs: opts.timeoutMs, fetchImpl });
+    results.push(r);
+    if (!opts.json) {
+      if (r.ok) console.log(`OK  TTFB ${r.ttfbMs}ms  总 ${r.totalMs}ms  ${r.tps != null ? `${r.tps} t/s` : r.charsPerSec != null ? `${r.charsPerSec} 字/秒` : "—"}`);
+      else console.log(`FAIL ${r.label} ${r.error ? `(${r.error.slice(0, 60)})` : ""}`);
+    }
+  }
+  const report = formatReport(results, { json: opts.json });
+  console.log("\n" + report.text);
+  if (opts.json) {
+    // json 已在 text 中输出一次（report.text 是 JSON），无需重复
+  }
+  const failed = results.filter((r) => !r.ok).length;
+  if (failed && !opts.json) {
+    console.log(`\n提示：失败 ${failed} 个多为 402余额不足/429限流/超时，可清冷却或换 Key 后重试`);
+  }
+  process.exit(failed ? 2 : 0);
+}

package/src/cli/commands/provider/index.js

@@ -187,6 +187,8 @@ export async function handleProvider(args) {
   if (await handleProviderAllowlist(id, sub, rest)) return true;
   const { handleProviderModels } = await import("./models.js");
   if (await handleProviderModels(id, sub, args, rest)) return true;
+  const { handleProviderBench } = await import("./bench.js");
+  if (await handleProviderBench(id, sub, rest, args)) return true;
   const { handleProviderKeys } = await import("./keys.js");
   await handleProviderKeys(id, sub, rest, args);
   return true;

package/src/cli/provider-row.js

@@ -0,0 +1,73 @@
+import { readFileSync, existsSync } from "node:fs";
+import { loadProviderKeys, loadProviderAuths, loadProviderConfigs, loadProviderAllowedModels, loadProviderShareKeys, loadProviderBaseUrl } from "../state.js";
+
+/**
+ * 聚合 genericIds：来自 providerConfigs + providerKeys + env MSLXDFF_*_KEY
+ * 纯逻辑抽取，便于单测；默认读真实 stateFile 与 env，测试可注入 stateFile
+ */
+export function buildProviderRows({ stateFile, env = process.env } = {}) {
+  // 复用 status 的聚合逻辑，但抽为纯函数
+  let configs = {};
+  try { configs = loadProviderConfigs(stateFile ? { file: stateFile } : undefined); } catch {}
+  const upstreamBase = env.UPSTREAM_BASE_URL || "https://opencode.ai";
+  const rows = [];
+  const opAllowed = (() => { try { return loadProviderAllowedModels("opencode", stateFile ? { file: stateFile } : undefined); } catch { return []; } })();
+  rows.push({ id: "opencode", enabled: true, baseUrl: upstreamBase, keys: [], allowed: opAllowed, share: false, note: "built-in, no key, cannot share", authCount: 0 });
+
+  const genericIds = new Set(Object.keys(configs).filter((id) => id !== "opencode"));
+  try {
+    const raw = JSON.parse(readFileSync(stateFile || "", "utf8"));
+    const pk = raw.providerKeys || {};
+    for (const id of Object.keys(pk)) if (id !== "opencode") genericIds.add(id);
+    const cfgRaw = raw.providerConfigs || {};
+    for (const id of Object.keys(cfgRaw)) if (id !== "opencode") genericIds.add(id);
+  } catch {}
+  for (const k of Object.keys(env)) {
+    const m = k.match(/^MSLXDFF_(.+)_KEY$/);
+    if (m) {
+      const id = m[1].toLowerCase().replace(/__/g, "-");
+      if (id !== "opencode") genericIds.add(id);
+    }
+  }
+
+  for (const gid of [...genericIds].sort()) {
+    const cfg = configs[gid];
+    let keys = [];
+    let baseUrl = "";
+    let share = false;
+    let allowed = [];
+    try { keys = loadProviderKeys(gid, stateFile ? { file: stateFile } : undefined); } catch {}
+    try { baseUrl = loadProviderBaseUrl(gid, stateFile ? { file: stateFile } : undefined) || cfg?.baseUrl || (gid === "openrouter" ? "https://openrouter.ai/api/v1" : gid === "workbuddy" ? "https://copilot.tencent.com" : ""); } catch { baseUrl = cfg?.baseUrl || ""; }
+    try { share = loadProviderShareKeys(gid, stateFile ? { file: stateFile } : undefined); } catch {}
+    try { allowed = loadProviderAllowedModels(gid, stateFile ? { file: stateFile } : undefined); } catch {}
+    let enabled = Boolean(baseUrl && keys.length) || (gid === "openrouter" && keys.length > 0);
+    const auths = gid === "workbuddy" ? (() => { try { return loadProviderAuths(gid, stateFile ? { file: stateFile } : undefined) || []; } catch { return []; } })() : [];
+    let note = "";
+    const isWorkbuddyStub = gid === "workbuddy" && (keys.includes("k-new") || String(baseUrl).includes("127.0.0.1") || (keys.length === 1 && keys[0].length < 20));
+    if (isWorkbuddyStub) {
+      enabled = false;
+      note = "测试桩 (key=k-new, baseUrl=127.0.0.1) — 请重跑 node workbuddy-token-auto.js 写入真实 JWT";
+    } else if (!enabled) {
+      if (!baseUrl && !keys.length) note = "no baseUrl, no keys";
+      else if (!baseUrl) note = "missing baseUrl";
+      else if (!keys.length) note = "no keys";
+    } else if (gid === "workbuddy" && auths.length && auths.length !== keys.length) {
+      note = `${auths.length} auth(s) / ${keys.length} key(s) — 数量不一致请重跑 workbuddy-token-auto.js`;
+    }
+    rows.push({ id: gid, enabled, baseUrl: baseUrl || "(none)", keys, allowed, share, note, authCount: auths.length });
+  }
+  return rows;
+}
+
+export function formatProviderRow(p) {
+  const dot = p.enabled ? "●" : "○";
+  const state = p.enabled ? "enabled " : "disabled";
+  let keysInfo;
+  if (p.id === "opencode") keysInfo = "无需 key (内置)";
+  else keysInfo = p.keys.length ? `${p.keys.length} key${p.keys.length > 1 ? "s" : ""} ${p.keys.map((k) => `${k.slice(0, 3)}…${k.slice(-3)}`).join(", ")}` : "0 keys";
+  const authInfo = p.authCount ? ` ${p.authCount} acc` : "";
+  const allowInfo = p.allowed.length ? `allow=${p.allowed.length}(${p.allowed.slice(0, 2).join(",")}${p.allowed.length > 2 ? "…" : ""})` : "allow=all";
+  const shareInfo = p.id === "opencode" ? "cannot share" : `share=${p.share ? "ON" : "off"}`;
+  const note = p.note ? `  (${p.note})` : "";
+  return `  ${dot} ${p.id.padEnd(12)} ${state}  ${keysInfo}${authInfo}  ${allowInfo.padEnd(18)}  baseUrl=${p.baseUrl}  ${shareInfo}${note}`;
+}