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

package/CHANGELOG.md

@@ -2,6 +2,39 @@
 
 # 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>

package/changelog/v1.json

@@ -1,4 +1,16 @@
 [
+  {
+    "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": [

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/lobehub",
-  "version": "2.0.0-next.196",
+  "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/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();

package/packages/database/src/models/__tests__/messages/messageWithTask.test.ts

@@ -5,7 +5,7 @@ import { ThreadStatus, ThreadType } from '@/types/index';
 import { 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 userId = 'message-task-user-test';
 const sessionId = 'message-task-session';

package/packages/database/src/models/__tests__/messages/queryWithMessageGroup.perf.test.ts

@@ -6,7 +6,7 @@ import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 import { messageGroups, messages, topics, users } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
 import { MessageModel } from '../../message';
-import { getTestDB } from '../_util';
+import { getTestDB } from '../../../core/getTestDB';
 
 const userId = 'message-query-perf-test-user';
 const topicId = 'perf-test-topic-1';

package/packages/database/src/models/__tests__/messages/queryWithMessageGroup.test.ts

@@ -6,7 +6,7 @@ import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 import { messageGroups, messages, topics, users } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
 import { MessageModel } from '../../message';
-import { getTestDB } from '../_util';
+import { getTestDB } from '../../../core/getTestDB';
 
 const userId = 'message-query-test-user';
 const topicId = 'test-topic-1';

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

@@ -4,7 +4,7 @@ import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 import { oauthHandoffs } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { OAuthHandoffModel } from '../oauthHandoff';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 const oauthHandoffModel = new OAuthHandoffModel(serverDB);

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

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

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

@@ -17,7 +17,7 @@ import {
 import { LobeChatDatabase } from '../../type';
 import { idGenerator } from '../../utils/idGenerator';
 import { SessionModel } from '../session';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

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

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

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

@@ -5,7 +5,7 @@ import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 import { sessions, threads, topics, users } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { ThreadModel } from '../thread';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const userId = 'thread-user-test';
 const otherUserId = 'other-user-test';

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

@@ -5,7 +5,7 @@ import { documents, sessions, topicDocuments, topics, users } from '../../schema
 import { LobeChatDatabase } from '../../type';
 import { DocumentModel } from '../document';
 import { TopicDocumentModel } from '../topicDocument';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

package/packages/database/src/models/__tests__/topics/topic.create.test.ts

@@ -4,7 +4,7 @@ import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 import { agents, messages, sessions, topics, users } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
 import { CreateTopicParams, TopicModel } from '../../topic';
-import { getTestDB } from '../_util';
+import { getTestDB } from '../../../core/getTestDB';
 
 const userId = 'topic-create-user';
 const userId2 = 'topic-create-user-2';

package/packages/database/src/models/__tests__/topics/topic.delete.test.ts

@@ -12,7 +12,7 @@ import {
 } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
 import { TopicModel } from '../../topic';
-import { getTestDB } from '../_util';
+import { getTestDB } from '../../../core/getTestDB';
 
 const userId = 'topic-delete-user';
 const userId2 = 'topic-delete-user-2';

package/packages/database/src/models/__tests__/topics/topic.query.test.ts

@@ -11,7 +11,7 @@ import {
 } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
 import { TopicModel } from '../../topic';
-import { getTestDB } from '../_util';
+import { getTestDB } from '../../../core/getTestDB';
 
 const userId = 'topic-query-user';
 const userId2 = 'topic-query-user-2';

package/packages/database/src/models/__tests__/topics/topic.stats.test.ts

@@ -3,7 +3,7 @@ import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 import { agents, agentsToSessions, messages, sessions, topics, users } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
 import { TopicModel } from '../../topic';
-import { getTestDB } from '../_util';
+import { getTestDB } from '../../../core/getTestDB';
 
 const userId = 'topic-stats-user';
 const userId2 = 'topic-stats-user-2';

package/packages/database/src/models/__tests__/topics/topic.update.test.ts

@@ -3,7 +3,7 @@ import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 import { sessions, topics, users } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
 import { TopicModel } from '../../topic';
-import { getTestDB } from '../_util';
+import { getTestDB } from '../../../core/getTestDB';
 
 const userId = 'topic-update-user';
 const sessionId = 'topic-update-session';

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

@@ -5,7 +5,7 @@ import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 import { nextauthAccounts, userSettings, users } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { ListUsersForMemoryExtractorCursor, UserModel, UserNotFoundError } from '../user';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const userId = 'user-model-test';
 const otherUserId = 'other-user-test';

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

@@ -24,7 +24,7 @@ import {
   CreateUserMemoryPreferenceParams,
   UserMemoryModel,
 } from '../userMemory';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 const serverDB: LobeChatDatabase = await getTestDB();
 

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

@@ -9,7 +9,7 @@ import { idGenerator } from '@/database/utils/idGenerator';
 import { userMemoriesIdentities, users } from '../../schemas';
 import { LobeChatDatabase } from '../../type';
 import { UserMemoryIdentityModel } from '../userMemory/identity';
-import { getTestDB } from './_util';
+import { getTestDB } from '../../core/getTestDB';
 
 // Helper to generate unique identity IDs
 const genIdentityId = () => `mem_${nanoid(12)}`;

package/packages/database/src/models/userMemory/__tests__/context.test.ts

@@ -3,7 +3,7 @@ import { beforeEach, describe, expect, it } from 'vitest';
 
 import { NewUserMemoryContext, userMemories, userMemoriesContexts, users } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
-import { getTestDB } from '../../__tests__/_util';
+import { getTestDB } from '../../../core/getTestDB';
 import { UserMemoryContextModel } from '../context';
 
 const userId = 'context-test-user';

package/packages/database/src/models/userMemory/__tests__/experience.test.ts

@@ -8,7 +8,7 @@ import {
   users,
 } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
-import { getTestDB } from '../../__tests__/_util';
+import { getTestDB } from '../../../core/getTestDB';
 import { UserMemoryExperienceModel } from '../experience';
 
 const userId = 'experience-test-user';

package/packages/database/src/models/userMemory/__tests__/identity.test.ts

@@ -9,7 +9,7 @@ import {
   users,
 } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
-import { getTestDB } from '../../__tests__/_util';
+import { getTestDB } from '../../../core/getTestDB';
 import { UserMemoryIdentityModel } from '../identity';
 
 const userId = 'identity-test-user';

package/packages/database/src/models/userMemory/__tests__/preference.test.ts

@@ -8,7 +8,7 @@ import {
   users,
 } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
-import { getTestDB } from '../../__tests__/_util';
+import { getTestDB } from '../../../core/getTestDB';
 import { UserMemoryPreferenceModel } from '../preference';
 
 const userId = 'preference-test-user';

package/packages/database/src/repositories/agentGroup/index.test.ts

@@ -2,7 +2,7 @@
 import { BUILTIN_AGENT_SLUGS } from '@lobechat/builtin-agents';
 import { beforeEach, describe, expect, it } from 'vitest';
 
-import { getTestDB } from '../../models/__tests__/_util';
+import { getTestDB } from '../../core/getTestDB';
 import { agents } from '../../schemas/agent';
 import { chatGroups, chatGroupsAgents } from '../../schemas/chatGroup';
 import { users } from '../../schemas/user';

package/packages/database/src/repositories/agentMigration/__tests__/agentMigrationRepo.test.ts

@@ -1,7 +1,7 @@
 import { eq, inArray } from 'drizzle-orm';
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
-import { getTestDB } from '../../../models/__tests__/_util';
+import { getTestDB } from '../../../core/getTestDB';
 import { agents, messages, sessions, topics, users } from '../../../schemas';
 import { LobeChatDatabase } from '../../../type';
 import { AgentMigrationRepo } from '../index';

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

@@ -8,7 +8,7 @@ import { AiProviderModelListItem, EnabledAiModel } from 'model-bank';
 import { DEFAULT_MODEL_PROVIDER_LIST } from 'model-bank/modelProviders';
 import { beforeAll, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { getTestDB } from '../../models/__tests__/_util';
+import { getTestDB } from '../../core/getTestDB';
 import { LobeChatDatabase } from '../../type';
 import { AiInfraRepos } from './index';
 

package/packages/database/src/repositories/compression/index.test.ts

@@ -2,7 +2,7 @@
 import { MessageGroupType } from '@lobechat/types';
 import { beforeEach, describe, expect, it } from 'vitest';
 
-import { getTestDB } from '../../models/__tests__/_util';
+import { getTestDB } from '../../core/getTestDB';
 import { messageGroups, messages } from '../../schemas/message';
 import { topics } from '../../schemas/topic';
 import { users } from '../../schemas/user';

package/packages/database/src/repositories/dataExporter/index.test.ts

@@ -1,6 +1,6 @@
 import { afterEach, beforeAll, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { getTestDB } from '../../models/__tests__/_util';
+import { getTestDB } from '../../core/getTestDB';
 import {
   agents,
   agentsKnowledgeBases,