@lobehub/chat 0.105.2 → 0.107.0

package/src/store/session/slices/chat/actions/translate.ts

@@ -1,113 +0,0 @@
-import { produce } from 'immer';
-import { StateCreator } from 'zustand/vanilla';
-
-import { chainLangDetect } from '@/chains/langDetect';
-import { chainTranslate } from '@/chains/translate';
-import { supportLocales } from '@/locales/options';
-import { chatService } from '@/services/chat';
-import { SessionStore } from '@/store/session';
-import { setNamespace } from '@/utils/storeDebug';
-
-import { sessionSelectors } from '../../session/selectors';
-
-const t = setNamespace('chat/translate');
-
-/**
- * 翻译事件
- */
-export interface ChatTranslateAction {
-  clearTTS: (id: string) => void;
-
-  clearTranslate: (id: string) => void;
-  /**
-   * 翻译消息
-   * @param id
-   */
-  translateMessage: (id: string, targetLang: string) => Promise<void>;
-  ttsMessage: (id: string, init?: boolean) => void;
-}
-
-export const chatTranslate: StateCreator<
-  SessionStore,
-  [['zustand/devtools', never]],
-  [],
-  ChatTranslateAction
-> = (set, get) => ({
-  clearTTS: (id) => {
-    get().dispatchMessage({
-      id,
-      key: 'tts',
-      type: 'updateMessageExtra',
-      value: null,
-    });
-  },
-
-  clearTranslate: (id) => {
-    get().dispatchMessage({
-      id,
-      key: 'translate',
-      type: 'updateMessageExtra',
-      value: null,
-    });
-  },
-
-  translateMessage: async (id, targetLang) => {
-    const { toggleChatLoading, dispatchMessage } = get();
-    const session = sessionSelectors.currentSession(get());
-    if (!session || !id) return;
-
-    const message = session.chats[id];
-    if (!message) return;
-
-    let content = '';
-    let from = '';
-
-    dispatchMessage({
-      id,
-      key: 'translate',
-      type: 'updateMessageExtra',
-      value: { content: '', to: targetLang },
-    });
-
-    toggleChatLoading(true, id, t('translateMessage(start)', { id }) as string);
-
-    // detect from language
-    chatService
-      .fetchPresetTaskResult({
-        params: chainLangDetect(message.content),
-      })
-      .then((data) => {
-        if (data && supportLocales.includes(data)) from = data;
-      });
-
-    // translate to target language
-    await chatService.fetchPresetTaskResult({
-      onMessageHandle: (text) => {
-        dispatchMessage({
-          id,
-          key: 'translate',
-          type: 'updateMessageExtra',
-          value: produce({ content: '', from, to: targetLang }, (draft) => {
-            content += text;
-            draft.content += content;
-          }),
-        });
-      },
-      params: chainTranslate(message.content, targetLang),
-    });
-
-    toggleChatLoading(false);
-  },
-
-  ttsMessage: (id, init) => {
-    const { dispatchMessage } = get();
-    dispatchMessage({
-      id,
-      key: 'tts',
-      type: 'updateMessageExtra',
-      value: {
-        init: Boolean(init),
-      },
-    });
-  },
-});

package/src/store/session/slices/chat/helpers.ts

@@ -1,9 +0,0 @@
-import { OpenAIChatMessage } from '@/types/openai/chat';
-import { encodeAsync } from '@/utils/tokenizer';
-
-export const getMessagesTokenCount = async (messages: OpenAIChatMessage[]) =>
-  encodeAsync(messages.map((m) => m.content).join(''));
-
-export const chatHelpers = {
-  getMessagesTokenCount,
-};

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

@@ -1,13 +0,0 @@
-export interface ChatState {
-  abortController?: AbortController;
-  activeTopicId?: string;
-  chatLoadingId?: string;
-  inputMessage: string;
-  renameTopicId?: string;
-  shareLoading?: boolean;
-  topicLoadingId?: string;
-}
-
-export const initialChatState: ChatState = {
-  inputMessage: '',
-};

