@lobehub/chat 1.114.2 → 1.114.4

package/docs/development/basic/image-generation-setup.zh-CN.mdx

@@ -0,0 +1,190 @@
+---
+title: 图像生成开发环境配置
+description: 配置本地环境以开发文本生图和图像处理功能
+---
+
+# 图像生成开发环境配置
+
+本指南帮助开发者配置本地环境，用于开发图像生成功能（文生图、图生图）等和文件存储能力。
+
+## 前置条件
+
+- 已安装并运行 Docker
+- 已安装 Node.js 和 pnpm
+- PostgreSQL 客户端工具（可选，用于调试）
+
+<Callout type="warning">
+  **安全提醒**：此配置仅适用于本地开发。使用的默认凭据和开放权限不适合生产环境。
+</Callout>
+
+## 快速配置
+
+运行提供的脚本来自动配置所有必需的服务：
+
+```bash
+# 配置 PostgreSQL 和 MinIO 用于图像存储
+./scripts/setup-image-generation-dev.sh
+
+# 启动开发服务器
+pnpm dev:desktop
+```
+
+此脚本将执行：
+
+1. 启动 PostgreSQL（本地开发无需身份验证）
+2. 运行数据库迁移以初始化模式
+3. 启动 MinIO（S3 兼容存储）
+4. 创建并配置存储桶
+5. 在 `.env.desktop` 中添加必要的 S3 环境变量
+
+## 架构概览
+
+图像生成功能需要：
+
+- **PostgreSQL**：存储生成图像的元数据
+- **MinIO/S3**：存储实际的图像文件
+- **服务器模式**：文件处理所需（`NEXT_PUBLIC_SERVICE_MODE=server`）
+
+## 环境配置
+
+以下环境变量会被配置脚本自动设置：
+
+```bash
+# S3 存储配置（本地开发使用 MinIO）
+S3_ACCESS_KEY_ID=minioadmin
+S3_SECRET_ACCESS_KEY=minioadmin
+S3_ENDPOINT=http://localhost:9000
+S3_BUCKET=lobe-chat
+S3_REGION=us-east-1
+S3_PUBLIC_DOMAIN=http://localhost:9000/lobe-chat
+S3_ENABLE_PATH_STYLE=1  # MinIO 必需
+```
+
+## 开发工作流
+
+### 1. 图像生成 API
+
+在开发图像生成功能时，生成的图像将：
+
+1. 由 AI 模型创建
+2. 通过预签名 URL 上传到 S3/MinIO
+3. 元数据存储在 PostgreSQL 中
+4. 通过公共 S3 URL 提供服务
+
+### 2. 文件存储结构
+
+```
+lobe-chat/           # S3 存储桶
+├── generated/       # 生成的图像
+│   └── {userId}/
+│       └── {sessionId}/
+│           └── {imageId}.png
+└── uploads/         # 用户上传的图像处理文件
+    └── {userId}/
+        └── {fileId}.{ext}
+```
+
+### 3. 测试您的实现
+
+配置环境后，您可以测试：
+
+```typescript
+// 示例：上传生成的图像
+const uploadUrl = await trpc.upload.createPresignedUrl.mutate({
+  filename: 'generated-image.png',
+  contentType: 'image/png',
+});
+
+// 上传到 S3
+await fetch(uploadUrl, {
+  method: 'PUT',
+  body: imageBlob,
+  headers: { 'Content-Type': 'image/png' },
+});
+```
+
+## 手动配置
+
+如果您希望手动配置服务：
+
+### PostgreSQL
+
+```bash
+docker run -d --name lobe-postgres \
+  -p 5432:5432 \
+  -e POSTGRES_HOST_AUTH_METHOD=trust \
+  -e POSTGRES_DB=postgres \
+  postgres:15
+```
+
+### MinIO
+
+```bash
+# 启动 MinIO
+docker run -d --name lobe-minio \
+  -p 9000:9000 -p 9001:9001 \
+  -e MINIO_ROOT_USER=minioadmin \
+  -e MINIO_ROOT_PASSWORD=minioadmin \
+  quay.io/minio/minio:RELEASE.2025-04-22T22-12-26Z \
+  server /data --console-address ":9001"
+
+# 创建存储桶
+docker run --rm \
+  --link lobe-minio:minio \
+  --entrypoint bash \
+  quay.io/minio/mc:RELEASE.2025-04-18T16-45-00Z \
+  -c "
+    mc config host add minio http://minio:9000 minioadmin minioadmin &&
+    mc mb minio/lobe-chat &&
+    mc anonymous set public minio/lobe-chat
+  "
+```
+
+## 服务地址
+
+- **PostgreSQL**：`postgres://postgres@localhost:5432/postgres`
+- **MinIO API**：`http://localhost:9000`
+- **MinIO 控制台**：`http://localhost:9001` (minioadmin/minioadmin)
+- **应用程序**：`http://localhost:3015`
+
+## 故障排除
+
+### 端口冲突
+
+如果端口已被占用：
+
+```bash
+# 检查端口使用情况
+lsof -i :5432  # PostgreSQL
+lsof -i :9000  # MinIO API
+lsof -i :9001  # MinIO 控制台
+```
+
+### 重置环境
+
+要完全重置开发环境：
+
+```bash
+# 停止并删除容器
+docker stop lobe-postgres lobe-minio
+docker rm lobe-postgres lobe-minio
+
+# 重新运行配置
+./scripts/setup-image-generation-dev.sh
+```
+
+### 数据库迁移
+
+配置脚本会自动运行迁移。如需手动运行：
+
+```bash
+pnpm db:migrate
+```
+
+注意：在使用 `pnpm dev:desktop` 的开发模式下，迁移也会在启动时自动运行。
+
+## 相关文档
+
+- [服务器数据库配置](/docs/self-hosting/server-database)
+- [S3 存储配置](/docs/self-hosting/advanced/s3)
+- [环境变量](/docs/self-hosting/environment-variables)

