@lobehub/lobehub 2.0.0-next.2 → 2.0.0-next.21

package/src/services/chat/__snapshots__/chat.test.ts.snap

@@ -1,110 +0,0 @@
-// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
-
-exports[`ChatService > createAssistantMessage > with tools messages > work with dalle3 1`] = `
-{
-  "enabledSearch": undefined,
-  "messages": [
-    {
-      "content": "<plugins description="The plugins you can use below">
-<collection name="DALL·E 3">
-<collection.instructions>Whenever a description of an image is given, use lobe-image-designer to create the images and then summarize the prompts used to generate the images in plain text. If the user does not ask for a specific number of images, default to creating four captions to send to lobe-image-designer that are written to be as diverse as possible.
-
-  All captions sent to lobe-image-designer must abide by the following policies:
-
-1. If the description is not in English, then translate it.
-2. Do not create more than 4 images, even if the user requests more.
-3. Don't create images of politicians or other public figures. Recommend other ideas instead.
-5. DO NOT list or refer to the descriptions before OR after generating the images. They should ONLY ever be written out ONCE, in the \`prompts\` field of the request. You do not need to ask for permission to generate, just do it!
-6. Always mention the image type (photo, oil painting, watercolor painting, illustration, cartoon, drawing, vector, render, etc.) at the beginning of the caption. Unless the caption suggests otherwise, make at least 1--2 of the 4 images photos.
-7. Diversify depictions of ALL images with people to include DESCENT and GENDER for EACH person using direct terms. Adjust only human descriptions.
-  - EXPLICITLY specify these attributes, not abstractly reference them.  The attributes should be specified in a minimal way and should directly describe their physical form.
-  - Your choices should be grounded in reality. For example, all of a given OCCUPATION should not be the same gender or race. Additionally, focus on creating diverse, inclusive, and exploratory scenes via the properties you choose during rewrites.  Make choices that may be insightful or unique sometimes.
-  - Use "various" or "diverse" ONLY IF the description refers to groups of more than 3 people. Do not change the number of people requested in the original description.
-  - Don't alter memes, fictional character origins, or unseen people. Maintain the original prompt's intent and prioritize quality.
-  - Do not create any imagery that would be offensive.
-
-8. Silently modify descriptions that include names or hints or references of specific people or celebrities by carefully selecting a few minimal modifications to substitute references to the people with generic descriptions that don't divulge any information about their identities, except for their genders and physiques. Do this EVEN WHEN the instructions ask for the prompt to not be changed. Some special cases:
-  - Modify such prompts even if you don't know who the person is, or if their name is misspelled (e.g. "Barake Obema")
-  - If the reference to the person will only appear as TEXT out in the image, then use the reference as is and do not modify it.
-  - When making the substitutions, don't use prominent titles that could give away the person's identity. E.g., instead of saying "president", "prime minister", or "chancellor", say "politician"; instead of saying "king", "queen", "emperor", or "empress", say "public figure"; instead of saying "Pope", say "religious figure"; and so on.
-  - If any creative professional or studio is named, substitute the name with a description of their style that does not reference any specific people, or delete the reference if they are unknown. DO NOT refer to the artist or studio's style.
-
-The prompt must intricately describe every part of the image in concrete, objective detail. THINK about what the end goal of the description is, and extrapolate that to what would make satisfying images.
-All descriptions sent to lobe-image-designer should be a paragraph of text that is extremely descriptive and detailed. Each should be more than 3 sentences long.</collection.instructions>
-<api identifier="lobe-image-designer____text2image____builtin">Create images from a text-only prompt.</api>
-</collection>
-</plugins>",
-      "role": "system",
-    },
-    {
-      "content": "https://vercel.com/ 请分析 chatGPT 关键词
-
-",
-      "role": "user",
-    },
-  ],
-  "model": "gpt-3.5-turbo-1106",
-  "tools": [
-    {
-      "function": {
-        "description": "Create images from a text-only prompt.",
-        "name": "lobe-image-designer____text2image____builtin",
-        "parameters": {
-          "properties": {
-            "prompts": {
-              "description": "The user's original image description, potentially modified to abide by the lobe-image-designer policies. If the user does not suggest a number of captions to create, create four of them. If creating multiple captions, make them as diverse as possible. If the user requested modifications to previous images, the captions should not simply be longer, but rather it should be refactored to integrate the suggestions into each of the captions. Generate no more than 4 images, even if the user requests more.",
-              "items": {
-                "type": "string",
-              },
-              "maxItems": 4,
-              "minItems": 1,
-              "type": "array",
-            },
-            "quality": {
-              "default": "standard",
-              "description": "The quality of the image that will be generated. hd creates images with finer details and greater consistency across the image.",
-              "enum": [
-                "standard",
-                "hd",
-              ],
-              "type": "string",
-            },
-            "seeds": {
-              "description": "A list of seeds to use for each prompt. If the user asks to modify a previous image, populate this field with the seed used to generate that image from the image lobe-image-designer metadata.",
-              "items": {
-                "type": "integer",
-              },
-              "type": "array",
-            },
-            "size": {
-              "default": "1024x1024",
-              "description": "The resolution of the requested image, which can be wide, square, or tall. Use 1024x1024 (square) as the default unless the prompt suggests a wide image, 1792x1024, or a full-body portrait, in which case 1024x1792 (tall) should be used instead. Always include this parameter in the request.",
-              "enum": [
-                "1792x1024",
-                "1024x1024",
-                "1024x1792",
-              ],
-              "type": "string",
-            },
-            "style": {
-              "default": "vivid",
-              "description": "The style of the generated images. Must be one of vivid or natural. Vivid causes the model to lean towards generating hyper-real and dramatic images. Natural causes the model to produce more natural, less hyper-real looking images.",
-              "enum": [
-                "vivid",
-                "natural",
-              ],
-              "type": "string",
-            },
-          },
-          "required": [
-            "prompts",
-          ],
-          "type": "object",
-        },
-      },
-      "type": "function",
-    },
-  ],
-  "top_p": 1,
-}
-`;

