@lobehub/chat 0.115.12 → 0.115.13

package/CHANGELOG.md

@@ -2,6 +2,31 @@
 
 # Changelog
 
+### [Version 0.115.13](https://github.com/lobehub/lobe-chat/compare/v0.115.12...v0.115.13)
+
+<sup>Released on **2023-12-26**</sup>
+
+#### 🐛 Bug Fixes
+
+- **misc**: Fix remove tts and translate not working.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **misc**: Fix remove tts and translate not working, closes [#818](https://github.com/lobehub/lobe-chat/issues/818) ([4a275e9](https://github.com/lobehub/lobe-chat/commit/4a275e9))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 0.115.12](https://github.com/lobehub/lobe-chat/compare/v0.115.11...v0.115.12)
 
 <sup>Released on **2023-12-25**</sup>

package/README.md

@@ -36,6 +36,7 @@ Supports one-click free deployment of your private ChatGPT/LLM web application.
 [![][share-whatsapp-shield]][share-whatsapp-link]
 [![][share-reddit-shield]][share-reddit-link]
 [![][share-weibo-shield]][share-weibo-link]
+[![][share-mastodon-shield]][share-mastodon-link]
 
 <sup>Pioneering the new age of thinking and creating. Built for you, the Super Individual.</sup>
 
@@ -625,6 +626,8 @@ This project is [MIT](./LICENSE) licensed.
 [pr-welcome-link]: https://github.com/lobehub/lobe-chat/pulls
 [pr-welcome-shield]: https://img.shields.io/badge/🤯_pr_welcome-%E2%86%92-ffcb47?labelColor=black&style=for-the-badge
 [profile-link]: https://github.com/lobehub
+[share-mastodon-link]: https://mastodon.social/share?text=Check%20this%20GitHub%20repository%20out%20%F0%9F%A4%AF%20LobeChat%20-%20An%20open-source,%20extensible%20(Function%20Calling),%20high-performance%20chatbot%20framework.%20It%20supports%20one-click%20free%20deployment%20of%20your%20private%20ChatGPT/LLM%20web%20application.%20https://github.com/lobehub/lobe-chat%20#chatbot%20#chatGPT%20#openAI
+[share-mastodon-shield]: https://img.shields.io/badge/-share%20on%20mastodon-black?labelColor=black&logo=mastodon&logoColor=white&style=flat-square
 [share-reddit-link]: https://www.reddit.com/submit?title=Check%20this%20GitHub%20repository%20out%20%F0%9F%A4%AF%20LobeChat%20-%20An%20open-source%2C%20extensible%20%28Function%20Calling%29%2C%20high-performance%20chatbot%20framework.%20It%20supports%20one-click%20free%20deployment%20of%20your%20private%20ChatGPT%2FLLM%20web%20application.%20%23chatbot%20%23chatGPT%20%23openAI&url=https%3A%2F%2Fgithub.com%2Flobehub%2Flobe-chat
 [share-reddit-shield]: https://img.shields.io/badge/-share%20on%20reddit-black?labelColor=black&logo=reddit&logoColor=white&style=flat-square
 [share-telegram-link]: https://t.me/share/url"?text=Check%20this%20GitHub%20repository%20out%20%F0%9F%A4%AF%20LobeChat%20-%20An%20open-source%2C%20extensible%20%28Function%20Calling%29%2C%20high-performance%20chatbot%20framework.%20It%20supports%20one-click%20free%20deployment%20of%20your%20private%20ChatGPT%2FLLM%20web%20application.%20%23chatbot%20%23chatGPT%20%23openAI&url=https%3A%2F%2Fgithub.com%2Flobehub%2Flobe-chat

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "0.115.12",
+  "version": "0.115.13",
   "description": "Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.",
   "keywords": [
     "framework",

package/src/database/models/message.ts

@@ -179,11 +179,7 @@ class _MessageModel extends BaseModel {
     tts,
     ...item
   }: DBModel<DB_Message>): ChatMessage => {
-    return {
-      ...item,
-      extra: { fromModel: fromModel, translate: translate, tts: tts },
-      meta: {},
-    };
+    return { ...item, extra: { fromModel, translate, tts }, meta: {} };
   };
 }
 

package/src/database/schemas/message.ts

