@lobehub/chat 1.90.4 → 1.91.1

package/src/database/migrations/meta/_journal.json

@@ -161,6 +161,13 @@
       "when": 1746724476380,
       "tag": "0022_add_documents",
       "breakpoints": true
+    },
+    {
+      "idx": 23,
+      "version": "7",
+      "when": 1748925630721,
+      "tag": "0023_remove_param_and_doubao",
+      "breakpoints": true
     }
   ],
   "version": "6"

package/src/database/repositories/dataImporter/__tests__/fixtures/with-client-id.json

@@ -15,9 +15,7 @@
               "name": "showMJ",
               "parameters": {
                 "type": "object",
-                "required": [
-                  "prompts"
-                ],
+                "required": ["prompts"],
                 "properties": {
                   "prompts": {
                     "type": "string",
@@ -29,11 +27,7 @@
             }
           ],
           "meta": {
-            "tags": [
-              "AI Image Generation",
-              "Midjourney",
-              "Visualization"
-            ],
+            "tags": ["AI Image Generation", "Midjourney", "Visualization"],
             "title": "Midjourney",
             "avatar": "https://cdn.discordapp.com/icons/662267976984297473/39128f6c9fc33f4c95a27d4c601ad7db.webp",
             "description": "Unleash creativity with Midjourney, effortlessly generating unique AI-driven images through simple prompts. Add visual flair to your conversations with Midjourney's artistic touch."
@@ -44,9 +38,7 @@
           "version": "1",
           "settings": {
             "type": "object",
-            "required": [
-              "MIDJOURNEY_PROXY_URL"
-            ],
+            "required": ["MIDJOURNEY_PROXY_URL"],
             "properties": {
               "MIDJOURNEY_PROXY_URL": {
                 "type": "string",
@@ -128,13 +120,7 @@
         "slug": "rapidly-radio-harbor-mountain",
         "title": "CAN",
         "description": "有经验的程序员，精通多种编程语言。提供代码解决方案、指导和实际示例，帮助用户实现他们的编程目标。\"我热爱编程。\"",
-        "tags": [
-          "programming",
-          "coding",
-          "programming-assistance",
-          "code-examples",
-          "guidance"
-        ],
+        "tags": ["programming", "coding", "programming-assistance", "code-examples", "guidance"],
         "avatar": "👨‍💻",
         "backgroundColor": null,
         "plugins": [],
@@ -174,9 +160,7 @@
         "slug": "crowd-society-south-shallow",
         "title": "",
         "description": "",
-        "tags": [
-          ""
-        ],
+        "tags": [""],
         "avatar": "",
         "backgroundColor": "rgba(0,0,0,0)",
         "plugins": [],
@@ -216,18 +200,10 @@
         "slug": "team-energy-settlers-west",
         "title": "量化交易顾问",
         "description": "擅长数字货币市场量化交易与策略优化",
-        "tags": [
-          "数字货币",
-          "量化交易",
-          "市场分析",
-          "风险管理",
-          "策略开发"
-        ],
+        "tags": ["数字货币", "量化交易", "市场分析", "风险管理", "策略开发"],
         "avatar": "📈",
         "backgroundColor": "rgba(0,0,0,0)",
-        "plugins": [
-          "web_search"
-        ],
+        "plugins": ["web_search"],
         "clientId": null,
         "chatConfig": {
           "displayMode": "chat",
@@ -264,11 +240,7 @@
         "slug": "load-essential-likely-travel",
         "title": "中英文互译助手",
         "description": "中英文翻译专家，追求翻译信达雅",
-        "tags": [
-          "翻译",
-          "中文",
-          "英文"
-        ],
+        "tags": ["翻译", "中文", "英文"],
         "avatar": "🌐",
         "backgroundColor": null,
         "plugins": [],
@@ -308,13 +280,7 @@
         "slug": "look-quickly-break-yard",
         "title": "全栈开发助手",
         "description": "此角色擅长全栈开发的深度思考与技术分析。",
-        "tags": [
-          "人工智能",
-          "全栈开发",
-          "思维模式",
-          "技术能力",
-          "解决方案"
-        ],
+        "tags": ["人工智能", "全栈开发", "思维模式", "技术能力", "解决方案"],
         "avatar": "🧩",
         "backgroundColor": "rgba(0,0,0,0)",
         "plugins": [],
@@ -354,9 +320,7 @@
         "slug": "everybody-those-unhappy-age",
         "title": "自适应全能产业顾问",
         "description": "你是一个能够适应各个行业和领域的全能人工智能助手。您的任务是根据用户指定的兴趣领域及其后续问题提供专家建议和信息。",
-        "tags": [
-          "产业专家、技术答疑"
-        ],
+        "tags": ["产业专家、技术答疑"],
         "avatar": "https://blogger.googleusercontent.com/img/b/R29vZ2xl/AVvXsEhJ5XrlGZKwN3Q_hEk139JOvb3Ieg5bC08jOqftLpESRRQ6_v4appLaa55PGR4g_1eK3A73UBrF_PaA8XsfswRgPPShCgZRkG8yHMvEIJNllUq3g14Pok0UGjtNZRVl3PNrLcbLxSfLX7TZ/s550/ai_shigoto_makaseru.png",
         "backgroundColor": null,
         "plugins": [],
@@ -396,10 +360,7 @@
         "slug": "swept-name-church-rays",
         "title": "AI代理生成器",
         "description": "擅长创建满足需求的AI代理角色描述。",
-        "tags": [
-          "ai-agent",
-          "角色创建"
-        ],
+        "tags": ["ai-agent", "角色创建"],
         "avatar": "🤖",
         "backgroundColor": null,
         "plugins": [],
@@ -439,11 +400,7 @@
         "slug": "continued-method-equator-attached",
         "title": "Stack Overflow 编程专家",
         "description": "",
-        "tags": [
-          "编程",
-          "专家",
-          "编程语言"
-        ],
+        "tags": ["编程", "专家", "编程语言"],
         "avatar": "🚀",
         "backgroundColor": null,
         "plugins": [],
@@ -486,9 +443,7 @@
         "tags": [],
         "avatar": null,
         "backgroundColor": null,
-        "plugins": [
-          "midjourney"
-        ],
+        "plugins": ["midjourney"],
         "clientId": null,
         "chatConfig": {
           "searchMode": "off",

package/src/features/ChatInput/ActionBar/Model/ControlsForm.tsx

@@ -14,12 +14,7 @@ import ContextCachingSwitch from './ContextCachingSwitch';
 import ReasoningEffortSlider from './ReasoningEffortSlider';
 import ReasoningTokenSlider from './ReasoningTokenSlider';
 
-interface ControlsProps {
-  setUpdating: (updating: boolean) => void;
-  updating: boolean;
-}
-
-const ControlsForm = memo<ControlsProps>(({ setUpdating }) => {
+const ControlsForm = memo(() => {
   const { t } = useTranslation('chat');
   const [model, provider, updateAgentChatConfig] = useAgentStore((s) => [
     agentSelectors.currentAgentModel(s),
@@ -111,9 +106,7 @@ const ControlsForm = memo<ControlsProps>(({ setUpdating }) => {
       }
       itemsType={'flat'}
       onValuesChange={async (_, values) => {
-        setUpdating(true);
         await updateAgentChatConfig(values);
-        setUpdating(false);
       }}
       size={'small'}
       style={{ fontSize: 12 }}

package/src/features/ChatInput/ActionBar/Model/index.tsx

@@ -1,8 +1,7 @@
 import { ModelIcon } from '@lobehub/icons';
-import { Icon } from '@lobehub/ui';
 import { createStyles } from 'antd-style';
-import { Loader2Icon, Settings2Icon } from 'lucide-react';
-import { memo, useState } from 'react';
+import { Settings2Icon } from 'lucide-react';
+import { memo } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Center, Flexbox } from 'react-layout-kit';
 
@@ -55,9 +54,7 @@ const useStyles = createStyles(({ css, token, cx }) => ({
 
 const ModelSwitch = memo(() => {
   const { t } = useTranslation('chat');
-  const { styles, cx, theme } = useStyles();
-  const [updating, setUpdating] = useState(false);
-  const [controlsUpdating, setControlsUpdating] = useState(false);
+  const { styles, cx } = useStyles();
 
   const [model, provider] = useAgentStore((s) => [
     agentSelectors.currentAgentModel(s),
@@ -70,28 +67,23 @@ const ModelSwitch = memo(() => {
 
   return (
     <Flexbox align={'center'} className={isModelHasExtendParams ? styles.container : ''} horizontal>
-      <ModelSwitchPanel setUpdating={setUpdating} updating={updating}>
+      <ModelSwitchPanel>
         <Center
           className={cx(styles.model, isModelHasExtendParams && styles.modelWithControl)}
           height={36}
           width={36}
         >
-          {updating ? (
-            <Icon color={theme.colorTextDescription} icon={Loader2Icon} size={18} spin />
-          ) : (
-            <div className={styles.icon}>
-              <ModelIcon model={model} size={22} />
-            </div>
-          )}
+          <div className={styles.icon}>
+            <ModelIcon model={model} size={22} />
+          </div>
         </Center>
       </ModelSwitchPanel>
 
       {isModelHasExtendParams && (
         <Action
           icon={Settings2Icon}
-          loading={controlsUpdating}
           popover={{
-            content: <ControlsForm setUpdating={setControlsUpdating} updating={controlsUpdating} />,
+            content: <ControlsForm />,
             minWidth: 350,
             placement: 'topLeft',
           }}

package/src/features/ChatInput/ActionBar/Search/Controls.tsx

@@ -59,11 +59,10 @@ interface NetworkOption {
   disable?: boolean;
   icon: LucideIcon;
   label: string;
-  setUpdating?: (loading: boolean) => void;
   value: SearchMode;
 }
 
-const Item = memo<NetworkOption>(({ value, description, icon, label, setUpdating }) => {
+const Item = memo<NetworkOption>(({ value, description, icon, label }) => {
   const { cx, styles } = useStyles();
   const [mode, updateAgentChatConfig] = useAgentStore((s) => [
     agentChatConfigSelectors.agentSearchMode(s),
@@ -78,9 +77,7 @@ const Item = memo<NetworkOption>(({ value, description, icon, label, setUpdating
       horizontal
       key={value}
       onClick={async () => {
-        setUpdating?.(true);
         await updateAgentChatConfig({ searchMode: value });
-        setUpdating?.(false);
       }}
     >
       <Center className={styles.icon} flex={'none'} height={32} width={32}>
@@ -94,12 +91,7 @@ const Item = memo<NetworkOption>(({ value, description, icon, label, setUpdating
   );
 });
 
-interface ControlsProps {
-  setUpdating: (updating: boolean) => void;
-  updating: boolean;
-}
-
-const Controls = memo<ControlsProps>(({ setUpdating }) => {
+const Controls = memo(() => {
   const { t } = useTranslation('chat');
   const [model, provider] = useAgentStore((s) => [
     agentSelectors.currentAgentModel(s),
@@ -131,11 +123,11 @@ const Controls = memo<ControlsProps>(({ setUpdating }) => {
   return (
     <Flexbox gap={4}>
       {options.map((option) => (
-        <Item setUpdating={setUpdating} {...option} key={option.value} />
+        <Item {...option} key={option.value} />
       ))}
       {showDivider && <Divider style={{ margin: 0 }} />}
       {isModelHasBuiltinSearchConfig && <ModelBuiltinSearch />}
-      {!supportFC && <FCSearchModel setLoading={setUpdating} />}
+      {!supportFC && <FCSearchModel />}
     </Flexbox>
   );
 });

package/src/features/ChatInput/ActionBar/Search/FCSearchModel.tsx

@@ -31,11 +31,7 @@ const useStyles = createStyles(({ css, token }) => ({
   `,
 }));
 
-interface FCSearchModelProps {
-  setLoading?: (loading: boolean) => void;
-}
-
-const FCSearchModel = memo<FCSearchModelProps>(({ setLoading }) => {
+const FCSearchModel = memo(() => {
   const { t } = useTranslation('chat');
   const { styles } = useStyles();
   const [searchFCModel, updateAgentChatConfig] = useAgentStore((s) => [
@@ -50,9 +46,7 @@ const FCSearchModel = memo<FCSearchModelProps>(({ setLoading }) => {
       </Flexbox>
       <FunctionCallingModelSelect
         onChange={async (value) => {
-          setLoading?.(true);
           await updateAgentChatConfig({ searchFCModel: value });
-          setLoading?.(false);
         }}
         style={{
           maxWidth: 160,

package/src/features/ChatInput/ActionBar/Search/index.tsx

@@ -1,7 +1,7 @@
 import { GlobeOffIcon } from '@lobehub/ui/icons';
 import { useTheme } from 'antd-style';
 import { Globe } from 'lucide-react';
-import { memo, useState } from 'react';
+import { memo } from 'react';
 import { useTranslation } from 'react-i18next';
 
 import { isDeprecatedEdition } from '@/const/version';
@@ -15,7 +15,6 @@ import Controls from './Controls';
 const Search = memo(() => {
   const { t } = useTranslation('chat');
   const [isLoading] = useAgentStore((s) => [agentSelectors.isAgentConfigLoading(s)]);
-  const [updating, setUpdating] = useState(false);
   const isAgentEnableSearch = useAgentEnableSearch();
   const theme = useTheme();
 
@@ -26,9 +25,8 @@ const Search = memo(() => {
     <Action
       color={isAgentEnableSearch ? theme.colorInfo : undefined}
       icon={isAgentEnableSearch ? Globe : GlobeOffIcon}
-      loading={updating}
       popover={{
-        content: <Controls setUpdating={setUpdating} updating={updating} />,
+        content: <Controls />,
         maxWidth: 320,
         minWidth: 320,
         placement: 'topLeft',

package/src/features/ModelSwitchPanel/index.tsx

@@ -44,11 +44,10 @@ interface IProps {
   children?: ReactNode;
   onOpenChange?: (open: boolean) => void;
   open?: boolean;
-  setUpdating?: (updating: boolean) => void;
   updating?: boolean;
 }
 
-const ModelSwitchPanel = memo<IProps>(({ children, setUpdating, onOpenChange, open }) => {
+const ModelSwitchPanel = memo<IProps>(({ children, onOpenChange, open }) => {
   const { t } = useTranslation('components');
   const { styles, theme } = useStyles();
   const [model, provider, updateAgentConfig] = useAgentStore((s) => [
@@ -66,9 +65,7 @@ const ModelSwitchPanel = memo<IProps>(({ children, setUpdating, onOpenChange, op
         key: menuKey(provider.id, model.id),
         label: <ModelItemRender {...model} {...model.abilities} />,
         onClick: async () => {
-          setUpdating?.(true);
           await updateAgentConfig({ model: model.id, provider: provider.id });
-          setUpdating?.(false);
         },
       }));
 

package/src/libs/model-runtime/anthropic/index.test.ts

@@ -305,12 +305,14 @@ describe('LobeAnthropicAI', () => {
 
       it('should build payload with tools and web search enabled', async () => {
         const tools: ChatCompletionTool[] = [
-          { function: { name: 'tool1', description: 'desc1' }, type: 'function' }
+          { function: { name: 'tool1', description: 'desc1' }, type: 'function' },
         ];
 
         const mockAnthropicTools = [{ name: 'tool1', description: 'desc1' }];
 
-        vi.spyOn(anthropicHelpers, 'buildAnthropicTools').mockReturnValue(mockAnthropicTools as any);
+        vi.spyOn(anthropicHelpers, 'buildAnthropicTools').mockReturnValue(
+          mockAnthropicTools as any,
+        );
 
         const payload: ChatStreamPayload = {
           messages: [{ content: 'Search and get info', role: 'user' }],

package/src/libs/model-runtime/utils/streams/anthropic.ts

@@ -1,7 +1,7 @@
 import Anthropic from '@anthropic-ai/sdk';
 import type { Stream } from '@anthropic-ai/sdk/streaming';
 
-import { ModelTokensUsage, CitationItem } from '@/types/message';
+import { CitationItem, ModelTokensUsage } from '@/types/message';
 
 import { ChatStreamCallbacks } from '../../types';
 import {
@@ -180,7 +180,7 @@ export const transformAnthropicStream = (
             context.returnedCitationArray.push({
               title: citations.title,
               url: citations.url,
-            } as CitationItem)
+            } as CitationItem);
           }
 
           return { data: null, id: context.id, type: 'text' };
@@ -219,15 +219,16 @@ export const transformAnthropicStream = (
 
     case 'message_stop': {
       return [
-        ...(context.returnedCitationArray?.length 
-          ? [{ 
-              data: { citations: context.returnedCitationArray }, 
-              id: context.id, 
-              type: 'grounding' 
-            }] 
-          : []
-        ),
-        { data: 'message_stop', id: context.id, type: 'stop' }
+        ...(context.returnedCitationArray?.length
+          ? [
+              {
+                data: { citations: context.returnedCitationArray },
+                id: context.id,
+                type: 'grounding',
+              },
+            ]
+          : []),
+        { data: 'message_stop', id: context.id, type: 'stop' },
       ] as any;
     }
 

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

@@ -227,10 +227,12 @@ export const transformOpenAIStream = (
             return [
               {
                 data: {
-                  citations: (citations as any[]).map((item) => ({
-                    title: typeof item === 'string' ? item : item.title,
-                    url: typeof item === 'string' ? item : item.url || item.link,
-                  })).filter(c => c.title && c.url), // Zhipu 内建搜索工具有时会返回空 link 引发程序崩溃
+                  citations: (citations as any[])
+                    .map((item) => ({
+                      title: typeof item === 'string' ? item : item.title,
+                      url: typeof item === 'string' ? item : item.url || item.link,
+                    }))
+                    .filter((c) => c.title && c.url), // Zhipu 内建搜索工具有时会返回空 link 引发程序崩溃
                 },
                 id: chunk.id,
                 type: 'grounding',

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

@@ -18,7 +18,7 @@ export interface StreamContext {
   returnedCitation?: boolean;
   /**
    * Claude's citations are inline and interleaved with text output.
-   * Each text segment may carry references to sources (e.g., web search results) 
+   * Each text segment may carry references to sources (e.g., web search results)
    * relevant to that specific portion of the generated content.
    * This array accumulates all citation items received during the streaming response.
    */

package/src/libs/model-runtime/utils/streams/spark.test.ts

@@ -40,7 +40,7 @@ describe('SparkAIStream', () => {
           },
         ],
       },
-    ]
+    ];
 
     const mockSparkStream = new ReadableStream({
       start(controller) {

package/src/libs/model-runtime/utils/streams/spark.ts

@@ -2,6 +2,7 @@ import OpenAI from 'openai';
 import type { Stream } from 'openai/streaming';
 
 import { ChatStreamCallbacks } from '../../types';
+import { convertUsage } from '../usageConverter';
 import {
   StreamProtocolChunk,
   StreamProtocolToolCallChunk,
@@ -11,8 +12,6 @@ import {
   generateToolCallId,
 } from './protocol';
 
-import { convertUsage } from '../usageConverter';
-
 export function transformSparkResponseToStream(data: OpenAI.ChatCompletion) {
   return new ReadableStream({
     start(controller) {

package/src/middleware.ts

@@ -136,7 +136,7 @@ const defaultMiddleware = (request: NextRequest) => {
 
 const isPublicRoute = createRouteMatcher([
   '/api/auth(.*)',
-  '/trpc/edge(.*)',
+  '/trpc(.*)',
   // next auth
   '/next-auth/(.*)',
   // clerk

package/src/server/services/search/impls/bocha/index.ts

@@ -0,0 +1,124 @@
+import { TRPCError } from '@trpc/server';
+import debug from 'debug';
+import urlJoin from 'url-join';
+
+import { SearchParams, UniformSearchResponse, UniformSearchResult } from '@/types/tool/search';
+
+import { SearchServiceImpl } from '../type';
+import { BochaSearchParameters, BochaResponse } from './type';
+
+const log = debug('lobe-search:Bocha');
+
+const timeRangeMapping = {
+  day: 'oneDay',
+  month: 'oneMonth',
+  week: 'oneWeek',
+  year: 'oneYear',
+};
+
+/**
+ * Bocha implementation of the search service
+ * Primarily used for web crawling
+ */
+export class BochaImpl implements SearchServiceImpl {
+  private get apiKey(): string | undefined {
+    return process.env.BOCHA_API_KEY;
+  }
+
+  private get baseUrl(): string {
+    // Assuming the base URL is consistent with the crawl endpoint
+    return 'https://api.bochaai.com/v1';
+  }
+
+  async query(query: string, params: SearchParams = {}): Promise<UniformSearchResponse> {
+    log('Starting Bocha query with query: "%s", params: %o', query, params);
+    const endpoint = urlJoin(this.baseUrl, '/web-search');
+
+    const defaultQueryParams: BochaSearchParameters = {
+      count: 15,
+      query,
+      summary: true,
+    };
+
+    let body: BochaSearchParameters = {
+      ...defaultQueryParams,
+      freshness:
+        params?.searchTimeRange && params.searchTimeRange !== 'anytime'
+          ? timeRangeMapping[params.searchTimeRange as keyof typeof timeRangeMapping] ?? undefined
+          : undefined,
+    };
+
+    log('Constructed request body: %o', body);
+
+    let response: Response;
+    const startAt = Date.now();
+    let costTime = 0;
+    try {
+      log('Sending request to endpoint: %s', endpoint);
+      response = await fetch(endpoint, {
+        body: JSON.stringify(body),
+        headers: {
+          'Authorization': this.apiKey ? `Bearer ${this.apiKey}` : '',
+          'Content-Type': 'application/json',
+        },
+        method: 'POST',
+      });
+      log('Received response with status: %d', response.status);
+      costTime = Date.now() - startAt;
+    } catch (error) {
+      log.extend('error')('Bocha fetch error: %o', error);
+      throw new TRPCError({
+        cause: error,
+        code: 'SERVICE_UNAVAILABLE',
+        message: 'Failed to connect to Bocha.',
+      });
+    }
+
+    if (!response.ok) {
+      const errorBody = await response.text();
+      log.extend('error')(
+        `Bocha request failed with status ${response.status}: %s`,
+        errorBody.length > 200 ? `${errorBody.slice(0, 200)}...` : errorBody,
+      );
+      throw new TRPCError({
+        cause: errorBody,
+        code: 'SERVICE_UNAVAILABLE',
+        message: `Bocha request failed: ${response.statusText}`,
+      });
+    }
+
+    try {
+      const bochaResponse = (await response.json()) as BochaResponse;
+
+      log('Parsed Bocha response: %o', bochaResponse);
+
+      const mappedResults = (bochaResponse.data.webPages.value || []).map(
+        (result): UniformSearchResult => ({
+          category: 'general', // Default category
+          content: result.summary || result.snippet || '', // Prioritize content, fallback to snippet
+          engines: ['bocha'], // Use 'bocha' as the engine name
+          parsedUrl: result.url ? new URL(result.url).hostname : '', // Basic URL parsing
+          score: 1, // Default score to 1
+          title: result.name || '',
+          url: result.url,
+        }),
+      );
+
+      log('Mapped %d results to SearchResult format', mappedResults.length);
+
+      return {
+        costTime,
+        query: query,
+        resultNumbers: mappedResults.length,
+        results: mappedResults,
+      };
+    } catch (error) {
+      log.extend('error')('Error parsing Bocha response: %o', error);
+      throw new TRPCError({
+        cause: error,
+        code: 'INTERNAL_SERVER_ERROR',
+        message: 'Failed to parse Bocha response.',
+      });
+    }
+  }
+}

package/src/server/services/search/impls/bocha/type.ts

@@ -0,0 +1,47 @@
+export interface BochaSearchParameters {
+  count?: number;
+  exclude?: string;
+  freshness?: string;
+  include?: string;
+  query: string;
+  summary?: boolean;
+}
+
+interface BochaQueryContext {
+  originalQuery: string;
+}
+
+interface BochaValue {
+  cachedPageUrl?: string;
+  dateLastCrawled?: string;
+  displayUrl?: string;
+  id?: string | null;
+  isFamilyFriendly?: boolean;
+  isNavigational?: boolean;
+  language?: string;
+  name: string;
+  siteName?: string;
+  snippet?: string;
+  summary?: string;
+  url: string;
+}
+
+interface BochaWebPages {
+  totalEstimatedMatches?: number;
+  value?: BochaValue[];
+  webSearchUrl?: string;
+}
+
+interface BochaData {
+  images?: any;
+  queryContext?: BochaQueryContext;
+  videos?: any;
+  webPages: BochaWebPages;
+}
+
+export interface BochaResponse {
+  code?: number;
+  data: BochaData;
+  log_id?: string;
+  msg?: string | null;
+}