@lobehub/chat 1.62.11 → 1.63.0

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;