@@ -24,7 +24,7 @@ export const DB_MessageSchema = z.object({
   plugin: PluginSchema.optional(),
   pluginState: z.any().optional(),
   fromModel: z.string().optional(),
-  translate: TranslateSchema.optional().or(z.null()),
+  translate: TranslateSchema.optional().or(z.literal(false)),
   tts: z.any().optional(),
 
   // foreign keys

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

@@ -153,21 +153,6 @@ describe('MessageService', () => {
     });
   });
 
-  describe('updateMessageContent', () => {
-    it('should update the content of a message', async () => {
-      // Setup
-      const newContent = 'Updated message content';
-      (MessageModel.update as Mock).mockResolvedValue({ ...mockMessage, content: newContent });
-
-      // Execute
-      const result = await messageService.updateMessageContent(mockMessageId, newContent);
-
-      // Assert
-      expect(MessageModel.update).toHaveBeenCalledWith(mockMessageId, { content: newContent });
-      expect(result).toEqual({ ...mockMessage, content: newContent });
-    });
-  });
-
   describe('removeMessages', () => {
     it('should batch remove messages by assistantId and topicId', async () => {
       // Setup
@@ -243,51 +228,6 @@ describe('MessageService', () => {
     });
   });
 
-  describe('updateMessageTranslate', () => {
-    it('should update the translate field of a message', async () => {
-      // Setup
-      const newTranslate = { content: 'Translated text', to: 'es' } as ChatTranslate;
-      (MessageModel.update as Mock).mockResolvedValue({ ...mockMessage, translate: newTranslate });
-
-      // Execute
-      const result = await messageService.updateMessageTranslate(mockMessageId, newTranslate);
-
-      // Assert
-      expect(MessageModel.update).toHaveBeenCalledWith(mockMessageId, { translate: newTranslate });
-      expect(result).toEqual({ ...mockMessage, translate: newTranslate });
-    });
-  });
-
-  describe('updateMessageTTS', () => {
-    it('should update the tts field of a message', async () => {
-      // Setup
-      const newTTS = { init: false } as ChatTTS;
-      (MessageModel.update as Mock).mockResolvedValue({ ...mockMessage, tts: newTTS });
-
-      // Execute
-      const result = await messageService.updateMessageTTS(mockMessageId, newTTS);
-
-      // Assert
-      expect(MessageModel.update).toHaveBeenCalledWith(mockMessageId, { tts: newTTS });
-      expect(result).toEqual({ ...mockMessage, tts: newTTS });
-    });
-  });
-
-  describe('updateMessageRole', () => {
-    it('should update the role of a message', async () => {
-      // Setup
-      const newRole = 'user';
-      (MessageModel.update as Mock).mockResolvedValue({ ...mockMessage, role: newRole });
-
-      // Execute
-      const result = await messageService.updateMessageRole(mockMessageId, newRole);
-
-      // Assert
-      expect(MessageModel.update).toHaveBeenCalledWith(mockMessageId, { role: newRole });
-      expect(result).toEqual({ ...mockMessage, role: newRole });
-    });
-  });
-
   describe('updateMessagePlugin', () => {
     it('should update the plugin payload of a message', async () => {
       // Setup

package/src/services/message.ts

@@ -1,13 +1,6 @@
 import { CreateMessageParams, MessageModel } from '@/database/models/message';
 import { DB_Message } from '@/database/schemas/message';
-import { LLMRoleType } from '@/types/llm';
-import {
-  ChatMessage,
-  ChatMessageError,
-  ChatPluginPayload,
-  ChatTTS,
-  ChatTranslate,
-} from '@/types/message';
+import { ChatMessage, ChatMessageError, ChatPluginPayload } from '@/types/message';
 
 export class MessageService {
   async create(data: CreateMessageParams) {
@@ -37,22 +30,10 @@ export class MessageService {
     return MessageModel.queryBySessionId(sessionId);
   }
 
-  async updateMessageContent(id: string, content: string) {
-    return MessageModel.update(id, { content });
-  }
-
   async updateMessageError(id: string, error: ChatMessageError) {
     return MessageModel.update(id, { error });
   }
 
-  async updateMessageTranslate(id: string, data: Partial<ChatTranslate> | null) {
-    return MessageModel.update(id, { translate: data as ChatTranslate });
-  }
-
-  async updateMessageTTS(id: string, data: Partial<ChatTTS> | null) {
-    return MessageModel.update(id, { tts: data as ChatTTS });
-  }
-
   async removeMessages(assistantId: string, topicId?: string) {
     return MessageModel.batchDelete(assistantId, topicId);
   }
@@ -69,10 +50,6 @@ export class MessageService {
     return MessageModel.update(id, message);
   }
 
-  async updateMessageRole(id: string, role: LLMRoleType) {
-    return MessageModel.update(id, { role });
-  }
-
   async updateMessagePlugin(id: string, plugin: ChatPluginPayload) {
     return MessageModel.update(id, { plugin });
   }

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

@@ -12,7 +12,7 @@ import { useChatStore } from '../../store';
 vi.mock('@/services/message', () => ({
   messageService: {
     updateMessageTTS: vi.fn(),
-    updateMessageTranslate: vi.fn(),
+    updateMessage: vi.fn(),
   },
 }));
 
@@ -59,7 +59,7 @@ describe('ChatEnhanceAction', () => {
         await result.current.clearTTS(messageId);
       });
 
-      expect(messageService.updateMessageTTS).toHaveBeenCalledWith(messageId, null);
+      expect(messageService.updateMessage).toHaveBeenCalledWith(messageId, { tts: false });
     });
   });
 
@@ -102,7 +102,7 @@ describe('ChatEnhanceAction', () => {
         await result.current.translateMessage(messageId, targetLang);
       });
 
-      expect(messageService.updateMessageTranslate).toHaveBeenCalled();
+      expect(messageService.updateMessage).toHaveBeenCalled();
     });
   });
 
