@lobehub/chat 1.91.1 → 1.91.2

package/src/libs/model-runtime/openai/index.ts

@@ -1,8 +1,7 @@
-import type { ChatModelCard } from '@/types/llm';
-
 import { ModelProvider } from '../types';
 import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 import { pruneReasoningPayload } from '../utils/openaiHelpers';
+import { processMultiProviderModelList } from '../utils/modelParse';
 
 export interface OpenAIModelCard {
   id: string;
@@ -45,45 +44,11 @@ export const LobeOpenAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_OPENAI_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
-
-    const functionCallKeywords = ['4o', '4.1', 'o3', 'o4'];
-
-    const visionKeywords = ['4o', '4.1', 'o4'];
-
-    const reasoningKeywords = ['o1', 'o3', 'o4'];
-
     const modelsPage = (await client.models.list()) as any;
     const modelList: OpenAIModelCard[] = modelsPage.data;
 
-    return modelList
-      .map((model) => {
-        const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
-          (m) => model.id.toLowerCase() === m.id.toLowerCase(),
-        );
-
-        return {
-          contextWindowTokens: knownModel?.contextWindowTokens ?? undefined,
-          displayName: knownModel?.displayName ?? undefined,
-          enabled: knownModel?.enabled || false,
-          functionCall:
-            (functionCallKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) &&
-              !model.id.toLowerCase().includes('audio')) ||
-            knownModel?.abilities?.functionCall ||
-            false,
-          id: model.id,
-          reasoning:
-            reasoningKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
-            knownModel?.abilities?.reasoning ||
-            false,
-          vision:
-            (visionKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) &&
-              !model.id.toLowerCase().includes('audio')) ||
-            knownModel?.abilities?.vision ||
-            false,
-        };
-      })
-      .filter(Boolean) as ChatModelCard[];
+    // 自动检测模型提供商并选择相应配置
+    return processMultiProviderModelList(modelList);
   },
   provider: ModelProvider.OpenAI,
 });

package/src/libs/model-runtime/openrouter/__snapshots__/index.test.ts.snap

