@lobehub/chat 0.147.11 → 0.147.13

package/src/services/__tests__/{plugin.test.ts → tool.test.ts}

@@ -1,13 +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 { globalHelpers } from '@/store/global/helpers';
-import { LobeTool } from '@/types/tool';
-import { LobeToolCustomPlugin } from '@/types/tool/plugin';
 
-import { InstallPluginParams, pluginService } from '../plugin';
+import { toolService } from '../tool';
 import openAPIV3 from './openai/OpenAPI_V3.json';
 import OpenAIPlugin from './openai/plugin.json';
 
@@ -18,21 +13,12 @@ vi.mock('@/store/global/helpers', () => ({
     getCurrentLanguage: vi.fn(),
   },
 }));
-vi.mock('@/database/models/plugin', () => ({
-  PluginModel: {
-    getList: vi.fn(),
-    create: vi.fn(),
-    delete: vi.fn(),
-    update: vi.fn(),
-    clear: vi.fn(),
-  },
-}));
 
 beforeEach(() => {
   vi.resetAllMocks();
 });
 
-describe('PluginService', () => {
+describe('ToolService', () => {
   describe('getPluginList', () => {
     it('should fetch and return the plugin list', async () => {
       // Arrange
@@ -45,7 +31,7 @@ describe('PluginService', () => {
       ) as any;
 
       // Act
-      const pluginList = await pluginService.getPluginList();
+      const pluginList = await toolService.getPluginList();
 
       // Assert
       expect(globalHelpers.getCurrentLanguage).toHaveBeenCalled();
@@ -60,7 +46,7 @@ describe('PluginService', () => {
       global.fetch = vi.fn(() => Promise.reject(new Error('Network error')));
 
       // Act & Assert
-      await expect(pluginService.getPluginList()).rejects.toThrow('Network error');
+      await expect(toolService.getPluginList()).rejects.toThrow('Network error');
     });
   });
 
@@ -112,7 +98,7 @@ describe('PluginService', () => {
         }),
       ) as any;
 
-      const manifest = await pluginService.getPluginManifest(manifestUrl);
+      const manifest = await toolService.getPluginManifest(manifestUrl);
 
       expect(fetch).toHaveBeenCalledWith(manifestUrl);
       expect(manifest).toEqual(fakeManifest);
@@ -120,7 +106,7 @@ describe('PluginService', () => {
 
     it('should return error on noManifest', async () => {
       try {
-        await pluginService.getPluginManifest();
+        await toolService.getPluginManifest();
       } catch (e) {
         expect(e).toEqual(new TypeError('noManifest'));
       }
@@ -138,7 +124,7 @@ describe('PluginService', () => {
       ) as any;
 
       try {
-        await pluginService.getPluginManifest(manifestUrl);
+        await toolService.getPluginManifest(manifestUrl);
       } catch (e) {
         expect(e).toEqual(new TypeError('manifestInvalid'));
       }
@@ -149,7 +135,7 @@ describe('PluginService', () => {
       global.fetch = vi.fn(() => Promise.reject(new Error('Network error')));
 
       try {
-        await pluginService.getPluginManifest(manifestUrl);
+        await toolService.getPluginManifest(manifestUrl);
       } catch (e) {
         expect(e).toEqual(new TypeError('fetchError'));
       }
@@ -170,7 +156,7 @@ describe('PluginService', () => {
       ) as any;
 
       try {
-        await pluginService.getPluginManifest(manifestUrl);
+        await toolService.getPluginManifest(manifestUrl);
       } catch (e) {
         expect(e).toEqual(new TypeError('urlError'));
       }
@@ -188,7 +174,7 @@ describe('PluginService', () => {
       ) as any;
 
       try {
-        await pluginService.getPluginManifest(manifestUrl);
+        await toolService.getPluginManifest(manifestUrl);
       } catch (e) {
         expect(e).toEqual(new TypeError('fetchError'));
       }
@@ -228,7 +214,7 @@ describe('PluginService', () => {
           }),
         ) as any;
 
-        const manifest = await pluginService.getPluginManifest(manifestUrl);
+        const manifest = await toolService.getPluginManifest(manifestUrl);
 
         expect(manifest).toMatchSnapshot();
       });
@@ -266,7 +252,7 @@ describe('PluginService', () => {
         ) as any;
 
         try {
-          await pluginService.getPluginManifest(manifestUrl);
+          await toolService.getPluginManifest(manifestUrl);
         } catch (e) {
           expect(e).toEqual(new TypeError('openAPIInvalid'));
         }
@@ -274,141 +260,8 @@ describe('PluginService', () => {
     });
   });
 
-  describe('installPlugin', () => {
-    it('should install a plugin', async () => {
-      // Arrange
-      const fakePlugin = {
-        identifier: 'test-plugin',
-        manifest: { name: 'TestPlugin', version: '1.0.0' } as unknown as LobeChatPluginManifest,
-        type: 'plugin',
-      } as InstallPluginParams;
-      vi.mocked(PluginModel.create).mockResolvedValue(fakePlugin);
-
-      // Act
-      const installedPlugin = await pluginService.installPlugin(fakePlugin);
-
-      // Assert
-      expect(PluginModel.create).toHaveBeenCalledWith(fakePlugin);
-      expect(installedPlugin).toEqual(fakePlugin);
-    });
-  });
-
-  describe('getInstalledPlugins', () => {
-    it('should return a list of installed plugins', async () => {
-      // Arrange
-      const fakePlugins = [{ identifier: 'test-plugin', type: 'plugin' }] as LobeTool[];
-      vi.mocked(PluginModel.getList).mockResolvedValue(fakePlugins as DB_Plugin[]);
-
-      // Act
-      const installedPlugins = await pluginService.getInstalledPlugins();
-
-      // Assert
-      expect(PluginModel.getList).toHaveBeenCalled();
-      expect(installedPlugins).toEqual(fakePlugins);
-    });
-  });
-
-  describe('uninstallPlugin', () => {
-    it('should uninstall a plugin', async () => {
-      // Arrange
-      const identifier = 'test-plugin';
-      vi.mocked(PluginModel.delete).mockResolvedValue();
-
-      // Act
-      const result = await pluginService.uninstallPlugin(identifier);
-
-      // Assert
-      expect(PluginModel.delete).toHaveBeenCalledWith(identifier);
-      expect(result).toBe(undefined);
-    });
-  });
-
-  describe('createCustomPlugin', () => {
-    it('should create a custom plugin', async () => {
-      // Arrange
-      const customPlugin = {
-        identifier: 'custom-plugin',
-        manifest: {},
-        type: 'customPlugin',
-      } as LobeToolCustomPlugin;
-      vi.mocked(PluginModel.create).mockResolvedValue(customPlugin);
-
-      // Act
-      const result = await pluginService.createCustomPlugin(customPlugin);
-
-      // Assert
-      expect(PluginModel.create).toHaveBeenCalledWith({
-        ...customPlugin,
-        type: 'customPlugin',
-      });
-      expect(result).toEqual(customPlugin);
-    });
-  });
-
-  describe('updatePlugin', () => {
-    it('should update a plugin', async () => {
-      // Arrange
-      const id = 'plugin-id';
-      const value = { settings: { ab: '1' } } as unknown as LobeToolCustomPlugin;
-      vi.mocked(PluginModel.update).mockResolvedValue(1);
-
-      // Act
-      const result = await pluginService.updatePlugin(id, value);
-
-      // Assert
-      expect(PluginModel.update).toHaveBeenCalledWith(id, value);
-      expect(result).toEqual(1);
-    });
-  });
-
-  describe('updatePluginManifest', () => {
-    it('should update a plugin manifest', async () => {
-      // Arrange
-      const id = 'plugin-id';
-      const manifest = { name: 'NewPluginManifest' } as unknown as LobeChatPluginManifest;
-      vi.mocked(PluginModel.update).mockResolvedValue(1);
-
-      // Act
-      const result = await pluginService.updatePluginManifest(id, manifest);
-
-      // Assert
-      expect(PluginModel.update).toHaveBeenCalledWith(id, { manifest });
-      expect(result).toEqual(1);
-    });
-  });
-
-  describe('removeAllPlugins', () => {
-    it('should remove all plugins', async () => {
-      // Arrange
-      vi.mocked(PluginModel.clear).mockResolvedValue(undefined);
-
-      // Act
-      const result = await pluginService.removeAllPlugins();
-
-      // Assert
-      expect(PluginModel.clear).toHaveBeenCalled();
-      expect(result).toBe(undefined);
-    });
-  });
-
-  describe('updatePluginSettings', () => {
-    it('should update plugin settings', async () => {
-      // Arrange
-      const id = 'plugin-id';
-      const settings = { color: 'blue' };
-      vi.mocked(PluginModel.update).mockResolvedValue(1);
-
-      // Act
-      const result = await pluginService.updatePluginSettings(id, settings);
-
-      // Assert
-      expect(PluginModel.update).toHaveBeenCalledWith(id, { settings });
-      expect(result).toEqual(1);
-    });
-  });
-
   it('can parse the OpenAI plugin', async () => {
-    const manifest = pluginService['convertOpenAIManifestToLobeManifest'](OpenAIPlugin as any);
+    const manifest = toolService['convertOpenAIManifestToLobeManifest'](OpenAIPlugin as any);
 
     expect(manifest).toMatchSnapshot();
   });

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 → file/client.ts}

@@ -1,11 +1,11 @@
-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';
 
-import { API_ENDPOINTS } from './_url';
+import { API_ENDPOINTS } from '../_url';
 
-class FileService {
+export class FileService {
   private isImage(fileType: string) {
     const imageRegex = /^image\//;
     return imageRegex.test(fileType);
@@ -84,5 +84,3 @@ class FileService {
     };
   }
 }
-
-export const fileService = new FileService();

package/src/services/{__tests__ → file}/file.test.ts

@@ -1,12 +1,14 @@
 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';
+import { FileService } from './client';
+
+const fileService = new FileService();
 
 // 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/file/index.ts

@@ -0,0 +1,3 @@
+import { FileService } from './client';
+
+export const fileService = new FileService();

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/{__tests__/message.test.ts → message/client.test.ts}

@@ -1,18 +1,14 @@
 import { Mock, describe, expect, it, vi } from 'vitest';
 
-import { CreateMessageParams, MessageModel } from '@/database/models/message';
-import {
-  ChatMessage,
-  ChatMessageError,
-  ChatPluginPayload,
-  ChatTTS,
-  ChatTranslate,
-} from '@/types/message';
+import { CreateMessageParams, MessageModel } from '@/database/client/models/message';
+import { ChatMessage, ChatMessageError, ChatPluginPayload } from '@/types/message';
 
-import { messageService } from '../message';
+import { MessageService } from './client';
+
+const messageService = new MessageService();
 
 // Mock the MessageModel
-vi.mock('@/database/models/message', () => {
+vi.mock('@/database/client/models/message', () => {
   return {
     MessageModel: {
       create: vi.fn(),

package/src/services/{message.ts → message/client.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);
@@ -67,5 +76,3 @@ export class MessageService {
     return MessageModel.queryAll();
   }
 }
-
-export const messageService = new MessageService();

package/src/services/message/index.ts

@@ -0,0 +1,4 @@
+import { MessageService } from './client';
+
+export const messageService = new MessageService();
+export type { CreateMessageParams } from './client';

package/src/services/plugin/client.ts

@@ -0,0 +1,44 @@
+import { LobeChatPluginManifest } from '@lobehub/chat-plugin-sdk';
+
+import { PluginModel } from '@/database/client/models/plugin';
+import { LobeTool } from '@/types/tool';
+import { LobeToolCustomPlugin } from '@/types/tool/plugin';
+
+export interface InstallPluginParams {
+  identifier: string;
+  manifest: LobeChatPluginManifest;
+  type: 'plugin' | 'customPlugin';
+}
+
+export class PluginService {
+  installPlugin = async (plugin: InstallPluginParams) => {
+    return PluginModel.create(plugin);
+  };
+
+  getInstalledPlugins = () => {
+    return PluginModel.getList() as Promise<LobeTool[]>;
+  };
+
+  uninstallPlugin(identifier: string) {
+    return PluginModel.delete(identifier);
+  }
+
+  async createCustomPlugin(customPlugin: LobeToolCustomPlugin) {
+    return PluginModel.create({ ...customPlugin, type: 'customPlugin' });
+  }
+
+  async updatePlugin(id: string, value: LobeToolCustomPlugin) {
+    return PluginModel.update(id, value);
+  }
+  async updatePluginManifest(id: string, manifest: LobeChatPluginManifest) {
+    return PluginModel.update(id, { manifest });
+  }
+
+  async removeAllPlugins() {
+    return PluginModel.clear();
+  }
+
+  async updatePluginSettings(id: string, settings: any) {
+    return PluginModel.update(id, { settings });
+  }
+}

package/src/services/plugin/index.ts

@@ -0,0 +1,5 @@
+import { PluginService } from './client';
+
+export type { InstallPluginParams } from './client';
+
+export const pluginService = new PluginService();

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

@@ -0,0 +1,162 @@
+import { LobeChatPluginManifest } from '@lobehub/chat-plugin-sdk';
+import { beforeEach, describe, expect, it, vi } from 'vitest';
+
+import { PluginModel } from '@/database/client/models/plugin';
+import { DB_Plugin } from '@/database/client/schemas/plugin';
+import { LobeTool } from '@/types/tool';
+import { LobeToolCustomPlugin } from '@/types/tool/plugin';
+
+import { InstallPluginParams, PluginService } from './client';
+
+const pluginService = new PluginService();
+
+// Mocking modules and functions
+
+vi.mock('@/database/client/models/plugin', () => ({
+  PluginModel: {
+    getList: vi.fn(),
+    create: vi.fn(),
+    delete: vi.fn(),
+    update: vi.fn(),
+    clear: vi.fn(),
+  },
+}));
+
+beforeEach(() => {
+  vi.resetAllMocks();
+});
+
+describe('PluginService', () => {
+  describe('installPlugin', () => {
+    it('should install a plugin', async () => {
+      // Arrange
+      const fakePlugin = {
+        identifier: 'test-plugin',
+        manifest: { name: 'TestPlugin', version: '1.0.0' } as unknown as LobeChatPluginManifest,
+        type: 'plugin',
+      } as InstallPluginParams;
+      vi.mocked(PluginModel.create).mockResolvedValue(fakePlugin);
+
+      // Act
+      const installedPlugin = await pluginService.installPlugin(fakePlugin);
+
+      // Assert
+      expect(PluginModel.create).toHaveBeenCalledWith(fakePlugin);
+      expect(installedPlugin).toEqual(fakePlugin);
+    });
+  });
+
+  describe('getInstalledPlugins', () => {
+    it('should return a list of installed plugins', async () => {
+      // Arrange
+      const fakePlugins = [{ identifier: 'test-plugin', type: 'plugin' }] as LobeTool[];
+      vi.mocked(PluginModel.getList).mockResolvedValue(fakePlugins as DB_Plugin[]);
+
+      // Act
+      const installedPlugins = await pluginService.getInstalledPlugins();
+
+      // Assert
+      expect(PluginModel.getList).toHaveBeenCalled();
+      expect(installedPlugins).toEqual(fakePlugins);
+    });
+  });
+
+  describe('uninstallPlugin', () => {
+    it('should uninstall a plugin', async () => {
+      // Arrange
+      const identifier = 'test-plugin';
+      vi.mocked(PluginModel.delete).mockResolvedValue();
+
+      // Act
+      const result = await pluginService.uninstallPlugin(identifier);
+
+      // Assert
+      expect(PluginModel.delete).toHaveBeenCalledWith(identifier);
+      expect(result).toBe(undefined);
+    });
+  });
+
+  describe('createCustomPlugin', () => {
+    it('should create a custom plugin', async () => {
+      // Arrange
+      const customPlugin = {
+        identifier: 'custom-plugin',
+        manifest: {},
+        type: 'customPlugin',
+      } as LobeToolCustomPlugin;
+      vi.mocked(PluginModel.create).mockResolvedValue(customPlugin);
+
+      // Act
+      const result = await pluginService.createCustomPlugin(customPlugin);
+
+      // Assert
+      expect(PluginModel.create).toHaveBeenCalledWith({
+        ...customPlugin,
+        type: 'customPlugin',
+      });
+      expect(result).toEqual(customPlugin);
+    });
+  });
+
+  describe('updatePlugin', () => {
+    it('should update a plugin', async () => {
+      // Arrange
+      const id = 'plugin-id';
+      const value = { settings: { ab: '1' } } as unknown as LobeToolCustomPlugin;
+      vi.mocked(PluginModel.update).mockResolvedValue(1);
+
+      // Act
+      const result = await pluginService.updatePlugin(id, value);
+
+      // Assert
+      expect(PluginModel.update).toHaveBeenCalledWith(id, value);
+      expect(result).toEqual(1);
+    });
+  });
+
+  describe('updatePluginManifest', () => {
+    it('should update a plugin manifest', async () => {
+      // Arrange
+      const id = 'plugin-id';
+      const manifest = { name: 'NewPluginManifest' } as unknown as LobeChatPluginManifest;
+      vi.mocked(PluginModel.update).mockResolvedValue(1);
+
+      // Act
+      const result = await pluginService.updatePluginManifest(id, manifest);
+
+      // Assert
+      expect(PluginModel.update).toHaveBeenCalledWith(id, { manifest });
+      expect(result).toEqual(1);
+    });
+  });
+
+  describe('removeAllPlugins', () => {
+    it('should remove all plugins', async () => {
+      // Arrange
+      vi.mocked(PluginModel.clear).mockResolvedValue(undefined);
+
+      // Act
+      const result = await pluginService.removeAllPlugins();
+
+      // Assert
+      expect(PluginModel.clear).toHaveBeenCalled();
+      expect(result).toBe(undefined);
+    });
+  });
+
+  describe('updatePluginSettings', () => {
+    it('should update plugin settings', async () => {
+      // Arrange
+      const id = 'plugin-id';
+      const settings = { color: 'blue' };
+      vi.mocked(PluginModel.update).mockResolvedValue(1);
+
+      // Act
+      const result = await pluginService.updatePluginSettings(id, settings);
+
+      // Assert
+      expect(PluginModel.update).toHaveBeenCalledWith(id, { settings });
+      expect(result).toEqual(1);
+    });
+  });
+});

package/src/services/{session.ts → session/client.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 {
@@ -14,7 +14,7 @@ import {
   SessionGroups,
 } from '@/types/session';
 
-class SessionService {
+export class SessionService {
   async createNewSession(
     type: LobeSessionType,
     defaultValue: Partial<LobeAgentSession>,
@@ -118,5 +118,3 @@ class SessionService {
     return SessionGroupModel.clear();
   }
 }
-
-export const sessionService = new SessionService();

package/src/services/session/index.ts

@@ -0,0 +1,3 @@
+import { SessionService } from './client';
+
+export const sessionService = new SessionService();

package/src/services/{__tests__ → session}/session.test.ts

@@ -1,14 +1,16 @@
 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';
+import { SessionService } from './client';
+
+const sessionService = new SessionService();
 
 // Mock the SessionModel
-vi.mock('@/database/models/session', () => {
+vi.mock('@/database/client/models/session', () => {
   return {
     SessionModel: {
       create: vi.fn(),
@@ -29,7 +31,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(),