@lobehub/chat 1.118.4 → 1.118.6

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

@@ -10,6 +10,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "whisper-1",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-02-27",
     "type": "stt",
@@ -23,6 +24,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "davinci-002",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-08-21",
     "type": "chat",
@@ -36,6 +38,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": true,
     "id": "gpt-3.5-turbo",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-02-28",
     "type": "chat",
@@ -65,6 +68,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
         ],
       },
     },
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-11-01",
     "type": "image",
@@ -78,6 +82,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "gpt-3.5-turbo-16k",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-05-10",
     "type": "chat",
@@ -91,6 +96,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "tts-1-hd-1106",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-11-03",
     "type": "chat",
@@ -104,6 +110,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "tts-1-hd",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-11-03",
     "type": "tts",
@@ -117,6 +124,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "gpt-3.5-turbo-16k-0613",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-05-30",
     "type": "chat",
@@ -130,6 +138,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "text-embedding-3-large",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2024-01-22",
     "type": "embedding",
@@ -143,6 +152,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "gpt-4-1106-vision-preview",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2024-03-26",
     "type": "chat",
@@ -156,6 +166,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "gpt-3.5-turbo-instruct-0914",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-09-07",
     "type": "chat",
@@ -169,6 +180,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": true,
     "id": "gpt-4-0125-preview",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2024-01-23",
     "type": "chat",
@@ -182,6 +194,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": true,
     "id": "gpt-4-turbo-preview",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2024-01-23",
     "type": "chat",
@@ -195,6 +208,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "gpt-3.5-turbo-instruct",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-08-24",
     "type": "chat",
@@ -208,6 +222,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "gpt-3.5-turbo-0301",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-03-01",
     "type": "chat",
@@ -221,6 +236,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "gpt-3.5-turbo-0613",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-06-12",
     "type": "chat",
@@ -234,6 +250,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "tts-1",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-04-19",
     "type": "tts",
@@ -260,6 +277,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
         ],
       },
     },
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-10-31",
     "type": "image",
@@ -273,6 +291,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": true,
     "id": "gpt-3.5-turbo-1106",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-11-02",
     "type": "chat",
@@ -286,6 +305,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": true,
     "id": "gpt-4-1106-preview",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-11-02",
     "type": "chat",
@@ -299,6 +319,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "babbage-002",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-08-21",
     "type": "chat",
@@ -312,6 +333,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "tts-1-1106",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-11-03",
     "type": "chat",
@@ -325,6 +347,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "gpt-4-vision-preview",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-11-02",
     "type": "chat",
@@ -338,6 +361,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "text-embedding-3-small",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2024-01-22",
     "type": "embedding",
@@ -351,6 +375,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": true,
     "id": "gpt-4",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-06-27",
     "type": "chat",
@@ -364,6 +389,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": false,
     "id": "text-embedding-ada-002",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2022-12-16",
     "type": "embedding",
@@ -377,6 +403,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": true,
     "id": "gpt-3.5-turbo-0125",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2024-01-23",
     "type": "chat",
@@ -390,6 +417,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "functionCall": true,
     "id": "gpt-4-0613",
     "maxOutput": undefined,
+    "pricing": undefined,
     "reasoning": false,
     "releasedAt": "2023-06-12",
     "type": "chat",

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

