@lobehub/lobehub 2.0.0-next.16 → 2.0.0-next.160

package/packages/database/src/core/electron.ts

@@ -1,18 +1,18 @@
 import { PGlite } from '@electric-sql/pglite';
 import { vector } from '@electric-sql/pglite/vector';
+import type { MigrationTableItem } from '@lobechat/types';
 import { drizzle as pgliteDrizzle } from 'drizzle-orm/pglite';
 import fs from 'node:fs';
 import { Md5 } from 'ts-md5';
 
 import { electronIpcClient } from '@/server/modules/ElectronIPCClient';
-import { MigrationTableItem } from '@/types/clientDB';
 
 import { DrizzleMigrationModel } from '../models/drizzleMigration';
 import * as schema from '../schemas';
 import { LobeChatDatabase } from '../type';
 import migrations from './migrations.json';
 
-// 用于实例管理的全局对象
+// Global object for instance management
 interface LobeGlobal {
   pgDB?: LobeChatDatabase;
   pgDBInitPromise?: Promise<LobeChatDatabase>;
@@ -22,7 +22,7 @@ interface LobeGlobal {
   };
 }
 
-// 确保 globalThis 有我们的命名空间
+// Ensure globalThis has our namespace
 declare global {
   // eslint-disable-next-line no-var
   var __LOBE__: LobeGlobal;
@@ -33,20 +33,20 @@ if (!globalThis.__LOBE__) {
 }
 
 /**
- * 尝试创建一个文件锁来确保单例模式
- * 返回 true 表示成功获取锁，false 表示已有其他实例正在运行
+ * Attempt to create a file lock to ensure singleton pattern
+ * Returns true if lock acquired successfully, false if another instance is already running
  */
 const acquireLock = async (dbPath: string): Promise<boolean> => {
   try {
-    // 数据库锁文件路径
+    // Database lock file path
     const lockPath = `${dbPath}.lock`;
 
-    // 尝试创建锁文件
+    // Attempt to create lock file
     if (!fs.existsSync(lockPath)) {
-      // 创建锁文件并写入当前进程 ID
+      // Create lock file and write current process ID
       fs.writeFileSync(lockPath, process.pid.toString(), 'utf8');
 
-      // 保存锁信息到全局对象
+      // Save lock information to global object
       if (!globalThis.__LOBE__.pgDBLock) {
         globalThis.__LOBE__.pgDBLock = {
           acquired: true,
@@ -58,19 +58,19 @@ const acquireLock = async (dbPath: string): Promise<boolean> => {
       return true;
     }
 
-    // 检查锁文件是否过期（超过5分钟未更新）
+    // Check if lock file has expired (not updated for over 5 minutes)
     const stats = fs.statSync(lockPath);
     const currentTime = Date.now();
     const modifiedTime = stats.mtime.getTime();
 
-    // 如果锁文件超过5分钟未更新，视为过期锁
+    // If lock file hasn't been updated for over 5 minutes, consider it expired
     if (currentTime - modifiedTime > 5 * 60 * 1000) {
-      // 删除过期锁文件
+      // Delete expired lock file
       fs.unlinkSync(lockPath);
-      // 重新创建锁文件
+      // Recreate lock file
       fs.writeFileSync(lockPath, process.pid.toString(), 'utf8');
 
-      // 保存锁信息到全局对象
+      // Save lock information to global object
       if (!globalThis.__LOBE__.pgDBLock) {
         globalThis.__LOBE__.pgDBLock = {
           acquired: true,
@@ -91,7 +91,7 @@ const acquireLock = async (dbPath: string): Promise<boolean> => {
 };
 
 /**
- * 释放文件锁
+ * Release file lock
  */
 const releaseLock = () => {
   if (globalThis.__LOBE__.pgDBLock?.acquired && globalThis.__LOBE__.pgDBLock.lockPath) {
@@ -105,7 +105,7 @@ const releaseLock = () => {
   }
 };
 
-// 在进程退出时释放锁
+// Release lock on process exit
 process.on('exit', releaseLock);
 process.on('SIGINT', () => {
   releaseLock();
@@ -129,15 +129,15 @@ const migrateDatabase = async (db: LobeChatDatabase): Promise<void> => {
 
     console.log('schemaHash:', hash);
 
-    // 如果哈希值相同，看下表是否全了
+    // If hash is the same, check if all tables exist
     if (hash === cacheHash) {
       try {
         const drizzleMigration = new DrizzleMigrationModel(db);
 
-        // 检查数据库中是否存在表
+        // Check if tables exist in the database
         const tableCount = await drizzleMigration.getTableCounts();
 
-        // 如果表数量大于0，则认为数据库已正确初始化
+        // If table count is greater than 0, assume database is properly initialized
         if (tableCount > 0) {
           console.log('✅ Electron DB schema already synced');
           return;
@@ -152,7 +152,7 @@ const migrateDatabase = async (db: LobeChatDatabase): Promise<void> => {
     console.log('🚀 Starting Electron DB migration...');
 
     try {
-      // 执行迁移
+      // Execute migration
       // @ts-expect-error
       await db.dialect.migrate(migrations, db.session, {});
 
@@ -162,10 +162,10 @@ const migrateDatabase = async (db: LobeChatDatabase): Promise<void> => {
     } catch (error) {
       console.error('❌ Electron database schema migration failed', error);
 
-      // 尝试查询迁移表数据
+      // Attempt to query migration table data
       let migrationsTableData: MigrationTableItem[] = [];
       try {
-        // 尝试查询迁移表
+        // Attempt to query migration table
         const drizzleMigration = new DrizzleMigrationModel(db);
         migrationsTableData = await drizzleMigration.getMigrationList();
       } catch (queryError) {
@@ -184,12 +184,12 @@ const migrateDatabase = async (db: LobeChatDatabase): Promise<void> => {
 };
 
 /**
- * 检查当前是否有活跃的数据库实例，如果有则尝试关闭它
+ * Check if there's an active database instance and attempt to close it if exists
  */
 const checkAndCleanupExistingInstance = async () => {
   if (globalThis.__LOBE__.pgDB) {
     try {
-      // 尝试关闭现有的 PGlite 实例 (如果客户端有 close 方法)
+      // Attempt to close existing PGlite instance (if client has close method)
       // @ts-expect-error
       const client = globalThis.__LOBE__.pgDB?.dialect?.client;
 
@@ -198,11 +198,11 @@ const checkAndCleanupExistingInstance = async () => {
         console.log('✅ Successfully closed previous PGlite instance');
       }
 
-      // 重置全局引用
+      // Reset global reference
       globalThis.__LOBE__.pgDB = undefined;
     } catch (error) {
       console.error('❌ Failed to close previous PGlite instance:', error);
-      // 继续执行，创建新实例
+      // Continue execution and create new instance
     }
   }
 };
@@ -220,19 +220,19 @@ export const getPgliteInstance = async (): Promise<LobeChatDatabase> => {
       }),
     );
 
-    // 已经初始化完成，直接返回实例
+    // Already initialized, return instance directly
     if (globalThis.__LOBE__.pgDB) return globalThis.__LOBE__.pgDB;
 
-    // 有初始化进行中的Promise，等待它完成
+    // Initialization promise in progress, wait for it to complete
     if (globalThis.__LOBE__.pgDBInitPromise) {
       console.log('Waiting for existing initialization promise to complete');
       return globalThis.__LOBE__.pgDBInitPromise;
     }
 
-    // 防止多次调用引起的竞态条件
+    // Prevent race conditions from multiple calls
     if (isInitializing) {
       console.log('Already initializing, waiting for result');
-      // 创建新的 Promise 等待初始化完成
+      // Create new Promise to wait for initialization to complete
       return new Promise((resolve, reject) => {
         const checkInterval = setInterval(() => {
           if (globalThis.__LOBE__.pgDB) {
@@ -248,12 +248,12 @@ export const getPgliteInstance = async (): Promise<LobeChatDatabase> => {
 
     isInitializing = true;
 
-    // 创建初始化Promise并保存
+    // Create and save initialization Promise
     globalThis.__LOBE__.pgDBInitPromise = (async () => {
-      // 再次检查，以防在等待过程中已有其他调用初始化成功
+      // Check again in case another call succeeded during wait
       if (globalThis.__LOBE__.pgDB) return globalThis.__LOBE__.pgDB;
 
-      // 先获取数据库路径
+      // Get database path first
       let dbPath: string = '';
       try {
         dbPath = await electronIpcClient.getDatabasePath();
@@ -263,34 +263,34 @@ export const getPgliteInstance = async (): Promise<LobeChatDatabase> => {
 
       console.log('Database path:', dbPath);
       try {
-        // 尝试获取数据库锁
+        // Attempt to acquire database lock
         const lockAcquired = await acquireLock(dbPath);
         if (!lockAcquired) {
           throw new Error('Cannot acquire database lock. Another instance might be using it.');
         }
 
-        // 检查并清理可能存在的旧实例
+        // Check and cleanup any existing old instances
         await checkAndCleanupExistingInstance();
 
-        // 创建新的 PGlite 实例
+        // Create new PGlite instance
         console.log('Creating new PGlite instance');
         const client = new PGlite(dbPath, {
           extensions: { vector },
-          // 增加选项以提高稳定性
+          // Add options to improve stability
           relaxedDurability: true,
         });
 
-        // 等待数据库就绪
+        // Wait for database to be ready
         await client.waitReady;
         console.log('PGlite state:', client.ready);
 
-        // 创建 Drizzle 数据库实例
+        // Create Drizzle database instance
         const db = pgliteDrizzle({ client, schema }) as unknown as LobeChatDatabase;
 
-        // 执行迁移
+        // Execute migration
         await migrateDatabase(db);
 
-        // 保存实例引用
+        // Save instance reference
         globalThis.__LOBE__.pgDB = db;
 
         console.log('✅ PGlite instance successfully initialized');
@@ -298,9 +298,9 @@ export const getPgliteInstance = async (): Promise<LobeChatDatabase> => {
         return db;
       } catch (error) {
         console.error('❌ Failed to initialize PGlite instance:', error);
-        // 清空初始化Promise，允许下次重试
+        // Clear initialization Promise to allow retry next time
         globalThis.__LOBE__.pgDBInitPromise = undefined;
-        // 释放可能已获取的锁
+        // Release potentially acquired lock
         releaseLock();
         throw error;
       } finally {

package/packages/database/src/core/migrations.json

@@ -746,5 +746,197 @@
     "bps": true,
     "folderMillis": 1761878697451,
     "hash": "2f740719356c4ea28a4f71b6089595871f6e185e056b43288a7d16f7d0aae196"
+  },
+  {
+    "sql": [
+      "CREATE INDEX IF NOT EXISTS \"agents_knowledge_bases_agent_id_idx\" ON \"agents_knowledge_bases\" USING btree (\"agent_id\");\n"
+    ],
+    "bps": true,
+    "folderMillis": 1762232313711,
+    "hash": "fd8e7ff36bf877ddc1a45d360edfe1fe9a579cd02d68d7664545d7df95b8eb09"
+  },
+  {
+    "sql": [
+      "ALTER TABLE \"ai_models\" ADD COLUMN IF NOT EXISTS \"settings\" jsonb DEFAULT '{}'::jsonb;"
+    ],
+    "bps": true,
+    "folderMillis": 1762251112601,
+    "hash": "923ccbdf46c32be9a981dabd348e6923b4a365444241e9b8cc174bf5b914cbc5"
+  },
+  {
+    "sql": ["ALTER TABLE \"agents\" ADD COLUMN IF NOT EXISTS \"market_identifier\" text;\n"],
+    "bps": true,
+    "folderMillis": 1762870034882,
+    "hash": "4178aacb4b8892b7fd15d29209bbf9b1d1f9d7c406ba796f27542c0bcd919680"
+  },
+  {
+    "sql": ["ALTER TABLE \"message_plugins\" ADD COLUMN IF NOT EXISTS \"intervention\" jsonb;\n"],
+    "bps": true,
+    "folderMillis": 1762911968658,
+    "hash": "552a032cc0e595277232e70b5f9338658585bafe9481ae8346a5f322b673a68b"
+  },
+  {
+    "sql": [
+      "ALTER TABLE \"documents\" ALTER COLUMN \"id\" SET DATA TYPE varchar(255);",
+      "\nALTER TABLE \"documents\" ADD COLUMN IF NOT EXISTS \"parent_id\" varchar(255);",
+      "\nALTER TABLE \"files\" ADD COLUMN IF NOT EXISTS \"parent_id\" varchar(255);",
+      "\nDO $$ BEGIN\n ALTER TABLE \"documents\" ADD CONSTRAINT \"documents_parent_id_documents_id_fk\" FOREIGN KEY (\"parent_id\") REFERENCES \"public\".\"documents\"(\"id\") ON DELETE set null ON UPDATE no action;\nEXCEPTION\n WHEN duplicate_object THEN null;\nEND $$;",
+      "\nDO $$ BEGIN\n ALTER TABLE \"files\" ADD CONSTRAINT \"files_parent_id_documents_id_fk\" FOREIGN KEY (\"parent_id\") REFERENCES \"public\".\"documents\"(\"id\") ON DELETE set null ON UPDATE no action;\nEXCEPTION\n WHEN duplicate_object THEN null;\nEND $$;",
+      "\nCREATE INDEX IF NOT EXISTS \"documents_parent_id_idx\" ON \"documents\" USING btree (\"parent_id\");",
+      "\nCREATE INDEX IF NOT EXISTS \"files_parent_id_idx\" ON \"files\" USING btree (\"parent_id\");"
+    ],
+    "bps": true,
+    "folderMillis": 1763453175961,
+    "hash": "6c2081c8ac22772a8276052c9ae0852a4c03463cbe998df26f85683ab356b914"
+  },
+  {
+    "sql": [
+      "ALTER TABLE \"documents\" ADD COLUMN IF NOT EXISTS \"slug\" varchar(255);",
+      "\nCREATE UNIQUE INDEX IF NOT EXISTS \"documents_slug_user_id_unique\" ON \"documents\" USING btree (\"slug\",\"user_id\") WHERE \"documents\".\"slug\" is not null;\n"
+    ],
+    "bps": true,
+    "folderMillis": 1763987922211,
+    "hash": "f823b521f4d25e5dc5ab238b372727d2d2d7f0aed27b5eabc8a9608ce4e50568"
+  },
+  {
+    "sql": ["ALTER TABLE \"agents\" ADD COLUMN IF NOT EXISTS \"editor_data\" jsonb;"],
+    "bps": true,
+    "folderMillis": 1764215503726,
+    "hash": "4188893a9083b3c7baebdbad0dd3f9d9400ede7584ca2394f5c64305dc9ec7b0"
+  },
+  {
+    "sql": [
+      "CREATE TABLE IF NOT EXISTS \"accounts\" (\n\t\"access_token\" text,\n\t\"access_token_expires_at\" timestamp,\n\t\"account_id\" text NOT NULL,\n\t\"created_at\" timestamp DEFAULT now() NOT NULL,\n\t\"id\" text PRIMARY KEY NOT NULL,\n\t\"id_token\" text,\n\t\"password\" text,\n\t\"provider_id\" text NOT NULL,\n\t\"refresh_token\" text,\n\t\"refresh_token_expires_at\" timestamp,\n\t\"scope\" text,\n\t\"updated_at\" timestamp NOT NULL,\n\t\"user_id\" text NOT NULL\n);\n",
+      "\nCREATE TABLE IF NOT EXISTS \"auth_sessions\" (\n\t\"created_at\" timestamp DEFAULT now() NOT NULL,\n\t\"expires_at\" timestamp NOT NULL,\n\t\"id\" text PRIMARY KEY NOT NULL,\n\t\"ip_address\" text,\n\t\"token\" text NOT NULL,\n\t\"updated_at\" timestamp NOT NULL,\n\t\"user_agent\" text,\n\t\"user_id\" text NOT NULL,\n\tCONSTRAINT \"auth_sessions_token_unique\" UNIQUE(\"token\")\n);\n",
+      "\nCREATE TABLE IF NOT EXISTS \"verifications\" (\n\t\"created_at\" timestamp DEFAULT now() NOT NULL,\n\t\"expires_at\" timestamp NOT NULL,\n\t\"id\" text PRIMARY KEY NOT NULL,\n\t\"identifier\" text NOT NULL,\n\t\"updated_at\" timestamp DEFAULT now() NOT NULL,\n\t\"value\" text NOT NULL\n);\n",
+      "\nALTER TABLE \"users\" ADD COLUMN IF NOT EXISTS \"email_verified\" boolean DEFAULT false NOT NULL;",
+      "\nDO $$ BEGIN\n ALTER TABLE \"accounts\" ADD CONSTRAINT \"accounts_user_id_users_id_fk\" FOREIGN KEY (\"user_id\") REFERENCES \"public\".\"users\"(\"id\") ON DELETE cascade ON UPDATE no action;\nEXCEPTION\n WHEN duplicate_object THEN null;\nEND $$;\n",
+      "\nDO $$ BEGIN\n ALTER TABLE \"auth_sessions\" ADD CONSTRAINT \"auth_sessions_user_id_users_id_fk\" FOREIGN KEY (\"user_id\") REFERENCES \"public\".\"users\"(\"id\") ON DELETE cascade ON UPDATE no action;\nEXCEPTION\n WHEN duplicate_object THEN null;\nEND $$;\n"
+    ],
+    "bps": true,
+    "folderMillis": 1764229953081,
+    "hash": "1532ebceae7b70550bc9c230fb0a65090aaa773bc7b873eefbc2ce2a815997e2"
+  },
+  {
+    "sql": [
+      "ALTER TABLE \"nextauth_accounts\" RENAME COLUMN \"userId\" TO \"user_id\";",
+      "\nALTER TABLE \"nextauth_authenticators\" RENAME COLUMN \"userId\" TO \"user_id\";",
+      "\nALTER TABLE \"nextauth_sessions\" RENAME COLUMN \"userId\" TO \"user_id\";",
+      "\nALTER TABLE \"nextauth_accounts\" DROP CONSTRAINT \"nextauth_accounts_userId_users_id_fk\";\n",
+      "\nALTER TABLE \"nextauth_authenticators\" DROP CONSTRAINT \"nextauth_authenticators_userId_users_id_fk\";\n",
+      "\nALTER TABLE \"nextauth_sessions\" DROP CONSTRAINT \"nextauth_sessions_userId_users_id_fk\";\n",
+      "\nALTER TABLE \"nextauth_authenticators\" DROP CONSTRAINT \"nextauth_authenticators_userId_credentialID_pk\";",
+      "\nALTER TABLE \"threads\" ALTER COLUMN \"status\" DROP DEFAULT;",
+      "\nALTER TABLE \"threads\" ALTER COLUMN \"source_message_id\" DROP NOT NULL;",
+      "\nALTER TABLE \"nextauth_authenticators\" ADD CONSTRAINT \"nextauth_authenticators_user_id_credentialID_pk\" PRIMARY KEY(\"user_id\",\"credentialID\");",
+      "\nALTER TABLE \"threads\" ADD COLUMN IF NOT EXISTS \"content\" text;",
+      "\nALTER TABLE \"threads\" ADD COLUMN IF NOT EXISTS \"editor_data\" jsonb;",
+      "\nALTER TABLE \"nextauth_accounts\" ADD CONSTRAINT \"nextauth_accounts_user_id_users_id_fk\" FOREIGN KEY (\"user_id\") REFERENCES \"public\".\"users\"(\"id\") ON DELETE cascade ON UPDATE no action;",
+      "\nALTER TABLE \"nextauth_authenticators\" ADD CONSTRAINT \"nextauth_authenticators_user_id_users_id_fk\" FOREIGN KEY (\"user_id\") REFERENCES \"public\".\"users\"(\"id\") ON DELETE cascade ON UPDATE no action;",
+      "\nALTER TABLE \"nextauth_sessions\" ADD CONSTRAINT \"nextauth_sessions_user_id_users_id_fk\" FOREIGN KEY (\"user_id\") REFERENCES \"public\".\"users\"(\"id\") ON DELETE cascade ON UPDATE no action;\n"
+    ],
+    "bps": true,
+    "folderMillis": 1764303057060,
+    "hash": "2c103eee82bdf329944fb622dd9c2b9f20df80eb54f23eb9254d2285de413099"
+  },
+  {
+    "sql": ["ALTER TABLE \"user_settings\" ADD COLUMN IF NOT EXISTS \"market\" jsonb;"],
+    "bps": true,
+    "folderMillis": 1764335703306,
+    "hash": "28c0d738c0b1fdf5fd871363be1a1477b4accbabdc140fe8dc6e9b339aae2c89"
+  },
+  {
+    "sql": [
+      "ALTER TABLE \"messages\" DROP CONSTRAINT \"messages_agent_id_agents_id_fk\";\n",
+      "\nALTER TABLE \"messages\" ADD COLUMN IF NOT EXISTS \"editor_data\" jsonb;",
+      "\nALTER TABLE \"topics\" ADD COLUMN IF NOT EXISTS \"content\" text;",
+      "\nALTER TABLE \"topics\" ADD COLUMN IF NOT EXISTS \"editor_data\" jsonb;",
+      "\nALTER TABLE \"topics\" ADD COLUMN IF NOT EXISTS \"agent_id\" text;",
+      "\nALTER TABLE \"messages\" ADD CONSTRAINT \"messages_agent_id_agents_id_fk\" FOREIGN KEY (\"agent_id\") REFERENCES \"public\".\"agents\"(\"id\") ON DELETE cascade ON UPDATE no action;",
+      "\nALTER TABLE \"topics\" ADD CONSTRAINT \"topics_agent_id_agents_id_fk\" FOREIGN KEY (\"agent_id\") REFERENCES \"public\".\"agents\"(\"id\") ON DELETE cascade ON UPDATE no action;",
+      "\nCREATE INDEX IF NOT EXISTS \"topics_agent_id_idx\" ON \"topics\" USING btree (\"agent_id\");\n"
+    ],
+    "bps": true,
+    "folderMillis": 1764500630663,
+    "hash": "94721bc06910a456a4756c9b0c27ef5d7ff55b7ea8c772acf58052c0155c693b"
+  },
+  {
+    "sql": [
+      "ALTER TABLE \"auth_sessions\" ADD COLUMN IF NOT EXISTS \"impersonated_by\" text;",
+      "\nALTER TABLE \"users\" ADD COLUMN IF NOT EXISTS \"role\" text;",
+      "\nALTER TABLE \"users\" ADD COLUMN IF NOT EXISTS \"banned\" boolean DEFAULT false;",
+      "\nALTER TABLE \"users\" ADD COLUMN IF NOT EXISTS \"ban_reason\" text;",
+      "\nALTER TABLE \"users\" ADD COLUMN IF NOT EXISTS \"ban_expires\" timestamp with time zone;\n"
+    ],
+    "bps": true,
+    "folderMillis": 1764511006123,
+    "hash": "523a8418ceeadbe80c34affee07cdfa2e8076e297473c55c73e74648a9be1e45"
+  },
+  {
+    "sql": [
+      "CREATE TABLE IF NOT EXISTS \"two_factor\" (\n  \"backup_codes\" text NOT NULL,\n  \"id\" text PRIMARY KEY NOT NULL,\n  \"secret\" text NOT NULL,\n  \"user_id\" text NOT NULL\n);\n",
+      "\nALTER TABLE \"users\" ADD COLUMN IF NOT EXISTS \"two_factor_enabled\" boolean DEFAULT false;\n",
+      "\nALTER TABLE \"users\" ADD COLUMN IF NOT EXISTS \"phone_number\" text;\n",
+      "\nALTER TABLE \"users\" ADD COLUMN IF NOT EXISTS \"phone_number_verified\" boolean;\n",
+      "\nDO $$\nBEGIN\n  IF NOT EXISTS (\n    SELECT 1 FROM pg_constraint WHERE conname = 'two_factor_user_id_users_id_fk'\n  ) THEN\n    ALTER TABLE \"two_factor\"\n      ADD CONSTRAINT \"two_factor_user_id_users_id_fk\"\n      FOREIGN KEY (\"user_id\") REFERENCES \"public\".\"users\"(\"id\") ON DELETE cascade ON UPDATE no action;\n  END IF;\nEND $$;\n",
+      "\nCREATE INDEX IF NOT EXISTS \"two_factor_secret_idx\" ON \"two_factor\" USING btree (\"secret\");\n",
+      "\nCREATE INDEX IF NOT EXISTS \"two_factor_user_id_idx\" ON \"two_factor\" USING btree (\"user_id\");\n",
+      "\nCREATE INDEX IF NOT EXISTS \"account_userId_idx\" ON \"accounts\" USING btree (\"user_id\");\n",
+      "\nCREATE INDEX IF NOT EXISTS \"auth_session_userId_idx\" ON \"auth_sessions\" USING btree (\"user_id\");\n",
+      "\nCREATE INDEX IF NOT EXISTS \"verification_identifier_idx\" ON \"verifications\" USING btree (\"identifier\");\n",
+      "\nDO $$\nBEGIN\n  IF NOT EXISTS (SELECT 1 FROM pg_constraint WHERE conname = 'users_email_unique') THEN\n    -- Normalize empty emails so the unique constraint can be created safely\n    UPDATE \"users\" SET \"email\" = NULL WHERE \"email\" = '';\n    ALTER TABLE \"users\" ADD CONSTRAINT \"users_email_unique\" UNIQUE (\"email\");\n  END IF;\nEND $$;\n",
+      "\nDO $$\nBEGIN\n  IF NOT EXISTS (SELECT 1 FROM pg_constraint WHERE conname = 'users_phone_number_unique') THEN\n    ALTER TABLE \"users\" ADD CONSTRAINT \"users_phone_number_unique\" UNIQUE (\"phone_number\");\n  END IF;\nEND $$;\n"
+    ],
+    "bps": true,
+    "folderMillis": 1764579351312,
+    "hash": "1d2536a9471bb87686b35053f98ba7762259a07c819dc4489bb4f3c7f27a4d8d"
+  },
+  {
+    "sql": [
+      "ALTER TABLE \"users\" DROP CONSTRAINT IF EXISTS \"users_phone_number_unique\";",
+      "\nALTER TABLE \"users\" DROP COLUMN IF EXISTS \"phone_number\";",
+      "\nALTER TABLE \"users\" DROP CONSTRAINT IF EXISTS \"users_phone_unique\";",
+      "\nALTER TABLE \"users\" ADD CONSTRAINT \"users_phone_unique\" UNIQUE(\"phone\");\n"
+    ],
+    "bps": true,
+    "folderMillis": 1764583392443,
+    "hash": "6a43d90ee1d2e1e008d1b8206f227aa05b9a2e2a8fc8c31ec608a3716c716a6c"
+  },
+  {
+    "sql": [
+      "ALTER TABLE \"agents\" DROP CONSTRAINT \"agents_slug_unique\";",
+      "\nCREATE UNIQUE INDEX IF NOT EXISTS \"agents_slug_user_id_unique\" ON \"agents\" USING btree (\"slug\",\"user_id\");\n"
+    ],
+    "bps": true,
+    "folderMillis": 1764685643024,
+    "hash": "6e7ac7f964eb03efa3cb0d2fd35ded23e25c3abf955c4c2a51418f8daef54af9"
+  },
+  {
+    "sql": [
+      "CREATE INDEX IF NOT EXISTS \"topics_extract_status_gin_idx\" ON \"topics\" USING gin ((metadata->'userMemoryExtractStatus') jsonb_path_ops);\n"
+    ],
+    "bps": true,
+    "folderMillis": 1764734167674,
+    "hash": "89c134be2948d3afc360d6bac11dea0c6fd5c902bf6093ed077033adb920fd02"
+  },
+  {
+    "sql": [
+      "ALTER TABLE \"user_installed_plugins\" ADD COLUMN IF NOT EXISTS \"source\" varchar(255);"
+    ],
+    "bps": true,
+    "folderMillis": 1764842015809,
+    "hash": "276514dc101fffc66794156f4ba644599a4ae5997a68b9e7f54452ee374bfa61"
+  },
+  {
+    "sql": [
+      "ALTER TABLE \"users\" ADD COLUMN IF NOT EXISTS \"normalized_email\" text;",
+      "\nCREATE INDEX IF NOT EXISTS \"users_email_idx\" ON \"users\" USING btree (\"email\");",
+      "\nCREATE INDEX IF NOT EXISTS \"users_username_idx\" ON \"users\" USING btree (\"username\");",
+      "\nCREATE UNIQUE INDEX IF NOT EXISTS \"users_normalized_email_unique_idx\" ON \"users\" USING btree (\"normalized_email\");",
+      "\n"
+    ],
+    "bps": true,
+    "folderMillis": 1764858574403,
+    "hash": "7838f9938b370867470e5e11807855253d23b11c2ac6aa9e90687844a356c949"
   }
 ]

package/packages/database/src/core/web-server.ts

@@ -10,7 +10,8 @@ import * as schema from '../schemas';
 import { LobeChatDatabase } from '../type';
 
 export const getDBInstance = (): LobeChatDatabase => {
-  if (!(process.env.NEXT_PUBLIC_SERVICE_MODE === 'server')) return {} as any;
+  // In test environment, return a mock instance to avoid initialization errors
+  if (process.env.NODE_ENV === 'test') return {} as LobeChatDatabase;
 
   if (!serverDBEnv.KEY_VAULTS_SECRET) {
     throw new Error(

package/packages/database/src/index.ts

@@ -1 +1,3 @@
+export * from './core/db-adaptor';
 export * from './type';
+export * from './utils/idGenerator';