@lobehub/lobehub 2.0.0-next.3 → 2.0.0-next.31

package/src/database/_deprecated/models/topic.ts

@@ -1,250 +0,0 @@
-import { BaseModel } from '@/database/_deprecated/core';
-import { DBModel } from '@/database/_deprecated/core/types/db';
-import { MessageModel } from '@/database/_deprecated/models/message';
-import { DB_Topic, DB_TopicSchema } from '@/database/_deprecated/schemas/topic';
-import { ChatTopic } from '@/types/topic';
-import { nanoid } from '@/utils/uuid';
-
-export interface CreateTopicParams {
-  favorite?: boolean;
-  messages?: string[];
-  sessionId: string;
-  title: string;
-}
-
-export interface QueryTopicParams {
-  containerId?: string | null; // sessionId or groupId
-  current?: number;
-  pageSize?: number;
-}
-
-class _TopicModel extends BaseModel {
-  constructor() {
-    super('topics', DB_TopicSchema);
-  }
-
-  // **************** Query *************** //
-
-  async query({
-    pageSize = 9999,
-    current = 0,
-    containerId,
-  }: QueryTopicParams): Promise<ChatTopic[]> {
-    const offset = current * pageSize;
-
-    // get all topics
-    const allTopics = containerId
-      ? await this.table.where('sessionId').equals(containerId).toArray()
-      : await this.table.toArray();
-
-    // 将所有主题按星标消息优先，时间倒序进行排序
-    const sortedTopics = allTopics.sort((a, b) => {
-      if (a.favorite && !b.favorite) return -1; // a是星标，b不是，a排前面
-      if (!a.favorite && b.favorite) return 1; // b是星标，a不是，b排前面
-
-      // 如果星标状态相同，则按时间倒序排序
-      return b.createdAt - a.createdAt;
-    });
-
-    // handle pageSize
-    const pagedTopics = sortedTopics.slice(offset, offset + pageSize);
-
-    return pagedTopics.map((i) => this.mapToChatTopic(i));
-  }
-
-  queryAll() {
-    return this.table.orderBy('updatedAt').toArray();
-  }
-
-  /**
-   * Query topics by keyword in title, message content, or translated content
-   * @param keyword The keyword to search for
-   * @param sessionId The currently activated session id.
-   */
-  async queryByKeyword(keyword: string, sessionId?: string): Promise<ChatTopic[]> {
-    if (!keyword) return [];
-
-    console.time('queryTopicsByKeyword');
-    const keywordLowerCase = keyword.toLowerCase();
-
-    // Find topics with matching title
-    const queryTable = sessionId ? this.table.where('sessionId').equals(sessionId) : this.table;
-    const matchingTopicsPromise = queryTable
-      .filter((topic) => topic.title.toLowerCase().includes(keywordLowerCase))
-      .toArray();
-
-    // Find messages with matching content or translate.content
-    const queryMessages = sessionId
-      ? this.db.messages.where('sessionId').equals(sessionId)
-      : this.db.messages;
-    const matchingMessagesPromise = queryMessages
-      .filter((message) => {
-        // check content
-        if (message.content.toLowerCase().includes(keywordLowerCase)) return true;
-
-        // check translate content
-        if (message.translate && message.translate.content) {
-          return message.translate.content.toLowerCase().includes(keywordLowerCase);
-        }
-
-        return false;
-      })
-      .toArray();
-
-    // Resolve both promises
-    const [matchingTopics, matchingMessages] = await Promise.all([
-      matchingTopicsPromise,
-      matchingMessagesPromise,
-    ]);
-
-    // Extract topic IDs from messages
-    const topicIdsFromMessages = matchingMessages.map((message) => message.topicId);
-
-    // Combine topic IDs from both sources
-    const combinedTopicIds = new Set([
-      ...topicIdsFromMessages,
-      ...matchingTopics.map((topic) => topic.id),
-    ]);
-
-    // Retrieve unique topics by IDs
-    const uniqueTopics = await this.table
-      .where('id')
-      .anyOf([...combinedTopicIds])
-      .toArray();
-
-    console.timeEnd('queryTopicsByKeyword');
-    return uniqueTopics.map((i) => ({ ...i, favorite: !!i.favorite }));
-  }
-
-  async findBySessionId(sessionId: string) {
-    return this.table.where({ sessionId }).toArray();
-  }
-
-  async findById(id: string): Promise<DBModel<DB_Topic>> {
-    return this.table.get(id);
-  }
-
-  async count() {
-    return this.table.count();
-  }
-
-  // **************** Create *************** //
-
-  async create({ title, favorite, sessionId, messages }: CreateTopicParams, id = nanoid()) {
-    const topic = await this._addWithSync(
-      { favorite: favorite ? 1 : 0, sessionId, title: title },
-      id,
-    );
-
-    // add topicId to these messages
-    if (messages) {
-      await MessageModel.batchUpdate(messages, { topicId: topic.id });
-    }
-
-    return topic;
-  }
-
-  async batchCreate(topics: CreateTopicParams[]) {
-    return this._batchAdd(topics.map((t) => ({ ...t, favorite: t.favorite ? 1 : 0 })));
-  }
-
-  async duplicateTopic(topicId: string, newTitle?: string) {
-    return this.db.transaction('rw', [this.db.topics, this.db.messages], async () => {
-      // Step 1: get DB_Topic
-      const topic = await this.findById(topicId);
-
-      if (!topic) {
-        throw new Error(`Topic with id ${topicId} not found`);
-      }
-
-      // Step 3: 查询与 `topic` 关联的 `messages`
-      const originalMessages = await MessageModel.queryByTopicId(topicId);
-
-      const duplicateMessages = await MessageModel.duplicateMessages(originalMessages);
-
-      const { id } = await this.create({
-        ...this.mapToChatTopic(topic),
-        messages: duplicateMessages.map((m) => m.id),
-        sessionId: topic.sessionId!,
-        title: newTitle || topic.title,
-      });
-
-      return id;
-    });
-  }
-
-  // **************** Delete *************** //
-
-  /**
-   * Deletes a topic and all messages associated with it.
-   */
-  async delete(id: string) {
-    return this.db.transaction('rw', [this.table, this.db.messages], async () => {
-      // Delete all messages associated with the topic
-      await MessageModel.batchDeleteByTopicId(id);
-
-      await this._deleteWithSync(id);
-    });
-  }
-
-  /**
-   * Deletes multiple topic based on the sessionId.
-   *
-   * @param {string} sessionId - The identifier of the assistant associated with the messages.
-   * @returns {Promise<void>}
-   */
-  async batchDeleteBySessionId(sessionId: string): Promise<void> {
-    // use sessionId as the filter criteria in the query.
-    const query = this.table.where('sessionId').equals(sessionId);
-
-    // Retrieve a collection of message IDs that satisfy the criteria
-    const topicIds = await query.primaryKeys();
-
-    // Use the bulkDelete method to delete all selected messages in bulk
-    return this._bulkDeleteWithSync(topicIds);
-  }
-  /**
-   * Deletes multiple topics and all messages associated with them in a transaction.
-   */
-  async batchDelete(topicIds: string[]) {
-    return this.db.transaction('rw', [this.table, this.db.messages], async () => {
-      // Iterate over each topicId and delete related messages, then delete the topic itself
-      for (const topicId of topicIds) {
-        // Delete all messages associated with the topic
-        await this.delete(topicId);
-      }
-    });
-  }
-
-  async clearTable() {
-    return this._clearWithSync();
-  }
-
-  // **************** Update *************** //
-  async update(id: string, data: Partial<DB_Topic>) {
-    return super._updateWithSync(id, data);
-  }
-
-  async toggleFavorite(id: string, newState?: boolean) {
-    const topic = await this.findById(id);
-    if (!topic) {
-      throw new Error(`Topic with id ${id} not found`);
-    }
-
-    // Toggle the 'favorite' status
-    const nextState = typeof newState !== 'undefined' ? newState : !topic.favorite;
-
-    await this.update(id, { favorite: nextState ? 1 : 0 });
-
-    return nextState;
-  }
-
-  // **************** Helper *************** //
-
-  private mapToChatTopic = (dbTopic: DBModel<DB_Topic>): ChatTopic => ({
-    ...dbTopic,
-    favorite: !!dbTopic.favorite,
-  });
-}
-
-export const TopicModel = new _TopicModel();

