@lobehub/chat 0.147.16 → 0.147.17

package/src/store/session/slices/agent/action.test.ts

@@ -0,0 +1,175 @@
+import { act, renderHook } from '@testing-library/react';
+import * as immer from 'immer';
+import { describe, expect, it, vi } from 'vitest';
+
+import { sessionService } from '@/services/session';
+import { useGlobalStore } from '@/store/global';
+import { useSessionStore } from '@/store/session';
+import { agentSelectors, sessionSelectors } from '@/store/session/selectors';
+
+describe('AgentSlice', () => {
+  describe('removePlugin', () => {
+    it('should call togglePlugin with the provided id and false', async () => {
+      const { result } = renderHook(() => useSessionStore());
+      const pluginId = 'plugin-id';
+      const togglePluginMock = vi.spyOn(result.current, 'togglePlugin');
+
+      await act(async () => {
+        await result.current.removePlugin(pluginId);
+      });
+
+      expect(togglePluginMock).toHaveBeenCalledWith(pluginId, false);
+      togglePluginMock.mockRestore();
+    });
+  });
+
+  describe('togglePlugin', () => {
+    it('should add plugin id to plugins array if not present and open is true or undefined', async () => {
+      const { result } = renderHook(() => useSessionStore());
+      const pluginId = 'plugin-id';
+      const updateAgentConfigMock = vi.spyOn(result.current, 'updateAgentConfig');
+
+      // 模拟当前配置不包含插件 ID
+      vi.spyOn(agentSelectors, 'currentAgentConfig').mockReturnValue({ plugins: [] } as any);
+
+      await act(async () => {
+        await result.current.togglePlugin(pluginId);
+      });
+
+      expect(updateAgentConfigMock).toHaveBeenCalledWith(
+        expect.objectContaining({ plugins: [pluginId] }),
+      );
+      updateAgentConfigMock.mockRestore();
+    });
+
+    it('should remove plugin id from plugins array if present and open is false', async () => {
+      const { result } = renderHook(() => useSessionStore());
+      const pluginId = 'plugin-id';
+      const updateAgentConfigMock = vi.spyOn(result.current, 'updateAgentConfig');
+
+      // 模拟当前配置包含插件 ID
+      vi.spyOn(agentSelectors, 'currentAgentConfig').mockReturnValue({
+        plugins: [pluginId],
+      } as any);
+
+      await act(async () => {
+        await result.current.togglePlugin(pluginId, false);
+      });
+
+      expect(updateAgentConfigMock).toHaveBeenCalledWith(expect.objectContaining({ plugins: [] }));
+      updateAgentConfigMock.mockRestore();
+    });
+
+    it('should not modify plugins array if plugin id is not present and open is false', async () => {
+      const { result } = renderHook(() => useSessionStore());
+      const pluginId = 'plugin-id';
+      const updateAgentConfigMock = vi.spyOn(result.current, 'updateAgentConfig');
+
+      // 模拟当前配置不包含插件 ID
+      vi.spyOn(agentSelectors, 'currentAgentConfig').mockReturnValue({ plugins: [] } as any);
+
+      await act(async () => {
+        await result.current.togglePlugin(pluginId, false);
+      });
+
+      expect(updateAgentConfigMock).toHaveBeenCalledWith(expect.objectContaining({ plugins: [] }));
+      updateAgentConfigMock.mockRestore();
+    });
+  });
+
+  describe('updateAgentConfig', () => {
+    it('should update global config if current session is inbox session', async () => {
+      const { result } = renderHook(() => useSessionStore());
+      const config = { model: 'gpt-3.5-turbo' };
+      const updateDefaultAgentMock = vi.spyOn(useGlobalStore.getState(), 'updateDefaultAgent');
+
+      // 模拟当前会话是收件箱会话
+      vi.spyOn(sessionSelectors, 'isInboxSession').mockReturnValue(true);
+
+      await act(async () => {
+        await result.current.updateAgentConfig(config);
+      });
+
+      expect(updateDefaultAgentMock).toHaveBeenCalledWith({ config });
+      updateDefaultAgentMock.mockRestore();
+    });
+
+    it('should update session config if current session is not inbox session', async () => {
+      const { result } = renderHook(() => useSessionStore());
+      const config = { model: 'gpt-3.5-turbo' };
+      const updateSessionConfigMock = vi.spyOn(sessionService, 'updateSessionConfig');
+      const refreshSessionsMock = vi.spyOn(result.current, 'refreshSessions');
+
+      // 模拟当前会话不是收件箱会话
+      vi.spyOn(sessionSelectors, 'isInboxSession').mockReturnValue(false);
+      vi.spyOn(sessionSelectors, 'currentSession').mockReturnValue({ id: 'session-id' } as any);
+      vi.spyOn(result.current, 'activeId', 'get').mockReturnValue('session-id');
+
+      await act(async () => {
+        await result.current.updateAgentConfig(config);
+      });
+
+      expect(updateSessionConfigMock).toHaveBeenCalledWith('session-id', config);
+      expect(refreshSessionsMock).toHaveBeenCalled();
+      updateSessionConfigMock.mockRestore();
+      refreshSessionsMock.mockRestore();
+    });
+
+    it('should not update config if there is no current session', async () => {
+      const { result } = renderHook(() => useSessionStore());
+      const config = { model: 'gpt-3.5-turbo' };
+      const updateSessionConfigMock = vi.spyOn(sessionService, 'updateSessionConfig');
+
+      // 模拟没有当前会话
+      vi.spyOn(sessionSelectors, 'currentSession').mockReturnValue(null as any);
+
+      await act(async () => {
+        await result.current.updateAgentConfig(config);
+      });
+
+      expect(updateSessionConfigMock).not.toHaveBeenCalled();
+      updateSessionConfigMock.mockRestore();
+    });
+  });
+
+  describe('updateAgentMeta', () => {
+    it('should not update meta if there is no current session', async () => {
+      const { result } = renderHook(() => useSessionStore());
+      const meta = { title: 'Test Agent' };
+      const updateSessionMock = vi.spyOn(sessionService, 'updateSession');
+      const refreshSessionsMock = vi.spyOn(result.current, 'refreshSessions');
+
+      // 模拟没有当前会话
+      vi.spyOn(sessionSelectors, 'currentSession').mockReturnValue(null as any);
+
+      await act(async () => {
+        await result.current.updateAgentMeta(meta as any);
+      });
+
+      expect(updateSessionMock).not.toHaveBeenCalled();
+      expect(refreshSessionsMock).not.toHaveBeenCalled();
+      updateSessionMock.mockRestore();
+      refreshSessionsMock.mockRestore();
+    });
+
+    it('should update session meta and refresh sessions', async () => {
+      const { result } = renderHook(() => useSessionStore());
+      const meta = { title: 'Test Agent' };
+      const updateSessionMock = vi.spyOn(sessionService, 'updateSession');
+      const refreshSessionsMock = vi.spyOn(result.current, 'refreshSessions');
+
+      // 模拟有当前会话
+      vi.spyOn(sessionSelectors, 'currentSession').mockReturnValue({ id: 'session-id' } as any);
+      vi.spyOn(result.current, 'activeId', 'get').mockReturnValue('session-id');
+
+      await act(async () => {
+        await result.current.updateAgentMeta(meta);
+      });
+
+      expect(updateSessionMock).toHaveBeenCalledWith('session-id', { meta });
+      expect(refreshSessionsMock).toHaveBeenCalled();
+      updateSessionMock.mockRestore();
+      refreshSessionsMock.mockRestore();
+    });
+  });
+});

