opencode-codebuddy-external-auth 1.0.17 → 1.0.18

package/dist/plugin.js

@@ -94,9 +94,52 @@ function resolveModel(inputModel) {
 let cachedTenantId = "";
 let cachedEnterpriseId = "";
 let cachedUserId = "";
-async function fetchEnterpriseId(accessToken, tenantId) {
+let cachedModelIds = [];
+let warnedModelFallback = false;
+function extractModelIds(payload) {
+    const results = [];
+    const collect = (list) => {
+        if (!Array.isArray(list))
+            return;
+        for (const item of list) {
+            if (!item)
+                continue;
+            if (typeof item === "string") {
+                results.push(item);
+                continue;
+            }
+            if (typeof item === "object") {
+                const id = item.id ||
+                    item.model ||
+                    item.modelId ||
+                    item.model_id ||
+                    item.code ||
+                    item.name;
+                if (typeof id === "string")
+                    results.push(id);
+                if (Array.isArray(item.models))
+                    collect(item.models);
+            }
+        }
+    };
+    const candidates = [
+        payload?.data?.models,
+        payload?.data?.modelList,
+        payload?.data?.availableModels,
+        payload?.models,
+        payload?.modelList,
+        payload?.availableModels,
+        payload?.data?.modelGroups,
+        payload?.modelGroups,
+    ];
+    for (const list of candidates) {
+        collect(list);
+    }
+    return Array.from(new Set(results)).filter(Boolean);
+}
+async function fetchConfigModels(accessToken, tenantId) {
     if (!tenantId)
-        return "";
+        return { enterpriseId: "", models: [] };
     try {
         const url = `${CONFIG.serverUrl}/console/enterprises/${tenantId}/config/models`;
         const response = await fetch(url, {
@@ -108,21 +151,44 @@ async function fetchEnterpriseId(accessToken, tenantId) {
             },
         });
         if (!response.ok) {
-            return "";
+            return { enterpriseId: "", models: [] };
         }
         const data = await response.json();
-        return data?.data?.enterpriseId || data?.enterpriseId || "";
+        const enterpriseId = data?.data?.enterpriseId || data?.enterpriseId || "";
+        const models = extractModelIds(data);
+        return { enterpriseId, models };
     }
     catch {
-        return "";
+        return { enterpriseId: "", models: [] };
+    }
+}
+async function getSupportedModels(accessToken, tenantId) {
+    if (cachedModelIds.length > 0)
+        return cachedModelIds;
+    const config = await fetchConfigModels(accessToken, tenantId);
+    if (config.enterpriseId && !cachedEnterpriseId) {
+        cachedEnterpriseId = config.enterpriseId;
+    }
+    if (config.models.length) {
+        cachedModelIds = config.models;
     }
+    return cachedModelIds;
+}
+function pickFallbackModel(models) {
+    if (models.includes("glm-4.7-ioa"))
+        return "glm-4.7-ioa";
+    return models[0] || "";
 }
 async function buildAuthHeaders(accessToken) {
     const tenantId = cachedTenantId || resolveTenantId(accessToken);
     let enterpriseId = cachedEnterpriseId || resolveEnterpriseId(accessToken);
     const userId = cachedUserId || resolveUserId(accessToken);
     if (!enterpriseId) {
-        enterpriseId = await fetchEnterpriseId(accessToken, tenantId);
+        const config = await fetchConfigModels(accessToken, tenantId);
+        enterpriseId = config.enterpriseId || enterpriseId;
+        if (config.models.length) {
+            cachedModelIds = config.models;
+        }
     }
     if (tenantId)
         cachedTenantId = tenantId;
@@ -378,10 +444,16 @@ async function executeViaAuthApi(openaiRequest, auth) {
         throw new Error("缺少 access token，无法调用 CodeBuddy API");
     }
     let accessToken = auth.access;
+    const tenantId = resolveTenantId(accessToken);
     const resolvedModel = resolveModel(openaiRequest.model);
+    const supportedModels = await getSupportedModels(accessToken, tenantId);
     if (!resolvedModel) {
-        throw new Error("未设置模型，请设置 CODEBUDDY_DEFAULT_MODEL 或在 OpenCode 选择模型");
+        throw new Error("未设置模型，请在 OpenCode 选择模型");
+    }
+    if (supportedModels.length && !supportedModels.includes(resolvedModel)) {
+        throw new Error(`[codebuddy-external] 模型 ${resolvedModel} 不在可用列表，请更换模型`);
     }
+    console.log(`[codebuddy-external] 使用模型: ${resolvedModel}`);
     const requestBody = {
         ...openaiRequest,
         model: resolvedModel,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-codebuddy-external-auth",
-  "version": "1.0.17",
+  "version": "1.0.18",
   "description": "OpenCode plugin for CodeBuddy External (IOA) authentication",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",