@ganziliang/kb-model-setup 0.1.0 → 0.1.2

package/dist/index.js

@@ -1,19 +1,113 @@
+const gatewayHost = "https://llm-gateway.zhizhengroup.com";
+const statsUrl = `${gatewayHost}/apiStats/api/user-stats`;
 const protocols = ["openai-responses", "anthropic-messages"];
+function parseApiId(value) {
+    const input = value.trim();
+    if (!input)
+        throw new Error("apiId 不能为空");
+    if (input.includes("apiId=")) {
+        try {
+            const apiId = new URL(input).searchParams.get("apiId");
+            if (apiId)
+                return apiId;
+        }
+        catch {
+            throw new Error("无法解析 api-stats 页面地址");
+        }
+    }
+    if (!/^[0-9a-f-]{20,}$/i.test(input)) {
+        throw new Error("apiId 格式不正确，请粘贴 api-stats 地址或 apiId 参数值");
+    }
+    return input;
+}
+function modelId(value) {
+    if (typeof value === "string")
+        return value.trim() || undefined;
+    if (!value || typeof value !== "object")
+        return undefined;
+    const item = value;
+    const id = item.id || item.model || item.modelId || item.name;
+    return typeof id === "string" && id.trim() ? id.trim() : undefined;
+}
+function modelList(value) {
+    if (Array.isArray(value))
+        return value;
+    if (!value || typeof value !== "object")
+        return [];
+    return Object.values(value).flatMap((entry) => Array.isArray(entry) ? entry : [entry]);
+}
+function availableModels(payload) {
+    const data = payload.data;
+    const models = data?.availableModels;
+    const toModels = (value) => {
+        const result = [];
+        const seen = new Set();
+        for (const entry of modelList(value)) {
+            const id = modelId(entry);
+            if (!id || seen.has(id))
+                continue;
+            seen.add(id);
+            result.push({ id, ...(entry && typeof entry === "object" ? entry : {}) });
+        }
+        return result;
+    };
+    return {
+        openai: toModels(models?.openai).filter((model) => model.id.toLowerCase().startsWith("gpt")),
+        anthropic: toModels(models?.anthropic ?? models?.availableModels).filter((model) => !model.id.toLowerCase().startsWith("gpt")),
+    };
+}
+async function fetchAvailableModels(apiId) {
+    const response = await fetch(statsUrl, {
+        method: "POST",
+        headers: {
+            accept: "application/json, text/plain, */*",
+            "content-type": "application/json",
+            origin: gatewayHost,
+            referer: `${gatewayHost}/admin-next/api-stats?apiId=${encodeURIComponent(apiId)}`,
+        },
+        body: JSON.stringify({ apiId }),
+    });
+    const body = await response.text();
+    let payload;
+    try {
+        payload = JSON.parse(body);
+    }
+    catch {
+        throw new Error(`模型查询接口返回了非 JSON 响应（HTTP ${response.status}）`);
+    }
+    if (!response.ok) {
+        const item = payload && typeof payload === "object" ? payload : {};
+        throw new Error(`模型查询失败（HTTP ${response.status}）${item.message ? `：${String(item.message)}` : ""}`);
+    }
+    if (!payload || typeof payload !== "object")
+        throw new Error("模型查询接口返回格式不正确");
+    const result = payload;
+    if (result.success === false)
+        throw new Error(String(result.message || result.error || "模型查询失败"));
+    return availableModels(result);
+}
 export function createModelSetupExtension() {
     return {
-        async configure(prompter, existing) {
-            const provider = await prompter.ask("Provider", existing?.provider ?? "");
-            const apiIndex = await prompter.select("API protocol", protocols, Math.max(0, protocols.indexOf(existing?.api ?? protocols[0])));
-            const model = await prompter.ask("Model", existing?.model ?? "");
-            const baseURL = await prompter.ask("Base URL", existing?.baseURL ?? "");
-            const apiKey = await prompter.ask("API key", existing?.apiKey ?? "");
-            const config = { provider, api: protocols[apiIndex], model, baseURL, apiKey };
-            const errors = this.validate(config);
-            if (errors.length) {
-                await prompter.notice(errors.join("\n"));
-                return this.configure(prompter, config);
-            }
-            return config;
+        async configure(prompter) {
+            const apiIdInput = await prompter.ask("粘贴 api-stats 页面地址或 apiId：", "https://llm-gateway.zhizhengroup.com/admin-next/api-stats?apiId=...");
+            if (!apiIdInput.trim())
+                throw new Error("未提供 apiId");
+            const apiId = parseApiId(apiIdInput);
+            await prompter.notice("正在查询账户和可用模型……");
+            const models = await fetchAvailableModels(apiId);
+            const preferredOpenAI = models.openai.find((model) => model.id.toLowerCase() === "gpt-5.6-luna") ?? models.openai[0];
+            const selected = preferredOpenAI
+                ? { provider: "company-gpt", api: "openai-responses", model: preferredOpenAI.id, baseURL: `${gatewayHost}/openai` }
+                : models.anthropic[0]
+                    ? { provider: "company-anthropic", api: "anthropic-messages", model: models.anthropic[0].id, baseURL: `${gatewayHost}/api` }
+                    : undefined;
+            if (!selected)
+                throw new Error("接口返回中没有发现可用模型");
+            await prompter.notice(`已发现模型，默认使用 ${selected.provider}/${selected.model}`);
+            const apiKey = await prompter.ask("请输入 API Key：", "API Key");
+            if (!apiKey.trim())
+                throw new Error("API Key 不能为空");
+            return { ...selected, apiKey: apiKey.trim() };
         },
         validate(config) {
             const errors = [];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ganziliang/kb-model-setup",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "Interactive model configuration for the kb local knowledge base agent",
   "type": "module",
   "main": "dist/index.js",
@@ -11,13 +11,17 @@
       "import": "./dist/index.js"
     }
   },
-  "files": ["dist"],
+  "files": [
+    "dist"
+  ],
   "scripts": {
     "build": "tsc -p tsconfig.json",
     "typecheck": "tsc -p tsconfig.json --noEmit",
     "test": "npm run build && node --test"
   },
-  "engines": { "node": ">=22.5.0" },
+  "engines": {
+    "node": ">=22.5.0"
+  },
   "devDependencies": {
     "@types/node": "^22.10.0",
     "typescript": "^5.7.2"