package/src/store/session/slices/agent/action.ts

@@ -78,7 +78,7 @@ export const createAgentSlice: StateCreator<
 
     const { activeId, refreshSessions } = get();
 
-    await sessionService.updateSessionMeta(activeId, meta);
+    await sessionService.updateSession(activeId, { meta });
     await refreshSessions();
   },
 });

package/src/store/session/slices/session/action.test.ts

@@ -1,8 +1,6 @@
 import { act, renderHook } from '@testing-library/react';
-import { AppRouterInstance } from 'next/dist/shared/lib/app-router-context.shared-runtime';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { INBOX_SESSION_ID } from '@/const/session';
 import { SESSION_CHAT_URL } from '@/const/url';
 import { sessionService } from '@/services/session';
 import { useSessionStore } from '@/store/session';
@@ -12,11 +10,12 @@ import { LobeSessionType } from '@/types/session';
 vi.mock('@/services/session', () => ({
   sessionService: {
     removeAllSessions: vi.fn(),
-    createNewSession: vi.fn(),
-    duplicateSession: vi.fn(),
+    createSession: vi.fn(),
+    cloneSession: vi.fn(),
     updateSessionGroup: vi.fn(),
     removeSession: vi.fn(),
-    getSessions: vi.fn(),
+    getAllSessions: vi.fn(),
+    updateSession: vi.fn(),
     updateSessionGroupId: vi.fn(),
     searchSessions: vi.fn(),
     updateSessionPinned: vi.fn(),
@@ -56,7 +55,7 @@ describe('SessionAction', () => {
     it('should create a new session and switch to it', async () => {
       const { result } = renderHook(() => useSessionStore());
       const newSessionId = 'new-session-id';
-      vi.mocked(sessionService.createNewSession).mockResolvedValue(newSessionId);
+      vi.mocked(sessionService.createSession).mockResolvedValue(newSessionId);
 
       let createdSessionId;
 
@@ -64,7 +63,7 @@ describe('SessionAction', () => {
         createdSessionId = await result.current.createSession({ config: { displayMode: 'docs' } });
       });
 
-      const call = vi.mocked(sessionService.createNewSession).mock.calls[0];
+      const call = vi.mocked(sessionService.createSession).mock.calls[0];
       expect(call[0]).toEqual(LobeSessionType.Agent);
       expect(call[1]).toMatchObject({ config: { displayMode: 'docs' } });
 
@@ -74,7 +73,7 @@ describe('SessionAction', () => {
     it('should create a new session but not switch to it if isSwitchSession is false', async () => {
       const { result } = renderHook(() => useSessionStore());
       const newSessionId = 'new-session-id';
-      vi.mocked(sessionService.createNewSession).mockResolvedValue(newSessionId);
+      vi.mocked(sessionService.createSession).mockResolvedValue(newSessionId);
 
       let createdSessionId;
 
@@ -85,7 +84,7 @@ describe('SessionAction', () => {
         );
       });
 
-      const call = vi.mocked(sessionService.createNewSession).mock.calls[0];
+      const call = vi.mocked(sessionService.createSession).mock.calls[0];
       expect(call[0]).toEqual(LobeSessionType.Agent);
       expect(call[1]).toMatchObject({ config: { displayMode: 'docs' } });
 
@@ -96,18 +95,18 @@ describe('SessionAction', () => {
     });
   });
 
-  describe('duplicateSession', () => {
+  describe('cloneSession', () => {
     it('should duplicate a session and switch to the new one', async () => {
       const { result } = renderHook(() => useSessionStore());
       const sessionId = 'session-id';
       const duplicatedSessionId = 'duplicated-session-id';
-      vi.mocked(sessionService.duplicateSession).mockResolvedValue(duplicatedSessionId);
+      vi.mocked(sessionService.cloneSession).mockResolvedValue(duplicatedSessionId);
 
       await act(async () => {
         await result.current.duplicateSession(sessionId);
       });
 
-      expect(sessionService.duplicateSession).toHaveBeenCalledWith(sessionId, undefined);
+      expect(sessionService.cloneSession).toHaveBeenCalledWith(sessionId, undefined);
     });
   });
 
@@ -147,7 +146,7 @@ describe('SessionAction', () => {
         await result.current.pinSession(sessionId, true);
       });
 
-      expect(sessionService.updateSessionPinned).toHaveBeenCalledWith(sessionId, true);
+      expect(sessionService.updateSession).toHaveBeenCalledWith(sessionId, { pinned: true });
       expect(mockRefresh).toHaveBeenCalled();
     });
 
@@ -159,7 +158,7 @@ describe('SessionAction', () => {
         await result.current.pinSession(sessionId, false);
       });
 
-      expect(sessionService.updateSessionPinned).toHaveBeenCalledWith(sessionId, false);
+      expect(sessionService.updateSession).toHaveBeenCalledWith(sessionId, { pinned: false });
       expect(mockRefresh).toHaveBeenCalled();
     });
   });
@@ -174,7 +173,7 @@ describe('SessionAction', () => {
         await result.current.updateSessionGroupId(sessionId, groupId);
       });
 
-      expect(sessionService.updateSessionGroupId).toHaveBeenCalledWith(sessionId, groupId);
+      expect(sessionService.updateSession).toHaveBeenCalledWith(sessionId, { group: groupId });
       expect(mockRefresh).toHaveBeenCalled();
     });
   });

package/src/store/session/slices/session/action.ts

@@ -91,7 +91,7 @@ export const createSessionSlice: StateCreator<
 
     const newSession: LobeAgentSession = merge(defaultAgent, agent);
 
-    const id = await sessionService.createNewSession(LobeSessionType.Agent, newSession);
+    const id = await sessionService.createSession(LobeSessionType.Agent, newSession);
     await refreshSessions();
 
     // Whether to goto  to the new session after creation, the default is to switch to
@@ -109,7 +109,7 @@ export const createSessionSlice: StateCreator<
 
     const newTitle = t('duplicateTitle', { ns: 'chat', title: title });
 
-    const newId = await sessionService.duplicateSession(id, newTitle);
+    const newId = await sessionService.cloneSession(id, newTitle);
 
     // duplicate Session Error
     if (!newId) {
@@ -122,7 +122,7 @@ export const createSessionSlice: StateCreator<
   },
 
   pinSession: async (sessionId, pinned) => {
-    await sessionService.updateSessionPinned(sessionId, pinned);
+    await sessionService.updateSession(sessionId, { pinned });
 
     await get().refreshSessions();
   },
@@ -142,7 +142,7 @@ export const createSessionSlice: StateCreator<
   },
 
   useFetchSessions: () =>