package/src/services/chatGroup/client.ts

@@ -1,63 +0,0 @@
-import { clientDB } from '@/database/client/db';
-import { ChatGroupModel } from '@/database/models/chatGroup';
-import {
-  ChatGroupAgentItem,
-  ChatGroupItem,
-  NewChatGroup,
-  NewChatGroupAgent,
-} from '@/database/schemas/chatGroup';
-import { BaseClientService } from '@/services/baseClientService';
-
-import { IChatGroupService } from './type';
-
-export class ClientService extends BaseClientService implements IChatGroupService {
-  private get chatGroupModel(): ChatGroupModel {
-    return new ChatGroupModel(clientDB as any, this.userId);
-  }
-
-  async createGroup(params: Omit<NewChatGroup, 'userId'>): Promise<ChatGroupItem> {
-    return this.chatGroupModel.create(params);
-  }
-
-  async updateGroup(id: string, value: Partial<ChatGroupItem>): Promise<ChatGroupItem> {
-    return this.chatGroupModel.update(id, value);
-  }
-
-  async deleteGroup(id: string): Promise<any> {
-    return this.chatGroupModel.delete(id);
-  }
-
-  async deleteAllGroups(): Promise<any> {
-    return this.chatGroupModel.deleteAll();
-  }
-
-  async getGroup(id: string): Promise<ChatGroupItem | undefined> {
-    return this.chatGroupModel.findById(id);
-  }
-
-  async getGroups(): Promise<ChatGroupItem[]> {
-    return this.chatGroupModel.queryWithMemberDetails();
-  }
-
-  async addAgentsToGroup(groupId: string, agentIds: string[]): Promise<ChatGroupAgentItem[]> {
-    return this.chatGroupModel.addAgentsToGroup(groupId, agentIds);
-  }
-
-  async removeAgentsFromGroup(groupId: string, agentIds: string[]): Promise<any> {
-    for (const agentId of agentIds) {
-      await this.chatGroupModel.removeAgentFromGroup(groupId, agentId);
-    }
-  }
-
-  async updateAgentInGroup(
-    groupId: string,
-    agentId: string,
-    updates: Partial<Pick<NewChatGroupAgent, 'order' | 'role'>>,
-  ): Promise<ChatGroupAgentItem> {
-    return this.chatGroupModel.updateAgentInGroup(groupId, agentId, updates);
-  }
-
-  async getGroupAgents(groupId: string): Promise<ChatGroupAgentItem[]> {
-    return this.chatGroupModel.getGroupAgents(groupId);
-  }
-}

