mslxdff 0.1.160 → 0.1.161

package/src/bench/via-probe.js

@@ -1,125 +1,125 @@
-import { joinUrl } from "../providers/base.js";
-import { computeMetrics, extractUsageFromJson } from "../metrics.js";
-import { createTransport } from "../transport/index.js";
+import { joinUrl } from "../providers/base.js";
+import { computeMetrics, extractUsageFromJson } from "../metrics.js";
+import { createTransport } from "../transport/index.js";
 import { compatFetch } from "../compat.js";
-
-function extractInnerMessage(bodyText) {
-  const t = String(bodyText || "");
-  try {
-    const j = JSON.parse(t);
-    const m = j?.error?.message || j?.error || j?.message || j?.data?.error || "";
-    if (typeof m === "string" && m.trim()) return m.trim().slice(0, 300);
-    if (typeof j?.error === "string") return j.error.slice(0, 300);
-  } catch {}
-  return t.slice(0, 300);
-}
-
-function classifyError(status, bodyText) {
-  const t = String(bodyText || "").slice(0, 500);
-  const low = t.toLowerCase();
-  if (status === 401) return { label: "鉴权失败", retryable: false };
-  if (status === 402 || /insufficient balance/i.test(t)) return { label: "余额不足", retryable: false };
-  if (low.includes("only available via cline")) return { label: "仅 Cline 客户端可用", retryable: false };
-  if (low.includes("invalid model format")) return { label: "模型格式错误", retryable: false };
-  if (status === 403) return { label: /insufficient/i.test(t) ? "余额不足" : "鉴权失败", retryable: false };
-  if (status === 429) return { label: "限流", retryable: true };
-  if (status >= 500) return { label: `上游错误 ${status}`, retryable: true };
-  if (status === 404) return { label: "模型不存在", retryable: false };
-  return { label: `HTTP ${status}`, retryable: false };
-}
-
-export async function viaProbe({
-  peerUrl,
-  token,
-  providerId,
-  model,
-  prompt = "hi",
-  maxTokens = 5,
-  timeoutMs = 30000,
-  fetchImpl = compatFetch,
-  clock = Date.now,
-  shareKeys,
-  shareKeysHeader,
-  relayTarget,
-  relayHeaders,
-  relayBody,
-  targetUrl,
-} = {}) {
-  const base = String(peerUrl || "").replace(/\/+$/, "");
-  if (!base) return { ok: false, label: "配置错误", error: "missing peerUrl", ttfbMs: null, totalMs: 0, tps: null, charsPerSec: null, tokens: null };
-  const tr = createTransport({ fetchImpl, keepAlive: false, retry: {}, timeoutMs });
-  const rt = String(relayTarget || targetUrl || "").trim();
-  if (rt) {
-    const rh = relayHeaders && typeof relayHeaders === "object" ? relayHeaders : {};
-    const rb = relayBody !== undefined ? relayBody : null;
-    const relayUrl = joinUrl(base, "/v1/relay");
-    const relayHeadersOut = { "Content-Type": "application/json", Accept: "application/json" };
-    if (token) relayHeadersOut.Authorization = `Bearer ${token}`;
-    const payload = { targetUrl: rt, method: "POST", headers: rh, body: rb };
-    try {
-      const res = await tr.request({ url: relayUrl, method: "POST", headers: relayHeadersOut, body: payload, stream: false });
-      const ttfbMs = res.ttfbMs;
-      const totalMs = res.totalMs;
-      if (!res.ok) {
-        let txt = ""; try { txt = await res.text(); } catch {}
-        const cls = classifyError(res.status, txt);
-        const msg = extractInnerMessage(txt) || `HTTP ${res.status}`;
-        return { ok: false, status: res.status, label: cls.label, error: msg, ttfbMs, totalMs, tps: null, charsPerSec: null, tokens: null };
-      }
-      let txt = ""; let j = {};
-      try { txt = await res.text(); j = JSON.parse(txt); } catch { j = {}; }
-      const relayStatus = res.headers.get("x-mslxdff-relay-status") ? Number(res.headers.get("x-mslxdff-relay-status")) : res.status;
-      if (relayStatus >= 400) {
-        const cls = classifyError(relayStatus, txt);
-        const msg = extractInnerMessage(txt) || `HTTP ${relayStatus}`;
-        return { ok: false, status: relayStatus, label: cls.label, error: msg, ttfbMs, totalMs, tps: null, charsPerSec: null, tokens: null };
-      }
-      const usage = extractUsageFromJson(j);
-      const content = j?.choices?.[0]?.message?.content || j?.choices?.[0]?.text || txt || "";
-      const chars = typeof content === "string" ? content.length : 0;
-      const pt = usage?.prompt_tokens ?? null;
-      const ct = usage?.completion_tokens ?? null;
-      const { tps, charsPerSec } = computeMetrics({ ttfbMs, totalMs, promptTokens: pt, completionTokens: ct, chars });
-      const totalTokens = usage?.total_tokens ?? (pt !== null && ct !== null ? pt + ct : null);
-      return { ok: true, status: relayStatus, label: "成功", ttfbMs, totalMs, tps, charsPerSec, tokens: { prompt: pt, completion: ct, total: totalTokens }, chars };
-    } catch (e) {
-      const msg = e?.message || String(e);
-      const isTimeout = /timeout|abort/i.test(msg);
-      return { ok: false, label: isTimeout ? "超时" : "网络错误", error: msg.slice(0, 300), ttfbMs: null, totalMs: null, tps: null, charsPerSec: null, tokens: null };
-    }
-  }
-  if (!model) return { ok: false, label: "配置错误", error: "missing model", ttfbMs: null, totalMs: 0, tps: null, charsPerSec: null, tokens: null };
-  let rawModel = String(model).trim();
-  if (providerId && rawModel.startsWith(`${providerId}/`)) rawModel = rawModel.slice(providerId.length + 1);
-  const url = joinUrl(base, "/v1/chat/completions");
-  const body = { model: rawModel, stream: false, messages: [{ role: "user", content: prompt }], max_tokens: maxTokens };
-  const headers = { "Content-Type": "application/json", Accept: "application/json" };
-  if (token) headers.Authorization = `Bearer ${token}`;
-  const sk = shareKeysHeader || shareKeys;
-  if (sk) headers["x-mslxdff-share-keys"] = String(sk);
-  try {
-    const res = await tr.request({ url, method: "POST", headers, body, stream: false });
-    const ttfbMs = res.ttfbMs;
-    const totalMs = res.totalMs;
-    if (!res.ok) {
-      let txt = ""; try { txt = await res.text(); } catch {}
-      const cls = classifyError(res.status, txt);
-      const msg = extractInnerMessage(txt) || `HTTP ${res.status}`;
-      return { ok: false, status: res.status, label: cls.label, error: msg, ttfbMs, totalMs, tps: null, charsPerSec: null, tokens: null };
-    }
-    let json = {}; let txt = "";
-    try { txt = await res.text(); json = JSON.parse(txt); } catch { json = {}; }
-    const usage = extractUsageFromJson(json);
-    const content = json?.choices?.[0]?.message?.content || json?.choices?.[0]?.text || txt || "";
-    const chars = typeof content === "string" ? content.length : 0;
-    const promptTokens = usage?.prompt_tokens ?? null;
-    const completionTokens = usage?.completion_tokens ?? null;
-    const { tps, charsPerSec } = computeMetrics({ ttfbMs, totalMs, promptTokens, completionTokens, chars });
-    const totalTokens = usage?.total_tokens ?? (promptTokens !== null && completionTokens !== null ? promptTokens + completionTokens : null);
-    return { ok: true, status: res.status, label: "成功", ttfbMs, totalMs, tps, charsPerSec, tokens: { prompt: promptTokens, completion: completionTokens, total: totalTokens }, chars };
-  } catch (e) {
-    const msg = e?.message || String(e);
-    const isTimeout = /timeout|abort/i.test(msg);
-    return { ok: false, label: isTimeout ? "超时" : "网络错误", error: msg.slice(0, 300), ttfbMs: null, totalMs: null, tps: null, charsPerSec: null, tokens: null };
-  }
-}
+
+function extractInnerMessage(bodyText) {
+  const t = String(bodyText || "");
+  try {
+    const j = JSON.parse(t);
+    const m = j?.error?.message || j?.error || j?.message || j?.data?.error || "";
+    if (typeof m === "string" && m.trim()) return m.trim().slice(0, 300);
+    if (typeof j?.error === "string") return j.error.slice(0, 300);
+  } catch {}
+  return t.slice(0, 300);
+}
+
+function classifyError(status, bodyText) {
+  const t = String(bodyText || "").slice(0, 500);
+  const low = t.toLowerCase();
+  if (status === 401) return { label: "鉴权失败", retryable: false };
+  if (status === 402 || /insufficient balance/i.test(t)) return { label: "余额不足", retryable: false };
+  if (low.includes("only available via cline")) return { label: "仅 Cline 客户端可用", retryable: false };
+  if (low.includes("invalid model format")) return { label: "模型格式错误", retryable: false };
+  if (status === 403) return { label: /insufficient/i.test(t) ? "余额不足" : "鉴权失败", retryable: false };
+  if (status === 429) return { label: "限流", retryable: true };
+  if (status >= 500) return { label: `上游错误 ${status}`, retryable: true };
+  if (status === 404) return { label: "模型不存在", retryable: false };
+  return { label: `HTTP ${status}`, retryable: false };
+}
+
+export async function viaProbe({
+  peerUrl,
+  token,
+  providerId,
+  model,
+  prompt = "hi",
+  maxTokens = 5,
+  timeoutMs = 30000,
+  fetchImpl = compatFetch,
+  clock = Date.now,
+  shareKeys,
+  shareKeysHeader,
+  relayTarget,
+  relayHeaders,
+  relayBody,
+  targetUrl,
+} = {}) {
+  const base = String(peerUrl || "").replace(/\/+$/, "");
+  if (!base) return { ok: false, label: "配置错误", error: "missing peerUrl", ttfbMs: null, totalMs: 0, tps: null, charsPerSec: null, tokens: null };
+  const tr = createTransport({ fetchImpl, keepAlive: false, retry: {}, timeoutMs });
+  const rt = String(relayTarget || targetUrl || "").trim();
+  if (rt) {
+    const rh = relayHeaders && typeof relayHeaders === "object" ? relayHeaders : {};
+    const rb = relayBody !== undefined ? relayBody : null;
+    const relayUrl = joinUrl(base, "/v1/relay");
+    const relayHeadersOut = { "Content-Type": "application/json", Accept: "application/json" };
+    if (token) relayHeadersOut.Authorization = `Bearer ${token}`;
+    const payload = { targetUrl: rt, method: "POST", headers: rh, body: rb };
+    try {
+      const res = await tr.request({ url: relayUrl, method: "POST", headers: relayHeadersOut, body: payload, stream: false });
+      const ttfbMs = res.ttfbMs;
+      const totalMs = res.totalMs;
+      if (!res.ok) {
+        let txt = ""; try { txt = await res.text(); } catch {}
+        const cls = classifyError(res.status, txt);
+        const msg = extractInnerMessage(txt) || `HTTP ${res.status}`;
+        return { ok: false, status: res.status, label: cls.label, error: msg, ttfbMs, totalMs, tps: null, charsPerSec: null, tokens: null };
+      }
+      let txt = ""; let j = {};
+      try { txt = await res.text(); j = JSON.parse(txt); } catch { j = {}; }
+      const relayStatus = res.headers.get("x-mslxdff-relay-status") ? Number(res.headers.get("x-mslxdff-relay-status")) : res.status;
+      if (relayStatus >= 400) {
+        const cls = classifyError(relayStatus, txt);
+        const msg = extractInnerMessage(txt) || `HTTP ${relayStatus}`;
+        return { ok: false, status: relayStatus, label: cls.label, error: msg, ttfbMs, totalMs, tps: null, charsPerSec: null, tokens: null };
+      }
+      const usage = extractUsageFromJson(j);
+      const content = j?.choices?.[0]?.message?.content || j?.choices?.[0]?.text || txt || "";
+      const chars = typeof content === "string" ? content.length : 0;
+      const pt = usage?.prompt_tokens ?? null;
+      const ct = usage?.completion_tokens ?? null;
+      const { tps, charsPerSec } = computeMetrics({ ttfbMs, totalMs, promptTokens: pt, completionTokens: ct, chars });
+      const totalTokens = usage?.total_tokens ?? (pt !== null && ct !== null ? pt + ct : null);
+      return { ok: true, status: relayStatus, label: "成功", ttfbMs, totalMs, tps, charsPerSec, tokens: { prompt: pt, completion: ct, total: totalTokens }, chars };
+    } catch (e) {
+      const msg = e?.message || String(e);
+      const isTimeout = /timeout|abort/i.test(msg);
+      return { ok: false, label: isTimeout ? "超时" : "网络错误", error: msg.slice(0, 300), ttfbMs: null, totalMs: null, tps: null, charsPerSec: null, tokens: null };
+    }
+  }
+  if (!model) return { ok: false, label: "配置错误", error: "missing model", ttfbMs: null, totalMs: 0, tps: null, charsPerSec: null, tokens: null };
+  let rawModel = String(model).trim();
+  if (providerId && rawModel.startsWith(`${providerId}/`)) rawModel = rawModel.slice(providerId.length + 1);
+  const url = joinUrl(base, "/v1/chat/completions");
+  const body = { model: rawModel, stream: false, messages: [{ role: "user", content: prompt }], max_tokens: maxTokens };
+  const headers = { "Content-Type": "application/json", Accept: "application/json" };
+  if (token) headers.Authorization = `Bearer ${token}`;
+  const sk = shareKeysHeader || shareKeys;
+  if (sk) headers["x-mslxdff-share-keys"] = String(sk);
+  try {
+    const res = await tr.request({ url, method: "POST", headers, body, stream: false });
+    const ttfbMs = res.ttfbMs;
+    const totalMs = res.totalMs;
+    if (!res.ok) {
+      let txt = ""; try { txt = await res.text(); } catch {}
+      const cls = classifyError(res.status, txt);
+      const msg = extractInnerMessage(txt) || `HTTP ${res.status}`;
+      return { ok: false, status: res.status, label: cls.label, error: msg, ttfbMs, totalMs, tps: null, charsPerSec: null, tokens: null };
+    }
+    let json = {}; let txt = "";
+    try { txt = await res.text(); json = JSON.parse(txt); } catch { json = {}; }
+    const usage = extractUsageFromJson(json);
+    const content = json?.choices?.[0]?.message?.content || json?.choices?.[0]?.text || txt || "";
+    const chars = typeof content === "string" ? content.length : 0;
+    const promptTokens = usage?.prompt_tokens ?? null;
+    const completionTokens = usage?.completion_tokens ?? null;
+    const { tps, charsPerSec } = computeMetrics({ ttfbMs, totalMs, promptTokens, completionTokens, chars });
+    const totalTokens = usage?.total_tokens ?? (promptTokens !== null && completionTokens !== null ? promptTokens + completionTokens : null);
+    return { ok: true, status: res.status, label: "成功", ttfbMs, totalMs, tps, charsPerSec, tokens: { prompt: promptTokens, completion: completionTokens, total: totalTokens }, chars };
+  } catch (e) {
+    const msg = e?.message || String(e);
+    const isTimeout = /timeout|abort/i.test(msg);
+    return { ok: false, label: isTimeout ? "超时" : "网络错误", error: msg.slice(0, 300), ttfbMs: null, totalMs: null, tps: null, charsPerSec: null, tokens: null };
+  }
+}