@@ -13,6 +13,7 @@ _These are free, rate-limited endpoints for [Reflection 70B](/models/mattshumer/
     "enabled": false,
     "functionCall": true,
     "id": "mattshumer/reflection-70b:free",
+    "maxOutput": undefined,
     "maxTokens": 4096,
     "pricing": {
       "input": 0,
@@ -38,6 +39,7 @@ _These are free, rate-limited endpoints for [Reflection 70B](/models/mattshumer/
     "enabled": false,
     "functionCall": false,
     "id": "mattshumer/reflection-70b:free",
+    "maxOutput": undefined,
     "maxTokens": 4096,
     "pricing": {
       "input": 0,
@@ -63,6 +65,7 @@ _These are free, rate-limited endpoints for [Reflection 70B](/models/mattshumer/
     "enabled": false,
     "functionCall": false,
     "id": "mattshumer/reflection-70b:free",
+    "maxOutput": undefined,
     "maxTokens": 4096,
     "pricing": {
       "input": 0,

package/src/libs/model-runtime/openrouter/index.ts

@@ -1,6 +1,7 @@
 import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
+import { processMultiProviderModelList } from '../utils/modelParse';
 import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 import { OpenRouterModelCard, OpenRouterModelExtraInfo, OpenRouterReasoning } from './type';
 
@@ -40,20 +41,9 @@ export const LobeOpenRouterAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_OPENROUTER_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
-
-    const reasoningKeywords = [
-      'deepseek/deepseek-r1',
-      'openai/o1',
-      'openai/o3',
-      'qwen/qvq',
-      'qwen/qwq',
-      'thinking',
-    ];
-
     const modelsPage = (await client.models.list()) as any;
     const modelList: OpenRouterModelCard[] = modelsPage.data;
-
+  
     const modelsExtraInfo: OpenRouterModelExtraInfo[] = [];
     try {
       const response = await fetch('https://openrouter.ai/api/frontend/models');
@@ -62,50 +52,51 @@ export const LobeOpenRouterAI = createOpenAICompatibleRuntime({
         modelsExtraInfo.push(...data['data']);
       }
     } catch (error) {
-      // 忽略 fetch 错误，使用空的 modelsExtraInfo 数组继续处理
       console.error('Failed to fetch OpenRouter frontend models:', error);
     }
-
-    return modelList
-      .map((model) => {
-        const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
-          (m) => model.id.toLowerCase() === m.id.toLowerCase(),
-        );
-        const extraInfo = modelsExtraInfo.find(
-          (m) => m.slug.toLowerCase() === model.id.toLowerCase(),
-        );
-
-        return {
-          contextWindowTokens: model.context_length,
-          description: model.description,
-          displayName: model.name,
-          enabled: knownModel?.enabled || false,
-          functionCall:
-            model.description.includes('function calling') ||
-            model.description.includes('tools') ||
-            extraInfo?.endpoint?.supports_tool_parameters ||
-            knownModel?.abilities?.functionCall ||
-            false,
-          id: model.id,
-          maxTokens:
-            typeof model.top_provider.max_completion_tokens === 'number'
-              ? model.top_provider.max_completion_tokens
-              : undefined,
-          pricing: {
-            input: formatPrice(model.pricing.prompt),
-            output: formatPrice(model.pricing.completion),
-          },
-          reasoning:
-            reasoningKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
-            extraInfo?.endpoint?.supports_reasoning ||
-            knownModel?.abilities?.reasoning ||
-            false,
-          releasedAt: new Date(model.created * 1000).toISOString().split('T')[0],
-          vision:
-            model.architecture.modality.includes('image') || knownModel?.abilities?.vision || false,
-        };
-      })
-      .filter(Boolean) as ChatModelCard[];
+  
+    // 解析模型能力
+    const baseModels = await processMultiProviderModelList(modelList);
+  
+    // 合并 OpenRouter 获取的模型信息
+    return baseModels.map((baseModel) => {
+      const model = modelList.find(m => m.id === baseModel.id);
+      const extraInfo = modelsExtraInfo.find(
+        (m) => m.slug.toLowerCase() === baseModel.id.toLowerCase(),
+      );
+  
+      if (!model) return baseModel;
+  
+      return {
+        ...baseModel,
+        contextWindowTokens: model.context_length,
+        description: model.description,
+        displayName: model.name,
+        functionCall:
+          baseModel.functionCall ||
+          model.description.includes('function calling') ||
+          model.description.includes('tools') ||
+          extraInfo?.endpoint?.supports_tool_parameters ||
+          false,
+        maxTokens:
+          typeof model.top_provider.max_completion_tokens === 'number'
+            ? model.top_provider.max_completion_tokens
+            : undefined,
+        pricing: {
+          input: formatPrice(model.pricing.prompt),
+          output: formatPrice(model.pricing.completion),
+        },
+        reasoning:
+          baseModel.reasoning ||
+          extraInfo?.endpoint?.supports_reasoning ||
+          false,
+        releasedAt: new Date(model.created * 1000).toISOString().split('T')[0],
+        vision:
+          baseModel.vision ||
+          model.architecture.modality.includes('image') ||
+          false,
+      };
+    }).filter(Boolean) as ChatModelCard[];
   },
   provider: ModelProvider.OpenRouter,
 });

package/src/libs/model-runtime/qwen/index.ts

@@ -1,6 +1,5 @@
-import type { ChatModelCard } from '@/types/llm';
-
 import { ModelProvider } from '../types';
+import { processMultiProviderModelList } from '../utils/modelParse';
 import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 import { QwenAIStream } from '../utils/streams';
 
@@ -78,52 +77,10 @@ export const LobeQwenAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_QWEN_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
-
-    const functionCallKeywords = [
-      'qwen-max',
-      'qwen-plus',
-      'qwen-turbo',
-      'qwen-long',
-      'qwen1.5',
-      'qwen2',
-      'qwen2.5',
-      'qwen3',
-    ];
-
-    const visionKeywords = ['qvq', 'vl'];
-
-    const reasoningKeywords = ['qvq', 'qwq', 'deepseek-r1', 'qwen3'];
-
     const modelsPage = (await client.models.list()) as any;
     const modelList: QwenModelCard[] = modelsPage.data;
 
-    return modelList
-      .map((model) => {
-        const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
-          (m) => model.id.toLowerCase() === m.id.toLowerCase(),
-        );
-
-        return {
-          contextWindowTokens: knownModel?.contextWindowTokens ?? undefined,
-          displayName: knownModel?.displayName ?? undefined,
-          enabled: knownModel?.enabled || false,
-          functionCall:
-            functionCallKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
-            knownModel?.abilities?.functionCall ||
-            false,
-          id: model.id,
-          reasoning:
-            reasoningKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
-            knownModel?.abilities?.reasoning ||
-            false,
-          vision:
-            visionKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
-            knownModel?.abilities?.vision ||
-            false,
-        };
-      })
-      .filter(Boolean) as ChatModelCard[];
+    return processMultiProviderModelList(modelList);
   },
   provider: ModelProvider.Qwen,
 });

