@lobehub/chat 0.147.10 → 0.147.12

package/CHANGELOG.md

@@ -2,6 +2,56 @@
 
 # Changelog
 
+### [Version 0.147.12](https://github.com/lobehub/lobe-chat/compare/v0.147.11...v0.147.12)
+
+<sup>Released on **2024-04-14**</sup>
+
+#### ♻ Code Refactoring
+
+- **misc**: Move client db to a new folder.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### Code refactoring
+
+- **misc**: Move client db to a new folder, closes [#2037](https://github.com/lobehub/lobe-chat/issues/2037) ([ebe65bb](https://github.com/lobehub/lobe-chat/commit/ebe65bb))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
+### [Version 0.147.11](https://github.com/lobehub/lobe-chat/compare/v0.147.10...v0.147.11)
+
+<sup>Released on **2024-04-14**</sup>
+
+#### 🐛 Bug Fixes
+
+- **misc**: Support drag or copy to upload file by model ability.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **misc**: Support drag or copy to upload file by model ability, closes [#2016](https://github.com/lobehub/lobe-chat/issues/2016) ([2abe37e](https://github.com/lobehub/lobe-chat/commit/2abe37e))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 0.147.10](https://github.com/lobehub/lobe-chat/compare/v0.147.9...v0.147.10)
 
 <sup>Released on **2024-04-13**</sup>

package/README.md

@@ -260,14 +260,14 @@ Our marketplace is not just a showcase platform but also a collaborative space.
 
 <!-- AGENT LIST -->
 
-| Recent Submits                                                                                                                                                                       | Description                                                                                                                                                                               |
-| ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------ | ----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
-| [Healthy Recipe Recommender](https://chat-preview.lobehub.com/market?agent=healthy-recipe-recommender)<br/><sup>By **[Greasen](https://github.com/Greasen)** on **2024-04-11**</sup> | Customized nutrition meals, scientific combinations, healthy eating, your personal nutritionist.<br/>`recipes` `fitness-meals` `nutrition-meals`                                          |
-| [SmartWeather Assistant](https://chat-preview.lobehub.com/market?agent=personal-weather-consultant)<br/><sup>By **[Greasen](https://github.com/Greasen)** on **2024-04-11**</sup>    | SmartWeather Assistant, your personal weather advisor, fashion guide, and positive energy booster!<br/>`weather` `assistant` `fashion`                                                    |
-| [Dirty Talk Learning Assistant](https://chat-preview.lobehub.com/market?agent=profanity-assistant)<br/><sup>By **[cokice](https://github.com/cokice)** on **2024-04-10**</sup>       | I only know how to curse, nothing else.<br/>`answer` `dirty-talk`                                                                                                                         |
-| [TadzGenius](https://chat-preview.lobehub.com/market?agent=tadz-genius)<br/><sup>By **[infoaitek24](https://github.com/infoaitek24)** on **2024-04-10**</sup>                        | Expert in business development and development practices in the Philippine market<br/>`business-development` `ai-assistant` `market-analysis` `strategic-planning` `customer-acquisition` |
-
-> 📊 Total agents: [<kbd>**236**</kbd> ](https://github.com/lobehub/lobe-chat-agents)
+| Recent Submits                                                                                                                                                                       | Description                                                                                                                                                                                                                                                                                                                                                                                                   |
+| ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------ | ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
+| [Biology Professor](https://chat-preview.lobehub.com/market?agent=bio-professor)<br/><sup>By **[luxiangze](https://github.com/luxiangze)** on **2024-04-13**</sup>                   | As a biology professor, you will receive questions and concepts about biology. Please explain these questions and concepts in specific and concise language, and try to use practical examples to illustrate them to help your audience better understand. Please ensure that your explanations are accurate and clear, and try to encourage creative and flexible answers. Respond in Chinese.<br/>`biology` |
+| [Master of Fortune Telling](https://chat-preview.lobehub.com/market?agent=fortune-teller)<br/><sup>By **[kamilkenrich](https://github.com/kamilkenrich)** on **2024-04-13**</sup>    | Specializes in numerology, divination, astrology, and blood type analysis<br/>`numerology` `divination` `astrology` `psychology` `blood-type` `constellation`                                                                                                                                                                                                                                                 |
+| [Healthy Recipe Recommender](https://chat-preview.lobehub.com/market?agent=healthy-recipe-recommender)<br/><sup>By **[Greasen](https://github.com/Greasen)** on **2024-04-11**</sup> | Customized nutrition meals, scientific combinations, healthy eating, your personal nutritionist.<br/>`recipes` `fitness-meals` `nutrition-meals`                                                                                                                                                                                                                                                              |
+| [SmartWeather Assistant](https://chat-preview.lobehub.com/market?agent=personal-weather-consultant)<br/><sup>By **[Greasen](https://github.com/Greasen)** on **2024-04-11**</sup>    | SmartWeather Assistant, your personal weather advisor, fashion guide, and positive energy booster!<br/>`weather` `assistant` `fashion`                                                                                                                                                                                                                                                                        |
+
+> 📊 Total agents: [<kbd>**238**</kbd> ](https://github.com/lobehub/lobe-chat-agents)
 
  <!-- AGENT LIST -->
 

package/README.zh-CN.md

@@ -248,14 +248,14 @@ LobeChat 的插件生态系统是其核心功能的重要扩展，它极大地
 
 <!-- AGENT LIST -->
 
-| 最近新增                                                                                                                                                                 | 助手说明                                                                                                                                                                                  |
-| ------------------------------------------------------------------------------------------------------------------------------------------------------------------------ | ----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
-| [健康食谱推荐师](https://chat-preview.lobehub.com/market?agent=healthy-recipe-recommender)<br/><sup>By **[Greasen](https://github.com/Greasen)** on **2024-04-11**</sup> | 精准定制营养餐，科学搭配，健康饮食，你的私人营养师。<br/>`食谱、健身餐、营养餐`                                                                                                           |
-| [智预小助手](https://chat-preview.lobehub.com/market?agent=personal-weather-consultant)<br/><sup>By **[Greasen](https://github.com/Greasen)** on **2024-04-11**</sup>    | 智预小助手，您的私人天气顾问，穿搭指南，正能量加油器！<br/>`天气` `助理、穿搭`                                                                                                            |
-| [脏话学习助手](https://chat-preview.lobehub.com/market?agent=profanity-assistant)<br/><sup>By **[cokice](https://github.com/cokice)** on **2024-04-10**</sup>            | 我只会骂人 别的都不会<br/>`回答` `脏话`                                                                                                                                                   |
-| [TadzGenius](https://chat-preview.lobehub.com/market?agent=tadz-genius)<br/><sup>By **[infoaitek24](https://github.com/infoaitek24)** on **2024-04-10**</sup>            | Expert in business development and development practices in the Philippine market<br/>`business-development` `ai-assistant` `market-analysis` `strategic-planning` `customer-acquisition` |
-
-> 📊 Total agents: [<kbd>**236**</kbd> ](https://github.com/lobehub/lobe-chat-agents)
+| 最近新增                                                                                                                                                                 | 助手说明                                                                                                                                                                                                                          |
+| ------------------------------------------------------------------------------------------------------------------------------------------------------------------------ | --------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
+| [生物学教授](https://chat-preview.lobehub.com/market?agent=bio-professor)<br/><sup>By **[luxiangze](https://github.com/luxiangze)** on **2024-04-13**</sup>              | 作为一位生物学教授，您将收到关于生物学的问题和概念，请用具体而简明的语言解释这些问题和概念，并尽量使用实际例子进行说明，以帮助您的听众更好地理解。请确保您的解释准确而清晰，并试图鼓励创造性和灵活性的答案。用中文回答<br/>`生物` |
+| [命理大师](https://chat-preview.lobehub.com/market?agent=fortune-teller)<br/><sup>By **[kamilkenrich](https://github.com/kamilkenrich)** on **2024-04-13**</sup>         | 擅长命理学、占卜预测、占星术和血型学分析<br/>`命理学、占卜、占星学、心理学、血型、星座`                                                                                                                                           |
+| [健康食谱推荐师](https://chat-preview.lobehub.com/market?agent=healthy-recipe-recommender)<br/><sup>By **[Greasen](https://github.com/Greasen)** on **2024-04-11**</sup> | 精准定制营养餐，科学搭配，健康饮食，你的私人营养师。<br/>`食谱、健身餐、营养餐`                                                                                                                                                   |
+| [智预小助手](https://chat-preview.lobehub.com/market?agent=personal-weather-consultant)<br/><sup>By **[Greasen](https://github.com/Greasen)** on **2024-04-11**</sup>    | 智预小助手，您的私人天气顾问，穿搭指南，正能量加油器！<br/>`天气` `助理、穿搭`                                                                                                                                                    |
+
+> 📊 Total agents: [<kbd>**238**</kbd> ](https://github.com/lobehub/lobe-chat-agents)
 
  <!-- AGENT LIST -->
 

package/contributing/Basic/Feature-Development.md

@@ -128,8 +128,8 @@ When building the LobeChat application, the Model is responsible for interacting
 In `src/database/model/sessionGroup.ts`, the `SessionGroupModel` is defined as follows:
 
 ```typescript
-import { BaseModel } from '@/database/core';
-import { DB_SessionGroup, DB_SessionGroupSchema } from '@/database/schemas/sessionGroup';
+import { BaseModel } from '@/database/client/core';
+import { DB_SessionGroup, DB_SessionGroupSchema } from '@/database/client/schemas/sessionGroup';
 import { nanoid } from '@/utils/uuid';
 
 class _SessionGroupModel extends BaseModel {

package/contributing/Basic/Feature-Development.zh-CN.md

@@ -128,8 +128,8 @@ export class LocalDB extends Dexie {
 在 `src/database/model/sessionGroup.ts` 中定义 `SessionGroupModel`：
 
 ```typescript
-import { BaseModel } from '@/database/core';
-import { DB_SessionGroup, DB_SessionGroupSchema } from '@/database/schemas/sessionGroup';
+import { BaseModel } from '@/database/client/core';
+import { DB_SessionGroup, DB_SessionGroupSchema } from '@/database/client/schemas/sessionGroup';
 import { nanoid } from '@/utils/uuid';
 
 class _SessionGroupModel extends BaseModel {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "0.147.10",
+  "version": "0.147.12",
   "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",
@@ -172,7 +172,7 @@
     "@next/eslint-plugin-next": "^14.1.4",
     "@peculiar/webcrypto": "^1.4.6",
     "@testing-library/jest-dom": "^6.4.2",
-    "@testing-library/react": "^14.3.0",
+    "@testing-library/react": "^15.0.0",
     "@types/chroma-js": "^2.4.4",
     "@types/debug": "^4.1.12",
     "@types/diff": "^5.0.9",

package/src/app/chat/(desktop)/features/ChatInput/Footer/DragUpload.tsx

@@ -7,6 +7,10 @@ import { useTranslation } from 'react-i18next';
 import { Center, Flexbox } from 'react-layout-kit';
 
 import { useFileStore } from '@/store/file';
+import { useGlobalStore } from '@/store/global';
+import { modelProviderSelectors } from '@/store/global/selectors';
+import { useSessionStore } from '@/store/session';
+import { agentSelectors } from '@/store/session/selectors';
 
 const useStyles = createStyles(({ css, token, stylish }) => {
   return {
@@ -56,6 +60,8 @@ const useStyles = createStyles(({ css, token, stylish }) => {
   };
 });
 
+
+
 const handleDragOver = (e: DragEvent) => {
   e.preventDefault();
 };
@@ -71,12 +77,16 @@ const DragUpload = memo(() => {
 
   const uploadFile = useFileStore((s) => s.uploadFile);
 
+  const model = useSessionStore(agentSelectors.currentAgentModel);
+
+  const enabledFiles = useGlobalStore(modelProviderSelectors.isModelEnabledFiles(model));
+
   const uploadImages = async (fileList: FileList | undefined) => {
     if (!fileList || fileList.length === 0) return;
 
     const pools = Array.from(fileList).map(async (file) => {
       // skip none-file items
-      if (!file.type.startsWith('image')) return;
+      if (!file.type.startsWith('image') && !enabledFiles) return;
       await uploadFile(file);
     });
 
@@ -126,6 +136,7 @@ const DragUpload = memo(() => {
     uploadImages(files);
   };
 
+
   useEffect(() => {
     window.addEventListener('dragenter', handleDragEnter);
     window.addEventListener('dragover', handleDragOver);
@@ -140,7 +151,7 @@ const DragUpload = memo(() => {
       window.removeEventListener('drop', handleDrop);
       window.removeEventListener('paste', handlePaste);
     };
-  }, []);
+  }, [handleDrop, handlePaste]);
 
   return (
     isDragging && (
@@ -164,3 +175,4 @@ const DragUpload = memo(() => {
 });
 
 export default DragUpload;
+

package/src/database/{core → client/core}/__tests__/db-upgrade.test.ts

@@ -1,7 +1,7 @@
 import Dexie from 'dexie';
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
-import { LocalDB } from '../db';
+import { BrowserDB } from '../db';
 import { dbSchemaV3 } from '../schemas';
 import { LOBE_CHAT_LOCAL_DB_NAME } from '../types/db';
 
@@ -28,7 +28,7 @@ describe('LocalDB migration', () => {
     dbV3.close();
 
     // 创建新的数据库实例，包含版本4的迁移逻辑
-    const dbV4 = new LocalDB();
+    const dbV4 = new BrowserDB();
     await dbV4.open();
 
     // 验证迁移后的数据

package/src/database/{core → client/core}/__tests__/db.test.ts

@@ -1,15 +1,14 @@
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
-import { DBModel } from '@/database/core/types/db';
-import { DB_File } from '@/database/schemas/files';
-
-import { LocalDB } from '../db';
+import { DB_File } from '../../schemas/files';
+import { BrowserDB } from '../db';
+import { DBModel } from '../types/db';
 
 describe('LocalDB', () => {
-  let db: LocalDB;
+  let db: BrowserDB;
 
   beforeEach(() => {
-    db = new LocalDB();
+    db = new BrowserDB();
   });
 
   afterEach(async () => {

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

@@ -1,15 +1,15 @@
 import Dexie, { Transaction } from 'dexie';
 
-import { DB_File } from '@/database/schemas/files';
-import { DB_Message } from '@/database/schemas/message';
-import { DB_Plugin } from '@/database/schemas/plugin';
-import { DB_Session } from '@/database/schemas/session';
-import { DB_SessionGroup } from '@/database/schemas/sessionGroup';
-import { DB_Topic } from '@/database/schemas/topic';
-import { DB_User } from '@/database/schemas/user';
 import { MigrationLLMSettings } from '@/migrations/FromV3ToV4';
 import { uuid } from '@/utils/uuid';
 
+import { DB_File } from '../schemas/files';
+import { DB_Message } from '../schemas/message';
+import { DB_Plugin } from '../schemas/plugin';
+import { DB_Session } from '../schemas/session';
+import { DB_SessionGroup } from '../schemas/sessionGroup';
+import { DB_Topic } from '../schemas/topic';
+import { DB_User } from '../schemas/user';
 import { migrateSettingsToUser } from './migrations/migrateSettingsToUser';
 import {
   dbSchemaV1,
@@ -33,14 +33,14 @@ export interface LobeDBSchemaMap {
 }
 
 // Define a local DB
-export class LocalDB extends Dexie {
-  public files: LobeDBTable<'files'>;
-  public sessions: LobeDBTable<'sessions'>;
-  public messages: LobeDBTable<'messages'>;
-  public topics: LobeDBTable<'topics'>;
-  public plugins: LobeDBTable<'plugins'>;
-  public sessionGroups: LobeDBTable<'sessionGroups'>;
-  public users: LobeDBTable<'users'>;
+export class BrowserDB extends Dexie {
+  public files: BrowserDBTable<'files'>;
+  public sessions: BrowserDBTable<'sessions'>;
+  public messages: BrowserDBTable<'messages'>;
+  public topics: BrowserDBTable<'topics'>;
+  public plugins: BrowserDBTable<'plugins'>;
+  public sessionGroups: BrowserDBTable<'sessionGroups'>;
+  public users: BrowserDBTable<'users'>;
 
   constructor() {
     super(LOBE_CHAT_LOCAL_DB_NAME);
@@ -155,7 +155,7 @@ export class LocalDB extends Dexie {
   };
 }
 
-export const LocalDBInstance = new LocalDB();
+export const browserDB = new BrowserDB();
 
 // ================================================ //
 // ================================================ //
@@ -164,10 +164,10 @@ export const LocalDBInstance = new LocalDB();
 // ================================================ //
 
 // types helper
-export type LocalDBSchema = {
+export type BrowserDBSchema = {
   [t in keyof LobeDBSchemaMap]: {
     model: LobeDBSchemaMap[t];
     table: Dexie.Table<DBModel<LobeDBSchemaMap[t]>, string>;
   };
 };
-type LobeDBTable<T extends keyof LobeDBSchemaMap> = LocalDBSchema[T]['table'];
+type BrowserDBTable<T extends keyof LobeDBSchemaMap> = BrowserDBSchema[T]['table'];

package/src/database/{core → client/core}/index.ts

@@ -1,3 +1,3 @@
-export { LocalDBInstance } from './db';
+export { browserDB } from './db';
 export * from './model';
 export { dataSync } from './sync';

package/src/database/{core → client/core}/migrations/migrateSettingsToUser/index.ts

@@ -1,4 +1,4 @@
-import { DB_Settings } from '@/database/schemas/user';
+import { DB_Settings } from '@/database/client/schemas/user';
 
 import { V4Settings } from './type';
 

package/src/database/{core → client/core}/model.ts

@@ -3,16 +3,16 @@ import { ZodObject } from 'zod';
 
 import { nanoid } from '@/utils/uuid';
 
-import { LocalDB, LocalDBInstance, LocalDBSchema } from './db';
+import { BrowserDB, BrowserDBSchema, browserDB } from './db';
 import { dataSync } from './sync';
 import { DBBaseFieldsSchema } from './types/db';
 
-export class BaseModel<N extends keyof LocalDBSchema = any, T = LocalDBSchema[N]['table']> {
-  protected readonly db: LocalDB;
+export class BaseModel<N extends keyof BrowserDBSchema = any, T = BrowserDBSchema[N]['table']> {
+  protected readonly db: BrowserDB;
   private readonly schema: ZodObject<any>;
-  private readonly _tableName: keyof LocalDBSchema;
+  private readonly _tableName: keyof BrowserDBSchema;
 
-  constructor(table: N, schema: ZodObject<any>, db = LocalDBInstance) {
+  constructor(table: N, schema: ZodObject<any>, db = browserDB) {
     this.db = db;
     this.schema = schema;
     this._tableName = table;
@@ -31,7 +31,7 @@ export class BaseModel<N extends keyof LocalDBSchema = any, T = LocalDBSchema[N]
   /**
    * create a new record
    */
-  protected async _addWithSync<T = LocalDBSchema[N]['model']>(
+  protected async _addWithSync<T = BrowserDBSchema[N]['model']>(
     data: T,
     id: string | number = nanoid(),
     primaryKey: string = 'id',
@@ -71,7 +71,7 @@ export class BaseModel<N extends keyof LocalDBSchema = any, T = LocalDBSchema[N]
    * @param options.generateId
    * @param options.createWithNewId
    */
-  protected async _batchAdd<T = LocalDBSchema[N]['model']>(
+  protected async _batchAdd<T = BrowserDBSchema[N]['model']>(
     dataArray: T[],
     options: {
       /**

package/src/database/{core → client/core}/sync.ts

@@ -11,7 +11,7 @@ import {
   StartDataSyncParams,
 } from '@/types/sync';
 
-import { LobeDBSchemaMap, LocalDBInstance } from './db';
+import { LobeDBSchemaMap, browserDB } from './db';
 
 const LOG_NAME_SPACE = 'DataSync';
 
@@ -201,7 +201,7 @@ class DataSync {
     onEvent: OnSyncEvent,
     onSyncStatusChange: OnSyncStatusChange,
   ) => {
-    const table = LocalDBInstance[tableKey];
+    const table = browserDB[tableKey];
     const yItemMap = this.getYMap(tableKey);
     const updateSyncEvent = throttle(onEvent, 1000);
 
@@ -249,7 +249,7 @@ class DataSync {
   };
 
   private loadDataFromDBtoYjs = async (tableKey: keyof LobeDBSchemaMap) => {
-    const table = LocalDBInstance[tableKey];
+    const table = browserDB[tableKey];
     const items = await table.toArray();
     const yItemMap = this.getYMap(tableKey);
 

package/src/database/{models → client/models}/__DEBUG.ts

@@ -1,9 +1,9 @@
 // This file is for debugging purposes only.
 // DON'T USE IT IN PRODUCTION.
 import { DEFAULT_AGENT_CONFIG } from '@/const/settings';
-import { DBModel } from '@/database/core/types/db';
-import { DB_Message } from '@/database/schemas/message';
-import { DB_Topic } from '@/database/schemas/topic';
+import { DBModel } from '@/database/client/core/types/db';
+import { DB_Message } from '@/database/client/schemas/message';
+import { DB_Topic } from '@/database/client/schemas/topic';
 
 import { BaseModel } from '../core';
 import { DB_Session, DB_SessionSchema } from '../schemas/session';

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

@@ -1,8 +1,7 @@
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
-import { DB_File } from '@/database/schemas/files';
-
-import { LocalDB } from '../../core/db';
+import { BrowserDB } from '../../core/db';
+import { DB_File } from '../../schemas/files';
 import { FileModel } from '../file';
 
 // Assuming LocalDB is already mocked or using an in-memory database
@@ -25,7 +24,7 @@ describe('_FileModel', () => {
 
   afterEach(async () => {
     // Clean up the database after each test
-    const db = new LocalDB();
+    const db = new BrowserDB();
     await db.files.clear();
     db.close();
   });

package/src/database/{models → client/models}/__tests__/session.test.ts

@@ -1,9 +1,9 @@
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
 import { DEFAULT_AGENT_CONFIG } from '@/const/settings';
-import { CreateMessageParams, MessageModel } from '@/database/models/message';
-import { SessionGroupModel } from '@/database/models/sessionGroup';
-import { TopicModel } from '@/database/models/topic';
+import { CreateMessageParams, MessageModel } from '@/database/client/models/message';
+import { SessionGroupModel } from '@/database/client/models/sessionGroup';
+import { TopicModel } from '@/database/client/models/topic';
 import { LobeAgentConfig } from '@/types/agent';
 import {
   LobeAgentSession,

package/src/database/{models → client/models}/__tests__/sessionGroup.test.ts

@@ -1,6 +1,6 @@
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
-import { SessionModel } from '@/database/models/session';
+import { SessionModel } from '@/database/client/models/session';
 import { SessionGroups } from '@/types/session';
 
 import { DB_SessionGroup } from '../../schemas/sessionGroup';

package/src/database/{models → client/models}/__tests__/topic.test.ts

@@ -1,9 +1,9 @@
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
-import { DBModel } from '@/database/core/types/db';
-import { CreateMessageParams, MessageModel } from '@/database/models/message';
-import { DB_Message } from '@/database/schemas/message';
-import { DB_Topic } from '@/database/schemas/topic';
+import { DBModel } from '@/database/client/core/types/db';
+import { CreateMessageParams, MessageModel } from '@/database/client/models/message';
+import { DB_Message } from '@/database/client/schemas/message';
+import { DB_Topic } from '@/database/client/schemas/topic';
 import { nanoid } from '@/utils/uuid';
 import * as uuidUtils from '@/utils/uuid';
 

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

@@ -1,4 +1,4 @@
-import { DB_File, DB_FileSchema } from '@/database/schemas/files';
+import { DB_File, DB_FileSchema } from '@/database/client/schemas/files';
 import { nanoid } from '@/utils/uuid';
 
 import { BaseModel } from '../core';

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

@@ -1,8 +1,8 @@
 import { DeepPartial } from 'utility-types';
 
-import { BaseModel } from '@/database/core';
-import { DBModel } from '@/database/core/types/db';
-import { DB_Message, DB_MessageSchema } from '@/database/schemas/message';
+import { BaseModel } from '@/database/client/core';
+import { DBModel } from '@/database/client/core/types/db';
+import { DB_Message, DB_MessageSchema } from '@/database/client/schemas/message';
 import { ChatMessage } from '@/types/message';
 import { nanoid } from '@/utils/uuid';
 

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

@@ -1,6 +1,6 @@
 import { LobeChatPluginManifest } from '@lobehub/chat-plugin-sdk';
 
-import { BaseModel } from '@/database/core';
+import { BaseModel } from '@/database/client/core';
 import { LobeTool } from '@/types/tool';
 import { merge } from '@/utils/merge';
 

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

@@ -1,9 +1,9 @@
 import { DeepPartial } from 'utility-types';
 
 import { DEFAULT_AGENT_LOBE_SESSION } from '@/const/session';
-import { BaseModel } from '@/database/core';
-import { DBModel } from '@/database/core/types/db';
-import { DB_Session, DB_SessionSchema } from '@/database/schemas/session';
+import { BaseModel } from '@/database/client/core';
+import { DBModel } from '@/database/client/core/types/db';
+import { DB_Session, DB_SessionSchema } from '@/database/client/schemas/session';
 import { LobeAgentConfig } from '@/types/agent';
 import {
   ChatSessionList,

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

@@ -1,5 +1,5 @@
-import { BaseModel } from '@/database/core';
-import { DB_SessionGroup, DB_SessionGroupSchema } from '@/database/schemas/sessionGroup';
+import { BaseModel } from '@/database/client/core';
+import { DB_SessionGroup, DB_SessionGroupSchema } from '@/database/client/schemas/sessionGroup';
 import { SessionGroups } from '@/types/session';
 import { nanoid } from '@/utils/uuid';
 

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

@@ -1,7 +1,7 @@
-import { BaseModel } from '@/database/core';
-import { DBModel } from '@/database/core/types/db';
-import { MessageModel } from '@/database/models/message';
-import { DB_Topic, DB_TopicSchema } from '@/database/schemas/topic';
+import { BaseModel } from '@/database/client/core';
+import { DBModel } from '@/database/client/core/types/db';
+import { MessageModel } from '@/database/client/models/message';
+import { DB_Topic, DB_TopicSchema } from '@/database/client/schemas/topic';
 import { ChatTopic } from '@/types/topic';
 import { nanoid } from '@/utils/uuid';
 

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

@@ -1,6 +1,6 @@
 import { DeepPartial } from 'utility-types';
 
-import { BaseModel } from '@/database/core';
+import { BaseModel } from '@/database/client/core';
 import { GlobalSettings } from '@/types/settings';
 import { uuid } from '@/utils/uuid';
 

package/src/database/{schemas → client/schemas}/user.ts

@@ -1,6 +1,6 @@
 import { z } from 'zod';
 
-import { AgentSchema } from '@/database/schemas/session';
+import { AgentSchema } from '@/database/client/schemas/session';
 import { LobeMetaDataSchema } from '@/types/meta';
 
 const settingsSchema = z.object({

package/src/services/__tests__/file.test.ts

@@ -1,12 +1,12 @@
 import { Mock, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { FileModel } from '@/database/models/file';
-import { DB_File } from '@/database/schemas/files';
+import { FileModel } from '@/database/client/models/file';
+import { DB_File } from '@/database/client/schemas/files';
 
 import { fileService } from '../file';
 
 // Mocks for the FileModel
-vi.mock('@/database/models/file', () => ({
+vi.mock('@/database/client/models/file', () => ({
   FileModel: {
     create: vi.fn(),
     delete: vi.fn(),

package/src/services/__tests__/message.test.ts

@@ -1,6 +1,6 @@
 import { Mock, describe, expect, it, vi } from 'vitest';
 
-import { CreateMessageParams, MessageModel } from '@/database/models/message';
+import { CreateMessageParams, MessageModel } from '@/database/client/models/message';
 import {
   ChatMessage,
   ChatMessageError,
@@ -12,7 +12,7 @@ import {
 import { messageService } from '../message';
 
 // Mock the MessageModel
-vi.mock('@/database/models/message', () => {
+vi.mock('@/database/client/models/message', () => {
   return {
     MessageModel: {
       create: vi.fn(),

package/src/services/__tests__/plugin.test.ts

@@ -1,8 +1,8 @@
 import { LobeChatPluginManifest } from '@lobehub/chat-plugin-sdk';
 import { Mock, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { PluginModel } from '@/database/models/plugin';
-import { DB_Plugin } from '@/database/schemas/plugin';
+import { PluginModel } from '@/database/client/models/plugin';
+import { DB_Plugin } from '@/database/client/schemas/plugin';
 import { globalHelpers } from '@/store/global/helpers';
 import { LobeTool } from '@/types/tool';
 import { LobeToolCustomPlugin } from '@/types/tool/plugin';
@@ -18,7 +18,7 @@ vi.mock('@/store/global/helpers', () => ({
     getCurrentLanguage: vi.fn(),
   },
 }));
-vi.mock('@/database/models/plugin', () => ({
+vi.mock('@/database/client/models/plugin', () => ({
   PluginModel: {
     getList: vi.fn(),
     create: vi.fn(),

package/src/services/__tests__/session.test.ts

@@ -1,14 +1,14 @@
 import { Mock, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { SessionModel } from '@/database/models/session';
-import { SessionGroupModel } from '@/database/models/sessionGroup';
+import { SessionModel } from '@/database/client/models/session';
+import { SessionGroupModel } from '@/database/client/models/sessionGroup';
 import { LobeAgentConfig } from '@/types/agent';
 import { LobeAgentSession, LobeSessionType, SessionGroups } from '@/types/session';
 
 import { sessionService } from '../session';
 
 // Mock the SessionModel
-vi.mock('@/database/models/session', () => {
+vi.mock('@/database/client/models/session', () => {
   return {
     SessionModel: {
       create: vi.fn(),
@@ -29,7 +29,7 @@ vi.mock('@/database/models/session', () => {
 });
 
 // Mock the SessionGroupModel
-vi.mock('@/database/models/sessionGroup', () => {
+vi.mock('@/database/client/models/sessionGroup', () => {
   return {
     SessionGroupModel: {
       create: vi.fn(),

package/src/services/__tests__/topic.test.ts

@@ -1,12 +1,12 @@
 import { Mock, beforeAll, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { CreateTopicParams, TopicModel } from '@/database/models/topic';
+import { CreateTopicParams, TopicModel } from '@/database/client/models/topic';
 import { ChatTopic } from '@/types/topic';
 
 import { topicService } from '../topic';
 
 // Mock the TopicModel
-vi.mock('@/database/models/topic', () => {
+vi.mock('@/database/client/models/topic', () => {
   return {
     TopicModel: {
       create: vi.fn(),

package/src/services/debug.ts

@@ -1,4 +1,4 @@
-import { DEBUG_MODEL } from '@/database/models/__DEBUG';
+import { DEBUG_MODEL } from '@/database/client/models/__DEBUG';
 
 class DebugService {
   async insertLargeDataToDB() {

package/src/services/file.ts

@@ -1,5 +1,5 @@
-import { FileModel } from '@/database/models/file';
-import { DB_File } from '@/database/schemas/files';
+import { FileModel } from '@/database/client/models/file';
+import { DB_File } from '@/database/client/schemas/files';
 import { FilePreview } from '@/types/files';
 import compressImage from '@/utils/compressImage';
 

package/src/services/global.ts

@@ -1,4 +1,4 @@
-import { dataSync } from '@/database/core';
+import { dataSync } from '@/database/client/core';
 import { GlobalServerConfig } from '@/types/serverConfig';
 import { StartDataSyncParams } from '@/types/sync';
 

package/src/services/message.ts

@@ -1,7 +1,16 @@
-import { CreateMessageParams, MessageModel } from '@/database/models/message';
-import { DB_Message } from '@/database/schemas/message';
+import { MessageModel } from '@/database/client/models/message';
+import { DB_Message } from '@/database/client/schemas/message';
 import { ChatMessage, ChatMessageError, ChatPluginPayload } from '@/types/message';
 
+export interface CreateMessageParams
+  extends Partial<Omit<ChatMessage, 'content' | 'role'>>,
+    Pick<ChatMessage, 'content' | 'role'> {
+  fromModel?: string;
+  fromProvider?: string;
+  sessionId: string;
+  traceId?: string;
+}
+
 export class MessageService {
   async create(data: CreateMessageParams) {
     const { id } = await MessageModel.create(data);

package/src/services/plugin.ts

@@ -4,7 +4,7 @@ import {
   pluginManifestSchema,
 } from '@lobehub/chat-plugin-sdk';
 
-import { PluginModel } from '@/database/models/plugin';
+import { PluginModel } from '@/database/client/models/plugin';
 import { globalHelpers } from '@/store/global/helpers';
 import { OpenAIPluginManifest } from '@/types/openai/plugin';
 import { LobeTool } from '@/types/tool';

package/src/services/session.ts

@@ -1,7 +1,7 @@
 import { DeepPartial } from 'utility-types';
 
-import { SessionModel } from '@/database/models/session';
-import { SessionGroupModel } from '@/database/models/sessionGroup';
+import { SessionModel } from '@/database/client/models/session';
+import { SessionGroupModel } from '@/database/client/models/sessionGroup';
 import { LobeAgentConfig } from '@/types/agent';
 import { MetaData } from '@/types/meta';
 import {

package/src/services/topic.ts

@@ -1,4 +1,4 @@
-import { CreateTopicParams, QueryTopicParams, TopicModel } from '@/database/models/topic';
+import { CreateTopicParams, QueryTopicParams, TopicModel } from '@/database/client/models/topic';
 import { ChatTopic } from '@/types/topic';
 
 class TopicService {

package/src/services/user.ts

@@ -1,6 +1,6 @@
 import { DeepPartial } from 'utility-types';
 
-import { UserModel } from '@/database/models/user';
+import { UserModel } from '@/database/client/models/user';
 import { GlobalSettings } from '@/types/settings';
 
 export interface UserConfig {

package/src/store/chat/slices/message/action.ts

@@ -7,10 +7,9 @@ import { StateCreator } from 'zustand/vanilla';
 
 import { LOADING_FLAT, isFunctionMessageAtStart, testFunctionMessageAtEnd } from '@/const/message';
 import { TraceEventType, TraceNameMap } from '@/const/trace';
-import { CreateMessageParams } from '@/database/models/message';
 import { useClientDataSWR } from '@/libs/swr';
 import { chatService } from '@/services/chat';
-import { messageService } from '@/services/message';
+import { CreateMessageParams, messageService } from '@/services/message';
 import { topicService } from '@/services/topic';
 import { traceService } from '@/services/trace';
 import { chatHelpers } from '@/store/chat/helpers';

package/src/store/chat/slices/plugin/action.ts

@@ -4,9 +4,8 @@ import { Md5 } from 'ts-md5';
 import { StateCreator } from 'zustand/vanilla';
 
 import { PLUGIN_SCHEMA_API_MD5_PREFIX, PLUGIN_SCHEMA_SEPARATOR } from '@/const/plugin';
-import { CreateMessageParams } from '@/database/models/message';
 import { chatService } from '@/services/chat';
-import { messageService } from '@/services/message';
+import { CreateMessageParams, messageService } from '@/services/message';
 import { ChatStore } from '@/store/chat/store';
 import { useToolStore } from '@/store/tool';
 import { pluginSelectors } from '@/store/tool/selectors';

package/src/types/sync.ts

@@ -1,4 +1,4 @@
-import { LobeDBSchemaMap } from '@/database/core/db';
+import { LobeDBSchemaMap } from '@/database/client/core/db';
 
 export type OnSyncEvent = (tableKey: keyof LobeDBSchemaMap) => void;
 export type OnSyncStatusChange = (status: PeerSyncStatus) => void;