@lobehub/chat 0.105.2 → 0.107.0

package/src/migrations/FromV1ToV2/fixtures/output-v2.json

@@ -0,0 +1,202 @@
+{
+  "exportType": "sessions",
+  "state": {
+    "sessions": [
+      {
+        "config": {
+          "displayMode": "chat",
+          "historyCount": 1,
+          "model": "gpt-3.5-turbo",
+          "params": {
+            "frequency_penalty": 0,
+            "presence_penalty": 0,
+            "temperature": 0.6,
+            "top_p": 1,
+            "max_tokens": 2200
+          },
+          "plugins": [],
+          "systemRole": "你是一名 Postgresql 的数据库专家。用户是一名数据库小白，你需要使用简单直白的方式告诉用户如何使用 ostgresql 和它相应的 orm 工具 primsa",
+          "enableCompressThreshold": false,
+          "enableMaxTokens": false,
+          "inputTemplate": ""
+        },
+        "createdAt": 1698463356269,
+        "id": "06cc3e20-e870-4099-a619-c07a849d742d",
+        "meta": {
+          "avatar": "🐘",
+          "backgroundColor": "rgba(0,0,0,0)",
+          "title": "Postgresql 指南",
+          "description": "PostgreSQL 数据库和 Prisma ORM 工具的简单使用指南",
+          "tags": ["数据库", "PostgreSQL", "教程", "ORM", "Prisma"]
+        },
+        "type": "agent",
+        "updatedAt": 1698463356269,
+        "group": "pinned"
+      }
+    ],
+    "topics": [
+      {
+        "createdAt": 1698463911747,
+        "id": "CZg3IEdy",
+        "title": "Prisma中的cuid函数作用是什么",
+        "sessionId": "06cc3e20-e870-4099-a619-c07a849d742d",
+        "updatedAt": 1698463915716
+      },
+      {
+        "createdAt": 1698754463117,
+        "id": "MNWhBEh5",
+        "title": "启动 PostgreSQL Docker 镜像方法",
+        "sessionId": "06cc3e20-e870-4099-a619-c07a849d742d",
+        "updatedAt": 1698754464867
+      }
+    ],
+    "messages": [
+      {
+        "content": "是的，你可以使用 Prisma 来自定义生成 `id` 的逻辑，例如使用 `nanoid` 库来生成唯一标识符。\n\n首先，你需要在项目中安装 `nanoid` 库：\n\n```\nnpm install nanoid\n```\n\n然后，你可以在 Prisma 的模型定义中使用 `@default` 指令，并提供一个自定义的默认值函数来生成 `id`。\n\n```prisma\nmodel Message {\n  id String @id @default(dbgenerated(\"nanoid(10)\"))\n\n  createdAt DateTime @default(now())\n  updatedAt DateTime @updatedAt\n}\n```\n\n在上述模型定义中，我们使用 `dbgenerated` 指令来定义一个自定义的默认值函数。函数 `nanoid(10)` 会生成一个长度为 10 的唯一标识符。\n\n请注意，使用自定义的 `id` 生成逻辑可能会对数据库性能产生一些影响，因为每次创建新记录时都需要执行自定义函数。因此，请确保你的自定义逻辑是高效的，并在实际使用中进行测试和评估性能。",
+        "createdAt": 1698463981313,
+        "id": "n2Tsl3Dm",
+        "meta": {},
+        "parentId": "JSZaJOGj",
+        "role": "assistant",
+        "updatedAt": 1698464004494,
+        "sessionId": "06cc3e20-e870-4099-a619-c07a849d742d",
+        "topicId": "CZg3IEdy",
+        "fromModel": "gpt-3.5-turbo"
+      },
+      {
+        "content": "如何指定 自定义函数的引入位置？",
+        "createdAt": 1698464173224,
+        "id": "xPft8Msa",
+        "meta": {},
+        "role": "user",
+        "updatedAt": 1698464173226,
+        "sessionId": "06cc3e20-e870-4099-a619-c07a849d742d",
+        "topicId": "CZg3IEdy"
+      },
+      {
+        "content": "在 Prisma 中，你可以通过在 `schema.prisma` 文件的顶部使用 `generator` 块来指定自定义函数的引入位置。\n\n例如，假设你想使用 `nanoid` 库来生成自定义的 `id`，你可以按照以下方式配置：\n\n```prisma\ngenerator client {\n  provider = \"prisma-client-js\"\n  output   = \"./node_modules/@prisma/client\"\n}\n\ngenerator dbgenerated {\n  provider = \"prisma-dbgenerator\"\n  output   = \"./node_modules/@prisma/client\"\n  binaryTargets = [\"native\"]\n  previewFeatures = [\"nativeTypes\"]\n  config = {\n    \"generator\": {\n      \"dbgenerated\": {\n        \"output\": \"./src/db/generated\",\n        \"provider\": \"./src/db/generated/provider.js\"\n      }\n    }\n  }\n}\n```\n\n在上述配置中，我们定义了两个生成器。`client` 生成器用于生成 Prisma 客户端代码，而 `dbgenerated` 生成器用于生成数据库相关的代码。\n\n注意到 `dbgenerated` 生成器的 `config` 部分中有一个 `provider` 字段，它指定了自定义函数的引入位置。在这个示例中，我们将自定义函数放在 `./src/db/generated/provider.js` 文件中。\n\n你可以根据你的项目结构和需要来调整路径和文件名，确保指定的 `provider` 路径正确引入了自定义函数。",
+        "createdAt": 1698464173228,
+        "id": "jZc2KKRs",
+        "meta": {},
+        "parentId": "xPft8Msa",
+        "role": "assistant",
+        "updatedAt": 1698464181896,
+        "sessionId": "06cc3e20-e870-4099-a619-c07a849d742d",
+        "topicId": "CZg3IEdy",
+        "fromModel": "gpt-3.5-turbo"
+      },
+      {
+        "content": "如何启动一个 postgres Docker 镜像？",
+        "createdAt": 1698754460196,
+        "id": "LjG9XSuq",
+        "meta": {},
+        "role": "user",
+        "updatedAt": 1698754463118,
+        "sessionId": "06cc3e20-e870-4099-a619-c07a849d742d",
+        "topicId": "MNWhBEh5"
+      },
+      {
+        "content": "要启动一个 PostgreSQL Docker 镜像，你可以按照以下步骤操作：\n\n1. 首先，确保你已经安装了 Docker，可以在终端中运行 `docker --version` 来检查。\n\n2. 在终端中运行以下命令以下载并启动一个 PostgreSQL 镜像：\n\n   ```shell\n   docker run --name my-postgres -e POSTGRES_PASSWORD=mysecretpassword -p 5432:5432 -d postgres\n   ```\n\n   这将从 Docker Hub 下载 PostgreSQL 镜像",
+        "createdAt": 1698754479751,
+        "id": "CL7E1Mc5",
+        "meta": {},
+        "parentId": "LjG9XSuq",
+        "role": "assistant",
+        "updatedAt": 1698754507690,
+        "sessionId": "06cc3e20-e870-4099-a619-c07a849d742d",
+        "topicId": "MNWhBEh5",
+        "fromModel": "gpt-3.5-turbo"
+      },
+      {
+        "content": "如何指定本地存储的目录到 /www/database/postgres",
+        "createdAt": 1698754591763,
+        "id": "UIvHZr7m",
+        "meta": {},
+        "role": "user",
+        "updatedAt": 1698754591764,
+        "sessionId": "06cc3e20-e870-4099-a619-c07a849d742d",
+        "topicId": "MNWhBEh5"
+      },
+      {
+        "content": "要将本地存储的目录映射到容器内的 `/var/lib/postgresql/data` 目录，你可以使用 Docker 的 `-v` 参数。\n\n以下是示例命令：\n\n```shell\ndocker run --name my-postgres -e POSTGRES_PASSWORD=mysecretpassword -p 5432:5432 -v /www/database/postgres:/var/lib/postgresql/data -d postgres\n```\n\n在这个命令中，我们使用了 `-v` 参数来指定本地目录 `/www/database/postgres` 映射到容器内的 `/var/lib/postgresql/data` 目录。这样，所有的 PostgreSQL 数据将会保存在本地指定的目录中。",
+        "createdAt": 1698754591765,
+        "id": "XLJOK44r",
+        "meta": {},
+        "parentId": "UIvHZr7m",
+        "role": "assistant",
+        "updatedAt": 1698754609275,
+        "sessionId": "06cc3e20-e870-4099-a619-c07a849d742d",
+        "topicId": "MNWhBEh5",
+        "fromModel": "gpt-3.5-turbo"
+      },
+      {
+        "content": "docker run --name postgres -e POSTGRES_PASSWORD=DSLapHYdzNrai5j6 -p 33300:5432 -v /www/database/postgres:/var/lib/postgresql/data -d postgres",
+        "createdAt": 1698754700529,
+        "id": "W3J6jQjm",
+        "meta": {},
+        "role": "user",
+        "updatedAt": 1698754700530,
+        "sessionId": "06cc3e20-e870-4099-a619-c07a849d742d",
+        "topicId": "MNWhBEh5"
+      },
+      {
+        "content": "如何启动 pgadmin",
+        "createdAt": 1698755749886,
+        "id": "L7mq0ehV",
+        "meta": {},
+        "role": "user",
+        "updatedAt": 1698755749888,
+        "sessionId": "06cc3e20-e870-4099-a619-c07a849d742d",
+        "topicId": "MNWhBEh5"
+      },
+      {
+        "content": "要启动 pgAdmin，你可以按照以下步骤操作：\n\n1. 首先，确保你已经安装了 Docker，并且已经成功启动了 PostgreSQL Docker 容器。\n\n2. 在终端中运行以下命令以启动 pgAdmin Docker 容器：\n\n   ```shell\n   docker run --name my-pgadmin -p 5050:80 -e PGADMIN_DEFAULT_EMAIL=myemail@example.com -e PGADMIN_DEFAULT_PASSWORD=mypassword -d dpage/pgadmin4\n   ```\n\n   在这个命令中，我们使用了 `-p` 参数来将容器的端口 80 映射到本地的端口 5050。你可以根据需要修改这个端口映射。\n\n   我们还使用了 `-e` 参数来设置 pgAdmin 的默认登录邮箱和密码。你可以根据需要修改这些值。\n\n3. 在浏览器中访问 `http://localhost:5050`，你将看到 pgAdmin 的登录界面。\n\n4. 使用在上一步中设置的邮箱和密码进行登录。\n\n5. 在 pgAdmin 中，你可以添加一个新的服务器连接，以连接到运行在 Docker 容器中的 PostgreSQL 数据库。要添加一个新的服务器连接，你需要提供以下信息：\n\n   - 主机名/地址：使用 Docker 容器的 IP 地址或者 `localhost`。\n   - 端口：使用 Docker 容器中 PostgreSQL 的映射端口，例如 33300。\n   - 用户名和密码：使用在启动 PostgreSQL 容器时设置的用户名和密码。\n\n   完成上述信息的填写后，点击保存并连接到 PostgreSQL 服务器。\n\n现在，你已经成功启动了 pgAdmin 并连接到了你的 PostgreSQL 数据库。你可以使用 pgAdmin 来管理和操作你的数据库。",
+        "createdAt": 1698755749889,
+        "id": "d5XTX9EQ",
+        "meta": {},
+        "parentId": "L7mq0ehV",
+        "role": "assistant",
+        "updatedAt": 1698755786183,
+        "sessionId": "06cc3e20-e870-4099-a619-c07a849d742d",
+        "topicId": "MNWhBEh5",
+        "fromModel": "gpt-3.5-turbo"
+      },
+      {
+        "content": "abcabc",
+        "createdAt": 1690650544852,
+        "id": "KPPDiRyW",
+        "meta": {},
+        "parentId": "42k72jMi",
+        "role": "function",
+        "updatedAt": 1690650572399,
+        "sessionId": "06cc3e20-e870-4099-a619-c07a849d742d",
+
+        "fromModel": "gpt-3.5-turbo-16k",
+        "plugin": {
+          "apiName": "websiteCrawler",
+          "arguments": "{\n  \"url\": \"https://mp.weixin.qq.com/s?__biz=MjM5MTA1MjAxMQ==&mid=2651264337&idx=1&sn=d7d9126578c74c912e1f0d42cb5629da&chksm=bd48ccd58a3f45c3f2cbc7d7b732c820b9e7cd6b547c06acc8170b233710b5fb5ed62f5fd94d&mpshare=1&scene=1&srcid=07294Mpw7C6JCLgtagL1cBDR&sharer_sharetime=1690622712877&sharer_shareid=0358058a42fc25387d28337fc3d22c3c#rd\"\n}",
+          "identifier": "websiteCrawler",
+          "type": "default"
+        }
+      },
+      {
+        "content": "bbbbb",
+        "createdAt": 1700065743405,
+        "id": "9cRjevRQ",
+        "meta": {},
+        "parentId": "3nDXtEKv",
+        "role": "function",
+        "sessionId": "06cc3e20-e870-4099-a619-c07a849d742d",
+        "updatedAt": 1700065751861,
+        "fromModel": "gpt-3.5-turbo-16k",
+        "plugin": {
+          "apiName": "getWebsiteContent",
+          "arguments": "{\n  \"url\": \"https://nodejs.org/api/packages.html#dual-package-hazard\"\n}",
+          "identifier": "website-crawler",
+          "type": "default"
+        }
+      }
+    ]
+  },
+  "version": 2
+}

