opencode-codebuddy-external-auth 1.0.16 → 1.0.18

package/dist/plugin.js

@@ -91,10 +91,111 @@ function resolveModel(inputModel) {
         return CONFIG.defaultModel;
     return inputModel || "";
 }
-function buildAuthHeaders(accessToken) {
-    const tenantId = resolveTenantId(accessToken);
-    const enterpriseId = resolveEnterpriseId(accessToken);
-    const userId = resolveUserId(accessToken);
+let cachedTenantId = "";
+let cachedEnterpriseId = "";
+let cachedUserId = "";
+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 { enterpriseId: "", models: [] };
+    try {
+        const url = `${CONFIG.serverUrl}/console/enterprises/${tenantId}/config/models`;
+        const response = await fetch(url, {
+            method: "GET",
+            headers: {
+                "Accept": "application/json, text/plain, */*",
+                "X-Requested-With": "XMLHttpRequest",
+                "Authorization": `Bearer ${accessToken}`,
+            },
+        });
+        if (!response.ok) {
+            return { enterpriseId: "", models: [] };
+        }
+        const data = await response.json();
+        const enterpriseId = data?.data?.enterpriseId || data?.enterpriseId || "";
+        const models = extractModelIds(data);
+        return { enterpriseId, models };
+    }
+    catch {
+        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) {
+        const config = await fetchConfigModels(accessToken, tenantId);
+        enterpriseId = config.enterpriseId || enterpriseId;
+        if (config.models.length) {
+            cachedModelIds = config.models;
+        }
+    }
+    if (tenantId)
+        cachedTenantId = tenantId;
+    if (enterpriseId)
+        cachedEnterpriseId = enterpriseId;
+    if (userId)
+        cachedUserId = userId;
     if (!tenantId && !warnedTenantId) {
         warnedTenantId = true;
         console.warn("[codebuddy-external] 未获取到 X-Tenant-Id，请设置 CODEBUDDY_TENANT_ID");
@@ -343,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,
@@ -354,9 +461,10 @@ async function executeViaAuthApi(openaiRequest, auth) {
         stream: openaiRequest.stream ?? true,
     };
     const doRequest = async (token) => {
+        const headers = await buildAuthHeaders(token);
         const response = await fetch(`${CONFIG.serverUrl}${CONFIG.chatCompletionsPath}`, {
             method: "POST",
-            headers: buildAuthHeaders(token),
+            headers,
             body: JSON.stringify(requestBody),
         });
         return response;

package/package.json

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