package/docs/development/database-schema.dbml

@@ -104,6 +104,19 @@ table ai_providers {
   }
 }
 
+table api_keys {
+  id integer [pk, not null]
+  name varchar(256) [not null]
+  key varchar(256) [not null, unique]
+  enabled boolean [default: true]
+  expires_at "timestamp with time zone"
+  last_used_at "timestamp with time zone"
+  user_id text [not null]
+  accessed_at "timestamp with time zone" [not null, default: `now()`]
+  created_at "timestamp with time zone" [not null, default: `now()`]
+  updated_at "timestamp with time zone" [not null, default: `now()`]
+}
+
 table async_tasks {
   id uuid [pk, not null, default: `gen_random_uuid()`]
   type text
@@ -702,6 +715,7 @@ table rbac_roles {
   description text
   is_system boolean [not null, default: false]
   is_active boolean [not null, default: true]
+  metadata jsonb [default: `{}`]
   accessed_at "timestamp with time zone" [not null, default: `now()`]
   created_at "timestamp with time zone" [not null, default: `now()`]
   updated_at "timestamp with time zone" [not null, default: `now()`]

package/drizzle.config.ts

@@ -22,8 +22,8 @@ export default {
     url: connectionString,
   },
   dialect: 'postgresql',
-  out: './src/database/migrations',
+  out: './packages/database/migrations',
 
-  schema: './src/database/schemas',
+  schema: './packages/database/src/schemas',
   strict: true,
 } satisfies Config;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.114.2",
+  "version": "1.114.4",
   "description": "Lobe Chat - an open-source, high-performance chatbot 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",
@@ -73,8 +73,6 @@
     "test": "npm run test-app && npm run test-server",
     "test-app": "vitest run --config vitest.config.ts",
     "test-app:coverage": "vitest run --config vitest.config.ts  --coverage",
-    "test-server": "vitest run --config vitest.config.server.ts",
-    "test-server:coverage": "vitest run --config vitest.config.server.ts --coverage",
     "test:update": "vitest -u",
     "type-check": "tsgo --noEmit",
     "webhook:ngrok": "ngrok http http://localhost:3011",
@@ -141,11 +139,14 @@
     "@icons-pack/react-simple-icons": "9.6.0",
     "@khmyznikov/pwa-install": "0.3.9",
     "@langchain/community": "^0.3.50",
+    "@lobechat/const": "workspace:*",
+    "@lobechat/database": "workspace:*",
     "@lobechat/electron-client-ipc": "workspace:*",
     "@lobechat/electron-server-ipc": "workspace:*",
     "@lobechat/file-loaders": "workspace:*",
     "@lobechat/model-runtime": "workspace:*",
     "@lobechat/prompts": "workspace:*",
+    "@lobechat/utils": "workspace:*",
     "@lobechat/web-crawler": "workspace:*",
     "@lobehub/analytics": "^1.6.0",
     "@lobehub/charts": "^2.0.0",
@@ -361,8 +362,7 @@
     "unified": "^11.0.5",
     "unist-util-visit": "^5.0.0",
     "vite": "^5.4.19",
-    "vitest": "^3.2.4",
-    "vitest-canvas-mock": "^0.3.3"
+    "vitest": "^3.2.4"
   },
   "packageManager": "pnpm@10.14.0",
   "publishConfig": {

package/packages/const/src/index.ts

@@ -1,2 +1,4 @@
+export * from './locale';
+export * from './message';
 export * from './settings';
 export * from './version';

package/{src/utils → packages/const/src/settings}/genUserLLMConfig.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it, vi } from 'vitest';
 
