@lobehub/chat 1.62.11 → 1.63.0

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

package/src/store/aiInfra/slices/aiModel/selectors.ts

@@ -1,5 +1,6 @@
 import { AIProviderStoreState } from '@/store/aiInfra/initialState';
 import { AiModelSourceEnum } from '@/types/aiModel';
+import { ModelSearchImplement } from '@/types/search';
 
 const aiProviderChatModelListIds = (s: AIProviderStoreState) =>
   s.aiProviderModelList.filter((item) => item.type === 'chat').map((item) => item.id);
@@ -69,20 +70,55 @@ const modelContextWindowTokens = (id: string, provider: string) => (s: AIProvide
   return model?.contextWindowTokens;
 };
 
+const modelExtendControls = (id: string, provider: string) => (s: AIProviderStoreState) => {
+  const model = getEnabledModelById(id, provider)(s);
+
+  return model?.settings?.extendControls;
+};
+
+const isModelHasExtendControls = (id: string, provider: string) => (s: AIProviderStoreState) => {
+  const controls = modelExtendControls(id, provider)(s);
+
+  return !!controls && controls.length > 0;
+};
+
+const isModelHasBuiltinSearch = (id: string, provider: string) => (s: AIProviderStoreState) => {
+  const model = getEnabledModelById(id, provider)(s);
+
+  return !!model?.settings?.searchImpl;
+};
+
+const isModelHasBuiltinSearchConfig =
+  (id: string, provider: string) => (s: AIProviderStoreState) => {
+    const model = getEnabledModelById(id, provider)(s);
+
+    return (
+      !!model?.settings?.searchImpl &&
+      [ModelSearchImplement.Tool, ModelSearchImplement.Params].includes(
+        model?.settings?.searchImpl as ModelSearchImplement,
+      )
+    );
+  };
+
 export const aiModelSelectors = {
   aiProviderChatModelListIds,
   disabledAiProviderModelList,
   enabledAiProviderModelList,
   filteredAiProviderModelList,
   getAiModelById,
+  getEnabledModelById,
   hasRemoteModels,
   isEmptyAiProviderModelList,
   isModelEnabled,
+  isModelHasBuiltinSearch,
+  isModelHasBuiltinSearchConfig,
   isModelHasContextWindowToken,
+  isModelHasExtendControls,
   isModelLoading,
   isModelSupportReasoning,
   isModelSupportToolUse,
   isModelSupportVision,
   modelContextWindowTokens,
+  modelExtendControls,
   totalAiProviderModelList,
 };

package/src/store/aiInfra/slices/aiProvider/initialState.ts

@@ -1,10 +1,10 @@
-import { EnabledProviderWithModels } from '@/types/aiModel';
+import { EnabledAiModel } from '@/types/aiModel';
 import {
   AiProviderDetailItem,
   AiProviderListItem,
   AiProviderRuntimeConfig,
-  EnabledAiModel,
   EnabledProvider,
+  EnabledProviderWithModels,
 } from '@/types/aiProvider';
 
 export interface AIProviderState {

package/src/store/aiInfra/slices/aiProvider/selectors.ts

@@ -87,6 +87,18 @@ const providerKeyVaults = (provider: string | undefined) => (s: AIProviderStoreS
   return s.aiProviderRuntimeConfig?.[provider]?.keyVaults;
 };
 
+const isProviderHasBuiltinSearch = (provider: string) => (s: AIProviderStoreState) => {
+  const config = providerConfigById(provider)(s);
+
+  return !!config?.settings.searchMode;
+};
+
+const isProviderHasBuiltinSearchConfig = (id: string) => (s: AIProviderStoreState) => {
+  const providerCfg = providerConfigById(id)(s);
+
+  return !!providerCfg?.settings.searchMode && providerCfg?.settings.searchMode !== 'internal';
+};
+
 export const aiProviderSelectors = {
   activeProviderConfig,
   disabledAiProviderList,
@@ -97,6 +109,8 @@ export const aiProviderSelectors = {
   isProviderConfigUpdating,
   isProviderEnabled,
   isProviderFetchOnClient,
+  isProviderHasBuiltinSearch,
+  isProviderHasBuiltinSearchConfig,
   isProviderLoading,
   providerConfigById,
   providerKeyVaults,

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

@@ -455,7 +455,7 @@ export const generateAIChat: StateCreator<
         await messageService.updateMessageError(messageId, error);
         await refreshMessages();
       },
-      onFinish: async (content, { traceId, observationId, toolCalls, reasoning, citations }) => {
+      onFinish: async (content, { traceId, observationId, toolCalls, reasoning, grounding }) => {
         // if there is traceId, update it
         if (traceId) {
           msgTraceId = traceId;
@@ -473,19 +473,29 @@ export const generateAIChat: StateCreator<
         await internal_updateMessageContent(messageId, content, {
           toolCalls,
           reasoning: !!reasoning ? { content: reasoning, duration } : undefined,
-          search: !!citations ? { citations } : undefined,
+          search: !!grounding?.citations ? grounding : undefined,
         });
       },
       onMessageHandle: async (chunk) => {
         switch (chunk.type) {
-          case 'citations': {
+          case 'grounding': {
             // if there is no citations, then stop
-            if (!chunk.citations || chunk.citations.length <= 0) return;
+            if (
+              !chunk.grounding ||
+              !chunk.grounding.citations ||
+              chunk.grounding.citations.length <= 0
+            )
+              return;
 
             internal_dispatchMessage({
               id: messageId,
               type: 'updateMessage',
-              value: { search: { citations: chunk.citations } },
+              value: {
+                search: {
+                  citations: chunk.grounding.citations,
+                  searchQueries: chunk.grounding.searchQueries,
+                },
+              },
             });
             break;
           }

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

@@ -16,10 +16,10 @@ import {
   ChatMessage,
   ChatMessageError,
   CreateMessageParams,
-  GroundingSearch,
   MessageToolCall,
   ModelReasoning,
 } from '@/types/message';
+import { GroundingSearch } from '@/types/search';
 import { TraceEventPayloads } from '@/types/trace';
 import { setNamespace } from '@/utils/storeDebug';
 import { nanoid } from '@/utils/uuid';

package/src/store/user/slices/modelList/selectors/modelProvider.ts

@@ -1,7 +1,7 @@
 import { uniqBy } from 'lodash-es';
 
 import { filterEnabledModels } from '@/config/modelProviders';
-import { EnabledProviderWithModels } from '@/types/aiModel';
+import { EnabledProviderWithModels } from '@/types/aiProvider';
 import { ChatModelCard, ModelProviderCard } from '@/types/llm';
 import { ServerModelProviderConfig } from '@/types/serverConfig';
 import { GlobalLLMProviderKey } from '@/types/user/settings';

package/src/store/user/slices/settings/selectors/__snapshots__/settings.test.ts.snap

@@ -76,6 +76,7 @@ exports[`settingsSelectors > defaultAgent > should merge DEFAULT_AGENT and s.set
       "enableCompressHistory": true,
       "enableHistoryCount": true,
       "historyCount": 8,
+      "searchMode": "off",
     },
     "model": "gpt-3.5-turbo",
     "params": {