package/src/bench/via-routes.js

@@ -1,87 +1,87 @@
-import { readFileSync, existsSync } from "node:fs";
-import { join, dirname } from "node:path";
-import os from "node:os";
-import { defaultStateFile } from "../state/store.js";
-import { atomicWriteSync } from "../state/persist.js";
-
-export function defaultViaRoutesFile() {
-  if (process.env.MSLXDFF_VIA_ROUTES_FILE) return String(process.env.MSLXDFF_VIA_ROUTES_FILE).trim();
-  const sf = defaultStateFile();
-  return join(dirname(sf), "via-routes.json");
-}
-
-function viaTtlMs() {
-  const raw = process.env.MSLXDFF_VIA_ROUTE_TTL_MS;
-  if (raw === undefined || raw === null || raw === "") return 0;
-  const s = String(raw).trim().toLowerCase();
-  if (s === "0" || s === "off" || s === "false" || s === "no" || s === "disable" || s === "disabled") return 0;
-  const n = Number(s);
-  if (Number.isInteger(n) && n >= 0) return n;
-  return 0;
-}
-
-export function loadViaRoutes(file) {
-  const f = file || defaultViaRoutesFile();
-  try {
-    if (!existsSync(f)) return { version: 1, at: null, routes: {}, meta: {} };
-    const j = JSON.parse(readFileSync(f, "utf8"));
-    if (j && typeof j === "object" && j.routes && typeof j.routes === "object") return j;
-    if (j && typeof j === "object" && !j.routes) return { version: 1, at: j.at || null, routes: j, meta: {} };
-    return { version: 1, at: null, routes: {}, meta: {} };
-  } catch {
-    return { version: 1, at: null, routes: {}, meta: {} };
-  }
-}
-
-export function getViaRoute(model, { file, ttlMs } = {}) {
-  const id = String(model || "").trim();
-  if (!id) return null;
-  const f = file || defaultViaRoutesFile();
-  const data = loadViaRoutes(f);
-  const entry = data.routes?.[id];
-  if (!entry) return null;
-  const t = ttlMs !== undefined ? ttlMs : viaTtlMs();
-  if (t > 0 && entry.at) {
-    const atMs = Date.parse(entry.at);
-    if (Number.isFinite(atMs) && Date.now() - atMs > t) return null;
-  }
-  return entry;
-}
-
-export function saveViaRoutes(results, { file, meta } = {}) {
-  const f = file || defaultViaRoutesFile();
-  const now = new Date().toISOString();
-  const prev = loadViaRoutes(f);
-  const nextRoutes = { ...(prev.routes || {}) };
-  for (const r of results || []) {
-    const id = String(r.model || r.id || "").trim();
-    if (!id) continue;
-    const best = String(r.best || "direct").trim() || "direct";
-    const direct = r.direct ? { ok: Boolean(r.direct.ok), ttfbMs: r.direct.ttfbMs ?? r.direct.totalMs ?? null, totalMs: r.direct.totalMs ?? null, label: r.direct.label || null, error: r.direct.error ? String(r.direct.error).slice(0, 300) : null } : null;
-    const via = {};
-    for (const [k, v] of Object.entries(r.via || {})) {
-      via[k] = v?.ok ? { ok: true, ttfbMs: v.ttfbMs ?? v.totalMs ?? null, totalMs: v.totalMs ?? null } : { ok: false, ttfbMs: v?.ttfbMs ?? null, totalMs: v?.totalMs ?? null, label: v?.label || v?.error || "offline" };
-    }
-    nextRoutes[id] = {
-      best,
-      direct,
-      via,
-      deltaMs: r.deltaMs ?? null,
-      provider: r.provider || id.split("/")[0] || "",
-      at: now,
-    };
-  }
-  const out = {
-    version: 1,
-    at: now,
-    routes: nextRoutes,
-    meta: meta || prev.meta || {},
-  };
-  atomicWriteSync(f, out);
-  return out;
-}
-
-export function clearViaRoutes(file) {
-  const f = file || defaultViaRoutesFile();
-  atomicWriteSync(f, { version: 1, at: new Date().toISOString(), routes: {}, meta: {} });
-}
+import { readFileSync, existsSync } from "node:fs";
+import { join, dirname } from "node:path";
+import os from "node:os";
+import { defaultStateFile } from "../state/store.js";
+import { atomicWriteSync } from "../state/persist.js";
+
+export function defaultViaRoutesFile() {
+  if (process.env.MSLXDFF_VIA_ROUTES_FILE) return String(process.env.MSLXDFF_VIA_ROUTES_FILE).trim();
+  const sf = defaultStateFile();
+  return join(dirname(sf), "via-routes.json");
+}
+
+function viaTtlMs() {
+  const raw = process.env.MSLXDFF_VIA_ROUTE_TTL_MS;
+  if (raw === undefined || raw === null || raw === "") return 0;
+  const s = String(raw).trim().toLowerCase();
+  if (s === "0" || s === "off" || s === "false" || s === "no" || s === "disable" || s === "disabled") return 0;
+  const n = Number(s);
+  if (Number.isInteger(n) && n >= 0) return n;
+  return 0;
+}
+
+export function loadViaRoutes(file) {
+  const f = file || defaultViaRoutesFile();
+  try {
+    if (!existsSync(f)) return { version: 1, at: null, routes: {}, meta: {} };
+    const j = JSON.parse(readFileSync(f, "utf8"));
+    if (j && typeof j === "object" && j.routes && typeof j.routes === "object") return j;
+    if (j && typeof j === "object" && !j.routes) return { version: 1, at: j.at || null, routes: j, meta: {} };
+    return { version: 1, at: null, routes: {}, meta: {} };
+  } catch {
+    return { version: 1, at: null, routes: {}, meta: {} };
+  }
+}
+
+export function getViaRoute(model, { file, ttlMs } = {}) {
+  const id = String(model || "").trim();
+  if (!id) return null;
+  const f = file || defaultViaRoutesFile();
+  const data = loadViaRoutes(f);
+  const entry = data.routes?.[id];
+  if (!entry) return null;
+  const t = ttlMs !== undefined ? ttlMs : viaTtlMs();
+  if (t > 0 && entry.at) {
+    const atMs = Date.parse(entry.at);
+    if (Number.isFinite(atMs) && Date.now() - atMs > t) return null;
+  }
+  return entry;
+}
+
+export function saveViaRoutes(results, { file, meta } = {}) {
+  const f = file || defaultViaRoutesFile();
+  const now = new Date().toISOString();
+  const prev = loadViaRoutes(f);
+  const nextRoutes = { ...(prev.routes || {}) };
+  for (const r of results || []) {
+    const id = String(r.model || r.id || "").trim();
+    if (!id) continue;
+    const best = String(r.best || "direct").trim() || "direct";
+    const direct = r.direct ? { ok: Boolean(r.direct.ok), ttfbMs: r.direct.ttfbMs ?? r.direct.totalMs ?? null, totalMs: r.direct.totalMs ?? null, label: r.direct.label || null, error: r.direct.error ? String(r.direct.error).slice(0, 300) : null } : null;
+    const via = {};
+    for (const [k, v] of Object.entries(r.via || {})) {
+      via[k] = v?.ok ? { ok: true, ttfbMs: v.ttfbMs ?? v.totalMs ?? null, totalMs: v.totalMs ?? null } : { ok: false, ttfbMs: v?.ttfbMs ?? null, totalMs: v?.totalMs ?? null, label: v?.label || v?.error || "offline" };
+    }
+    nextRoutes[id] = {
+      best,
+      direct,
+      via,
+      deltaMs: r.deltaMs ?? null,
+      provider: r.provider || id.split("/")[0] || "",
+      at: now,
+    };
+  }
+  const out = {
+    version: 1,
+    at: now,
+    routes: nextRoutes,
+    meta: meta || prev.meta || {},
+  };
+  atomicWriteSync(f, out);
+  return out;
+}
+
+export function clearViaRoutes(file) {
+  const f = file || defaultViaRoutesFile();
+  atomicWriteSync(f, { version: 1, at: new Date().toISOString(), routes: {}, meta: {} });
+}

