@lobehub/chat 1.36.46 → 1.37.1

package/CHANGELOG.md

@@ -2,6 +2,56 @@
 
 # Changelog
 
+### [Version 1.37.1](https://github.com/lobehub/lobe-chat/compare/v1.37.0...v1.37.1)
+
+<sup>Released on **2024-12-22**</sup>
+
+#### ♻ Code Refactoring
+
+- **misc**: Refactor the client service to deprecated.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### Code refactoring
+
+- **misc**: Refactor the client service to deprecated, closes [#5132](https://github.com/lobehub/lobe-chat/issues/5132) ([e603234](https://github.com/lobehub/lobe-chat/commit/e603234))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
+## [Version 1.37.0](https://github.com/lobehub/lobe-chat/compare/v1.36.46...v1.37.0)
+
+<sup>Released on **2024-12-22**</sup>
+
+#### ✨ Features
+
+- **misc**: Support to use pglite as client db.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's improved
+
+- **misc**: Support to use pglite as client db, closes [#4873](https://github.com/lobehub/lobe-chat/issues/4873) ([4131f20](https://github.com/lobehub/lobe-chat/commit/4131f20))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 1.36.46](https://github.com/lobehub/lobe-chat/compare/v1.36.45...v1.36.46)
 
 <sup>Released on **2024-12-21**</sup>

package/README.ja-JP.md

@@ -263,14 +263,14 @@ LobeChat のプラグインエコシステムは、そのコア機能の重要
 
 <!-- PLUGIN LIST -->
 
-| 最近追加                                                                                                                             | 説明                                                                                                                                                 |
-| ------------------------------------------------------------------------------------------------------------------------------------ | ---------------------------------------------------------------------------------------------------------------------------------------------------- |
-| [Google CSE](https://lobechat.com/discover/plugin/google-cse)<br/><sup>By **vsnthdev** on **2024-12-02**</sup>                       | 公式の CSE API を通じて Google を検索します。<br/>`ウェブ` `検索`                                                                                    |
-| [話す](https://lobechat.com/discover/plugin/speak)<br/><sup>By **speak** on **2024-12-02**</sup>                                     | Speak は、AI パワードの言語チューターで、他の言語で何でも言う方法を学ぶことができます。<br/>`教育` `言語`                                            |
-| [Tongyi Wanxiang 画像生成器](https://lobechat.com/discover/plugin/alps-tongyi-image)<br/><sup>By **YoungTx** on **2024-08-09**</sup> | このプラグインは、Alibaba の Tongyi Wanxiang モデルを使用して、テキストプロンプトに基づいて画像を生成します。<br/>`画像` `トンギ` `ワンシャン`       |
-| [ショッピングツール](https://lobechat.com/discover/plugin/ShoppingTools)<br/><sup>By **shoppingtools** on **2024-07-19**</sup>       | eBay や AliExpress で商品を検索し、eBay のイベントやクーポンを見つけます。迅速な例を取得します。<br/>`ショッピング` `e-bay` `ali-express` `クーポン` |
-
-> 📊 Total plugins: [<kbd>**47**</kbd>](https://lobechat.com/discover/plugins)
+| 最近追加                                                                                                                             | 説明                                                                                                                                           |
+| ------------------------------------------------------------------------------------------------------------------------------------ | ---------------------------------------------------------------------------------------------------------------------------------------------- |
+| [PortfolioMeta](https://lobechat.com/discover/plugin/StockData)<br/><sup>By **portfoliometa** on **2024-12-22**</sup>                | 株を分析し、包括的なリアルタイムの投資データと分析を取得します。<br/>`stock`                                                                   |
+| [Google CSE](https://lobechat.com/discover/plugin/google-cse)<br/><sup>By **vsnthdev** on **2024-12-02**</sup>                       | 公式の CSE API を通じて Google を検索します。<br/>`ウェブ` `検索`                                                                              |
+| [話す](https://lobechat.com/discover/plugin/speak)<br/><sup>By **speak** on **2024-12-02**</sup>                                     | Speak は、AI パワードの言語チューターで、他の言語で何でも言う方法を学ぶことができます。<br/>`教育` `言語`                                      |
+| [Tongyi Wanxiang 画像生成器](https://lobechat.com/discover/plugin/alps-tongyi-image)<br/><sup>By **YoungTx** on **2024-08-09**</sup> | このプラグインは、Alibaba の Tongyi Wanxiang モデルを使用して、テキストプロンプトに基づいて画像を生成します。<br/>`画像` `トンギ` `ワンシャン` |
+
+> 📊 Total plugins: [<kbd>**48**</kbd>](https://lobechat.com/discover/plugins)
 
  <!-- PLUGIN LIST -->
 

package/README.md

@@ -280,14 +280,14 @@ In addition, these plugins are not limited to news aggregation, but can also ext
 
 <!-- PLUGIN LIST -->
 
-| Recent Submits                                                                                                                            | Description                                                                                                                               |
-| ----------------------------------------------------------------------------------------------------------------------------------------- | ----------------------------------------------------------------------------------------------------------------------------------------- |
-| [Google CSE](https://lobechat.com/discover/plugin/google-cse)<br/><sup>By **vsnthdev** on **2024-12-02**</sup>                            | Searches Google through their official CSE API.<br/>`web` `search`                                                                        |
-| [Speak](https://lobechat.com/discover/plugin/speak)<br/><sup>By **speak** on **2024-12-02**</sup>                                         | Learn how to say anything in another language with Speak, your AI-powered language tutor.<br/>`education` `language`                      |
-| [Tongyi wanxiang Image Generator](https://lobechat.com/discover/plugin/alps-tongyi-image)<br/><sup>By **YoungTx** on **2024-08-09**</sup> | This plugin uses Alibaba's Tongyi Wanxiang model to generate images based on text prompts.<br/>`image` `tongyi` `wanxiang`                |
-| [Shopping tools](https://lobechat.com/discover/plugin/ShoppingTools)<br/><sup>By **shoppingtools** on **2024-07-19**</sup>                | Search for products on eBay & AliExpress, find eBay events & coupons. Get prompt examples.<br/>`shopping` `e-bay` `ali-express` `coupons` |
-
-> 📊 Total plugins: [<kbd>**47**</kbd>](https://lobechat.com/discover/plugins)
+| Recent Submits                                                                                                                            | Description                                                                                                                |
+| ----------------------------------------------------------------------------------------------------------------------------------------- | -------------------------------------------------------------------------------------------------------------------------- |
+| [PortfolioMeta](https://lobechat.com/discover/plugin/StockData)<br/><sup>By **portfoliometa** on **2024-12-22**</sup>                     | Analyze stocks and get comprehensive real-time investment data and analytics.<br/>`stock`                                  |
+| [Google CSE](https://lobechat.com/discover/plugin/google-cse)<br/><sup>By **vsnthdev** on **2024-12-02**</sup>                            | Searches Google through their official CSE API.<br/>`web` `search`                                                         |
+| [Speak](https://lobechat.com/discover/plugin/speak)<br/><sup>By **speak** on **2024-12-02**</sup>                                         | Learn how to say anything in another language with Speak, your AI-powered language tutor.<br/>`education` `language`       |
+| [Tongyi wanxiang Image Generator](https://lobechat.com/discover/plugin/alps-tongyi-image)<br/><sup>By **YoungTx** on **2024-08-09**</sup> | This plugin uses Alibaba's Tongyi Wanxiang model to generate images based on text prompts.<br/>`image` `tongyi` `wanxiang` |
+
+> 📊 Total plugins: [<kbd>**48**</kbd>](https://lobechat.com/discover/plugins)
 
  <!-- PLUGIN LIST -->
 

package/README.zh-CN.md

@@ -273,14 +273,14 @@ LobeChat 的插件生态系统是其核心功能的重要扩展，它极大地
 
 <!-- PLUGIN LIST -->
 
-| 最近新增                                                                                                                     | 描述                                                                                                               |
-| ---------------------------------------------------------------------------------------------------------------------------- | ------------------------------------------------------------------------------------------------------------------ |
-| [谷歌自定义搜索引擎](https://lobechat.com/discover/plugin/google-cse)<br/><sup>By **vsnthdev** on **2024-12-02**</sup>       | 通过他们的官方自定义搜索引擎 API 搜索谷歌。<br/>`网络` `搜索`                                                      |
-| [Speak](https://lobechat.com/discover/plugin/speak)<br/><sup>By **speak** on **2024-12-02**</sup>                            | 使用 Speak，您的 AI 语言导师，学习如何用另一种语言说任何事情。<br/>`教育` `语言`                                   |
-| [通义万象图像生成器](https://lobechat.com/discover/plugin/alps-tongyi-image)<br/><sup>By **YoungTx** on **2024-08-09**</sup> | 此插件使用阿里巴巴的通义万象模型根据文本提示生成图像。<br/>`图像` `通义` `万象`                                    |
-| [购物工具](https://lobechat.com/discover/plugin/ShoppingTools)<br/><sup>By **shoppingtools** on **2024-07-19**</sup>         | 在 eBay 和 AliExpress 上搜索产品，查找 eBay 活动和优惠券。获取快速示例。<br/>`购物` `e-bay` `ali-express` `优惠券` |
-
-> 📊 Total plugins: [<kbd>**47**</kbd>](https://lobechat.com/discover/plugins)
+| 最近新增                                                                                                                     | 描述                                                                             |
+| ---------------------------------------------------------------------------------------------------------------------------- | -------------------------------------------------------------------------------- |
+| [PortfolioMeta](https://lobechat.com/discover/plugin/StockData)<br/><sup>By **portfoliometa** on **2024-12-22**</sup>        | 分析股票并获取全面的实时投资数据和分析。<br/>`股票`                              |
+| [谷歌自定义搜索引擎](https://lobechat.com/discover/plugin/google-cse)<br/><sup>By **vsnthdev** on **2024-12-02**</sup>       | 通过他们的官方自定义搜索引擎 API 搜索谷歌。<br/>`网络` `搜索`                    |
+| [Speak](https://lobechat.com/discover/plugin/speak)<br/><sup>By **speak** on **2024-12-02**</sup>                            | 使用 Speak，您的 AI 语言导师，学习如何用另一种语言说任何事情。<br/>`教育` `语言` |
+| [通义万象图像生成器](https://lobechat.com/discover/plugin/alps-tongyi-image)<br/><sup>By **YoungTx** on **2024-08-09**</sup> | 此插件使用阿里巴巴的通义万象模型根据文本提示生成图像。<br/>`图像` `通义` `万象`  |
+
+> 📊 Total plugins: [<kbd>**48**</kbd>](https://lobechat.com/discover/plugins)
 
  <!-- PLUGIN LIST -->
 

package/changelog/v1.json

@@ -1,4 +1,22 @@
 [
+  {
+    "children": {
+      "improvements": [
+        "Refactor the client service to deprecated."
+      ]
+    },
+    "date": "2024-12-22",
+    "version": "1.37.1"
+  },
+  {
+    "children": {
+      "features": [
+        "Support to use pglite as client db."
+      ]
+    },
+    "date": "2024-12-22",
+    "version": "1.37.0"
+  },
   {
     "children": {
       "improvements": [

package/next.config.mjs

@@ -6,6 +6,7 @@ import ReactComponentName from 'react-scan/react-component-name/webpack';
 const isProd = process.env.NODE_ENV === 'production';
 const buildWithDocker = process.env.DOCKER === 'true';
 const enableReactScan = !!process.env.REACT_SCAN_MONITOR_API_KEY;
+const isUsePglite = process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite';
 
 // if you need to proxy the api endpoint to remote server
 const API_PROXY_ENDPOINT = process.env.API_PROXY_ENDPOINT || '';
@@ -26,6 +27,7 @@ const nextConfig = {
       'gpt-tokenizer',
       'chroma-js',
     ],
+    serverComponentsExternalPackages: ['@electric-sql/pglite'],
     webVitalsAttribution: ['CLS', 'LCP'],
   },
 
@@ -180,7 +182,8 @@ const nextConfig = {
       layers: true,
     };
 
-    if (enableReactScan) {
+    // 开启该插件会导致 pglite 的 fs bundler 被改表
+    if (enableReactScan && !isUsePglite) {
       config.plugins.push(ReactComponentName({}));
     }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.36.46",
+  "version": "1.37.1",
   "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",
@@ -32,7 +32,8 @@
     "build-sitemap": "tsx ./scripts/buildSitemapIndex/index.ts",
     "build:analyze": "ANALYZE=true next build",
     "build:docker": "DOCKER=true next build && npm run build-sitemap",
-    "db:generate": "drizzle-kit generate",
+    "db:generate": "drizzle-kit generate && npm run db:generate-client",
+    "db:generate-client": "tsx ./scripts/migrateClientDB/compile-migrations.ts",
     "db:migrate": "MIGRATION_DB=1 tsx ./scripts/migrateServerDB/index.ts",
     "db:push": "drizzle-kit push",
     "db:push-test": "NODE_ENV=test drizzle-kit push",
@@ -117,6 +118,7 @@
     "@clerk/themes": "^2.1.37",
     "@codesandbox/sandpack-react": "^2.19.9",
     "@cyntler/react-doc-viewer": "^1.17.0",
+    "@electric-sql/pglite": "0.2.13",
     "@google/generative-ai": "^0.21.0",
     "@huggingface/inference": "^2.8.1",
     "@icons-pack/react-simple-icons": "9.6.0",
@@ -307,7 +309,7 @@
     "vitest": "~1.2.2",
     "vitest-canvas-mock": "^0.3.3"
   },
-  "packageManager": "pnpm@9.15.0",
+  "packageManager": "pnpm@9.15.1",
   "publishConfig": {
     "access": "public",
     "registry": "https://registry.npmjs.org"

package/scripts/migrateClientDB/compile-migrations.ts

@@ -0,0 +1,14 @@
+import { readMigrationFiles } from 'drizzle-orm/migrator';
+import { writeFileSync } from 'node:fs';
+import { join } from 'node:path';
+
+const dbBase = join(__dirname, '../../src/database');
+const migrationsFolder = join(dbBase, './migrations');
+const migrations = readMigrationFiles({ migrationsFolder: migrationsFolder });
+
+writeFileSync(
+  join(dbBase, './client/migrations.json'),
+  JSON.stringify(migrations, null, 2), // null, 2 adds indentation for better readability
+);
+
+console.log('🏁 client migrations.json compiled!');

package/src/app/(main)/(mobile)/me/(home)/layout.tsx

@@ -1,6 +1,7 @@
 import { PropsWithChildren } from 'react';
 
 import MobileContentLayout from '@/components/server/MobileNavLayout';
+import InitClientDB from '@/features/InitClientDB';
 
 import Header from './features/Header';
 
@@ -8,6 +9,7 @@ const Layout = ({ children }: PropsWithChildren) => {
   return (
     <MobileContentLayout header={<Header />} withNav>
       {children}
+      <InitClientDB />
     </MobileContentLayout>
   );
 };

package/src/app/(main)/chat/_layout/Desktop/index.tsx

@@ -1,6 +1,7 @@
 import { Flexbox } from 'react-layout-kit';
 
-import Migration from '../../features/Migration';
+import InitClientDB from '@/features/InitClientDB';
+
 import { LayoutProps } from '../type';
 import SessionPanel from './SessionPanel';
 
@@ -18,7 +19,7 @@ const Layout = ({ children, session }: LayoutProps) => {
           {children}
         </Flexbox>
       </Flexbox>
-      <Migration />
+      <InitClientDB bottom={60} />
       {/* ↓ cloud slot ↓ */}
 
       {/* ↑ cloud slot ↑ */}

package/src/app/(main)/chat/_layout/Mobile.tsx

@@ -1,10 +1,10 @@
 'use client';
 
 import { createStyles } from 'antd-style';
-import { memo } from 'react';
+import { Suspense, memo } from 'react';
 import { Flexbox } from 'react-layout-kit';
 
-import Migration from '@/app/(main)/chat/features/Migration';
+import InitClientDB from '@/features/InitClientDB';
 import { useQuery } from '@/hooks/useQuery';
 
 import { LayoutProps } from './type';
@@ -39,7 +39,9 @@ const Layout = memo<LayoutProps>(({ children, session }) => {
       >
         {children}
       </Flexbox>
-      <Migration />
+      <Suspense fallback={null}>
+        <InitClientDB bottom={100} />
+      </Suspense>
     </>
   );
 });

package/src/app/(main)/chat/features/Migration/DBReader.ts

@@ -0,0 +1,290 @@
+import {
+  ImportMessage,
+  ImportSession,
+  ImportSessionGroup,
+  ImportTopic,
+  ImporterEntryData,
+} from '@/types/importer';
+
+interface V2DB_File {
+  /**
+   * create Time
+   */
+  createdAt: number;
+  /**
+   * file data array buffer
+   */
+  data: ArrayBuffer;
+  /**
+   * file type
+   * @example 'image/png'
+   */
+  fileType: string;
+  id: string;
+  metadata: any;
+  /**
+   * file name
+   * @example 'test.png'
+   */
+  name: string;
+  /**
+   * the mode database save the file
+   * local mean save the raw file into data
+   * url mean upload the file to a cdn and then save the url
+   */
+  saveMode: 'local' | 'url';
+  /**
+   * file size
+   */
+  size: number;
+  /**
+   * file url if saveMode is url
+   */
+  url: string;
+}
+
+interface V2DB_MESSAGE {
+  content: string;
+  createdAt: number;
+  error?: any;
+  favorite: 0 | 1;
+  files?: string[];
+  fromModel?: string;
+  fromProvider?: string;
+  id: string;
+  observationId?: string;
+  // foreign keys
+  parentId?: string;
+  plugin?: any;
+  pluginError?: any;
+  pluginState?: any;
+
+  quotaId?: string;
+  role: string;
+  sessionId?: string;
+  tool_call_id?: string;
+  tools?: object[];
+  topicId?: string;
+
+  traceId?: string;
+  translate?: object | false;
+  tts?: any;
+  updatedAt: number;
+}
+
+interface DB_Plugin {
+  createdAt: number;
+  id: string;
+  identifier: string;
+  manifest?: object;
+  settings?: object;
+  type: 'plugin' | 'customPlugin';
+  updatedAt: number;
+}
+
+interface DB_Session {
+  config: object;
+  createdAt: number;
+  group?: string;
+  // 原 Agent 类型
+  id: string;
+  meta: object;
+  pinned?: number;
+  type?: 'agent' | 'group';
+  updatedAt: number;
+}
+
+interface DB_SessionGroup {
+  createdAt: number;
+  id: string;
+  name: string;
+  sort?: number;
+  updatedAt: number;
+}
+
+interface DB_Topic {
+  createdAt: number;
+  favorite?: number;
+  id: string;
+  sessionId?: string;
+  title: string;
+  updatedAt: number;
+}
+
+interface DB_User {
+  avatar?: string;
+  createdAt: number;
+  id: string;
+  settings: object;
+  updatedAt: number;
+  uuid: string;
+}
+
+interface MigrationData {
+  files: V2DB_File[];
+  messages: V2DB_MESSAGE[];
+  plugins: DB_Plugin[];
+  sessionGroups: DB_SessionGroup[];
+  sessions: DB_Session[];
+  topics: DB_Topic[];
+  users: DB_User[];
+}
+
+const LOBE_CHAT_LOCAL_DB_NAME = 'LOBE_CHAT_DB';
+
+const V2DB_LASET_SCHEMA_VERSION = 7;
+export class V2DBReader {
+  private dbName: string = LOBE_CHAT_LOCAL_DB_NAME;
+  private storeNames: string[];
+
+  constructor(storeNames: string[]) {
+    this.storeNames = storeNames;
+  }
+
+  /**
+   * 读取所有数据
+   */
+  async readAllData(): Promise<MigrationData> {
+    try {
+      // 打开数据库连接
+      const db = await this.openDB();
+
+      // 并行读取所有表的数据
+      const results = await Promise.all(
+        this.storeNames.map((storeName) => this.readStore(db, storeName)),
+      );
+
+      // 构建返回结果
+      const migrationData = this.storeNames.reduce((acc, storeName, index) => {
+        // @ts-expect-error
+        acc[storeName] = results[index];
+        return acc;
+      }, {} as MigrationData);
+
+      // 关闭数据库连接
+      db.close();
+
+      return migrationData;
+    } catch (error) {
+      console.error('读取数据库失败:', error);
+      throw error;
+    }
+  }
+
+  async convertToImportData(data: MigrationData): Promise<ImporterEntryData> {
+    // 转换 messages
+    const messages = data.messages.map(
+      (msg): ImportMessage => ({
+        // 使用原有的 id
+        content: msg.content,
+        createdAt: msg.createdAt,
+        // 处理 error
+        error: msg.error || msg.pluginError,
+
+        // 处理额外信息
+        extra: {
+          fromModel: msg.fromModel,
+          fromProvider: msg.fromProvider,
+          translate: msg.translate as any,
+          tts: msg.tts,
+        },
+
+        files: msg.files,
+        id: msg.id,
+
+        // 复制原有字段
+        observationId: msg.observationId,
+        parentId: msg.parentId,
+        plugin: msg.plugin,
+        pluginState: msg.pluginState,
+        quotaId: msg.quotaId,
+        role: msg.role as any,
+        sessionId: msg.sessionId,
+        tool_call_id: msg.tool_call_id,
+        tools: msg.tools as any,
+
+        topicId: msg.topicId,
+
+        traceId: msg.traceId,
+
+        updatedAt: msg.updatedAt,
+      }),
+    );
+
+    // 转换 sessionGroups
+    const sessionGroups = data.sessionGroups.map(
+      (group): ImportSessionGroup => ({
+        createdAt: group.createdAt,
+        id: group.id,
+        // 使用原有的 id
+        name: group.name,
+        sort: group.sort || null,
+        updatedAt: group.updatedAt,
+      }),
+    );
+
+    // 转换 sessions
+    const sessions = data.sessions.map(
+      (session): ImportSession => ({
+        // 使用原有的 id
+        config: session.config as any,
+        createdAt: new Date(session.createdAt).toString(),
+        group: session.group,
+        id: session.id,
+        meta: session.meta as any,
+        pinned: session.pinned ? true : undefined,
+        type: session.type || 'agent',
+        updatedAt: new Date(session.updatedAt).toString(),
+      }),
+    );
+
+    const topics = data.topics.map(
+      (topic): ImportTopic => ({
+        ...topic,
+        favorite: topic.favorite ? true : undefined,
+      }),
+    );
+
+    return {
+      messages,
+      sessionGroups,
+      sessions,
+      topics,
+      version: V2DB_LASET_SCHEMA_VERSION,
+    };
+  }
+
+  /**
+   * 打开数据库
+   */
+  private openDB(): Promise<IDBDatabase> {
+    return new Promise((resolve, reject) => {
+      const request = indexedDB.open(this.dbName);
+
+      // eslint-disable-next-line unicorn/prefer-add-event-listener
+      request.onerror = () => {
+        reject(request.error);
+      };
+      request.onsuccess = () => resolve(request.result);
+    });
+  }
+
+  /**
+   * 读取单个存储对象的所有数据
+   */
+  private readStore(db: IDBDatabase, storeName: string): Promise<any[]> {
+    return new Promise((resolve, reject) => {
+      try {
+        const transaction = db.transaction(storeName, 'readonly');
+        const store = transaction.objectStore(storeName);
+        const request = store.getAll();
+
+        // eslint-disable-next-line unicorn/prefer-add-event-listener
+        request.onerror = () => reject(request.error);
+        request.onsuccess = () => resolve(request.result);
+      } catch (error) {
+        reject(error);
+      }
+    });
+  }
+}

package/src/app/(main)/chat/features/Migration/UpgradeButton.tsx

@@ -1,14 +1,12 @@
 import { Button } from 'antd';
-import { createStore, set } from 'idb-keyval';
 import { ReactNode, memo } from 'react';
 import { useTranslation } from 'react-i18next';
 
-import { Migration } from '@/migrations';
 import { configService } from '@/services/config';
 import { useChatStore } from '@/store/chat';
 import { useSessionStore } from '@/store/session';
 
-import { MIGRATE_KEY, MigrationError, UpgradeStatus, V1DB_NAME, V1DB_TABLE_NAME } from './const';
+import { MigrationError, UpgradeStatus } from './const';
 
 export interface UpgradeButtonProps {
   children?: ReactNode;
@@ -31,21 +29,19 @@ const UpgradeButton = memo<UpgradeButtonProps>(
 
     const upgrade = async () => {
       try {
-        const data = Migration.migrate({ state, version: 1 });
-
         setUpgradeStatus(UpgradeStatus.UPGRADING);
 
         await configService.importConfigState({
           exportType: 'sessions',
-          state: data.state,
-          version: 2,
+          state: state,
+          version: 7,
         });
 
         await refreshSession();
         await refreshMessages();
         await refreshTopic();
 
-        await set(MIGRATE_KEY, true, createStore(V1DB_NAME, V1DB_TABLE_NAME));
+        localStorage.setItem('V2DB_IS_MIGRATED', '1');
 
         setUpgradeStatus(UpgradeStatus.UPGRADED);