@lobehub/lobehub 2.0.0-next.12 → 2.0.0-next.14

package/src/services/session/_deprecated.test.ts

@@ -1,440 +0,0 @@
-import { Mock, beforeEach, describe, expect, it, vi } from 'vitest';
-
-import { SessionModel } from '@/database/_deprecated/models/session';
-import { SessionGroupModel } from '@/database/_deprecated/models/sessionGroup';
-import { LobeAgentConfig } from '@/types/agent';
-import { LobeAgentSession, LobeSessionType, SessionGroups } from '@/types/session';
-
-import { ClientService } from './_deprecated';
-
-const sessionService = new ClientService();
-
-// Mock the SessionModel
-vi.mock('@/database/_deprecated/models/session', () => {
-  return {
-    SessionModel: {
-      create: vi.fn(),
-      query: vi.fn(),
-      delete: vi.fn(),
-      clearTable: vi.fn(),
-      update: vi.fn(),
-      count: vi.fn(),
-      batchCreate: vi.fn(),
-      findById: vi.fn(),
-      isEmpty: vi.fn(),
-      queryByKeyword: vi.fn(),
-      updateConfig: vi.fn(),
-      queryByGroupIds: vi.fn(),
-      updatePinned: vi.fn(),
-      duplicate: vi.fn(),
-      queryWithGroups: vi.fn(),
-    },
-  };
-});
-
-// Mock the SessionGroupModel
-vi.mock('@/database/_deprecated/models/sessionGroup', () => {
-  return {
-    SessionGroupModel: {
-      create: vi.fn(),
-      query: vi.fn(),
-      delete: vi.fn(),
-      clear: vi.fn(),
-      update: vi.fn(),
-      batchCreate: vi.fn(),
-      isEmpty: vi.fn(),
-      updateOrder: vi.fn(),
-      queryByKeyword: vi.fn(),
-      updateConfig: vi.fn(),
-      queryByGroupIds: vi.fn(),
-    },
-  };
-});
-
-describe('SessionService', () => {
-  const mockSessionId = 'mock-session-id';
-  const mockSession = {
-    id: mockSessionId,
-    type: 'agent',
-    meta: { title: 'Mock Session' },
-  } as LobeAgentSession;
-  const mockSessions = [mockSession];
-
-  beforeEach(() => {
-    // Reset all mocks before running each test case
-    vi.resetAllMocks();
-  });
-
-  describe('createSession', () => {
-    it('should create a new session and return its id', async () => {
-      // Setup
-      const sessionType = LobeSessionType.Agent;
-      const defaultValue = { meta: { title: 'New Session' } } as Partial<LobeAgentSession>;
-      (SessionModel.create as Mock).mockResolvedValue(mockSession);
-
-      // Execute
-      const sessionId = await sessionService.createSession(sessionType, defaultValue);
-
-      // Assert
-      expect(SessionModel.create).toHaveBeenCalledWith(sessionType, defaultValue);
-      expect(sessionId).toBe(mockSessionId);
-    });
-
-    it('should throw an error if session creation fails', async () => {
-      // Setup
-      const sessionType = LobeSessionType.Agent;
-      const defaultValue = { meta: { title: 'New Session' } } as Partial<LobeAgentSession>;
-      (SessionModel.create as Mock).mockResolvedValue(null);
-
-      // Execute & Assert
-      await expect(sessionService.createSession(sessionType, defaultValue)).rejects.toThrow(
-        'session create Error',
-      );
-    });
-  });
-
-  describe('batchCreateSessions', () => {
-    it('should batch create sessions', async () => {
-      // Setup
-      (SessionModel.batchCreate as Mock).mockResolvedValue(mockSessions);
-
-      // Execute
-      const result = await sessionService.batchCreateSessions(mockSessions);
-
-      // Assert
-      expect(SessionModel.batchCreate).toHaveBeenCalledWith(mockSessions);
-      expect(result).toBe(mockSessions);
-    });
-  });
-
-  describe('getSessionsByType', () => {
-    it('should retrieve sessions with their group ids', async () => {
-      // Setup
-      (SessionModel.query as Mock).mockResolvedValue(mockSessions);
-
-      // Execute
-      const sessions = await sessionService.getSessionsByType();
-
-      // Assert
-      expect(SessionModel.query).toHaveBeenCalled();
-      expect(sessions).toBe(mockSessions);
-    });
-
-    it('should retrieve all agent sessions', async () => {
-      // Setup
-      // Assuming that SessionModel.query has been modified to accept filters
-      const agentSessions = mockSessions.filter((session) => session.type === 'agent');
-      (SessionModel.query as Mock).mockResolvedValue(agentSessions);
-
-      // Execute
-      const result = await sessionService.getSessionsByType('agent');
-
-      // Assert
-      // Assuming that SessionModel.query would be called with a filter for agents
-      expect(SessionModel.query).toHaveBeenCalled(); // Add filter argument if applicable
-      expect(result).toBe(agentSessions);
-    });
-  });
-
-  describe('removeSession', () => {
-    it('should remove a session by its id', async () => {
-      // Setup
-      (SessionModel.delete as Mock).mockResolvedValue(true);
-
-      // Execute
-      const result = await sessionService.removeSession(mockSessionId);
-
-      // Assert
-      expect(SessionModel.delete).toHaveBeenCalledWith(mockSessionId);
-      expect(result).toBe(true);
-    });
-  });
-
-  describe('removeAllSessions', () => {
-    it('should clear all sessions from the table', async () => {
-      // Setup
-      (SessionModel.clearTable as Mock).mockResolvedValue(true);
-
-      // Execute
-      const result = await sessionService.removeAllSessions();
-
-      // Assert
-      expect(SessionModel.clearTable).toHaveBeenCalled();
-      expect(result).toBe(true);
-    });
-  });
-
-  describe('updateSession', () => {
-    it('should update the group of a session', async () => {
-      // Setup
-      const groupId = 'new-group';
-      (SessionModel.update as Mock).mockResolvedValue({ ...mockSession, group: groupId });
-
-      // Execute
-      const result = await sessionService.updateSession(mockSessionId, { group: groupId });
-
-      // Assert
-      expect(SessionModel.update).toHaveBeenCalledWith(mockSessionId, { group: groupId });
-      expect(result).toEqual({ ...mockSession, group: groupId });
-    });
-
-    it('should update the meta of a session', async () => {
-      // Setup
-      const newMeta = { description: 'Updated description' };
-      (SessionModel.update as Mock).mockResolvedValue({ ...mockSession, meta: newMeta });
-
-      // Execute
-      const result = await sessionService.updateSession(mockSessionId, { meta: newMeta });
-
-      // Assert
-      expect(SessionModel.update).toHaveBeenCalledWith(mockSessionId, { meta: newMeta });
-      expect(result).toEqual({ ...mockSession, meta: newMeta });
-    });
-
-    it('should update the pinned status of a session', async () => {
-      // Setup
-      const pinned = true;
-
-      // Execute
-      await sessionService.updateSession(mockSessionId, { pinned });
-
-      // Assert
-      expect(SessionModel.update).toHaveBeenCalledWith(mockSessionId, { pinned: 1 });
-    });
-  });
-
-  describe('updateSessionConfig', () => {
-    it('should update the config of a session', async () => {
-      // Setup
-      const newConfig = { model: 'abc' } as LobeAgentConfig;
-      (SessionModel.updateConfig as Mock).mockResolvedValue({ ...mockSession, config: newConfig });
-
-      // Execute
-      const result = await sessionService.updateSessionConfig(mockSessionId, newConfig);
-
-      // Assert
-      expect(SessionModel.updateConfig).toHaveBeenCalledWith(mockSessionId, newConfig);
-      expect(result).toEqual({ ...mockSession, config: newConfig });
-    });
-  });
-
-  describe('countSessions', () => {
-    it('should return false if no sessions exist', async () => {
-      // Setup
-      (SessionModel.count as Mock).mockResolvedValue(0);
-
-      // Execute
-      const result = await sessionService.countSessions();
-
-      // Assert
-      expect(SessionModel.count).toHaveBeenCalled();
-      expect(result).toBe(0);
-    });
-
-    it('should return true if sessions exist', async () => {
-      // Setup
-      (SessionModel.count as Mock).mockResolvedValue(1);
-
-      // Execute
-      const result = await sessionService.countSessions();
-
-      // Assert
-      expect(SessionModel.count).toHaveBeenCalled();
-      expect(result).toBe(1);
-    });
-  });
-
-  describe('hasSessions', () => {
-    it('should return false if no sessions exist', async () => {
-      // Setup
-      (SessionModel.count as Mock).mockResolvedValue(0);
-
-      // Execute
-      const result = await sessionService.hasSessions();
-
-      // Assert
-      expect(SessionModel.count).toHaveBeenCalled();
-      expect(result).toBe(false);
-    });
-
-    it('should return true if sessions exist', async () => {
-      // Setup
-      (SessionModel.count as Mock).mockResolvedValue(1);
-
-      // Execute
-      const result = await sessionService.hasSessions();
-
-      // Assert
-      expect(SessionModel.count).toHaveBeenCalled();
-      expect(result).toBe(true);
-    });
-  });
-
-  describe('searchSessions', () => {
-    it('should return sessions that match the keyword', async () => {
-      // Setup
-      const keyword = 'search';
-      (SessionModel.queryByKeyword as Mock).mockResolvedValue(mockSessions);
-
-      // Execute
-      const result = await sessionService.searchSessions(keyword);
-
-      // Assert
-      expect(SessionModel.queryByKeyword).toHaveBeenCalledWith(keyword);
-      expect(result).toBe(mockSessions);
-    });
-  });
-
-  describe('cloneSession', () => {
-    it('should duplicate a session and return its id', async () => {
-      // Setup
-      const newTitle = 'Duplicated Session';
-      (SessionModel.duplicate as Mock).mockResolvedValue({
-        ...mockSession,
-        id: 'duplicated-session-id',
-      });
-
-      // Execute
-      const duplicatedSessionId = await sessionService.cloneSession(mockSessionId, newTitle);
-
-      // Assert
-      expect(SessionModel.duplicate).toHaveBeenCalledWith(mockSessionId, newTitle);
-      expect(duplicatedSessionId).toBe('duplicated-session-id');
-    });
-  });
-
-  describe('getGroupedSessions', () => {
-    it('should retrieve sessions with their group', async () => {
-      // Setup
-      (SessionModel.queryWithGroups as Mock).mockResolvedValue(mockSessions);
-
-      // Execute
-      const sessionsWithGroup = await sessionService.getGroupedSessions();
-
-      // Assert
-      expect(SessionModel.queryWithGroups).toHaveBeenCalled();
-      expect(sessionsWithGroup).toBe(mockSessions);
-    });
-  });
-
-  // SessionGroup related tests
-  describe('createSessionGroup', () => {
-    it('should create a new session group and return its id', async () => {
-      // Setup
-      const groupName = 'New Group';
-      const sort = 1;
-      (SessionGroupModel.create as Mock).mockResolvedValue({
-        id: 'new-group-id',
-        name: groupName,
-        sort,
-      });
-
-      // Execute
-      const groupId = await sessionService.createSessionGroup(groupName, sort);
-
-      // Assert
-      expect(SessionGroupModel.create).toHaveBeenCalledWith(groupName, sort);
-      expect(groupId).toBe('new-group-id');
-    });
-  });
-
-  describe('batchCreateSessionGroups', () => {
-    it('should batch create session groups', async () => {
-      // Setup
-      const groups = [
-        { id: 'group-1', name: 'Group 1', sort: 1 },
-        { id: 'group-2', name: 'Group 2', sort: 2 },
-      ] as SessionGroups;
-
-      (SessionGroupModel.batchCreate as Mock).mockResolvedValue(groups);
-
-      // Execute
-      const result = await sessionService.batchCreateSessionGroups(groups);
-
-      // Assert
-      expect(SessionGroupModel.batchCreate).toHaveBeenCalledWith(groups);
-      expect(result).toBe(groups);
-    });
-  });
-
-  describe('removeSessionGroup', () => {
-    it('should remove a session group by its id', async () => {
-      // Setup
-      const removeChildren = true;
-      (SessionGroupModel.delete as Mock).mockResolvedValue(true);
-
-      // Execute
-      const result = await sessionService.removeSessionGroup('group-id', removeChildren);
-
-      // Assert
-      expect(SessionGroupModel.delete).toHaveBeenCalledWith('group-id', removeChildren);
-      expect(result).toBe(true);
-    });
-  });
-
-  describe('clearSessionGroups', () => {
-    it('should clear all session groups', async () => {
-      // Setup
-      (SessionGroupModel.clear as Mock).mockResolvedValue(true);
-
-      // Execute
-      const result = await sessionService.removeSessionGroups();
-
-      // Assert
-      expect(SessionGroupModel.clear).toHaveBeenCalled();
-      expect(result).toBe(true);
-    });
-  });
-
-  describe('getSessionGroups', () => {
-    it('should retrieve all session groups', async () => {
-      // Setup
-      const groups = [
-        { id: 'group-1', name: 'Group 1', sort: 1 },
-        { id: 'group-2', name: 'Group 2', sort: 2 },
-      ];
-      (SessionGroupModel.query as Mock).mockResolvedValue(groups);
-
-      // Execute
-      const result = await sessionService.getSessionGroups();
-
-      // Assert
-      expect(SessionGroupModel.query).toHaveBeenCalled();
-      expect(result).toBe(groups);
-    });
-  });
-
-  describe('updateSessionGroup', () => {
-    it('should update a session group', async () => {
-      // Setup
-      const groupId = 'group-1';
-      const data = { name: 'Updated Group', sort: 2 };
-      (SessionGroupModel.update as Mock).mockResolvedValue({ id: groupId, ...data });
-
-      // Execute
-      const result = await sessionService.updateSessionGroup(groupId, data);
-
-      // Assert
-      expect(SessionGroupModel.update).toHaveBeenCalledWith(groupId, data);
-      expect(result).toEqual({ id: groupId, ...data });
-    });
-  });
-
-  describe('updateSessionGroupOrder', () => {
-    it('should update the order of session groups', async () => {
-      // Setup
-      const sortMap = [
-        { id: 'group-1', sort: 2 },
-        { id: 'group-2', sort: 1 },
-      ];
-      (SessionGroupModel.updateOrder as Mock).mockResolvedValue(true);
-
-      // Execute
-      const result = await sessionService.updateSessionGroupOrder(sortMap);
-
-      // Assert
-      expect(SessionGroupModel.updateOrder).toHaveBeenCalledWith(sortMap);
-      expect(result).toBe(true);
-    });
-  });
-});

