@lobehub/chat 1.123.3 → 1.124.0

package/src/store/chat/slices/aiChat/actions/__tests__/generateAIChatV2.test.ts

@@ -187,18 +187,21 @@ describe('generateAIChatV2 actions', () => {
         await result.current.sendMessage({ message, files });
       });
 
-      expect(aiChatService.sendMessageInServer).toHaveBeenCalledWith({
-        newAssistantMessage: {
-          model: DEFAULT_MODEL,
-          provider: DEFAULT_PROVIDER,
+      expect(aiChatService.sendMessageInServer).toHaveBeenCalledWith(
+        {
+          newAssistantMessage: {
+            model: DEFAULT_MODEL,
+            provider: DEFAULT_PROVIDER,
+          },
+          newUserMessage: {
+            content: message,
+            files: files.map((f) => f.id),
+          },
+          sessionId: mockState.activeId,
+          topicId: mockState.activeTopicId,
         },
-        newUserMessage: {
-          content: message,
-          files: files.map((f) => f.id),
-        },
-        sessionId: mockState.activeId,
-        topicId: mockState.activeTopicId,
-      });
+        expect.anything(),
+      );
       expect(result.current.internal_execAgentRuntime).toHaveBeenCalled();
     });
 
@@ -270,18 +273,21 @@ describe('generateAIChatV2 actions', () => {
         await result.current.sendMessage({ message: '', files: [{ id: 'file-1' }] as any });
       });
 
