@vfvrpq/llm-cli 1.1.0 → 1.2.1

package/lib/http.js

@@ -0,0 +1,50 @@
+"use strict";
+// HTTP 请求（fetch 封装 + 错误映射）与 SSE 流解析。
+const { ApiError, friendlyHTTPError } = require("./errors");
+
+const STREAM_IDLE_MS = 120000; // 流式模式下两次数据块之间的最大等待（毫秒）
+const PLAIN_MS = 300000;       // 非流式模式的整体等待（毫秒）
+const TIMEOUT_REASON = "llm-cli-idle-timeout";
+
+function isAbort(err) {
+  return err && (err.name === "AbortError" || err.code === "ABORT_ERR");
+}
+
+async function apiFetch(url, apiKey, payload, method, signal, hints) {
+  const headers = { Authorization: `Bearer ${apiKey}`, "Content-Type": "application/json" };
+  let body;
+  if (payload !== null && payload !== undefined) {
+    body = JSON.stringify(payload);
+    if (payload.stream) headers.Accept = "text/event-stream";
+  }
+  let resp;
+  try {
+    resp = await fetch(url, { method, headers, body, signal });
+  } catch (err) {
+    if (isAbort(err)) throw err;
+    const cause = err.cause ? (err.cause.message || String(err.cause)) : err.message;
+    throw new ApiError(`无法连接服务器：${cause}（请检查网络或 --base-url）`);
+  }
+  if (!resp.ok) throw new ApiError(await friendlyHTTPError(resp, hints));
+  return resp;
+}
+
+// 解析 OpenAI 风格 SSE：逐行产出 data: 负载，遇到 [DONE] 结束
+async function* iterSSE(body) {
+  const decoder = new TextDecoder();
+  let buf = "";
+  for await (const chunk of body) {
+    buf += decoder.decode(chunk, { stream: true });
+    let idx;
+    while ((idx = buf.indexOf("\n")) >= 0) {
+      const line = buf.slice(0, idx).trim();
+      buf = buf.slice(idx + 1);
+      if (!line.startsWith("data:")) continue;
+      const data = line.slice(5).trim();
+      if (data === "[DONE]") return;
+      if (data) yield data;
+    }
+  }
+}
+
+module.exports = { STREAM_IDLE_MS, PLAIN_MS, TIMEOUT_REASON, isAbort, apiFetch, iterSSE };

package/lib/registry.js

@@ -0,0 +1,77 @@
+"use strict";
+// 供应商注册表操作：选择、模型名推断、运行时解析、Key 解析与掩码。
+const { ApiError, DEFAULT_HINTS } = require("./errors");
+
+function normalizeProvider(p, registry) {
+  const key = String(p || "").trim().toLowerCase();
+  const alias = { zhipu: "glm", moonshot: "kimi", xiaomi: "mimo", volc: "ark" };
+  const name = alias[key] || key;
+  if (!registry[name]) {
+    throw new ApiError(`未知供应商 "${p}"，可选：${Object.keys(registry).join(" / ")}`);
+  }
+  return name;
+}
+
+// 没显式指定 --provider 时，按模型名前缀推断（如 glm-* / deepseek-* / mimo-*）
+function inferProvider(model, registry) {
+  const m = String(model || "").toLowerCase();
+  for (const [id, pc] of Object.entries(registry)) {
+    if ((pc.prefixes || []).some((p) => m.startsWith(p))) return id;
+  }
+  return null;
+}
+
+function providerOf(args, cfg, ctx) {
+  if (ctx.fixedProvider) return ctx.fixedProvider;
+  if (args.provider) return normalizeProvider(args.provider, ctx.registry);
+  const inferred = ctx.features.provider ? inferProvider(args.model, ctx.registry) : null;
+  const fallback = cfg.default_provider || Object.keys(ctx.registry)[0];
+  return normalizeProvider(inferred || fallback, ctx.registry);
+}
+
+function resolveRuntime(args, cfg, ctx) {
+  const provider = providerOf(args, cfg, ctx);
+  const pc = ctx.registry[provider];
+  const conf = (cfg.providers && cfg.providers[provider]) || {};
+  let apiKey = args.api_key;
+  if (!apiKey) {
+    for (const k of pc.envKeys) {
+      const v = process.env[k];
+      if (v) { apiKey = v; break; }
+    }
+  }
+  if (!apiKey) apiKey = conf.api_key;
+  const model = args.model || conf.model || pc.defaultModel;
+  if (!model) {
+    throw new ApiError(`[${provider}] 没有内置默认模型，必须用 -m 指定。${pc.hint ? "\n" + pc.hint : ""}`);
+  }
+  return {
+    provider,
+    label: pc.label,
+    apiKey,
+    model,
+    baseUrl: args.base_url || conf.base_url || pc.defaultBase,
+    pc,
+    hints: { ...DEFAULT_HINTS, ...(pc.hints || {}) },
+  };
+}
+
+function requireKey(apiKey, rt) {
+  if (apiKey) return apiKey;
+  const pc = rt.pc;
+  throw new ApiError(
+    `尚未配置 [${rt.provider}] 的 API Key，请任选其一：\n` +
+    `  1. config set api-key${` --provider ${rt.provider}`}   （推荐）\n` +
+    `  2. 设置环境变量 ${pc.envKeys[0]}\n` +
+    `  3. 临时使用：--api-key <你的Key>\n` +
+    `Key 获取：${pc.keyUrl}`
+  );
+}
+
+function mask(key) {
+  if (!key) return "(未设置)";
+  if (key.length <= 8) return key.slice(0, 2) + "****";
+  return key.slice(0, 6) + "..." + key.slice(-4);
+}
+
+module.exports = { normalizeProvider, inferProvider, providerOf, resolveRuntime, requireKey, mask };

