@lobehub/chat 1.77.2 → 1.77.4

package/packages/electron-server-ipc/src/ipcServer.ts

@@ -0,0 +1,126 @@
+import fs from 'node:fs';
+import net from 'node:net';
+import os from 'node:os';
+import path from 'node:path';
+
+import { SOCK_FILE, SOCK_INFO_FILE, WINDOW_PIPE_FILE } from './const';
+import { IElectronIPCMethods } from './types';
+
+export type IPCEventMethod = (
+  params: any,
+  context: { id: string; method: string; socket: net.Socket },
+) => Promise<any>;
+
+export type ElectronIPCEventHandler = {
+  [key in IElectronIPCMethods]: IPCEventMethod;
+};
+
+export class ElectronIPCServer {
+  private server: net.Server;
+  private socketPath: string;
+  private eventHandler: ElectronIPCEventHandler;
+
+  constructor(eventHandler: ElectronIPCEventHandler) {
+    const isWindows = process.platform === 'win32';
+    // 创建唯一的套接字路径，避免冲突
+    this.socketPath = isWindows ? WINDOW_PIPE_FILE : path.join(os.tmpdir(), SOCK_FILE);
+
+    // 如果是 Unix 套接字，确保文件不存在
+    if (!isWindows && fs.existsSync(this.socketPath)) {
+      fs.unlinkSync(this.socketPath);
+    }
+
+    // 创建服务器
+    this.server = net.createServer(this.handleConnection.bind(this));
+
+    this.eventHandler = eventHandler;
+  }
+
+  // 启动服务器
+  public start(): Promise<void> {
+    return new Promise((resolve, reject) => {
+      this.server.on('error', (err) => {
+        console.error('IPC Server error:', err);
+        reject(err);
+      });
+
+      this.server.listen(this.socketPath, () => {
+        console.log(`Electron IPC server listening on ${this.socketPath}`);
+
+        // 将套接字路径写入临时文件，供 Next.js 服务端读取
+        const tempDir = os.tmpdir();
+        const socketInfoPath = path.join(tempDir, SOCK_INFO_FILE);
+        fs.writeFileSync(socketInfoPath, JSON.stringify({ socketPath: this.socketPath }), 'utf8');
+
+        resolve();
+      });
+    });
+  }
+
+  // 处理客户端连接
+  private handleConnection(socket: net.Socket): void {
+    let dataBuffer = '';
+
+    socket.on('data', (data) => {
+      dataBuffer += data.toString();
+
+      try {
+        // 尝试解析 JSON 消息
+        const message = JSON.parse(dataBuffer);
+        dataBuffer = ''; // 重置缓冲区
+
+        // 处理请求
+        this.handleRequest(socket, message);
+      } catch {
+        // 如果不是有效的 JSON，可能是消息不完整，继续等待
+      }
+    });
+
+    socket.on('error', (err) => {
+      console.error('Socket error:', err);
+    });
+  }
+
+  // 处理客户端请求
+  private handleRequest = async (socket: net.Socket, request: any) => {
+    const { id, method, params } = request;
+
+    // 根据请求方法执行相应的操作
+    const eventHandler = this.eventHandler[method as IElectronIPCMethods];
+    if (!eventHandler) return;
+
+    try {
+      const data = await eventHandler(params, { id, method, socket });
+
+      this.sendResult(socket, id, data);
+    } catch (err) {
+      this.sendError(socket, id, `Failed to handle method(${method}): ${(err as Error).message}`);
+    }
+  };
+
+  // 发送结果
+  private sendResult(socket: net.Socket, id: string, result: any): void {
+    socket.write(JSON.stringify({ id, result }));
+  }
+
+  // 发送错误
+  private sendError(socket: net.Socket, id: string, error: string): void {
+    socket.write(JSON.stringify({ error, id }));
+  }
+
+  // 关闭服务器
+  public close(): Promise<void> {
+    return new Promise((resolve) => {
+      this.server.close(() => {
+        console.log('Electron IPC server closed');
+
+        // 删除套接字文件（Unix 平台）
+        if (process.platform !== 'win32' && fs.existsSync(this.socketPath)) {
+          fs.unlinkSync(this.socketPath);
+        }
+
+        resolve();
+      });
+    });
+  }
+}

package/packages/electron-server-ipc/src/types/event.ts

