@lobehub/lobehub 2.0.0-next.195 → 2.0.0-next.197

package/CHANGELOG.md

@@ -2,6 +2,64 @@
 
 # Changelog
 
+## [Version 2.0.0-next.197](https://github.com/lobehub/lobe-chat/compare/v2.0.0-next.196...v2.0.0-next.197)
+
+<sup>Released on **2026-01-03**</sup>
+
+#### ♻ Code Refactoring
+
+- **misc**: Remove client db and refactor test.
+
+#### 🐛 Bug Fixes
+
+- **misc**: Fix file upload issue.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### Code refactoring
+
+- **misc**: Remove client db and refactor test, closes [#11123](https://github.com/lobehub/lobe-chat/issues/11123) ([bb2799d](https://github.com/lobehub/lobe-chat/commit/bb2799d))
+
+#### What's fixed
+
+- **misc**: Fix file upload issue, closes [#11122](https://github.com/lobehub/lobe-chat/issues/11122) ([1ae327a](https://github.com/lobehub/lobe-chat/commit/1ae327a))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
+## [Version 2.0.0-next.196](https://github.com/lobehub/lobe-chat/compare/v2.0.0-next.195...v2.0.0-next.196)
+
+<sup>Released on **2026-01-03**</sup>
+
+#### ♻ Code Refactoring
+
+- **misc**: Refactor to remove access code.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### Code refactoring
+
+- **misc**: Refactor to remove access code, closes [#11120](https://github.com/lobehub/lobe-chat/issues/11120) ([0e9f98c](https://github.com/lobehub/lobe-chat/commit/0e9f98c))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ## [Version 2.0.0-next.195](https://github.com/lobehub/lobe-chat/compare/v2.0.0-next.194...v2.0.0-next.195)
 
 <sup>Released on **2026-01-03**</sup>

package/changelog/v1.json

@@ -1,4 +1,25 @@
 [
+  {
+    "children": {
+      "improvements": [
+        "Remove client db and refactor test."
+      ],
+      "fixes": [
+        "Fix file upload issue."
+      ]
+    },
+    "date": "2026-01-03",
+    "version": "2.0.0-next.197"
+  },
+  {
+    "children": {
+      "improvements": [
+        "Refactor to remove access code."
+      ]
+    },
+    "date": "2026-01-03",
+    "version": "2.0.0-next.196"
+  },
   {
     "children": {
       "fixes": [

package/locales/ar/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "الأسئلة الافتتاحية",
   "settingOpening.title": "إعدادات البداية",
   "settingPlugin.title": "قائمة المهارات",
-  "settingSystem.accessCode.desc": "تم تفعيل الوصول المشفر من قبل المسؤول",
-  "settingSystem.accessCode.placeholder": "أدخل كلمة مرور الوصول",
-  "settingSystem.accessCode.title": "كلمة مرور الوصول",
   "settingSystem.oauth.info.desc": "تم تسجيل الدخول",
   "settingSystem.oauth.info.title": "معلومات الحساب",
   "settingSystem.oauth.signin.action": "تسجيل الدخول",

package/locales/bg-BG/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "Начални въпроси",
   "settingOpening.title": "Настройки за начало",
   "settingPlugin.title": "Списък с умения",
-  "settingSystem.accessCode.desc": "Достъпът с шифроване е активиран от администратора",
-  "settingSystem.accessCode.placeholder": "Въведете парола за достъп",
-  "settingSystem.accessCode.title": "Парола за достъп",
   "settingSystem.oauth.info.desc": "Вход изпълнен",
   "settingSystem.oauth.info.title": "Информация за акаунта",
   "settingSystem.oauth.signin.action": "Вход",

package/locales/de-DE/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "Einstiegsfragen",
   "settingOpening.title": "Begrüßungseinstellungen",
   "settingPlugin.title": "Fähigkeitenliste",
-  "settingSystem.accessCode.desc": "Zugriff mit Verschlüsselung ist vom Administrator aktiviert",
-  "settingSystem.accessCode.placeholder": "Zugangspasswort eingeben",
-  "settingSystem.accessCode.title": "Zugangspasswort",
   "settingSystem.oauth.info.desc": "Angemeldet",
   "settingSystem.oauth.info.title": "Kontoinformationen",
   "settingSystem.oauth.signin.action": "Anmelden",

package/locales/en-US/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "Opening Questions",
   "settingOpening.title": "Opening Settings",
   "settingPlugin.title": "Skill List",
-  "settingSystem.accessCode.desc": "Encryption access is enabled by the administrator",
-  "settingSystem.accessCode.placeholder": "Enter access password",
-  "settingSystem.accessCode.title": "Access Password",
   "settingSystem.oauth.info.desc": "Logged in",
   "settingSystem.oauth.info.title": "Account Information",
   "settingSystem.oauth.signin.action": "Sign In",

package/locales/es-ES/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "Preguntas de Apertura",
   "settingOpening.title": "Configuración de Apertura",
   "settingPlugin.title": "Lista de Habilidades",
-  "settingSystem.accessCode.desc": "El administrador ha habilitado el acceso cifrado",
-  "settingSystem.accessCode.placeholder": "Introduce la contraseña de acceso",
-  "settingSystem.accessCode.title": "Contraseña de Acceso",
   "settingSystem.oauth.info.desc": "Sesión iniciada",
   "settingSystem.oauth.info.title": "Información de la Cuenta",
   "settingSystem.oauth.signin.action": "Iniciar Sesión",

package/locales/fa-IR/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "سؤالات آغازین",
   "settingOpening.title": "تنظیمات آغازین",
   "settingPlugin.title": "فهرست مهارت‌ها",
-  "settingSystem.accessCode.desc": "دسترسی رمزگذاری‌شده توسط مدیر فعال شده است",
-  "settingSystem.accessCode.placeholder": "رمز عبور دسترسی را وارد کنید",
-  "settingSystem.accessCode.title": "رمز عبور دسترسی",
   "settingSystem.oauth.info.desc": "وارد شده‌اید",
   "settingSystem.oauth.info.title": "اطلاعات حساب",
   "settingSystem.oauth.signin.action": "ورود",

package/locales/fr-FR/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "Questions d’accueil",
   "settingOpening.title": "Paramètres d’accueil",
   "settingPlugin.title": "Liste des compétences",
-  "settingSystem.accessCode.desc": "L’accès chiffré est activé par l’administrateur",
-  "settingSystem.accessCode.placeholder": "Saisir le mot de passe d’accès",
-  "settingSystem.accessCode.title": "Mot de passe d’accès",
   "settingSystem.oauth.info.desc": "Connecté",
   "settingSystem.oauth.info.title": "Informations du compte",
   "settingSystem.oauth.signin.action": "Se connecter",

package/locales/it-IT/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "Domande di Apertura",
   "settingOpening.title": "Impostazioni di Apertura",
   "settingPlugin.title": "Elenco Competenze",
-  "settingSystem.accessCode.desc": "L'accesso crittografato è abilitato dall'amministratore",
-  "settingSystem.accessCode.placeholder": "Inserisci la password di accesso",
-  "settingSystem.accessCode.title": "Password di Accesso",
   "settingSystem.oauth.info.desc": "Accesso effettuato",
   "settingSystem.oauth.info.title": "Informazioni Account",
   "settingSystem.oauth.signin.action": "Accedi",

package/locales/ja-JP/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "オープニング質問",
   "settingOpening.title": "オープニング設定",
   "settingPlugin.title": "スキルリスト",
-  "settingSystem.accessCode.desc": "管理者が暗号化アクセスを有効にしています",
-  "settingSystem.accessCode.placeholder": "アクセスコードを入力してください",
-  "settingSystem.accessCode.title": "アクセスコード",
   "settingSystem.oauth.info.desc": "ログイン済み",
   "settingSystem.oauth.info.title": "アカウント情報",
   "settingSystem.oauth.signin.action": "ログイン",

package/locales/ko-KR/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "오프닝 질문",
   "settingOpening.title": "오프닝 설정",
   "settingPlugin.title": "기능 목록",
-  "settingSystem.accessCode.desc": "관리자가 암호화된 액세스를 활성화했습니다",
-  "settingSystem.accessCode.placeholder": "액세스 암호를 입력하세요",
-  "settingSystem.accessCode.title": "액세스 암호",
   "settingSystem.oauth.info.desc": "로그인됨",
   "settingSystem.oauth.info.title": "계정 정보",
   "settingSystem.oauth.signin.action": "로그인",

package/locales/nl-NL/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "Openingsvragen",
   "settingOpening.title": "Openingsinstellingen",
   "settingPlugin.title": "Vaardighedenlijst",
-  "settingSystem.accessCode.desc": "Versleutelde toegang is ingeschakeld door de beheerder",
-  "settingSystem.accessCode.placeholder": "Voer toegangswachtwoord in",
-  "settingSystem.accessCode.title": "Toegangswachtwoord",
   "settingSystem.oauth.info.desc": "Ingelogd",
   "settingSystem.oauth.info.title": "Accountinformatie",
   "settingSystem.oauth.signin.action": "Inloggen",

package/locales/pl-PL/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "Pytania wprowadzające",
   "settingOpening.title": "Ustawienia powitania",
   "settingPlugin.title": "Lista umiejętności",
-  "settingSystem.accessCode.desc": "Dostęp szyfrowany został włączony przez administratora",
-  "settingSystem.accessCode.placeholder": "Wprowadź hasło dostępu",
-  "settingSystem.accessCode.title": "Hasło dostępu",
   "settingSystem.oauth.info.desc": "Zalogowano",
   "settingSystem.oauth.info.title": "Informacje o koncie",
   "settingSystem.oauth.signin.action": "Zaloguj się",

package/locales/pt-BR/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "Perguntas de Abertura",
   "settingOpening.title": "Configurações de Abertura",
   "settingPlugin.title": "Lista de Habilidades",
-  "settingSystem.accessCode.desc": "O acesso criptografado foi ativado pelo administrador",
-  "settingSystem.accessCode.placeholder": "Digite a senha de acesso",
-  "settingSystem.accessCode.title": "Senha de Acesso",
   "settingSystem.oauth.info.desc": "Conectado",
   "settingSystem.oauth.info.title": "Informações da Conta",
   "settingSystem.oauth.signin.action": "Entrar",

package/locales/ru-RU/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "Вводные вопросы",
   "settingOpening.title": "Настройки приветствия",
   "settingPlugin.title": "Список навыков",
-  "settingSystem.accessCode.desc": "Доступ по паролю включён администратором",
-  "settingSystem.accessCode.placeholder": "Введите пароль доступа",
-  "settingSystem.accessCode.title": "Пароль доступа",
   "settingSystem.oauth.info.desc": "Вы вошли в систему",
   "settingSystem.oauth.info.title": "Информация об аккаунте",
   "settingSystem.oauth.signin.action": "Войти",

package/locales/tr-TR/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "Açılış Soruları",
   "settingOpening.title": "Açılış Ayarları",
   "settingPlugin.title": "Yetenek Listesi",
-  "settingSystem.accessCode.desc": "Şifreli erişim, yönetici tarafından etkinleştirilmiştir",
-  "settingSystem.accessCode.placeholder": "Erişim şifresini girin",
-  "settingSystem.accessCode.title": "Erişim Şifresi",
   "settingSystem.oauth.info.desc": "Giriş yapıldı",
   "settingSystem.oauth.info.title": "Hesap Bilgileri",
   "settingSystem.oauth.signin.action": "Giriş Yap",

package/locales/vi-VN/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "Câu Hỏi Mở Đầu",
   "settingOpening.title": "Cài Đặt Mở Đầu",
   "settingPlugin.title": "Danh Sách Kỹ Năng",
-  "settingSystem.accessCode.desc": "Mã truy cập được mã hóa do quản trị viên bật",
-  "settingSystem.accessCode.placeholder": "Nhập mật khẩu truy cập",
-  "settingSystem.accessCode.title": "Mật Khẩu Truy Cập",
   "settingSystem.oauth.info.desc": "Đã đăng nhập",
   "settingSystem.oauth.info.title": "Thông Tin Tài Khoản",
   "settingSystem.oauth.signin.action": "Đăng Nhập",

package/locales/zh-CN/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "开场问题",
   "settingOpening.title": "开场设置",
   "settingPlugin.title": "技能列表",
-  "settingSystem.accessCode.desc": "管理员已开启加密访问",
-  "settingSystem.accessCode.placeholder": "请输入访问密码",
-  "settingSystem.accessCode.title": "访问密码",
   "settingSystem.oauth.info.desc": "已登录",
   "settingSystem.oauth.info.title": "账户信息",
   "settingSystem.oauth.signin.action": "登录",

package/locales/zh-TW/setting.json

@@ -384,9 +384,6 @@
   "settingOpening.openingQuestions.title": "開場問題",
   "settingOpening.title": "開場設定",
   "settingPlugin.title": "插件列表",
-  "settingSystem.accessCode.desc": "管理員已開啟加密訪問",
-  "settingSystem.accessCode.placeholder": "請輸入訪問密碼",
-  "settingSystem.accessCode.title": "訪問密碼",
   "settingSystem.oauth.info.desc": "已登錄",
   "settingSystem.oauth.info.title": "帳戶資訊",
   "settingSystem.oauth.signin.action": "登錄",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/lobehub",
-  "version": "2.0.0-next.195",
+  "version": "2.0.0-next.197",
   "description": "LobeHub - an open-source,comprehensive AI Agent framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.",
   "keywords": [
     "framework",
@@ -42,8 +42,7 @@
     "build:electron": "cross-env NODE_OPTIONS=--max-old-space-size=8192 NEXT_PUBLIC_IS_DESKTOP_APP=1 tsx scripts/electronWorkflow/buildNextApp.mts",
     "build:vercel": "npm run prebuild && cross-env NODE_OPTIONS=--max-old-space-size=6144 next build --webpack",
     "clean:node_modules": "bash -lc 'set -e; echo \"Removing all node_modules...\"; rm -rf node_modules; pnpm -r exec rm -rf node_modules; rm -rf apps/desktop/node_modules; echo \"All node_modules removed.\"'",
-    "db:generate": "drizzle-kit generate && npm run db:generate-client && npm run workflow:dbml",
-    "db:generate-client": "tsx ./scripts/migrateClientDB/compile-migrations.ts",
+    "db:generate": "drizzle-kit generate && npm run workflow:dbml",
     "db:migrate": "MIGRATION_DB=1 tsx ./scripts/migrateServerDB/index.ts",
     "db:studio": "drizzle-kit studio",
     "db:visualize": "dbdocs build docs/development/database-schema.dbml --project lobe-chat",

package/packages/const/src/fetch.ts

@@ -6,8 +6,6 @@ export const USE_AZURE_OPENAI = 'X-use-azure-openai';
 
 export const AZURE_OPENAI_API_VERSION = 'X-azure-openai-api-version';
 
-export const LOBE_CHAT_ACCESS_CODE = 'X-lobe-chat-access-code';
-
 export const OAUTH_AUTHORIZED = 'X-oauth-authorized';
 
 /**
@@ -16,7 +14,6 @@ export const OAUTH_AUTHORIZED = 'X-oauth-authorized';
 export const getOpenAIAuthFromRequest = (req: Request) => {
   const apiKey = req.headers.get(OPENAI_API_KEY_HEADER_KEY);
   const endpoint = req.headers.get(OPENAI_END_POINT);
-  const accessCode = req.headers.get(LOBE_CHAT_ACCESS_CODE);
   const useAzureStr = req.headers.get(USE_AZURE_OPENAI);
   const apiVersion = req.headers.get(AZURE_OPENAI_API_VERSION);
   const oauthAuthorizedStr = req.headers.get(OAUTH_AUTHORIZED);
@@ -25,5 +22,5 @@ export const getOpenAIAuthFromRequest = (req: Request) => {
   const oauthAuthorized = !!oauthAuthorizedStr;
   const useAzure = !!useAzureStr;
 
-  return { accessCode, apiKey, apiVersion, endpoint, oauthAuthorized, useAzure, userId };
+  return { apiKey, apiVersion, endpoint, oauthAuthorized, useAzure, userId };
 };

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

@@ -0,0 +1,50 @@
+import { PGlite } from '@electric-sql/pglite';
+import { vector } from '@electric-sql/pglite/vector';
+import { drizzle as pgliteDrizzle } from 'drizzle-orm/pglite';
+import { migrate as pgliteMigrate } from 'drizzle-orm/pglite/migrator';
+import { drizzle as nodeDrizzle } from 'drizzle-orm/node-postgres';
+import { migrate as nodeMigrate } from 'drizzle-orm/node-postgres/migrator';
+import { join } from 'node:path';
+import { Pool as NodePool } from 'pg';
+
+import { serverDBEnv } from '@/config/db';
+
+import * as schema from '../schemas';
+import { LobeChatDatabase } from '../type';
+
+const migrationsFolder = join(__dirname, '../../migrations');
+
+const isServerDBMode = process.env.TEST_SERVER_DB === '1';
+
+let testClientDB: ReturnType<typeof pgliteDrizzle<typeof schema>> | null = null;
+let testServerDB: ReturnType<typeof nodeDrizzle<typeof schema>> | null = null;
+
+export const getTestDB = async (): Promise<LobeChatDatabase> => {
+  // Server DB mode (node-postgres)
+  if (isServerDBMode) {
+    if (testServerDB) return testServerDB as unknown as LobeChatDatabase;
+
+    const connectionString = serverDBEnv.DATABASE_TEST_URL;
+
+    if (!connectionString) {
+      throw new Error('DATABASE_TEST_URL is not set');
+    }
+
+    const client = new NodePool({ connectionString });
+    testServerDB = nodeDrizzle(client, { schema });
+
+    await nodeMigrate(testServerDB, { migrationsFolder });
+
+    return testServerDB as unknown as LobeChatDatabase;
+  }
+
+  // Client DB mode (PGlite)
+  if (testClientDB) return testClientDB as unknown as LobeChatDatabase;
+
+  const pglite = new PGlite({ extensions: { vector } });
+  testClientDB = pgliteDrizzle({ client: pglite, schema });
+
+  await pgliteMigrate(testClientDB, { migrationsFolder });
+
+  return testClientDB as unknown as LobeChatDatabase;
+};

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

@@ -2,10 +2,10 @@
 import { eq } from 'drizzle-orm';
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
+import { getTestDB } from '../../core/getTestDB';
 import { sessionGroups, users } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { SessionGroupModel } from '../sessionGroup';
-import { getTestDB } from './_util';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

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

@@ -16,7 +16,7 @@ import {
 } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { AgentModel } from '../agent';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

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

@@ -6,7 +6,7 @@ import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 import { AiModelSelectItem, NewAiModelItem, aiModels, users } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { AiModelModel } from '../aiModel';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

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

@@ -8,7 +8,7 @@ import { sleep } from '@/utils/sleep';
 import { aiProviders, users } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { AiProviderModel } from '../aiProvider';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

package/packages/database/src/models/__tests__/apiKey.test.ts

@@ -5,7 +5,7 @@ import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 import { apiKeys, users } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { ApiKeyModel } from '../apiKey';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

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

@@ -7,7 +7,7 @@ import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 import { asyncTasks, users } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { AsyncTaskModel } from '../asyncTask';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

package/packages/database/src/models/__tests__/chatGroup.test.ts

@@ -12,7 +12,7 @@ import {
   users,
 } from '../../schemas';
 import { ChatGroupModel } from '../chatGroup';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const userId = 'test-user';
 const otherUserId = 'other-user';

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

@@ -6,7 +6,7 @@ import { LobeChatDatabase } from '../../type';import { uuid } from '@/utils/uuid
 
 import { chunks, embeddings, fileChunks, files, unstructuredChunks, users } from '../../schemas';
 import { ChunkModel } from '../chunk';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 import { codeEmbedding, designThinkingQuery, designThinkingQuery2 } from './fixtures/embedding';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/packages/database/src/models/__tests__/document.test.ts

@@ -5,7 +5,7 @@ import { documents, files, users } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { DocumentModel } from '../document';
 import { FileModel } from '../file';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

package/packages/database/src/models/__tests__/drizzleMigration.test.ts

@@ -3,7 +3,7 @@ import { beforeEach, describe, expect, it } from 'vitest';
 
 import { LobeChatDatabase } from '../../type';
 import { DrizzleMigrationModel } from '../drizzleMigration';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

package/packages/database/src/models/__tests__/embedding.test.ts

@@ -4,7 +4,7 @@ import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 import { chunks, embeddings, users } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { EmbeddingModel } from '../embedding';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 import { designThinkingQuery } from './fixtures/embedding';
 
 const userId = 'embedding-user-test';

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

@@ -15,7 +15,7 @@ import {
 } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { FileModel } from '../file';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

package/packages/database/src/models/__tests__/generation.test.ts

@@ -15,7 +15,7 @@ import {
 } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { GenerationModel } from '../generation';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

package/packages/database/src/models/__tests__/generationBatch.test.ts

@@ -13,7 +13,7 @@ import {
 } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { GenerationBatchModel } from '../generationBatch';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

package/packages/database/src/models/__tests__/generationTopic.test.ts

@@ -6,7 +6,7 @@ import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 import { generationBatches, generationTopics, generations, users } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { GenerationTopicModel } from '../generationTopic';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 // Mock FileService
 const mockGetFullFileUrl = vi.fn();

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

@@ -14,7 +14,7 @@ import {
 } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { KnowledgeBaseModel } from '../knowledgeBase';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

package/packages/database/src/models/__tests__/messages/message.create.test.ts

@@ -19,7 +19,7 @@ import {
 } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
 import { MessageModel } from '../../message';
-import { getTestDB } from '../_util';
+import { getTestDB } from '../../../core/getTestDB';
 import { codeEmbedding } from '../fixtures/embedding';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/packages/database/src/models/__tests__/messages/message.delete.test.ts

@@ -18,7 +18,7 @@ import {
 } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
 import { MessageModel } from '../../message';
-import { getTestDB } from '../_util';
+import { getTestDB } from '../../../core/getTestDB';
 import { codeEmbedding } from '../fixtures/embedding';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/packages/database/src/models/__tests__/messages/message.query.test.ts

@@ -28,7 +28,7 @@ import {
 } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
 import { MessageModel } from '../../message';
-import { getTestDB } from '../_util';
+import { getTestDB } from '../../../core/getTestDB';
 import { codeEmbedding } from '../fixtures/embedding';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/packages/database/src/models/__tests__/messages/message.stats.test.ts

@@ -7,7 +7,7 @@ import { uuid } from '@/utils/uuid';
 import { embeddings, files, messageQueries, messages, sessions, users } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
 import { MessageModel } from '../../message';
-import { getTestDB } from '../_util';
+import { getTestDB } from '../../../core/getTestDB';
 import { codeEmbedding } from '../fixtures/embedding';
 
 const serverDB: LobeChatDatabase = await getTestDB();

package/packages/database/src/models/__tests__/messages/message.thread-query.test.ts

@@ -4,7 +4,7 @@ import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 import { agents, messages, sessions, threads, topics, users } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
 import { MessageModel } from '../../message';
-import { getTestDB } from '../_util';
+import { getTestDB } from '../../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

package/packages/database/src/models/__tests__/messages/message.update.test.ts

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