package/src/services/export/_deprecated.ts

@@ -1,155 +0,0 @@
-import { Migration } from '@/migrations';
-import { messageService } from '@/services/message';
-import { sessionService } from '@/services/session';
-import { topicService } from '@/services/topic';
-import { useSessionStore } from '@/store/session';
-import { sessionSelectors } from '@/store/session/selectors';
-import { useUserStore } from '@/store/user';
-import { settingsSelectors } from '@/store/user/selectors';
-import {
-  ConfigFileAgents,
-  ConfigFileAll,
-  ConfigFileSessions,
-  ConfigFileSettings,
-  ConfigFileSingleSession,
-  ConfigModelMap,
-  ExportType,
-} from '@/types/exportConfig';
-
-type CreateConfigFileState<T extends ExportType> = ConfigModelMap[T]['state'];
-
-type CreateConfigFile<T extends ExportType> = ConfigModelMap[T]['file'];
-
-const createConfigFile = <T extends ExportType>(
-  type: T,
-  state: CreateConfigFileState<T>,
-): CreateConfigFile<T> => {
-  switch (type) {
-    case 'agents': {
-      return {
-        exportType: 'agents',
-        state,
-        version: Migration.targetVersion,
-      } as ConfigFileAgents;
-    }
-
-    case 'sessions': {
-      return {
-        exportType: 'sessions',
-        state,
-        version: Migration.targetVersion,
-      } as ConfigFileSessions;
-    }
-
-    case 'settings': {
-      return {
-        exportType: 'settings',
-        state,
-        version: Migration.targetVersion,
-      } as ConfigFileSettings;
-    }
-
-    case 'singleSession': {
-      return {
-        exportType: 'sessions',
-        state,
-        version: Migration.targetVersion,
-      } as ConfigFileSingleSession;
-    }
-
-    case 'all': {
-      return {
-        exportType: 'all',
-        state,
-        version: Migration.targetVersion,
-      } as ConfigFileAll;
-    }
-  }
-
-  throw new Error('缺少正确的导出类型，请检查实现...');
-};
-
-/**
- * @deprecated
- */
-export class ConfigService {
-  /**
-   * export all agents
-   */
-  exportAgents = async () => {
-    const agents = await sessionService.getSessionsByType('agent');
-    const sessionGroups = await sessionService.getSessionGroups();
-
-    return createConfigFile('agents', { sessionGroups, sessions: agents });
-  };
-
-  /**
-   * export all sessions
-   */
-  exportSessions = async () => {
-    const sessions = await sessionService.getSessionsByType();
-    const sessionGroups = await sessionService.getSessionGroups();
-    const messages = await messageService.getAllMessages();
-    const topics = await topicService.getAllTopics();
-
-    return createConfigFile('sessions', { messages, sessionGroups, sessions, topics });
-  };
-
-  /**
-   * export a session
-   */
-  exportSingleSession = async (id: string) => {
-    const session = this.getSession(id);
-    if (!session) return;
-
-    const messages = await messageService.getAllMessagesInSession(id);
-    const topics = await topicService.getTopics({ containerId: id });
-
-    const config = createConfigFile('singleSession', { messages, sessions: [session], topics });
-    return { config, title: `${session.meta?.title}-session` };
-  };
-
-  exportSingleAgent = async (id: string) => {
-    const agent = this.getAgent(id);
-    if (!agent) return;
-
-    const config = createConfigFile('agents', { sessionGroups: [], sessions: [agent] });
-
-    return { config, title: `${agent.meta?.title}-session` };
-  };
-
-  /**
-   * export settings
-   */
-  exportSettings = async () => {
-    const settings = this.getSettings();
-
-    return createConfigFile('settings', { settings });
-  };
-
-  /**
-   * export all data
-   */
-  exportAll = async () => {
-    const sessions = await sessionService.getSessionsByType();
-    const sessionGroups = await sessionService.getSessionGroups();
-    const messages = await messageService.getAllMessages();
-    const topics = await topicService.getAllTopics();
-    const settings = this.getSettings();
-
-    return createConfigFile('all', { messages, sessionGroups, sessions, settings, topics });
-  };
-
-  private getSettings = () => settingsSelectors.exportSettings(useUserStore.getState());
-
-  private getSession = (id: string) =>
-    sessionSelectors.getSessionById(id)(useSessionStore.getState());
-
-  private getAgent = (id: string) =>
-    sessionSelectors.getSessionById(id)(useSessionStore.getState());
-}
-
-/**
- * @deprecated
- */
-export const configService = new ConfigService();