@@ -0,0 +1,18 @@
+/* eslint-disable typescript-sort-keys/interface, sort-keys-fix/sort-keys-fix */
+export const ElectronIPCMethods = {
+  getDatabasePath: 'getDatabasePath',
+  getUserDataPath: 'getUserDataPath',
+
+  getDatabaseSchemaHash: 'getDatabaseSchemaHash',
+  setDatabaseSchemaHash: 'setDatabaseSchemaHash',
+} as const;
+
+export type IElectronIPCMethods = keyof typeof ElectronIPCMethods;
+
+export interface IpcDispatchEvent {
+  getDatabasePath: () => Promise<string>;
+  getUserDataPath: () => Promise<string>;
+
+  getDatabaseSchemaHash: () => Promise<string | undefined>;
+  setDatabaseSchemaHash: (hash: string) => Promise<void>;
+}

package/packages/electron-server-ipc/src/types/index.ts

@@ -0,0 +1 @@
+export * from './event';

package/scripts/migrateServerDB/index.ts

@@ -3,7 +3,7 @@ import { migrate as neonMigrate } from 'drizzle-orm/neon-serverless/migrator';
 import { migrate as nodeMigrate } from 'drizzle-orm/node-postgres/migrator';
 import { join } from 'node:path';
 
-import { serverDB } from '../../src/database/server/core/db';
+import { serverDB } from '../../src/database/server';
 import { DB_FAIL_INIT_HINT, PGVECTOR_HINT } from './errorHint';
 
 // Read the `.env` file if it exists, or a file specified by the

package/src/app/[variants]/(main)/repos/[id]/@menu/default.tsx

@@ -1,8 +1,8 @@
 import { notFound } from 'next/navigation';
 import { Flexbox } from 'react-layout-kit';
 
+import { KnowledgeBaseModel } from '@/database/models/knowledgeBase';
 import { serverDB } from '@/database/server';
-import { KnowledgeBaseModel } from '@/database/server/models/knowledgeBase';
 
 import Head from './Head';
 import Menu from './Menu';

package/src/app/[variants]/(main)/repos/[id]/page.tsx

@@ -1,7 +1,7 @@
 import { redirect } from 'next/navigation';
 
+import { KnowledgeBaseModel } from '@/database/models/knowledgeBase';
 import { serverDB } from '@/database/server';
-import { KnowledgeBaseModel } from '@/database/server/models/knowledgeBase';
 import FileManager from '@/features/FileManager';
 import { PagePropsWithId } from '@/types/next';
 

package/src/components/Branding/ProductLogo/index.tsx

@@ -1,14 +1,14 @@
-import { LobeChat, LobeChatProps } from '@lobehub/ui/brand';
+import { LobeHub, LobeHubProps } from '@lobehub/ui/brand';
 import { memo } from 'react';
 
 import { isCustomBranding } from '@/const/version';
 
 import CustomLogo from './Custom';
 
