@lobehub/chat 1.28.6 → 1.29.0

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

@@ -28,6 +28,8 @@ import {
 import { AgentRuntime } from '@/libs/agent-runtime';
 import { useToolStore } from '@/store/tool';
 import { UserStore } from '@/store/user';
+import { useUserStore } from '@/store/user';
+import { modelConfigSelectors } from '@/store/user/selectors';
 import { UserSettingsState, initialSettingsState } from '@/store/user/slices/settings/initialState';
 import { DalleManifest } from '@/tools/dalle';
 import { ChatMessage } from '@/types/message';
@@ -35,8 +37,6 @@ import { ChatStreamPayload, type OpenAIChatMessage } from '@/types/openai/chat';
 import { LobeTool } from '@/types/tool';
 
 import { chatService, initializeWithClientStore } from '../chat';
-import { useUserStore } from '@/store/user';
-import {modelConfigSelectors} from "@/store/user/selectors";
 
 // Mocking external dependencies
 vi.mock('i18next', () => ({
@@ -150,8 +150,6 @@ describe('ChatService', () => {
               },
             ],
           }, // Message with files
-          { content: 'Hi', role: 'tool', plugin: { identifier: 'plugin1', apiName: 'api1' } }, // Message with tool role
-          { content: 'Hey', role: 'assistant' }, // Regular user message
         ] as ChatMessage[];
 
         const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
@@ -177,15 +175,6 @@ describe('ChatService', () => {
                 ],
                 role: 'user',
               },
-              {
-                content: 'Hi',
-                name: 'plugin1____api1',
-                role: 'tool',
-              },
-              {
-                content: 'Hey',
-                role: 'assistant',
-              },
             ],
             model: 'gpt-4-vision-preview',
           },
