@lobehub/chat 1.91.0 → 1.91.2

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/utils/streams/anthropic.ts

@@ -1,7 +1,7 @@
 import Anthropic from '@anthropic-ai/sdk';
 import type { Stream } from '@anthropic-ai/sdk/streaming';
 
-import { ModelTokensUsage, CitationItem } from '@/types/message';
+import { CitationItem, ModelTokensUsage } from '@/types/message';
 
 import { ChatStreamCallbacks } from '../../types';
 import {
@@ -180,7 +180,7 @@ export const transformAnthropicStream = (
             context.returnedCitationArray.push({
               title: citations.title,
               url: citations.url,
-            } as CitationItem)
+            } as CitationItem);
           }
 
           return { data: null, id: context.id, type: 'text' };
@@ -219,15 +219,16 @@ export const transformAnthropicStream = (
 
     case 'message_stop': {
       return [
-        ...(context.returnedCitationArray?.length 
-          ? [{ 
-              data: { citations: context.returnedCitationArray }, 
-              id: context.id, 
-              type: 'grounding' 
-            }] 
-          : []
-        ),
-        { data: 'message_stop', id: context.id, type: 'stop' }
+        ...(context.returnedCitationArray?.length
+          ? [
+              {
+                data: { citations: context.returnedCitationArray },
+                id: context.id,
+                type: 'grounding',
+              },
+            ]
+          : []),
+        { data: 'message_stop', id: context.id, type: 'stop' },
       ] as any;
     }
 

package/src/libs/model-runtime/utils/streams/openai.ts

@@ -227,10 +227,12 @@ export const transformOpenAIStream = (
             return [
               {
                 data: {
-                  citations: (citations as any[]).map((item) => ({
-                    title: typeof item === 'string' ? item : item.title,
-                    url: typeof item === 'string' ? item : item.url || item.link,
-                  })).filter(c => c.title && c.url), // Zhipu 内建搜索工具有时会返回空 link 引发程序崩溃
+                  citations: (citations as any[])
+                    .map((item) => ({
+                      title: typeof item === 'string' ? item : item.title,
+                      url: typeof item === 'string' ? item : item.url || item.link,
+                    }))
+                    .filter((c) => c.title && c.url), // Zhipu 内建搜索工具有时会返回空 link 引发程序崩溃
                 },
                 id: chunk.id,
                 type: 'grounding',

package/src/libs/model-runtime/utils/streams/protocol.ts

@@ -18,7 +18,7 @@ export interface StreamContext {
   returnedCitation?: boolean;
   /**
    * Claude's citations are inline and interleaved with text output.
-   * Each text segment may carry references to sources (e.g., web search results) 
+   * Each text segment may carry references to sources (e.g., web search results)
    * relevant to that specific portion of the generated content.
    * This array accumulates all citation items received during the streaming response.
    */

package/src/libs/model-runtime/utils/streams/spark.test.ts

@@ -40,7 +40,7 @@ describe('SparkAIStream', () => {
           },
         ],
       },
-    ]
+    ];
 
     const mockSparkStream = new ReadableStream({
       start(controller) {

package/src/libs/model-runtime/utils/streams/spark.ts

@@ -2,6 +2,7 @@ import OpenAI from 'openai';
 import type { Stream } from 'openai/streaming';
 
 import { ChatStreamCallbacks } from '../../types';
+import { convertUsage } from '../usageConverter';
 import {
   StreamProtocolChunk,
   StreamProtocolToolCallChunk,
@@ -11,8 +12,6 @@ import {
   generateToolCallId,
 } from './protocol';
 
-import { convertUsage } from '../usageConverter';
-
 export function transformSparkResponseToStream(data: OpenAI.ChatCompletion) {
   return new ReadableStream({
     start(controller) {

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,
 });

package/src/middleware.ts

@@ -136,7 +136,7 @@ const defaultMiddleware = (request: NextRequest) => {
 
 const isPublicRoute = createRouteMatcher([
   '/api/auth(.*)',
-  '/trpc/edge(.*)',
+  '/trpc(.*)',
   // next auth
   '/next-auth/(.*)',
   // clerk

package/src/server/services/user/index.ts

@@ -100,16 +100,15 @@ export class UserService {
     const s3 = new S3();
     const s3FileUrl = `user/avatar/${id}/${image}`;
 
-    try{
+    try {
       const file = await s3.getFileByteArray(s3FileUrl);
       if (!file) {
         return null;
       }
       const fileBuffer = Buffer.from(file);
       return fileBuffer;
-    }
-    catch (error) {
+    } catch (error) {
       pino.error('Failed to get user avatar:', error);
     }
-  }
+  };
 }

package/src/services/__tests__/assistant.test.ts

@@ -36,7 +36,7 @@ describe('AssistantService', () => {
       // Arrange
       const fakeResponse = { agents: [{ name: 'TestAssisstant' }] };
       (globalHelpers.getCurrentLanguage as Mock).mockReturnValue('tt');
-      
+
       (edgeClient.market.getAgentIndex.query as Mock).mockResolvedValue(fakeResponse);
 
       // Act
@@ -50,21 +50,19 @@ describe('AssistantService', () => {
     it('should handle fetch error', async () => {
       // Arrange
       (globalHelpers.getCurrentLanguage as Mock).mockReturnValue('en');
-      (edgeClient.market.getAgentIndex.query as Mock).mockRejectedValue(
-        new Error('Network error'),
-      );
+      (edgeClient.market.getAgentIndex.query as Mock).mockRejectedValue(new Error('Network error'));
 
       // Act & Assert
       await expect(assistantService.getAssistantList()).rejects.toThrow('Network error');
     });
   });
-  
+
   describe('getAssistantById', () => {
     it('should fetch and return the assistant by id', async () => {
       // Arrange
       const fakeResponse = { identifier: 'test-assisstant' };
       (globalHelpers.getCurrentLanguage as Mock).mockReturnValue('tt');
-      
+
       (edgeClient.market.getAgent.query as Mock).mockResolvedValue(fakeResponse);
 
       // Act

package/src/services/__tests__/tool.test.ts

@@ -49,7 +49,9 @@ describe('ToolService', () => {
     it('should handle fetch error', async () => {
       // Arrange
       (globalHelpers.getCurrentLanguage as Mock).mockReturnValue('en');
-      (edgeClient.market.getPluginIndex.query as Mock).mockRejectedValue(new Error('Network error'));
+      (edgeClient.market.getPluginIndex.query as Mock).mockRejectedValue(
+        new Error('Network error'),
+      );
 
       // Act & Assert
       await expect(toolService.getToolList()).rejects.toThrow('Network error');

package/src/store/user/slices/auth/selectors.ts

@@ -2,9 +2,9 @@ import { t } from 'i18next';
 
 import { enableAuth, enableClerk, enableNextAuth } from '@/const/auth';
 import { BRANDING_NAME } from '@/const/branding';
+import { isDesktop } from '@/const/version';
 import { UserStore } from '@/store/user';
 import { LobeUser } from '@/types/user';
-import { isDesktop } from '@/const/version';
 
 const DEFAULT_USERNAME = BRANDING_NAME;
 

package/src/store/user/slices/common/action.test.ts

@@ -83,7 +83,7 @@ describe('createCommonSlice', () => {
         settings: {
           general: { fontSize: 14 },
         },
-        email: 'test@example.com'
+        email: 'test@example.com',
       };
 
       vi.spyOn(userService, 'getUserState').mockResolvedValueOnce(mockUserState);

package/src/tools/web-browsing/index.ts

@@ -25,13 +25,7 @@ export const WebBrowsingManifest: BuiltinToolManifest = {
           searchCategories: {
             description: 'The search categories you can set:',
             items: {
-              enum: [
-                'general',
-                'images',
-                'news',
-                'science',
-                'videos',
-              ],
+              enum: ['general', 'images', 'news', 'science', 'videos'],
               type: 'string',
             },
             type: 'array',