package/src/migrations/FromV1ToV2/index.ts

@@ -0,0 +1,82 @@
+import { INBOX_SESSION_ID } from '@/const/session';
+import type { Migration, MigrationData } from '@/migrations/VersionController';
+
+import { V1Chat, V1ConfigState, V1Session, V1Topic } from './types/v1';
+import { V2ConfigState, V2Message, V2Session, V2Topic } from './types/v2';
+
+export class MigrationV1ToV2 implements Migration {
+  // from this version to start migration
+  version = 1;
+
+  migrateSession = (
+    session: V1Session,
+    finalMessages: V2Message[],
+    finalTopics: V2Topic[],
+  ): V2Session => {
+    const { chats, topics, createAt, updateAt, pinned, ...i } = session;
+
+    if (chats) {
+      for (const chat of Object.values(chats)) {
+        finalMessages.push(this.migrationMessage(chat, i.id));
+      }
+    }
+
+    if (topics) {
+      for (const chat of Object.values(topics)) {
+        finalTopics.push(this.migrationTopic(i.id, chat));
+      }
+    }
+
+    return { ...i, createdAt: createAt, group: pinned ? 'pinned' : 'default', updatedAt: updateAt };
+  };
+
+  migrate(data: MigrationData<V1ConfigState>): MigrationData<V2ConfigState> {
+    const { sessions: v1Session = {}, inbox } = data.state;
+    const v2Messages: V2Message[] = [];
+    const v2Topics: V2Topic[] = [];
+
+    const sessions: V2Session[] = Object.values(v1Session).map((s) =>
+      this.migrateSession(s, v2Messages, v2Topics),
+    );
+
+    if (inbox) {
+      this.migrateSession({ ...inbox, id: INBOX_SESSION_ID }, v2Messages, v2Topics);
+    }
+
+    return {
+      ...data,
+      state: {
+        ...data.state,
+        messages: v2Messages,
+        sessions,
+        topics: v2Topics,
+      },
+    };
+  }
+
+  migrationMessage = (
+    { createAt, extra, updateAt, name, function_call, ...chat }: V1Chat,
+    sessionId: string,
+  ): V2Message => ({
+    ...chat,
+    createdAt: createAt,
+    plugin: function_call
+      ? {
+          apiName: function_call.name,
+          arguments: function_call.arguments,
+          identifier: name!,
+          type: 'default',
+        }
+      : chat.plugin,
+    sessionId,
+    updatedAt: updateAt,
+    ...extra,
+  });
+
+  migrationTopic = (sessionId: string, { createAt, updateAt, ...topic }: V1Topic): V2Topic => ({
+    ...topic,
+    createdAt: createAt,
+    sessionId,
+    updatedAt: updateAt,
+  });
+}

