@lobehub/chat 1.28.6 → 1.29.1

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

@@ -0,0 +1,63 @@
+import { act, renderHook } from '@testing-library/react';
+import { Mock, afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+
+import { messageService } from '@/services/message';
+
+import { useChatStore } from '../../store';
+
+// Mock messageService 和 chatService
+vi.mock('@/services/message', () => ({
+  messageService: {
+    updateMessageTTS: vi.fn(),
+    updateMessageTranslate: vi.fn(),
+    updateMessage: vi.fn(),
+  },
+}));
+
+vi.mock('@/services/chat', () => ({
+  chatService: {
+    fetchPresetTaskResult: vi.fn(),
+  },
+}));
+
+vi.mock('@/chains/langDetect', () => ({
+  chainLangDetect: vi.fn(),
+}));
+
+vi.mock('@/chains/translate', () => ({
+  chainTranslate: vi.fn(),
+}));
+
+// Mock supportLocales
+vi.mock('@/locales/options', () => ({
+  supportLocales: ['en-US', 'zh-CN'],
+}));
+
+beforeEach(() => {
+  vi.clearAllMocks();
+  useChatStore.setState(
+    {
+      // ... 初始状态
+    },
+    false,
+  );
+});
+
+afterEach(() => {
+  vi.restoreAllMocks();
+});
+
+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);
+    });
+  });
+});

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

@@ -0,0 +1,35 @@
+import { StateCreator } from 'zustand/vanilla';
+
+import { messageService } from '@/services/message';
+import { ChatStore } from '@/store/chat/store';
+import { ChatTTS } from '@/types/message';
+
+/**
+ * enhance chat action like translate,tts
+ */
+export interface ChatTTSAction {
+  clearTTS: (id: string) => Promise<void>;
+  ttsMessage: (
+    id: string,
+    state?: { contentMd5?: string; file?: string; voice?: string },
+  ) => Promise<void>;
+  updateMessageTTS: (id: string, data: Partial<ChatTTS> | false) => Promise<void>;
+}
+
+export const chatTTS: StateCreator<ChatStore, [['zustand/devtools', never]], [], ChatTTSAction> = (
+  set,
+  get,
+) => ({
+  clearTTS: async (id) => {
+    await get().updateMessageTTS(id, false);
+  },
+
+  ttsMessage: async (id, state = {}) => {
+    await get().updateMessageTTS(id, state);
+  },
+
+  updateMessageTTS: async (id, data) => {
+    await messageService.updateMessageTTS(id, data);
+    await get().refreshMessages();
+  },
+});

package/src/store/chat/store.ts

@@ -8,19 +8,21 @@ import { createDevtools } from '../middleware/createDevtools';
 import { ChatStoreState, initialState } from './initialState';
 import { ChatBuiltinToolAction, chatToolSlice } from './slices/builtinTool/action';
 import { ChatPortalAction, chatPortalSlice } from './slices/portal/action';
-import { ChatEnhanceAction, chatEnhance } from './slices/enchance/action';
+import { ChatTranslateAction, chatTranslate } from './slices/translate/action';
 import { ChatMessageAction, chatMessage } from './slices/message/action';
 import { ChatPluginAction, chatPlugin } from './slices/plugin/action';
 import { ShareAction, chatShare } from './slices/share/action';
 import { ChatTopicAction, chatTopic } from './slices/topic/action';
 import { chatAiChat, ChatAIChatAction } from './slices/aiChat/actions';
+import { chatTTS, ChatTTSAction } from './slices/tts/action';
 
 export interface ChatStoreAction
   extends ChatMessageAction,
     ChatAIChatAction,
     ChatTopicAction,
     ShareAction,
-    ChatEnhanceAction,
+    ChatTranslateAction,
+    ChatTTSAction,
     ChatPluginAction,
     ChatBuiltinToolAction,
     ChatPortalAction {}
