@lobehub/chat 1.91.1 → 1.91.3

package/src/libs/model-runtime/utils/modelParse.ts

@@ -0,0 +1,186 @@
+import type { ChatModelCard } from '@/types/llm';
+
+export interface ModelProcessorConfig {
+  excludeKeywords?: readonly string[]; // 对符合的模型不添加标签
+  functionCallKeywords?: readonly string[];
+  reasoningKeywords?: readonly string[];
+  visionKeywords?: readonly string[];
+}
+
+// 模型标签关键词配置
+export const MODEL_LIST_CONFIGS = {
+  anthropic: {
+    functionCallKeywords: ['claude'],
+    reasoningKeywords: ['-3-7', '3.7', '-4'],
+    visionKeywords: ['claude'],
+  },
+  deepseek: {
+    functionCallKeywords: ['v3', 'r1'],
+    reasoningKeywords: ['r1'],
+  },
+  google: {
+    functionCallKeywords: ['gemini'],
+    reasoningKeywords: ['thinking', '-2.5-'],
+    visionKeywords: ['gemini', 'learnlm'],
+  },
+  llama: {
+    functionCallKeywords: ['llama-3.2', 'llama-3.3', 'llama-4'],
+    reasoningKeywords: [],
+    visionKeywords: [],
+  },
+  openai: {
+    excludeKeywords: ['audio'],
+    functionCallKeywords: ['4o', '4.1', 'o3', 'o4'],
+    reasoningKeywords: ['o1', 'o3', 'o4'],
+    visionKeywords: ['4o', '4.1', 'o4'],
+  },
+  qwen: {
+    functionCallKeywords: [
+      'qwen-max',
+      'qwen-plus',
+      'qwen-turbo',
+      'qwen-long',
+      'qwen1.5',
+      'qwen2',
+      'qwen2.5',
+      'qwen3',
+    ],
+    reasoningKeywords: ['qvq', 'qwq', 'qwen3'],
+    visionKeywords: ['qvq', 'vl'],
+  },
+  volcengine: {
+    functionCallKeywords: ['doubao-1.5'],
+    reasoningKeywords: ['thinking', '-r1'],
+    visionKeywords: ['vision', '-m'],
+  },
+  zeroone: {
+    functionCallKeywords: ['fc'],
+    visionKeywords: ['vision'],
+  },
+  zhipu: {
+    functionCallKeywords: ['glm-4', 'glm-z1'],
+    reasoningKeywords: ['glm-zero', 'glm-z1'],
+    visionKeywords: ['glm-4v'],
+  },
+} as const;
+
+// 模型提供商关键词配置
+export const PROVIDER_DETECTION_CONFIG = {
+  anthropic: ['claude'],
+  deepseek: ['deepseek'],
+  google: ['gemini'],
+  llama: ['llama'],
+  openai: ['o1', 'o3', 'o4', 'gpt-'],
+  qwen: ['qwen', 'qwq', 'qvq'],
+  volcengine: ['doubao'],
+  zeroone: ['yi-'],
+  zhipu: ['glm'],
+} as const;
+
+/**
+ * 检测单个模型的提供商类型
+ * @param modelId 模型ID
+ * @returns 检测到的提供商配置键名，默认为 'openai'
+ */
+export const detectModelProvider = (modelId: string): keyof typeof MODEL_LIST_CONFIGS => {
+  const lowerModelId = modelId.toLowerCase();
+
+  for (const [provider, keywords] of Object.entries(PROVIDER_DETECTION_CONFIG)) {
+    const hasKeyword = keywords.some((keyword) => lowerModelId.includes(keyword));
+
+    if (hasKeyword && provider in MODEL_LIST_CONFIGS) {
+      return provider as keyof typeof MODEL_LIST_CONFIGS;
+    }
+  }
+
+  return 'openai';
+};
+
+/**
+ * 处理模型卡片的通用逻辑
+ */
+const processModelCard = (
+  model: { [key: string]: any; id: string },
+  config: ModelProcessorConfig,
+  knownModel?: any,
+): ChatModelCard => {
+  const {
+    functionCallKeywords = [],
+    visionKeywords = [],
+    reasoningKeywords = [],
+    excludeKeywords = [],
+  } = config;
+
+  const isExcludedModel = excludeKeywords.some((keyword) =>
+    model.id.toLowerCase().includes(keyword),
+  );
+
+  return {
+    contextWindowTokens: model.contextWindowTokens ?? knownModel?.contextWindowTokens ?? undefined,
+    displayName: model.displayName ?? knownModel?.displayName ?? model.id,
+    enabled: knownModel?.enabled || false,
+    functionCall:
+      (functionCallKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) &&
+        !isExcludedModel) ||
+      knownModel?.abilities?.functionCall ||
+      false,
+    id: model.id,
+    maxOutput: model.maxOutput ?? knownModel?.maxOutput ?? undefined,
+    reasoning:
+      reasoningKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
+      knownModel?.abilities?.reasoning ||
+      false,
+    vision:
+      (visionKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) &&
+        !isExcludedModel) ||
+      knownModel?.abilities?.vision ||
+      false,
+  };
+};
+
+/**
+ * 处理单一提供商的模型列表
+ * @param modelList 模型列表
+ * @param config 提供商配置
+ * @returns 处理后的模型卡片列表
+ */
+export const processModelList = async (
+  modelList: Array<{ id: string }>,
+  config: ModelProcessorConfig,
+): Promise<ChatModelCard[]> => {
+  const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+
+  return modelList
+    .map((model) => {
+      const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
+        (m) => model.id.toLowerCase() === m.id.toLowerCase(),
+      );
+
+      return processModelCard(model, config, knownModel);
+    })
+    .filter(Boolean);
+};
+
+/**
+ * 处理混合提供商的模型列表
+ * @param modelList 模型列表
+ * @returns 处理后的模型卡片列表
+ */
+export const processMultiProviderModelList = async (
+  modelList: Array<{ id: string }>,
+): Promise<ChatModelCard[]> => {
+  const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+
+  return modelList
+    .map((model) => {
+      const detectedProvider = detectModelProvider(model.id);
+      const config = MODEL_LIST_CONFIGS[detectedProvider];
+
+      const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
+        (m) => model.id.toLowerCase() === m.id.toLowerCase(),
+      );
+
+      return processModelCard(model, config, knownModel);
+    })
+    .filter(Boolean);
+};

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

