@lobehub/chat 0.154.6 → 0.155.0

package/src/services/topic/client.test.ts

@@ -1,5 +1,6 @@
 import { Mock, beforeAll, beforeEach, describe, expect, it, vi } from 'vitest';
 
+import { SessionModel } from '@/database/client/models/session';
 import { CreateTopicParams, TopicModel } from '@/database/client/models/topic';
 import { ChatTopic } from '@/types/topic';
 
@@ -13,6 +14,7 @@ vi.mock('@/database/client/models/topic', () => {
       create: vi.fn(),
       query: vi.fn(),
       delete: vi.fn(),
+      count: vi.fn(),
       batchDeleteBySessionId: vi.fn(),
       batchDelete: vi.fn(),
       clearTable: vi.fn(),
@@ -214,4 +216,30 @@ describe('TopicService', () => {
       expect(result).toBe(mockTopics);
     });
   });
+
+  describe('countTopics', () => {
+    it('should return false if no topics exist', async () => {
+      // Setup
+      (TopicModel.count as Mock).mockResolvedValue(0);
+
+      // Execute
+      const result = await topicService.countTopics();
+
+      // Assert
+      expect(TopicModel.count).toHaveBeenCalled();
+      expect(result).toBe(0);
+    });
+
+    it('should return true if topics exist', async () => {
+      // Setup
+      (TopicModel.count as Mock).mockResolvedValue(1);
+
+      // Execute
+      const result = await topicService.countTopics();
+
+      // Assert
+      expect(TopicModel.count).toHaveBeenCalled();
+      expect(result).toBe(1);
+    });
+  });
 });

package/src/services/topic/client.ts

@@ -34,6 +34,10 @@ export class ClientService implements ITopicService {
     return TopicModel.queryAll();
   }
 
+  async countTopics() {
+    return TopicModel.count();
+  }
+
   async updateTopicFavorite(id: string, favorite?: boolean) {
     return this.updateTopic(id, { favorite });
   }

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

