@lobehub/chat 0.105.2 → 0.107.0

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

@@ -1,179 +0,0 @@
-import { produce } from 'immer';
-
-import { ChatTopic, ChatTopicMap } from '@/types/topic';
-
-import { ChatTopicDispatch, topicReducer } from './topic';
-
-describe('topicReducer', () => {
-  let state: ChatTopicMap;
-
-  beforeEach(() => {
-    state = {};
-  });
-
-  describe('addChatTopic', () => {
-    it('should add a new ChatTopic object to state', () => {
-      const payload: ChatTopicDispatch = {
-        type: 'addChatTopic',
-        topic: {
-          id: '1',
-          title: 'Test Topic',
-          createAt: Date.now(),
-          updateAt: Date.now(),
-        },
-      };
-
-      const newState = topicReducer(state, payload);
-
-      expect(newState).toMatchObject({
-        '1': payload.topic,
-      });
-    });
-
-    it('should add a ChatTopic object with correct id', () => {
-      const payload: ChatTopicDispatch = {
-        type: 'addChatTopic',
-        topic: {
-          id: '1',
-          title: 'Test Topic',
-          createAt: Date.now(),
-          updateAt: Date.now(),
-        },
-      };
-
-      const newState = topicReducer(state, payload);
-
-      expect(newState['1']).toBeDefined();
-    });
-
-    it('should add a ChatTopic object with correct properties', () => {
-      const payload: ChatTopicDispatch = {
-        type: 'addChatTopic',
-        topic: {
-          id: '1',
-          title: 'Test Topic',
-          createAt: Date.now(),
-          updateAt: Date.now(),
-        },
-      };
-
-      const newState = topicReducer(state, payload);
-
-      expect(newState['1']).toMatchObject(payload.topic);
-    });
-  });
-
-  describe('updateChatTopic', () => {
-    it('should update the ChatTopic object in state', () => {
-      const topic: ChatTopic = {
-        id: '1',
-        title: 'Test Topic',
-        createAt: Date.now(),
-        updateAt: Date.now(),
-      };
-
-      state['1'] = topic;
-
-      const payload: ChatTopicDispatch = {
-        type: 'updateChatTopic',
-        id: '1',
-        key: 'title',
-        value: 'Updated Topic',
-      };
-
-      const newState = topicReducer(state, payload);
-
-      expect(newState['1'].title).toBe('Updated Topic');
-    });
-
-    it('should update the ChatTopic object with correct properties', () => {
-      const topic: ChatTopic = {
-        id: '1',
-        title: 'Test Topic',
-        createAt: Date.now() - 1,
-        updateAt: Date.now() - 1, // 设定比当前时间前面一点
-      };
-
-      state['1'] = topic;
-
-      const payload: ChatTopicDispatch = {
-        type: 'updateChatTopic',
-        id: '1',
-        key: 'title',
-        value: 'Updated Topic',
-      };
-
-      const newState = topicReducer(state, payload);
-
-      expect(newState['1'].updateAt).toBeGreaterThan(topic.updateAt);
-    });
-  });
-
-  describe('deleteChatTopic', () => {
-    it('should delete the specified ChatTopic object from state', () => {
-      const topic: ChatTopic = {
-        id: '1',
-        title: 'Test Topic',
-        createAt: Date.now(),
-        updateAt: Date.now(),
-      };
-
-      state['1'] = topic;
-
-      const payload: ChatTopicDispatch = {
-        type: 'deleteChatTopic',
-        id: '1',
-      };
-
-      const newState = topicReducer(state, payload);
-
-      expect(newState['1']).toBeUndefined();
-    });
-  });
-
-  describe('default', () => {
-    it('should return the original state object', () => {
-      const payload = {
-        type: 'unknown',
-      } as unknown as ChatTopicDispatch;
-
-      const newState = topicReducer(state, payload);
-
-      expect(newState).toBe(state);
-    });
-  });
-
-  describe('produce', () => {
-    it('should generate immutable state object', () => {
-      const payload: ChatTopicDispatch = {
-        type: 'addChatTopic',
-        topic: {
-          id: '1',
-          title: 'Test Topic',
-          createAt: Date.now(),
-          updateAt: Date.now(),
-        },
-      };
-
-      const newState = topicReducer(state, payload);
-
-      expect(newState).not.toBe(state);
-    });
-
-    it('should not modify the original state object', () => {
-      const payload: ChatTopicDispatch = {
-        type: 'addChatTopic',
-        topic: {
-          id: '1',
-          title: 'Test Topic',
-          createAt: Date.now(),
-          updateAt: Date.now(),
-        },
-      };
-
-      const newState = topicReducer(state, payload);
-
-      expect(state).toMatchObject({});
-    });
-  });
-});

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