package/src/services/export/client.ts

@@ -1,15 +0,0 @@
-import { clientDB } from '@/database/client/db';
-import { DataExporterRepos } from '@/database/repositories/dataExporter';
-import { BaseClientService } from '@/services/baseClientService';
-
-export class ClientService extends BaseClientService {
-  private get dataExporterRepos(): DataExporterRepos {
-    return new DataExporterRepos(clientDB as any, this.userId);
-  }
-
-  exportData = async () => {
-    const data = await this.dataExporterRepos.export();
-
-    return { data, url: undefined };
-  };
-}

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

@@ -1,119 +0,0 @@
-import { Mock, beforeAll, beforeEach, describe, expect, it, vi } from 'vitest';
-
-import { FileModel } from '@/database/_deprecated/models/file';
-import { DB_File } from '@/database/_deprecated/schemas/files';
-import { fileEnv } from '@/envs/file';
-import { clientS3Storage } from '@/services/file/ClientS3';
-import { serverConfigSelectors } from '@/store/serverConfig/selectors';
-import { createServerConfigStore } from '@/store/serverConfig/store';
-
-import { ClientService } from './_deprecated';
-
-const fileService = new ClientService();
-
-beforeAll(() => {
-  createServerConfigStore();
-});
-// Mocks for the FileModel
-vi.mock('@/database/_deprecated/models/file', () => ({
-  FileModel: {
-    create: vi.fn(),
-    delete: vi.fn(),
-    findById: vi.fn(),
-    clear: vi.fn(),
-  },
-}));
-
-let s3Domain: string;
-
-vi.mock('@/config/file', () => ({
-  fileEnv: {
-    get NEXT_PUBLIC_S3_DOMAIN() {
-      return s3Domain;
-    },
-  },
-}));
-
-// Mocks for the URL and Blob objects
-global.URL.createObjectURL = vi.fn();
-global.Blob = vi.fn();
-
-beforeEach(() => {
-  // Reset all mocks before each test
-  vi.resetAllMocks();
-  s3Domain = '';
-});
-
-describe('FileService', () => {
-  it('createFile should save the file to the database', async () => {
-    const localFile = {
-      name: 'test',
-      fileType: 'image/png',
-      url: 'client-s3://123',
-      size: 1,
-      hash: '123',
-    };
-
-    await clientS3Storage.putObject(
-      '123',
-      new File([new ArrayBuffer(1)], 'test.png', { type: 'image/png' }),
-    );
-
-    (FileModel.create as Mock).mockResolvedValue(localFile);
-
-    const result = await fileService.createFile(localFile);
-
-    expect(result).toEqual({ url: 'data:image/png;base64,AA==' });
-  });
-
-  it('removeFile should delete the file from the database', async () => {
-    const fileId = '1';
-    (FileModel.delete as Mock).mockResolvedValue(true);
-
-    const result = await fileService.removeFile(fileId);
-
-    expect(FileModel.delete).toHaveBeenCalledWith(fileId);
-    expect(result).toBe(true);
-  });
-
-  describe('getFile', () => {
-    it('should retrieve and convert local file info to FilePreview', async () => {
-      const fileId = '1';
-      const fileData = {
-        name: 'test',
-        data: new ArrayBuffer(1),
-        fileType: 'image/png',
-        saveMode: 'local',
-        size: 1,
-        createdAt: 1,
-        updatedAt: 2,
-      } as DB_File;
-
-      (FileModel.findById as Mock).mockResolvedValue(fileData);
-      (global.URL.createObjectURL as Mock).mockReturnValue('blob:test');
-      (global.Blob as Mock).mockImplementation(() => ['test']);
-
-      const result = await fileService.getFile(fileId);
-
-      expect(FileModel.findById).toHaveBeenCalledWith(fileId);
-      expect(result).toEqual({
-        createdAt: new Date(1),
-        id: '1',
-        size: 1,
-        type: 'image/png',
-        name: 'test',
-        url: 'blob:test',
-        updatedAt: new Date(2),
-      });
-    });
-
-    it('should throw an error when the file is not found', async () => {
-      const fileId = 'non-existent';
-      (FileModel.findById as Mock).mockResolvedValue(null);
-
-      const getFilePromise = fileService.getFile(fileId);
-
-      await expect(getFilePromise).rejects.toThrow('file not found');
-    });
-  });
-});