package/src/bench/workbuddy-bench.js

@@ -1,55 +1,55 @@
-import { computeMetrics } from "../metrics.js";
-import { createTransport } from "../transport/index.js";
+import { computeMetrics } from "../metrics.js";
+import { createTransport } from "../transport/index.js";
 import { compatFetch } from "../compat.js";
-
-function buildWorkbuddyHeaders(apiKey, auth) {
-  const h = {
-    "Content-Type": "application/json",
-    Accept: "text/event-stream",
-    "User-Agent": "CLI/2.115.0 WorkBuddy/2.115.0",
-    Origin: "https://www.codebuddy.cn",
-    Referer: "https://www.codebuddy.cn/",
-    "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;
-}
-
-function sseContent(obj) {
-  const c = obj?.choices?.[0]?.delta?.content || obj?.choices?.[0]?.message?.content || "";
-  return typeof c === "string" ? c : "";
-}
-
-export async function workbuddyBenchOne({ baseUrl, chatPath = "/v2/chat/completions", model, apiKey, auth, prompt = "hi", maxTokens = 5, timeoutMs = 30000, fetchImpl = compatFetch }) {
-  const tr = createTransport({ fetchImpl, keepAlive: false, retry: {}, timeoutMs });
-  let ttfbMs = null;
-  let totalMs = null;
-  let content = "";
-  try {
-    const url = String(baseUrl).replace(/\/+$/, "") + String(chatPath || "/v2/chat/completions");
-    const headers = buildWorkbuddyHeaders(apiKey, auth);
-    const rawModel = String(model || "").trim();
-    const body = { model: rawModel, stream: true, messages: [{ role: "user", content: prompt }], max_tokens: maxTokens };
-    const res = await tr.request({ url, method: "POST", headers, body, stream: true });
-    if (!res.ok) {
-      let txt = "";
-      try { txt = await res.text(); } catch {}
-      const label = res.status === 401 ? "鉴权失败" : res.status === 402 ? "余额不足" : res.status === 429 ? "限流" : res.status >= 500 ? `上游错误 ${res.status}` : `HTTP ${res.status}`;
-      return { id: model, ok: false, status: res.status, label, error: txt.slice(0, 300), ttfbMs, totalMs: res.totalMs, tps: null, charsPerSec: null, tokens: null };
-    }
-    for await (const ev of res.stream()) {
-      try { content += sseContent(JSON.parse(ev)); } catch {}
-    }
-    ttfbMs = res.ttfbMs;
-    totalMs = res.totalMs;
-    const chars = content.length;
-    const { tps, charsPerSec } = computeMetrics({ ttfbMs, totalMs, promptTokens: null, completionTokens: null, chars });
-    return { id: model, ok: true, status: 200, label: "成功", ttfbMs, totalMs, tps, charsPerSec, tokens: null, chars };
-  } catch (e) {
-    const msg = e?.message || String(e);
-    return { id: model, ok: false, label: /timeout|abort/i.test(msg) ? "超时" : "网络错误", error: msg.slice(0, 300), ttfbMs, totalMs, tps: null, charsPerSec: null, tokens: null };
-  }
-}
+
+function buildWorkbuddyHeaders(apiKey, auth) {
+  const h = {
+    "Content-Type": "application/json",
+    Accept: "text/event-stream",
+    "User-Agent": "CLI/2.115.0 WorkBuddy/2.115.0",
+    Origin: "https://www.codebuddy.cn",
+    Referer: "https://www.codebuddy.cn/",
+    "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;
+}
+
+function sseContent(obj) {
+  const c = obj?.choices?.[0]?.delta?.content || obj?.choices?.[0]?.message?.content || "";
+  return typeof c === "string" ? c : "";
+}
+
+export async function workbuddyBenchOne({ baseUrl, chatPath = "/v2/chat/completions", model, apiKey, auth, prompt = "hi", maxTokens = 5, timeoutMs = 30000, fetchImpl = compatFetch }) {
+  const tr = createTransport({ fetchImpl, keepAlive: false, retry: {}, timeoutMs });
+  let ttfbMs = null;
+  let totalMs = null;
+  let content = "";
+  try {
+    const url = String(baseUrl).replace(/\/+$/, "") + String(chatPath || "/v2/chat/completions");
+    const headers = buildWorkbuddyHeaders(apiKey, auth);
+    const rawModel = String(model || "").trim();
+    const body = { model: rawModel, stream: true, messages: [{ role: "user", content: prompt }], max_tokens: maxTokens };
+    const res = await tr.request({ url, method: "POST", headers, body, stream: true });
+    if (!res.ok) {
+      let txt = "";
+      try { txt = await res.text(); } catch {}
+      const label = res.status === 401 ? "鉴权失败" : res.status === 402 ? "余额不足" : res.status === 429 ? "限流" : res.status >= 500 ? `上游错误 ${res.status}` : `HTTP ${res.status}`;
+      return { id: model, ok: false, status: res.status, label, error: txt.slice(0, 300), ttfbMs, totalMs: res.totalMs, tps: null, charsPerSec: null, tokens: null };
+    }
+    for await (const ev of res.stream()) {
+      try { content += sseContent(JSON.parse(ev)); } catch {}
+    }
+    ttfbMs = res.ttfbMs;
+    totalMs = res.totalMs;
+    const chars = content.length;
+    const { tps, charsPerSec } = computeMetrics({ ttfbMs, totalMs, promptTokens: null, completionTokens: null, chars });
+    return { id: model, ok: true, status: 200, label: "成功", ttfbMs, totalMs, tps, charsPerSec, tokens: null, chars };
+  } catch (e) {
+    const msg = e?.message || String(e);
+    return { id: model, ok: false, label: /timeout|abort/i.test(msg) ? "超时" : "网络错误", error: msg.slice(0, 300), ttfbMs, totalMs, tps: null, charsPerSec: null, tokens: null };
+  }
+}