-export const ProductLogo = memo<LobeChatProps>((props) => {
+export const ProductLogo = memo<LobeHubProps>((props) => {
   if (isCustomBranding) {
     return <CustomLogo {...props} />;
   }
 
-  return <LobeChat {...props} />;
+  return <LobeHub {...props} />;
 });

package/src/database/core/db-adaptor.ts

@@ -0,0 +1,6 @@
+// import { isDesktop } from '@/const/version';
+import { getDBInstance } from '@/database/core/web-server';
+
+// import { getPgliteInstance } from './electron';
+
+export const serverDB = getDBInstance();

package/src/database/{server/core → core}/dbForTest.ts

@@ -9,9 +9,9 @@ import ws from 'ws';
 
 import { serverDBEnv } from '@/config/db';
 
-import * as schema from '../../schemas';
+import * as schema from '../schemas';
 
-const migrationsFolder = join(__dirname, '../../migrations');
+const migrationsFolder = join(__dirname, '../migrations');
 
 export const getTestDBInstance = async () => {
   let connectionString = serverDBEnv.DATABASE_TEST_URL;

package/src/database/{server/core/db.ts → core/web-server.ts}

@@ -1,15 +1,16 @@
 import { Pool as NeonPool, neonConfig } from '@neondatabase/serverless';
-import { NeonDatabase, drizzle as neonDrizzle } from 'drizzle-orm/neon-serverless';
+import { drizzle as neonDrizzle } from 'drizzle-orm/neon-serverless';
 import { drizzle as nodeDrizzle } from 'drizzle-orm/node-postgres';
 import { Pool as NodePool } from 'pg';
 import ws from 'ws';
 
 import { serverDBEnv } from '@/config/db';
 import { isServerMode } from '@/const/version';
+import * as schema from '@/database/schemas';
 
-import * as schema from '../../schemas';
+import { LobeChatDatabase } from '../type';
 
-export const getDBInstance = (): NeonDatabase<typeof schema> => {
+export const getDBInstance = (): LobeChatDatabase => {
   if (!isServerMode) return {} as any;
 
   if (!serverDBEnv.KEY_VAULTS_SECRET) {
@@ -40,5 +41,3 @@ If you don't have it, please run \`openssl rand -base64 32\` to create one.
   const client = new NeonPool({ connectionString });
   return neonDrizzle(client, { schema });
 };
-
-export const serverDB = getDBInstance();

package/src/database/models/__tests__/_test_template.ts

@@ -2,12 +2,13 @@
 import { eq } from 'drizzle-orm/expressions';
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
-import { getTestDBInstance } from '@/database/server/core/dbForTest';
+import { LobeChatDatabase } from '@/database/type';
 
 import { sessionGroups, users } from '../../schemas';
-import { SessionGroupModel } from '../../server/models/sessionGroup';
+import { SessionGroupModel } from '../sessionGroup';
+import { getTestDB } from './_util';
 
-let serverDB = await getTestDBInstance();
+const serverDB: LobeChatDatabase = await getTestDB();
 
 const userId = 'session-group-model-test-user-id';
 const sessionGroupModel = new SessionGroupModel(serverDB, userId);

package/src/database/models/__tests__/_util.ts

@@ -1,8 +1,8 @@
 import { clientDB, initializeDB } from '@/database/client/db';
-import { getTestDBInstance } from '@/database/server/core/dbForTest';
+import { getTestDBInstance } from '@/database/core/dbForTest';
 import { LobeChatDatabase } from '@/database/type';
 
-export const isServerDBMode = process.env.TEST_SERVER_DB === '1';
+const isServerDBMode = process.env.TEST_SERVER_DB === '1';
 
 export const getTestDB = async () => {
   if (isServerDBMode) return await getTestDBInstance();

package/src/database/models/__tests__/agent.test.ts

@@ -14,7 +14,7 @@ import {
   sessions,
   users,
 } from '../../schemas';
-import { AgentModel } from '../../server/models/agent';
+import { AgentModel } from '../agent';
 import { getTestDB } from './_util';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/src/database/models/__tests__/aiModel.test.ts

@@ -6,7 +6,7 @@ import { LobeChatDatabase } from '@/database/type';
 import { AiProviderModelListItem } from '@/types/aiModel';
 
 import { AiModelSelectItem, NewAiModelItem, aiModels, users } from '../../schemas';
-import { AiModelModel } from '../../server/models/aiModel';
+import { AiModelModel } from '../aiModel';
 import { getTestDB } from './_util';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/src/database/models/__tests__/aiProvider.test.ts

@@ -6,7 +6,7 @@ import { LobeChatDatabase } from '@/database/type';
 import { ModelProvider } from '@/libs/agent-runtime';
 
 import { aiProviders, users } from '../../schemas';
-import { AiProviderModel } from '../../server/models/aiProvider';
+import { AiProviderModel } from '../aiProvider';
 import { getTestDB } from './_util';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/src/database/models/__tests__/asyncTask.test.ts

@@ -6,7 +6,7 @@ import { LobeChatDatabase } from '@/database/type';
 import { AsyncTaskStatus, AsyncTaskType } from '@/types/asyncTask';
 
 import { asyncTasks, users } from '../../schemas';
-import { ASYNC_TASK_TIMEOUT, AsyncTaskModel } from '../../server/models/asyncTask';
+import { ASYNC_TASK_TIMEOUT, AsyncTaskModel } from '../asyncTask';
 import { getTestDB } from './_util';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/src/database/models/__tests__/chunk.test.ts

@@ -6,7 +6,7 @@ import { LobeChatDatabase } from '@/database/type';
 import { uuid } from '@/utils/uuid';
 
 import { chunks, embeddings, fileChunks, files, unstructuredChunks, users } from '../../schemas';
-import { ChunkModel } from '../../server/models/chunk';
+import { ChunkModel } from '../chunk';
 import { getTestDB } from './_util';
 import { codeEmbedding, designThinkingQuery, designThinkingQuery2 } from './fixtures/embedding';
 

package/src/database/models/__tests__/file.test.ts

@@ -6,7 +6,7 @@ import { LobeChatDatabase } from '@/database/type';
 import { FilesTabs, SortType } from '@/types/files';
 
 import { files, globalFiles, knowledgeBaseFiles, knowledgeBases, users } from '../../schemas';
-import { FileModel } from '../../server/models/file';
+import { FileModel } from '../file';
 import { getTestDB } from './_util';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/src/database/models/__tests__/knowledgeBase.test.ts

@@ -13,7 +13,7 @@ import {
   knowledgeBases,
   users,
 } from '../../schemas';
-import { KnowledgeBaseModel } from '../../server/models/knowledgeBase';
+import { KnowledgeBaseModel } from '../knowledgeBase';
 import { getTestDB } from './_util';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/src/database/models/__tests__/message.test.ts

@@ -23,7 +23,7 @@ import {
   topics,
   users,
 } from '../../schemas';
-import { MessageModel } from '../../server/models/message';
+import { MessageModel } from '../message';
 import { codeEmbedding } from './fixtures/embedding';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/src/database/models/__tests__/plugin.test.ts

@@ -4,7 +4,7 @@ import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 import { LobeChatDatabase } from '@/database/type';
 
 import { NewInstalledPlugin, userInstalledPlugins, users } from '../../schemas';
-import { PluginModel } from '../../server/models/plugin';
+import { PluginModel } from '../plugin';
 import { getTestDB } from './_util';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/src/database/models/__tests__/session.test.ts

@@ -2,7 +2,6 @@ import { and, eq, inArray } from 'drizzle-orm/expressions';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { DEFAULT_AGENT_CONFIG } from '@/const/settings';
-import { getTestDBInstance } from '@/database/server/core/dbForTest';
 import { LobeChatDatabase } from '@/database/type';
 import { idGenerator } from '@/database/utils/idGenerator';
 
@@ -17,7 +16,7 @@ import {
   topics,
   users,
 } from '../../schemas';
-import { SessionModel } from '../../server/models/session';
+import { SessionModel } from '../session';
 import { getTestDB } from './_util';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/src/database/models/__tests__/sessionGroup.test.ts

@@ -5,7 +5,7 @@ import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 import { LobeChatDatabase } from '@/database/type';
 
 import { sessionGroups, users } from '../../schemas';
-import { SessionGroupModel } from '../../server/models/sessionGroup';
+import { SessionGroupModel } from '../sessionGroup';
 import { getTestDB } from './_util';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/src/database/models/__tests__/topic.test.ts

@@ -4,7 +4,7 @@ import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 import { LobeChatDatabase } from '@/database/type';
 
 import { messages, sessions, topics, users } from '../../schemas';
-import { CreateTopicParams, TopicModel } from '../../server/models/topic';
+import { CreateTopicParams, TopicModel } from '../topic';
 import { getTestDB } from './_util';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/src/database/{server/models → models}/_template.ts

@@ -2,7 +2,7 @@ import { and, desc, eq } from 'drizzle-orm/expressions';
 
 import { LobeChatDatabase } from '@/database/type';
 
-import { NewSessionGroup, SessionGroupItem, sessionGroups } from '../../schemas';
+import { NewSessionGroup, SessionGroupItem, sessionGroups } from '../schemas';
 
 export class TemplateModel {
   private userId: string;

package/src/database/{server/models → models}/aiModel.ts

@@ -9,7 +9,7 @@ import {
   ToggleAiModelEnableParams,
 } from '@/types/aiModel';
 
-import { AiModelSelectItem, NewAiModelItem, aiModels } from '../../schemas';
+import { AiModelSelectItem, NewAiModelItem, aiModels } from '../schemas';
 
 export class AiModelModel {
   private userId: string;

package/src/database/{server/models → models}/aiProvider.ts

@@ -13,7 +13,7 @@ import {
 } from '@/types/aiProvider';
 import { merge } from '@/utils/merge';
 
-import { AiProviderSelectItem, aiModels, aiProviders } from '../../schemas';
+import { AiProviderSelectItem, aiModels, aiProviders } from '../schemas';
 
 type DecryptUserKeyVaults = (encryptKeyVaultsStr: string | null) => Promise<any>;
 

package/src/database/{server/models → models}/asyncTask.ts

@@ -8,7 +8,7 @@ import {
   AsyncTaskType,
 } from '@/types/asyncTask';
 
-import { AsyncTaskSelectItem, NewAsyncTaskItem, asyncTasks } from '../../schemas';
+import { AsyncTaskSelectItem, NewAsyncTaskItem, asyncTasks } from '../schemas';
 
 // set timeout to about 5 minutes, and give 2s padding time
 export const ASYNC_TASK_TIMEOUT = 298 * 1000;

package/src/database/{server/models → models}/chunk.ts

@@ -13,7 +13,7 @@ import {
   fileChunks,
   files,
   unstructuredChunks,
-} from '../../schemas';
+} from '../schemas';
 
 export class ChunkModel {
   private userId: string;

package/src/database/{server/models → models}/embedding.ts

@@ -3,7 +3,7 @@ import { and, eq } from 'drizzle-orm/expressions';
 
 import { LobeChatDatabase } from '@/database/type';
 
-import { NewEmbeddingsItem, embeddings } from '../../schemas';
+import { NewEmbeddingsItem, embeddings } from '../schemas';
 
 export class EmbeddingModel {
   private userId: string;

package/src/database/{server/models → models}/file.ts

@@ -15,7 +15,7 @@ import {
   files,
   globalFiles,
   knowledgeBaseFiles,
-} from '../../schemas';
+} from '../schemas';
 
 export class FileModel {
   private readonly userId: string;

package/src/database/{server/models → models}/knowledgeBase.ts

@@ -3,7 +3,7 @@ import { and, desc, eq, inArray } from 'drizzle-orm/expressions';
 import { LobeChatDatabase } from '@/database/type';
 import { KnowledgeBaseItem } from '@/types/knowledgeBase';
 
-import { NewKnowledgeBase, knowledgeBaseFiles, knowledgeBases } from '../../schemas';
+import { NewKnowledgeBase, knowledgeBaseFiles, knowledgeBases } from '../schemas';
 
 export class KnowledgeBaseModel {
   private userId: string;

package/src/database/{server/models → models}/message.ts

@@ -40,7 +40,7 @@ import {
   messageTranslates,
   messages,
   messagesFiles,
-} from '../../schemas';
+} from '../schemas';
 
 export interface QueryMessageParams {
   current?: number;

package/src/database/{server/models → models}/plugin.ts

@@ -2,7 +2,7 @@ import { and, desc, eq } from 'drizzle-orm/expressions';
 
 import { LobeChatDatabase } from '@/database/type';
 
-import { InstalledPluginItem, NewInstalledPlugin, userInstalledPlugins } from '../../schemas';
+import { InstalledPluginItem, NewInstalledPlugin, userInstalledPlugins } from '../schemas';
 
 export class PluginModel {
   private userId: string;
@@ -31,7 +31,9 @@ export class PluginModel {
   delete = async (id: string) => {
     return this.db
       .delete(userInstalledPlugins)
-      .where(and(eq(userInstalledPlugins.identifier, id), eq(userInstalledPlugins.userId, this.userId)));
+      .where(
+        and(eq(userInstalledPlugins.identifier, id), eq(userInstalledPlugins.userId, this.userId)),
+      );
   };
 
   deleteAll = async () => {
@@ -56,7 +58,10 @@ export class PluginModel {
 
   findById = async (id: string) => {
     return this.db.query.userInstalledPlugins.findFirst({
-      where: and(eq(userInstalledPlugins.identifier, id), eq(userInstalledPlugins.userId, this.userId)),
+      where: and(
+        eq(userInstalledPlugins.identifier, id),
+        eq(userInstalledPlugins.userId, this.userId),
+      ),
     });
   };
 
@@ -64,6 +69,8 @@ export class PluginModel {
     return this.db
       .update(userInstalledPlugins)
       .set({ ...value, updatedAt: new Date() })
-      .where(and(eq(userInstalledPlugins.identifier, id), eq(userInstalledPlugins.userId, this.userId)));
+      .where(
+        and(eq(userInstalledPlugins.identifier, id), eq(userInstalledPlugins.userId, this.userId)),
+      );
   };
 }

package/src/database/{server/models → models}/session.ts

@@ -27,7 +27,7 @@ import {
   sessionGroups,
   sessions,
   topics,
-} from '../../schemas';
+} from '../schemas';
 
 export class SessionModel {
   private userId: string;

package/src/database/{server/models → models}/sessionGroup.ts

@@ -3,7 +3,7 @@ import { and, asc, desc, eq } from 'drizzle-orm/expressions';
 import { LobeChatDatabase } from '@/database/type';
 import { idGenerator } from '@/database/utils/idGenerator';
 
-import { SessionGroupItem, sessionGroups } from '../../schemas';
+import { SessionGroupItem, sessionGroups } from '../schemas';
 
 export class SessionGroupModel {
   private userId: string;

package/src/database/{server/models → models}/thread.ts

@@ -3,7 +3,7 @@ import { and, desc, eq } from 'drizzle-orm/expressions';
 import { LobeChatDatabase } from '@/database/type';
 import { CreateThreadParams, ThreadStatus } from '@/types/topic';
 
-import { ThreadItem, threads } from '../../schemas';
+import { ThreadItem, threads } from '../schemas';
 
 const queryColumns = {
   createdAt: threads.createdAt,

package/src/database/{server/models → models}/topic.ts

@@ -12,7 +12,7 @@ import { idGenerator } from '@/database/utils/idGenerator';
 import { MessageItem } from '@/types/message';
 import { TopicRankItem } from '@/types/topic';
 
-import { TopicItem, messages, topics } from '../../schemas';
+import { TopicItem, messages, topics } from '../schemas';
 
 export interface CreateTopicParams {
   favorite?: boolean;

package/src/database/{server/models → models}/user.ts

@@ -17,7 +17,7 @@ import {
   nextauthAccounts,
   userSettings,
   users,
-} from '../../schemas';
+} from '../schemas';
 
 type DecryptUserKeyVaults = (
   encryptKeyVaultsStr: string | null,

package/src/database/repositories/aiInfra/index.ts

@@ -2,8 +2,8 @@ import { isEmpty } from 'lodash-es';
 import pMap from 'p-map';
 
 import { DEFAULT_MODEL_PROVIDER_LIST } from '@/config/modelProviders';
-import { AiModelModel } from '@/database/server/models/aiModel';
-import { AiProviderModel } from '@/database/server/models/aiProvider';
+import { AiModelModel } from '@/database/models/aiModel';
+import { AiProviderModel } from '@/database/models/aiProvider';
 import { LobeChatDatabase } from '@/database/type';
 import {
   AIChatModelCard,

package/src/database/repositories/dataImporter/deprecated/__tests__/index.test.ts

@@ -2,6 +2,7 @@
 import { eq, inArray } from 'drizzle-orm/expressions';
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 
+import { getTestDBInstance } from '@/database/core/dbForTest';
 import {
   agents,
   agentsToSessions,
@@ -11,7 +12,6 @@ import {
   topics,
   users,
 } from '@/database/schemas';
-import { getTestDBInstance } from '@/database/server/core/dbForTest';
 import { CURRENT_CONFIG_VERSION } from '@/migrations';
 import { ImporterEntryData } from '@/types/importer';
 

package/src/database/server/index.ts

@@ -1 +1 @@
-export { serverDB } from './core/db';
+export { serverDB } from '../core/db-adaptor';

package/src/database/server/models/__tests__/nextauth.test.ts

@@ -8,6 +8,7 @@ import type {
 import { eq } from 'drizzle-orm/expressions';
 import { afterAll, afterEach, beforeAll, beforeEach, describe, expect, it, vi } from 'vitest';
 
+import { getTestDBInstance } from '@/database/core/dbForTest';
 import {
   nextauthAccounts,
   nextauthAuthenticators,
@@ -15,7 +16,6 @@ import {
   nextauthVerificationTokens,
   users,
 } from '@/database/schemas';
-import { getTestDBInstance } from '@/database/server/core/dbForTest';
 import { LobeChatDatabase } from '@/database/type';
 import { LobeNextAuthDbAdapter } from '@/libs/next-auth/adapter';