@@ -115,25 +115,7 @@ describe('ChatEnhanceAction', () => {
         await result.current.clearTranslate(messageId);
       });
 
-      expect(messageService.updateMessageTranslate).toHaveBeenCalledWith(messageId, null);
-    });
-  });
-
-  describe('ttsMessage', () => {
-    it('should update TTS state for a message and refresh messages', async () => {
-      const { result } = renderHook(() => useChatStore());
-      const messageId = 'message-id';
-      const ttsState = {
-        contentMd5: 'some-md5',
-        file: 'path-to-tts-file',
-        voice: 'voice-type',
-      };
-
-      await act(async () => {
-        await result.current.ttsMessage(messageId, ttsState);
-      });
-
-      expect(messageService.updateMessageTTS).toHaveBeenCalledWith(messageId, ttsState);
+      expect(messageService.updateMessage).toHaveBeenCalledWith(messageId, { translate: false });
     });
   });
 });

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

@@ -11,6 +11,7 @@ import { imageGenerationService } from '@/services/imageGeneration';
 import { messageService } from '@/services/message';
 import { chatSelectors } from '@/store/chat/selectors';
 import { ChatStore } from '@/store/chat/store';
+import { ChatTTS, ChatTranslate } from '@/types/message';
 import { DallEImageItem } from '@/types/tool/dalle';
 import { setNamespace } from '@/utils/storeDebug';
 
@@ -31,6 +32,8 @@ export interface ChatEnhanceAction {
     state?: { contentMd5?: string; file?: string; voice?: string },
   ) => Promise<void>;
   updateImageItem: (id: string, updater: (data: DallEImageItem[]) => void) => Promise<void>;
+  updateMessageTTS: (id: string, data: Partial<ChatTTS> | false) => Promise<void>;
+  updateMessageTranslate: (id: string, data: Partial<ChatTranslate> | false) => Promise<void>;
 }
 
 export const chatEnhance: StateCreator<