package/src/libs/model-runtime/siliconcloud/index.ts

@@ -1,7 +1,6 @@
-import type { ChatModelCard } from '@/types/llm';
-
 import { AgentRuntimeErrorType } from '../error';
 import { ChatCompletionErrorPayload, ModelProvider } from '../types';
+import { processMultiProviderModelList } from '../utils/modelParse';
 import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface SiliconCloudModelCard {
@@ -69,58 +68,10 @@ export const LobeSiliconCloudAI = createOpenAICompatibleRuntime({
     invalidAPIKey: AgentRuntimeErrorType.InvalidProviderAPIKey,
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
-
-    const functionCallKeywords = [
-      'qwen/qwen3',
-      'qwen/qwen2.5',
-      'thudm/glm-4',
-      'deepseek-ai/deepseek',
-      'internlm/internlm2_5',
-      'meta-llama/meta-llama-3.1',
-      'meta-llama/meta-llama-3.3',
-    ];
-
-    const visionKeywords = [
-      'opengvlab/internvl',
-      'qwen/qvq',
-      'qwen/qwen2-vl',
-      'teleai/telemm',
-      'deepseek-ai/deepseek-vl',
-    ];
-
-    const reasoningKeywords = ['deepseek-ai/deepseek-r1', 'qwen/qvq', 'qwen/qwq', 'qwen/qwen3'];
-
     const modelsPage = (await client.models.list()) as any;
     const modelList: SiliconCloudModelCard[] = modelsPage.data;
 
-    return modelList
-      .map((model) => {
-        const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
-          (m) => model.id.toLowerCase() === m.id.toLowerCase(),
-        );
-
-        return {
-          contextWindowTokens: knownModel?.contextWindowTokens ?? undefined,
-          displayName: knownModel?.displayName ?? undefined,
-          enabled: knownModel?.enabled || false,
-          functionCall:
-            (functionCallKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) &&
-              !model.id.toLowerCase().includes('deepseek-r1')) ||
-            knownModel?.abilities?.functionCall ||
-            false,
-          id: model.id,
-          reasoning:
-            reasoningKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
-            knownModel?.abilities?.reasoning ||
-            false,
-          vision:
-            visionKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
-            knownModel?.abilities?.vision ||
-            false,
-        };
-      })
-      .filter(Boolean) as ChatModelCard[];
+    return processMultiProviderModelList(modelList);
   },
   provider: ModelProvider.SiliconCloud,
 });