package/src/store/session/slices/chat/reducers/message.test.ts

@@ -1,319 +0,0 @@
-import { ChatMessageMap } from '@/types/chatMessage';
-
-import { MessageDispatch, messagesReducer } from './message';
-
-describe('messagesReducer', () => {
-  let initialState: ChatMessageMap;
-
-  beforeEach(() => {
-    initialState = {
-      message1: {
-        id: 'message1',
-        content: 'Hello World',
-        createAt: 1629264000000,
-        updateAt: 1629264000000,
-        role: 'user',
-      },
-      message2: {
-        id: 'message2',
-        content: 'How are you?',
-        createAt: 1629264000000,
-        updateAt: 1629264000000,
-        role: 'system',
-      },
-    } as unknown as ChatMessageMap;
-  });
-
-  describe('addMessage', () => {
-    it('should add a new message to the state', () => {
-      const payload: MessageDispatch = {
-        type: 'addMessage',
-        message: 'New Message',
-        role: 'user',
-        id: 'message3',
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(Object.keys(newState)).toHaveLength(3);
-      expect(newState).toHaveProperty('message1');
-      expect(newState).toHaveProperty('message2');
-      expect(newState).toHaveProperty('message3');
-      expect(newState.message3).toEqual({
-        id: 'message3',
-        content: 'New Message',
-        meta: {},
-        createAt: expect.any(Number),
-        updateAt: expect.any(Number),
-        role: 'user',
-      });
-    });
-
-    it('should use the provided id when adding a new message', () => {
-      const payload: MessageDispatch = {
-        type: 'addMessage',
-        id: 'customId',
-        message: 'New Message',
-        role: 'user',
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(Object.keys(newState)).toHaveLength(3);
-      expect(newState).toHaveProperty('message1');
-      expect(newState).toHaveProperty('message2');
-      expect(newState).toHaveProperty('customId');
-      expect(newState.customId).toEqual({
-        id: 'customId',
-        content: 'New Message',
-        meta: {},
-        createAt: expect.any(Number),
-        updateAt: expect.any(Number),
-        role: 'user',
-      });
-    });
-
-    it('should use the provided parentId when adding a new message', () => {
-      const payload: MessageDispatch = {
-        type: 'addMessage',
-        message: 'New Message',
-        id: 'message3',
-        role: 'user',
-        parentId: 'message1',
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(newState.message3).toEqual({
-        id: 'message3',
-        content: 'New Message',
-        meta: {},
-        createAt: expect.any(Number),
-        updateAt: expect.any(Number),
-        role: 'user',
-        parentId: 'message1',
-      });
-    });
-
-    it('should use the provided quotaId when adding a new message', () => {
-      const payload: MessageDispatch = {
-        type: 'addMessage',
-        message: 'New Message',
-        id: 'message3',
-        role: 'user',
-        quotaId: 'message2',
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(newState.message3).toEqual({
-        id: 'message3',
-        meta: {},
-        content: 'New Message',
-        createAt: expect.any(Number),
-        updateAt: expect.any(Number),
-        role: 'user',
-        quotaId: 'message2',
-      });
-    });
-
-    it('should use the provided parentId and quotaId when adding a new message', () => {
-      const payload: MessageDispatch = {
-        type: 'addMessage',
-        message: 'New Message',
-        id: 'message3',
-        role: 'user',
-        parentId: 'message1',
-        quotaId: 'message2',
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(newState.message3).toEqual({
-        id: 'message3',
-        content: 'New Message',
-        meta: {},
-        createAt: expect.any(Number),
-        updateAt: expect.any(Number),
-        role: 'user',
-        parentId: 'message1',
-        quotaId: 'message2',
-      });
-    });
-  });
-
-  describe('deleteMessage', () => {
-    it('should remove the specified message from the state', () => {
-      const payload: MessageDispatch = {
-        type: 'deleteMessage',
-        id: 'message1',
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(Object.keys(newState)).toHaveLength(1);
-      expect(newState).not.toHaveProperty('message1');
-      expect(newState).toHaveProperty('message2');
-    });
-
-    it('should not modify the state if the specified message does not exist', () => {
-      const payload: MessageDispatch = {
-        type: 'deleteMessage',
-        id: 'nonexistentMessage',
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(newState).toEqual(initialState);
-    });
-
-    it('should not modify the state if the specified message does not exist', () => {
-      const payload: MessageDispatch = {
-        type: 'deleteMessage',
-        id: 'nonexistentMessage',
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(newState).toEqual(initialState);
-    });
-  });
-
-  describe('updateMessage', () => {
-    it('should update the specified message with the provided value', () => {
-      const payload: MessageDispatch = {
-        type: 'updateMessage',
-        id: 'message1',
-        key: 'content',
-        value: 'Updated Message',
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(newState.message1.content).toBe('Updated Message');
-      expect(newState.message1.updateAt).toBeGreaterThan(initialState.message1.updateAt);
-    });
-
-    it('should not modify the state if the specified message does not exist', () => {
-      const payload: MessageDispatch = {
-        type: 'updateMessage',
-        id: 'nonexistentMessage',
-        key: 'content',
-        value: 'Updated Message',
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(newState).toEqual(initialState);
-    });
-
-    it('should not modify the state if the specified message does not exist', () => {
-      const payload: MessageDispatch = {
-        type: 'updateMessage',
-        id: 'nonexistentMessage',
-        key: 'content',
-        value: 'Updated Message',
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(newState).toEqual(initialState);
-    });
-  });
-
-  describe('updateMessageExtra', () => {
-    it('should update the specified message extra property with the provided value', () => {
-      const payload: MessageDispatch = {
-        type: 'updateMessageExtra',
-        id: 'message1',
-        key: 'translate',
-        value: { target: 'en', to: 'zh' },
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(newState.message1.extra!.translate).toEqual({ target: 'en', to: 'zh' });
-      expect(newState.message1.updateAt).toBeGreaterThan(initialState.message1.updateAt);
-    });
-
-    it('should not modify the state if the specified message does not exist', () => {
-      const payload: MessageDispatch = {
-        type: 'updateMessageExtra',
-        id: 'nonexistentMessage',
-        key: 'translate',
-        value: { target: 'en', to: 'zh' },
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(newState).toEqual(initialState);
-    });
-
-    it('should not modify the state if the specified message does not exist', () => {
-      const payload: MessageDispatch = {
-        type: 'updateMessageExtra',
-        id: 'nonexistentMessage',
-        key: 'translate',
-        value: { target: 'en', to: 'zh' },
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(newState).toEqual(initialState);
-    });
-  });
-
-  describe('resetMessages', () => {
-    it('should reset the state to an empty object', () => {
-      const payload: MessageDispatch = {
-        type: 'resetMessages',
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(newState).toEqual({});
-    });
-
-    it('should delete messages with the specified topicId', () => {
-      const initialState = {
-        message1: {
-          id: 'message1',
-          content: 'Hello World',
-          createAt: 1629264000000,
-          updateAt: 1629264000000,
-          role: 'user',
-          topicId: 'topic1',
-        },
-        message2: {
-          id: 'message2',
-          content: 'How are you?',
-          createAt: 1629264000000,
-          updateAt: 1629264000000,
-          role: 'system',
-        },
-      } as unknown as ChatMessageMap;
-
-      const payload: MessageDispatch = {
-        type: 'resetMessages',
-        topicId: 'topic1',
-      };
-
-      const newState = messagesReducer(initialState, payload);
-
-      expect(Object.keys(newState)).toHaveLength(1);
-      expect(newState).not.toHaveProperty('message1');
-      expect(newState).toHaveProperty('message2');
-    });
-  });
-
-  describe('unimplemented type', () => {
-    it('should throw an error when an unimplemented type is provided', () => {
-      // @ts-ignore
-      const payload: MessageDispatch = { type: 'unimplementedType' };
-
-      expect(() => messagesReducer(initialState, payload)).toThrowError(
-        '暂未实现的 type，请检查 reducer',
-      );
-    });
-  });
-});

package/src/store/session/slices/chat/reducers/message.ts

@@ -1,156 +0,0 @@
-import isEqual from 'fast-deep-equal';
-import { produce } from 'immer';
-
-import { ChatMessage, ChatMessageMap } from '@/types/chatMessage';
-import { LLMRoleType } from '@/types/llm';
-import { MetaData } from '@/types/meta';
-import { merge } from '@/utils/merge';
-import { nanoid } from '@/utils/uuid';
-
-interface AddMessage {
-  id?: string;
-  message: string;
-  meta?: MetaData;
-  parentId?: string;
-  quotaId?: string;
-  role: LLMRoleType;
-  type: 'addMessage';
-}
-
-interface DeleteMessage {
-  id: string;
-  type: 'deleteMessage';
-}
-
-interface ResetMessages {
-  topicId?: string;
-  type: 'resetMessages';
-}
-
-interface UpdateMessage {
-  id: string;
-  key: keyof ChatMessage;
-  type: 'updateMessage';
-  value: ChatMessage[keyof ChatMessage];
-}
-interface UpdateMessageExtra {
-  id: string;
-  key: string;
-  type: 'updateMessageExtra';
-  value: any;
-}
-
-interface UpdatePluginState {
-  id: string;
-  key: string;
-  type: 'updatePluginState';
-  value: any;
-}
-
-export type MessageDispatch =
-  | AddMessage
-  | DeleteMessage
-  | ResetMessages
-  | UpdateMessage
-  | UpdateMessageExtra
-  | UpdatePluginState;
-
-export const messagesReducer = (
-  state: ChatMessageMap,
-  payload: MessageDispatch,
-): ChatMessageMap => {
-  switch (payload.type) {
-    case 'addMessage': {
-      return produce(state, (draftState) => {
-        const mid = payload.id || nanoid();
-
-        draftState[mid] = {
-          content: payload.message,
-          createAt: Date.now(),
-          id: mid,
-          meta: payload.meta || {},
-          parentId: payload.parentId,
-          quotaId: payload.quotaId,
-          role: payload.role,
-          updateAt: Date.now(),
-        };
-      });
-    }
-
-    case 'deleteMessage': {
-      return produce(state, (draftState) => {
-        delete draftState[payload.id];
-      });
-    }
-
-    case 'updateMessage': {
-      return produce(state, (draftState) => {
-        const { id, key, value } = payload;
-        const message = draftState[id];
-        if (!message) return;
-
-        // @ts-ignore
-        message[key] = value;
-        message.updateAt = Date.now();
-      });
-    }
-
-    case 'updateMessageExtra': {
-      return produce(state, (draftState) => {
-        const { id, key, value } = payload;
-        const message = draftState[id];
-        if (!message) return;
-
-        if (!message.extra) {
-          message.extra = { [key]: value } as any;
-        } else {
-          message.extra[key] = value;
-        }
-
-        message.updateAt = Date.now();
-      });
-    }
-
-    case 'updatePluginState': {
-      return produce(state, (draftState) => {
-        const { id, key, value } = payload;
-        const message = draftState[id];
-        if (!message) return;
-
-        let newState;
-        if (!message.pluginState) {
-          newState = { [key]: value } as any;
-        } else {
-          newState = merge(message.pluginState, { [key]: value });
-        }
-
-        if (isEqual(message.pluginState, newState)) return;
-
-        message.pluginState = newState;
-        message.updateAt = Date.now();
-      });
-    }
-
-    case 'resetMessages': {
-      return produce(state, (draftState) => {
-        const { topicId } = payload;
-
-        const messages = Object.values(draftState).filter((message) => {
-          // 如果没有 topicId，说明是清空默认对话里的消息
-          if (!topicId) return !message.topicId;
-
-          return message.topicId === topicId;
-        });
-
-        // 删除上述找到的消息
-        for (const message of messages) {
-          delete draftState[message.id];
-        }
-      });
-    }
-
-    default: {
-      throw new Error('暂未实现的 type，请检查 reducer');
-    }
-  }
-};