-      expect(aiChatService.sendMessageInServer).toHaveBeenCalledWith({
-        newAssistantMessage: {
-          model: DEFAULT_MODEL,
-          provider: DEFAULT_PROVIDER,
-        },
-        newUserMessage: {
-          content: '',
-          files: ['file-1'],
+      expect(aiChatService.sendMessageInServer).toHaveBeenCalledWith(
+        {
+          newAssistantMessage: {
+            model: DEFAULT_MODEL,
+            provider: DEFAULT_PROVIDER,
+          },
+          newUserMessage: {
+            content: '',
+            files: ['file-1'],
+          },
+          sessionId: 'session-id',
+          topicId: 'topic-id',
         },
-        sessionId: 'session-id',
-        topicId: 'topic-id',
-      });
+        expect.anything(),
+      );
     });
 
     it('当同时有文件和消息内容时,正确发送消息并关联文件', async () => {
@@ -291,18 +297,21 @@ describe('generateAIChatV2 actions', () => {
         await result.current.sendMessage({ message: 'test', files: [{ id: 'file-1' }] as any });
       });
 
-      expect(aiChatService.sendMessageInServer).toHaveBeenCalledWith({
-        newAssistantMessage: {
-          model: DEFAULT_MODEL,
-          provider: DEFAULT_PROVIDER,
-        },
-        newUserMessage: {
-          content: 'test',
-          files: ['file-1'],
+      expect(aiChatService.sendMessageInServer).toHaveBeenCalledWith(
+        {
+          newAssistantMessage: {
+            model: DEFAULT_MODEL,
+            provider: DEFAULT_PROVIDER,
+          },
+          newUserMessage: {
+            content: 'test',
+            files: ['file-1'],
+          },
+          sessionId: 'session-id',
+          topicId: 'topic-id',
         },
-        sessionId: 'session-id',
-        topicId: 'topic-id',
-      });
+        expect.anything(),
+      );
     });
 
     it('当 createMessage 抛出错误时,正确处理错误而不影响整个应用', async () => {

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

@@ -7,9 +7,12 @@ import {
   ChatTopic,
   MessageSemanticSearchChunk,
   SendMessageParams,
+  SendMessageServerResponse,
   TraceNameMap,
 } from '@lobechat/types';
+import { TRPCClientError } from '@trpc/client';
 import { t } from 'i18next';
+import { produce } from 'immer';
 import { StateCreator } from 'zustand/vanilla';
 
 import { aiChatService } from '@/services/aiChat';
@@ -18,6 +21,7 @@ import { messageService } from '@/services/message';
 import { getAgentStoreState } from '@/store/agent';
 import { agentChatConfigSelectors, agentSelectors } from '@/store/agent/slices/chat';
 import { aiModelSelectors, aiProviderSelectors, getAiInfraStoreState } from '@/store/aiInfra';
+import { MainSendMessageOperation } from '@/store/chat/slices/aiChat/initialState';
 import type { ChatStore } from '@/store/chat/store';
 import { getSessionStoreState } from '@/store/session';
 import { WebBrowsingManifest } from '@/tools/web-browsing';
@@ -33,6 +37,11 @@ export interface AIGenerateV2Action {
    * Sends a new message to the AI chat system
    */
   sendMessageInServer: (params: SendMessageParams) => Promise<void>;
+  /**
+   * Cancels sendMessageInServer operation for a specific topic/session
+   */
+  cancelSendMessageInServer: (topicId?: string) => void;
+  clearSendMessageError: () => void;
   internal_refreshAiChat: (params: {
     topics?: ChatTopic[];
     messages: ChatMessage[];
@@ -57,6 +66,19 @@ export interface AIGenerateV2Action {
     inPortalThread?: boolean;
     traceId?: string;
   }) => Promise<void>;
+  /**
+   * Toggle sendMessageInServer operation state
+   */
+  internal_toggleSendMessageOperation: (
+    key: string | { sessionId: string; topicId?: string | null },
+    loading: boolean,
+    cancelReason?: string,
+  ) => AbortController | undefined;
+  internal_updateSendMessageOperation: (
+    key: string | { sessionId: string; topicId?: string | null },
+    value: Partial<MainSendMessageOperation> | null,
+    actionName?: any,
+  ) => void;
 }
 
 export const generateAIChatV2: StateCreator<
@@ -66,7 +88,8 @@ export const generateAIChatV2: StateCreator<
   AIGenerateV2Action
 > = (set, get) => ({
   sendMessageInServer: async ({ message, files, onlyAddUserMessage, isWelcomeQuestion }) => {
-    const { activeTopicId, activeId, activeThreadId, internal_execAgentRuntime } = get();
+    const { activeTopicId, activeId, activeThreadId, internal_execAgentRuntime, mainInputEditor } =
+      get();
     if (!activeId) return;
 
     const fileIdList = files?.map((f) => f.id);
@@ -95,44 +118,78 @@ export const generateAIChatV2: StateCreator<
       topicId: activeTopicId,
       threadId: activeThreadId,
     });
-
     get().internal_toggleMessageLoading(true, tempId);
-    set({ isCreatingMessage: true }, false, 'creatingMessage/start');
 
-    const { model, provider } = agentSelectors.currentAgentConfig(getAgentStoreState());
+    const operationKey = messageMapKey(activeId, activeTopicId);
 
-    const data = await aiChatService.sendMessageInServer({
-      newUserMessage: {
-        content: message,
-        files: fileIdList,
-      },
-      // if there is activeTopicId，then add topicId to message
-      topicId: activeTopicId,
-      threadId: activeThreadId,
-      newTopic: !activeTopicId
-        ? {
-            topicMessageIds: messages.map((m) => m.id),
-            title: t('defaultTitle', { ns: 'topic' }),
-          }
-        : undefined,
-      sessionId: activeId === INBOX_SESSION_ID ? undefined : activeId,
-      newAssistantMessage: { model, provider: provider! },
-    });
+    // Start tracking sendMessageInServer operation with AbortController
+    const abortController = get().internal_toggleSendMessageOperation(operationKey, true)!;
 
-    // refresh the total data
-    get().internal_refreshAiChat({
-      messages: data.messages,
-      topics: data.topics,
-      sessionId: activeId,
-      topicId: data.topicId,
-    });
-    get().internal_dispatchMessage({ type: 'deleteMessage', id: tempId });
+    const jsonState = mainInputEditor?.getJSONState();
+    get().internal_updateSendMessageOperation(
+      operationKey,
+      { inputSendErrorMsg: undefined, inputEditorTempState: jsonState },
+      'creatingMessage/start',
+    );
+
+    let data: SendMessageServerResponse | undefined;
+    try {
+      const { model, provider } = agentSelectors.currentAgentConfig(getAgentStoreState());
+      data = await aiChatService.sendMessageInServer(
+        {
+          newUserMessage: {
+            content: message,
+            files: fileIdList,
+          },
+          // if there is activeTopicId，then add topicId to message
+          topicId: activeTopicId,
+          threadId: activeThreadId,
+          newTopic: !activeTopicId
+            ? {
+                topicMessageIds: messages.map((m) => m.id),
+                title: t('defaultTitle', { ns: 'topic' }),
+              }
+            : undefined,
+          sessionId: activeId === INBOX_SESSION_ID ? undefined : activeId,
+          newAssistantMessage: { model, provider: provider! },
+        },
+        abortController,
+      );
+      // refresh the total data
+      get().internal_refreshAiChat({
+        messages: data.messages,
+        topics: data.topics,
+        sessionId: activeId,
+        topicId: data.topicId,
+      });
 
-    if (!activeTopicId) {
-      await get().switchTopic(data.topicId!, true);
+      if (!activeTopicId) {
+        await get().switchTopic(data.topicId!, true);
+      }
+    } catch (e) {
+      if (e instanceof TRPCClientError) {
+        const isAbort = e.message.includes('aborted') || e.name === 'AbortError';
+        // Check if error is due to cancellation
+        if (!isAbort) {
+          get().internal_updateSendMessageOperation(operationKey, { inputSendErrorMsg: e.message });
+          get().mainInputEditor?.setJSONState(jsonState);
+        }
+      }
+    } finally {
+      // Stop tracking sendMessageInServer operation
+      get().internal_toggleSendMessageOperation(operationKey, false);
     }
 
+    // remove temporally message
+    get().internal_dispatchMessage({ type: 'deleteMessage', id: tempId });
     get().internal_toggleMessageLoading(false, tempId);
+    get().internal_updateSendMessageOperation(
+      operationKey,
+      { inputEditorTempState: null },
+      'creatingMessage/finished',
+    );
+
+    if (!data) return;
 
     //  update assistant update to make it rerank
     getSessionStoreState().triggerSessionUpdate(get().activeId);
@@ -143,6 +200,7 @@ export const generateAIChatV2: StateCreator<
       .activeBaseChats(get())
       .filter((item) => item.id !== data.assistantMessageId);
 
+    if (data.topicId) get().internal_updateTopicLoading(data.topicId, true);
     try {
       await internal_execAgentRuntime({
         messages: baseMessages,
@@ -152,7 +210,6 @@ export const generateAIChatV2: StateCreator<
         ragQuery: get().internal_shouldUseRAG() ? message : undefined,
         threadId: activeThreadId,
       });
-      set({ isCreatingMessage: false }, false, 'creatingMessage/stop');
 
       const summaryTitle = async () => {
         // check activeTopic and then auto update topic title
@@ -161,9 +218,9 @@ export const generateAIChatV2: StateCreator<
           return;
         }
 
-        if (!activeTopicId) return;
+        if (!data.topicId) return;
 
-        const topic = topicSelectors.getTopicById(activeTopicId)(get());
+        const topic = topicSelectors.getTopicById(data.topicId)(get());
 
         if (topic && !topic.title) {
           const chats = chatSelectors.getBaseChatsByKey(messageMapKey(activeId, topic.id))(get());
@@ -181,10 +238,38 @@ export const generateAIChatV2: StateCreator<
       await Promise.all([summaryTitle(), addFilesToAgent()]);
     } catch (e) {
       console.error(e);
-      set({ isCreatingMessage: false }, false, 'creatingMessage/stop');
+    } finally {
+      if (data.topicId) get().internal_updateTopicLoading(data.topicId, false);
     }
   },
 
+  cancelSendMessageInServer: (topicId?: string) => {
+    const { activeId, activeTopicId } = get();
+
+    // Determine which operation to cancel
+    const targetTopicId = topicId ?? activeTopicId;
+    const operationKey = messageMapKey(activeId, targetTopicId);
+
+    // Cancel the specific operation
+    get().internal_toggleSendMessageOperation(
+      operationKey,
+      false,
+      'User cancelled sendMessageInServer operation',
+    );
+
+    // Only clear creating message state if it's the active session
+    if (operationKey === messageMapKey(activeId, activeTopicId)) {
+      const editorTempState = get().mainSendMessageOperations[operationKey]?.inputEditorTempState;
+      get().mainInputEditor?.setJSONState(editorTempState);
+    }
+  },
+  clearSendMessageError: () => {
+    get().internal_updateSendMessageOperation(
+      { sessionId: get().activeId, topicId: get().activeTopicId },
+      null,
+      'clearSendMessageError',
+    );
+  },
   internal_refreshAiChat: ({ topics, messages, sessionId, topicId }) => {
     set(
       {
@@ -407,4 +492,58 @@ export const generateAIChatV2: StateCreator<
       await get().internal_summaryHistory(historyMessages);
     }
   },
+
+  internal_updateSendMessageOperation: (key, value, actionName) => {
+    const operationKey = typeof key === 'string' ? key : messageMapKey(key.sessionId, key.topicId);
+
+    set(
+      produce((draft) => {
+        if (!draft.mainSendMessageOperations[operationKey])
+          draft.mainSendMessageOperations[operationKey] = value;
+        else {
+          if (value === null) {
+            delete draft.mainSendMessageOperations[operationKey];
+          } else {
+            draft.mainSendMessageOperations[operationKey] = {
+              ...draft.mainSendMessageOperations[operationKey],
+              ...value,
+            };
+          }
+        }
+      }),
+      false,
+      actionName ?? n('updateSendMessageOperation', { operationKey, value }),
+    );
+  },
+  internal_toggleSendMessageOperation: (key, loading: boolean, cancelReason?: string) => {
+    if (loading) {
+      const abortController = new AbortController();
+
+      get().internal_updateSendMessageOperation(
+        key,
+        { isLoading: true, abortController },
+        n('toggleSendMessageOperation(start)', { key }),
+      );
+
+      return abortController;
+    } else {
+      const operationKey =
+        typeof key === 'string' ? key : messageMapKey(key.sessionId, key.topicId);
+
+      const operation = get().mainSendMessageOperations[operationKey];
+
+      // If cancelReason is provided, abort the operation first
+      if (cancelReason && operation?.isLoading) {
+        operation.abortController?.abort(cancelReason);
+      }
+
+      get().internal_updateSendMessageOperation(
+        key,
+        { isLoading: false, abortController: null },
+        n('toggleSendMessageOperation(stop)', { key, cancelReason }),
+      );
+
+      return undefined;
+    }
+  },
 });

package/src/store/chat/slices/aiChat/initialState.ts

@@ -1,3 +1,12 @@
+import type { ChatInputEditor } from '@/features/ChatInput';
+
+export interface MainSendMessageOperation {
+  abortController?: AbortController | null;
+  inputEditorTempState?: any | null;
+  inputSendErrorMsg?: string;
+  isLoading: boolean;
+}
+
 export interface ChatAIChatState {
   /**
    * is the AI message is generating
@@ -6,6 +15,12 @@ export interface ChatAIChatState {
   chatLoadingIdsAbortController?: AbortController;
   inputFiles: File[];
   inputMessage: string;
+  mainInputEditor: ChatInputEditor | null;
+  /**
+   * sendMessageInServer operations map, keyed by sessionId|topicId
+   * Contains both loading state and AbortController
+   */
+  mainSendMessageOperations: Record<string, MainSendMessageOperation>;
   messageInToolsCallingIds: string[];
   /**
    * is the message is in RAG flow
@@ -17,6 +32,7 @@ export interface ChatAIChatState {
    */
   reasoningLoadingIds: string[];
   searchWorkflowLoadingIds: string[];
+  threadInputEditor: ChatInputEditor | null;
   /**
    * the tool calling stream ids
    */
@@ -27,10 +43,13 @@ export const initialAiChatState: ChatAIChatState = {
   chatLoadingIds: [],
   inputFiles: [],
   inputMessage: '',
+  mainInputEditor: null,
+  mainSendMessageOperations: {},
   messageInToolsCallingIds: [],
   messageRAGLoadingIds: [],
   pluginApiLoadingIds: [],
   reasoningLoadingIds: [],
   searchWorkflowLoadingIds: [],
+  threadInputEditor: null,
   toolCallingStreamIds: {},
 };

package/src/store/chat/slices/aiChat/selectors.ts

@@ -1,3 +1,5 @@
+import { messageMapKey } from '@/store/chat/utils/messageMapKey';
+
 import type { ChatStoreState } from '../../initialState';
 
 const isMessageInReasoning = (id: string) => (s: ChatStoreState) =>
@@ -9,8 +11,24 @@ const isMessageInSearchWorkflow = (id: string) => (s: ChatStoreState) =>
 const isIntentUnderstanding = (id: string) => (s: ChatStoreState) =>
   isMessageInSearchWorkflow(id)(s);
 
+const isCurrentSendMessageLoading = (s: ChatStoreState) => {
+  const operationKey = messageMapKey(s.activeId, s.activeTopicId);
+  return s.mainSendMessageOperations[operationKey]?.isLoading || false;
+};
+
+const isCurrentSendMessageError = (s: ChatStoreState) => {
+  const operationKey = messageMapKey(s.activeId, s.activeTopicId);
+  return s.mainSendMessageOperations[operationKey]?.inputSendErrorMsg;
+};
+
+const isSendMessageLoadingForTopic = (topicKey: string) => (s: ChatStoreState) =>
+  s.mainSendMessageOperations[topicKey]?.isLoading ?? false;
+
 export const aiChatSelectors = {
+  isCurrentSendMessageError,
+  isCurrentSendMessageLoading,
   isIntentUnderstanding,
   isMessageInReasoning,
   isMessageInSearchWorkflow,
+  isSendMessageLoadingForTopic,
 };

package/src/store/global/action.test.ts

@@ -252,13 +252,14 @@ describe('createPreferenceSlice', () => {
   describe('updatePreference', () => {
     it('should update status', () => {
       const { result } = renderHook(() => useGlobalStore());
-      const status = { inputHeight: 200 };
+      const status = { wideScreen: false };
 
       act(() => {
+        useGlobalStore.setState({ isStatusInit: true });
         result.current.updateSystemStatus(status);
       });
 
-      expect(result.current.status.inputHeight).toEqual(200);
+      expect(result.current.status.wideScreen).toEqual(false);
     });
   });
 
@@ -394,7 +395,7 @@ describe('createPreferenceSlice', () => {
     it('should update with data', async () => {
       const { result } = renderHook(() => useGlobalStore());
       vi.spyOn(useGlobalStore.getState().statusStorage, 'getFromLocalStorage').mockReturnValueOnce({
-        inputHeight: 300,
+        wideScreen: false,
       } as any);
 
       const { result: hooks } = renderHook(() => result.current.useInitSystemStatus(), {
@@ -402,10 +403,10 @@ describe('createPreferenceSlice', () => {
       });
 
       await waitFor(() => {
-        expect(hooks.current.data).toEqual({ inputHeight: 300 });
+        expect(hooks.current.data).toEqual({ wideScreen: false });
       });
 
-      expect(result.current.status.inputHeight).toEqual(300);
+      expect(result.current.status.wideScreen).toEqual(false);
     });
   });
 

package/src/store/global/actions/__tests__/general.test.ts

@@ -34,7 +34,7 @@ describe('generalActionSlice', () => {
       const { result } = renderHook(() => useGlobalStore());
 
       act(() => {
-        result.current.updateSystemStatus({ inputHeight: 200 });
+        result.current.updateSystemStatus({ wideScreen: false });
       });
 
       expect(result.current.status).toEqual(initialState.status);
@@ -45,10 +45,10 @@ describe('generalActionSlice', () => {
 
       act(() => {
         useGlobalStore.setState({ isStatusInit: true });
-        result.current.updateSystemStatus({ inputHeight: 200 });
+        result.current.updateSystemStatus({ wideScreen: false });
       });
 
-      expect(result.current.status.inputHeight).toBe(200);
+      expect(result.current.status.wideScreen).toBe(false);
     });
 
     it('should not update if new status equals current status', () => {
@@ -57,7 +57,7 @@ describe('generalActionSlice', () => {
 
       act(() => {
         useGlobalStore.setState({ isStatusInit: true });
-        result.current.updateSystemStatus({ inputHeight: initialState.status.inputHeight });
+        result.current.updateSystemStatus({ wideScreen: initialState.status.wideScreen });
       });
 
       expect(saveToLocalStorageSpy).not.toHaveBeenCalled();
@@ -69,11 +69,11 @@ describe('generalActionSlice', () => {
 
       act(() => {
         useGlobalStore.setState({ isStatusInit: true });
-        result.current.updateSystemStatus({ inputHeight: 300 });
+        result.current.updateSystemStatus({ wideScreen: false });
       });
 
       expect(saveToLocalStorageSpy).toHaveBeenCalledWith(
-        expect.objectContaining({ inputHeight: 300 }),
+        expect.objectContaining({ wideScreen: false }),
       );
     });
 

package/src/store/global/actions/workspacePane.ts

@@ -16,6 +16,7 @@ export interface GlobalWorkspacePaneAction {
   toggleMobilePortal: (visible?: boolean) => void;
   toggleMobileTopic: (visible?: boolean) => void;
   toggleSystemRole: (visible?: boolean) => void;
+  toggleWideScreen: (enable?: boolean) => void;
   toggleZenMode: () => void;
 }
 
@@ -80,6 +81,11 @@ export const globalWorkspaceSlice: StateCreator<
 
     get().updateSystemStatus({ showSystemRole }, n('toggleMobileTopic', newValue));
   },
+  toggleWideScreen: (newValue) => {
+    const wideScreen = typeof newValue === 'boolean' ? newValue : !get().status.wideScreen;
+
+    get().updateSystemStatus({ wideScreen }, n('toggleWideScreen', newValue));
+  },
   toggleZenMode: () => {
     const { status } = get();
     const nextZenMode = !status.zenMode;

package/src/store/global/initialState.ts

@@ -55,7 +55,6 @@ export interface SystemStatus {
   hideThreadLimitAlert?: boolean;
   imagePanelWidth: number;
   imageTopicPanelWidth?: number;
-  inputHeight: number;
   /**
    * 应用初始化时不启用 PGLite，只有当用户手动开启时才启用
    */
@@ -79,7 +78,7 @@ export interface SystemStatus {
    * theme mode
    */
   themeMode?: ThemeMode;
-  threadInputHeight: number;
+  wideScreen?: boolean;
   zenMode?: boolean;
 }
 
@@ -114,7 +113,6 @@ export const INITIAL_STATUS = {
   hideThreadLimitAlert: false,
   imagePanelWidth: 320,
   imageTopicPanelWidth: 80,
-  inputHeight: 200,
   mobileShowTopic: false,
   portalWidth: 400,
   sessionsWidth: 320,
@@ -127,7 +125,7 @@ export const INITIAL_STATUS = {
   showSystemRole: false,
   systemRoleExpandedMap: {},
   themeMode: 'auto',
-  threadInputHeight: 200,
+  wideScreen: true,
   zenMode: false,
 } satisfies SystemStatus;
 

package/src/store/global/selectors/systemStatus.test.ts

@@ -69,8 +69,7 @@ describe('systemStatusSelectors', () => {
       expect(systemStatusSelectors.sessionWidth(s)).toBe(300);
       expect(systemStatusSelectors.portalWidth(s)).toBe(500);
       expect(systemStatusSelectors.filePanelWidth(s)).toBe(400);
-      expect(systemStatusSelectors.inputHeight(s)).toBe(150);
-      expect(systemStatusSelectors.threadInputHeight(s)).toBe(100);
+      expect(systemStatusSelectors.wideScreen(s)).toBe(true);
     });
 
     it('should handle zen mode effects', () => {

package/src/store/global/selectors/systemStatus.ts

@@ -28,9 +28,7 @@ const portalWidth = (s: GlobalState) => s.status.portalWidth || 400;
 const filePanelWidth = (s: GlobalState) => s.status.filePanelWidth;
 const imagePanelWidth = (s: GlobalState) => s.status.imagePanelWidth;
 const imageTopicPanelWidth = (s: GlobalState) => s.status.imageTopicPanelWidth;
-const inputHeight = (s: GlobalState) => s.status.inputHeight;
-const threadInputHeight = (s: GlobalState) => s.status.threadInputHeight;
-
+const wideScreen = (s: GlobalState) => s.status.wideScreen;
 const isPgliteNotEnabled = (s: GlobalState) =>
   isUsePgliteDB && !isServerMode && s.isStatusInit && !s.status.isEnablePglite;
 
@@ -69,7 +67,6 @@ export const systemStatusSelectors = {
   imagePanelWidth,
   imageTopicPanelWidth,
   inZenMode,
-  inputHeight,
   isDBInited,
   isPgliteInited,
   isPgliteNotEnabled,
@@ -90,5 +87,5 @@ export const systemStatusSelectors = {
   showSystemRole,
   systemStatus,
   themeMode,
-  threadInputHeight,
+  wideScreen,
 };

package/src/app/(backend)/webapi/chat/anthropic/route.test.ts

@@ -1,30 +0,0 @@
-// @vitest-environment edge-runtime
-import { describe, expect, it, vi } from 'vitest';
-
-import { POST as UniverseRoute } from '../[provider]/route';
-import { POST, preferredRegion, runtime } from './route';
-
-// 模拟 '../[provider]/route'
-vi.mock('../[provider]/route', () => ({
-  POST: vi.fn().mockResolvedValue('mocked response'),
-}));
-
-describe('Configuration tests', () => {
-  it('should have runtime set to "edge"', () => {
-    expect(runtime).toBe('edge');
-  });
-
-  it('should contain specific regions in preferredRegion', () => {
-    expect(preferredRegion).not.contain(['hk1']);
-  });
-});
-
-describe('Anthropic POST function tests', () => {
-  it('should call UniverseRoute with correct parameters', async () => {
-    const mockRequest = new Request('https://example.com', { method: 'POST' });
-    await POST(mockRequest);
-    expect(UniverseRoute).toHaveBeenCalledWith(mockRequest, {
-      params: Promise.resolve({ provider: 'anthropic' }),
-    });
-  });
-});