@@ -40,13 +43,11 @@ export const chatEnhance: StateCreator<
   ChatEnhanceAction
 > = (set, get) => ({
   clearTTS: async (id) => {
-    await messageService.updateMessageTTS(id, null);
-    await get().refreshMessages();
+    await get().updateMessageTTS(id, false);
   },
 
   clearTranslate: async (id) => {
-    await messageService.updateMessageTranslate(id, null);
-    await get().refreshMessages();
+    await get().updateMessageTranslate(id, false);
   },
 
   generateImageFromPrompts: async (items, messageId) => {
@@ -94,16 +95,14 @@ export const chatEnhance: StateCreator<
       n('toggleDallEImageLoading'),
     );
   },
-
   translateMessage: async (id, targetLang) => {
-    const { toggleChatLoading, dispatchMessage, refreshMessages } = get();
+    const { toggleChatLoading, updateMessageTranslate, dispatchMessage } = get();
 
     const message = chatSelectors.getMessageById(id)(get());
     if (!message) return;
 
     // create translate extra
-    await messageService.updateMessageTranslate(id, { content: '', from: '', to: targetLang });
-    await refreshMessages();
+    await updateMessageTranslate(id, { content: '', from: '', to: targetLang });
 
     toggleChatLoading(true, id, n('translateMessage(start)', { id }) as string);
 
@@ -118,8 +117,7 @@ export const chatEnhance: StateCreator<
       .then(async (data) => {
         if (data && supportLocales.includes(data)) from = data;
 
-        await messageService.updateMessageTranslate(id, { content, from, to: targetLang });
-        await refreshMessages();
+        await updateMessageTranslate(id, { content, from, to: targetLang });
       });
 
     // translate to target language
@@ -138,15 +136,13 @@ export const chatEnhance: StateCreator<
       params: chainTranslate(message.content, targetLang),
     });
 
-    await messageService.updateMessageTranslate(id, { content, from, to: targetLang });
-    await refreshMessages();
+    await updateMessageTranslate(id, { content, from, to: targetLang });
 
     toggleChatLoading(false);
   },
 
   ttsMessage: async (id, state = {}) => {
-    await messageService.updateMessageTTS(id, state);
-    await get().refreshMessages();
+    await get().updateMessageTTS(id, state);
   },
 
   updateImageItem: async (id, updater) => {
@@ -156,7 +152,15 @@ export const chatEnhance: StateCreator<
     const data: DallEImageItem[] = JSON.parse(message.content);
 
     const nextContent = produce(data, updater);
-    await messageService.updateMessageContent(id, JSON.stringify(nextContent));
+    await get().updateMessageContent(id, JSON.stringify(nextContent));
+  },
+
+  updateMessageTTS: async (id, data) => {
+    await messageService.updateMessage(id, { tts: data as ChatTTS });
+    await get().refreshMessages();
+  },
+  updateMessageTranslate: async (id, data) => {
+    await messageService.updateMessage(id, { translate: data as ChatTranslate });
     await get().refreshMessages();
   },
 });

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

@@ -20,7 +20,7 @@ vi.mock('@/services/message', () => ({
     createAssistantMessage: vi.fn(() => Promise.resolve('content-content-content')),
     removeMessages: vi.fn(() => Promise.resolve()),
     create: vi.fn(() => Promise.resolve('new-message-id')),
-    updateMessageContent: vi.fn(),
+    updateMessage: vi.fn(),
     clearAllMessage: vi.fn(() => Promise.resolve()),
   },
 }));
@@ -303,7 +303,7 @@ describe('chatMessage actions', () => {
         await result.current.updateMessageContent(messageId, newContent);
       });
 