package/src/database/_deprecated/models/user.ts

@@ -1,69 +0,0 @@
-import type { PartialDeep } from 'type-fest';
-
-import { BaseModel } from '@/database/_deprecated/core';
-import { LobeAgentConfig } from '@/types/agent';
-import { uuid } from '@/utils/uuid';
-
-import { DB_Settings, DB_User, DB_UserSchema } from '../schemas/user';
-
-class _UserModel extends BaseModel {
-  constructor() {
-    super('users', DB_UserSchema);
-  }
-  // **************** Query *************** //
-
-  getUser = async (): Promise<DB_User & { id: number }> => {
-    const noUser = !(await this.table.count());
-
-    if (noUser) await this.table.put({ uuid: uuid() });
-
-    const list = (await this.table.toArray()) as (DB_User & { id: number })[];
-
-    return list[0];
-  };
-
-  getAgentConfig = async () => {
-    const user = await this.getUser();
-
-    return user.settings?.defaultAgent?.config as LobeAgentConfig;
-  };
-  // **************** Create *************** //
-
-  create = async (user: DB_User) => {
-    return this.table.put(user);
-  };
-
-  // **************** Delete *************** //
-
-  clear() {
-    return this.table.clear();
-  }
-
-  // **************** Update *************** //
-
-  async updateSettings(settings: PartialDeep<DB_Settings>) {
-    const user = await this.getUser();
-
-    return this.update(user.id, { settings: settings as any });
-  }
-
-  async resetSettings() {
-    const user = await this.getUser();
-
-    return this.update(user.id, { avatar: undefined, settings: undefined });
-  }
-
-  async updateAvatar(avatar: string) {
-    const user = await this.getUser();
-
-    return this.update(user.id, { avatar });
-  }
-
-  // **************** Helper *************** //
-
-  private update = async (id: number, value: PartialDeep<DB_User>) => {
-    return this.table.update(id, value);
-  };
-}
-
-export const UserModel = new _UserModel();

