@lobehub/chat 1.90.2 → 1.90.4

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

@@ -1,7 +1,7 @@
 import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 import { pruneReasoningPayload } from '../utils/openaiHelpers';
 
 export interface OpenAIModelCard {
@@ -10,13 +10,13 @@ export interface OpenAIModelCard {
 
 const prunePrefixes = ['o1', 'o3', 'o4'];
 
-export const LobeOpenAI = LobeOpenAICompatibleFactory({
+export const LobeOpenAI = createOpenAICompatibleRuntime({
   baseURL: 'https://api.openai.com/v1',
   chatCompletion: {
     handlePayload: (payload) => {
       const { model } = payload;
 
-      if (prunePrefixes.some(prefix => model.startsWith(prefix))) {
+      if (prunePrefixes.some((prefix) => model.startsWith(prefix))) {
         return pruneReasoningPayload(payload) as any;
       }
 

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

@@ -1,7 +1,7 @@
 import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 import { OpenRouterModelCard, OpenRouterModelExtraInfo, OpenRouterReasoning } from './type';
 
 const formatPrice = (price: string) => {
@@ -9,7 +9,7 @@ const formatPrice = (price: string) => {
   return Number((Number(price) * 1e6).toPrecision(5));
 };
 
-export const LobeOpenRouterAI = LobeOpenAICompatibleFactory({
+export const LobeOpenRouterAI = createOpenAICompatibleRuntime({
   baseURL: 'https://openrouter.ai/api/v1',
   chatCompletion: {
     handlePayload: (payload) => {

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

@@ -1,9 +1,9 @@
 import OpenAI from 'openai';
 
 import { ChatStreamPayload, ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
-export const LobePerplexityAI = LobeOpenAICompatibleFactory({
+export const LobePerplexityAI = createOpenAICompatibleRuntime({
   baseURL: 'https://api.perplexity.ai',
   chatCompletion: {
     handlePayload: (payload: ChatStreamPayload) => {

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

@@ -1,10 +1,10 @@
 import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 import { PPIOModelCard } from './type';
 
-export const LobePPIOAI = LobeOpenAICompatibleFactory({
+export const LobePPIOAI = createOpenAICompatibleRuntime({
   baseURL: 'https://api.ppinfra.com/v3/openai',
   constructorOptions: {
     defaultHeaders: {

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

@@ -1,13 +1,13 @@
 import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface QiniuModelCard {
   id: string;
 }
 
-export const LobeQiniuAI = LobeOpenAICompatibleFactory({
+export const LobeQiniuAI = createOpenAICompatibleRuntime({
   apiKey: 'placeholder-to-avoid-error',
   baseURL: 'https://api.qnaigc.com/v1',
   debug: {

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

@@ -1,7 +1,7 @@
 import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 import { QwenAIStream } from '../utils/streams';
 
 export interface QwenModelCard {
@@ -20,7 +20,7 @@ export const QwenLegacyModels = new Set([
   'qwen-1.8b-longcontext-chat',
 ]);
 
-export const LobeQwenAI = LobeOpenAICompatibleFactory({
+export const LobeQwenAI = createOpenAICompatibleRuntime({
   baseURL: 'https://dashscope.aliyuncs.com/compatible-mode/v1',
   chatCompletion: {
     handlePayload: (payload) => {

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

@@ -1,7 +1,7 @@
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
-export const LobeSambaNovaAI = LobeOpenAICompatibleFactory({
+export const LobeSambaNovaAI = createOpenAICompatibleRuntime({
   baseURL: 'https://api.sambanova.ai/v1',
   debug: {
     chatCompletion: () => process.env.DEBUG_SAMBANOVA_CHAT_COMPLETION === '1',

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

@@ -3,13 +3,13 @@ import OpenAI from 'openai';
 import type { ChatModelCard } from '@/types/llm';
 
 import { ChatStreamPayload, ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface Search1APIModelCard {
   id: string;
 }
 
-export const LobeSearch1API = LobeOpenAICompatibleFactory({
+export const LobeSearch1API = createOpenAICompatibleRuntime({
   baseURL: 'https://api.search1api.com/v1',
   chatCompletion: {
     handlePayload: (payload: ChatStreamPayload) => {

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

@@ -1,14 +1,14 @@
 import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 import { convertSenseNovaMessage } from '../utils/sensenovaHelpers';
 
 export interface SenseNovaModelCard {
   id: string;
 }
 
-export const LobeSenseNovaAI = LobeOpenAICompatibleFactory({
+export const LobeSenseNovaAI = createOpenAICompatibleRuntime({
   baseURL: 'https://api.sensenova.cn/compatible-mode/v1',
   chatCompletion: {
     handlePayload: (payload) => {

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

@@ -2,13 +2,13 @@ import type { ChatModelCard } from '@/types/llm';
 
 import { AgentRuntimeErrorType } from '../error';
 import { ChatCompletionErrorPayload, ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface SiliconCloudModelCard {
   id: string;
 }
 
-export const LobeSiliconCloudAI = LobeOpenAICompatibleFactory({
+export const LobeSiliconCloudAI = createOpenAICompatibleRuntime({
   baseURL: 'https://api.siliconflow.cn/v1',
   chatCompletion: {
     handleError: (error: any): Omit<ChatCompletionErrorPayload, 'provider'> | undefined => {

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

@@ -1,26 +1,28 @@
 import { ChatStreamPayload, ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 import { SparkAIStream, transformSparkResponseToStream } from '../utils/streams';
 
-export const LobeSparkAI = LobeOpenAICompatibleFactory({
+export const LobeSparkAI = createOpenAICompatibleRuntime({
   baseURL: 'https://spark-api-open.xf-yun.com/v1',
   chatCompletion: {
     handlePayload: (payload: ChatStreamPayload) => {
       const { enabledSearch, tools, ...rest } = payload;
 
-      const sparkTools = enabledSearch ? [
-        ...(tools || []),
-        {
-          type: "web_search",
-          web_search: {
-            enable: true,
-            search_mode: process.env.SPARK_SEARCH_MODE || "normal", // normal or deep
-            /*
+      const sparkTools = enabledSearch
+        ? [
+            ...(tools || []),
+            {
+              type: 'web_search',
+              web_search: {
+                enable: true,
+                search_mode: process.env.SPARK_SEARCH_MODE || 'normal', // normal or deep
+                /*
             show_ref_label: true,
             */
-          },
-        }
-      ] : tools;
+              },
+            },
+          ]
+        : tools;
 
       return {
         ...rest,

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

@@ -1,13 +1,13 @@
 import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface StepfunModelCard {
   id: string;
 }
 
-export const LobeStepfunAI = LobeOpenAICompatibleFactory({
+export const LobeStepfunAI = createOpenAICompatibleRuntime({
   baseURL: 'https://api.stepfun.com/v1',
   chatCompletion: {
     handlePayload: (payload) => {

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

@@ -1,9 +1,9 @@
 import OpenAI from 'openai';
 
 import { ChatStreamPayload, ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
-export const LobeTaichuAI = LobeOpenAICompatibleFactory({
+export const LobeTaichuAI = createOpenAICompatibleRuntime({
   baseURL: 'https://ai-maas.wair.ac.cn/maas/v1',
   chatCompletion: {
     handlePayload: (payload: ChatStreamPayload) => {

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

@@ -1,13 +1,13 @@
 import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface TencentCloudModelCard {
   id: string;
 }
 
-export const LobeTencentCloudAI = LobeOpenAICompatibleFactory({
+export const LobeTencentCloudAI = createOpenAICompatibleRuntime({
   baseURL: 'https://api.lkeap.cloud.tencent.com/v1',
   debug: {
     chatCompletion: () => process.env.DEBUG_TENCENT_CLOUD_CHAT_COMPLETION === '1',

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

@@ -1,10 +1,10 @@
 import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 import { TogetherAIModel } from './type';
 
-export const LobeTogetherAI = LobeOpenAICompatibleFactory({
+export const LobeTogetherAI = createOpenAICompatibleRuntime({
   baseURL: 'https://api.together.xyz/v1',
   constructorOptions: {
     defaultHeaders: {

package/src/libs/model-runtime/types/chat.ts

@@ -137,7 +137,7 @@ export interface ChatStreamPayload {
   top_p?: number;
 }
 
-export interface ChatCompetitionOptions {
+export interface ChatMethodOptions {
   callback?: ChatStreamCallbacks;
   /**
    * response headers

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

@@ -1,7 +1,7 @@
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
-export const LobeUpstageAI = LobeOpenAICompatibleFactory({
+export const LobeUpstageAI = createOpenAICompatibleRuntime({
   baseURL: 'https://api.upstage.ai/v1/solar',
   debug: {
     chatCompletion: () => process.env.DEBUG_UPSTAGE_CHAT_COMPLETION === '1',

package/src/libs/model-runtime/utils/openaiCompatibleFactory/index.test.ts

@@ -14,7 +14,7 @@ import officalOpenAIModels from '@/libs/model-runtime/openai/fixtures/openai-mod
 import { sleep } from '@/utils/sleep';
 
 import * as debugStreamModule from '../debugStream';
-import { LobeOpenAICompatibleFactory } from './index';
+import { createOpenAICompatibleRuntime } from './index';
 
 const provider = 'groq';
 const defaultBaseURL = 'https://api.groq.com/openai/v1';
@@ -26,7 +26,7 @@ vi.spyOn(console, 'error').mockImplementation(() => {});
 
 let instance: LobeOpenAICompatibleRuntime;
 
-const LobeMockProvider = LobeOpenAICompatibleFactory({
+const LobeMockProvider = createOpenAICompatibleRuntime({
   baseURL: defaultBaseURL,
   chatCompletion: {
     handleError: (error) => {
@@ -426,7 +426,7 @@ describe('LobeOpenAICompatibleFactory', () => {
 
     describe('noUserId option', () => {
       it('should not add user to payload when noUserId is true', async () => {
-        const LobeMockProvider = LobeOpenAICompatibleFactory({
+        const LobeMockProvider = createOpenAICompatibleRuntime({
           baseURL: 'https://api.mistral.ai/v1',
           chatCompletion: {
             noUserId: true,
@@ -457,7 +457,7 @@ describe('LobeOpenAICompatibleFactory', () => {
       });
 
       it('should add user to payload when noUserId is false', async () => {
-        const LobeMockProvider = LobeOpenAICompatibleFactory({
+        const LobeMockProvider = createOpenAICompatibleRuntime({
           baseURL: 'https://api.mistral.ai/v1',
           chatCompletion: {
             noUserId: false,
@@ -488,7 +488,7 @@ describe('LobeOpenAICompatibleFactory', () => {
       });
 
       it('should add user to payload when noUserId is not set in chatCompletion', async () => {
-        const LobeMockProvider = LobeOpenAICompatibleFactory({
+        const LobeMockProvider = createOpenAICompatibleRuntime({
           baseURL: 'https://api.mistral.ai/v1',
           provider: ModelProvider.Mistral,
         });
@@ -831,7 +831,7 @@ describe('LobeOpenAICompatibleFactory', () => {
         },
       );
 
-      const LobeMockProvider = LobeOpenAICompatibleFactory({
+      const LobeMockProvider = createOpenAICompatibleRuntime({
         baseURL: 'https://api.test.com/v1',
         chatCompletion: {
           handleStream: customStreamHandler,
@@ -890,7 +890,7 @@ describe('LobeOpenAICompatibleFactory', () => {
         });
       });
 
-      const LobeMockProvider = LobeOpenAICompatibleFactory({
+      const LobeMockProvider = createOpenAICompatibleRuntime({
         baseURL: 'https://api.test.com/v1',
         chatCompletion: {
           handleTransformResponseToStream: customTransformHandler,

package/src/libs/model-runtime/utils/openaiCompatibleFactory/index.ts

@@ -9,8 +9,8 @@ import type { ChatModelCard } from '@/types/llm';
 import { LobeRuntimeAI } from '../../BaseAI';
 import { AgentRuntimeErrorType, ILobeAgentRuntimeErrorType } from '../../error';
 import type {
-  ChatCompetitionOptions,
   ChatCompletionErrorPayload,
+  ChatMethodOptions,
   ChatStreamPayload,
   Embeddings,
   EmbeddingsOptions,
@@ -150,7 +150,7 @@ export function transformResponseToStream(data: OpenAI.ChatCompletion) {
   });
 }
 
-export const LobeOpenAICompatibleFactory = <T extends Record<string, any> = any>({
+export const createOpenAICompatibleRuntime = <T extends Record<string, any> = any>({
   provider,
   baseURL: DEFAULT_BASE_URL,
   apiKey: DEFAULT_API_LEY,
@@ -199,7 +199,7 @@ export const LobeOpenAICompatibleFactory = <T extends Record<string, any> = any>
       this.id = options.id || provider;
     }
 
-    async chat({ responseMode, ...payload }: ChatStreamPayload, options?: ChatCompetitionOptions) {
+    async chat({ responseMode, ...payload }: ChatStreamPayload, options?: ChatMethodOptions) {
       try {
         const inputStartAt = Date.now();
         const postPayload = chatCompletion?.handlePayload

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

@@ -1,13 +1,13 @@
 import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface VLLMModelCard {
   id: string;
 }
 
-export const LobeVLLMAI = LobeOpenAICompatibleFactory({
+export const LobeVLLMAI = createOpenAICompatibleRuntime({
   baseURL: 'http://localhost:8000/v1',
   debug: {
     chatCompletion: () => process.env.DEBUG_VLLM_CHAT_COMPLETION === '1',

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

@@ -1,7 +1,7 @@
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
-export const LobeVolcengineAI = LobeOpenAICompatibleFactory({
+export const LobeVolcengineAI = createOpenAICompatibleRuntime({
   baseURL: 'https://ark.cn-beijing.volces.com/api/v3',
   chatCompletion: {
     handlePayload: (payload) => {

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

@@ -1,7 +1,7 @@
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
-export const LobeWenxinAI = LobeOpenAICompatibleFactory({
+export const LobeWenxinAI = createOpenAICompatibleRuntime({
   baseURL: 'https://qianfan.baidubce.com/v2',
   chatCompletion: {
     handlePayload: (payload) => {

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

@@ -1,13 +1,13 @@
 import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface XAIModelCard {
   id: string;
 }
 
-export const LobeXAI = LobeOpenAICompatibleFactory({
+export const LobeXAI = createOpenAICompatibleRuntime({
   baseURL: 'https://api.x.ai/v1',
   chatCompletion: {
     handlePayload: (payload) => {
@@ -21,7 +21,10 @@ export const LobeXAI = LobeOpenAICompatibleFactory({
         stream: true,
         ...(enabledSearch && {
           search_parameters: {
-            max_search_results: Math.min(Math.max(parseInt(process.env.XAI_MAX_SEARCH_RESULTS ?? '15', 10), 1), 30),
+            max_search_results: Math.min(
+              Math.max(parseInt(process.env.XAI_MAX_SEARCH_RESULTS ?? '15', 10), 1),
+              30,
+            ),
             mode: 'auto',
             return_citations: true,
             sources: [

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

@@ -1,7 +1,7 @@
 import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface XinferenceModelCard {
   context_length: number;
@@ -12,7 +12,7 @@ export interface XinferenceModelCard {
   name: string;
 }
 
-export const LobeXinferenceAI = LobeOpenAICompatibleFactory({
+export const LobeXinferenceAI = createOpenAICompatibleRuntime({
   baseURL: 'http://localhost:9997/v1',
   debug: {
     chatCompletion: () => process.env.DEBUG_XINFERENCE_CHAT_COMPLETION === '1',

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

@@ -1,13 +1,13 @@
 import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface ZeroOneModelCard {
   id: string;
 }
 
-export const LobeZeroOneAI = LobeOpenAICompatibleFactory({
+export const LobeZeroOneAI = createOpenAICompatibleRuntime({
   baseURL: 'https://api.lingyiwanwu.com/v1',
   debug: {
     chatCompletion: () => process.env.DEBUG_ZEROONE_CHAT_COMPLETION === '1',

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

@@ -1,7 +1,7 @@
 import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
-import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface ZhipuModelCard {
   description: string;
@@ -9,7 +9,7 @@ export interface ZhipuModelCard {
   modelName: string;
 }
 
-export const LobeZhipuAI = LobeOpenAICompatibleFactory({
+export const LobeZhipuAI = createOpenAICompatibleRuntime({
   baseURL: 'https://open.bigmodel.cn/api/paas/v4',
   chatCompletion: {
     handlePayload: (payload) => {

package/src/middleware.ts

@@ -206,7 +206,9 @@ const clerkAuthMiddleware = clerkMiddleware(
   async (auth, req) => {
     logClerk('Clerk middleware processing request: %s %s', req.method, req.url);
 
-    const isProtected = isProtectedRoute(req);
+    // when enable auth protection, only public route is not protected, others are all protected
+    const isProtected = appEnv.ENABLE_AUTH_PROTECTION ? !isPublicRoute(req) : isProtectedRoute(req);
+
     logClerk('Route protection status: %s, %s', req.url, isProtected ? 'protected' : 'public');
 
     if (isProtected) {

package/src/server/globalConfig/index.ts

@@ -40,6 +40,9 @@ export const getServerGlobalConfig = async () => {
         enabled: isDesktop ? true : undefined,
         fetchOnClient: isDesktop ? false : !process.env.OLLAMA_PROXY_URL,
       },
+      qwen: {
+        withDeploymentName: true,
+      },
       tencentcloud: {
         enabledKey: 'ENABLED_TENCENT_CLOUD',
         modelListKey: 'TENCENT_CLOUD_MODEL_LIST',

package/src/server/routers/tools/search.test.ts

@@ -104,8 +104,7 @@ describe('searchRouter', () => {
         query: 'test query',
       });
 
-      expect(result).toEqual({
-        costTime: 0,
+      expect(result).toMatchObject({
         query: 'test query',
         results: [
           {
@@ -139,8 +138,7 @@ describe('searchRouter', () => {
         query: 'test query',
       });
 
-      expect(result).toEqual({
-        costTime: 0,
+      expect(result).toMatchObject({
         query: 'test query',
         results: [
           {

package/src/services/chat.ts

@@ -300,6 +300,7 @@ class ChatService {
       ModelProvider.Azure,
       ModelProvider.Volcengine,
       ModelProvider.AzureAI,
+      ModelProvider.Qwen,
     ] as string[];
 
     if (providersWithDeploymentName.includes(provider)) {

package/src/store/chat/slices/aiChat/actions/__tests__/generateAIChat.test.ts

@@ -671,14 +671,16 @@ describe('chatMessage actions', () => {
       (chatService.createAssistantMessage as Mock).mockResolvedValue(aiResponse);
       const spy = vi.spyOn(chatService, 'createAssistantMessageStream');
       // 模拟消息创建
-      (messageService.createMessage as Mock).mockResolvedValue('assistant-message-id');
+      const createMessageSpyOn = vi
+        .spyOn(messageService, 'createMessage')
+        .mockResolvedValue('assistant-message-id');
 
       await act(async () => {
         await result.current.internal_coreProcessMessage(messages, userMessage.id);
       });
 
       // 验证是否创建了代表 AI 响应的消息
-      expect(messageService.createMessage).toHaveBeenCalledWith(
+      expect(createMessageSpyOn).toHaveBeenCalledWith(
         expect.objectContaining({
           role: 'assistant',
           content: LOADING_FLAT,

package/src/store/chat/slices/message/action.test.ts

@@ -369,11 +369,12 @@ describe('chatMessage actions', () => {
       const messageId = 'message-id';
       const newContent = 'Updated content';
 
+      const spy = vi.spyOn(messageService, 'updateMessage');
       await act(async () => {
         await result.current.internal_updateMessageContent(messageId, newContent);
       });
 
-      expect(messageService.updateMessage).toHaveBeenCalledWith(messageId, { content: newContent });
+      expect(spy).toHaveBeenCalledWith(messageId, { content: newContent });
     });
 
     it('should dispatch message update action', async () => {

package/src/store/chat/slices/topic/action.test.ts

@@ -264,7 +264,8 @@ describe('topic action', () => {
       const topicId = 'topic-id';
       const newTitle = 'Updated Topic Title';
       // Mock the topicService.updateTitle to resolve immediately
-      (topicService.updateTopic as Mock).mockResolvedValue(undefined);
+
+      const spyOn = vi.spyOn(topicService, 'updateTopic');
 
       const { result } = renderHook(() => useChatStore());
 
@@ -276,7 +277,7 @@ describe('topic action', () => {
       });
 
       // Verify that the topicService.updateTitle was called with correct parameters
-      expect(topicService.updateTopic).toHaveBeenCalledWith(topicId, {
+      expect(spyOn).toHaveBeenCalledWith(topicId, {
         title: 'Updated Topic Title',
       });
 

package/src/types/aiProvider.ts

@@ -23,6 +23,7 @@ export const AiProviderSDKEnum = {
   Huggingface: 'huggingface',
   Ollama: 'ollama',
   Openai: 'openai',
+  Qwen: 'qwen',
   Volcengine: 'volcengine',
 } as const;