package/src/migrations/FromV1ToV2/migrations.test.ts

@@ -0,0 +1,224 @@
+import { V1Config, V1ConfigState, V1Session } from '@/migrations/FromV1ToV2/types/v1';
+import { MigrationData, VersionController } from '@/migrations/VersionController';
+
+import inputV1Data from './fixtures/input-v1-session.json';
+import outputV2Data from './fixtures/output-v2.json';
+import { MigrationV1ToV2 } from './index';
+
+describe('MigrationV1ToV2', () => {
+  let migrations;
+  let versionController: VersionController<any>;
+
+  beforeEach(() => {
+    migrations = [MigrationV1ToV2];
+    versionController = new VersionController(migrations, 2);
+  });
+
+  it('should migrate data correctly through multiple versions', () => {
+    const data: MigrationData = inputV1Data;
+
+    const migratedData = versionController.migrate(data);
+
+    expect(migratedData.version).toEqual(outputV2Data.version);
+    expect(migratedData.state.sessions).toEqual(outputV2Data.state.sessions);
+    expect(migratedData.state.topics).toEqual(outputV2Data.state.topics);
+    expect(migratedData.state.messages).toEqual(outputV2Data.state.messages);
+  });
+
+  it('should work correctly with session with no topic', () => {
+    const data: MigrationData<V1ConfigState> = {
+      state: {
+        sessions: {
+          'f8a620ef-c44f-403e-892c-e97fb745255e': {
+            chats: {},
+            config: {
+              model: 'gpt-3.5-turbo',
+              params: {
+                temperature: 0.6,
+              },
+              systemRole:
+                '你是一名前端专家。现在我们正在实现一个 zustand store。该store包含 agents、chats、sessionTree 三个关键的数据。它们的类型定义如下：\n\n```ts\n\n\nexport interface ChatSessionState {\n  sessionTree: SessionTree[];\n  chats: ChatContextMap;\n  agents: ChatAgentMap;\n}\n\ninterface SessionTree {\n  agentId: string;\n  chats: string[];\n}\n\nexport type ChatContextMap = Record<string, ChatContext>;\nexport type ChatAgentMap = Record<string, ChatAgent>;\n\n```',
+              displayMode: 'chat',
+              plugins: [],
+            } as unknown as V1Config,
+            createAt: 1690016491289,
+            id: 'f8a620ef-c44f-403e-892c-e97fb745255e',
+            meta: {
+              title: '前端 zustand store 专家',
+              description: '你需要实现一个 zustand store 的功能',
+              avatar: '输出: 🧪',
+            },
+            type: 'agent',
+            updateAt: 1690016491289,
+          },
+        },
+      },
+      version: 1,
+    };
+
+    const migratedData = versionController.migrate(data);
+    expect(migratedData.version).toEqual(2);
+    expect(migratedData.state.sessions).toEqual([
+      {
+        group: 'default',
+        config: {
+          model: 'gpt-3.5-turbo',
+          params: {
+            temperature: 0.6,
+          },
+          systemRole:
+            '你是一名前端专家。现在我们正在实现一个 zustand store。该store包含 agents、chats、sessionTree 三个关键的数据。它们的类型定义如下：\n\n```ts\n\n\nexport interface ChatSessionState {\n  sessionTree: SessionTree[];\n  chats: ChatContextMap;\n  agents: ChatAgentMap;\n}\n\ninterface SessionTree {\n  agentId: string;\n  chats: string[];\n}\n\nexport type ChatContextMap = Record<string, ChatContext>;\nexport type ChatAgentMap = Record<string, ChatAgent>;\n\n```',
+          displayMode: 'chat',
+          plugins: [],
+        },
+        createdAt: 1690016491289,
+        id: 'f8a620ef-c44f-403e-892c-e97fb745255e',
+        meta: {
+          title: '前端 zustand store 专家',
+          description: '你需要实现一个 zustand store 的功能',
+          avatar: '输出: 🧪',
+        },
+        type: 'agent',
+        updatedAt: 1690016491289,
+      },
+    ]);
+  });
+
+  it('should add inbox messages', () => {
+    const data: MigrationData<V1ConfigState> = {
+      state: {
+        inbox: {
+          type: 'agent',
+          updateAt: 12,
+          createAt: 12,
+          id: 'inbox',
+          meta: {},
+          config: {} as V1Config,
+          chats: {
+            HD4krtKa: {
+              content: '1',
+              createAt: 1692891912689,
+              id: 'HD4krtKa',
+              meta: {},
+              role: 'user',
+              updateAt: 1693114820406,
+              topicId: 'jvfmUEwF',
+            },
+            K4AVcvGB: {
+              content: '2',
+              createAt: 1692891912699,
+              id: 'K4AVcvGB',
+              meta: {},
+              parentId: 'HD4krtKa',
+              role: 'assistant',
+              updateAt: 1693114820418,
+              extra: { fromModel: 'gpt-3.5-turbo' },
+              topicId: 'jvfmUEwF',
+            },
+            QaAUgIGC: {
+              content: '3',
+              createAt: 1693043755942,
+              id: 'QaAUgIGC',
+              meta: {},
+              role: 'user',
+              updateAt: 1693114820428,
+              topicId: 'jvfmUEwF',
+            },
+            jF4p75eF: {
+              content: '4',
+              createAt: 1693043755952,
+              id: 'jF4p75eF',
+              meta: {},
+              parentId: 'QaAUgIGC',
+              role: 'assistant',
+              updateAt: 1693114820439,
+              extra: { fromModel: 'gpt-3.5-turbo' },
+              topicId: 'jvfmUEwF',
+            },
+          },
+          topics: {
+            jvfmUEwF: {
+              createAt: 1693114820394,
+              id: 'jvfmUEwF',
+              title: 'JSX错误：children属性类型错误',
+              updateAt: 1693114821388,
+            },
+            IVfDVB5g: {
+              createAt: 1693228301335,
+              id: 'IVfDVB5g',
+              title: '上游服务端错误状态码\n下游服务器错误状态码',
+              updateAt: 1693228303288,
+            },
+          },
+        },
+      },
+      version: 1,
+    };
+
+    const migratedData = versionController.migrate(data);
+
+    expect(migratedData.version).toEqual(2);
+    expect(migratedData.state.messages).toEqual([
+      {
+        content: '1',
+        createdAt: 1692891912689,
+        id: 'HD4krtKa',
+        sessionId: 'inbox',
+        meta: {},
+        role: 'user',
+        updatedAt: 1693114820406,
+        topicId: 'jvfmUEwF',
+      },
+      {
+        content: '2',
+        createdAt: 1692891912699,
+        id: 'K4AVcvGB',
+        meta: {},
+        sessionId: 'inbox',
+        parentId: 'HD4krtKa',
+        role: 'assistant',
+        updatedAt: 1693114820418,
+        fromModel: 'gpt-3.5-turbo',
+        topicId: 'jvfmUEwF',
+      },
+      {
+        content: '3',
+        createdAt: 1693043755942,
+        id: 'QaAUgIGC',
+        sessionId: 'inbox',
+        meta: {},
+        role: 'user',
+        updatedAt: 1693114820428,
+        topicId: 'jvfmUEwF',
+      },
+      {
+        content: '4',
+        createdAt: 1693043755952,
+        sessionId: 'inbox',
+        id: 'jF4p75eF',
+        meta: {},
+        parentId: 'QaAUgIGC',
+        role: 'assistant',
+        updatedAt: 1693114820439,
+        fromModel: 'gpt-3.5-turbo',
+        topicId: 'jvfmUEwF',
+      },
+    ]);
+    expect(migratedData.state.topics).toEqual([
+      {
+        createdAt: 1693114820394,
+        id: 'jvfmUEwF',
+        title: 'JSX错误：children属性类型错误',
+        sessionId: 'inbox',
+        updatedAt: 1693114821388,
+      },
+      {
+        createdAt: 1693228301335,
+        id: 'IVfDVB5g',
+        title: '上游服务端错误状态码\n下游服务器错误状态码',
+        sessionId: 'inbox',
+        updatedAt: 1693228303288,
+      },
+    ]);
+  });
+});