package/src/database/_deprecated/schemas/files.ts

@@ -1,39 +0,0 @@
-import { z } from 'zod';
-
-export const DB_FileSchema = z.object({
-  /**
-   * create Time
-   */
-  createdAt: z.number().optional(),
-  /**
-   * file data array buffer
-   */
-  data: z.instanceof(ArrayBuffer).optional(),
-  /**
-   * file type
-   * @example 'image/png'
-   */
-  fileType: z.string(),
-  metadata: z.any().optional(),
-  /**
-   * file name
-   * @example 'test.png'
-   */
-  name: z.string(),
-  /**
-   * the mode database save the file
-   * local mean save the raw file into data
-   * url mean upload the file to a cdn and then save the url
-   */
-  saveMode: z.enum(['local', 'url']),
-  /**
-   * file size
-   */
-  size: z.number(),
-  /**
-   * file url if saveMode is url
-   */
-  url: z.string().optional(),
-});
-
-export type DB_File = z.infer<typeof DB_FileSchema>;

package/src/database/_deprecated/schemas/message.ts

@@ -1,50 +0,0 @@
-/* eslint-disable sort-keys-fix/sort-keys-fix  */
-import { z } from 'zod';
-
-const TranslateSchema = z.object({
-  from: z.string().optional(),
-  to: z.string(),
-  content: z.string().optional(),
-});
-
-const PluginSchema = z.object({
-  identifier: z.string(),
-  arguments: z.string(),
-  apiName: z.string(),
-  type: z.enum(['default', 'markdown', 'standalone', 'builtin', 'mcp']).default('default'),
-});
-
-const ToolCallSchema = PluginSchema.extend({
-  id: z.string(),
-});
-
-export const DB_MessageSchema = z.object({
-  role: z.enum(['user', 'system', 'assistant', 'tool']),
-  content: z.string(),
-  files: z.array(z.string()).optional(),
-  favorite: z.number().int().min(0).max(1).optional(),
-  error: z.any().optional(),
-
-  tools: z.array(ToolCallSchema).optional(),
-  tool_call_id: z.string().optional(),
-
-  plugin: PluginSchema.optional(),
-  pluginState: z.any().optional(),
-  pluginError: z.any().optional(),
-
-  fromModel: z.string().optional(),
-  fromProvider: z.string().optional(),
-  translate: TranslateSchema.optional().or(z.literal(false)).or(z.null()),
-  tts: z.any().optional(),
-
-  traceId: z.string().optional(),
-  observationId: z.string().optional(),
-
-  // foreign keys
-  parentId: z.string().optional(),
-  quotaId: z.string().optional(),
-  sessionId: z.string(),
-  topicId: z.string().nullable().optional(),
-});
-
-export type DB_Message = z.infer<typeof DB_MessageSchema>;

package/src/database/_deprecated/schemas/plugin.ts