@@ -1,81 +0,0 @@
-import { produce } from 'immer';
-
-import { ChatTopic, ChatTopicMap } from '@/types/topic';
-
-interface AddChatTopicAction {
-  topic: ChatTopic;
-  type: 'addChatTopic';
-}
-
-interface UpdateChatTopicAction {
-  id: string;
-  key: keyof ChatTopic;
-  type: 'updateChatTopic';
-  value: any;
-}
-
-interface FavorChatTopicAction {
-  favorite: boolean;
-  id: string;
-  type: 'favorChatTopic';
-}
-
-interface DeleteChatTopicAction {
-  id: string;
-  type: 'deleteChatTopic';
-}
-
-export type ChatTopicDispatch =
-  | AddChatTopicAction
-  | UpdateChatTopicAction
-  | DeleteChatTopicAction
-  | FavorChatTopicAction;
-
-export const topicReducer = (state: ChatTopicMap, payload: ChatTopicDispatch): ChatTopicMap => {
-  switch (payload.type) {
-    case 'addChatTopic': {
-      return produce(state, (draftState) => {
-        draftState[payload.topic.id] = payload.topic;
-      });
-    }
-
-    case 'updateChatTopic': {
-      return produce(state, (draftState) => {
-        const { key, value, id } = payload;
-
-        if (!draftState[id]) return;
-
-        const topic = draftState[id];
-
-        if (value !== undefined) {
-          // @ts-ignore
-          topic[key] = value;
-
-          topic.updateAt = Date.now();
-        }
-      });
-    }
-
-    case 'favorChatTopic': {
-      return produce(state, (draftState) => {
-        const { favorite, id } = payload;
-
-        if (!draftState[id]) return;
-
-        const topic = draftState[id];
-
-        topic.favorite = favorite;
-      });
-    }
-
-    case 'deleteChatTopic': {
-      return produce(state, (draftState) => {
-        delete draftState[payload.id];
-      });
-    }
-
-    default: {
-      return state;
-    }
-  }
-};

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

