自动选择模型

bot-test-bot · bot-test-bot · commit bade06ff1849 · 2025-03-15T10:35:07.000+08:00
diff --git a/src/renderer/components/ChatWindow.jsx b/src/renderer/components/ChatWindow.jsx
@@ -603,14 +603,76 @@ const ChatWindow = memo(({ session, onUpdateSession }) => {
       })
       .filter((item) => item.models.length > 0); // 只返回有启用模型的提供商
   };
+  const providerModels = getProviderModels();
 
   // 获取当前选择的模型
   const getCurrentModel = () => {
+    // 首先检查配置中是否有providerId和modelId
     if (!config.providerId || !config.modelId) return null;
+    // 检查当前选择的提供商是否存在于可用提供商列表中
+    const provider = providerModels.find(
+      (item) => item.provider.id === config.providerId
+    );
+    if (!provider) return null;
+
+    // 检查当前选择的模型是否存在于可用模型列表中
+    const modelExists = provider.models.some(
+      (model) => model.id === config.modelId
+    );
+    if (!modelExists) return null;
+
+    // 只有当提供商和模型都存在且启用时，才返回完整的选择值
     return `${config.providerId}|${config.modelId}`;
   };
 
-  const providerModels = getProviderModels();
+  // 检查当前选中的模型是否可用，如果不可用则选择第一个可用模型
+  useEffect(() => {
+    if (!config.providerId || !config.modelId || providerModels.length === 0)
+      return;
+
+    // 检查当前选择的提供商是否存在于可用提供商列表中
+    const provider = providerModels.find(
+      (item) => item.provider.id === config.providerId
+    );
+
+    // 如果提供商不存在或没有模型，选择第一个可用模型
+    if (!provider) {
+      // 如果有可用的提供商和模型，选择第一个
+      if (providerModels.length > 0 && providerModels[0].models.length > 0) {
+        const firstProvider = providerModels[0];
+        const firstModel = firstProvider.models[0];
+        handleModelChange(`${firstProvider.provider.id}|${firstModel.id}`);
+        console.log(`自动选择了第一个可用模型: ${firstModel.name}`);
+        message.info(t("chat.autoSelectedModel", { model: firstModel.name }));
+      }
+      return;
+    }
+
+    // 检查当前选择的模型是否存在于可用模型列表中
+    const modelExists = provider.models.some(
+      (model) => model.id === config.modelId
+    );
+
+    // 如果模型不存在，选择该提供商的第一个可用模型
+    if (!modelExists) {
+      if (provider.models.length > 0) {
+        const firstModel = provider.models[0];
+        handleModelChange(`${provider.provider.id}|${firstModel.id}`);
+        console.log(`当前模型不可用，自动选择了: ${firstModel.name}`);
+        message.info(t("chat.autoSelectedModel", { model: firstModel.name }));
+      } else if (
+        providerModels.length > 0 &&
+        providerModels[0].models.length > 0
+      ) {
+        // 如果该提供商没有可用模型，选择第一个可用的提供商的第一个模型
+        const firstProvider = providerModels[0];
+        const firstModel = firstProvider.models[0];
+        handleModelChange(`${firstProvider.provider.id}|${firstModel.id}`);
+        console.log(`自动选择了第一个可用模型: ${firstModel.name}`);
+        message.info(t("chat.autoSelectedModel", { model: firstModel.name }));
+      }
+    }
+  }, [config.providerId, config.modelId, providerModels, handleModelChange]);
 
   return (
     <div className="chat-window">
diff --git a/src/renderer/hooks/useUserConfig.js b/src/renderer/hooks/useUserConfig.js
@@ -277,24 +277,54 @@ if (typeof window !== "undefined") {
 export function isAIConfigured() {
   const config = getUserConfig();
   const providersConfig = getProvidersConfig();
-  console.log("providersConfig", providersConfig);
-  console.log("config", config);
+  console.log("检查AI配置:", { userConfig: config, providersConfig });
+
   // 检查是否选择了提供商和模型
   if (!config.providerId || !config.modelId) {
+    console.log("未选择提供商或模型");
     return false;
   }
 
   // 检查提供商是否存在且已启用
   const provider = providersConfig[config.providerId];
-  if (!provider || provider.enabled === false) {
+  if (!provider) {
+    console.log("提供商不存在:", config.providerId);
+    return false;
+  }
+
+  // 如果提供商被明确禁用，则返回false
+  if (provider.enabled === false) {
+    console.log("提供商被禁用:", config.providerId);
     return false;
   }
 
-  // 检查模型是否存在
-  const providerConfig = provider || {};
+  // 检查模型是否存在于配置中
+  const savedModels = provider.models || [];
+  const modelExists = savedModels.some(
+    (m) => m.id === config.modelId && m.enabled !== false && m.deleted !== true
+  );
+
+  if (!modelExists) {
+    console.log("模型不存在或被禁用:", config.modelId);
+
+    // 在这里不返回false，因为模型可能在系统定义但不在用户配置中
+    // 后续会通过getAllProviders自动选择合适的模型
+  }
+
+  // 如果是模拟响应模式(mock)或开发环境，不需要API密钥
+  if (provider.mockResponse || process.env.NODE_ENV === "development") {
+    return true;
+  }
+
+  // API密钥检查变为可选，如果提供商有baseUrl且能正常工作，即使没有API密钥也可以接受
+  // 对于某些提供商，如果baseUrl有效，可以不需要apiKey
+  if (provider.baseUrl) {
+    return true;
+  }
 
-  // 检查 API 密钥是否已设置
-  if (!providerConfig.apiKey && !providerConfig.mockResponse) {
+  // 如果都不满足，检查是否有API密钥
+  if (!provider.apiKey) {
+    console.log("提供商没有API密钥:", config.providerId);
     return false;
   }
 
diff --git a/src/renderer/i18n/locales/en/translation.json b/src/renderer/i18n/locales/en/translation.json
@@ -117,7 +117,6 @@
     "toolsFoundSuccessMsg": "Successfully retrieved {{count}} tools",
     "refreshToolsFailedMsg": "Failed to refresh tool list: {{message}}",
     "loadMessagesFailed": "Failed to load messages",
-    "sendMessageFailed": "Failed to send message",
     "missingProviderOrModel": "Please select a service provider and model first",
     "providerDisabled": "Service provider {{name}} is disabled, please enable it in settings first",
     "unsupportedProvider": "Unsupported provider: {{name}}",
@@ -144,7 +143,7 @@
     "addProvider": "Add Provider",
     "apiSettings": "API Settings",
     "apiSettingsInfo": "Provider API Configuration",
-    "apiSettingsDescription": "Configure the API authentication and endpoint for this service provider. You need to set the API key to use this provider's models.",
+    "apiSettingsDescription": "Configure the API authentication and endpoint for this service provider.",
     "apiBaseUrl": "API Base URL",
     "enterApiBaseUrl": "Enter API Base URL",
     "enterApiKey": "Enter API Key"
@@ -188,6 +187,8 @@
     "noToolCallsDirectComplete": "No tool calls, direct complete callback",
     "apiCallFailed": "AI API call failed",
     "requestCancelled": "Request cancelled by user",
+    "autoSelectedModel": "Auto-selected model: {{model}}",
+    "sendMessageFailed": "Failed to send message",
     "mcpTools": {
       "useTool": "Use MCP Tool",
       "loading": "Loading tools...",
diff --git a/src/renderer/i18n/locales/zh-CN/translation.json b/src/renderer/i18n/locales/zh-CN/translation.json
@@ -117,7 +117,6 @@
     "toolsFoundSuccessMsg": "成功获取到 {{count}} 个工具",
     "refreshToolsFailedMsg": "刷新工具列表失败：{{message}}",
     "loadMessagesFailed": "加载消息失败",
-    "sendMessageFailed": "发送消息失败",
     "missingProviderOrModel": "请先选择服务提供商和模型",
     "providerDisabled": "服务提供商 {{name}} 已被禁用，请先在设置中启用",
     "unsupportedProvider": "不支持的提供商: {{name}}",
@@ -144,7 +143,7 @@
     "addProvider": "添加供应商",
     "apiSettings": "API 设置",
     "apiSettingsInfo": "供应商 API 配置",
-    "apiSettingsDescription": "配置此服务提供商的 API 认证和端点。您需要设置 API 密钥才能使用该提供商的模型。",
+    "apiSettingsDescription": "配置此服务提供商的 API 认证和端点。",
     "apiBaseUrl": "API 基础 URL",
     "enterApiBaseUrl": "输入 API 基础 URL",
     "enterApiKey": "输入 API 密钥"
@@ -188,6 +187,8 @@
     "noToolCallsDirectComplete": "没有工具调用，直接调用onComplete回调",
     "apiCallFailed": "调用 AI API 失败",
     "requestCancelled": "请求被用户取消",
+    "autoSelectedModel": "已自动选择模型: {{model}}",
+    "sendMessageFailed": "发送消息失败",
     "mcpTools": {
       "useTool": "使用MCP工具",
       "loading": "加载工具中...",