-import { ModelProviderCard } from '@/types/llm';
+import { ModelProviderCard } from '@/types/index';
 
 import { genUserLLMConfig } from './genUserLLMConfig';
 

package/{src/utils → packages/const/src/settings}/genUserLLMConfig.ts

@@ -1,8 +1,8 @@
 import { ModelProvider } from '@lobechat/model-runtime';
+import { UserModelProviderConfig } from '@lobechat/types';
 
 import * as ProviderCards from '@/config/modelProviders';
 import { ModelProviderCard } from '@/types/llm';
-import { UserModelProviderConfig } from '@/types/user/settings';
 
 export const genUserLLMConfig = (specificConfig: Record<any, any>): UserModelProviderConfig => {
   return Object.keys(ModelProvider).reduce((config, providerKey) => {

package/packages/const/src/settings/index.ts

@@ -9,8 +9,6 @@ import { DEFAULT_SYSTEM_AGENT_CONFIG } from './systemAgent';
 import { DEFAULT_TOOL_CONFIG } from './tool';
 import { DEFAULT_TTS_CONFIG } from './tts';
 
-export const COOKIE_CACHE_DAYS = 30;
-
 export * from './agent';
 export * from './hotkey';
 export * from './llm';

package/packages/const/src/settings/llm.ts

@@ -1,6 +1,6 @@
 import { ModelProvider } from '@lobechat/model-runtime';
 
-import { genUserLLMConfig } from '@/utils/genUserLLMConfig';
+import { genUserLLMConfig } from './genUserLLMConfig';
 
 export const DEFAULT_LLM_CONFIG = genUserLLMConfig({
   lmstudio: {

package/{src → packages}/database/migrations/0028_oauth_handoffs.sql

@@ -1,4 +1,4 @@
-CREATE TABLE "oauth_handoffs" (
+CREATE TABLE IF NOT EXISTS "oauth_handoffs" (
 	"id" text PRIMARY KEY NOT NULL,
 	"client" varchar(50) NOT NULL,
 	"payload" jsonb NOT NULL,

package/{src → packages}/database/migrations/0029_add_apikey_manage.sql

@@ -12,5 +12,5 @@ CREATE TABLE IF NOT EXISTS "api_keys" (
 	CONSTRAINT "api_keys_key_unique" UNIQUE("key")
 );
 --> statement-breakpoint
-ALTER TABLE "rbac_roles" ADD COLUMN "metadata" jsonb DEFAULT '{}'::jsonb;--> statement-breakpoint
+ALTER TABLE "rbac_roles" ADD COLUMN IF NOT EXISTS "metadata" jsonb DEFAULT '{}'::jsonb;--> statement-breakpoint
 ALTER TABLE "api_keys" ADD CONSTRAINT "api_keys_user_id_users_id_fk" FOREIGN KEY ("user_id") REFERENCES "public"."users"("id") ON DELETE cascade ON UPDATE no action;

package/packages/database/package.json

@@ -0,0 +1,24 @@
+{
+  "name": "@lobechat/database",
+  "version": "1.0.0",
+  "private": true,
+  "main": "src/index.ts",
+  "types": "src/index.ts",
+  "scripts": {
+    "test": "vitest",
+    "test:coverage": "vitest --coverage"
+  },
+  "dependencies": {
+    "@electric-sql/pglite": "^0.2.17",
+    "@lobechat/const": "workspace:*",
+    "@lobechat/types": "workspace:*",
+    "@lobechat/utils": "workspace:*",
+    "dayjs": "^1.11.13",
+    "drizzle-orm": "^0.44.4",
+    "nanoid": "^5.1.5",
+    "pg": "^8.16.3",
+    "random-words": "^2.0.1",
+    "ts-md5": "^2.0.1",
+    "ws": "^8.18.3"
+  }
+}

package/{src/database → packages/database/src}/client/db.ts

@@ -2,7 +2,6 @@ import { sql } from 'drizzle-orm';
 import { PgliteDatabase, drizzle } from 'drizzle-orm/pglite';
 import { Md5 } from 'ts-md5';
 
-import { DrizzleMigrationModel } from '@/database/models/drizzleMigration';
 import {
   ClientDBLoadingProgress,
   DatabaseLoadingState,
@@ -11,8 +10,9 @@ import {
 } from '@/types/clientDB';
 import { sleep } from '@/utils/sleep';
 
+import migrations from '../core/migrations.json';
+import { DrizzleMigrationModel } from '../models/drizzleMigration';
 import * as schema from '../schemas';
-import migrations from './migrations.json';
 
 const pgliteSchemaHashCache = 'LOBE_CHAT_PGLITE_SCHEMA_HASH';
 

package/{src/database → packages/database/src}/core/db-adaptor.ts

@@ -1,8 +1,8 @@
-import { isDesktop } from '@/const/version';
-import { getDBInstance } from '@/database/core/web-server';
-import { LobeChatDatabase } from '@/database/type';
+import { isDesktop } from '@lobechat/const';
 
+import { LobeChatDatabase } from '../type';
 import { getPgliteInstance } from './electron';
+import { getDBInstance } from './web-server';
 
 /**
  * 懒加载数据库实例

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

@@ -11,7 +11,7 @@ import { serverDBEnv } from '@/config/db';
 
 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 → packages/database/src}/core/electron.ts

@@ -4,13 +4,13 @@ import { drizzle as pgliteDrizzle } from 'drizzle-orm/pglite';
 import fs from 'node:fs';
 import { Md5 } from 'ts-md5';
 
-import { DrizzleMigrationModel } from '@/database/models/drizzleMigration';
-import * as schema from '@/database/schemas';
 import { electronIpcClient } from '@/server/modules/ElectronIPCClient';
 import { MigrationTableItem } from '@/types/clientDB';
 
-import migrations from '../client/migrations.json';
+import { DrizzleMigrationModel } from '../models/drizzleMigration';
+import * as schema from '../schemas';
 import { LobeChatDatabase } from '../type';
+import migrations from './migrations.json';
 
 // 用于实例管理的全局对象
 interface LobeGlobal {

package/{src/database/client → packages/database/src/core}/migrations.json

@@ -544,10 +544,20 @@
   },
   {
     "sql": [
-      "CREATE TABLE \"oauth_handoffs\" (\n\t\"id\" text PRIMARY KEY NOT NULL,\n\t\"client\" varchar(50) NOT NULL,\n\t\"payload\" jsonb NOT NULL,\n\t\"accessed_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"created_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"updated_at\" timestamp with time zone DEFAULT now() NOT NULL\n);\n"
+      "CREATE TABLE IF NOT EXISTS \"oauth_handoffs\" (\n\t\"id\" text PRIMARY KEY NOT NULL,\n\t\"client\" varchar(50) NOT NULL,\n\t\"payload\" jsonb NOT NULL,\n\t\"accessed_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"created_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"updated_at\" timestamp with time zone DEFAULT now() NOT NULL\n);\n"
     ],
     "bps": true,
     "folderMillis": 1752567402506,
-    "hash": "8ba3ae52ed72e8aad1623dbcf47ca26a8406ebffc6d5284abff94ea994b59c04"
+    "hash": "83c410b18ef5c8667b4bdfd7880ef7db4c3278d826e6e87e9d3e05dde67fe8e1"
+  },
+  {
+    "sql": [
+      "CREATE TABLE IF NOT EXISTS \"api_keys\" (\n\t\"id\" integer PRIMARY KEY GENERATED BY DEFAULT AS IDENTITY (sequence name \"api_keys_id_seq\" INCREMENT BY 1 MINVALUE 1 MAXVALUE 2147483647 START WITH 1 CACHE 1),\n\t\"name\" varchar(256) NOT NULL,\n\t\"key\" varchar(256) NOT NULL,\n\t\"enabled\" boolean DEFAULT true,\n\t\"expires_at\" timestamp with time zone,\n\t\"last_used_at\" timestamp with time zone,\n\t\"user_id\" text NOT NULL,\n\t\"accessed_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"created_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"updated_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\tCONSTRAINT \"api_keys_key_unique\" UNIQUE(\"key\")\n);\n",
+      "\nALTER TABLE \"rbac_roles\" ADD COLUMN IF NOT EXISTS \"metadata\" jsonb DEFAULT '{}'::jsonb;",
+      "\nALTER TABLE \"api_keys\" ADD CONSTRAINT \"api_keys_user_id_users_id_fk\" FOREIGN KEY (\"user_id\") REFERENCES \"public\".\"users\"(\"id\") ON DELETE cascade ON UPDATE no action;\n"
+    ],
+    "bps": true,
+    "folderMillis": 1753201379817,
+    "hash": "fe5c0d7c2768189771c42ef93693fc1d58586b468c4bdde7fb6f2dc58cc9931c"
   }
 ]

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

@@ -5,13 +5,12 @@ 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 = (): LobeChatDatabase => {
-  if (!isServerMode) return {} as any;
+  if (!(process.env.NEXT_PUBLIC_SERVICE_MODE === 'server')) return {} as any;
 
   if (!serverDBEnv.KEY_VAULTS_SECRET) {
     throw new Error(

package/{src/database → packages/database/src}/models/__tests__/_test_template.ts

@@ -2,9 +2,8 @@
 import { eq } from 'drizzle-orm';
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
-import { LobeChatDatabase } from '@/database/type';
-
 import { sessionGroups, users } from '../../schemas';
+import { LobeChatDatabase } from '../../type';
 import { SessionGroupModel } from '../sessionGroup';
 import { getTestDB } from './_util';
 

package/{src/database → packages/database/src}/models/__tests__/_util.ts

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

package/{src/database → packages/database/src}/models/__tests__/agent.test.ts

@@ -2,8 +2,7 @@
 import { eq } from 'drizzle-orm';
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
-import { LobeChatDatabase } from '@/database/type';
-
+import { LobeChatDatabase } from '../../type';
 import {
   agents,
   agentsFiles,

package/{src/database → packages/database/src}/models/__tests__/aiModel.test.ts

@@ -2,10 +2,10 @@
 import { eq } from 'drizzle-orm';
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
-import { LobeChatDatabase } from '@/database/type';
 import { AiProviderModelListItem } from '@/types/aiModel';
 
 import { AiModelSelectItem, NewAiModelItem, aiModels, users } from '../../schemas';
+import { LobeChatDatabase } from '../../type';
 import { AiModelModel } from '../aiModel';
 import { getTestDB } from './_util';
 

package/{src/database → packages/database/src}/models/__tests__/aiProvider.test.ts

@@ -3,10 +3,10 @@ import { ModelProvider } from '@lobechat/model-runtime';
 import { eq } from 'drizzle-orm';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { LobeChatDatabase } from '@/database/type';
 import { sleep } from '@/utils/sleep';
 
 import { aiProviders, users } from '../../schemas';
+import { LobeChatDatabase } from '../../type';
 import { AiProviderModel } from '../aiProvider';
 import { getTestDB } from './_util';
 

package/{src/database → packages/database/src}/models/__tests__/asyncTask.test.ts

@@ -2,10 +2,10 @@
 import { eq } from 'drizzle-orm';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { LobeChatDatabase } from '@/database/type';
 import { AsyncTaskStatus, AsyncTaskType } from '@/types/asyncTask';
 
 import { asyncTasks, users } from '../../schemas';
+import { LobeChatDatabase } from '../../type';
 import { ASYNC_TASK_TIMEOUT, AsyncTaskModel } from '../asyncTask';
 import { getTestDB } from './_util';
 

package/{src/database → packages/database/src}/models/__tests__/chunk.test.ts

@@ -2,8 +2,7 @@
 import { eq } from 'drizzle-orm';
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
-import { LobeChatDatabase } from '@/database/type';
-import { uuid } from '@/utils/uuid';
+import { LobeChatDatabase } from '../../type';import { uuid } from '@/utils/uuid';
 
 import { chunks, embeddings, fileChunks, files, unstructuredChunks, users } from '../../schemas';
 import { ChunkModel } from '../chunk';

package/{src/database → packages/database/src}/models/__tests__/file.test.ts

@@ -2,10 +2,10 @@
 import { eq, inArray } from 'drizzle-orm';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { LobeChatDatabase } from '@/database/type';
 import { FilesTabs, SortType } from '@/types/files';
 
 import { files, globalFiles, knowledgeBaseFiles, knowledgeBases, users } from '../../schemas';
+import { LobeChatDatabase } from '../../type';
 import { FileModel } from '../file';
 import { getTestDB } from './_util';
 

package/{src/database → packages/database/src}/models/__tests__/generation.test.ts

@@ -2,8 +2,7 @@
 import { eq } from 'drizzle-orm';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { LobeChatDatabase } from '@/database/type';
-import { AsyncTaskStatus } from '@/types/asyncTask';
+import { LobeChatDatabase } from '../../type';import { AsyncTaskStatus } from '@/types/asyncTask';
 import { FileSource } from '@/types/files';
 import { ImageGenerationAsset } from '@/types/generation';
 

package/{src/database → packages/database/src}/models/__tests__/generationBatch.test.ts

@@ -2,8 +2,7 @@
 import { eq } from 'drizzle-orm';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { LobeChatDatabase } from '@/database/type';
-import { AsyncTaskStatus } from '@/types/asyncTask';
+import { LobeChatDatabase } from '../../type';import { AsyncTaskStatus } from '@/types/asyncTask';
 import { GenerationConfig } from '@/types/generation';
 
 import {

package/{src/database → packages/database/src}/models/__tests__/generationTopic.test.ts

@@ -2,8 +2,7 @@
 import { eq } from 'drizzle-orm';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { LobeChatDatabase } from '@/database/type';
-import { FileService } from '@/server/services/file';
+import { LobeChatDatabase } from '../../type';import { FileService } from '@/server/services/file';
 import { ImageGenerationTopic } from '@/types/generation';
 
 import { generationBatches, generationTopics, generations, users } from '../../schemas';

package/{src/database → packages/database/src}/models/__tests__/knowledgeBase.test.ts

@@ -2,8 +2,7 @@
 import { and, eq } from 'drizzle-orm';
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
-import { LobeChatDatabase } from '@/database/type';
-import { sleep } from '@/utils/sleep';
+import { LobeChatDatabase } from '../../type';import { sleep } from '@/utils/sleep';
 
 import {
   NewKnowledgeBase,

package/{src/database → packages/database/src}/models/__tests__/message.test.ts

@@ -2,11 +2,10 @@ import dayjs from 'dayjs';
 import { eq } from 'drizzle-orm';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { getTestDB } from '@/database/models/__tests__/_util';
-import { LobeChatDatabase } from '@/database/type';
 import { MessageItem } from '@/types/message';
 import { uuid } from '@/utils/uuid';
 
+import { getTestDB } from '../../models/__tests__/_util';
 import {
   chunks,
   embeddings,
@@ -23,6 +22,7 @@ import {
   topics,
   users,
 } from '../../schemas';
+import { LobeChatDatabase } from '../../type';
 import { MessageModel } from '../message';
 import { codeEmbedding } from './fixtures/embedding';
 

package/{src/database → packages/database/src}/models/__tests__/plugin.test.ts

@@ -1,8 +1,7 @@
 // @vitest-environment node
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { LobeChatDatabase } from '@/database/type';
-
+import { LobeChatDatabase } from '../../type';
 import { NewInstalledPlugin, userInstalledPlugins, users } from '../../schemas';
 import { PluginModel } from '../plugin';
 import { getTestDB } from './_util';