-      expect(messageService.updateMessageContent).toHaveBeenCalledWith(messageId, newContent);
+      expect(messageService.updateMessage).toHaveBeenCalledWith(messageId, { content: newContent });
     });
 
     it('should dispatch message update action', async () => {

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

@@ -204,7 +204,7 @@ export const chatMessage: StateCreator<
     // refs: https://medium.com/@kyledeguzmanx/what-are-optimistic-updates-483662c3e171
     dispatchMessage({ id, key: 'content', type: 'updateMessage', value: content });
 
-    await messageService.updateMessageContent(id, content);
+    await messageService.updateMessage(id, { content });
     await refreshMessages();
   },
   useFetchMessages: (sessionId, activeTopicId) =>
@@ -265,7 +265,7 @@ export const chatMessage: StateCreator<
       if (functionCallAtEnd) {
         // create a new separate message and remove the function call from the prev message
 
-        await messageService.updateMessageContent(mid, content.replace(functionCallContent, ''));
+        await get().updateMessageContent(mid, content.replace(functionCallContent, ''));
 
         const functionMessage: CreateMessageParams = {
           role: 'function',
@@ -294,7 +294,7 @@ export const chatMessage: StateCreator<
     set({ messages }, false, n(`dispatchMessage/${payload.type}`, payload));
   },
   fetchAIChatMessage: async (messages, assistantId) => {
-    const { toggleChatLoading, refreshMessages } = get();
+    const { toggleChatLoading, refreshMessages, updateMessageContent } = get();
 
     const abortController = toggleChatLoading(
       true,
@@ -369,14 +369,12 @@ export const chatMessage: StateCreator<
       },
       onFinish: async (content) => {
         // update the content after fetch result
-        await messageService.updateMessageContent(assistantId, content);
-        await refreshMessages();
+        await updateMessageContent(assistantId, content);
       },
       onMessageHandle: async (text) => {
         output += text;
 
-        await messageService.updateMessageContent(assistantId, output);
-        await refreshMessages();
+        await updateMessageContent(assistantId, output);
 
         // is this message is just a function call
         if (isFunctionMessageAtStart(output)) isFunctionCall = true;

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

@@ -58,7 +58,7 @@ describe('ChatPluginAction', () => {
       });
 
       // 验证 messageService.updateMessageContent 是否被正确调用
-      expect(messageService.updateMessageContent).toHaveBeenCalledWith(messageId, newContent);
+      expect(messageService.updateMessage).toHaveBeenCalledWith(messageId, { content: newContent });
 
       // 验证 refreshMessages 是否被调用
       expect(result.current.refreshMessages).toHaveBeenCalled();
@@ -95,10 +95,9 @@ describe('ChatPluginAction', () => {
         expect.any(String),
       );
       expect(chatService.runPluginApi).toHaveBeenCalledWith(pluginPayload, { signal: undefined });
-      expect(messageService.updateMessageContent).toHaveBeenCalledWith(
-        messageId,
-        pluginApiResponse,
-      );
+      expect(messageService.updateMessage).toHaveBeenCalledWith(messageId, {
+        content: pluginApiResponse,
+      });
       expect(initialState.refreshMessages).toHaveBeenCalled();
       expect(initialState.coreProcessMessage).toHaveBeenCalled();
       expect(initialState.toggleChatLoading).toHaveBeenCalledWith(false);

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

@@ -30,24 +30,22 @@ export const chatPlugin: StateCreator<
   ChatPluginAction
 > = (set, get) => ({
   fillPluginMessageContent: async (id, content) => {
-    const { coreProcessMessage, refreshMessages } = get();
+    const { coreProcessMessage, updateMessageContent } = get();
 
-    await messageService.updateMessageContent(id, content);
-    await refreshMessages();
+    await updateMessageContent(id, content);
 
     const chats = chatSelectors.currentChats(get());
     await coreProcessMessage(chats, id);
   },
   invokeBuiltinTool: async (id, payload) => {
-    const { toggleChatLoading, refreshMessages } = get();
+    const { toggleChatLoading, updateMessageContent } = get();
     const params = JSON.parse(payload.arguments);
     toggleChatLoading(true, id, n('invokeBuiltinTool') as string);
     const data = await useToolStore.getState().invokeBuiltinTool(payload.apiName, params);
     toggleChatLoading(false);
 
     if (data) {
-      await messageService.updateMessageContent(id, data);
-      await refreshMessages();
+      await updateMessageContent(id, data);
     }
 
     // postToolCalling
@@ -58,7 +56,7 @@ export const chatPlugin: StateCreator<
     await action(id, JSON.parse(data));
   },
   invokeDefaultTypePlugin: async (id, payload) => {
-    const { refreshMessages, coreProcessMessage, toggleChatLoading } = get();
+    const { updateMessageContent, refreshMessages, coreProcessMessage, toggleChatLoading } = get();
     let data: string;
 
     try {
@@ -79,8 +77,7 @@ export const chatPlugin: StateCreator<
     // 如果报错则结束了
     if (!data) return;
 
-    await messageService.updateMessageContent(id, data);
-    await refreshMessages();
+    await updateMessageContent(id, data);
 
     const chats = chatSelectors.currentChats(get());
     await coreProcessMessage(chats, id);

package/src/types/message/index.ts

@@ -37,7 +37,7 @@ export interface ChatMessage extends BaseDataModel {
   extra?: {
     fromModel?: string;
     // 翻译
-    translate?: ChatTranslate | null;
+    translate?: ChatTranslate | false;
     // TTS
     tts?: ChatTTS;
   } & Record<string, any>;