package/src/services/session/_deprecated.ts

@@ -1,190 +0,0 @@
-import type { PartialDeep } from 'type-fest';
-
-import { INBOX_SESSION_ID } from '@/const/session';
-import { DEFAULT_AGENT_CONFIG } from '@/const/settings';
-import { SessionModel } from '@/database/_deprecated/models/session';
-import { SessionGroupModel } from '@/database/_deprecated/models/sessionGroup';
-import { UserModel } from '@/database/_deprecated/models/user';
-import { useUserStore } from '@/store/user';
-import { LobeAgentChatConfig, LobeAgentConfig } from '@/types/agent';
-import { MetaData } from '@/types/meta';
-import {
-  ChatSessionList,
-  LobeAgentSession,
-  LobeSessionType,
-  LobeSessions,
-  SessionGroupItem,
-  SessionGroups,
-} from '@/types/session';
-import { merge } from '@/utils/merge';
-
-import { ISessionService } from './type';
-
-export class ClientService implements ISessionService {
-  async createSession(
-    type: LobeSessionType,
-    defaultValue: Partial<LobeAgentSession>,
-  ): Promise<string> {
-    const item = await SessionModel.create(type, defaultValue);
-    if (!item) {
-      throw new Error('session create Error');
-    }
-    return item.id;
-  }
-
-  async batchCreateSessions(importSessions: LobeSessions) {
-    // @ts-ignore
-    return SessionModel.batchCreate(importSessions);
-  }
-
-  async cloneSession(id: string, newTitle: string): Promise<string | undefined> {
-    const res = await SessionModel.duplicate(id, newTitle);
-
-    if (res) return res?.id;
-  }
-
-  async getGroupedSessions(): Promise<ChatSessionList> {
-    return SessionModel.queryWithGroups();
-  }
-
-  async getSessionConfig(id: string): Promise<LobeAgentConfig> {
-    if (!id || id === INBOX_SESSION_ID) {
-      return (await UserModel.getAgentConfig()) || DEFAULT_AGENT_CONFIG;
-    }
-
-    const res = await SessionModel.findById(id);
-
-    if (!res) throw new Error('Session not found');
-
-    return res.config as LobeAgentConfig;
-  }
-
-  async getSessionsByType(type: 'agent' | 'group' | 'all' = 'all'): Promise<LobeSessions> {
-    switch (type) {
-      // TODO: add a filter to get only agents or agents
-      case 'group': {
-        return SessionModel.query();
-      }
-      case 'agent': {
-        return SessionModel.query();
-      }
-
-      case 'all': {
-        return SessionModel.query();
-      }
-    }
-  }
-
-  async getAllAgents(): Promise<LobeSessions> {
-    // TODO: add a filter to get only agents
-    return SessionModel.query();
-  }
-
-  async countSessions() {
-    return SessionModel.count();
-  }
-
-  // @ts-ignore
-  async rankSessions() {
-    throw new Error('Method not implemented.');
-  }
-
-  async hasSessions() {
-    return (await this.countSessions()) !== 0;
-  }
-
-  async searchSessions(keyword: string) {
-    return SessionModel.queryByKeyword(keyword);
-  }
-
-  async updateSession(
-    id: string,
-    data: Partial<Pick<LobeAgentSession, 'group' | 'meta' | 'pinned'>>,
-  ) {
-    const pinned = typeof data.pinned === 'boolean' ? (data.pinned ? 1 : 0) : undefined;
-    const prev = await SessionModel.findById(id);
-
-    return SessionModel.update(id, merge(prev, { ...data, pinned }));
-  }
-
-  async updateSessionConfig(
-    activeId: string,
-    config: PartialDeep<LobeAgentConfig>,
-    // eslint-disable-next-line @typescript-eslint/no-unused-vars
-    _?: AbortSignal,
-  ) {
-    // TODO: 需要删除这部分处理逻辑
-    // 后续直接给用户创建一个 inbox 的 session
-    if (activeId === INBOX_SESSION_ID) {
-      return useUserStore.getState().updateDefaultAgent({ config });
-    }
-
-    return SessionModel.updateConfig(activeId, config);
-  }
-
-  async updateSessionMeta(
-    activeId: string,
-    meta: Partial<MetaData>,
-    // eslint-disable-next-line @typescript-eslint/no-unused-vars
-    _?: AbortSignal,
-  ) {
-    // inbox 不允许修改 meta
-    if (activeId === INBOX_SESSION_ID) return;
-
-    return SessionModel.update(activeId, { meta });
-  }
-
-  async updateSessionChatConfig(
-    activeId: string,
-    config: PartialDeep<LobeAgentChatConfig>,
-    // eslint-disable-next-line @typescript-eslint/no-unused-vars
-    _?: AbortSignal,
-  ) {
-    return this.updateSessionConfig(activeId, { chatConfig: config });
-  }
-
-  async removeSession(id: string) {
-    return SessionModel.delete(id);
-  }
-
-  async removeAllSessions() {
-    return SessionModel.clearTable();
-  }
-
-  // ************************************** //
-  // ***********  SessionGroup  *********** //
-  // ************************************** //
-
-  async createSessionGroup(name: string, sort?: number) {
-    const item = await SessionGroupModel.create(name, sort);
-    if (!item) {
-      throw new Error('session group create Error');
-    }
-
-    return item.id;
-  }
-
-  async batchCreateSessionGroups(groups: SessionGroups) {
-    return SessionGroupModel.batchCreate(groups);
-  }
-
-  async removeSessionGroup(id: string, removeChildren?: boolean) {
-    return SessionGroupModel.delete(id, removeChildren);
-  }
-
-  async updateSessionGroup(id: string, data: Partial<SessionGroupItem>) {
-    return SessionGroupModel.update(id, data as any);
-  }
-
-  async updateSessionGroupOrder(sortMap: { id: string; sort: number }[]) {
-    return SessionGroupModel.updateOrder(sortMap);
-  }
-
-  async getSessionGroups(): Promise<SessionGroupItem[]> {
-    return SessionGroupModel.query();
-  }
-
-  async removeSessionGroups() {
-    return SessionGroupModel.clear();
-  }
-}