@lobehub/chat 1.37.0 → 1.37.2

package/src/services/session/pglite.ts

@@ -1,184 +0,0 @@
-import { DeepPartial } from 'utility-types';
-
-import { INBOX_SESSION_ID } from '@/const/session';
-import { clientDB } from '@/database/client/db';
-import { AgentItem } from '@/database/schemas';
-import { SessionModel } from '@/database/server/models/session';
-import { SessionGroupModel } from '@/database/server/models/sessionGroup';
-import { BaseClientService } from '@/services/baseClientService';
-import { LobeAgentChatConfig, LobeAgentConfig } from '@/types/agent';
-import { MetaData } from '@/types/meta';
-import {
-  ChatSessionList,
-  LobeAgentSession,
-  LobeSessionType,
-  LobeSessions,
-  SessionGroupItem,
-  SessionGroups,
-  UpdateSessionParams,
-} from '@/types/session';
-
-import { ISessionService } from './type';
-
-export class ClientService extends BaseClientService implements ISessionService {
-  private get sessionModel(): SessionModel {
-    return new SessionModel(clientDB as any, this.userId);
-  }
-
-  private get sessionGroupModel(): SessionGroupModel {
-    return new SessionGroupModel(clientDB as any, this.userId);
-  }
-
-  async createSession(type: LobeSessionType, data: Partial<LobeAgentSession>): Promise<string> {
-    const { config, group, meta, ...session } = data;
-
-    const item = await this.sessionModel.create({
-      config: { ...config, ...meta } as any,
-      session: { ...session, groupId: group },
-      type,
-    });
-    if (!item) {
-      throw new Error('session create Error');
-    }
-    return item.id;
-  }
-
-  async batchCreateSessions(importSessions: LobeSessions) {
-    // @ts-ignore
-    return this.sessionModel.batchCreate(importSessions);
-  }
-
-  async cloneSession(id: string, newTitle: string): Promise<string | undefined> {
-    const res = await this.sessionModel.duplicate(id, newTitle);
-
-    if (res) return res?.id;
-  }
-
-  async getGroupedSessions(): Promise<ChatSessionList> {
-    return this.sessionModel.queryWithGroups();
-  }
-
-  async getSessionConfig(id: string): Promise<LobeAgentConfig> {
-    const res = await this.sessionModel.findByIdOrSlug(id);
-
-    if (!res) throw new Error('Session not found');
-
-    return res.agent 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': {
-        // @ts-ignore
-        return this.sessionModel.query();
-      }
-      case 'agent': {
-        // @ts-ignore
-        return this.sessionModel.query();
-      }
-
-      case 'all': {
-        // @ts-ignore
-        return this.sessionModel.query();
-      }
-    }
-  }
-
-  async countSessions() {
-    return this.sessionModel.count();
-  }
-
-  async searchSessions(keyword: string) {
-    return this.sessionModel.queryByKeyword(keyword);
-  }
-
-  async updateSession(id: string, value: Partial<UpdateSessionParams>) {
-    return this.sessionModel.update(id, {
-      ...value,
-      groupId: value.group === 'default' ? null : value.group,
-    });
-  }
-
-  async updateSessionConfig(
-    activeId: string,
-    config: DeepPartial<LobeAgentConfig>,
-    // eslint-disable-next-line @typescript-eslint/no-unused-vars
-    _?: AbortSignal,
-  ) {
-    const session = await this.sessionModel.findByIdOrSlug(activeId);
-    if (!session || !config) return;
-
-    return this.sessionModel.updateConfig(session.agent.id, config as AgentItem);
-  }
-
-  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 this.sessionModel.update(activeId, meta);
-  }
-
-  async updateSessionChatConfig(
-    activeId: string,
-    config: DeepPartial<LobeAgentChatConfig>,
-    // eslint-disable-next-line @typescript-eslint/no-unused-vars
-    _?: AbortSignal,
-  ) {
-    return this.updateSessionConfig(activeId, { chatConfig: config });
-  }
-
-  async removeSession(id: string) {
-    return this.sessionModel.delete(id);
-  }
-
-  async removeAllSessions() {
-    return this.sessionModel.deleteAll();
-  }
-
-  // ************************************** //
-  // ***********  SessionGroup  *********** //
-  // ************************************** //
-
-  async createSessionGroup(name: string, sort?: number) {
-    const item = await this.sessionGroupModel.create({ name, sort });
-    if (!item) {
-      throw new Error('session group create Error');
-    }
-
-    return item.id;
-  }
-
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async batchCreateSessionGroups(_groups: SessionGroups) {
-    return { added: 0, ids: [], skips: [], success: true };
-  }
-
-  async removeSessionGroup(id: string) {
-    return await this.sessionGroupModel.delete(id);
-  }
-
-  async updateSessionGroup(id: string, data: Partial<SessionGroupItem>) {
-    return this.sessionGroupModel.update(id, data);
-  }
-
-  async updateSessionGroupOrder(sortMap: { id: string; sort: number }[]) {
-    return this.sessionGroupModel.updateOrder(sortMap);
-  }
-
-  async getSessionGroups(): Promise<SessionGroupItem[]> {
-    return this.sessionGroupModel.query();
-  }
-
-  async removeSessionGroups() {
-    return this.sessionGroupModel.deleteAll();
-  }
-}

package/src/services/topic/pglite.ts