@@ -196,7 +185,6 @@ describe('ChatService', () => {
       it('should not include image with vision models when can not find the image', async () => {
         const messages = [
           { content: 'Hello', role: 'user', files: ['file2'] }, // Message with files
-          { content: 'Hi', role: 'tool', plugin: { identifier: 'plugin1', apiName: 'api1' } }, // Message with function role
           { content: 'Hey', role: 'assistant' }, // Regular user message
         ] as ChatMessage[];
 
@@ -207,7 +195,6 @@ describe('ChatService', () => {
           {
             messages: [
               { content: 'Hello', role: 'user' },
-              { content: 'Hi', name: 'plugin1____api1', role: 'tool' },
               { content: 'Hey', role: 'assistant' },
             ],
           },
@@ -536,7 +523,9 @@ describe('ChatService', () => {
       };
 
       vi.spyOn(useUserStore, 'getState').mockImplementationOnce(() => mockUserStore as any);
-      vi.spyOn(modelConfigSelectors, 'isProviderFetchOnClient').mockImplementationOnce(mockModelConfigSelectors.isProviderFetchOnClient);
+      vi.spyOn(modelConfigSelectors, 'isProviderFetchOnClient').mockImplementationOnce(
+        mockModelConfigSelectors.isProviderFetchOnClient,
+      );
 
       const params: Partial<ChatStreamPayload> = {
         model: 'test-model',
@@ -550,18 +539,15 @@ describe('ChatService', () => {
         ...params,
       };
 
-      const result = await chatService.getChatCompletion(params,options);
+      const result = await chatService.getChatCompletion(params, options);
 
-      expect(global.fetch).toHaveBeenCalledWith(
-        expect.any(String),
-        {
-          body: JSON.stringify(expectedPayload),
-          headers: expect.objectContaining({
-            'Content-Type': 'application/json',
-          }),
-          method: 'POST',
-        },
-      );
+      expect(global.fetch).toHaveBeenCalledWith(expect.any(String), {
+        body: JSON.stringify(expectedPayload),
+        headers: expect.objectContaining({
+          'Content-Type': 'application/json',
+        }),
+        method: 'POST',
+      });
       expect(result.status).toBe(401);
     });
 
@@ -683,14 +669,6 @@ describe('ChatService', () => {
               id: 'tool_call_nXxXHW8Z',
               type: 'function',
             },
-            {
-              function: {
-                arguments: '{"query":"LobeHub","searchEngines":["bilibili"]}',
-                name: 'lobe-web-browsing____searchWithSearXNG____builtin',
-              },
-              id: 'tool_call_2f3CEKz9',
-              type: 'function',
-            },
           ],
         },
         {
@@ -749,14 +727,6 @@ describe('ChatService', () => {
               id: 'tool_call_nXxXHW8Z',
               type: 'function',
             },
-            {
-              function: {
-                arguments: '{"query":"LobeHub","searchEngines":["bilibili"]}',
-                name: 'lobe-web-browsing____searchWithSearXNG____builtin',
-              },
-              id: 'tool_call_2f3CEKz9',
-              type: 'function',
-            },
           ],
         },
         {
@@ -771,12 +741,6 @@ describe('ChatService', () => {
           role: 'tool',
           tool_call_id: 'tool_call_nXxXHW8Z',
         },
-        {
-          content: '[]',
-          name: 'lobe-web-browsing____searchWithSearXNG____builtin',
-          role: 'tool',
-          tool_call_id: 'tool_call_2f3CEKz9',
-        },
         {
           content: 'LobeHub 是一个专注于设计和开发现代人工智能生成内容（AIGC）工具和组件的团队。',
           role: 'assistant',
@@ -826,7 +790,6 @@ describe('ChatService', () => {
               },
             ],
           }, // Message with files
-          { content: 'Hi', role: 'tool', plugin: { identifier: 'plugin1', apiName: 'api1' } }, // Message with tool role
           { content: 'Hey', role: 'assistant' }, // Regular user message
         ] as ChatMessage[];
 
@@ -861,11 +824,6 @@ describe('ChatService', () => {
             ],
             role: 'user',
           },
-          {
-            content: 'Hi',
-            name: 'plugin1____api1',
-            role: 'tool',
-          },
           {
             content: 'Hey',
             role: 'assistant',
@@ -894,7 +852,6 @@ describe('ChatService', () => {
             },
           ],
         }, // Message with files
-        { content: 'Hi', role: 'tool', plugin: { identifier: 'plugin1', apiName: 'api1' } }, // Message with tool role
         { content: 'Hey', role: 'assistant' }, // Regular user message
       ] as ChatMessage[];
 
@@ -929,11 +886,6 @@ describe('ChatService', () => {
               ],
               role: 'user',
             },
-            {
-              content: 'Hi',
-              name: 'plugin1____api1',
-              role: 'tool',
-            },
             {
               content: 'Hey',
               role: 'assistant',

package/src/services/chat.ts

@@ -8,8 +8,14 @@ import { INBOX_SESSION_ID } from '@/const/session';
 import { DEFAULT_AGENT_CONFIG } from '@/const/settings';
 import { TracePayload, TraceTagMap } from '@/const/trace';
 import { isServerMode } from '@/const/version';
-import {AgentRuntime, AgentRuntimeError, ChatCompletionErrorPayload, ModelProvider} from '@/libs/agent-runtime';
+import {
+  AgentRuntime,
+  AgentRuntimeError,
+  ChatCompletionErrorPayload,
+  ModelProvider,
+} from '@/libs/agent-runtime';
 import { filesPrompts } from '@/prompts/files';
+import { BuiltinSystemRolePrompts } from '@/prompts/systemRole';
 import { useSessionStore } from '@/store/session';
 import { sessionMetaSelectors } from '@/store/session/selectors';
 import { useToolStore } from '@/store/tool';
@@ -34,6 +40,7 @@ import { createHeaderWithAuth, getProviderAuthPayload } from './_auth';
 import { API_ENDPOINTS } from './_url';
 
 interface FetchOptions extends FetchSSEOptions {
+  historySummary?: string;
   isWelcomeQuestion?: boolean;
   signal?: AbortSignal | undefined;
   trace?: TracePayload;
@@ -60,6 +67,7 @@ interface FetchAITaskResultParams extends FetchSSEOptions {
 
 interface CreateAssistantMessageStream extends FetchSSEOptions {
   abortController?: AbortController;
+  historySummary?: string;
   isWelcomeQuestion?: boolean;
   params: GetChatCompletionPayload;
   trace?: TracePayload;
@@ -234,8 +242,10 @@ class ChatService {
     onFinish,
     trace,
     isWelcomeQuestion,
+    historySummary,
   }: CreateAssistantMessageStream) => {
     await this.createAssistantMessage(params, {
+      historySummary,
       isWelcomeQuestion,
       onAbort,
       onErrorHandle,
@@ -463,7 +473,7 @@ class ChatService {
         }
 
         default: {
-          return { content: m.content, role: m.role };
+          return { content: m.content, role: m.role as any };
         }
       }
     });
@@ -483,9 +493,11 @@ class ChatService {
       const toolsSystemRoles =
         hasFC && toolSelectors.enabledSystemRoles(tools)(useToolStore.getState());
 
-      const injectSystemRoles = [inboxGuideSystemRole, toolsSystemRoles]
-        .filter(Boolean)
-        .join('\n\n');
+      const injectSystemRoles = BuiltinSystemRolePrompts({
+        historySummary: options?.historySummary,
+        plugins: toolsSystemRoles as string,
+        welcome: inboxGuideSystemRole as string,
+      });
 
       if (!injectSystemRoles) return;
 
@@ -549,18 +561,40 @@ class ChatService {
    * see https://github.com/lobehub/lobe-chat/pull/3155
    */
   private reorderToolMessages = (messages: OpenAIChatMessage[]): OpenAIChatMessage[] => {
-    const reorderedMessages: OpenAIChatMessage[] = [];
-    const toolMessages: Record<string, OpenAIChatMessage> = {};
+    // 1. 先收集所有 assistant 消息中的有效 tool_call_id
+    const validToolCallIds = new Set<string>();
+    messages.forEach((message) => {
+      if (message.role === 'assistant' && message.tool_calls) {
+        message.tool_calls.forEach((toolCall) => {
+          validToolCallIds.add(toolCall.id);
+        });
+      }
+    });
 
-    // 1. collect all tool messages
+    // 2. 收集所有有效的 tool 消息
+    const toolMessages: Record<string, OpenAIChatMessage> = {};
     messages.forEach((message) => {
-      if (message.role === 'tool' && message.tool_call_id) {
+      if (
+        message.role === 'tool' &&
+        message.tool_call_id &&
+        validToolCallIds.has(message.tool_call_id)
+      ) {
         toolMessages[message.tool_call_id] = message;
       }
     });
 
-    // 2. reorder messages
+    // 3. 重新排序消息
+    const reorderedMessages: OpenAIChatMessage[] = [];
     messages.forEach((message) => {
+      // 跳过无效的 tool 消息
+      if (
+        message.role === 'tool' &&
+        (!message.tool_call_id || !validToolCallIds.has(message.tool_call_id))
+      ) {
+        return;
+      }
+
+      // 检查是否已经添加过该 tool 消息
       const hasPushed = reorderedMessages.some(
         (m) => !!message.tool_call_id && m.tool_call_id === message.tool_call_id,
       );
@@ -569,12 +603,12 @@ class ChatService {
 
       reorderedMessages.push(message);
 
+      // 如果是 assistant 消息且有 tool_calls，添加对应的 tool 消息
       if (message.role === 'assistant' && message.tool_calls) {
         message.tool_calls.forEach((toolCall) => {
           const correspondingToolMessage = toolMessages[toolCall.id];
           if (correspondingToolMessage) {
             reorderedMessages.push(correspondingToolMessage);
-            // 从 toolMessages 中删除已处理的消息，避免重复
             delete toolMessages[toolCall.id];
           }
         });

package/src/store/agent/slices/chat/__snapshots__/selectors.test.ts.snap

@@ -6,7 +6,9 @@ exports[`agentSelectors > defaultAgentConfig > should merge DEFAULT_AGENT_CONFIG
     "autoCreateTopicThreshold": 2,
     "displayMode": "chat",
     "enableAutoCreateTopic": true,
-    "historyCount": 1,
+    "enableCompressHistory": true,
+    "enableHistoryCount": true,
+    "historyCount": 8,
   },
   "model": "gpt-3.5-turbo",
   "params": {

package/src/store/chat/helpers.ts

@@ -12,15 +12,19 @@ export const getMessageById = (messages: ChatMessage[], id: string) =>
 const getSlicedMessagesWithConfig = (
   messages: ChatMessage[],
   config: LobeAgentChatConfig,
+  includeNewUserMessage?: boolean,
 ): ChatMessage[] => {
   // if historyCount is not enabled or set to 0, return all messages
   if (!config.enableHistoryCount || !config.historyCount) return messages;
 
+  // if user send message, history will include this message so the total length should +1
+  const messagesCount = !!includeNewUserMessage ? config.historyCount + 1 : config.historyCount;
+
   // if historyCount is negative, return empty array
-  if (config.historyCount <= 0) return [];
+  if (messagesCount <= 0) return [];
 
   // if historyCount is positive, return last N messages
-  return messages.slice(-config.historyCount);
+  return messages.slice(-messagesCount);
 };
 
 export const chatHelpers = {

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

@@ -5,13 +5,13 @@ import { template } from 'lodash-es';
 import { StateCreator } from 'zustand/vanilla';
 
 import { LOADING_FLAT, MESSAGE_CANCEL_FLAT } from '@/const/message';
+import { DEFAULT_AGENT_CHAT_CONFIG } from '@/const/settings';
 import { TraceEventType, TraceNameMap } from '@/const/trace';
 import { isServerMode } from '@/const/version';
 import { knowledgeBaseQAPrompts } from '@/prompts/knowledgeBaseQA';
 import { chatService } from '@/services/chat';
 import { messageService } from '@/services/message';
 import { useAgentStore } from '@/store/agent';
-import { agentSelectors } from '@/store/agent/selectors';
 import { chatHelpers } from '@/store/chat/helpers';
 import { ChatStore } from '@/store/chat/store';
 import { messageMapKey } from '@/store/chat/utils/messageMapKey';
@@ -21,6 +21,7 @@ import { MessageSemanticSearchChunk } from '@/types/rag';
 import { setNamespace } from '@/utils/storeDebug';
 
 import { chatSelectors, topicSelectors } from '../../../selectors';
+import { getAgentChatConfig, getAgentConfig, getAgentKnowledge } from './helpers';
 
 const n = setNamespace('ai');
 
@@ -93,10 +94,6 @@ export interface AIGenerateAction {
   internal_toggleToolCallingStreaming: (id: string, streaming: boolean[] | undefined) => void;
 }
 
-const getAgentConfig = () => agentSelectors.currentAgentConfig(useAgentStore.getState());
-const getAgentChatConfig = () => agentSelectors.currentAgentChatConfig(useAgentStore.getState());
-const getAgentKnowledge = () => agentSelectors.currentEnabledKnowledge(useAgentStore.getState());
-
 export const generateAIChat: StateCreator<
   ChatStore,
   [['zustand/devtools', never]],
@@ -111,7 +108,7 @@ export const generateAIChat: StateCreator<
     // trace the delete and regenerate message
     get().internal_traceMessage(id, { eventType: TraceEventType.DeleteAndRegenerateMessage });
   },
-  regenerateMessage: async (id: string) => {
+  regenerateMessage: async (id) => {
     const traceId = chatSelectors.getTraceIdByMessageId(id)(get());
     await get().internal_resendMessage(id, traceId);
 
@@ -265,7 +262,7 @@ export const generateAIChat: StateCreator<
     // create a new array to avoid the original messages array change
     const messages = [...originalMessages];
 
-    const { model, provider } = getAgentConfig();
+    const { model, provider, chatConfig } = getAgentConfig();
 
     let fileChunks: MessageSemanticSearchChunk[] | undefined;
     let ragQueryId;
@@ -325,6 +322,20 @@ export const generateAIChat: StateCreator<
       await refreshMessages();
       await triggerToolCalls(assistantId);
     }
+
+    // 5. summary history if context messages is larger than historyCount
+    const historyCount =
+      chatConfig.historyCount || (DEFAULT_AGENT_CHAT_CONFIG.historyCount as number);
+
+    if (
+      chatConfig.enableHistoryCount &&
+      chatConfig.enableCompressHistory &&
+      originalMessages.length > historyCount
+    ) {
+      const historyMessages = originalMessages.slice(0, -historyCount);
+
+      await get().internal_summaryHistory(historyMessages);
+    }
   },
   internal_fetchAIChatMessage: async (messages, assistantId, params) => {
     const {
@@ -351,7 +362,7 @@ export const generateAIChat: StateCreator<
     // ================================== //
 
     // 1. slice messages with config
-    let preprocessMsgs = chatHelpers.getSlicedMessagesWithConfig(messages, chatConfig);
+    let preprocessMsgs = chatHelpers.getSlicedMessagesWithConfig(messages, chatConfig, true);
 
     // 2. replace inputMessage template
     preprocessMsgs = !chatConfig.inputTemplate
@@ -394,6 +405,7 @@ export const generateAIChat: StateCreator<
     let msgTraceId: string | undefined;
     let output = '';
 
+    const historySummary = topicSelectors.currentActiveTopicSummary(get());
     await chatService.createAssistantMessageStream({
       abortController,
       params: {
@@ -403,6 +415,7 @@ export const generateAIChat: StateCreator<
         ...agentConfig.params,
         plugins: agentConfig.plugins,
       },
+      historySummary: historySummary?.content,
       trace: {
         traceId: params?.traceId,
         sessionId: get().activeId,

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

@@ -0,0 +1,9 @@
+import { useAgentStore } from '@/store/agent';
+import { agentSelectors } from '@/store/agent/selectors';
+
+export const getAgentConfig = () => agentSelectors.currentAgentConfig(useAgentStore.getState());
+export const getAgentChatConfig = () =>
+  agentSelectors.currentAgentChatConfig(useAgentStore.getState());
+
+export const getAgentKnowledge = () =>
+  agentSelectors.currentEnabledKnowledge(useAgentStore.getState());

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

@@ -3,9 +3,10 @@ import { StateCreator } from 'zustand/vanilla';
 import { ChatStore } from '@/store/chat/store';
 
 import { AIGenerateAction, generateAIChat } from './generateAIChat';
+import { ChatMemoryAction, chatMemory } from './memory';
 import { ChatRAGAction, chatRag } from './rag';
 
-export interface ChatAIChatAction extends ChatRAGAction, AIGenerateAction {
+export interface ChatAIChatAction extends ChatRAGAction, ChatMemoryAction, AIGenerateAction {
   /**/
 }
 
@@ -17,4 +18,5 @@ export const chatAiChat: StateCreator<
 > = (...params) => ({
   ...chatRag(...params),
   ...generateAIChat(...params),
+  ...chatMemory(...params),
 });

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

@@ -0,0 +1,52 @@
+import { StateCreator } from 'zustand/vanilla';
+
+import { chainSummaryHistory } from '@/chains/summaryHistory';
+import { chatService } from '@/services/chat';
+import { topicService } from '@/services/topic';
+import { ChatStore } from '@/store/chat';
+import { useUserStore } from '@/store/user';
+import { systemAgentSelectors } from '@/store/user/selectors';
+import { ChatMessage } from '@/types/message';
+
+import { getAgentConfig } from './helpers';
+
+export interface ChatMemoryAction {
+  internal_summaryHistory: (messages: ChatMessage[]) => Promise<void>;
+}
+
+export const chatMemory: StateCreator<
+  ChatStore,
+  [['zustand/devtools', never]],
+  [],
+  ChatMemoryAction
+> = (set, get) => ({
+  internal_summaryHistory: async (messages) => {
+    const topicId = get().activeTopicId;
+    if (messages.length <= 1 || !topicId) return;
+
+    const { model, provider } = getAgentConfig();
+
+    const historyCompressConfig = systemAgentSelectors.historyCompress(useUserStore.getState());
+
+    let historySummary = '';
+    await chatService.fetchPresetTaskResult({
+      onFinish: async (text) => {
+        historySummary = text;
+      },
+
+      params: {
+        ...chainSummaryHistory(messages),
+        model: historyCompressConfig.model,
+        provider: historyCompressConfig.provider,
+        stream: false,
+      },
+    });
+
+    await topicService.updateTopic(topicId, {
+      metadata: { model, provider },
+      summary: historySummary,
+    });
+    await get().refreshTopic();
+    await get().refreshMessages();
+  },
+});

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

@@ -75,9 +75,7 @@ const showInboxWelcome = (s: ChatStoreState): boolean => {
   if (!isInbox) return false;
 
   const data = currentChats(s);
-  const isBrandNewChat = data.length === 0;
-
-  return isBrandNewChat;
+  return data.length === 0;
 };
 
 // Custom message for new assistant initialization

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

@@ -1,36 +1,47 @@
 import { t } from 'i18next';
 
-import { ChatTopic, GroupedTopic } from '@/types/topic';
+import { ChatTopic, ChatTopicSummary, GroupedTopic } from '@/types/topic';
 import { groupTopicsByTime } from '@/utils/client/topic';
 
-import { ChatStore } from '../../store';
+import { ChatStoreState } from '../../initialState';
 
-const currentTopics = (s: ChatStore): ChatTopic[] | undefined => s.topicMaps[s.activeId];
+const currentTopics = (s: ChatStoreState): ChatTopic[] | undefined => s.topicMaps[s.activeId];
 
-const currentActiveTopic = (s: ChatStore): ChatTopic | undefined => {
+const currentActiveTopic = (s: ChatStoreState): ChatTopic | undefined => {
   return currentTopics(s)?.find((topic) => topic.id === s.activeTopicId);
 };
-const searchTopics = (s: ChatStore): ChatTopic[] => s.searchTopics;
+const searchTopics = (s: ChatStoreState): ChatTopic[] => s.searchTopics;
 
-const displayTopics = (s: ChatStore): ChatTopic[] | undefined =>
+const displayTopics = (s: ChatStoreState): ChatTopic[] | undefined =>
   s.isSearchingTopic ? searchTopics(s) : currentTopics(s);
 
-const currentFavTopics = (s: ChatStore): ChatTopic[] =>
+const currentFavTopics = (s: ChatStoreState): ChatTopic[] =>
   currentTopics(s)?.filter((s) => s.favorite) || [];
 
-const currentUnFavTopics = (s: ChatStore): ChatTopic[] =>
+const currentUnFavTopics = (s: ChatStoreState): ChatTopic[] =>
   currentTopics(s)?.filter((s) => !s.favorite) || [];
 
-const currentTopicLength = (s: ChatStore): number => currentTopics(s)?.length || 0;
+const currentTopicLength = (s: ChatStoreState): number => currentTopics(s)?.length || 0;
 
 const getTopicById =
   (id: string) =>
-  (s: ChatStore): ChatTopic | undefined =>
+  (s: ChatStoreState): ChatTopic | undefined =>
     currentTopics(s)?.find((topic) => topic.id === id);
 
-const isCreatingTopic = (s: ChatStore) => s.creatingTopic;
+const currentActiveTopicSummary = (s: ChatStoreState): ChatTopicSummary | undefined => {
+  const activeTopic = currentActiveTopic(s);
+  if (!activeTopic) return undefined;
 
-const groupedTopicsSelector = (s: ChatStore): GroupedTopic[] => {
+  return {
+    content: activeTopic.summary || '',
+    model: activeTopic.metadata?.model || '',
+    provider: activeTopic.metadata?.provider || '',
+  };
+};
+
+const isCreatingTopic = (s: ChatStoreState) => s.creatingTopic;
+
+const groupedTopicsSelector = (s: ChatStoreState): GroupedTopic[] => {
   const topics = currentTopics(s);
 
   if (!topics) return [];
@@ -51,6 +62,7 @@ const groupedTopicsSelector = (s: ChatStore): GroupedTopic[] => {
 
 export const topicSelectors = {
   currentActiveTopic,
+  currentActiveTopicSummary,
   currentTopicLength,
   currentTopics,
   currentUnFavTopics,

package/src/store/chat/slices/{enchance → translate}/action.test.ts

@@ -52,19 +52,6 @@ afterEach(() => {
 });
 
 describe('ChatEnhanceAction', () => {
-  describe('clearTTS', () => {
-    it('should clear TTS for a message and refresh messages', async () => {
-      const { result } = renderHook(() => useChatStore());
-      const messageId = 'message-id';
-
-      await act(async () => {
-        await result.current.clearTTS(messageId);
-      });
-
-      expect(messageService.updateMessageTTS).toHaveBeenCalledWith(messageId, false);
-    });
-  });
-
   describe('translateMessage', () => {
     it('should translate a message to the target language and refresh messages', async () => {
       const { result } = renderHook(() => useChatStore());

package/src/store/chat/slices/{enchance → translate}/action.ts

@@ -11,38 +11,28 @@ import { chatSelectors } from '@/store/chat/selectors';
 import { ChatStore } from '@/store/chat/store';
 import { useUserStore } from '@/store/user';
 import { systemAgentSelectors } from '@/store/user/selectors';
-import { ChatTTS, ChatTranslate } from '@/types/message';
+import { ChatTranslate } from '@/types/message';
 import { merge } from '@/utils/merge';
 import { setNamespace } from '@/utils/storeDebug';
 
 const n = setNamespace('enhance');
 
 /**
- * enhance chat action like translate,tts
+ * chat translate
  */
-export interface ChatEnhanceAction {
-  clearTTS: (id: string) => Promise<void>;
+export interface ChatTranslateAction {
   clearTranslate: (id: string) => Promise<void>;
   getCurrentTracePayload: (data: Partial<TracePayload>) => TracePayload;
   translateMessage: (id: string, targetLang: string) => Promise<void>;
-  ttsMessage: (
-    id: string,
-    state?: { contentMd5?: string; file?: string; voice?: string },
-  ) => Promise<void>;
-  updateMessageTTS: (id: string, data: Partial<ChatTTS> | false) => Promise<void>;
   updateMessageTranslate: (id: string, data: Partial<ChatTranslate> | false) => Promise<void>;
 }
 
-export const chatEnhance: StateCreator<
+export const chatTranslate: StateCreator<
   ChatStore,
   [['zustand/devtools', never]],
   [],
-  ChatEnhanceAction
+  ChatTranslateAction
 > = (set, get) => ({
-  clearTTS: async (id) => {
-    await get().updateMessageTTS(id, false);
-  },
-
   clearTranslate: async (id) => {
     await get().updateMessageTranslate(id, false);
   },
@@ -107,15 +97,6 @@ export const chatEnhance: StateCreator<
     });
   },
 
-  ttsMessage: async (id, state = {}) => {
-    await get().updateMessageTTS(id, state);
-  },
-
-  updateMessageTTS: async (id, data) => {
-    await messageService.updateMessageTTS(id, data);
-    await get().refreshMessages();
-  },
-
   updateMessageTranslate: async (id, data) => {
     await messageService.updateMessageTranslate(id, data);