@lobehub/chat 1.136.9 → 1.136.11

package/packages/model-runtime/src/providers/openai/__snapshots__/index.test.ts.snap

@@ -9,11 +9,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "whisper-1",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-02-27",
+    "search": false,
     "type": "stt",
+    "video": false,
     "vision": false,
   },
   {
@@ -23,11 +26,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "davinci-002",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-08-21",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -37,11 +43,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": true,
     "id": "gpt-3.5-turbo",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-02-28",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -51,6 +60,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "dall-e-2",
+    "imageOutput": false,
     "maxOutput": undefined,
     "parameters": {
       "imageUrl": {
@@ -71,7 +81,9 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-11-01",
+    "search": false,
     "type": "image",
+    "video": false,
     "vision": false,
   },
   {
@@ -81,11 +93,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "gpt-3.5-turbo-16k",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-05-10",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -95,11 +110,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "tts-1-hd-1106",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-11-03",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -109,11 +127,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "tts-1-hd",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-11-03",
+    "search": false,
     "type": "tts",
+    "video": false,
     "vision": false,
   },
   {
@@ -123,11 +144,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "gpt-3.5-turbo-16k-0613",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-05-30",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -137,11 +161,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "text-embedding-3-large",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2024-01-22",
+    "search": false,
     "type": "embedding",
+    "video": false,
     "vision": false,
   },
   {
@@ -151,11 +178,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "gpt-4-1106-vision-preview",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2024-03-26",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -165,11 +195,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "gpt-3.5-turbo-instruct-0914",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-09-07",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -179,11 +212,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": true,
     "id": "gpt-4-0125-preview",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2024-01-23",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -193,11 +229,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": true,
     "id": "gpt-4-turbo-preview",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2024-01-23",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -207,11 +246,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "gpt-3.5-turbo-instruct",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-08-24",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -221,11 +263,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "gpt-3.5-turbo-0301",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-03-01",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -235,11 +280,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "gpt-3.5-turbo-0613",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-06-12",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -249,11 +297,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "tts-1",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-04-19",
+    "search": false,
     "type": "tts",
+    "video": false,
     "vision": false,
   },
   {
@@ -263,6 +314,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "dall-e-3",
+    "imageOutput": false,
     "maxOutput": undefined,
     "parameters": {
       "prompt": {
@@ -287,7 +339,9 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-10-31",
+    "search": false,
     "type": "image",
+    "video": false,
     "vision": false,
   },
   {
@@ -297,11 +351,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": true,
     "id": "gpt-3.5-turbo-1106",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-11-02",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -311,11 +368,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": true,
     "id": "gpt-4-1106-preview",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-11-02",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -325,11 +385,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "babbage-002",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-08-21",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -339,11 +402,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "tts-1-1106",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-11-03",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -353,11 +419,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "gpt-4-vision-preview",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-11-02",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": true,
   },
   {
@@ -367,11 +436,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "text-embedding-3-small",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2024-01-22",
+    "search": false,
     "type": "embedding",
+    "video": false,
     "vision": false,
   },
   {
@@ -381,11 +453,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": true,
     "id": "gpt-4",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-06-27",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -395,11 +470,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": false,
     "id": "text-embedding-ada-002",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2022-12-16",
+    "search": false,
     "type": "embedding",
+    "video": false,
     "vision": false,
   },
   {
@@ -409,11 +487,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": true,
     "id": "gpt-3.5-turbo-0125",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2024-01-23",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
   {
@@ -423,11 +504,14 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "enabled": false,
     "functionCall": true,
     "id": "gpt-4-0613",
+    "imageOutput": false,
     "maxOutput": undefined,
     "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-06-12",
+    "search": false,
     "type": "chat",
+    "video": false,
     "vision": false,
   },
 ]

package/packages/model-runtime/src/utils/modelParse.test.ts

@@ -99,6 +99,14 @@ const mockDefaultModelList: (Partial<ChatModelCard> & { id: string })[] = [
 // Mock the import
 vi.mock('model-bank', () => ({
   LOBE_DEFAULT_MODEL_LIST: mockDefaultModelList,
+  // 新增 provider 专用清单，供 findKnownModelByProvider 使用
+  google: [
+    {
+      id: 'gemini-2.5-pro',
+      displayName: 'Gemini 2.5 Pro',
+      abilities: { search: true, functionCall: true, reasoning: true, vision: true },
+    },
+  ],
 }));
 
 describe('modelParse', () => {
@@ -151,6 +159,14 @@ describe('modelParse', () => {
     });
   });
 
+  // New sanity tests for your added config
+  describe('MODEL_LIST_CONFIGS sanity', () => {
+    it('google.imageOutputKeywords should include "-image-" for image-output capability inference', () => {
+      expect(MODEL_LIST_CONFIGS.google.imageOutputKeywords).toBeDefined();
+      expect(MODEL_LIST_CONFIGS.google.imageOutputKeywords).toContain('-image-');
+    });
+  });
+
   describe('processModelList', () => {
     it('should process a list of models with the given provider config', async () => {
       const modelList = [{ id: 'gpt-4o' }, { id: 'gpt-3.5-turbo' }];
@@ -220,6 +236,52 @@ describe('modelParse', () => {
       expect(Array.isArray(result)).toBe(true);
     });
 
+    // New search & imageOutput focused tests for single provider path
+    describe('search and imageOutput (processModelList)', () => {
+      it('openai: default search keywords should make "*-search" models support search', async () => {
+        // openai config does not define searchKeywords, so DEFAULT_SEARCH_KEYWORDS ['-search'] applies
+        const out = await processModelList([{ id: 'gpt-4o-search' }], MODEL_LIST_CONFIGS.openai, 'openai');
+        expect(out).toHaveLength(1);
+        expect(out[0].search).toBe(true);
+      });
+
+      it('openai: models without "-search" should not get search by default', async () => {
+        const out = await processModelList([{ id: 'gpt-4o' }], MODEL_LIST_CONFIGS.openai, 'openai');
+        expect(out).toHaveLength(1);
+        expect(out[0].search).toBe(false);
+      });
+
+      it('openai: "-search" models with excluded keywords (audio) should not get search', async () => {
+        const out = await processModelList(
+          [{ id: 'gpt-4o-search-audio' }],
+          MODEL_LIST_CONFIGS.openai,
+          'openai',
+        );
+        expect(out).toHaveLength(1);
+        expect(out[0].search).toBe(false);
+      });
+
+      it('google: gemini-* with "-image-" in id should infer imageOutput=true via keywords and remain chat type', async () => {
+        const out = await processModelList(
+          [{ id: 'gemini-2.5-image-pro' }],
+          MODEL_LIST_CONFIGS.google,
+          'google',
+        );
+        expect(out).toHaveLength(1);
+        expect(out[0].imageOutput).toBe(true);
+        // due to '!gemini' exclusion in IMAGE_MODEL_KEYWORDS
+        expect(out[0].type).toBe('chat');
+      });
+
+      it('google: gemini-* without "-image-" should not infer imageOutput and get search=true via known google model', async () => {
+        const out = await processModelList([{ id: 'gemini-2.5-pro' }], MODEL_LIST_CONFIGS.google, 'google');
+        expect(out).toHaveLength(1);
+        expect(out[0].displayName).toBe('Gemini 2.5 Pro');
+        expect(out[0].search).toBe(true);
+        expect(out[0].imageOutput).toBe(false);
+      });
+    });
+
     describe('Detailed capability and property processing in processModelList', () => {
       const config = MODEL_LIST_CONFIGS.openai;
 
@@ -411,6 +473,47 @@ describe('modelParse', () => {
       expect(model.reasoning).toBe(true); // 从 knownModel.abilities.reasoning
     });
 
+    // New search & imageOutput focused tests for multi-provider path
+    describe('search and imageOutput (processMultiProviderModelList)', () => {
+      it('non-google provider gemini-2.5-pro should still get search=true via known google model', async () => {
+        // Simulate a mixed/custom list. Even if called with providerid='openai',
+        // detectModelProvider('gemini-2.5-pro') => 'google'
+        // knownModel (google list) has abilities.search=true, so final search=true
+        const out = await processMultiProviderModelList([{ id: 'gemini-2.5-pro' }], 'openai');
+        expect(out).toHaveLength(1);
+        const m = out[0];
+        expect(m.id).toBe('gemini-2.5-pro');
+        expect(m.displayName).toBe('Gemini 2.5 Pro');
+        expect(m.search).toBe(true);
+      });
+
+      it('default search keywords should make "*-search" models support search', async () => {
+        const out = await processMultiProviderModelList([{ id: 'gpt-4o-search'}]);
+        expect(out).toHaveLength(1);
+        expect(out[0].search).toBe(true);
+      });
+
+      it('google: gemini-* with "-image-" in id should infer imageOutput=true via keywords', async () => {
+        const out = await processMultiProviderModelList([{ id: 'gemini-2.5-image-pro' }]);
+        expect(out).toHaveLength(1);
+        expect(out[0].imageOutput).toBe(true);
+        // and still a chat model due to "!gemini" image-type exclusion
+        expect(out[0].type).toBe('chat');
+      });
+
+      it('openai: "-search" + "audio" should be suppressed by excludeKeywords', async () => {
+        const out = await processMultiProviderModelList([{ id: 'gpt-4o-search-audio' }], 'openai');
+        expect(out).toHaveLength(1);
+        expect(out[0].search).toBe(false);
+      });
+
+      it('google: gemini-* without "-image-" should not infer imageOutput', async () => {
+        const out = await processMultiProviderModelList([{ id: 'gemini-2.5-pro' }]);
+        expect(out).toHaveLength(1);
+        expect(out[0].imageOutput).toBe(false);
+      });
+    });
+
     describe('Extended tests for detectModelProvider', () => {
       it('should handle unusual casing patterns', () => {
         expect(detectModelProvider('gPt-4')).toBe('openai');