@lobehub/chat 1.11.4 → 1.11.6

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

@@ -2,6 +2,8 @@ import { readableFromAsyncIterable } from 'ai';
 import OpenAI from 'openai';
 import type { Stream } from 'openai/streaming';
 
+import { ChatMessageError } from '@/types/message';
+
 import { ChatStreamCallbacks } from '../../types';
 import {
   StreamProtocolChunk,
@@ -15,53 +17,74 @@ import {
 export const transformOpenAIStream = (chunk: OpenAI.ChatCompletionChunk): StreamProtocolChunk => {
   // maybe need another structure to add support for multiple choices
 
-  const item = chunk.choices[0];
-  if (!item) {
-    return { data: chunk, id: chunk.id, type: 'data' };
-  }
+  try {
+    const item = chunk.choices[0];
+    if (!item) {
+      return { data: chunk, id: chunk.id, type: 'data' };
+    }
 
-  if (typeof item.delta?.content === 'string') {
-    return { data: item.delta.content, id: chunk.id, type: 'text' };
-  }
+    if (typeof item.delta?.content === 'string') {
+      return { data: item.delta.content, id: chunk.id, type: 'text' };
+    }
+
+    if (item.delta?.tool_calls) {
+      return {
+        data: item.delta.tool_calls.map(
+          (value, index): StreamToolCallChunkData => ({
+            function: value.function,
+            id: value.id || generateToolCallId(index, value.function?.name),
+
+            // mistral's tool calling don't have index and function field, it's data like:
+            // [{"id":"xbhnmTtY7","function":{"name":"lobe-image-designer____text2image____builtin","arguments":"{\"prompts\": [\"A photo of a small, fluffy dog with a playful expression and wagging tail.\", \"A watercolor painting of a small, energetic dog with a glossy coat and bright eyes.\", \"A vector illustration of a small, adorable dog with a short snout and perky ears.\", \"A drawing of a small, scruffy dog with a mischievous grin and a wagging tail.\"], \"quality\": \"standard\", \"seeds\": [123456, 654321, 111222, 333444], \"size\": \"1024x1024\", \"style\": \"vivid\"}"}}]
+
+            // minimax's tool calling don't have index field, it's data like:
+            // [{"id":"call_function_4752059746","type":"function","function":{"name":"lobe-image-designer____text2image____builtin","arguments":"{\"prompts\": [\"一个流浪的地球，背景是浩瀚"}}]
+
+            // so we need to add these default values
+            index: typeof value.index !== 'undefined' ? value.index : index,
+            type: value.type || 'function',
+          }),
+        ),
+        id: chunk.id,
+        type: 'tool_calls',
+      } as StreamProtocolToolCallChunk;
+    }
 
-  if (item.delta?.tool_calls) {
+    // 给定结束原因
+    if (item.finish_reason) {
+      return { data: item.finish_reason, id: chunk.id, type: 'stop' };
+    }
+
+    if (item.delta?.content === null) {
+      return { data: item.delta, id: chunk.id, type: 'data' };
+    }
+
+    // 其余情况下，返回 delta 和 index
     return {
-      data: item.delta.tool_calls.map(
-        (value, index): StreamToolCallChunkData => ({
-          function: value.function,
-          id: value.id || generateToolCallId(index, value.function?.name),
-
-          // mistral's tool calling don't have index and function field, it's data like:
-          // [{"id":"xbhnmTtY7","function":{"name":"lobe-image-designer____text2image____builtin","arguments":"{\"prompts\": [\"A photo of a small, fluffy dog with a playful expression and wagging tail.\", \"A watercolor painting of a small, energetic dog with a glossy coat and bright eyes.\", \"A vector illustration of a small, adorable dog with a short snout and perky ears.\", \"A drawing of a small, scruffy dog with a mischievous grin and a wagging tail.\"], \"quality\": \"standard\", \"seeds\": [123456, 654321, 111222, 333444], \"size\": \"1024x1024\", \"style\": \"vivid\"}"}}]
-
-          // minimax's tool calling don't have index field, it's data like:
-          // [{"id":"call_function_4752059746","type":"function","function":{"name":"lobe-image-designer____text2image____builtin","arguments":"{\"prompts\": [\"一个流浪的地球，背景是浩瀚"}}]
-
-          // so we need to add these default values
-          index: typeof value.index !== 'undefined' ? value.index : index,
-          type: value.type || 'function',
-        }),
-      ),
+      data: { delta: item.delta, id: chunk.id, index: item.index },
       id: chunk.id,
-      type: 'tool_calls',
-    } as StreamProtocolToolCallChunk;
-  }
+      type: 'data',
+    };
+  } catch (e) {
+    const errorName = 'StreamChunkError';
+    console.error(`[${errorName}]`, e);
+    console.error(`[${errorName}] raw chunk:`, chunk);
 
-  // 给定结束原因
-  if (item.finish_reason) {
-    return { data: item.finish_reason, id: chunk.id, type: 'stop' };
-  }
+    const err = e as Error;
 
-  if (item.delta?.content === null) {
-    return { data: item.delta, id: chunk.id, type: 'data' };
-  }
+    /* eslint-disable sort-keys-fix/sort-keys-fix */
+    const errorData = {
+      body: {
+        message:
+          'chat response streaming chunk parse error, please contact your API Provider to fix it.',
+        context: { error: { message: err.message, name: err.name }, chunk },
+      },
+      type: 'StreamChunkError',
+    } as ChatMessageError;
+    /* eslint-enable */
 
-  // 其余情况下，返回 delta 和 index
-  return {
-    data: { delta: item.delta, id: chunk.id, index: item.index },
-    id: chunk.id,
-    type: 'data',
-  };
+    return { data: errorData, id: chunk.id, type: 'error' };
+  }
 };
 
 const chatStreamable = async function* (stream: AsyncIterable<OpenAI.ChatCompletionChunk>) {

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

@@ -13,7 +13,7 @@ export interface StreamStack {
 export interface StreamProtocolChunk {
   data: any;
   id?: string;
-  type: 'text' | 'tool_calls' | 'data' | 'stop';
+  type: 'text' | 'tool_calls' | 'data' | 'stop' | 'error';
 }
 
 export interface StreamToolCallChunkData {

package/src/libs/trpc/client/edge.ts

@@ -13,6 +13,7 @@ export const edgeClient = createTRPCClient<EdgeRouter>({
 
         return createHeaderWithAuth();
       },
+      maxURLLength: 2083,
       transformer: superjson,
       url: withBasePath('/trpc/edge'),
     }),

package/src/libs/trpc/middleware/userAuth.ts

@@ -6,6 +6,7 @@ export const userAuth = trpc.middleware(async (opts) => {
   const { ctx } = opts;
   // `ctx.user` is nullable
   if (!ctx.userId) {
+    console.log('clerk auth:', ctx.clerkAuth);
     throw new TRPCError({ code: 'UNAUTHORIZED' });
   }
 

package/src/locales/default/error.ts

@@ -85,10 +85,15 @@ export default {
      * @deprecated
      */
     NoOpenAIAPIKey: 'OpenAI API Key 不正确或为空，请添加自定义 OpenAI API Key',
+    /**
+     * @deprecated
+     */
     OpenAIBizError: '请求 OpenAI 服务出错，请根据以下信息排查或重试',
 
     InvalidBedrockCredentials: 'Bedrock 鉴权未通过，请检查 AccessKeyId/SecretAccessKey 后重试',
-
+    StreamChunkError:
+      '流式请求的消息块解析错误，请检查当前 API 接口是否符合标准规范，或联系你的 API 供应商咨询',
+    UnknownChatFetchError: '很抱歉，遇到未知请求错误，请根据以下信息排查或重试',
     InvalidOllamaArgs: 'Ollama 配置不正确，请检查 Ollama 配置后重试',
     OllamaBizError: '请求 Ollama 服务出错，请根据以下信息排查或重试',
     OllamaServiceUnavailable:

package/src/server/ld.ts

@@ -1,11 +1,9 @@
 import urlJoin from 'url-join';
 
-import { getAppConfig } from '@/config/app';
 import { EMAIL_BUSINESS, EMAIL_SUPPORT, OFFICIAL_SITE, OFFICIAL_URL, X } from '@/const/url';
 
 import pkg from '../../package.json';
 
-const { SITE_URL = OFFICIAL_URL } = getAppConfig();
 const LAST_MODIFIED = new Date().toISOString();
 export const AUTHOR_LIST = {
   arvinxx: {
@@ -70,7 +68,7 @@ class Ld {
 
   genOrganization() {
     return {
-      '@id': this.getId(SITE_URL, '#organization'),
+      '@id': this.getId(OFFICIAL_URL, '#organization'),
       '@type': 'Organization',
       'alternateName': 'LobeChat',
       'contactPoint': {
@@ -102,7 +100,7 @@ class Ld {
 
   getAuthors(ids: string[] = []) {
     const defaultAuthor = {
-      '@id': this.getId(SITE_URL, '#organization'),
+      '@id': this.getId(OFFICIAL_URL, '#organization'),
       '@type': 'Organization',
     };
     if (!ids || ids.length === 0) return defaultAuthor;
@@ -142,7 +140,7 @@ class Ld {
       '@id': fixedUrl,
       '@type': 'WebPage',
       'about': {
-        '@id': this.getId(SITE_URL, '#organization'),
+        '@id': this.getId(OFFICIAL_URL, '#organization'),
       },
       'breadcrumbs': {
         '@id': this.getId(fixedUrl, '#breadcrumb'),
@@ -155,7 +153,7 @@ class Ld {
       },
       'inLanguage': 'en-US',
       'isPartOf': {
-        '@id': this.getId(SITE_URL, '#website'),
+        '@id': this.getId(OFFICIAL_URL, '#website'),
       },
       'name': this.fixTitle(title),
       'primaryImageOfPage': {
@@ -188,15 +186,15 @@ class Ld {
 
   genWebSite() {
     const baseInfo: any = {
-      '@id': this.getId(SITE_URL, '#website'),
+      '@id': this.getId(OFFICIAL_URL, '#website'),
       '@type': 'WebSite',
       'description': pkg.description,
       'inLanguage': 'en-US',
       'name': 'LobeChat',
       'publisher': {
-        '@id': this.getId(SITE_URL, '#organization'),
+        '@id': this.getId(OFFICIAL_URL, '#organization'),
       },
-      'url': SITE_URL,
+      'url': OFFICIAL_URL,
     };
 
     return baseInfo;
@@ -211,7 +209,7 @@ class Ld {
   }
 
   private fixUrl(url: string) {
-    return urlJoin(SITE_URL, url);
+    return urlJoin(OFFICIAL_URL, url);
   }
 }
 

package/src/services/__tests__/chat.test.ts

@@ -577,7 +577,6 @@ Get data from users`,
         body: JSON.stringify(expectedPayload),
         headers: expect.any(Object),
         method: 'POST',
-        signal: expect.any(AbortSignal),
       });
     });
 

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

@@ -4,6 +4,7 @@ import { SWRResponse, mutate } from 'swr';
 import { DeepPartial } from 'utility-types';
 import { StateCreator } from 'zustand/vanilla';
 
+import { MESSAGE_CANCEL_FLAT } from '@/const/message';
 import { INBOX_SESSION_ID } from '@/const/session';
 import { DEFAULT_AGENT_CONFIG } from '@/const/settings';
 import { useClientDataSWR, useOnlyFetchOnceSWR } from '@/libs/swr';
@@ -168,7 +169,7 @@ export const createChatSlice: StateCreator<
 
   internal_createAbortController: (key) => {
     const abortController = get()[key] as AbortController;
-    if (abortController) abortController.abort('canceled');
+    if (abortController) abortController.abort(MESSAGE_CANCEL_FLAT);
     const controller = new AbortController();
     set({ [key]: controller }, false, 'internal_createAbortController');
 

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

@@ -958,10 +958,10 @@ describe('chatMessage actions', () => {
     it('should not do anything if there is no abortController', async () => {
       const { result } = renderHook(() => useChatStore());
 
-      // 确保没有设置 abortController
-      useChatStore.setState({ abortController: undefined });
-
       await act(async () => {
+        // 确保没有设置 abortController
+        useChatStore.setState({ abortController: undefined });
+
         result.current.stopGenerateMessage();
       });
 
@@ -1089,7 +1089,7 @@ describe('chatMessage actions', () => {
 
       // Mock fetch to reject with an error
       const errorMessage = 'Error fetching AI response';
-      vi.mocked(fetch).mockRejectedValue(new Error(errorMessage));
+      vi.mocked(fetch).mockRejectedValueOnce(new Error(errorMessage));
 
       await act(async () => {
         expect(

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

@@ -7,7 +7,7 @@ import { template } from 'lodash-es';
 import { SWRResponse, mutate } from 'swr';
 import { StateCreator } from 'zustand/vanilla';
 
-import { LOADING_FLAT } from '@/const/message';
+import { LOADING_FLAT, MESSAGE_CANCEL_FLAT } from '@/const/message';
 import { TraceEventType, TraceNameMap } from '@/const/trace';
 import { useClientDataSWR } from '@/libs/swr';
 import { chatService } from '@/services/chat';
@@ -399,7 +399,7 @@ export const chatMessage: StateCreator<
     const { abortController, internal_toggleChatLoading } = get();
     if (!abortController) return;
 
-    abortController.abort();
+    abortController.abort(MESSAGE_CANCEL_FLAT);
 
     internal_toggleChatLoading(false, undefined, n('stopGenerateMessage') as string);
   },

package/src/store/session/slices/session/action.ts

@@ -5,6 +5,7 @@ import { DeepPartial } from 'utility-types';
 import { StateCreator } from 'zustand/vanilla';
 
 import { message } from '@/components/AntdStaticMethods';
+import { MESSAGE_CANCEL_FLAT } from '@/const/message';
 import { DEFAULT_AGENT_LOBE_SESSION, INBOX_SESSION_ID } from '@/const/session';
 import { useClientDataSWR } from '@/libs/swr';
 import { sessionService } from '@/services/session';
@@ -188,7 +189,7 @@ export const createSessionSlice: StateCreator<
     const { activeId, refreshSessions } = get();
 
     const abortController = get().signalSessionMeta as AbortController;
-    if (abortController) abortController.abort('canceled');
+    if (abortController) abortController.abort(MESSAGE_CANCEL_FLAT);
     const controller = new AbortController();
     set({ signalSessionMeta: controller }, false, 'updateSessionMetaSignal');
 

package/src/store/tool/slices/plugin/action.ts

@@ -2,6 +2,7 @@ import { Schema, ValidationResult } from '@cfworker/json-schema';
 import useSWR, { SWRResponse } from 'swr';
 import { StateCreator } from 'zustand/vanilla';
 
+import { MESSAGE_CANCEL_FLAT } from '@/const/message';
 import { pluginService } from '@/services/plugin';
 import { merge } from '@/utils/merge';
 
@@ -46,7 +47,7 @@ export const createPluginSlice: StateCreator<
   },
   updatePluginSettings: async (id, settings) => {
     const signal = get().updatePluginSettingsSignal;
-    if (signal) signal.abort('canceled');
+    if (signal) signal.abort(MESSAGE_CANCEL_FLAT);
 
     const newSignal = new AbortController();
 

package/src/store/user/slices/settings/action.ts

@@ -3,6 +3,7 @@ import isEqual from 'fast-deep-equal';
 import { DeepPartial } from 'utility-types';
 import type { StateCreator } from 'zustand/vanilla';
 
+import { MESSAGE_CANCEL_FLAT } from '@/const/message';
 import { shareService } from '@/services/share';
 import { userService } from '@/services/user';
 import type { UserStore } from '@/store/user';
@@ -63,7 +64,8 @@ export const createSettingsSlice: StateCreator<
 
   internal_createSignal: () => {
     const abortController = get().updateSettingsSignal;
-    if (abortController && !abortController.signal.aborted) abortController.abort('canceled');
+    if (abortController && !abortController.signal.aborted)
+      abortController.abort(MESSAGE_CANCEL_FLAT);
 
     const newSignal = new AbortController();
 

package/src/types/fetch.ts

@@ -12,6 +12,7 @@ export const ChatErrorType = {
   NoOpenAIAPIKey: 'NoOpenAIAPIKey',
   OllamaServiceUnavailable: 'OllamaServiceUnavailable', // 未启动/检测到 Ollama 服务
   PluginFailToTransformArguments: 'PluginFailToTransformArguments',
+  UnknownChatFetchError: 'UnknownChatFetchError',
 
   // ******* 客户端错误 ******* //
   BadRequest: 400,

package/src/utils/downloadFile.ts

@@ -0,0 +1,18 @@
+export const downloadFile = async (url: string, fileName: string) => {
+  try {
+    const res = await fetch(url);
+    const blob = await res.blob();
+
+    const blobUrl = window.URL.createObjectURL(blob);
+    const link = document.createElement('a');
+    link.href = blobUrl;
+    link.download = fileName;
+    link.style.display = 'none';
+    document.body.append(link);
+    link.click();
+    link.remove();
+    window.URL.revokeObjectURL(blobUrl);
+  } catch (error) {
+    console.error('Download failed:', error);
+  }
+};