@@ -36,7 +38,8 @@ const createStore: StateCreator<ChatStore, [['zustand/devtools', never]]> = (...
   ...chatAiChat(...params),
   ...chatTopic(...params),
   ...chatShare(...params),
-  ...chatEnhance(...params),
+  ...chatTranslate(...params),
+  ...chatTTS(...params),
   ...chatToolSlice(...params),
   ...chatPlugin(...params),
   ...chatPortalSlice(...params),

package/src/store/file/reducers/uploadFileList.test.ts

@@ -0,0 +1,197 @@
+import { describe, expect, it } from 'vitest';
+
+import { UploadFileItem } from '@/types/files/upload';
+
+import { uploadFileListReducer } from './uploadFileList';
+
+// 创建测试用的文件项
+const createMockFile = (id: string): UploadFileItem => ({
+  id,
+  file: new File(['test'], 'test.txt'),
+  status: 'pending',
+  uploadState: { speed: 2, restTime: 2, progress: 10 },
+});
+
+describe('uploadFileListReducer', () => {
+  describe('addFile action', () => {
+    it('should add a file at the end by default', () => {
+      const initialState: UploadFileItem[] = [createMockFile('1')];
+      const newFile = createMockFile('2');
+
+      const result = uploadFileListReducer(initialState, {
+        type: 'addFile',
+        file: newFile,
+      });
+
+      expect(result).toHaveLength(2);
+      expect(result[1]).toEqual(newFile);
+    });
+
+    it('should add a file at the start when atStart is true', () => {
+      const initialState: UploadFileItem[] = [createMockFile('1')];
+      const newFile = createMockFile('2');
+
+      const result = uploadFileListReducer(initialState, {
+        type: 'addFile',
+        file: newFile,
+        atStart: true,
+      });
+
+      expect(result).toHaveLength(2);
+      expect(result[0]).toEqual(newFile);
+    });
+  });
+
+  describe('addFiles action', () => {
+    it('should add multiple files at the end by default', () => {
+      const initialState: UploadFileItem[] = [createMockFile('1')];
+      const newFiles = [createMockFile('2'), createMockFile('3')];
+
+      const result = uploadFileListReducer(initialState, {
+        type: 'addFiles',
+        files: newFiles,
+      });
+
+      expect(result).toHaveLength(3);
+      expect(result.slice(1)).toEqual(newFiles);
+    });
+
+    it('should add multiple files at the start when atStart is true', () => {
+      const initialState: UploadFileItem[] = [createMockFile('1')];
+      const newFiles = [createMockFile('2'), createMockFile('3')];
+
+      const result = uploadFileListReducer(initialState, {
+        type: 'addFiles',
+        files: newFiles,
+        atStart: true,
+      });
+
+      expect(result).toHaveLength(3);
+      expect(result.slice(0, 2)).toEqual(newFiles.reverse());
+    });
+  });
+
+  describe('updateFile action', () => {
+    it('should update file properties', () => {
+      const initialState: UploadFileItem[] = [createMockFile('1')];
+      const updateValue = { name: 'updated.txt' } as File;
+
+      const result = uploadFileListReducer(initialState, {
+        type: 'updateFile',
+        id: '1',
+        value: { file: updateValue },
+      });
+
+      expect(result[0].file.name).toBe('updated.txt');
+    });
+
+    it('should not modify state if file id is not found', () => {
+      const initialState: UploadFileItem[] = [createMockFile('1')];
+
+      const result = uploadFileListReducer(initialState, {
+        type: 'updateFile',
+        id: 'non-existent',
+        value: { file: { name: 'updated.txt' } as File },
+      });
+
+      expect(result).toEqual(initialState);
+    });
+  });
+
+  describe('updateFileStatus action', () => {
+    it('should update file status', () => {
+      const initialState: UploadFileItem[] = [createMockFile('1')];
+
+      const result = uploadFileListReducer(initialState, {
+        type: 'updateFileStatus',
+        id: '1',
+        status: 'success',
+      });
+
+      expect(result[0].status).toBe('success');
+    });
+  });
+
+  describe('updateFileUploadState action', () => {
+    it('should update file upload state', () => {
+      const initialState: UploadFileItem[] = [createMockFile('1')];
+
+      const result = uploadFileListReducer(initialState, {
+        type: 'updateFileUploadState',
+        id: '1',
+        uploadState: { progress: 12, restTime: 12, speed: 12 },
+      });
+
+      expect(result[0].uploadState).toEqual({
+        progress: 12,
+        restTime: 12,
+        speed: 12,
+      });
+    });
+  });
+
+  describe('removeFile action', () => {
+    it('should remove a file by id', () => {
+      const initialState: UploadFileItem[] = [createMockFile('1'), createMockFile('2')];
+
+      const result = uploadFileListReducer(initialState, {
+        type: 'removeFile',
+        id: '1',
+      });
+
+      expect(result).toHaveLength(1);
+      expect(result[0].id).toBe('2');
+    });
+
+    it('should not modify state if file id is not found', () => {
+      const initialState: UploadFileItem[] = [createMockFile('1')];
+
+      const result = uploadFileListReducer(initialState, {
+        type: 'removeFile',
+        id: 'non-existent',
+      });
+
+      expect(result).toEqual(initialState);
+    });
+  });
+
+  describe('removeFiles action', () => {
+    it('should remove multiple files by ids', () => {
+      const initialState: UploadFileItem[] = [
+        createMockFile('1'),
+        createMockFile('2'),
+        createMockFile('3'),
+      ];
+
+      const result = uploadFileListReducer(initialState, {
+        type: 'removeFiles',
+        ids: ['1', '2'],
+      });
+
+      expect(result).toHaveLength(1);
+      expect(result[0].id).toBe('3');
+    });
+
+    it('should not modify state if no file ids are found', () => {
+      const initialState: UploadFileItem[] = [createMockFile('1')];
+
+      const result = uploadFileListReducer(initialState, {
+        type: 'removeFiles',
+        ids: ['non-existent'],
+      });
+
+      expect(result).toEqual(initialState);
+    });
+  });
+
+  describe('error handling', () => {
+    it('should throw error for unhandled action type', () => {
+      const initialState: UploadFileItem[] = [];
+
+      // @ts-expect-error Testing invalid action type
+      expect(() => uploadFileListReducer(initialState, { type: 'invalid' })).toThrow(
+        'Unhandled action type',
+      );
+    });
+  });
+});

package/src/store/user/slices/settings/selectors/__snapshots__/settings.test.ts.snap

@@ -73,7 +73,9 @@ exports[`settingsSelectors > defaultAgent > should merge DEFAULT_AGENT and s.set
       "autoCreateTopicThreshold": 2,
       "displayMode": "chat",
       "enableAutoCreateTopic": true,
-      "historyCount": 1,
+      "enableCompressHistory": true,
+      "enableHistoryCount": true,
+      "historyCount": 8,
     },
     "model": "gpt-3.5-turbo",
     "params": {

package/src/store/user/slices/settings/selectors/systemAgent.ts

@@ -11,9 +11,11 @@ const translation = (s: UserStore) => currentSystemAgent(s).translation;
 const topic = (s: UserStore) => currentSystemAgent(s).topic;
 const agentMeta = (s: UserStore) => currentSystemAgent(s).agentMeta;
 const queryRewrite = (s: UserStore) => currentSystemAgent(s).queryRewrite;
+const historyCompress = (s: UserStore) => currentSystemAgent(s).historyCompress;
 
 export const systemAgentSelectors = {
   agentMeta,
+  historyCompress,
   queryRewrite,
   topic,
   translation,

package/src/types/agent/index.ts

@@ -56,13 +56,12 @@ export interface LobeAgentConfig {
 
 export interface LobeAgentChatConfig {
   autoCreateTopicThreshold: number;
-  compressThreshold?: number;
   displayMode?: 'chat' | 'docs';
   enableAutoCreateTopic?: boolean;
   /**
    * 历史消息长度压缩阈值
    */
-  enableCompressThreshold?: boolean;
+  enableCompressHistory?: boolean;
   /**
    * 开启历史记录条数
    */
@@ -78,10 +77,9 @@ export interface LobeAgentChatConfig {
 
 export const AgentChatConfigSchema = z.object({
   autoCreateTopicThreshold: z.number().default(2),
-  compressThreshold: z.number().optional(),
   displayMode: z.enum(['chat', 'docs']).optional(),
   enableAutoCreateTopic: z.boolean().optional(),
-  enableCompressThreshold: z.boolean().optional(),
+  enableCompressHistory: z.boolean().optional(),
   enableHistoryCount: z.boolean().optional(),
   enableMaxTokens: z.boolean().optional(),
   historyCount: z.number().optional(),

package/src/types/topic.ts

@@ -24,8 +24,21 @@ export interface GroupedTopic {
   title?: string;
 }
 
+export interface ChatTopicMetadata {
+  model?: string;
+  provider?: string;
+}
+
+export interface ChatTopicSummary {
+  content: string;
+  model: string;
+  provider: string;
+}
+
 export interface ChatTopic extends Omit<BaseDataModel, 'meta'> {
   favorite?: boolean;
+  historySummary?: string;
+  metadata?: ChatTopicMetadata;
   sessionId?: string;
   title: string;
 }

package/src/types/user/settings/systemAgent.ts

@@ -11,6 +11,7 @@ export interface QueryRewriteSystemAgent extends Omit<SystemAgentItem, 'enabled'
 
 export interface UserSystemAgentConfig {
   agentMeta: SystemAgentItem;
+  historyCompress: SystemAgentItem;
   queryRewrite: QueryRewriteSystemAgent;
   topic: SystemAgentItem;
   translation: SystemAgentItem;

package/src/utils/tokenizer/client.ts

@@ -17,7 +17,7 @@ export const clientEncodeAsync = (str: string): Promise<number> =>
       return;
     }
 
-    const id = Date.now().toString();
+    const id = str;
 
     const handleMessage = (event: MessageEvent) => {
       if (event.data.id === id) {