package/lib/run.js

@@ -0,0 +1,73 @@
+"use strict";
+// 入口引导：解析参数 → 加载配置 → 分发子命令。三个入口脚本都通过 run() 启动。
+const { parseArgs, printHelp } = require("./args");
+const { enableColors, colors } = require("./colors");
+const { ApiError } = require("./errors");
+const { loadUnified, saveUnified, resolveConfigPath, historyPathFor } = require("./config");
+const ask = require("./commands/ask");
+const chat = require("./commands/chat");
+const models = require("./commands/models");
+const configCmd = require("./commands/config-cmd");
+
+async function run(entry) {
+  if (typeof fetch === "undefined") {
+    console.error("需要 Node.js 18+（内置 fetch）。当前 Node 版本过旧，请升级后使用。");
+    process.exitCode = 1;
+    return;
+  }
+  if (process.platform === "win32") {
+    for (const s of [process.stdout, process.stderr]) {
+      try { s.reconfigure({ errors: "replace" }); } catch {}
+    }
+  }
+
+  let args;
+  try {
+    args = parseArgs(process.argv.slice(2), entry.features);
+  } catch (e) {
+    console.error(`${colors.red}${e.message}${colors.reset}`);
+    printHelp(entry);
+    process.exitCode = 1;
+    return;
+  }
+  enableColors(args.no_color);
+  if (args.version) {
+    console.log(`${entry.name} ${entry.version}`);
+    return;
+  }
+  if (args.help || args._.length === 0) {
+    printHelp(entry);
+    return;
+  }
+
+  const command = args._.shift();
+  const configFile = resolveConfigPath(entry);
+  const cfg = loadUnified(configFile, { singleProvider: entry.fixedProvider });
+  const ctx = {
+    entry,
+    registry: entry.registry,
+    features: entry.features,
+    fixedProvider: entry.fixedProvider,
+    configFile,
+    historyFile: historyPathFor(configFile),
+    saveCfg: (c) => saveUnified(configFile, c, { singleProvider: entry.fixedProvider }),
+  };
+
+  try {
+    if (command === "ask") await ask(args, cfg, ctx);
+    else if (command === "chat") await chat(args, cfg, ctx);
+    else if (command === "models") await models(args, cfg, ctx);
+    else if (command === "config") await configCmd(args, cfg, ctx);
+    else throw new ApiError(`未知命令 "${command}"，可用：ask / chat / models / config`);
+  } catch (e) {
+    if (e instanceof ApiError) {
+      console.error(`${colors.red}${e.message}${colors.reset}`);
+      process.exitCode = 1;
+    } else {
+      console.error(e);
+      process.exitCode = 1;
+    }
+  }
+}
+
+module.exports = { run };