@lobehub/lobehub 2.0.0-next.26 → 2.0.0-next.27

package/src/services/session/type.ts

@@ -1,82 +0,0 @@
-/* eslint-disable typescript-sort-keys/interface */
-import type { PartialDeep } from 'type-fest';
-
-import { LobeAgentChatConfig, LobeAgentConfig } from '@/types/agent';
-import { MetaData } from '@/types/meta';
-import { BatchTaskResult } from '@/types/service';
-import {
-  ChatSessionList,
-  LobeAgentSession,
-  LobeSessionType,
-  LobeSessions,
-  SessionGroupItem,
-  SessionGroups,
-  SessionRankItem,
-  UpdateSessionParams,
-} from '@/types/session';
-
-export interface ISessionService {
-  hasSessions(): Promise<boolean>;
-  createSession(type: LobeSessionType, defaultValue: Partial<LobeAgentSession>): Promise<string>;
-
-  /**
-   * 需要废弃
-   * @deprecated
-   */
-  batchCreateSessions(importSessions: LobeSessions): Promise<any>;
-  cloneSession(id: string, newTitle: string): Promise<string | undefined>;
-
-  getGroupedSessions(): Promise<ChatSessionList>;
-
-  /**
-   * @deprecated
-   */
-  getSessionsByType(type?: 'agent' | 'group' | 'all'): Promise<LobeSessions>;
-  countSessions(params?: {
-    endDate?: string;
-    range?: [string, string];
-    startDate?: string;
-  }): Promise<number>;
-  rankSessions(limit?: number): Promise<SessionRankItem[]>;
-  searchSessions(keyword: string): Promise<LobeSessions>;
-
-  updateSession(id: string, data: Partial<UpdateSessionParams>): Promise<any>;
-
-  getSessionConfig(id: string): Promise<LobeAgentConfig>;
-  updateSessionConfig(
-    id: string,
-    config: PartialDeep<LobeAgentConfig>,
-    signal?: AbortSignal,
-  ): Promise<any>;
-
-  updateSessionMeta(id: string, meta: Partial<MetaData>, signal?: AbortSignal): Promise<any>;
-
-  updateSessionChatConfig(
-    id: string,
-    config: Partial<LobeAgentChatConfig>,
-    signal?: AbortSignal,
-  ): Promise<any>;
-
-  removeSession(id: string): Promise<any>;
-  removeAllSessions(): Promise<any>;
-
-  // ************************************** //
-  // ***********  SessionGroup  *********** //
-  // ************************************** //
-
-  createSessionGroup(name: string, sort?: number): Promise<string>;
-
-  /**
-   * 需要废弃
-   * @deprecated
-   */
-  batchCreateSessionGroups(groups: SessionGroups): Promise<BatchTaskResult>;
-
-  getSessionGroups(): Promise<SessionGroupItem[]>;
-
-  updateSessionGroup(id: string, data: Partial<SessionGroupItem>): Promise<any>;
-  updateSessionGroupOrder(sortMap: { id: string; sort: number }[]): Promise<any>;
-
-  removeSessionGroup(id: string, removeChildren?: boolean): Promise<any>;
-  removeSessionGroups(): Promise<any>;
-}

package/src/services/thread/server.ts

@@ -1,32 +0,0 @@
-import { INBOX_SESSION_ID } from '@/const/session';
-import { lambdaClient } from '@/libs/trpc/client';
-
-import { IThreadService } from './type';
-
-export class ServerService implements IThreadService {
-  getThreads: IThreadService['getThreads'] = (topicId) => {
-    return lambdaClient.thread.getThreads.query({ topicId });
-  };
-
-  createThreadWithMessage: IThreadService['createThreadWithMessage'] = async ({
-    message,
-    ...params
-  }) => {
-    return lambdaClient.thread.createThreadWithMessage.mutate({
-      ...params,
-      message: { ...message, sessionId: this.toDbSessionId(message.sessionId) },
-    });
-  };
-
-  updateThread: IThreadService['updateThread'] = async (id, data) => {
-    return lambdaClient.thread.updateThread.mutate({ id, value: data });
-  };
-
-  removeThread: IThreadService['removeThread'] = async (id) => {
-    return lambdaClient.thread.removeThread.mutate({ id });
-  };
-
-  private toDbSessionId = (sessionId: string | undefined) => {
-    return sessionId === INBOX_SESSION_ID ? null : sessionId;
-  };
-}