-    useClientDataSWR<ChatSessionList>(FETCH_SESSIONS_KEY, sessionService.getSessionsWithGroup, {
+    useClientDataSWR<ChatSessionList>(FETCH_SESSIONS_KEY, sessionService.getGroupedSessions, {
       onSuccess: (data) => {
         // 由于 https://github.com/lobehub/lobe-chat/pull/541 的关系
         // 只有触发了 refreshSessions 才会更新 sessions，进而触发页面 rerender

package/src/store/session/slices/sessionGroup/action.test.ts

@@ -31,7 +31,7 @@ describe('createSessionGroupSlice', () => {
 
   describe('clearSessionGroups', () => {
     it('should clear session groups and refresh sessions', async () => {
-      vi.spyOn(sessionService, 'clearSessionGroups');
+      vi.spyOn(sessionService, 'removeSessionGroups');
       const spyOnRefreshSessions = vi.spyOn(useSessionStore.getState(), 'refreshSessions');
 
       const { result } = renderHook(() => useSessionStore());
@@ -40,7 +40,7 @@ describe('createSessionGroupSlice', () => {
         await result.current.clearSessionGroups();
       });
 
-      expect(sessionService.clearSessionGroups).toHaveBeenCalled();
+      expect(sessionService.removeSessionGroups).toHaveBeenCalled();
       expect(spyOnRefreshSessions).toHaveBeenCalled();
     });
   });
@@ -66,7 +66,7 @@ describe('createSessionGroupSlice', () => {
     it('should update a session group id and refresh sessions', async () => {
       const mockSessionId = 'session-id';
       const mockGroupId = 'group-id';
-      vi.spyOn(sessionService, 'updateSessionGroupId');
+      vi.spyOn(sessionService, 'updateSession');
       const spyOnRefreshSessions = vi.spyOn(useSessionStore.getState(), 'refreshSessions');
 
       const { result } = renderHook(() => useSessionStore());
@@ -75,7 +75,9 @@ describe('createSessionGroupSlice', () => {
         await result.current.updateSessionGroupId(mockSessionId, mockGroupId);
       });
 
-      expect(sessionService.updateSessionGroupId).toHaveBeenCalledWith(mockSessionId, mockGroupId);
+      expect(sessionService.updateSession).toHaveBeenCalledWith(mockSessionId, {
+        group: mockGroupId,
+      });
       expect(spyOnRefreshSessions).toHaveBeenCalled();
     });
   });

package/src/store/session/slices/sessionGroup/action.ts

@@ -28,7 +28,7 @@ export const createSessionGroupSlice: StateCreator<
   },
 
   clearSessionGroups: async () => {
-    await sessionService.clearSessionGroups();
+    await sessionService.removeSessionGroups();
     await get().refreshSessions();
   },
 
@@ -36,8 +36,8 @@ export const createSessionGroupSlice: StateCreator<
     await sessionService.removeSessionGroup(id);
     await get().refreshSessions();
   },
-  updateSessionGroupId: async (sessionId, groupId) => {
-    await sessionService.updateSessionGroupId(sessionId, groupId);
+  updateSessionGroupId: async (sessionId, group) => {
+    await sessionService.updateSession(sessionId, { group });
 
     await get().refreshSessions();
   },