package/src/migrations/FromV1ToV2/types/v1.ts

@@ -0,0 +1,78 @@
+export interface V1MetaData {
+  avatar?: string;
+  backgroundColor?: string;
+  description?: string;
+  tags?: string[];
+  title?: string;
+}
+
+export interface V1Chat {
+  content: string;
+  createAt: number;
+  extra?: {
+    fromModel: string;
+  };
+  function_call?: {
+    arguments: string;
+    name: string;
+  };
+  id: string;
+  meta: V1MetaData;
+  // function call name
+  name?: string;
+  parentId?: string;
+  plugin?: {
+    apiName: string;
+    arguments: string;
+    identifier: string;
+    type: 'default';
+  };
+  role: string;
+  topicId: string;
+  updateAt: number;
+}
+
+export interface V1Session {
+  chats: Record<string, V1Chat>;
+  config: V1Config;
+  createAt: number;
+  id: string;
+  meta: V1MetaData;
+  pinned?: boolean;
+  topics?: Record<string, V1Topic>;
+  type: 'agent';
+  updateAt: number;
+}
+
+export interface V1Config {
+  displayMode: string;
+  enableCompressThreshold: boolean;
+  enableMaxTokens: boolean;
+  historyCount: number;
+  inputTemplate: string;
+  model: string;
+  params: Partial<Params>;
+  plugins: string[];
+  systemRole: string;
+}
+
+export interface Params {
+  frequency_penalty: number;
+  max_tokens: number;
+  presence_penalty: number;
+  temperature: number;
+  top_p: number;
+}
+
+export interface V1Topic {
+  createAt: number;
+  favorite?: boolean;
+  id: string;
+  title: string;
+  updateAt: number;
+}
+
+export interface V1ConfigState {
+  inbox?: V1Session;
+  sessions?: Record<string, V1Session>;
+}