package/src/services/thread/type.ts

@@ -1,21 +0,0 @@
-/* eslint-disable typescript-sort-keys/interface */
-import { CreateMessageParams } from '@lobechat/types';
-
-import { CreateThreadParams, ThreadItem } from '@/types/topic';
-
-interface CreateThreadWithMessageParams extends CreateThreadParams {
-  message: CreateMessageParams;
-}
-
-export interface IThreadService {
-  getThreads(topicId: string): Promise<ThreadItem[]>;
-
-  createThreadWithMessage({
-    message,
-    ...params
-  }: CreateThreadWithMessageParams): Promise<{ messageId: string; threadId: string }>;
-
-  updateThread(id: string, data: Partial<ThreadItem>): Promise<any>;
-  //
-  removeThread(id: string): Promise<any>;
-}

package/src/services/topic/server.ts

@@ -1,57 +0,0 @@
-import { INBOX_SESSION_ID } from '@/const/session';
-import { lambdaClient } from '@/libs/trpc/client';
-import { ITopicService } from '@/services/topic/type';
-
-export class ServerService implements ITopicService {
-  createTopic: ITopicService['createTopic'] = (params) =>
-    lambdaClient.topic.createTopic.mutate({
-      ...params,
-      sessionId: this.toDbSessionId(params.sessionId),
-    });
-
-  batchCreateTopics: ITopicService['batchCreateTopics'] = (importTopics) =>
-    lambdaClient.topic.batchCreateTopics.mutate(importTopics);
-
-  cloneTopic: ITopicService['cloneTopic'] = (id, newTitle) =>
-    lambdaClient.topic.cloneTopic.mutate({ id, newTitle });
-
-  getTopics: ITopicService['getTopics'] = (params) =>
-    lambdaClient.topic.getTopics.query({
-      ...params,
-      containerId: this.toDbSessionId(params.containerId),
-    }) as any;
-
-  getAllTopics: ITopicService['getAllTopics'] = () =>
-    lambdaClient.topic.getAllTopics.query() as any;
-
-  countTopics: ITopicService['countTopics'] = async (params) => {
-    return lambdaClient.topic.countTopics.query(params);
-  };
-
-  rankTopics: ITopicService['rankTopics'] = async (limit) => {
-    return lambdaClient.topic.rankTopics.query(limit);
-  };
-
-  searchTopics: ITopicService['searchTopics'] = (keywords, sessionId) =>
-    lambdaClient.topic.searchTopics.query({
-      keywords,
-      sessionId: this.toDbSessionId(sessionId),
-    }) as any;
-
-  updateTopic: ITopicService['updateTopic'] = (id, data) =>
-    lambdaClient.topic.updateTopic.mutate({ id, value: data });
-
-  removeTopic: ITopicService['removeTopic'] = (id) => lambdaClient.topic.removeTopic.mutate({ id });
-
-  removeTopics: ITopicService['removeTopics'] = (sessionId) =>
-    lambdaClient.topic.batchDeleteBySessionId.mutate({ id: this.toDbSessionId(sessionId) });
-
-  batchRemoveTopics: ITopicService['batchRemoveTopics'] = (topics) =>
-    lambdaClient.topic.batchDelete.mutate({ ids: topics });
-
-  removeAllTopic: ITopicService['removeAllTopic'] = () =>
-    lambdaClient.topic.removeAllTopics.mutate();
-
-  private toDbSessionId = (sessionId?: string | null) =>
-    sessionId === INBOX_SESSION_ID ? null : sessionId;
-}

package/src/services/topic/type.ts