@@ -1,5 +1,10 @@
 import { ModelProvider } from '../types';
 import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
+import { MODEL_LIST_CONFIGS, processModelList } from '../utils/modelParse';
+
+export interface VolcengineModelCard {
+  id: string;
+}
 
 export const LobeVolcengineAI = createOpenAICompatibleRuntime({
   baseURL: 'https://ark.cn-beijing.volces.com/api/v3',
@@ -24,5 +29,11 @@ export const LobeVolcengineAI = createOpenAICompatibleRuntime({
   debug: {
     chatCompletion: () => process.env.DEBUG_VOLCENGINE_CHAT_COMPLETION === '1',
   },
+  models: async ({ client }) => {
+    const modelsPage = (await client.models.list()) as any;
+    const modelList: VolcengineModelCard[] = modelsPage.data;
+
+    return processModelList(modelList, MODEL_LIST_CONFIGS.volcengine);
+  },
   provider: ModelProvider.Volcengine,
 });

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

@@ -1,6 +1,5 @@
-import type { ChatModelCard } from '@/types/llm';
-
 import { ModelProvider } from '../types';
+import { MODEL_LIST_CONFIGS, processModelList } from '../utils/modelParse';
 import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface ZeroOneModelCard {
@@ -13,30 +12,10 @@ export const LobeZeroOneAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_ZEROONE_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
-
     const modelsPage = (await client.models.list()) as any;
     const modelList: ZeroOneModelCard[] = 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:
-            model.id.toLowerCase().includes('fc') || knownModel?.abilities?.functionCall || false,
-          id: model.id,
-          reasoning: knownModel?.abilities?.reasoning || false,
-          vision:
-            model.id.toLowerCase().includes('vision') || knownModel?.abilities?.vision || false,
-        };
-      })
-      .filter(Boolean) as ChatModelCard[];
+    return processModelList(modelList, MODEL_LIST_CONFIGS.zeroone);
   },
   provider: ModelProvider.ZeroOne,
 });

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

@@ -1,6 +1,5 @@
-import type { ChatModelCard } from '@/types/llm';
-
 import { ModelProvider } from '../types';
+import { MODEL_LIST_CONFIGS, processModelList } from '../utils/modelParse';
 import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface ZhipuModelCard {
@@ -60,10 +59,6 @@ export const LobeZhipuAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_ZHIPU_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
-
-    const reasoningKeywords = ['glm-zero', 'glm-z1'];
-
     // ref: https://open.bigmodel.cn/console/modelcenter/square
     const url = 'https://open.bigmodel.cn/api/fine-tuning/model_center/list?pageSize=100&pageNum=1';
     const response = await fetch(url, {
@@ -78,34 +73,12 @@ export const LobeZhipuAI = createOpenAICompatibleRuntime({
 
     const modelList: ZhipuModelCard[] = json.rows;
 
-    return modelList
-      .map((model) => {
-        const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
-          (m) => model.modelCode.toLowerCase() === m.id.toLowerCase(),
-        );
-
-        return {
-          contextWindowTokens: knownModel?.contextWindowTokens ?? undefined,
-          description: model.description,
-          displayName: model.modelName,
-          enabled: knownModel?.enabled || false,
-          functionCall:
-            (model.modelCode.toLowerCase().includes('glm-4') &&
-              !model.modelCode.toLowerCase().includes('glm-4v')) ||
-            knownModel?.abilities?.functionCall ||
-            false,
-          id: model.modelCode,
-          reasoning:
-            reasoningKeywords.some((keyword) => model.modelCode.toLowerCase().includes(keyword)) ||
-            knownModel?.abilities?.reasoning ||
-            false,
-          vision:
-            model.modelCode.toLowerCase().includes('glm-4v') ||
-            knownModel?.abilities?.vision ||
-            false,
-        };
-      })
-      .filter(Boolean) as ChatModelCard[];
+    const standardModelList = modelList.map((model) => ({
+      description: model.description,
+      displayName: model.modelName,
+      id: model.modelCode,
+    }));
+    return processModelList(standardModelList, MODEL_LIST_CONFIGS.zhipu);
   },
   provider: ModelProvider.ZhiPu,
 });