package/src/migrations/FromV1ToV2/types/v2.ts

@@ -0,0 +1,52 @@
+import { V1Config } from './v1';
+
+export interface V2ConfigState {
+  messages: V2Message[];
+  sessions: V2Session[];
+  topics: V2Topic[];
+}
+
+export interface V2Session {
+  config: V1Config;
+  createdAt: number;
+  group: 'default' | 'pinned';
+  id: string;
+  meta: Meta;
+  type: string;
+  updatedAt: number;
+}
+
+export interface Meta {
+  avatar?: string;
+  backgroundColor?: string;
+  description?: string;
+  tags?: string[];
+  title?: string;
+}
+
+export interface V2Topic {
+  createdAt: number;
+  id: string;
+  sessionId: string;
+  title: string;
+  updatedAt: number;
+}
+
+export interface V2Message {
+  content: string;
+  createdAt: number;
+  fromModel?: string;
+  id: string;
+  meta: Meta;
+  parentId?: string;
+  plugin?: {
+    apiName: string;
+    arguments: string;
+    identifier: string;
+    type: 'default';
+  };
+  role: string;
+  sessionId: string;
+  topicId: string;
+  updatedAt: number;
+}

package/src/{utils → migrations}/VersionController.test.ts

@@ -55,9 +55,7 @@ describe('VersionController', () => {
       state: { value: 10 },
     };
 
-    expect(() => versionController.migrate(data as any)).toThrow(
-      '导入数据缺少版本号，请检查文件后重试',
-    );
+    expect(() => versionController.migrate(data as any)).toThrow();
   });
 
   it('should migrate data correctly through multiple versions', () => {

package/src/{utils → migrations}/VersionController.ts

@@ -1,3 +1,5 @@
+import { t } from 'i18next';
+
 /**
  * 迁移接口
  * @template T - 状态类型
@@ -46,12 +48,13 @@ export class VersionController<T> {
   migrate(data: MigrationData<T>): MigrationData<T> {
     let nextData = data;
     const targetVersion = this.targetVersion || this.migrations.length;
-    if (data.version === undefined) throw new Error('导入数据缺少版本号，请检查文件后重试');
+    if (data.version === undefined)
+      throw new Error(t('migrateError.missVersion', { ns: 'migration' }));
     const currentVersion = data.version;
 
     for (let i = currentVersion || 0; i < targetVersion; i++) {
       const migration = this.migrations.find((m) => m.version === i);
-      if (!migration) throw new Error('程序出错');
+      if (!migration) throw new Error(t('migrateError.noMigration', { ns: 'migration' }));
 
       nextData = migration.migrate(nextData);