package/src/services/file/_deprecated.ts

@@ -1,80 +0,0 @@
-import { FileModel } from '@/database/_deprecated/models/file';
-import { clientS3Storage } from '@/services/file/ClientS3';
-import { FileItem, UploadFileParams } from '@/types/files';
-
-import { IFileService } from './type';
-
-export class ClientService implements IFileService {
-  async createFile(file: UploadFileParams) {
-    // save to local storage
-    // we may want to save to a remote server later
-    const res = await FileModel.create({
-      createdAt: Date.now(),
-      data: undefined,
-      fileHash: file.hash,
-      fileType: file.fileType,
-      metadata: file.metadata,
-      name: file.name,
-      saveMode: 'url',
-      size: file.size,
-      url: file.url,
-    } as any);
-
-    // get file to base64 url
-    const base64 = await this.getBase64ByFileHash(file.hash!);
-
-    return {
-      id: res.id,
-      url: `data:${file.fileType};base64,${base64}`,
-    };
-  }
-
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async checkFileHash(_hash: string) {
-    return { isExist: false, metadata: {}, url: '' };
-  }
-
-  async getFile(id: string): Promise<FileItem> {
-    const item = await FileModel.findById(id);
-    if (!item) {
-      throw new Error('file not found');
-    }
-
-    // arrayBuffer to blob or base64 to blob
-    const blob = !!item.data
-      ? new Blob([item.data!], { type: item.fileType })
-      : // @ts-ignore
-        new Blob([Buffer.from(item.base64!, 'base64')], { type: item.fileType });
-
-    const url = URL.createObjectURL(blob);
-
-    return {
-      createdAt: new Date(item.createdAt),
-      id,
-      name: item.name,
-      size: item.size,
-      type: item.fileType,
-      updatedAt: new Date(item.updatedAt),
-      url,
-    };
-  }
-
-  async removeFile(id: string) {
-    return FileModel.delete(id);
-  }
-
-  async removeFiles(ids: string[]) {
-    await Promise.all(ids.map((id) => FileModel.delete(id)));
-  }
-
-  async removeAllFiles() {
-    return FileModel.clear();
-  }
-
-  private async getBase64ByFileHash(hash: string) {
-    const fileItem = await clientS3Storage.getObject(hash);
-    if (!fileItem) throw new Error('file not found');
-
-    return Buffer.from(await fileItem.arrayBuffer()).toString('base64');
-  }
-}