@@ -2,9 +2,6 @@
 import OpenAI from 'openai';
 import { Mock, afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-// 引入模块以便于对函数进行spy
-import { ChatStreamCallbacks } from '@/libs/model-runtime';
-
 import * as debugStreamModule from '../utils/debugStream';
 import officalOpenAIModels from './fixtures/openai-models.json';
 import { LobeOpenAI } from './index';

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

@@ -1,5 +1,51 @@
 // Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
 
+exports[`LobeOpenRouterAI > chat > should handle fetch error gracefully 1`] = `
+[
+  {
+    "contextWindowTokens": 131072,
+    "description": "Reflection Llama-3.1 70B is trained with a new technique called Reflection-Tuning that teaches a LLM to detect mistakes in its reasoning and correct course.
+
+The model was trained on synthetic data.
+
+_These are free, rate-limited endpoints for [Reflection 70B](/models/mattshumer/reflection-70b). Outputs may be cached. Read about rate limits [here](/docs/limits)._",
+    "displayName": "Reflection 70B",
+    "enabled": false,
+    "functionCall": false,
+    "id": "mattshumer/reflection-70b:free",
+    "maxOutput": undefined,
+    "pricing": undefined,
+    "reasoning": false,
+    "releasedAt": "1970-01-20",
+    "type": "chat",
+    "vision": false,
+  },
+]
+`;
+
+exports[`LobeOpenRouterAI > chat > should handle fetch failure gracefully 1`] = `
+[
+  {
+    "contextWindowTokens": 131072,
+    "description": "Reflection Llama-3.1 70B is trained with a new technique called Reflection-Tuning that teaches a LLM to detect mistakes in its reasoning and correct course.
+
+The model was trained on synthetic data.
+
+_These are free, rate-limited endpoints for [Reflection 70B](/models/mattshumer/reflection-70b). Outputs may be cached. Read about rate limits [here](/docs/limits)._",
+    "displayName": "Reflection 70B",
+    "enabled": false,
+    "functionCall": false,
+    "id": "mattshumer/reflection-70b:free",
+    "maxOutput": undefined,
+    "pricing": undefined,
+    "reasoning": false,
+    "releasedAt": "1970-01-20",
+    "type": "chat",
+    "vision": false,
+  },
+]
+`;
+
 exports[`LobeOpenRouterAI > models > should get models with frontend models data 1`] = `
 [
   {

package/packages/model-runtime/src/openrouter/index.test.ts

@@ -4,7 +4,6 @@ import { Mock, afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 import { LobeOpenAICompatibleRuntime } from '@/libs/model-runtime';
 import { testProvider } from '@/libs/model-runtime/providerTestUtils';
 
-import frontendModels from './fixtures/frontendModels.json';
 import models from './fixtures/models.json';
 import { LobeOpenRouterAI } from './index';
 
@@ -30,11 +29,17 @@ let instance: LobeOpenAICompatibleRuntime;
 beforeEach(() => {
   instance = new LobeOpenRouterAI({ apiKey: 'test' });
 
-  // 使用 vi.spyOn 来模拟 chat.completions.create 方法
-  vi.spyOn(instance['client'].chat.completions, 'create').mockResolvedValue(
-    new ReadableStream() as any,
-  );
-  vi.spyOn(instance['client'].models, 'list').mockResolvedValue({ data: [] } as any);
+  // 用一个完整的假 client 覆盖实例的 client，避免访问深层属性时出现初始化顺序问题
+  instance['client'] = {
+    chat: {
+      completions: {
+        create: vi.fn().mockResolvedValue(Promise.resolve(new ReadableStream())),
+      },
+    },
+    models: {
+      list: vi.fn().mockResolvedValue({ data: [] }),
+    },
+  } as any;
 });
 
 afterEach(() => {
@@ -44,12 +49,12 @@ afterEach(() => {
 describe('LobeOpenRouterAI', () => {
   describe('init', () => {
     it('should correctly initialize with a custom base URL', async () => {
-      const instance = new LobeOpenRouterAI({
+      const inst = new LobeOpenRouterAI({
         apiKey: 'test_api_key',
         baseURL: 'https://api.abc.com/v1',
       });
-      expect(instance).toBeInstanceOf(LobeOpenRouterAI);
-      expect(instance.baseURL).toEqual('https://api.abc.com/v1');
+      expect(inst).toBeInstanceOf(LobeOpenRouterAI);
+      expect(inst.baseURL).toEqual('https://api.abc.com/v1');
     });
   });
 
@@ -135,35 +140,6 @@ describe('LobeOpenRouterAI', () => {
       );
       expect(result).toBeInstanceOf(Response);
     });
-  });
-
-  describe('models', () => {
-    it('should get models with frontend models data', async () => {
-      // mock the models.list method
-      (instance['client'].models.list as Mock).mockResolvedValue({ data: models });
-
-      // 模拟成功的 fetch 响应
-      vi.stubGlobal(
-        'fetch',
-        vi.fn().mockResolvedValue({
-          ok: true,
-          json: vi.fn().mockResolvedValue(frontendModels),
-        }),
-      );
-
-      const list = await instance.models();
-
-      // 验证 fetch 被正确调用
-      expect(fetch).toHaveBeenCalledWith('https://openrouter.ai/api/frontend/models');
-
-      // 验证模型列表中包含了从前端 API 获取的额外信息
-      const reflectionModel = list.find((model) => model.id === 'mattshumer/reflection-70b:free');
-      expect(reflectionModel).toBeDefined();
-      expect(reflectionModel?.reasoning).toBe(true);
-      expect(reflectionModel?.functionCall).toBe(true);
-
-      expect(list).toMatchSnapshot();
-    });
 
     it('should handle fetch failure gracefully', async () => {
       // mock the models.list method
@@ -179,10 +155,10 @@ describe('LobeOpenRouterAI', () => {
 
       const list = await instance.models();
 
-      // 验证即使 fetch 失败，方法仍然能返回有效的模型列表
+      // 验证在当前实现中，当 frontend fetch 返回非 ok 时，会返回空列表
       expect(fetch).toHaveBeenCalledWith('https://openrouter.ai/api/frontend/models');
-      expect(list.length).toBeGreaterThan(0); // 确保返回了模型列表
-      expect(list).toMatchSnapshot();
+      expect(list.length).toBe(0);
+      expect(list).toEqual([]);
     });
 
     it('should handle fetch error gracefully', async () => {
@@ -190,15 +166,15 @@ describe('LobeOpenRouterAI', () => {
       (instance['client'].models.list as Mock).mockResolvedValue({ data: models });
 
       // 在测试环境中，需要先修改 fetch 的实现，确保错误被捕获
-      vi.spyOn(global, 'fetch').mockImplementation(() => {
+      vi.spyOn(globalThis, 'fetch').mockImplementation(() => {
         throw new Error('Network error');
       });
 
       const list = await instance.models();
 
-      // 验证即使 fetch 出错，方法仍然能返回有效的模型列表
-      expect(list.length).toBeGreaterThan(0); // 确保返回了模型列表
-      expect(list).toMatchSnapshot();
+      // 验证在当前实现中，当 frontend fetch 抛错时，会返回空列表
+      expect(list.length).toBe(0);
+      expect(list).toEqual([]);
     });
   });
 });

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

@@ -1,9 +1,9 @@
-import OpenRouterModels from '@/config/aiModels/openrouter';
+import { openrouter as OpenRouterModels } from 'model-bank';
 
 import { ModelProvider } from '../types';
 import { processMultiProviderModelList } from '../utils/modelParse';
 import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
-import { OpenRouterModelCard, OpenRouterModelExtraInfo, OpenRouterReasoning } from './type';
+import { OpenRouterModelCard, OpenRouterReasoning } from './type';
 
 const formatPrice = (price: string) => {
   if (price === '-1') return undefined;
@@ -55,48 +55,45 @@ export const LobeOpenRouterAI = createOpenAICompatibleRuntime({
   debug: {
     chatCompletion: () => process.env.DEBUG_OPENROUTER_CHAT_COMPLETION === '1',
   },
-  models: async ({ client }) => {
-    const modelsPage = (await client.models.list()) as any;
-    const modelList: OpenRouterModelCard[] = modelsPage.data;
+  models: async () => {
+    let modelList: OpenRouterModelCard[] = [];
 
-    const modelsExtraInfo: OpenRouterModelExtraInfo[] = [];
     try {
       const response = await fetch('https://openrouter.ai/api/frontend/models');
       if (response.ok) {
         const data = await response.json();
-        modelsExtraInfo.push(...data['data']);
+        modelList = data['data'];
       }
     } catch (error) {
       console.error('Failed to fetch OpenRouter frontend models:', error);
+      return [];
     }
 
-    // 先处理抓取的模型信息，转换为标准格式
+    // 处理前端获取的模型信息，转换为标准格式
     const formattedModels = modelList.map((model) => {
-      const extraInfo = modelsExtraInfo.find(
-        (m) => m.slug.toLowerCase() === model.id.toLowerCase(),
-      );
+      const displayName = model.slug?.toLowerCase().includes('deepseek')
+        ? (model.name ?? model.slug)
+        : (model.short_name ?? model.name ?? model.slug);
 
       return {
         contextWindowTokens: model.context_length,
         description: model.description,
-        displayName: model.name,
-        functionCall:
-          model.description.includes('function calling') ||
-          model.description.includes('tools') ||
-          extraInfo?.endpoint?.supports_tool_parameters ||
-          false,
-        id: model.id,
+        displayName,
+        functionCall: model.endpoint?.supports_tool_parameters || false,
+        id: model.slug,
         maxOutput:
-          typeof model.top_provider.max_completion_tokens === 'number'
-            ? model.top_provider.max_completion_tokens
+          typeof model.endpoint?.max_completion_tokens === 'number'
+            ? model.endpoint.max_completion_tokens
             : undefined,
         pricing: {
-          input: formatPrice(model.pricing.prompt),
-          output: formatPrice(model.pricing.completion),
+          input: formatPrice(model.endpoint?.pricing?.prompt),
+          output: formatPrice(model.endpoint?.pricing?.completion),
         },
-        reasoning: extraInfo?.endpoint?.supports_reasoning || false,
-        releasedAt: new Date(model.created * 1000).toISOString().split('T')[0],
-        vision: model.architecture.modality.includes('image') || false,
+        reasoning: model.endpoint?.supports_reasoning || false,
+        releasedAt: new Date(model.created_at).toISOString().split('T')[0],
+        vision:
+          (Array.isArray(model.input_modalities) && model.input_modalities.includes('image')) ||
+          false,
       };
     });
 

package/packages/model-runtime/src/openrouter/type.ts

@@ -5,39 +5,27 @@ interface ModelPricing {
   request: string;
 }
 
-interface ModelArchitecture {
-  instruct_type: string | null;
-  modality: string;
-  tokenizer: string;
-}
-
-interface ModelTopProvider {
-  is_moderated: boolean;
-  max_completion_tokens: number | null;
-}
-
 export interface OpenRouterModelCard {
-  architecture: ModelArchitecture;
   context_length: number;
-  created: number;
-  description: string;
-  id: string;
-  name: string;
-  per_request_limits: any | null;
-  pricing: ModelPricing;
-  top_provider: ModelTopProvider;
+  created_at: number;
+  description?: string;
+  endpoint: OpenRouterModelEndpoint;
+  input_modalities?: string[];
+  name?: string;
+  output_modalities?: string[];
+  per_request_limits?: any | null;
+  short_name?: string;
+  slug: string;
 }
 
 interface OpenRouterModelEndpoint {
+  max_completion_tokens: number | null;
+  pricing: ModelPricing;
+  supported_parameters: string[];
   supports_reasoning?: boolean;
   supports_tool_parameters?: boolean;
 }
 
-export interface OpenRouterModelExtraInfo {
-  endpoint?: OpenRouterModelEndpoint;
-  slug: string;
-}
-
 interface OpenRouterOpenAIReasoning {
   effort: 'high' | 'medium' | 'low';
   exclude?: boolean;

package/packages/model-runtime/src/ppio/index.ts

@@ -15,7 +15,7 @@ export const LobePPIOAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_PPIO_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+    const { LOBE_DEFAULT_MODEL_LIST } = await import('model-bank');
 
     const reasoningKeywords = ['deepseek-r1'];
 

package/packages/model-runtime/src/search1api/index.ts

@@ -37,7 +37,7 @@ export const LobeSearch1API = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_SEARCH1API_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+    const { LOBE_DEFAULT_MODEL_LIST } = await import('model-bank');
 
     const modelsPage = (await client.models.list()) as any;
     const modelList: Search1APIModelCard[] = modelsPage.data;

package/packages/model-runtime/src/sensenova/index.ts

@@ -54,7 +54,7 @@ export const LobeSenseNovaAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_SENSENOVA_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+    const { LOBE_DEFAULT_MODEL_LIST } = await import('model-bank');
 
     const functionCallKeywords = ['1202'];
 

package/packages/model-runtime/src/stepfun/index.ts

@@ -36,7 +36,7 @@ export const LobeStepfunAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_STEPFUN_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+    const { LOBE_DEFAULT_MODEL_LIST } = await import('model-bank');
 
     // ref: https://platform.stepfun.com/docs/llm/modeloverview
     const functionCallKeywords = ['step-1-', 'step-1o-', 'step-1v-', 'step-2-'];

package/packages/model-runtime/src/tencentcloud/index.ts

@@ -13,7 +13,7 @@ export const LobeTencentCloudAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_TENCENT_CLOUD_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+    const { LOBE_DEFAULT_MODEL_LIST } = await import('model-bank');
 
     const reasoningKeywords = ['deepseek-r1'];
 

package/packages/model-runtime/src/togetherai/index.ts

@@ -16,7 +16,7 @@ export const LobeTogetherAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_TOGETHERAI_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+    const { LOBE_DEFAULT_MODEL_LIST } = await import('model-bank');
 
     const visionKeywords = ['qvq', 'vision'];
 

package/packages/model-runtime/src/types/image.ts

@@ -1,4 +1,4 @@
-import { RuntimeImageGenParams } from '@/libs/standard-parameters/index';
+import { RuntimeImageGenParams } from 'model-bank';
 
 export type CreateImagePayload = {
   model: string;

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

@@ -98,7 +98,7 @@ const mockDefaultModelList: (Partial<ChatModelCard> & { id: string })[] = [
 ];
 
 // Mock the import
-vi.mock('@/config/aiModels', () => ({
+vi.mock('model-bank', () => ({
   LOBE_DEFAULT_MODEL_LIST: mockDefaultModelList,
 }));
 
@@ -394,7 +394,7 @@ describe('modelParse', () => {
           reasoning: true,
         },
       };
-      const mockModule = await import('@/config/aiModels');
+      const mockModule = await import('model-bank');
       mockModule.LOBE_DEFAULT_MODEL_LIST.push(tempMockEntry as any);
 
       const modelList = [{ id: modelId }];
@@ -481,7 +481,7 @@ describe('modelParse', () => {
           contextWindowTokens: 5000,
         };
 
-        const mockModule = await import('@/config/aiModels');
+        const mockModule = await import('model-bank');
         mockModule.LOBE_DEFAULT_MODEL_LIST.push(tempModelEntry as any);
 
         const config = MODEL_LIST_CONFIGS.openai;
@@ -706,7 +706,7 @@ describe('modelParse', () => {
           reasoning: true,
         },
       };
-      const mockModule = await import('@/config/aiModels');
+      const mockModule = await import('model-bank');
       mockModule.LOBE_DEFAULT_MODEL_LIST.push(tempMockEntry as any);
 
       const modelList = [{ id: modelId }];
@@ -735,7 +735,7 @@ describe('modelParse', () => {
           reasoning: false,
         },
       };
-      const mockModule = await import('@/config/aiModels');
+      const mockModule = await import('model-bank');
       mockModule.LOBE_DEFAULT_MODEL_LIST.push(tempMockEntry as any);
 
       const modelList = [{ id: modelId }];