@@ -1,85 +0,0 @@
-import { INBOX_SESSION_ID } from '@/const/session';
-import { clientDB } from '@/database/client/db';
-import { TopicModel } from '@/database/server/models/topic';
-import { BaseClientService } from '@/services/baseClientService';
-import { ChatTopic } from '@/types/topic';
-
-import { CreateTopicParams, ITopicService, QueryTopicParams } from './type';
-
-export class ClientService extends BaseClientService implements ITopicService {
-  private get topicModel(): TopicModel {
-    return new TopicModel(clientDB as any, this.userId);
-  }
-
-  async createTopic(params: CreateTopicParams): Promise<string> {
-    const item = await this.topicModel.create({
-      ...params,
-      sessionId: this.toDbSessionId(params.sessionId),
-    } as any);
-
-    if (!item) {
-      throw new Error('topic create Error');
-    }
-
-    return item.id;
-  }
-
-  async batchCreateTopics(importTopics: ChatTopic[]) {
-    const data = await this.topicModel.batchCreate(importTopics as any);
-
-    return { added: data.length, ids: [], skips: [], success: true };
-  }
-
-  async cloneTopic(id: string, newTitle?: string) {
-    const data = await this.topicModel.duplicate(id, newTitle);
-    return data.topic.id;
-  }
-
-  async getTopics(params: QueryTopicParams) {
-    const data = await this.topicModel.query({
-      ...params,
-      sessionId: this.toDbSessionId(params.sessionId),
-    });
-    return data as unknown as Promise<ChatTopic[]>;
-  }
-
-  async searchTopics(keyword: string, sessionId?: string) {
-    const data = await this.topicModel.queryByKeyword(keyword, this.toDbSessionId(sessionId));
-
-    return data as unknown as Promise<ChatTopic[]>;
-  }
-
-  async getAllTopics() {
-    const data = await this.topicModel.queryAll();
-
-    return data as unknown as Promise<ChatTopic[]>;
-  }
-
-  async countTopics() {
-    return this.topicModel.count();
-  }
-
-  async updateTopic(id: string, data: Partial<ChatTopic>) {
-    return this.topicModel.update(id, data as any);
-  }
-
-  async removeTopic(id: string) {
-    return this.topicModel.delete(id);
-  }
-
-  async removeTopics(sessionId: string) {
-    return this.topicModel.batchDeleteBySessionId(this.toDbSessionId(sessionId));
-  }
-
-  async batchRemoveTopics(topics: string[]) {
-    return this.topicModel.batchDelete(topics);
-  }
-
-  async removeAllTopic() {
-    return this.topicModel.deleteAll();
-  }
-
-  private toDbSessionId(sessionId?: string | null) {
-    return sessionId === INBOX_SESSION_ID ? null : sessionId;
-  }
-}

package/src/services/user/pglite.ts

@@ -1,92 +0,0 @@
-import { DeepPartial } from 'utility-types';
-
-import { clientDB } from '@/database/client/db';
-import { users } from '@/database/schemas';
-import { MessageModel } from '@/database/server/models/message';
-import { SessionModel } from '@/database/server/models/session';
-import { UserModel } from '@/database/server/models/user';
-import { BaseClientService } from '@/services/baseClientService';
-import { UserGuide, UserInitializationState, UserPreference } from '@/types/user';
-import { UserSettings } from '@/types/user/settings';
-import { AsyncLocalStorage } from '@/utils/localStorage';
-
-import { IUserService } from './type';
-
-export class ClientService extends BaseClientService implements IUserService {
-  private preferenceStorage: AsyncLocalStorage<UserPreference>;
-
-  private get userModel(): UserModel {
-    return new UserModel(clientDB as any, this.userId);
-  }
-  private get messageModel(): MessageModel {
-    return new MessageModel(clientDB as any, this.userId);
-  }
-  private get sessionModel(): SessionModel {
-    return new SessionModel(clientDB as any, this.userId);
-  }
-
-  constructor(userId?: string) {
-    super(userId);
-    this.preferenceStorage = new AsyncLocalStorage('LOBE_PREFERENCE');
-  }
-
-  async getUserState(): Promise<UserInitializationState> {
-    // if user not exist in the db, create one to make sure the user exist
-    await this.makeSureUserExist();
-
-    const state = await this.userModel.getUserState((encryptKeyVaultsStr) =>
-      encryptKeyVaultsStr ? JSON.parse(encryptKeyVaultsStr) : {},
-    );
-
-    const user = await UserModel.findById(clientDB as any, this.userId);
-    const messageCount = await this.messageModel.count();
-    const sessionCount = await this.sessionModel.count();
-
-    return {
-      ...state,
-      avatar: user?.avatar as string,
-      canEnablePWAGuide: messageCount >= 4,
-      canEnableTrace: messageCount >= 4,
-      hasConversation: messageCount > 0 || sessionCount > 0,
-      isOnboard: true,
-      preference: await this.preferenceStorage.getFromLocalStorage(),
-    };
-  }
-
-  updateUserSettings = async (value: DeepPartial<UserSettings>) => {
-    const { keyVaults, ...res } = value;
-
-    return this.userModel.updateSetting({ ...res, keyVaults: JSON.stringify(keyVaults) });
-  };
-
-  resetUserSettings = async () => {
-    return this.userModel.deleteSetting();
-  };
-
-  async updateAvatar(avatar: string) {
-    await this.userModel.updateUser({ avatar });
-  }
-
-  async updatePreference(preference: Partial<UserPreference>) {
-    await this.preferenceStorage.saveToLocalStorage(preference);
-  }
-
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars,unused-imports/no-unused-vars
-  async updateGuide(guide: Partial<UserGuide>) {
-    throw new Error('Method not implemented.');
-  }
-
-  async makeSureUserExist() {
-    const existUsers = await clientDB.query.users.findMany();
-
-    let user: { id: string };
-    if (existUsers.length === 0) {
-      const result = await clientDB.insert(users).values({ id: this.userId }).returning();
-      user = result[0];
-    } else {
-      user = existUsers[0];
-    }
-
-    return user;
-  }
-}