@@ -1,12 +0,0 @@
-/* eslint-disable sort-keys-fix/sort-keys-fix  */
-import { z } from 'zod';
-
-export const DB_PluginSchema = z.object({
-  identifier: z.string(),
-  id: z.string(),
-  type: z.enum(['plugin', 'customPlugin']),
-  manifest: z.any().optional(),
-  settings: z.any().optional(),
-});
-
-export type DB_Plugin = z.infer<typeof DB_PluginSchema>;

package/src/database/_deprecated/schemas/session.ts

@@ -1,54 +0,0 @@
-import { z } from 'zod';
-
-import { DEFAULT_MODEL } from '@/const/settings';
-import { AgentChatConfigSchema } from '@/types/agent';
-import { LobeMetaDataSchema } from '@/types/meta';
-
-const fewShotsSchema = z.array(
-  z.object({
-    content: z.string(),
-    role: z.string(),
-  }),
-);
-
-const ttsSchema = z.object({
-  showAllLocaleVoice: z.boolean().optional(),
-  sttLocale: z.string().default('auto'),
-  ttsService: z.string().default('openai'),
-  voice: z
-    .object({
-      edge: z.string().optional(),
-      microsoft: z.string().optional(),
-      openai: z.string().default(''),
-    })
-    .optional(),
-});
-
-export const AgentSchema = z.object({
-  chatConfig: AgentChatConfigSchema,
-  fewShots: fewShotsSchema.optional(),
-  model: z.string().default(DEFAULT_MODEL),
-  openingMessage: z.string().optional(),
-  openingQuestions: z.array(z.string()).default([]).optional(),
-  params: z.object({
-    frequency_penalty: z.number().default(0).optional(),
-    max_tokens: z.number().optional(),
-    presence_penalty: z.number().default(0).optional(),
-    temperature: z.number().default(1).optional(),
-    top_p: z.number().default(1).optional(),
-  }),
-  plugins: z.array(z.string()).optional(),
-  provider: z.string().default('openai').optional(),
-  systemRole: z.string().default(''),
-  tts: ttsSchema.optional(),
-});
-
-export const DB_SessionSchema = z.object({
-  config: AgentSchema,
-  group: z.string().default('default'),
-  meta: LobeMetaDataSchema,
-  pinned: z.number().int().min(0).max(1).optional(),
-  type: z.enum(['agent', 'group']).default('agent'),
-});
-
-export type DB_Session = z.infer<typeof DB_SessionSchema>;

package/src/database/_deprecated/schemas/sessionGroup.ts

@@ -1,8 +0,0 @@
-import { z } from 'zod';
-
-export const DB_SessionGroupSchema = z.object({
-  name: z.string(),
-  sort: z.number().optional(),
-});
-
-export type DB_SessionGroup = z.infer<typeof DB_SessionGroupSchema>;

package/src/database/_deprecated/schemas/topic.ts

@@ -1,12 +0,0 @@
-/* eslint-disable sort-keys-fix/sort-keys-fix  */
-import { z } from 'zod';
-
-export const DB_TopicSchema = z.object({
-  title: z.string(),
-  favorite: z.number().int().default(0),
-
-  // foreign keys
-  sessionId: z.string().optional(),
-});
-
-export type DB_Topic = z.infer<typeof DB_TopicSchema>;

package/src/database/_deprecated/schemas/user.ts

@@ -1,40 +0,0 @@
-import { z } from 'zod';
-
-import { AgentSchema } from '@/database/_deprecated/schemas/session';
-import { LobeMetaDataSchema } from '@/types/meta';
-
-const generalSechma = z.object({
-  fontSize: z.number().default(14),
-  language: z.string(),
-  neutralColor: z.string().optional(),
-  password: z.string(),
-  themeMode: z.string(),
-});
-
-const settingsSchema = z.object({
-  defaultAgent: z.object({
-    config: AgentSchema,
-    meta: LobeMetaDataSchema,
-  }),
-  general: generalSechma.partial().optional(),
-  keyVaults: z.any().optional(),
-  languageModel: z.any().optional(),
-  tts: z.object({
-    openAI: z.object({
-      sttModel: z.string(),
-      ttsModel: z.string(),
-    }),
-    sttAutoStop: z.boolean(),
-    sttServer: z.string(),
-  }),
-});
-
-export const DB_UserSchema = z.object({
-  avatar: z.string().optional(),
-  settings: settingsSchema.partial(),
-  uuid: z.string(),
-});
-
-export type DB_User = z.infer<typeof DB_UserSchema>;
-
-export type DB_Settings = z.infer<typeof settingsSchema>;