@@ -1,40 +0,0 @@
-/* eslint-disable typescript-sort-keys/interface */
-import { BatchTaskResult } from '@/types/service';
-import { ChatTopic, TopicRankItem } from '@/types/topic';
-
-export interface CreateTopicParams {
-  favorite?: boolean;
-  groupId?: string | null;
-  messages?: string[];
-  sessionId?: string | null;
-  title: string;
-}
-
-export interface QueryTopicParams {
-  current?: number;
-  containerId?: string | null; // sessionId or groupId
-  pageSize?: number;
-}
-
-export interface ITopicService {
-  createTopic(params: CreateTopicParams): Promise<string>;
-  batchCreateTopics(importTopics: ChatTopic[]): Promise<BatchTaskResult>;
-  cloneTopic(id: string, newTitle?: string): Promise<string>;
-
-  getTopics(params: QueryTopicParams): Promise<ChatTopic[]>;
-  getAllTopics(): Promise<ChatTopic[]>;
-  countTopics(params?: {
-    endDate?: string;
-    range?: [string, string];
-    startDate?: string;
-  }): Promise<number>;
-  rankTopics(limit?: number): Promise<TopicRankItem[]>;
-  searchTopics(keyword: string, sessionId?: string, groupId?: string): Promise<ChatTopic[]>;
-
-  updateTopic(id: string, data: Partial<ChatTopic>): Promise<any>;
-
-  removeTopic(id: string): Promise<any>;
-  removeTopics(sessionId: string): Promise<any>;
-  batchRemoveTopics(topics: string[]): Promise<any>;
-  removeAllTopic(): Promise<any>;
-}

package/src/services/user/server.test.ts

@@ -1,149 +0,0 @@
-import type { PartialDeep } from 'type-fest';
-import { describe, expect, it, vi } from 'vitest';
-
-import { lambdaClient } from '@/libs/trpc/client';
-import { UserInitializationState, UserPreference } from '@/types/user';
-import { UserSettings } from '@/types/user/settings';
-
-import { ServerService } from './server';
-
-vi.mock('@/libs/trpc/client', () => ({
-  lambdaClient: {
-    user: {
-      getUserRegistrationDuration: {
-        query: vi.fn(),
-      },
-      getUserState: {
-        query: vi.fn(),
-      },
-      getUserSSOProviders: {
-        query: vi.fn(),
-      },
-      unlinkSSOProvider: {
-        mutate: vi.fn(),
-      },
-      makeUserOnboarded: {
-        mutate: vi.fn(),
-      },
-      updatePreference: {
-        mutate: vi.fn(),
-      },
-      updateGuide: {
-        mutate: vi.fn(),
-      },
-      updateSettings: {
-        mutate: vi.fn(),
-      },
-      resetSettings: {
-        mutate: vi.fn(),
-      },
-    },
-  },
-}));
-
-describe('ServerService', () => {
-  const service = new ServerService();
-
-  it('should get user registration duration', async () => {
-    const mockData = {
-      createdAt: '2023-01-01',
-      duration: 100,
-      updatedAt: '2023-01-02',
-    };
-    vi.mocked(lambdaClient.user.getUserRegistrationDuration.query).mockResolvedValue(mockData);
-
-    const result = await service.getUserRegistrationDuration();
-    expect(result).toEqual(mockData);
-  });
-
-  it('should get user state', async () => {
-    const mockState: UserInitializationState = {
-      isOnboard: true,
-      preference: {
-        telemetry: true,
-      },
-      settings: {},
-    };
-    vi.mocked(lambdaClient.user.getUserState.query).mockResolvedValue(mockState);
-
-    const result = await service.getUserState();
-    expect(result).toEqual(mockState);
-  });
-
-  it('should get user SSO providers', async () => {
-    const mockProviders = [
-      {
-        provider: 'google',
-        providerAccountId: '123',
-        userId: 'user1',
-        type: 'oauth' as const,
-        access_token: 'token',
-        token_type: 'bearer' as const,
-        expires_at: 123,
-        scope: 'email profile',
-      },
-    ];
-    vi.mocked(lambdaClient.user.getUserSSOProviders.query).mockResolvedValue(mockProviders);
-
-    const result = await service.getUserSSOProviders();
-    expect(result).toEqual(mockProviders);
-  });
-
-  it('should unlink SSO provider', async () => {
-    const provider = 'google';
-    const providerAccountId = '123';
-    await service.unlinkSSOProvider(provider, providerAccountId);
-
-    expect(lambdaClient.user.unlinkSSOProvider.mutate).toHaveBeenCalledWith({
-      provider,
-      providerAccountId,
-    });
-  });
-
-  it('should make user onboarded', async () => {
-    await service.makeUserOnboarded();
-    expect(lambdaClient.user.makeUserOnboarded.mutate).toHaveBeenCalled();
-  });
-
-  it('should update user preference', async () => {
-    const preference: Partial<UserPreference> = {
-      telemetry: true,
-      useCmdEnterToSend: true,
-    };
-    await service.updatePreference(preference);
-    expect(lambdaClient.user.updatePreference.mutate).toHaveBeenCalledWith(preference);
-  });
-
-  it('should update user guide', async () => {
-    const guide = {
-      moveSettingsToAvatar: true,
-      topic: false,
-      uploadFileInKnowledgeBase: true,
-    };
-    await service.updateGuide(guide);
-    expect(lambdaClient.user.updateGuide.mutate).toHaveBeenCalledWith(guide);
-  });
-
-  it('should update user settings', async () => {
-    const settings: PartialDeep<UserSettings> = {
-      defaultAgent: {
-        config: {
-          model: 'gpt-4',
-          provider: 'openai',
-        },
-        meta: {
-          avatar: 'avatar',
-          description: 'test agent',
-        },
-      },
-    };
-    const signal = new AbortController().signal;
-    await service.updateUserSettings(settings, signal);
-    expect(lambdaClient.user.updateSettings.mutate).toHaveBeenCalledWith(settings, { signal });
-  });
-
-  it('should reset user settings', async () => {
-    await service.resetUserSettings();
-    expect(lambdaClient.user.resetSettings.mutate).toHaveBeenCalled();
-  });
-});