@@ -1,111 +0,0 @@
-import { LobePluginType } from '@lobehub/chat-plugin-sdk';
-import { t } from 'i18next';
-
-import { DEFAULT_INBOX_AVATAR, DEFAULT_USER_AVATAR } from '@/const/meta';
-import { INBOX_SESSION_ID } from '@/const/session';
-import { useGlobalStore } from '@/store/global';
-import { ChatMessage } from '@/types/chatMessage';
-
-import type { SessionStore } from '../../../store';
-import { agentSelectors } from '../../agent';
-import { sessionSelectors } from '../../session/selectors';
-import { getSlicedMessagesWithConfig } from '../utils';
-import { currentTopics } from './topic';
-import { organizeChats } from './utils';
-
-export const getChatsById =
-  (id: string) =>
-  (s: SessionStore): ChatMessage[] => {
-    const session = sessionSelectors.getSessionById(id)(s);
-
-    if (!session) return [];
-
-    return organizeChats(session, {
-      meta: {
-        assistant: {
-          avatar: agentSelectors.currentAgentAvatar(s),
-          backgroundColor: agentSelectors.currentAgentBackgroundColor(s),
-        },
-        user: {
-          avatar: useGlobalStore.getState().settings.avatar || DEFAULT_USER_AVATAR,
-        },
-      },
-      topicId: s.activeTopicId,
-    });
-  };
-
-// 当前激活的消息列表
-export const currentChats = (s: SessionStore): ChatMessage[] => {
-  if (!s.activeId) return [];
-
-  return getChatsById(s.activeId)(s);
-};
-
-const initTime = Date.now();
-// 针对新助手添加初始化时的自定义消息
-export const currentChatsWithGuideMessage = (s: SessionStore): ChatMessage[] => {
-  const data = currentChats(s);
-  const noTopic = currentTopics(s);
-
-  const isBrandNewChat = data.length === 0 && noTopic;
-
-  if (!isBrandNewChat) return data;
-
-  const [activeId, isInbox] = [s.activeId, s.activeId === INBOX_SESSION_ID];
-  const meta = agentSelectors.currentAgentMeta(s);
-
-  const inboxMsg = t('inbox.defaultMessage', { ns: 'chat' });
-  const agentSystemRoleMsg = t('agentDefaultMessageWithSystemRole', {
-    name: meta.title || t('defaultAgent'),
-    ns: 'chat',
-    systemRole: meta.description,
-  });
-  const agentMsg = t('agentDefaultMessage', {
-    id: activeId,
-    name: meta.title || t('defaultAgent'),
-    ns: 'chat',
-  });
-
-  const emptyInboxGuideMessage = {
-    content: isInbox ? inboxMsg : !!meta.description ? agentSystemRoleMsg : agentMsg,
-    createAt: initTime,
-    extra: {},
-    id: 'default',
-    meta: meta || {
-      avatar: DEFAULT_INBOX_AVATAR,
-    },
-    role: 'assistant',
-    updateAt: initTime,
-  } as ChatMessage;
-
-  return [emptyInboxGuideMessage];
-};
-
-export const currentChatsWithHistoryConfig = (s: SessionStore): ChatMessage[] => {
-  const chats = currentChats(s);
-  const config = agentSelectors.currentAgentConfig(s);
-
-  return getSlicedMessagesWithConfig(chats, config);
-};
-
-export const chatsMessageString = (s: SessionStore): string => {
-  const chats = currentChatsWithHistoryConfig(s);
-  return chats.map((m) => m.content).join('');
-};
-
-export const getFunctionMessageProps =
-  ({ plugin, content, id }: Pick<ChatMessage, 'plugin' | 'content' | 'id'>) =>
-  (s: SessionStore) => ({
-    arguments: plugin?.arguments,
-    command: plugin,
-    content,
-    id: plugin?.identifier,
-    loading: id === s.chatLoadingId,
-    type: plugin?.type as LobePluginType,
-  });
-
-export const getMessageById = (id: string) => (s: SessionStore) => {
-  for (const e of Object.values(s.sessions)) {
-    if (e.chats[id]) return e.chats[id];
-  }
-};

package/src/store/session/slices/chat/selectors/index.ts

@@ -1,26 +0,0 @@
-import {
-  chatsMessageString,
-  currentChats,
-  currentChatsWithGuideMessage,
-  currentChatsWithHistoryConfig,
-  getChatsById,
-  getFunctionMessageProps,
-  getMessageById,
-} from './chat';
-import { currentTopicLength, currentTopics, getTopicMessages } from './topic';
-
-export const chatSelectors = {
-  chatsMessageString,
-  currentChats,
-  currentChatsWithGuideMessage,
-  currentChatsWithHistoryConfig,
-  getChatsById,
-  getFunctionMessageProps,
-  getMessageById,
-};
-
-export const topicSelectors = {
-  currentTopicLength,
-  currentTopics,
-  getTopicMessages,
-};

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

@@ -1,30 +0,0 @@
-// 展示在聊天框中的消息
-import { ChatTopic } from '@/types/topic';
-
-import { SessionStore } from '../../../store';
-import { sessionSelectors } from '../../session/selectors';
-import { organizeChats } from './utils';
-
-export const currentTopics = (s: SessionStore): ChatTopic[] => {
-  const session = sessionSelectors.currentSession(s);
-  if (!session) return [];
-
-  const topics = Object.values(session.topics || {});
-
-  // 按时间倒序
-  const favTopics = topics.filter((t) => t.favorite).sort((a, b) => b.updateAt - a.updateAt);
-  const defaultTopics = topics.filter((t) => !t.favorite).sort((a, b) => b.updateAt - a.updateAt);
-
-  return [...favTopics, ...defaultTopics];
-};
-
-export const currentTopicLength = (s: SessionStore): number => {
-  return currentTopics(s).length;
-};
-
-export const getTopicMessages = (topicId: string) => (s: SessionStore) => {
-  const session = sessionSelectors.currentSession(s);
-  if (!session) return [];
-
-  return organizeChats(session, { topicId });
-};