@lobehub/chat 1.62.10 → 1.63.0

package/src/libs/agent-runtime/google/index.ts

@@ -5,12 +5,13 @@ import {
   FunctionDeclaration,
   Tool as GoogleFunctionCallTool,
   GoogleGenerativeAI,
+  GoogleSearchRetrievalTool,
   Part,
   SchemaType,
 } from '@google/generative-ai';
 
-import type { ChatModelCard } from '@/types/llm';
 import { VertexAIStream } from '@/libs/agent-runtime/utils/streams/vertex-ai';
+import type { ChatModelCard } from '@/types/llm';
 import { imageUrlToBase64 } from '@/utils/imageToBase64';
 import { safeParseJSON } from '@/utils/safeParseJSON';
 
@@ -86,7 +87,7 @@ export class LobeGoogleAI implements LobeRuntimeAI {
       const payload = this.buildPayload(rawPayload);
       const model = payload.model;
 
-      const contents = await this.buildGoogleMessages(payload.messages, model);
+      const contents = await this.buildGoogleMessages(payload.messages);
 
       const geminiStreamResult = await this.client
         .getGenerativeModel(
@@ -123,7 +124,7 @@ export class LobeGoogleAI implements LobeRuntimeAI {
         .generateContentStream({
           contents,
           systemInstruction: payload.system as string,
-          tools: this.buildGoogleTools(payload.tools),
+          tools: this.buildGoogleTools(payload.tools, payload),
         });
 
       const googleStream = convertIterableToStream(geminiStreamResult.stream);
@@ -168,26 +169,30 @@ export class LobeGoogleAI implements LobeRuntimeAI {
       .map((model) => {
         const modelName = model.name.replace(/^models\//, '');
 
-        const knownModel = LOBE_DEFAULT_MODEL_LIST.find((m) => modelName.toLowerCase() === m.id.toLowerCase());
+        const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
+          (m) => modelName.toLowerCase() === m.id.toLowerCase(),
+        );
 
         return {
           contextWindowTokens: model.inputTokenLimit + model.outputTokenLimit,
           displayName: model.displayName,
           enabled: knownModel?.enabled || false,
           functionCall:
-            modelName.toLowerCase().includes('gemini') && !modelName.toLowerCase().includes('thinking')
-            || knownModel?.abilities?.functionCall
-            || false,
+            (modelName.toLowerCase().includes('gemini') &&
+              !modelName.toLowerCase().includes('thinking')) ||
+            knownModel?.abilities?.functionCall ||
+            false,
           id: modelName,
           reasoning:
-            modelName.toLowerCase().includes('thinking')
-            || knownModel?.abilities?.reasoning
-            || false,
+            modelName.toLowerCase().includes('thinking') ||
+            knownModel?.abilities?.reasoning ||
+            false,
           vision:
-            modelName.toLowerCase().includes('vision')
-            || (modelName.toLowerCase().includes('gemini') && !modelName.toLowerCase().includes('gemini-1.0'))
-            || knownModel?.abilities?.vision
-            || false,
+            modelName.toLowerCase().includes('vision') ||
+            (modelName.toLowerCase().includes('gemini') &&
+              !modelName.toLowerCase().includes('gemini-1.0')) ||
+            knownModel?.abilities?.vision ||
+            false,
         };
       })
       .filter(Boolean) as ChatModelCard[];
@@ -266,43 +271,7 @@ export class LobeGoogleAI implements LobeRuntimeAI {
   };
 
   // convert messages from the OpenAI format to Google GenAI SDK
-  private buildGoogleMessages = async (
-    messages: OpenAIChatMessage[],
-    model: string,
-  ): Promise<Content[]> => {
-    // if the model is gemini-1.0 we need to pair messages
-    if (model.startsWith('gemini-1.0')) {
-      const contents: Content[] = [];
-      let lastRole = 'model';
-
-      for (const message of messages) {
-        // current to filter function message
-        if (message.role === 'function') {
-          continue;
-        }
-        const googleMessage = await this.convertOAIMessagesToGoogleMessage(message);
-
-        // if the last message is a model message and the current message is a model message
-        // then we need to add a user message to separate them
-        if (lastRole === googleMessage.role) {
-          contents.push({ parts: [{ text: '' }], role: lastRole === 'user' ? 'model' : 'user' });
-        }
-
-        // add the current message to the contents
-        contents.push(googleMessage);
-
-        // update the last role
-        lastRole = googleMessage.role;
-      }
-
-      // if the last message is a user message, then we need to add a model message to separate them
-      if (lastRole === 'model') {
-        contents.push({ parts: [{ text: '' }], role: 'user' });
-      }
-
-      return contents;
-    }
-
+  private buildGoogleMessages = async (messages: OpenAIChatMessage[]): Promise<Content[]> => {
     const pools = messages
       .filter((message) => message.role !== 'function')
       .map(async (msg) => await this.convertOAIMessagesToGoogleMessage(msg));
@@ -353,7 +322,13 @@ export class LobeGoogleAI implements LobeRuntimeAI {
 
   private buildGoogleTools(
     tools: ChatCompletionTool[] | undefined,
+    payload?: ChatStreamPayload,
   ): GoogleFunctionCallTool[] | undefined {
+    // 目前 Tools (例如 googleSearch) 无法与其他 FunctionCall 同时使用
+    if (payload?.enabledSearch) {
+      return [{ googleSearch: {} } as GoogleSearchRetrievalTool];
+    }
+
     if (!tools || tools.length === 0) return;
 
     return [

package/src/libs/agent-runtime/novita/__snapshots__/index.test.ts.snap

@@ -26,7 +26,7 @@ exports[`NovitaAI > models > should get models 1`] = `
     "contextWindowTokens": 8192,
     "description": "Meta's latest class of models, Llama 3.1, launched with a variety of sizes and configurations. The 8B instruct-tuned version is particularly fast and efficient. It has demonstrated strong performance in human evaluations, outperforming several leading closed-source models.",
     "displayName": "meta-llama/llama-3.1-8b-instruct",
-    "enabled": true,
+    "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3.1-8b-instruct",
     "reasoning": false,
@@ -36,7 +36,7 @@ exports[`NovitaAI > models > should get models 1`] = `
     "contextWindowTokens": 8192,
     "description": "Meta's latest class of models, Llama 3.1, has launched with a variety of sizes and configurations. The 70B instruct-tuned version is optimized for high-quality dialogue use cases. It has demonstrated strong performance in human evaluations compared to leading closed-source models.",
     "displayName": "meta-llama/llama-3.1-70b-instruct",
-    "enabled": true,
+    "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3.1-70b-instruct",
     "reasoning": false,
@@ -46,7 +46,7 @@ exports[`NovitaAI > models > should get models 1`] = `
     "contextWindowTokens": 32768,
     "description": "Meta's latest class of models, Llama 3.1, launched with a variety of sizes and configurations. This 405B instruct-tuned version is optimized for high-quality dialogue use cases. It has demonstrated strong performance compared to leading closed-source models, including GPT-4o and Claude 3.5 Sonnet, in evaluations.",
     "displayName": "meta-llama/llama-3.1-405b-instruct",
-    "enabled": true,
+    "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3.1-405b-instruct",
     "reasoning": false,

package/src/libs/agent-runtime/openrouter/__snapshots__/index.test.ts.snap

@@ -407,7 +407,7 @@ It has demonstrated strong performance compared to leading closed-source models
 
 To read more about the model release, [click here](https://ai.meta.com/blog/meta-llama-3/). Usage of this model is subject to [Meta's Acceptable Use Policy](https://llama.meta.com/llama3/use-policy/).",
     "displayName": "Meta: Llama 3.1 70B Instruct",
-    "enabled": true,
+    "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3.1-70b-instruct",
     "maxTokens": undefined,
@@ -439,7 +439,7 @@ It has demonstrated strong performance compared to leading closed-source models
 
 To read more about the model release, [click here](https://ai.meta.com/blog/meta-llama-3/). Usage of this model is subject to [Meta's Acceptable Use Policy](https://llama.meta.com/llama3/use-policy/).",
     "displayName": "Meta: Llama 3.1 8B Instruct",
-    "enabled": true,
+    "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3.1-8b-instruct",
     "maxTokens": undefined,
@@ -456,7 +456,7 @@ It has demonstrated strong performance compared to leading closed-source models
 
 To read more about the model release, [click here](https://ai.meta.com/blog/meta-llama-3/). Usage of this model is subject to [Meta's Acceptable Use Policy](https://llama.meta.com/llama3/use-policy/).",
     "displayName": "Meta: Llama 3.1 405B Instruct",
-    "enabled": true,
+    "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3.1-405b-instruct",
     "maxTokens": undefined,

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

@@ -1,9 +1,9 @@
+import type { ChatModelCard } from '@/types/llm';
+
 import { ModelProvider } from '../types';
 import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
 import { OpenRouterModelCard } from './type';
 
-import type { ChatModelCard } from '@/types/llm';
-
 export const LobeOpenRouterAI = LobeOpenAICompatibleFactory({
   baseURL: 'https://openrouter.ai/api/v1',
   chatCompletion: {
@@ -11,6 +11,7 @@ export const LobeOpenRouterAI = LobeOpenAICompatibleFactory({
       return {
         ...payload,
         include_reasoning: true,
+        model: payload.enabledSearch ? `${payload.model}:online` : payload.model,
         stream: payload.stream ?? true,
       } as any;
     },
@@ -27,10 +28,7 @@ export const LobeOpenRouterAI = LobeOpenAICompatibleFactory({
   models: async ({ client }) => {
     const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
 
-    const visionKeywords = [
-      'qwen/qvq',
-      'vision'
-    ];
+    const visionKeywords = ['qwen/qvq', 'vision'];
 
     const reasoningKeywords = [
       'deepseek/deepseek-r1',
@@ -41,12 +39,14 @@ export const LobeOpenRouterAI = LobeOpenAICompatibleFactory({
       'thinking',
     ];
 
-    const modelsPage = await client.models.list() as any;
+    const modelsPage = (await client.models.list()) as any;
     const modelList: OpenRouterModelCard[] = modelsPage.data;
 
     return modelList
       .map((model) => {
-        const knownModel = LOBE_DEFAULT_MODEL_LIST.find((m) => model.id.toLowerCase() === m.id.toLowerCase());
+        const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
+          (m) => model.id.toLowerCase() === m.id.toLowerCase(),
+        );
 
         return {
           contextWindowTokens: model.context_length,
@@ -54,25 +54,25 @@ export const LobeOpenRouterAI = LobeOpenAICompatibleFactory({
           displayName: model.name,
           enabled: knownModel?.enabled || false,
           functionCall:
-            model.description.includes('function calling')
-            || model.description.includes('tools')
-            || knownModel?.abilities?.functionCall
-            || false,
+            model.description.includes('function calling') ||
+            model.description.includes('tools') ||
+            knownModel?.abilities?.functionCall ||
+            false,
           id: model.id,
           maxTokens:
             typeof model.top_provider.max_completion_tokens === 'number'
               ? model.top_provider.max_completion_tokens
               : undefined,
           reasoning:
-            reasoningKeywords.some(keyword => model.id.toLowerCase().includes(keyword))
-            || knownModel?.abilities?.reasoning
-            || false,
+            reasoningKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
+            knownModel?.abilities?.reasoning ||
+            false,
           vision:
-            model.description.includes('vision')
-            || model.description.includes('multimodal')
-            || visionKeywords.some(keyword => model.id.toLowerCase().includes(keyword))
-            || knownModel?.abilities?.vision
-            || false,
+            model.description.includes('vision') ||
+            model.description.includes('multimodal') ||
+            visionKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
+            knownModel?.abilities?.vision ||
+            false,
         };
       })
       .filter(Boolean) as ChatModelCard[];

package/src/libs/agent-runtime/perplexity/index.test.ts

@@ -203,8 +203,8 @@ describe('LobePerplexityAI', () => {
       expect(stream).toEqual(
         [
           'id: 506d64fb-e7f2-4d94-b80f-158369e9446d',
-          'event: citations',
-          'data: [{"title":"https://www.weather.com.cn/weather/101210101.shtml","url":"https://www.weather.com.cn/weather/101210101.shtml"},{"title":"https://tianqi.moji.com/weather/china/zhejiang/hangzhou","url":"https://tianqi.moji.com/weather/china/zhejiang/hangzhou"},{"title":"https://weather.cma.cn/web/weather/58457.html","url":"https://weather.cma.cn/web/weather/58457.html"},{"title":"https://tianqi.so.com/weather/101210101","url":"https://tianqi.so.com/weather/101210101"},{"title":"https://www.accuweather.com/zh/cn/hangzhou/106832/weather-forecast/106832","url":"https://www.accuweather.com/zh/cn/hangzhou/106832/weather-forecast/106832"},{"title":"https://www.hzqx.com","url":"https://www.hzqx.com"},{"title":"https://www.hzqx.com/pc/hztq/","url":"https://www.hzqx.com/pc/hztq/"}]\n',
+          'event: grounding',
+          'data: {"citations":[{"title":"https://www.weather.com.cn/weather/101210101.shtml","url":"https://www.weather.com.cn/weather/101210101.shtml"},{"title":"https://tianqi.moji.com/weather/china/zhejiang/hangzhou","url":"https://tianqi.moji.com/weather/china/zhejiang/hangzhou"},{"title":"https://weather.cma.cn/web/weather/58457.html","url":"https://weather.cma.cn/web/weather/58457.html"},{"title":"https://tianqi.so.com/weather/101210101","url":"https://tianqi.so.com/weather/101210101"},{"title":"https://www.accuweather.com/zh/cn/hangzhou/106832/weather-forecast/106832","url":"https://www.accuweather.com/zh/cn/hangzhou/106832/weather-forecast/106832"},{"title":"https://www.hzqx.com","url":"https://www.hzqx.com"},{"title":"https://www.hzqx.com/pc/hztq/","url":"https://www.hzqx.com/pc/hztq/"}]}\n',
           'id: 506d64fb-e7f2-4d94-b80f-158369e9446d',
           'event: text',
           'data: "杭州今"\n',

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

@@ -1,24 +1,13 @@
+import type { ChatModelCard } from '@/types/llm';
+
 import { ModelProvider } from '../types';
 import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
-
 import { QwenAIStream } from '../utils/streams';
 
-import type { ChatModelCard } from '@/types/llm';
-
 export interface QwenModelCard {
   id: string;
 }
 
-/*
-  QwenEnableSearchModelSeries: An array of Qwen model series that support the enable_search parameter.
-  Currently, enable_search is only supported on Qwen commercial series, excluding Qwen-VL and Qwen-Long series.
-*/
-export const QwenEnableSearchModelSeries = [
-  'qwen-max',
-  'qwen-plus',
-  'qwen-turbo',
-];
-
 /*
   QwenLegacyModels: A set of legacy Qwen models that do not support presence_penalty.
   Currently, presence_penalty is only supported on Qwen commercial models and open-source models starting from Qwen 1.5 and later.
@@ -35,30 +24,34 @@ export const LobeQwenAI = LobeOpenAICompatibleFactory({
   baseURL: 'https://dashscope.aliyuncs.com/compatible-mode/v1',
   chatCompletion: {
     handlePayload: (payload) => {
-      const { model, presence_penalty, temperature, top_p, ...rest } = payload;
+      const { model, presence_penalty, temperature, top_p, enabledSearch, ...rest } = payload;
 
       return {
         ...rest,
         frequency_penalty: undefined,
         model,
-        presence_penalty:
-          QwenLegacyModels.has(model)
-            ? undefined
-            : (presence_penalty !== undefined && presence_penalty >= -2 && presence_penalty <= 2)
-              ? presence_penalty
-              : undefined,
+        presence_penalty: QwenLegacyModels.has(model)
+          ? undefined
+          : presence_penalty !== undefined && presence_penalty >= -2 && presence_penalty <= 2
+            ? presence_penalty
+            : undefined,
         stream: !payload.tools,
-        temperature: (temperature !== undefined && temperature >= 0 && temperature < 2) ? temperature : undefined,
-        ...(model.startsWith('qvq') || model.startsWith('qwen-vl') ? {
-          top_p: (top_p !== undefined && top_p > 0 && top_p <= 1) ? top_p : undefined,
-        } : {
-          top_p: (top_p !== undefined && top_p > 0 && top_p < 1) ? top_p : undefined,
-        }),
-        ...(process.env.QWEN_ENABLE_SEARCH === '1' && QwenEnableSearchModelSeries.some(prefix => model.startsWith(prefix)) && {
-          enable_search: true,
+        temperature:
+          temperature !== undefined && temperature >= 0 && temperature < 2
+            ? temperature
+            : undefined,
+        ...(model.startsWith('qvq') || model.startsWith('qwen-vl')
+          ? {
+              top_p: top_p !== undefined && top_p > 0 && top_p <= 1 ? top_p : undefined,
+            }
+          : {
+              top_p: top_p !== undefined && top_p > 0 && top_p < 1 ? top_p : undefined,
+            }),
+        ...(enabledSearch && {
+          enable_search: enabledSearch,
           search_options: {
             search_strategy: process.env.QWEN_SEARCH_STRATEGY || 'standard', // standard or pro
-          }
+          },
         }),
         ...(payload.tools && {
           parallel_tool_calls: true,
@@ -73,48 +66,38 @@ export const LobeQwenAI = LobeOpenAICompatibleFactory({
   models: async ({ client }) => {
     const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
 
-    const functionCallKeywords = [
-      'qwen-max',
-      'qwen-plus',
-      'qwen-turbo',
-      'qwen2.5',
-    ];
+    const functionCallKeywords = ['qwen-max', 'qwen-plus', 'qwen-turbo', 'qwen2.5'];
 
-    const visionKeywords = [
-      'qvq',
-      'vl',
-    ];
+    const visionKeywords = ['qvq', 'vl'];
 
-    const reasoningKeywords = [
-      'qvq',
-      'qwq',
-      'deepseek-r1'
-    ];
+    const reasoningKeywords = ['qvq', 'qwq', 'deepseek-r1'];
 
-    const modelsPage = await client.models.list() as any;
+    const modelsPage = (await client.models.list()) as any;
     const modelList: QwenModelCard[] = modelsPage.data;
 
     return modelList
       .map((model) => {
-        const knownModel = LOBE_DEFAULT_MODEL_LIST.find((m) => model.id.toLowerCase() === m.id.toLowerCase());
+        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:
-            functionCallKeywords.some(keyword => model.id.toLowerCase().includes(keyword))
-            || knownModel?.abilities?.functionCall
-            || false,
+            functionCallKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
+            knownModel?.abilities?.functionCall ||
+            false,
           id: model.id,
           reasoning:
-            reasoningKeywords.some(keyword => model.id.toLowerCase().includes(keyword))
-            || knownModel?.abilities?.reasoning
-            || false,
+            reasoningKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
+            knownModel?.abilities?.reasoning ||
+            false,
           vision:
-            visionKeywords.some(keyword => model.id.toLowerCase().includes(keyword))
-            || knownModel?.abilities?.vision
-            || false,
+            visionKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
+            knownModel?.abilities?.vision ||
+            false,
         };
       })
       .filter(Boolean) as ChatModelCard[];

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

@@ -38,6 +38,10 @@ export interface OpenAIChatMessage {
  * @title Chat Stream Payload
  */
 export interface ChatStreamPayload {
+  /**
+   * 是否开启搜索
+   */
+  enabledSearch?: boolean;
   /**
    * @title 控制生成文本中的惩罚系数，用于减少重复性
    * @default 0
@@ -68,11 +72,11 @@ export interface ChatStreamPayload {
    * @default 0
    */
   presence_penalty?: number;
+
   /**
    * @default openai
    */
   provider?: string;
-
   responseMode?: 'streamText' | 'json';
   /**
    * @title 是否开启流式请求
@@ -85,8 +89,8 @@ export interface ChatStreamPayload {
    */
   temperature: number;
   tool_choice?: string;
-  tools?: ChatCompletionTool[];
 
+  tools?: ChatCompletionTool[];
   /**
    * @title 控制生成文本中最高概率的单个令牌
    * @default 1

package/src/libs/agent-runtime/utils/streams/google-ai.ts

@@ -1,5 +1,6 @@
 import { EnhancedGenerateContentResponse } from '@google/generative-ai';
 
+import { GroundingSearch } from '@/types/search';
 import { nanoid } from '@/utils/uuid';
 
 import { ChatStreamCallbacks } from '../../types';
@@ -14,8 +15,8 @@ import {
 
 const transformGoogleGenerativeAIStream = (
   chunk: EnhancedGenerateContentResponse,
-  stack: StreamContext,
-): StreamProtocolChunk => {
+  context: StreamContext,
+): StreamProtocolChunk | StreamProtocolChunk[] => {
   // maybe need another structure to add support for multiple choices
   const functionCalls = chunk.functionCalls();
 
@@ -32,15 +33,39 @@ const transformGoogleGenerativeAIStream = (
           type: 'function',
         }),
       ),
-      id: stack.id,
+      id: context.id,
       type: 'tool_calls',
     };
   }
   const text = chunk.text();
 
+  if (chunk.candidates && chunk.candidates[0].groundingMetadata) {
+    const { webSearchQueries, groundingSupports, groundingChunks } =
+      chunk.candidates[0].groundingMetadata;
+    console.log({ groundingChunks, groundingSupports, webSearchQueries });
+
+    return [
+      { data: text, id: context.id, type: 'text' },
+      {
+        data: {
+          citations: groundingChunks?.map((chunk) => ({
+            // google 返回的 uri 是经过 google 自己处理过的 url，因此无法展现真实的 favicon
+            // 需要使用 title 作为替换
+            favicon: chunk.web?.title,
+            title: chunk.web?.title,
+            url: chunk.web?.uri,
+          })),
+          searchQueries: webSearchQueries,
+        } as GroundingSearch,
+        id: context.id,
+        type: 'grounding',
+      },
+    ];
+  }
+
   return {
     data: text,
-    id: stack?.id,
+    id: context?.id,
     type: 'text',
   };
 };

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

@@ -127,7 +127,7 @@ export const transformOpenAIStream = (
           );
 
           return [
-            { data: citations, id: chunk.id, type: 'citations' },
+            { data: { citations }, id: chunk.id, type: 'grounding' },
             { data: content, id: chunk.id, type: 'text' },
           ];
         }

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

@@ -30,7 +30,7 @@ export interface StreamProtocolChunk {
     // Model Thinking
     | 'reasoning'
     // Search or Grounding
-    | 'citations'
+    | 'grounding'
     // stop signal
     | 'stop'
     // Error

package/src/locales/default/chat.ts

@@ -32,6 +32,9 @@ export default {
   },
   duplicateTitle: '{{title}} 副本',
   emptyAgent: '暂无助手',
+  extendControls: {
+    title: '模型扩展功能',
+  },
   historyRange: '历史范围',
   historySummary: '历史消息总结',
   inbox: {
@@ -86,6 +89,31 @@ export default {
   },
   regenerate: '重新生成',
   roleAndArchive: '角色与记录',
+  search: {
+    grounding: {
+      searchQueries: '搜索关键词',
+      title: '已搜索到 {{count}} 个结果',
+    },
+
+    mode: {
+      auto: {
+        desc: '根据对话内容智能判断是否需要搜索',
+        title: '智能联网',
+      },
+      disable: '当前模型不支持函数调用，因此无法使用智能联网功能',
+      off: {
+        desc: '仅使用模型的基础知识，不进行网络搜索',
+        title: '关闭联网',
+      },
+      on: {
+        desc: '持续进行网络搜索，获取最新信息',
+        title: '始终联网',
+      },
+      useModelBuiltin: '使用模型内置搜索引擎',
+    },
+
+    title: '联网搜索',
+  },
   searchAgentPlaceholder: '搜索助手...',
   sendPlaceholder: '输入聊天内容...',
   sessionGroup: {

package/src/services/chat.ts

@@ -18,6 +18,7 @@ import {
 import { filesPrompts } from '@/prompts/files';
 import { BuiltinSystemRolePrompts } from '@/prompts/systemRole';
 import { aiModelSelectors, aiProviderSelectors, useAiInfraStore } from '@/store/aiInfra';
+import { getAgentChatConfig } from '@/store/chat/slices/aiChat/actions/helpers';
 import { useSessionStore } from '@/store/session';
 import { sessionMetaSelectors } from '@/store/session/selectors';
 import { useToolStore } from '@/store/tool';
@@ -224,6 +225,8 @@ class ChatService {
 
     const { provider = ModelProvider.OpenAI, ...res } = params;
 
+    // =================== process model =================== //
+    // ===================================================== //
     let model = res.model || DEFAULT_AGENT_CONFIG.model;
 
     // if the provider is Azure, get the deployment name as the request model
@@ -238,6 +241,13 @@ class ChatService {
       model = findDeploymentName(model, provider);
     }
 
+    // =================== process search =================== //
+    // ===================================================== //
+    const chatConfig = getAgentChatConfig();
+    if (chatConfig.searchMode !== 'off') {
+      res.enabledSearch = true;
+    }
+
     const payload = merge(
       { model: DEFAULT_AGENT_CONFIG.model, stream: true, ...DEFAULT_AGENT_CONFIG.params },
       { ...res, model },

package/src/store/agent/slices/chat/__snapshots__/selectors.test.ts.snap

@@ -9,6 +9,7 @@ exports[`agentSelectors > defaultAgentConfig > should merge DEFAULT_AGENT_CONFIG
     "enableCompressHistory": true,
     "enableHistoryCount": true,
     "historyCount": 8,
+    "searchMode": "off",
   },
   "model": "gpt-3.5-turbo",
   "params": {

package/src/store/agent/slices/chat/selectors.ts

@@ -107,6 +107,8 @@ const currentEnabledKnowledge = (s: AgentStore) => {
   ] as KnowledgeItem[];
 };
 
+const agentSearchMode = (s: AgentStore) => currentAgentChatConfig(s).searchMode || 'off';
+
 const hasSystemRole = (s: AgentStore) => {
   const config = currentAgentConfig(s);
 
@@ -140,7 +142,10 @@ const currentKnowledgeIds = (s: AgentStore) => {
 
 const isAgentConfigLoading = (s: AgentStore) => !s.agentConfigInitMap[s.activeId];
 
+const isAgentEnableSearch = (s: AgentStore) => agentSearchMode(s) !== 'off';
+
 export const agentSelectors = {
+  agentSearchMode,
   currentAgentChatConfig,
   currentAgentConfig,
   currentAgentFiles,
@@ -160,5 +165,6 @@ export const agentSelectors = {
   inboxAgentConfig,
   inboxAgentModel,
   isAgentConfigLoading,
+  isAgentEnableSearch,
   isInboxSession,
 };