package/src/services/user/server.ts

@@ -1,47 +0,0 @@
-import { lambdaClient } from '@/libs/trpc/client';
-import { IUserService } from '@/services/user/type';
-
-export class ServerService implements IUserService {
-  getUserRegistrationDuration: IUserService['getUserRegistrationDuration'] = async () => {
-    return lambdaClient.user.getUserRegistrationDuration.query();
-  };
-
-  getUserState: IUserService['getUserState'] = async () => {
-    return lambdaClient.user.getUserState.query();
-  };
-
-  getUserSSOProviders: IUserService['getUserSSOProviders'] = async () => {
-    return lambdaClient.user.getUserSSOProviders.query();
-  };
-
-  unlinkSSOProvider: IUserService['unlinkSSOProvider'] = async (
-    provider: string,
-    providerAccountId: string,
-  ) => {
-    return lambdaClient.user.unlinkSSOProvider.mutate({ provider, providerAccountId });
-  };
-
-  makeUserOnboarded = async () => {
-    return lambdaClient.user.makeUserOnboarded.mutate();
-  };
-
-  updateAvatar: IUserService['updateAvatar'] = async (avatar) => {
-    return lambdaClient.user.updateAvatar.mutate(avatar);
-  };
-
-  updatePreference: IUserService['updatePreference'] = async (preference) => {
-    return lambdaClient.user.updatePreference.mutate(preference);
-  };
-
-  updateGuide: IUserService['updateGuide'] = async (guide) => {
-    return lambdaClient.user.updateGuide.mutate(guide);
-  };
-
-  updateUserSettings: IUserService['updateUserSettings'] = async (value, signal) => {
-    return lambdaClient.user.updateSettings.mutate(value, { signal });
-  };
-
-  resetUserSettings: IUserService['resetUserSettings'] = async () => {
-    return lambdaClient.user.resetSettings.mutate();
-  };
-}

package/src/services/user/type.ts

@@ -1,21 +0,0 @@
-import type { AdapterAccount } from 'next-auth/adapters';
-import type { PartialDeep } from 'type-fest';
-
-import { UserGuide, UserInitializationState, UserPreference } from '@/types/user';
-import { UserSettings } from '@/types/user/settings';
-
-export interface IUserService {
-  getUserRegistrationDuration: () => Promise<{
-    createdAt: string;
-    duration: number;
-    updatedAt: string;
-  }>;
-  getUserSSOProviders: () => Promise<AdapterAccount[]>;
-  getUserState: () => Promise<UserInitializationState>;
-  resetUserSettings: () => Promise<any>;
-  unlinkSSOProvider: (provider: string, providerAccountId: string) => Promise<any>;
-  updateAvatar: (avatar: string) => Promise<any>;
-  updateGuide: (guide: Partial<UserGuide>) => Promise<any>;
-  updatePreference: (preference: Partial<UserPreference>) => Promise<any>;
-  updateUserSettings: (value: PartialDeep<UserSettings>, signal?: AbortSignal) => Promise<any>;
-}