@lobehub/chat 1.62.10 → 1.63.0

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": {

package/src/types/agent/index.ts

@@ -3,6 +3,7 @@ import { z } from 'zod';
 import { FileItem } from '@/types/files';
 import { KnowledgeBaseItem } from '@/types/knowledgeBase';
 import { FewShots, LLMParams } from '@/types/llm';
+import { SearchMode } from '@/types/search';
 
 export type TTSServer = 'openai' | 'edge' | 'microsoft';
 
@@ -78,6 +79,8 @@ export interface LobeAgentChatConfig {
    */
   historyCount?: number;
   inputTemplate?: string;
+  searchMode?: SearchMode;
+  useModelBuiltinSearch?: boolean;
 }
 
 export const AgentChatConfigSchema = z.object({
@@ -89,6 +92,7 @@ export const AgentChatConfigSchema = z.object({
   enableMaxTokens: z.boolean().optional(),
   enableReasoningEffort: z.boolean().optional(),
   historyCount: z.number().optional(),
+  searchMode: z.enum(['off', 'on', 'auto']).optional(),
 });
 
 export type LobeAgentConfigKeys =

package/src/types/aiModel.ts

@@ -1,7 +1,5 @@
 import { z } from 'zod';
 
-import { AiProviderSourceType } from '@/types/aiProvider';
-
 export type ModelPriceCurrency = 'CNY' | 'USD';
 
 export const AiModelSourceEnum = {
@@ -133,6 +131,29 @@ export interface AiModelConfig {
    * used in azure and doubao
    */
   deploymentName?: string;
+
+  /**
+   * qwen series model enabled search
+   */
+  enabledSearch?: boolean;
+}
+
+export interface ExtendedControl {
+  key: string;
+  requestParams: string | string[];
+  type: 'params' | 'tool';
+  valueType: 'boolean';
+}
+
+export type ModelSearchImplementType = 'tool' | 'params' | 'internal';
+
+export interface AiModelSettings {
+  extendControls?: ExtendedControl[];
+  /**
+   * 模型层实现搜索的方式
+   */
+  searchImpl?: ModelSearchImplementType;
+  searchProvider?: string;
 }
 
 export interface AIChatModelCard extends AIBaseModelCard {
@@ -140,6 +161,7 @@ export interface AIChatModelCard extends AIBaseModelCard {
   config?: AiModelConfig;
   maxOutput?: number;
   pricing?: ChatModelPricing;
+  settings?: AiModelSettings;
   type: 'chat';
 }
 
@@ -306,17 +328,22 @@ export type ToggleAiModelEnableParams = z.infer<typeof ToggleAiModelEnableSchema
 
 //
 
-interface AiModelForSelect {
+export interface AiModelForSelect {
   abilities: ModelAbilities;
   contextWindowTokens?: number;
   displayName?: string;
   id: string;
 }
 
-export interface EnabledProviderWithModels {
-  children: AiModelForSelect[];
+export interface EnabledAiModel {
+  abilities: ModelAbilities;
+  config?: AiModelConfig;
+  contextWindowTokens?: number;
+  displayName?: string;
+  enabled?: boolean;
   id: string;
-  logo?: string;
-  name: string;
-  source: AiProviderSourceType;
+  providerId: string;
+  settings?: AiModelSettings;
+  sort?: number;
+  type: AiModelType;
 }

package/src/types/aiProvider.ts

@@ -1,6 +1,6 @@
 import { z } from 'zod';
 
-import { AiModelConfig, AiModelType, ModelAbilities } from '@/types/aiModel';
+import { AiModelForSelect, EnabledAiModel, ModelSearchImplementType } from '@/types/aiModel';
 import { SmoothingParams } from '@/types/llm';
 
 export const AiProviderSourceEnum = {
@@ -96,6 +96,7 @@ export interface AiProviderSettings {
    * default openai
    */
   sdkType?: AiProviderSDKType;
+  searchMode?: ModelSearchImplementType;
   showAddNewModel?: boolean;
   /**
    * whether show api key in the provider config
@@ -199,16 +200,12 @@ export interface EnabledProvider {
   source: AiProviderSourceType;
 }
 
-export interface EnabledAiModel {
-  abilities: ModelAbilities;
-  config?: AiModelConfig;
-  contextWindowTokens?: number;
-  displayName?: string;
-  enabled?: boolean;
+export interface EnabledProviderWithModels {
+  children: AiModelForSelect[];
   id: string;
-  providerId: string;
-  sort?: number;
-  type: AiModelType;
+  logo?: string;
+  name: string;
+  source: AiProviderSourceType;
 }
 
 export interface AiProviderRuntimeConfig {

package/src/types/message/base.ts

@@ -1,3 +1,5 @@
+import { GroundingSearch } from '@/types/search';
+
 export interface CitationItem {
   id?: string;
   onlyUrl?: boolean;
@@ -5,11 +7,6 @@ export interface CitationItem {
   url: string;
 }
 
-export interface GroundingSearch {
-  citations?: CitationItem[];
-  searchQueries?: string[];
-}
-
 export interface ModelReasoning {
   content?: string;
   duration?: number;

package/src/types/message/chat.ts

@@ -2,11 +2,13 @@ import { IPluginErrorType } from '@lobehub/chat-plugin-sdk';
 
 import { ILobeAgentRuntimeErrorType } from '@/libs/agent-runtime';
 import { ErrorType } from '@/types/fetch';
-import { GroundingSearch, MessageRoleType, ModelReasoning } from '@/types/message/base';
-import { ChatPluginPayload, ChatToolPayload } from '@/types/message/tools';
-import { Translate } from '@/types/message/translate';
 import { MetaData } from '@/types/meta';
 import { MessageSemanticSearchChunk } from '@/types/rag';
+import { GroundingSearch } from '@/types/search';
+
+import { MessageRoleType, ModelReasoning } from './base';
+import { ChatPluginPayload, ChatToolPayload } from './tools';
+import { Translate } from './translate';
 
 /**
  * 聊天消息错误对象

package/src/types/openai/chat.ts

@@ -42,6 +42,10 @@ export interface OpenAIChatMessage {
  * @title Chat Stream Payload
  */
 export interface ChatStreamPayload {
+  /**
+   * 是否开启搜索
+   */
+  enabledSearch?: boolean;
   /**
    * @title 控制生成文本中的惩罚系数，用于减少重复性
    * @default 0
@@ -64,6 +68,7 @@ export interface ChatStreamPayload {
    */
   n?: number;
   /**
+   * @deprecated
    * 开启的插件列表
    */
   plugins?: string[];

package/src/types/search.ts

@@ -0,0 +1,29 @@
+export type SearchMode = 'off' | 'auto' | 'on';
+
+export enum ModelSearchImplement {
+  /**
+   * 模型内置了搜索功能
+   * 类似 Jina 、PPLX 等模型的搜索模式，让调用方无感知
+   */
+  Internal = 'internal',
+  /**
+   * 使用参数开关的方式，例如 Qwen、Google、OpenRouter，搜索结果在
+   */
+  Params = 'params',
+  /**
+   * 使用工具调用的方式
+   */
+  Tool = 'tool',
+}
+
+export interface CitationItem {
+  favicon?: string;
+  id?: string;
+  title?: string;
+  url: string;
+}
+
+export interface GroundingSearch {
+  citations?: CitationItem[];
+  searchQueries?: string[];
+}

package/src/utils/fetch/fetchSSE.ts

@@ -6,11 +6,11 @@ import { ChatErrorType } from '@/types/fetch';
 import { SmoothingParams } from '@/types/llm';
 import {
   ChatMessageError,
-  CitationItem,
   MessageToolCall,
   MessageToolCallChunk,
   MessageToolCallSchema,
 } from '@/types/message';
+import { GroundingSearch } from '@/types/search';
 
 import { fetchEventSource } from './fetchEventSource';
 import { getMessageError } from './parseError';
@@ -21,7 +21,7 @@ type SSEFinishType = 'done' | 'error' | 'abort';
 export type OnFinishHandler = (
   text: string,
   context: {
-    citations?: CitationItem[];
+    grounding?: GroundingSearch;
     observationId?: string | null;
     reasoning?: string;
     toolCalls?: MessageToolCall[];
@@ -40,9 +40,9 @@ export interface MessageReasoningChunk {
   type: 'reasoning';
 }
 
-export interface MessageCitationsChunk {
-  citations: CitationItem[];
-  type: 'citations';
+export interface MessageGroundingChunk {
+  grounding: GroundingSearch;
+  type: 'grounding';
 }
 
 interface MessageToolCallsChunk {
@@ -57,7 +57,7 @@ export interface FetchSSEOptions {
   onErrorHandle?: (error: ChatMessageError) => void;
   onFinish?: OnFinishHandler;
   onMessageHandle?: (
-    chunk: MessageTextChunk | MessageToolCallsChunk | MessageReasoningChunk | MessageCitationsChunk,
+    chunk: MessageTextChunk | MessageToolCallsChunk | MessageReasoningChunk | MessageGroundingChunk,
   ) => void;
   smoothing?: SmoothingParams | boolean;
 }
@@ -286,7 +286,7 @@ export const fetchSSE = async (url: string, options: RequestInit & FetchSSEOptio
     startSpeed: smoothingSpeed,
   });
 
-  let citations: CitationItem[] | undefined = undefined;
+  let grounding: GroundingSearch | undefined = undefined;
   await fetchEventSource(url, {
     body: options.body,
     fetch: options?.fetcher,
@@ -359,9 +359,9 @@ export const fetchSSE = async (url: string, options: RequestInit & FetchSSEOptio
           break;
         }
 
-        case 'citations': {
-          citations = data;
-          options.onMessageHandle?.({ citations: data, type: 'citations' });
+        case 'grounding': {
+          grounding = data;
+          options.onMessageHandle?.({ grounding: data, type: 'grounding' });
           break;
         }
 
@@ -434,7 +434,7 @@ export const fetchSSE = async (url: string, options: RequestInit & FetchSSEOptio
       }
 
       await options?.onFinish?.(output, {
-        citations,
+        grounding,
         observationId,
         reasoning: !!thinking ? thinking : undefined,
         toolCalls,

package/src/features/ChatInput/ActionBar/ModelSwitch.tsx

@@ -1,20 +0,0 @@
-import { ActionIcon } from '@lobehub/ui';
-import { Brain } from 'lucide-react';
-import { memo } from 'react';
-import { useTranslation } from 'react-i18next';
-
-import ModelSwitchPanel from '@/features/ModelSwitchPanel';
-
-const ModelSwitch = memo(() => {
-  const { t } = useTranslation('chat');
-
-  return (
-    <ModelSwitchPanel>
-      <ActionIcon icon={Brain} placement={'bottom'} title={t('ModelSwitch.title')} />
-    </ModelSwitchPanel>
-  );
-});
-
-ModelSwitch.displayName = 'ModelSwitch';
-
-export default ModelSwitch;