@@ -49,7 +49,7 @@ export const chatEnhance: StateCreator<
     ...data,
   }),
   translateMessage: async (id, targetLang) => {
-    const { toggleChatLoading, updateMessageTranslate, dispatchMessage } = get();
+    const { internal_toggleChatLoading, updateMessageTranslate, internal_dispatchMessage } = get();
 
     const message = chatSelectors.getMessageById(id)(get());
     if (!message) return;
@@ -57,7 +57,7 @@ export const chatEnhance: StateCreator<
     // create translate extra
     await updateMessageTranslate(id, { content: '', from: '', to: targetLang });
 
-    toggleChatLoading(true, id, n('translateMessage(start)', { id }) as string);
+    internal_toggleChatLoading(true, id, n('translateMessage(start)', { id }) as string);
 
     let content = '';
     let from = '';
@@ -77,7 +77,7 @@ export const chatEnhance: StateCreator<
     // translate to target language
     await chatService.fetchPresetTaskResult({
       onMessageHandle: (text) => {
-        dispatchMessage({
+        internal_dispatchMessage({
           id,
           key: 'translate',
           type: 'updateMessageExtra',
@@ -93,7 +93,7 @@ export const chatEnhance: StateCreator<
 
     await updateMessageTranslate(id, { content, from, to: targetLang });
 
-    toggleChatLoading(false);
+    internal_toggleChatLoading(false);
   },
 
   ttsMessage: async (id, state = {}) => {

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

@@ -53,7 +53,7 @@ vi.mock('@/store/chat/selectors', () => ({
   },
 }));
 
-const realCoreProcessMessage = useChatStore.getState().coreProcessMessage;
+const realCoreProcessMessage = useChatStore.getState().internal_coreProcessMessage;
 const realRefreshMessages = useChatStore.getState().refreshMessages;
 // Mock state
 const mockState = {
@@ -62,7 +62,7 @@ const mockState = {
   messages: [],
   refreshMessages: vi.fn(),
   refreshTopic: vi.fn(),
-  coreProcessMessage: vi.fn(),
+  internal_coreProcessMessage: vi.fn(),
   saveToTopic: vi.fn(),
 };
 
@@ -191,7 +191,7 @@ describe('chatMessage actions', () => {
 
       expect(messageService.createMessage).not.toHaveBeenCalled();
       expect(result.current.refreshMessages).not.toHaveBeenCalled();
-      expect(result.current.coreProcessMessage).not.toHaveBeenCalled();
+      expect(result.current.internal_coreProcessMessage).not.toHaveBeenCalled();
     });
 
     it('should not send message if message is empty and there are no files', async () => {
@@ -204,7 +204,7 @@ describe('chatMessage actions', () => {
 
       expect(messageService.createMessage).not.toHaveBeenCalled();
       expect(result.current.refreshMessages).not.toHaveBeenCalled();
-      expect(result.current.coreProcessMessage).not.toHaveBeenCalled();
+      expect(result.current.internal_coreProcessMessage).not.toHaveBeenCalled();
     });
 
     it('should not send message if message is empty and there are empty files', async () => {
@@ -217,10 +217,10 @@ describe('chatMessage actions', () => {
 
       expect(messageService.createMessage).not.toHaveBeenCalled();
       expect(result.current.refreshMessages).not.toHaveBeenCalled();
-      expect(result.current.coreProcessMessage).not.toHaveBeenCalled();
+      expect(result.current.internal_coreProcessMessage).not.toHaveBeenCalled();
     });
 
-    it('should create message and call coreProcessMessage if message or files are provided', async () => {
+    it('should create message and call internal_coreProcessMessage if message or files are provided', async () => {
       const { result } = renderHook(() => useChatStore());
       const message = 'Test message';
       const files = [{ id: 'file-id', url: 'file-url' }];
@@ -240,7 +240,7 @@ describe('chatMessage actions', () => {
         topicId: mockState.activeTopicId,
       });
       expect(result.current.refreshMessages).toHaveBeenCalled();
-      expect(result.current.coreProcessMessage).toHaveBeenCalled();
+      expect(result.current.internal_coreProcessMessage).toHaveBeenCalled();
     });
 
     describe('auto-create topic', () => {
@@ -365,7 +365,7 @@ describe('chatMessage actions', () => {
     });
   });
 
-  describe('internalResendMessage action', () => {
+  describe('internal_resendMessage action', () => {
     it('should resend a message by id and refresh messages', async () => {
       const { result } = renderHook(() => useChatStore());
       const messageId = 'message-id';
@@ -377,15 +377,19 @@ describe('chatMessage actions', () => {
         // ... other messages
       ]);
 
-      // Mock the coreProcessMessage function to resolve immediately
-      mockState.coreProcessMessage.mockResolvedValue(undefined);
+      // Mock the internal_coreProcessMessage function to resolve immediately
+      mockState.internal_coreProcessMessage.mockResolvedValue(undefined);
 
       await act(async () => {
-        await result.current.internalResendMessage(messageId);
+        await result.current.internal_resendMessage(messageId);
       });
 
       expect(messageService.removeMessage).not.toHaveBeenCalledWith(messageId);
-      expect(mockState.coreProcessMessage).toHaveBeenCalledWith(expect.any(Array), messageId, {});
+      expect(mockState.internal_coreProcessMessage).toHaveBeenCalledWith(
+        expect.any(Array),
+        messageId,
+        {},
+      );
     });
 
     it('should not perform any action if the message id does not exist', async () => {
@@ -398,23 +402,23 @@ describe('chatMessage actions', () => {
       ]);
 
       await act(async () => {
-        await result.current.internalResendMessage(messageId);
+        await result.current.internal_resendMessage(messageId);
       });
 
       expect(messageService.removeMessage).not.toHaveBeenCalledWith(messageId);
-      expect(mockState.coreProcessMessage).not.toHaveBeenCalled();
+      expect(mockState.internal_coreProcessMessage).not.toHaveBeenCalled();
       expect(mockState.refreshMessages).not.toHaveBeenCalled();
     });
   });
 
-  describe('internalUpdateMessageContent action', () => {
-    it('should call messageService.internalUpdateMessageContent with correct parameters', async () => {
+  describe('internal_updateMessageContent action', () => {
+    it('should call messageService.internal_updateMessageContent with correct parameters', async () => {
       const { result } = renderHook(() => useChatStore());
       const messageId = 'message-id';
       const newContent = 'Updated content';
 
       await act(async () => {
-        await result.current.internalUpdateMessageContent(messageId, newContent);
+        await result.current.internal_updateMessageContent(messageId, newContent);
       });
 
       expect(messageService.updateMessage).toHaveBeenCalledWith(messageId, { content: newContent });
@@ -424,13 +428,13 @@ describe('chatMessage actions', () => {
       const { result } = renderHook(() => useChatStore());
       const messageId = 'message-id';
       const newContent = 'Updated content';
-      const dispatchMessageSpy = vi.spyOn(result.current, 'dispatchMessage');
+      const internal_dispatchMessageSpy = vi.spyOn(result.current, 'internal_dispatchMessage');
 
       await act(async () => {
-        await result.current.internalUpdateMessageContent(messageId, newContent);
+        await result.current.internal_updateMessageContent(messageId, newContent);
       });
 
-      expect(dispatchMessageSpy).toHaveBeenCalledWith({
+      expect(internal_dispatchMessageSpy).toHaveBeenCalledWith({
         id: messageId,
         key: 'content',
         type: 'updateMessage',
@@ -444,16 +448,16 @@ describe('chatMessage actions', () => {
       const newContent = 'Updated content';
 
       await act(async () => {
-        await result.current.internalUpdateMessageContent(messageId, newContent);
+        await result.current.internal_updateMessageContent(messageId, newContent);
       });
 
       expect(result.current.refreshMessages).toHaveBeenCalled();
     });
   });
 
-  describe('coreProcessMessage action', () => {
+  describe('internal_coreProcessMessage action', () => {
     it('should handle the core AI message processing', async () => {
-      useChatStore.setState({ coreProcessMessage: realCoreProcessMessage });
+      useChatStore.setState({ internal_coreProcessMessage: realCoreProcessMessage });
 
       const { result } = renderHook(() => useChatStore());
       const userMessage = {
@@ -473,7 +477,7 @@ describe('chatMessage actions', () => {
       (messageService.createMessage as Mock).mockResolvedValue('assistant-message-id');
 
       await act(async () => {
-        await result.current.coreProcessMessage(messages, userMessage.id);
+        await result.current.internal_coreProcessMessage(messages, userMessage.id);
       });
 
       // 验证是否创建了代表 AI 响应的消息
@@ -499,7 +503,7 @@ describe('chatMessage actions', () => {
   describe('stopGenerateMessage action', () => {
     it('should stop generating message and set loading states correctly', async () => {
       const { result } = renderHook(() => useChatStore());
-      const toggleChatLoadingSpy = vi.spyOn(result.current, 'toggleChatLoading');
+      const internal_toggleChatLoadingSpy = vi.spyOn(result.current, 'internal_toggleChatLoading');
       const abortController = new AbortController();
 
       act(() => {
@@ -511,7 +515,11 @@ describe('chatMessage actions', () => {
       });
 
       expect(abortController.signal.aborted).toBe(true);
-      expect(toggleChatLoadingSpy).toHaveBeenCalledWith(false, undefined, expect.any(String));
+      expect(internal_toggleChatLoadingSpy).toHaveBeenCalledWith(
+        false,
+        undefined,
+        expect.any(String),
+      );
     });
 
     it('should not do anything if there is no abortController', async () => {
@@ -598,7 +606,7 @@ describe('chatMessage actions', () => {
     });
   });
 
-  describe('fetchAIChatMessage', () => {
+  describe('internal_fetchAIChatMessage', () => {
     it('should fetch AI chat message and return content', async () => {
       const { result } = renderHook(() => useChatStore());
       const messages = [{ id: 'message-id', content: 'Hello', role: 'user' }] as ChatMessage[];
@@ -608,7 +616,10 @@ describe('chatMessage actions', () => {
       (fetch as Mock).mockResolvedValueOnce(new Response(aiResponse));
 
       await act(async () => {
-        const response = await result.current.fetchAIChatMessage(messages, assistantMessageId);
+        const response = await result.current.internal_fetchAIChatMessage(
+          messages,
+          assistantMessageId,
+        );
         expect(response.content).toEqual(aiResponse);
         expect(response.isFunctionCall).toEqual(false);
         expect(response.functionCallAtEnd).toEqual(false);
@@ -627,7 +638,10 @@ describe('chatMessage actions', () => {
       vi.mocked(fetch).mockResolvedValueOnce(new Response(aiResponse));
 
       await act(async () => {
-        const response = await result.current.fetchAIChatMessage(messages, assistantMessageId);
+        const response = await result.current.internal_fetchAIChatMessage(
+          messages,
+          assistantMessageId,
+        );
         expect(response.content).toEqual(aiResponse);
         expect(response.isFunctionCall).toEqual(true);
         expect(response.functionCallAtEnd).toEqual(false);
@@ -646,7 +660,10 @@ describe('chatMessage actions', () => {
       vi.mocked(fetch).mockResolvedValue(new Response(aiResponse));
 
       await act(async () => {
-        const response = await result.current.fetchAIChatMessage(messages, assistantMessageId);
+        const response = await result.current.internal_fetchAIChatMessage(
+          messages,
+          assistantMessageId,
+        );
         expect(response.content).toEqual(aiResponse);
         expect(response.isFunctionCall).toEqual(true);
         expect(response.functionCallAtEnd).toEqual(true);
@@ -667,19 +684,19 @@ describe('chatMessage actions', () => {
 
       await act(async () => {
         await expect(
-          result.current.fetchAIChatMessage(messages, assistantMessageId),
+          result.current.internal_fetchAIChatMessage(messages, assistantMessageId),
         ).rejects.toThrow(errorMessage);
       });
     });
   });
 
-  describe('toggleChatLoading', () => {
+  describe('internal_toggleChatLoading', () => {
     it('should set loading state and create an AbortController when loading is true', () => {
       const { result } = renderHook(() => useChatStore());
       const action = 'loading-action';
 
       act(() => {
-        result.current.toggleChatLoading(true, 'message-id', action);
+        result.current.internal_toggleChatLoading(true, 'message-id', action);
       });
 
       const state = useChatStore.getState();
@@ -693,12 +710,12 @@ describe('chatMessage actions', () => {
 
       // Set initial loading state
       act(() => {
-        result.current.toggleChatLoading(true, 'message-id', 'start-loading-action');
+        result.current.internal_toggleChatLoading(true, 'message-id', 'start-loading-action');
       });
 
       // Stop loading
       act(() => {
-        result.current.toggleChatLoading(false, undefined, action);
+        result.current.internal_toggleChatLoading(false, undefined, action);
       });
 
       const state = useChatStore.getState();
@@ -711,7 +728,7 @@ describe('chatMessage actions', () => {
       const addEventListenerSpy = vi.spyOn(window, 'addEventListener');
 
       act(() => {
-        result.current.toggleChatLoading(true, 'message-id', 'loading-action');
+        result.current.internal_toggleChatLoading(true, 'message-id', 'loading-action');
       });
 
       expect(addEventListenerSpy).toHaveBeenCalledWith('beforeunload', expect.any(Function));
@@ -723,8 +740,8 @@ describe('chatMessage actions', () => {
 
       // Start and then stop loading to trigger the removal of the event listener
       act(() => {
-        result.current.toggleChatLoading(true, 'message-id', 'start-loading-action');
-        result.current.toggleChatLoading(false, undefined, 'stop-loading-action');
+        result.current.internal_toggleChatLoading(true, 'message-id', 'start-loading-action');
+        result.current.internal_toggleChatLoading(false, undefined, 'stop-loading-action');
       });
 
       expect(removeEventListenerSpy).toHaveBeenCalledWith('beforeunload', expect.any(Function));
@@ -736,7 +753,7 @@ describe('chatMessage actions', () => {
 
       act(() => {
         useChatStore.setState({ abortController });
-        result.current.toggleChatLoading(true, 'message-id', 'loading-action');
+        result.current.internal_toggleChatLoading(true, 'message-id', 'loading-action');
       });
 
       const state = useChatStore.getState();