新增信息id

fix: #4735

.eslintignore

@@ -1,5 +0,0 @@
-node_modules
-dist
-out
-.gitignore
-scripts/cloudflare-worker.js

.eslintrc.cjs

@@ -1,22 +0,0 @@
-module.exports = {
-  plugins: ['unused-imports', 'simple-import-sort'],
-  extends: [
-    'eslint:recommended',
-    'plugin:react/recommended',
-    'plugin:react/jsx-runtime',
-    'plugin:react-hooks/recommended',
-    '@electron-toolkit/eslint-config-ts/recommended',
-    '@electron-toolkit/eslint-config-prettier'
-  ],
-  rules: {
-    '@typescript-eslint/explicit-function-return-type': 'off',
-    '@typescript-eslint/no-explicit-any': 'off',
-    'unused-imports/no-unused-imports': 'error',
-    '@typescript-eslint/no-non-null-asserted-optional-chain': 'off',
-    'react/prop-types': 'off',
-    'simple-import-sort/imports': 'error',
-    'simple-import-sort/exports': 'error',
-    'react/no-is-mounted': 'off',
-    'prettier/prettier': ['error', { endOfLine: 'auto' }]
-  }
-}

.gitattributes

@@ -0,0 +1,2 @@
+/.yarn/**            linguist-vendored
+/.yarn/releases/*    binary

.github/ISSUE_TEMPLATE/0_bug_report.yml

@@ -1,4 +1,4 @@
-name: 🐛 Bug Report
+name: 🐛 Bug Report (English)
 description: Create a report to help us improve
 title: '[Bug]: '
 labels: ['bug']

.github/ISSUE_TEMPLATE/1_feature_request.yml

@@ -1,4 +1,4 @@
-name: 💡 Feature Request
+name: 💡 Feature Request (English)
 description: Suggest an idea for this project
 title: '[Feature]: '
 labels: ['enhancement']

.github/workflows/nightly-build.yml

@@ -0,0 +1,264 @@
+name: Nightly Build
+
+on:
+  workflow_dispatch:
+  schedule:
+    - cron: '0 17 * * *' # 1:00 BJ Time
+
+permissions:
+  contents: write
+
+jobs:
+  nightly-build:
+    runs-on: ${{ matrix.os }}
+
+    strategy:
+      matrix:
+        os: [macos-latest, windows-latest, ubuntu-latest]
+      fail-fast: false
+
+    steps:
+      - name: Check out Git repository
+        uses: actions/checkout@v4
+
+      - name: Install Node.js
+        uses: actions/setup-node@v4
+        with:
+          node-version: 20
+
+      - name: Install corepack
+        run: corepack enable && corepack prepare yarn@4.6.0 --activate
+
+      - name: Get yarn cache directory path
+        id: yarn-cache-dir-path
+        run: echo "dir=$(yarn config get cacheFolder)" >> $GITHUB_OUTPUT
+
+      - name: Cache yarn dependencies
+        uses: actions/cache@v4
+        with:
+          path: |
+            ${{ steps.yarn-cache-dir-path.outputs.dir }}
+            node_modules
+          key: ${{ runner.os }}-yarn-${{ hashFiles('**/yarn.lock') }}
+          restore-keys: |
+            ${{ runner.os }}-yarn-
+
+      - name: Install Dependencies
+        run: yarn install
+
+      - name: Generate date tag
+        id: date
+        run: echo "date=$(date +'%Y%m%d')" >> $GITHUB_OUTPUT
+        shell: bash
+
+      - name: Build Linux
+        if: matrix.os == 'ubuntu-latest'
+        run: |
+          yarn build:npm linux
+          yarn build:linux
+        env:
+          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          RENDERER_VITE_AIHUBMIX_SECRET: ${{ vars.RENDERER_VITE_AIHUBMIX_SECRET }}
+
+      - name: Build Mac
+        if: matrix.os == 'macos-latest'
+        run: |
+          yarn build:npm mac
+          yarn build:mac
+        env:
+          CSC_LINK: ${{ secrets.CSC_LINK }}
+          CSC_KEY_PASSWORD: ${{ secrets.CSC_KEY_PASSWORD }}
+          APPLE_ID: ${{ vars.APPLE_ID }}
+          APPLE_APP_SPECIFIC_PASSWORD: ${{ vars.APPLE_APP_SPECIFIC_PASSWORD }}
+          APPLE_TEAM_ID: ${{ vars.APPLE_TEAM_ID }}
+          RENDERER_VITE_AIHUBMIX_SECRET: ${{ vars.RENDERER_VITE_AIHUBMIX_SECRET }}
+          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Build Windows
+        if: matrix.os == 'windows-latest'
+        run: |
+          yarn build:npm windows
+          yarn build:win:x64
+          yarn build:win:arm64
+        env:
+          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          RENDERER_VITE_AIHUBMIX_SECRET: ${{ vars.RENDERER_VITE_AIHUBMIX_SECRET }}
+
+      - name: Replace spaces in filenames
+        run: node scripts/replace-spaces.js
+
+      - name: Rename artifacts with nightly format
+        shell: bash
+        run: |
+          mkdir -p renamed-artifacts
+          DATE=${{ steps.date.outputs.date }}
+
+          # Windows artifacts - based on actual file naming pattern
+          if [ "${{ matrix.os }}" == "windows-latest" ]; then
+            # Setup installer
+            find dist -name "*setup.exe" -exec cp {} renamed-artifacts/cherry-studio-nightly-${DATE}-setup.exe \;
+            
+            # Portable exe
+            find dist -name "*portable.exe" -exec cp {} renamed-artifacts/cherry-studio-nightly-${DATE}-portable.exe \;
+
+            # Rename blockmap files to match the new exe names
+            if [ -f "dist/*setup.exe.blockmap" ]; then
+              cp dist/*setup.exe.blockmap renamed-artifacts/cherry-studio-nightly-${DATE}-setup.exe.blockmap || true
+            fi
+          fi
+
+          # macOS artifacts
+          if [ "${{ matrix.os }}" == "macos-latest" ]; then
+            # 处理arm64架构文件
+            find dist -name "*-arm64.dmg" -exec cp {} renamed-artifacts/cherry-studio-nightly-${DATE}-arm64.dmg \;
+            find dist -name "*-arm64.dmg.blockmap" -exec cp {} renamed-artifacts/cherry-studio-nightly-${DATE}-arm64.dmg.blockmap \;
+            find dist -name "*-arm64.zip" -exec cp {} renamed-artifacts/cherry-studio-nightly-${DATE}-arm64.zip \;
+            find dist -name "*-arm64.zip.blockmap" -exec cp {} renamed-artifacts/cherry-studio-nightly-${DATE}-arm64.zip.blockmap \;
+            
+            # 处理x64架构文件
+            find dist -name "*-x64.dmg" -exec cp {} renamed-artifacts/cherry-studio-nightly-${DATE}-x64.dmg \;
+            find dist -name "*-x64.dmg.blockmap" -exec cp {} renamed-artifacts/cherry-studio-nightly-${DATE}-x64.dmg.blockmap \;
+            find dist -name "*-x64.zip" -exec cp {} renamed-artifacts/cherry-studio-nightly-${DATE}-x64.zip \;
+            find dist -name "*-x64.zip.blockmap" -exec cp {} renamed-artifacts/cherry-studio-nightly-${DATE}-x64.zip.blockmap \;
+          fi
+
+          # Linux artifacts
+          if [ "${{ matrix.os }}" == "ubuntu-latest" ]; then
+            find dist -name "*.AppImage" -exec cp {} renamed-artifacts/cherry-studio-nightly-${DATE}.AppImage \;
+            find dist -name "*.snap" -exec cp {} renamed-artifacts/cherry-studio-nightly-${DATE}.snap \;
+            find dist -name "*.deb" -exec cp {} renamed-artifacts/cherry-studio-nightly-${DATE}.deb \;
+            find dist -name "*.rpm" -exec cp {} renamed-artifacts/cherry-studio-nightly-${DATE}.rpm \;
+            find dist -name "*.tar.gz" -exec cp {} renamed-artifacts/cherry-studio-nightly-${DATE}.tar.gz \;
+          fi
+
+          # Copy update files
+          cp dist/latest*.yml renamed-artifacts/ || true
+
+      # Generate SHA256 checksums (Windows)
+      - name: Generate SHA256 checksums (Windows)
+        if: runner.os == 'Windows'
+        shell: pwsh
+        run: |
+          cd renamed-artifacts
+          echo "# SHA256 checksums for Windows - $(Get-Date -Format 'yyyy-MM-dd')" > SHA256SUMS.txt
+          Get-ChildItem -File | Where-Object { $_.Name -ne 'SHA256SUMS.txt' } | ForEach-Object {
+            $file = $_.Name
+            $hash = (Get-FileHash -Algorithm SHA256 $file).Hash.ToLower()
+            Add-Content -Path SHA256SUMS.txt -Value "$hash  $file"
+          }
+          cat SHA256SUMS.txt
+
+      # Generate SHA256 checksums (macOS/Linux)
+      - name: Generate SHA256 checksums (macOS/Linux)
+        if: runner.os != 'Windows'
+        shell: bash
+        run: |
+          cd renamed-artifacts
+          echo "# SHA256 checksums for ${{ runner.os }} - $(date +'%Y-%m-%d')" > SHA256SUMS.txt
+          if command -v shasum &>/dev/null; then
+            # macOS
+            shasum -a 256 * 2>/dev/null | grep -v SHA256SUMS.txt >> SHA256SUMS.txt || echo "No files to hash" >> SHA256SUMS.txt
+          else
+            # Linux
+            sha256sum * 2>/dev/null | grep -v SHA256SUMS.txt >> SHA256SUMS.txt || echo "No files to hash" >> SHA256SUMS.txt
+          fi
+          cat SHA256SUMS.txt
+
+      - name: List files to be uploaded
+        shell: bash
+        run: |
+          echo "准备上传的文件:"
+          if [ -x "$(command -v tree)" ]; then
+            tree renamed-artifacts
+          elif [ "$RUNNER_OS" == "Windows" ]; then
+            dir renamed-artifacts
+          else
+            ls -la renamed-artifacts
+          fi
+          echo "总计: $(find renamed-artifacts -type f | wc -l) 个文件"
+
+      - name: Upload artifacts
+        uses: actions/upload-artifact@v4
+        with:
+          name: cherry-studio-nightly-${{ steps.date.outputs.date }}-${{ matrix.os }}
+          path: renamed-artifacts/*
+          retention-days: 3 # 保留3天
+          compression-level: 8
+
+  Build-Summary:
+    needs: nightly-build
+    if: always()
+    runs-on: ubuntu-latest
+    steps:
+      - name: Get date tag
+        id: date
+        run: echo "date=$(date +'%Y%m%d')" >> $GITHUB_OUTPUT
+        shell: bash
+
+      - name: Download all artifacts
+        uses: actions/download-artifact@v4
+        with:
+          path: all-artifacts
+          merge-multiple: false
+        continue-on-error: true
+
+      - name: Create summary report
+        run: |
+          echo "## ⚠️ 警告：这是每日构建版本" >> $GITHUB_STEP_SUMMARY
+          echo "此版本为自动构建的不稳定版本，仅供测试使用。不建议在生产环境中使用。" >> $GITHUB_STEP_SUMMARY
+          echo "安装此版本前请务必备份数据，并做好数据迁移准备。" >> $GITHUB_STEP_SUMMARY
+          echo "构建日期：$(date +'%Y-%m-%d')" >> $GITHUB_STEP_SUMMARY
+          echo "" >> $GITHUB_STEP_SUMMARY
+
+          echo "## 📦 安装包校验和" >> $GITHUB_STEP_SUMMARY
+          echo "请在下载后验证文件完整性。提供 SHA256 校验和。" >> $GITHUB_STEP_SUMMARY
+          echo "" >> $GITHUB_STEP_SUMMARY
+
+          # Check each platform's artifacts and show checksums if available
+
+          # Windows
+          WIN_ARTIFACT_DIR="all-artifacts/cherry-studio-nightly-${{ steps.date.outputs.date }}-windows-latest"
+          if [ -d "$WIN_ARTIFACT_DIR" ] && [ -f "$WIN_ARTIFACT_DIR/SHA256SUMS.txt" ]; then
+            echo "### Windows 安装包" >> $GITHUB_STEP_SUMMARY
+            echo '```' >> $GITHUB_STEP_SUMMARY
+            cat "$WIN_ARTIFACT_DIR/SHA256SUMS.txt" >> $GITHUB_STEP_SUMMARY
+            echo '```' >> $GITHUB_STEP_SUMMARY
+            echo "" >> $GITHUB_STEP_SUMMARY
+          else
+            echo "### Windows 安装包" >> $GITHUB_STEP_SUMMARY
+            echo "❌ Windows 构建未成功完成或未生成校验和。" >> $GITHUB_STEP_SUMMARY
+            echo "" >> $GITHUB_STEP_SUMMARY
+          fi
+
+          # macOS
+          MAC_ARTIFACT_DIR="all-artifacts/cherry-studio-nightly-${{ steps.date.outputs.date }}-macos-latest"
+          if [ -d "$MAC_ARTIFACT_DIR" ] && [ -f "$MAC_ARTIFACT_DIR/SHA256SUMS.txt" ]; then
+            echo "### macOS 安装包" >> $GITHUB_STEP_SUMMARY
+            echo '```' >> $GITHUB_STEP_SUMMARY
+            cat "$MAC_ARTIFACT_DIR/SHA256SUMS.txt" >> $GITHUB_STEP_SUMMARY
+            echo '```' >> $GITHUB_STEP_SUMMARY
+            echo "" >> $GITHUB_STEP_SUMMARY
+          else
+            echo "### macOS 安装包" >> $GITHUB_STEP_SUMMARY
+            echo "❌ macOS 构建未成功完成或未生成校验和。" >> $GITHUB_STEP_SUMMARY
+            echo "" >> $GITHUB_STEP_SUMMARY
+          fi
+
+          # Linux
+          LINUX_ARTIFACT_DIR="all-artifacts/cherry-studio-nightly-${{ steps.date.outputs.date }}-ubuntu-latest"
+          if [ -d "$LINUX_ARTIFACT_DIR" ] && [ -f "$LINUX_ARTIFACT_DIR/SHA256SUMS.txt" ]; then
+            echo "### Linux 安装包" >> $GITHUB_STEP_SUMMARY
+            echo '```' >> $GITHUB_STEP_SUMMARY
+            cat "$LINUX_ARTIFACT_DIR/SHA256SUMS.txt" >> $GITHUB_STEP_SUMMARY
+            echo '```' >> $GITHUB_STEP_SUMMARY
+            echo "" >> $GITHUB_STEP_SUMMARY
+          else
+            echo "### Linux 安装包" >> $GITHUB_STEP_SUMMARY
+            echo "❌ Linux 构建未成功完成或未生成校验和。" >> $GITHUB_STEP_SUMMARY
+            echo "" >> $GITHUB_STEP_SUMMARY
+          fi
+
+          echo "## ⚠️ Warning: This is a nightly build version" >> $GITHUB_STEP_SUMMARY
+          echo "This version is an unstable version built automatically and is only for testing. It is not recommended to use it in a production environment." >> $GITHUB_STEP_SUMMARY
+          echo "Please backup your data before installing this version and prepare for data migration." >> $GITHUB_STEP_SUMMARY
+          echo "Build date: $(date +'%Y-%m-%d')" >> $GITHUB_STEP_SUMMARY

.github/workflows/pr-ci.yml

@@ -0,0 +1,46 @@
+name: Pull Request CI
+
+on:
+  workflow_dispatch:
+  pull_request:
+    branches:
+      - main
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Check out Git repository
+        uses: actions/checkout@v4
+
+      - name: Install Node.js
+        uses: actions/setup-node@v4
+        with:
+          node-version: 20
+
+      - name: Install corepack
+        run: corepack enable && corepack prepare yarn@4.6.0 --activate
+
+      - name: Get yarn cache directory path
+        id: yarn-cache-dir-path
+        run: echo "dir=$(yarn config get cacheFolder)" >> $GITHUB_OUTPUT
+
+      - name: Cache yarn dependencies
+        uses: actions/cache@v4
+        with:
+          path: |
+            ${{ steps.yarn-cache-dir-path.outputs.dir }}
+            node_modules
+          key: ${{ runner.os }}-yarn-${{ hashFiles('**/yarn.lock') }}
+          restore-keys: |
+            ${{ runner.os }}-yarn-
+
+      - name: Install Dependencies
+        run: yarn install
+
+      - name: Build Check
+        run: yarn build:check
+
+      - name: Lint Check
+        run: yarn lint

.github/workflows/release.yml

@@ -38,19 +38,19 @@ jobs:
           fi
 
       - name: Install Node.js
-        uses: actions/setup-node@v3
+        uses: actions/setup-node@v4
         with:
           node-version: 20
 
       - name: Install corepack
-        run: corepack enable && corepack prepare yarn@4.3.1 --activate
+        run: corepack enable && corepack prepare yarn@4.6.0 --activate
 
       - name: Get yarn cache directory path
         id: yarn-cache-dir-path
         run: echo "dir=$(yarn config get cacheFolder)" >> $GITHUB_OUTPUT
 
       - name: Cache yarn dependencies
-        uses: actions/cache@v3
+        uses: actions/cache@v4
         with:
           path: |
             ${{ steps.yarn-cache-dir-path.outputs.dir }}
@@ -88,7 +88,9 @@ jobs:
 
       - name: Build Windows
         if: matrix.os == 'windows-latest'
-        run: yarn build:win
+        run: |
+          yarn build:npm windows
+          yarn build:win
         env:
           GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
           RENDERER_VITE_AIHUBMIX_SECRET: ${{ vars.RENDERER_VITE_AIHUBMIX_SECRET }}

.gitignore

@@ -35,7 +35,6 @@ Thumbs.db
 node_modules
 dist
 out
-build/icons
 stats.html
 
 # ENV

.husky/pre-commit

@@ -0,0 +1 @@
+yarn lint-staged

.npmrc

@@ -0,0 +1 @@
+electron_mirror=https://npmmirror.com/mirrors/electron/

.prettierignore

@@ -6,3 +6,4 @@ tsconfig.json
 tsconfig.*.json
 CHANGELOG*.md
 agents.json
+src/renderer/src/integration/nutstore/sso/lib

.vscode/settings.json

@@ -4,7 +4,8 @@
     "source.fixAll.eslint": "explicit"
   },
   "search.exclude": {
-    "**/dist/**": true
+    "**/dist/**": true,
+    ".yarn/releases/**": true
   },
   "[javascript]": {
     "editor.defaultFormatter": "esbenp.prettier-vscode"

.yarn/patches/@llm-tools-embedjs-loader-markdown-npm-0.1.28-81647ffac6.patch

@@ -1,13 +0,0 @@
-diff --git a/src/markdown-loader.js b/src/markdown-loader.js
-index eaf30b114a273e68abbb92c8b07018495e63f4cb..4b06519bdb51845e4693fe877da9de01c7a81039 100644
---- a/src/markdown-loader.js
-+++ b/src/markdown-loader.js
-@@ -21,7 +21,7 @@ export class MarkdownLoader extends BaseLoader {
-             ? (await getSafe(this.filePathOrUrl, { format: 'buffer' })).body
-             : await streamToBuffer(fs.createReadStream(this.filePathOrUrl));
-         this.debug('MarkdownLoader stream created');
--        const result = micromark(buffer, { extensions: [gfm(), mdxJsx()], htmlExtensions: [gfmHtml()] });
-+        const result = micromark(buffer, { extensions: [gfm()], htmlExtensions: [gfmHtml()] });
-         this.debug('Markdown parsed...');
-         const webLoader = new WebLoader({
-             urlOrContent: result,

.yarn/patches/@llm-tools-embedjs-npm-0.1.28-8e4393fa2d.patch

@@ -1,158 +0,0 @@
-diff --git a/src/loaders/local-path-loader.d.ts b/src/loaders/local-path-loader.d.ts
-index 48c20e68c469cd309be2dc8f28e44c1bd04a26e9..1c16d83bcbf9b7140292793d6cbb8c04281949d9 100644
---- a/src/loaders/local-path-loader.d.ts
-+++ b/src/loaders/local-path-loader.d.ts
-@@ -4,8 +4,10 @@ export declare class LocalPathLoader extends BaseLoader<{
- }> {
-     private readonly debug;
-     private readonly path;
--    constructor({ path }: {
-+    constructor({ path, chunkSize, chunkOverlap }: {
-         path: string;
-+        chunkSize?: number;
-+        chunkOverlap?: number;
-     });
-     getUnfilteredChunks(): AsyncGenerator<{
-         metadata: {
-diff --git a/src/loaders/local-path-loader.js b/src/loaders/local-path-loader.js
-index 4cf8a6bd1d890244c8ec49d4a05ee3bd58861c79..ec8215b01195a21ef20f3c5d56ecc99f186bb596 100644
---- a/src/loaders/local-path-loader.js
-+++ b/src/loaders/local-path-loader.js
-@@ -8,8 +8,8 @@ import { BaseLoader } from '@llm-tools/embedjs-interfaces';
- export class LocalPathLoader extends BaseLoader {
-     debug = createDebugMessages('embedjs:loader:LocalPathLoader');
-     path;
--    constructor({ path }) {
--        super(`LocalPathLoader_${md5(path)}`, { path });
-+    constructor({ path, chunkSize, chunkOverlap }) {
-+        super(`LocalPathLoader_${md5(path)}`, { path }, chunkSize ?? 1000, chunkOverlap ?? 0);
-         this.path = path;
-     }
-     async *getUnfilteredChunks() {
-@@ -36,10 +36,12 @@ export class LocalPathLoader extends BaseLoader {
-                 const extension = currentPath.split('.').pop().toLowerCase();
-                 if (extension === 'md' || extension === 'mdx')
-                     mime = 'text/markdown';
-+                if (extension === 'txt')
-+                    mime = 'text/plain';
-                 this.debug(`File '${this.path}' mime type updated to 'text/markdown'`);
-             }
-             try {
--                const loader = await createLoaderFromMimeType(currentPath, mime);
-+                const loader = await createLoaderFromMimeType(currentPath, mime, this.chunkSize, this.chunkOverlap);
-                 for await (const result of await loader.getUnfilteredChunks()) {
-                     yield {
-                         pageContent: result.pageContent,
-diff --git a/src/util/mime.d.ts b/src/util/mime.d.ts
-index 57f56a1b8edc98366af9f84d671676c41c2f01ca..14be3b5727cff6eb1978838045e9a788f8f53bfb 100644
---- a/src/util/mime.d.ts
-+++ b/src/util/mime.d.ts
-@@ -1,2 +1,2 @@
- import { BaseLoader } from '@llm-tools/embedjs-interfaces';
--export declare function createLoaderFromMimeType(loaderData: string, mimeType: string): Promise<BaseLoader>;
-+export declare function createLoaderFromMimeType(loaderData: string, mimeType: string, chunkSize?: number, chunkOverlap?: number): Promise<BaseLoader>;
-diff --git a/src/util/mime.js b/src/util/mime.js
-index b6426a859968e2bf6206795f70333e90ae27aeb7..16ae2adb863f8d7abfa757f1c5cc39f6bb1c44fa 100644
---- a/src/util/mime.js
-+++ b/src/util/mime.js
-@@ -1,7 +1,9 @@
- import mime from 'mime';
- import createDebugMessages from 'debug';
- import { TextLoader } from '../loaders/text-loader.js';
--export async function createLoaderFromMimeType(loaderData, mimeType) {
-+import fs from 'node:fs'
-+
-+export async function createLoaderFromMimeType(loaderData, mimeType, chunkSize, chunkOverlap) {
-     createDebugMessages('embedjs:util:createLoaderFromMimeType')(`Incoming mime type '${mimeType}'`);
-     switch (mimeType) {
-         case 'application/msword':
-@@ -10,7 +12,7 @@ export async function createLoaderFromMimeType(loaderData, mimeType) {
-                 throw new Error('Package `@llm-tools/embedjs-loader-msoffice` needs to be installed to load docx files');
-             });
-             createDebugMessages('embedjs:util:createLoaderFromMimeType')('Dynamically imported DocxLoader');
--            return new DocxLoader({ filePathOrUrl: loaderData });
-+            return new DocxLoader({ filePathOrUrl: loaderData, chunkSize, chunkOverlap });
-         }
-         case 'application/vnd.ms-excel':
-         case 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet': {
-@@ -18,21 +20,21 @@ export async function createLoaderFromMimeType(loaderData, mimeType) {
-                 throw new Error('Package `@llm-tools/embedjs-loader-msoffice` needs to be installed to load excel files');
-             });
-             createDebugMessages('embedjs:util:createLoaderFromMimeType')('Dynamically imported ExcelLoader');
--            return new ExcelLoader({ filePathOrUrl: loaderData });
-+            return new ExcelLoader({ filePathOrUrl: loaderData, chunkSize, chunkOverlap  });
-         }
-         case 'application/pdf': {
-             const { PdfLoader } = await import('@llm-tools/embedjs-loader-pdf').catch(() => {
-                 throw new Error('Package `@llm-tools/embedjs-loader-pdf` needs to be installed to load PDF files');
-             });
-             createDebugMessages('embedjs:util:createLoaderFromMimeType')('Dynamically imported PdfLoader');
--            return new PdfLoader({ filePathOrUrl: loaderData });
-+            return new PdfLoader({ filePathOrUrl: loaderData, chunkSize, chunkOverlap });
-         }
-         case 'application/vnd.openxmlformats-officedocument.presentationml.presentation': {
-             const { PptLoader } = await import('@llm-tools/embedjs-loader-msoffice').catch(() => {
-                 throw new Error('Package `@llm-tools/embedjs-loader-msoffice` needs to be installed to load pptx files');
-             });
-             createDebugMessages('embedjs:util:createLoaderFromMimeType')('Dynamically imported PptLoader');
--            return new PptLoader({ filePathOrUrl: loaderData });
-+            return new PptLoader({ filePathOrUrl: loaderData, chunkSize, chunkOverlap });
-         }
-         case 'text/plain': {
-             const fineType = mime.getType(loaderData);
-@@ -42,24 +44,24 @@ export async function createLoaderFromMimeType(loaderData, mimeType) {
-                     throw new Error('Package `@llm-tools/embedjs-loader-csv` needs to be installed to load CSV files');
-                 });
-                 createDebugMessages('embedjs:util:createLoaderFromMimeType')('Dynamically imported CsvLoader');
--                return new CsvLoader({ filePathOrUrl: loaderData });
-+                return new CsvLoader({ filePathOrUrl: loaderData, chunkSize, chunkOverlap });
-             }
--            else
--                return new TextLoader({ text: loaderData });
-+            const content = fs.readFileSync(loaderData, 'utf-8');
-+            return new TextLoader({ text: content, chunkSize, chunkOverlap });
-         }
-         case 'application/csv': {
-             const { CsvLoader } = await import('@llm-tools/embedjs-loader-csv').catch(() => {
-                 throw new Error('Package `@llm-tools/embedjs-loader-csv` needs to be installed to load CSV files');
-             });
-             createDebugMessages('embedjs:util:createLoaderFromMimeType')('Dynamically imported CsvLoader');
--            return new CsvLoader({ filePathOrUrl: loaderData });
-+            return new CsvLoader({ filePathOrUrl: loaderData, chunkSize, chunkOverlap });
-         }
-         case 'text/html': {
-             const { WebLoader } = await import('@llm-tools/embedjs-loader-web').catch(() => {
-                 throw new Error('Package `@llm-tools/embedjs-loader-web` needs to be installed to load web documents');
-             });
-             createDebugMessages('embedjs:util:createLoaderFromMimeType')('Dynamically imported WebLoader');
--            return new WebLoader({ urlOrContent: loaderData });
-+            return new WebLoader({ urlOrContent: loaderData, chunkSize, chunkOverlap });
-         }
-         case 'text/xml': {
-             const { SitemapLoader } = await import('@llm-tools/embedjs-loader-sitemap').catch(() => {
-@@ -67,14 +69,14 @@ export async function createLoaderFromMimeType(loaderData, mimeType) {
-             });
-             createDebugMessages('embedjs:util:createLoaderFromMimeType')('Dynamically imported SitemapLoader');
-             if (await SitemapLoader.test(loaderData)) {
--                return new SitemapLoader({ url: loaderData });
-+                return new SitemapLoader({ url: loaderData, chunkSize, chunkOverlap });
-             }
-             //This is not a Sitemap but is still XML
-             const { XmlLoader } = await import('@llm-tools/embedjs-loader-xml').catch(() => {
-                 throw new Error('Package `@llm-tools/embedjs-loader-xml` needs to be installed to load XML documents');
-             });
-             createDebugMessages('embedjs:util:createLoaderFromMimeType')('Dynamically imported XmlLoader');
--            return new XmlLoader({ filePathOrUrl: loaderData });
-+            return new XmlLoader({ filePathOrUrl: loaderData, chunkSize, chunkOverlap });
-         }
-         case 'text/x-markdown':
-         case 'text/markdown': {
-@@ -82,7 +84,7 @@ export async function createLoaderFromMimeType(loaderData, mimeType) {
-                 throw new Error('Package `@llm-tools/embedjs-loader-markdown` needs to be installed to load markdown files');
-             });
-             createDebugMessages('embedjs:util:createLoaderFromMimeType')('Dynamically imported MarkdownLoader');
--            return new MarkdownLoader({ filePathOrUrl: loaderData });
-+            return new MarkdownLoader({ filePathOrUrl: loaderData, chunkSize, chunkOverlap });
-         }
-         case 'image/png':
-         case 'image/jpeg': {

.yarn/patches/epub-npm-1.3.0-8325494ffe.patch

@@ -0,0 +1,53 @@
+diff --git a/epub.js b/epub.js
+index 50efff7678ca4879ed639d3bb70fd37e7477fd16..accbe689cd200bd59475dd20fca596511d0f33e0 100644
+--- a/epub.js
++++ b/epub.js
+@@ -3,9 +3,28 @@ var xml2jsOptions = xml2js.defaults['0.1'];
+ var EventEmitter = require('events').EventEmitter;
+ 
+ try {
+-    // zipfile is an optional dependency:
+-    var ZipFile = require("zipfile").ZipFile;
+-} catch (err) {
++    var zipread = require("zipread");
++    var ZipFile = function(filename) {
++        var zip = zipread(filename);
++        this.zip = zip;
++        var files = zip.files;
++
++        files = Object.values(files).filter((file) => {
++            return !file.dir;
++        }).map((file) => {
++            return file.name;
++        });
++
++        this.names = files;
++        this.count = this.names.length;
++    };
++    ZipFile.prototype.readFile = function(name, cb) {
++        this.zip.readFile(name
++            , function(err, buffer) {
++                return cb(null, buffer);
++            });
++    };
++} catch(err) {
+     // Mock zipfile using pure-JS adm-zip:
+     var AdmZip = require('adm-zip');
+ 
+diff --git a/package.json b/package.json
+index 8c3dccf0caac8913a2edabd7049b25bb9063c905..57bac3b71ddd73916adbdf00b049089181db5bcb 100644
+--- a/package.json
++++ b/package.json
+@@ -40,10 +40,8 @@
+   ],
+   "dependencies": {
+     "adm-zip": "^0.4.11",
+-    "xml2js": "^0.4.23"
+-  },
+-  "optionalDependencies": {
+-    "zipfile": "^0.5.11"
++    "xml2js": "^0.4.23",
++    "zipread": "^1.3.3"
+   },
+   "devDependencies": {
+     "@types/mocha": "^5.2.5",

.yarn/patches/libsql-npm-0.4.7-444e260fb1.patch

@@ -0,0 +1,17 @@
+diff --git a/index.js b/index.js
+index 4e8423491ab51a9eb9fee22182e4ea0fcc3d3d3b..2846c5d4354c130d478dc99565b3ecd6d85b7d2e 100644
+--- a/index.js
++++ b/index.js
+@@ -19,7 +19,11 @@ function requireNative() {
+       break;
+     }
+   }
+-  return require(`@libsql/${target}`);
++  if (target === "win32-arm64-msvc") {
++    return require(`@strongtz/win32-arm64-msvc`);
++  } else {
++    return require(`@libsql/${target}`);
++  }
+ }
+ 
+ const {

.yarn/patches/openai-npm-4.87.3-2b30a7685f.patch

@@ -1,8 +1,8 @@
 diff --git a/core.js b/core.js
-index e75a18281ce8f051990c5a50bc1076afdddf91a3..e62f796791a155f23d054e74a429516c14d6e11b 100644
+index ebb071d31cd5a14792b62814df072c5971e83300..31e1062d4a7f2422ffec79cf96a35dbb69fe89cb 100644
 --- a/core.js
 +++ b/core.js
-@@ -156,7 +156,7 @@ class APIClient {
+@@ -157,7 +157,7 @@ class APIClient {
              Accept: 'application/json',
              'Content-Type': 'application/json',
              'User-Agent': this.getUserAgent(),
@@ -12,10 +12,10 @@ index e75a18281ce8f051990c5a50bc1076afdddf91a3..e62f796791a155f23d054e74a429516c
          };
      }
 diff --git a/core.mjs b/core.mjs
-index fcef58eb502664c41a77483a00db8adaf29b2817..18c5d6ed4be86b3640931277bdc27700006764d7 100644
+index 9c1a0264dcd73a85de1cf81df4efab9ce9ee2ab7..33f9f1f237f2eb2667a05dae1a7e3dc916f6bfff 100644
 --- a/core.mjs
 +++ b/core.mjs
-@@ -149,7 +149,7 @@ export class APIClient {
+@@ -150,7 +150,7 @@ export class APIClient {
              Accept: 'application/json',
              'Content-Type': 'application/json',
              'User-Agent': this.getUserAgent(),

.yarn/patches/pkce-challenge-npm-4.1.0-fbc51695a3.patch

@@ -0,0 +1,18 @@
+diff --git a/dist/index.node.js b/dist/index.node.js
+index bb108cbc210af5b99e864fd1dd8c555e948ecf7a..8ef8c1aab59215c21d161c0e52125724528ecab8 100644
+--- a/dist/index.node.js
++++ b/dist/index.node.js
+@@ -1,8 +1,11 @@
+ let crypto;
+ crypto =
+     globalThis.crypto?.webcrypto ?? // Node.js 16 REPL has globalThis.crypto as node:crypto
+-        globalThis.crypto ?? // Node.js 18+ 
+-        (await import("node:crypto")).webcrypto; // Node.js 16 non-REPL
++        globalThis.crypto ?? // Node.js 18+
++        (async() => {
++            const crypto = await import("node:crypto");
++            return crypto.webcrypto;
++        })();
+ /**
+  * Creates an array of length `size` of random bytes
+  * @param size

.yarnrc.yml

@@ -3,3 +3,5 @@ enableImmutableInstalls: false
 httpTimeout: 300000
 
 nodeLinker: node-modules
+
+yarnPath: .yarn/releases/yarn-4.6.0.cjs

CONTRIBUTING.md

@@ -40,6 +40,6 @@
 如果您有任何问题或建议，欢迎通过以下方式联系我们：
 
 - 微信：kangfenmao
-- [GitHub Issues](https://github.com/kangfenmao/cherry-studio/issues)
+- [GitHub Issues](https://github.com/CherryHQ/cherry-studio/issues)
 
 感谢您的支持和贡献！我们期待与您一起将 Cherry Studio 打造成更好的产品。

LICENSE

@@ -1,19 +1,16 @@
-## Cherry Studio 用户协议
-
-欢迎使用 Cherry Studio 桌面 AI 客户端工具。请仔细阅读以下协议条款，继续使用本软件即表示您同意本协议内容。
-
 **许可协议**
 
-本软件采用 Apache License 2.0 许可。除 Apache License 2.0 规定的条款外，您在使用 Cherry Studio 时还应遵守以下附加条款：
+采用 Apache License 2.0 修改版许可，并附加以下条件：
 
 **一. 商用许可**
 
-1. **免费商用**：用户在不修改代码的情况下，可以免费用于商业目的。
-2. **商业授权**：如果您满足以下任意条件之一，需取得商业授权：
-   1. 对本软件进行二次修改、开发（包括但不限于修改应用名称、logo、代码以及功能）。
-   2. 为企业客户提供多租户服务，且该服务支持 10 人或以上的使用。
-   3. 预装或集成到硬件设备或产品中进行捆绑销售。
-   4. 政府或教育机构的大规模采购项目，特别是涉及安全、数据隐私等敏感需求时。
+在以下任何一种情况下，您需要联系我们并获得明确的书面商业授权后，方可继续使用 Cherry Studio 材料：
+
+1. **修改与衍生**： 您对 Cherry Studio 材料进行修改或基于其进行衍生开发（包括但不限于修改应用名称、Logo、代码、功能、界面，数据等）。
+2. **企业服务**： 在您的企业内部，或为企业客户提供基于 Cherry Studio 的服务，且该服务支持 10 人及以上累计用户使用。
+3. **硬件捆绑销售**： 您将 Cherry Studio 预装或集成到硬件设备或产品中进行捆绑销售。
+4. **政府或教育机构大规模采购**： 您的使用场景属于政府或教育机构的大规模采购项目，特别是涉及安全、数据隐私等敏感需求时。
+5. **面向公众的公有云服务**：基于 Cherry Studio，提供面向公众的公有云服务。
 
 **二. 贡献者协议**
 
@@ -33,47 +30,33 @@
 
 ---
 
-根据 Apache 许可证 2.0 版（“许可证”）进行许可；除非符合许可证，否则您不得使用此文件。您可以在以下网址获取许可证副本：
-
-http://www.apache.org/licenses/LICENSE-2.0
-
-除非适用法律要求或书面同意，软件根据许可证分发的内容以“原样”分发，不附带任何明示或暗示的保证或条件。请参阅特定语言管理权限的许可证和许可证下的限制。
-
-## Cherry Studio User Agreement
-
-Welcome to Cherry Studio, a desktop AI client tool. Please read the following agreement carefully. By continuing to use this software, you agree to the terms outlined below.
 
 **License Agreement**
 
-This software is licensed under the **Apache License 2.0**. In addition to the terms of the Apache License 2.0, the following additional terms apply to the use of Cherry Studio:
+This software is licensed under a modified version of the Apache License 2.0, with the following additional conditions。
 
-**I. Commercial Use License**
+**I. Commercial Licensing**
 
-1. **Free Commercial Use**: Users can use the software for commercial purposes without modifying the code.
-2. **Commercial License Required**: A commercial license is required if any of the following conditions are met:
-   1. You modify, develop, or alter the software, including but not limited to changes to the application name, logo, code, or functionality.
-   2. You provide multi-tenant services to enterprise customers with 10 or more users.
-   3. You pre-install or integrate the software into hardware devices or products and bundle it for sale.
-   4. You are engaging in large-scale procurement for government or educational institutions, especially involving security, data privacy, or other sensitive requirements.
+You must contact us and obtain explicit written commercial authorization to continue using Cherry Studio materials under any of the following circumstances:
+
+1. **Modifications and Derivatives:** You modify Cherry Studio materials or perform derivative development based on them (including but not limited to changing the application’s name, logo, code, functionality, user interface, data, etc.).
+2. **Enterprise Services:** You use Cherry Studio internally within your enterprise, or you provide Cherry Studio-based services for enterprise customers, and such services support cumulative usage by 10 or more users.
+3. **Hardware Bundling and Sales:** You pre-install or integrate Cherry Studio into hardware devices or products for bundled sale.
+4. **Large-scale Procurement by Government or Educational Institutions:** Your usage scenario involves large-scale procurement projects by government or educational institutions, especially in cases involving sensitive requirements such as security and data privacy.
+5. **Public Cloud Services:** You provide public cloud-based product services utilizing Cherry Studio.
 
 **II. Contributor Agreement**
 
-As a contributor to Cherry Studio, you agree to the following:
+As a contributor to Cherry Studio, you must agree to the following terms:
 
-1. **License Adjustment**: The producer reserves the right to adjust the open-source license as needed, making it stricter or more lenient.
-2. **Commercial Use**: Any code you contribute may be used for commercial purposes, including but not limited to cloud business operations.
+1. **License Adjustments:** The producer reserves the right to adjust the open-source license as necessary, making it more strict or permissive.
+2. **Commercial Usage:** Your contributed code may be used commercially, including but not limited to cloud business operations.
 
 **III. Other Terms**
 
-1. The interpretation of these terms is subject to the discretion of Cherry Studio developers.
-2. These terms may be updated, and users will be notified through the software when changes occur.
+1. Cherry Studio developers reserve the right of final interpretation of these agreement terms.
+2. This agreement may be updated according to practical circumstances, and users will be notified of updates through this software.
 
-For any questions or to request a commercial license, please contact the Cherry Studio development team.
+If you have any questions or need to apply for commercial authorization, please contact the Cherry Studio development team.
 
-Apart from the specific conditions mentioned above, all other rights and restrictions follow the Apache License 2.0. Detailed information about the Apache License 2.0 can be found at http://www.apache.org/licenses/LICENSE-2.0.
-
----
-
-Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.
+Other than these specific conditions, all remaining rights and restrictions follow the Apache License 2.0. For more detailed information regarding Apache License 2.0, please visit http://www.apache.org/licenses/LICENSE-2.0.

README.md

@@ -1,21 +1,26 @@
 <h1 align="center">
-  <a href="https://github.com/kangfenmao/cherry-studio/releases">
-    <img src="https://github.com/kangfenmao/cherry-studio/blob/main/build/icon.png?raw=true" width="150" height="150" alt="banner" /><br>
+  <a href="https://github.com/CherryHQ/cherry-studio/releases">
+    <img src="https://github.com/CherryHQ/cherry-studio/blob/main/build/icon.png?raw=true" width="150" height="150" alt="banner" /><br>
   </a>
 </h1>
   <p align="center">English | <a href="./docs/README.zh.md">中文</a> | <a href="./docs/README.ja.md">日本語</a><br></p>
 <div align="center">
  <a href="https://trendshift.io/repositories/11772" target="_blank"><img src="https://trendshift.io/api/badge/repositories/11772" alt="kangfenmao%2Fcherry-studio | Trendshift" style="width: 250px; height: 55px;" width="250" height="55"/></a>
+ <a href="https://www.producthunt.com/posts/cherry-studio?embed=true&utm_source=badge-featured&utm_medium=badge&utm_souce=badge-cherry&#0045;studio" target="_blank"><img src="https://api.producthunt.com/widgets/embed-image/v1/featured.svg?post_id=496640&theme=light" alt="Cherry&#0032;Studio - AI&#0032;Chatbots&#0044;&#0032;AI&#0032;Desktop&#0032;Client | Product Hunt" style="width: 250px; height: 54px;" width="250" height="54" /></a>
 </div>
 
 # 🍒 Cherry Studio
 
 Cherry Studio is a desktop client that supports for multiple LLM providers, available on Windows, Mac and Linux.
 
-👏 Join [Telegram Group](https://t.me/CherryStudioAI)｜[Discord](https://discord.gg/wez8HtpxqQ) | [QQ Group(1025067911)](https://qm.qq.com/q/RIBAO2pPKS)
+👏 Join [Telegram Group](https://t.me/CherryStudioAI)｜[Discord](https://discord.gg/wez8HtpxqQ) | [QQ Group(472019156)](https://qm.qq.com/q/CbZiBWwCXu)
 
 ❤️ Like Cherry Studio? Give it a star 🌟 or [Sponsor](docs/sponsor.md) to support the development!
 
+# 📖 Guide
+
+https://docs.cherry-ai.com
+
 # 🌠 Screenshot
 
 ![](https://github.com/user-attachments/assets/28585d83-4bf0-4714-b561-8c7bf57cc600)
@@ -28,45 +33,48 @@ Cherry Studio is a desktop client that supports for multiple LLM providers, avai
 
 1. **Diverse LLM Provider Support**:
 
-   - ☁️ Major LLM Cloud Services: OpenAI, Gemini, Anthropic, and more
-   - 🔗 AI Web Service Integration: Claude, Peplexity, Poe, and others
-   - 💻 Local Model Support with Ollama, LM Studio
+- ☁️ Major LLM Cloud Services: OpenAI, Gemini, Anthropic, and more
+- 🔗 AI Web Service Integration: Claude, Peplexity, Poe, and others
+- 💻 Local Model Support with Ollama, LM Studio
 
 2. **AI Assistants & Conversations**:
 
-   - 📚 300+ Pre-configured AI Assistants
-   - 🤖 Custom Assistant Creation
-   - 💬 Multi-model Simultaneous Conversations
+- 📚 300+ Pre-configured AI Assistants
+- 🤖 Custom Assistant Creation
+- 💬 Multi-model Simultaneous Conversations
 
 3. **Document & Data Processing**:
 
-   - 📄 Support for Text, Images, Office, PDF, and more
-   - ☁️ WebDAV File Management and Backup
-   - 📊 Mermaid Chart Visualization
-   - 💻 Code Syntax Highlighting
+- 📄 Support for Text, Images, Office, PDF, and more
+- ☁️ WebDAV File Management and Backup
+- 📊 Mermaid Chart Visualization
+- 💻 Code Syntax Highlighting
 
 4. **Practical Tools Integration**:
 
-   - 🔍 Global Search Functionality
-   - 📝 Topic Management System
-   - 🔤 AI-powered Translation
-   - 🎯 Drag-and-drop Sorting
-   - 🔌 Mini Program Support
+- 🔍 Global Search Functionality
+- 📝 Topic Management System
+- 🔤 AI-powered Translation
+- 🎯 Drag-and-drop Sorting
+- 🔌 Mini Program Support
+- ⚙️ MCP(Model Context Protocol) Server
 
 5. **Enhanced User Experience**:
-   - 🖥️ Cross-platform Support for Windows, Mac, and Linux
-   - 📦 Ready to Use, No Environment Setup Required
-   - 🎨 Light/Dark Themes and Transparent Window
-   - 📝 Complete Markdown Rendering
-   - 🤲 Easy Content Sharing
+
+- 🖥️ Cross-platform Support for Windows, Mac, and Linux
+- 📦 Ready to Use, No Environment Setup Required
+- 🎨 Light/Dark Themes and Transparent Window
+- 📝 Complete Markdown Rendering
+- 🤲 Easy Content Sharing
 
 # 📝 TODO
 
 - [x] Quick popup (read clipboard, quick question, explain, translate, summarize)
 - [x] Comparison of multi-model answers
 - [x] Support login using SSO provided by service providers
-- [ ] All models support networking (in development...)
-- [ ] Launch of the first official version
+- [x] All models support networking
+- [x] Launch of the first official version
+- [x] Bug fixes and improvements (In progress...)
 - [ ] Plugin functionality (JavaScript)
 - [ ] Browser extension (highlight text to translate, summarize, add to knowledge base)
 - [ ] iOS & Android client
@@ -74,38 +82,17 @@ Cherry Studio is a desktop client that supports for multiple LLM providers, avai
 - [ ] Voice input and output (AI call)
 - [ ] Data backup supports custom backup content
 
+# 🌈 Theme
+
+- Theme Gallery: https://cherrycss.com
+- Aero Theme: https://github.com/hakadao/CherryStudio-Aero
+- PaperMaterial Theme: https://github.com/rainoffallingstar/CherryStudio-PaperMaterial
+
+Welcome PR for more themes
+
 # 🖥️ Develop
 
-## IDE Setup
-
-[Cursor](https://www.cursor.com/) + [ESLint](https://marketplace.visualstudio.com/items?itemName=dbaeumer.vscode-eslint) + [Prettier](https://marketplace.visualstudio.com/items?itemName=esbenp.prettier-vscode)
-
-## Project Setup
-
-### Install
-
-```bash
-$ yarn
-```
-
-### Development
-
-```bash
-$ yarn dev
-```
-
-### Build
-
-```bash
-# For windows
-$ yarn build:win
-
-# For macOS
-$ yarn build:mac
-
-# For Linux
-$ yarn build:linux
-```
+Refer to the [development documentation](docs/dev.md)
 
 # 🤝 Contributing
 
@@ -134,20 +121,18 @@ Thank you for your support and contributions!
 
 - [one-api](https://github.com/songquanpeng/one-api):LLM API management and distribution system, supporting mainstream models like OpenAI, Azure, and Anthropic. Features unified API interface, suitable for key management and secondary distribution.
 
+- [ublacklist](https://github.com/iorate/ublacklist):Blocks specific sites from appearing in Google search results
+
 # 🚀 Contributors
 
-<a href="https://github.com/kangfenmao/cherry-studio/graphs/contributors">
+<a href="https://github.com/CherryHQ/cherry-studio/graphs/contributors">
   <img src="https://contrib.rocks/image?repo=kangfenmao/cherry-studio" />
 </a>
 <br /><br />
 
 # 🌐 Community
 
-[Telegram](https://t.me/CherryStudioAI) | [Email](mailto:kangfenmao@gmail.com) | [Twitter](https://x.com/kangfenmao)
-
-# 📣 Product Hunt
-
-<a href="https://www.producthunt.com/posts/cherry-studio?embed=true&utm_source=badge-featured&utm_medium=badge&utm_souce=badge-cherry&#0045;studio" target="_blank"><img src="https://api.producthunt.com/widgets/embed-image/v1/featured.svg?post_id=496640&theme=light" alt="Cherry&#0032;Studio - AI&#0032;Chatbots&#0044;&#0032;AI&#0032;Desktop&#0032;Client | Product Hunt" style="width: 250px; height: 54px;" width="250" height="54" /></a>
+[Telegram](https://t.me/CherryStudioAI) | [Email](mailto:support@cherry-ai.com) | [Twitter](https://x.com/kangfenmao)
 
 # ☕ Sponsor
 
@@ -157,6 +142,10 @@ Thank you for your support and contributions!
 
 [LICENSE](./LICENSE)
 
+# ✉️ Contact
+
+yinsenho@cherry-ai.com
+
 # ⭐️ Star History
 
 [![Star History Chart](https://api.star-history.com/svg?repos=kangfenmao/cherry-studio&type=Timeline)](https://star-history.com/#kangfenmao/cherry-studio&Timeline)

dev-app-update.yml

@@ -1,8 +1,8 @@
 # provider: generic
 # url: http://127.0.0.1:8080
 # updaterCacheDirName: cherry-studio-updater
-# provider: github
-# repo: cherry-studio
-# owner: kangfenmao
-provider: generic
-url: https://cherrystudio.ocool.online
+provider: github
+repo: cherry-studio
+owner: kangfenmao
+# provider: generic
+# url: https://cherrystudio.ocool.online

docs/README.ja.md

@@ -1,21 +1,26 @@
 <h1 align="center">
-  <a href="https://github.com/kangfenmao/cherry-studio/releases">
-    <img src="https://github.com/kangfenmao/cherry-studio/blob/main/build/icon.png?raw=true" width="150" height="150" alt="banner" />
+  <a href="https://github.com/CherryHQ/cherry-studio/releases">
+    <img src="https://github.com/CherryHQ/cherry-studio/blob/main/build/icon.png?raw=true" width="150" height="150" alt="banner" />
   </a>
 </h1>
-<div align="center">
-  <a href="./README.md">English</a> | <a href="./README.zh.md">中文</a> | 日本語
-</div>
+<p align="center">
+  <a href="https://github.com/CherryHQ/cherry-studio">English</a> | <a href="./README.zh.md">中文</a> | 日本語 <br>
+</p>
 <div align="center">
  <a href="https://trendshift.io/repositories/11772" target="_blank"><img src="https://trendshift.io/api/badge/repositories/11772" alt="kangfenmao%2Fcherry-studio | Trendshift" style="width: 250px; height: 55px;" width="250" height="55"/></a>
+ <a href="https://www.producthunt.com/posts/cherry-studio?embed=true&utm_source=badge-featured&utm_medium=badge&utm_souce=badge-cherry&#0045;studio" target="_blank"><img src="https://api.producthunt.com/widgets/embed-image/v1/featured.svg?post_id=496640&theme=light" alt="Cherry&#0032;Studio - AI&#0032;Chatbots&#0044;&#0032;AI&#0032;Desktop&#0032;Client | Product Hunt" style="width: 250px; height: 54px;" width="250" height="54" /></a>
 </div>
 # 🍒 Cherry Studio
 
-Cherry Studioは、複数のLLMプロバイダーをサポートするデスクトップクライアントで、Windows、Mac、Linuxで利用可能です。
+Cherry Studio は、複数の LLM プロバイダーをサポートするデスクトップクライアントで、Windows、Mac、Linux で利用可能です。
 
-👏 [Telegram](https://t.me/CherryStudioAI)｜[Discord](https://discord.gg/wez8HtpxqQ) | [QQグループ(1025067911)](https://qm.qq.com/q/RIBAO2pPKS)
+👏 [Telegram](https://t.me/CherryStudioAI)｜[Discord](https://discord.gg/wez8HtpxqQ) | [QQグループ(472019156)](https://qm.qq.com/q/CbZiBWwCXu)
 
-❤️ Cherry Studioをお気に入りにしましたか？小さな星をつけてください 🌟 または [スポンサー](sponsor.md) をして開発をサポートしてください！❤️
+❤️ Cherry Studio をお気に入りにしましたか？小さな星をつけてください 🌟 または [スポンサー](sponsor.md) をして開発をサポートしてください！❤️
+
+# 📖 ガイド
+
+https://docs.cherry-ai.com
 
 # 🌠 スクリーンショット
 
@@ -29,96 +34,77 @@ Cherry Studioは、複数のLLMプロバイダーをサポートするデスク
 
 1. **多様な LLM サービス対応**：
 
-   - ☁️ 主要な LLM クラウドサービス対応：OpenAI、Gemini、Anthropic など
-   - 🔗 AI Web サービス統合：Claude、Peplexity、Poe など
-   - 💻 Ollama、LM Studio によるローカルモデル実行対応
+- ☁️ 主要な LLM クラウドサービス対応：OpenAI、Gemini、Anthropic など
+- 🔗 AI Web サービス統合：Claude、Peplexity、Poe など
+- 💻 Ollama、LM Studio によるローカルモデル実行対応
 
 2. **AI アシスタントと対話**：
 
-   - 📚 300+ の事前設定済み AI アシスタント
-   - 🤖 カスタム AI アシスタントの作成
-   - 💬 複数モデルでの同時対話機能
+- 📚 300+ の事前設定済み AI アシスタント
+- 🤖 カスタム AI アシスタントの作成
+- 💬 複数モデルでの同時対話機能
 
 3. **文書とデータ処理**：
 
-   - 📄 テキスト、画像、Office、PDF など多様な形式対応
-   - ☁️ WebDAV によるファイル管理とバックアップ
-   - 📊 Mermaid による図表作成
-   - 💻 コードハイライト機能
+- 📄 テキスト、画像、Office、PDF など多様な形式対応
+- ☁️ WebDAV によるファイル管理とバックアップ
+- 📊 Mermaid による図表作成
+- 💻 コードハイライト機能
 
 4. **実用的なツール統合**：
 
-   - 🔍 グローバル検索機能
-   - 📝 トピック管理システム
-   - 🔤 AI による翻訳機能
-   - 🎯 ドラッグ＆ドロップによる整理
-   - 🔌 ミニプログラム対応
+- 🔍 グローバル検索機能
+- 📝 トピック管理システム
+- 🔤 AI による翻訳機能
+- 🎯 ドラッグ＆ドロップによる整理
+- 🔌 ミニプログラム対応
+- ⚙️ MCP（モデルコンテキストプロトコル） サービス
 
 5. **優れたユーザー体験**：
-   - 🖥️ Windows、Mac、Linux のクロスプラットフォーム対応
-   - 📦 環境構築不要ですぐに使用可能
-   - 🎨 ライト/ダークテーマと透明ウィンドウ対応
-   - 📝 完全な Markdown レンダリング
-   - 🤲 簡単な共有機能
+
+- 🖥️ Windows、Mac、Linux のクロスプラットフォーム対応
+- 📦 環境構築不要ですぐに使用可能
+- 🎨 ライト/ダークテーマと透明ウィンドウ対応
+- 📝 完全な Markdown レンダリング
+- 🤲 簡単な共有機能
 
 # 📝 TODO
 
 - [x] クイックポップアップ（クリップボードの読み取り、簡単な質問、説明、翻訳、要約）
 - [x] 複数モデルの回答の比較
-- [x] サービスプロバイダーが提供するSSOを使用したログインをサポート
-- [ ] すべてのモデルがネットワークをサポート（開発中...）
-- [ ] 最初の公式バージョンのリリース
+- [x] サービスプロバイダーが提供する SSO を使用したログインをサポート
+- [x] すべてのモデルがネットワークをサポート
+- [x] 最初の公式バージョンのリリース
+- [ ] 錯誤修復と改善 (開発中...)
 - [ ] プラグイン機能（JavaScript）
 - [ ] ブラウザ拡張機能（テキストをハイライトして翻訳、要約、ナレッジベースに追加）
 - [ ] iOS & Android クライアント
 - [ ] AIノート
-- [ ] 音声入出力（AIコール）
+- [ ] 音声入出力（AI コール）
 - [ ] データバックアップはカスタムバックアップコンテンツをサポート
 
+# 🌈 テーマ
+
+テーマギャラリー: https://cherrycss.com  
+Aero テーマ: https://github.com/hakadao/CherryStudio-Aero
+
+より多くのテーマのPRを歓迎します
+
 # 🖥️ 開発
 
-## IDEの設定
-
-[Cursor](https://www.cursor.com/) + [ESLint](https://marketplace.visualstudio.com/items?itemName=dbaeumer.vscode-eslint) + [Prettier](https://marketplace.visualstudio.com/items?itemName=esbenp.prettier-vscode)
-
-## プロジェクトの設定
-
-### インストール
-
-```bash
-$ yarn
-```
-
-### 開発
-
-```bash
-$ yarn dev
-```
-
-### ビルド
-
-```bash
-# Windowsの場合
-$ yarn build:win
-
-# macOSの場合
-$ yarn build:mac
-
-# Linuxの場合
-$ yarn build:linux
-```
+参考[開発ドキュメント](dev.md)
 
 # 🤝 貢献
 
-Cherry Studioへの貢献を歓迎します！以下の方法で貢献できます：
+Cherry Studio への貢献を歓迎します！以下の方法で貢献できます：
 
 1. **コードの貢献**：新機能を開発するか、既存のコードを最適化します。
 2. **バグの修正**：見つけたバグを修正します。
-3. **問題の管理**：GitHubの問題を管理するのを手伝います。
+3. **問題の管理**：GitHub の問題を管理するのを手伝います。
 4. **製品デザイン**：デザインの議論に参加します。
 5. **ドキュメントの作成**：ユーザーマニュアルやガイドを改善します。
 6. **コミュニティの参加**：ディスカッションに参加し、ユーザーを支援します。
-7. **使用の促進**：Cherry Studioを広めます。
+7. **使用の促進**：Cherry Studio を広めます。
 
 ## 始め方
 
@@ -127,27 +113,23 @@ Cherry Studioへの貢献を歓迎します！以下の方法で貢献できま
 3. **変更を提出**：変更をコミットしてプッシュします。
 4. **プルリクエストを開く**：変更内容と理由を説明します。
 
-詳細なガイドラインについては、[貢献ガイド](./CONTRIBUTING.md)をご覧ください。
+詳細なガイドラインについては、[貢献ガイド](../CONTRIBUTING.md)をご覧ください。
 
 ご支援と貢献に感謝します！
 
 ## 関連頁版
 
-- [one-api](https://github.com/songquanpeng/one-api):LLM APIの管理・配信システム。OpenAI、Azure、Anthropicなどの主要モデルに対応し、統一APIインターフェースを提供。APIキー管理と再配布に利用可能。
+- [one-api](https://github.com/songquanpeng/one-api)：LLM API の管理・配信システム。OpenAI、Azure、Anthropic などの主要モデルに対応し、統一 API インターフェースを提供。API キー管理と再配布に利用可能。
 
 # 🚀 コントリビューター
 
-<a href="https://github.com/kangfenmao/cherry-studio/graphs/contributors">
+<a href="https://github.com/CherryHQ/cherry-studio/graphs/contributors">
   <img src="https://contrib.rocks/image?repo=kangfenmao/cherry-studio" />
 </a>
 
 # コミュニティ
 
-[Telegram](https://t.me/CherryStudioAI) | [Email](mailto:kangfenmao@gmail.com) | [Twitter](https://x.com/kangfenmao)
-
-# 📣 プロダクトハント
-
-<a href="https://www.producthunt.com/posts/cherry-studio?embed=true&utm_source=badge-featured&utm_medium=badge&utm_souce=badge-cherry&#0045;studio" target="_blank"><img src="https://api.producthunt.com/widgets/embed-image/v1/featured.svg?post_id=496640&theme=light" alt="Cherry&#0032;Studio - AI&#0032;Chatbots&#0044;&#0032;AI&#0032;Desktop&#0032;Client | Product Hunt" style="width: 250px; height: 54px;" width="250" height="54" /></a>
+[Telegram](https://t.me/CherryStudioAI) | [Email](mailto:support@cherry-ai.com) | [Twitter](https://x.com/kangfenmao)
 
 # スポンサー
 
@@ -157,6 +139,10 @@ Cherry Studioへの貢献を歓迎します！以下の方法で貢献できま
 
 [LICENSE](../LICENSE)
 
+# ✉️ お問い合わせ
+
+yinsenho@cherry-ai.com
+
 # ⭐️ スター履歴
 
 [![Star History Chart](https://api.star-history.com/svg?repos=kangfenmao/cherry-studio&type=Timeline)](https://star-history.com/#kangfenmao/cherry-studio&Timeline)

docs/README.zh.md

@@ -1,22 +1,27 @@
 <h1 align="center">
-  <a href="https://github.com/kangfenmao/cherry-studio/releases">
-    <img src="https://github.com/kangfenmao/cherry-studio/blob/main/build/icon.png?raw=true" width="150" height="150" alt="banner" />
+  <a href="https://github.com/CherryHQ/cherry-studio/releases">
+    <img src="https://github.com/CherryHQ/cherry-studio/blob/main/build/icon.png?raw=true" width="150" height="150" alt="banner" />
   </a>
 </h1>
-<div align="center">
-  中文 / <a href="https://github.com/kangfenmao/cherry-studio">English</a> / <a href="./README.ja.md">日本語</a>
-</div>
+<p align="center">
+  <a href="https://github.com/CherryHQ/cherry-studio">English</a> | 中文 | <a href="./README.ja.md">日本語</a><br></p>
 <div align="center">
  <a href="https://trendshift.io/repositories/11772" target="_blank"><img src="https://trendshift.io/api/badge/repositories/11772" alt="kangfenmao%2Fcherry-studio | Trendshift" style="width: 250px; height: 55px;" width="250" height="55"/></a>
+ <a href="https://www.producthunt.com/posts/cherry-studio?embed=true&utm_source=badge-featured&utm_medium=badge&utm_souce=badge-cherry&#0045;studio" target="_blank"><img src="https://api.producthunt.com/widgets/embed-image/v1/featured.svg?post_id=496640&theme=light" alt="Cherry&#0032;Studio - AI&#0032;Chatbots&#0044;&#0032;AI&#0032;Desktop&#0032;Client | Product Hunt" style="width: 250px; height: 54px;" width="250" height="54" /></a>
 </div>
+
 # 🍒 Cherry Studio
 
 Cherry Studio 是一款支持多个大语言模型（LLM）服务商的桌面客户端，兼容 Windows、Mac 和 Linux 系统。
 
-👏 欢迎加入 [Telegram 群组](https://t.me/CherryStudioAI)｜[Discord](https://discord.gg/wez8HtpxqQ) | [QQ群(1025067911)](https://qm.qq.com/q/RIBAO2pPKS)
+👏 欢迎加入 [Telegram 群组](https://t.me/CherryStudioAI)｜[Discord](https://discord.gg/wez8HtpxqQ) | [QQ群(472019156)](https://qm.qq.com/q/CbZiBWwCXu)
 
 ❤️ 喜欢 Cherry Studio? 点亮小星星 🌟 或 [赞助开发者](sponsor.md)! ❤️
 
+# 📖 使用教程
+
+https://docs.cherry-ai.com
+
 # 🌠 界面
 
 ![](https://github.com/user-attachments/assets/28585d83-4bf0-4714-b561-8c7bf57cc600)
@@ -29,84 +34,65 @@ Cherry Studio 是一款支持多个大语言模型（LLM）服务商的桌面客
 
 1. **多样化 LLM 服务支持**：
 
-   - ☁️ 支持主流 LLM 云服务：OpenAI、Gemini、Anthropic、硅基流动等
-   - 🔗 集成流行 AI Web 服务：Claude、Peplexity、Poe、腾讯元宝、知乎直答等
-   - 💻 支持 Ollama、LM Studio 本地模型部署
+- ☁️ 支持主流 LLM 云服务：OpenAI、Gemini、Anthropic、硅基流动等
+- 🔗 集成流行 AI Web 服务：Claude、Peplexity、Poe、腾讯元宝、知乎直答等
+- 💻 支持 Ollama、LM Studio 本地模型部署
 
 2. **智能助手与对话**：
 
-   - 📚 内置 300+ 预配置 AI 助手
-   - 🤖 支持自定义创建专属助手
-   - 💬 多模型同时对话，获得多样化观点
+- 📚 内置 300+ 预配置 AI 助手
+- 🤖 支持自定义创建专属助手
+- 💬 多模型同时对话，获得多样化观点
 
 3. **文档与数据处理**：
 
-   - 📄 支持文本、图片、Office、PDF 等多种格式
-   - ☁️ WebDAV 文件管理与数据备份
-   - 📊 Mermaid 图表可视化
-   - 💻 代码高亮显示
+- 📄 支持文本、图片、Office、PDF 等多种格式
+- ☁️ WebDAV 文件管理与数据备份
+- 📊 Mermaid 图表可视化
+- 💻 代码高亮显示
 
 4. **实用工具集成**：
 
-   - 🔍 全局搜索功能
-   - 📝 话题管理系统
-   - 🔤 AI 驱动的翻译功能
-   - 🎯 拖拽排序
-   - 🔌 小程序支持
+- 🔍 全局搜索功能
+- 📝 话题管理系统
+- 🔤 AI 驱动的翻译功能
+- 🎯 拖拽排序
+- 🔌 小程序支持
+- ⚙️ MCP(模型上下文协议) 服务
 
 5. **优质使用体验**：
-   - 🖥️ Windows、Mac、Linux 跨平台支持
-   - 📦 开箱即用，无需配置环境
-   - 🎨 支持明暗主题与透明窗口
-   - 📝 完整的 Markdown 渲染
-   - 🤲 便捷的内容分享功能
+
+- 🖥️ Windows、Mac、Linux 跨平台支持
+- 📦 开箱即用，无需配置环境
+- 🎨 支持明暗主题与透明窗口
+- 📝 完整的 Markdown 渲染
+- 🤲 便捷的内容分享功能
 
 # 📝 待辦事項
 
-- [x] 快捷彈窗 (讀取剪貼簿、快速提問、解釋、翻譯、總結）
-- [x] 多模型回答對比
-- [x] 支援使用服務供應商提供的 SSO 進行登入
-- [ ] 全部模型支援連網（開發中...）
-- [ ] 推出第一個正式版
+- [x] 快捷弹窗（读取剪贴板、快速提问、解释、翻译、总结）
+- [x] 多模型回答对比
+- [x] 支持使用服务供应商提供的 SSO 进行登入
+- [x] 全部模型支持连网（开发中...）
+- [x] 推出第一个正式版
+- [x] 错误修复和改进（开发中...）
 - [ ] 插件功能（JavaScript）
-- [ ] 瀏覽器插件（劃詞翻譯、總結、新增至知識庫）
-- [ ] iOS & Android 客戶端
-- [ ] AI 筆記
-- [ ] 語音輸入輸出（AI 通話）
-- [ ] 資料備份支援自訂備份內容
+- [ ] 浏览器插件（划词翻译、总结、新增至知识库）
+- [ ] iOS & Android 客户端
+- [ ] AI 笔记
+- [ ] 语音输入输出（AI 通话）
+- [ ] 数据备份支持自定义备份内容
+
+# 🌈 主题
+
+主题库：https://cherrycss.com  
+Aero 主题：https://github.com/hakadao/CherryStudio-Aero
+
+欢迎 PR 更多主题
 
 # 🖥️ 开发
 
-## IDE 设置
-
-[Cursor](https://www.cursor.com/) + [ESLint](https://marketplace.visualstudio.com/items?itemName=dbaeumer.vscode-eslint) + [Prettier](https://marketplace.visualstudio.com/items?itemName=esbenp.prettier-vscode)
-
-## 项目设置
-
-### 安装
-
-```bash
-$ yarn
-```
-
-### 开发
-
-```bash
-$ yarn dev
-```
-
-### 构建
-
-```bash
-# Windows
-$ yarn build:win
-
-# macOS
-$ yarn build:mac
-
-# Linux
-$ yarn build:linux
-```
+参考[开发文档](dev.md)
 
 # 🤝 贡献
 
@@ -127,28 +113,24 @@ $ yarn build:linux
 3. **提交更改**：提交并推送您的更改。
 4. **打开 Pull Request**：描述您的更改和原因。
 
-有关更详细的指南，请参阅我们的 [贡献指南](./CONTRIBUTING.md)。
+有关更详细的指南，请参阅我们的 [贡献指南](../CONTRIBUTING.md)。
 
 感谢您的支持和贡献！
 
 ## 相关项目
 
-- [one-api](https://github.com/songquanpeng/one-api):LLM API管理及分发系统，支持OpenAI、Azure、Anthropic等主流模型，统一API接口，可用于密钥管理与二次分发。
+- [one-api](https://github.com/songquanpeng/one-api)：LLM API 管理及分发系统，支持 OpenAI、Azure、Anthropic 等主流模型，统一 API 接口，可用于密钥管理与二次分发。
 
 # 🚀 贡献者
 
-<a href="https://github.com/kangfenmao/cherry-studio/graphs/contributors">
+<a href="https://github.com/CherryHQ/cherry-studio/graphs/contributors">
   <img src="https://contrib.rocks/image?repo=kangfenmao/cherry-studio" />
 </a>
 <br /><br />
 
 # 🌐 社区
 
-[Telegram](https://t.me/CherryStudioAI) | [Email](mailto:kangfenmao@gmail.com) | [Twitter](https://x.com/kangfenmao)
-
-# 📣 产品猎人
-
-<a href="https://www.producthunt.com/posts/cherry-studio?embed=true&utm_source=badge-featured&utm_medium=badge&utm_souce=badge-cherry&#0045;studio" target="_blank"><img src="https://api.producthunt.com/widgets/embed-image/v1/featured.svg?post_id=496640&theme=light" alt="Cherry&#0032;Studio - AI&#0032;Chatbots&#0044;&#0032;AI&#0032;Desktop&#0032;Client | Product Hunt" style="width: 250px; height: 54px;" width="250" height="54" /></a>
+[Telegram](https://t.me/CherryStudioAI) | [Email](mailto:support@cherry-ai.com) | [Twitter](https://x.com/kangfenmao)
 
 # ☕ 赞助
 
@@ -158,6 +140,10 @@ $ yarn build:linux
 
 [LICENSE](../LICENSE)
 
+# ✉️ 联系我们
+
+yinsenho@cherry-ai.com
+
 # ⭐️ Star 记录
 
 [![Star History Chart](https://api.star-history.com/svg?repos=kangfenmao/cherry-studio&type=Timeline)](https://star-history.com/#kangfenmao/cherry-studio&Timeline)

docs/dev.md

@@ -0,0 +1,51 @@
+# 🖥️ Develop
+
+## IDE Setup
+
+[Cursor](https://www.cursor.com/) + [ESLint](https://marketplace.visualstudio.com/items?itemName=dbaeumer.vscode-eslint) + [Prettier](https://marketplace.visualstudio.com/items?itemName=esbenp.prettier-vscode)
+
+## Project Setup
+
+### Install
+
+```bash
+yarn
+```
+
+### Development
+
+### Setup Node.js
+
+Download and install [Node.js v20.x.x](https://nodejs.org/en/download)
+
+### Setup Yarn
+
+```bash
+corepack enable
+corepack prepare yarn@4.6.0 --activate
+```
+
+### Install Dependencies
+
+```bash
+yarn install
+```
+
+### Start
+
+```bash
+yarn dev
+```
+
+### Build
+
+```bash
+# For windows
+$ yarn build:win
+
+# For macOS
+$ yarn build:mac
+
+# For Linux
+$ yarn build:linux
+```

docs/technical/KnowledgeService.md

@@ -0,0 +1,174 @@
+我来详细介绍 KnowledgeService 中控制并发处理的项目数量和工作负载的部分。
+
+## 并发控制与工作负载管理机制
+
+KnowledgeService 实现了一个精细的任务队列系统，用于控制并发处理的项目数量和工作负载。这个系统主要通过以下几个关键部分实现：
+
+### 1. 关键变量和限制
+
+```typescript
+private workload = 0
+private processingItemCount = 0
+private knowledgeItemProcessingQueueMappingPromise: Map<LoaderTaskOfSet, () => void> = new Map()
+private static MAXIMUM_WORKLOAD = 1024 * 1024 * 80  // 约80MB
+private static MAXIMUM_PROCESSING_ITEM_COUNT = 30
+```
+
+- `workload`: 跟踪当前正在处理的总工作量（以字节为单位）
+- `processingItemCount`: 跟踪当前正在处理的项目数量
+- `MAXIMUM_WORKLOAD`: 设置最大工作负载为80MB
+- `MAXIMUM_PROCESSING_ITEM_COUNT`: 设置最大并发处理项目数为30个
+
+### 2. 工作负载评估
+
+每个任务都有一个评估工作负载的机制，通过 `evaluateTaskWorkload` 属性来表示：
+
+```typescript
+interface EvaluateTaskWorkload {
+  workload: number
+}
+```
+
+不同类型的任务有不同的工作负载评估方式：
+
+- 文件任务：使用文件大小作为工作负载 `{ workload: file.size }`
+- URL任务：使用固定值 `{ workload: 1024 * 1024 * 2 }` (约2MB)
+- 网站地图任务：使用固定值 `{ workload: 1024 * 1024 * 20 }` (约20MB)
+- 笔记任务：使用文本内容的字节长度 `{ workload: contentBytes.length }`
+
+### 3. 任务状态管理
+
+任务通过状态枚举来跟踪其生命周期：
+
+```typescript
+enum LoaderTaskItemState {
+  PENDING, // 等待处理
+  PROCESSING, // 正在处理
+  DONE // 已完成
+}
+```
+
+### 4. 任务队列处理核心逻辑
+
+核心的队列处理逻辑在 `processingQueueHandle` 方法中：
+
+```typescript
+private processingQueueHandle() {
+  const getSubtasksUntilMaximumLoad = (): QueueTaskItem[] => {
+    const queueTaskList: QueueTaskItem[] = []
+    that: for (const [task, resolve] of this.knowledgeItemProcessingQueueMappingPromise) {
+      for (const item of task.loaderTasks) {
+        if (this.maximumLoad()) {
+          break that
+        }
+
+        const { state, task: taskPromise, evaluateTaskWorkload } = item
+
+        if (state !== LoaderTaskItemState.PENDING) {
+          continue
+        }
+
+        const { workload } = evaluateTaskWorkload
+        this.workload += workload
+        this.processingItemCount += 1
+        item.state = LoaderTaskItemState.PROCESSING
+        queueTaskList.push({
+          taskPromise: () =>
+            taskPromise().then(() => {
+              this.workload -= workload
+              this.processingItemCount -= 1
+              task.loaderTasks.delete(item)
+              if (task.loaderTasks.size === 0) {
+                this.knowledgeItemProcessingQueueMappingPromise.delete(task)
+                resolve()
+              }
+              this.processingQueueHandle()
+            }),
+          resolve: () => {},
+          evaluateTaskWorkload
+        })
+      }
+    }
+    return queueTaskList
+  }
+
+  const subTasks = getSubtasksUntilMaximumLoad()
+  if (subTasks.length > 0) {
+    const subTaskPromises = subTasks.map(({ taskPromise }) => taskPromise())
+    Promise.all(subTaskPromises).then(() => {
+      subTasks.forEach(({ resolve }) => resolve())
+    })
+  }
+}
+```
+
+这个方法的工作流程是：
+
+1. 遍历所有待处理的任务集合
+2. 对于每个任务集合中的每个子任务：
+   - 检查是否已达到最大负载（通过 `maximumLoad()` 方法）
+   - 如果任务状态为 PENDING，则：
+     - 增加当前工作负载和处理项目计数
+     - 将任务状态更新为 PROCESSING
+     - 将任务添加到待执行队列
+3. 执行所有收集到的子任务
+4. 当子任务完成时：
+   - 减少工作负载和处理项目计数
+   - 从任务集合中移除已完成的任务
+   - 如果任务集合为空，则解析相应的 Promise
+   - 递归调用 `processingQueueHandle()` 以处理更多任务
+
+### 5. 负载检查
+
+```typescript
+private maximumLoad() {
+  return (
+    this.processingItemCount >= KnowledgeService.MAXIMUM_PROCESSING_ITEM_COUNT ||
+    this.workload >= KnowledgeService.MAXIMUM_WORKLOAD
+  )
+}
+```
+
+这个方法检查当前是否已达到最大负载，通过两个条件：
+
+- 处理项目数量达到上限（30个）
+- 总工作负载达到上限（80MB）
+
+### 6. 任务添加与执行流程
+
+当添加新任务时，流程如下：
+
+1. 创建任务（根据类型不同创建不同的任务）
+2. 通过 `appendProcessingQueue` 将任务添加到队列
+3. 调用 `processingQueueHandle` 开始处理队列中的任务
+
+```typescript
+private appendProcessingQueue(task: LoaderTask): Promise<LoaderReturn> {
+  return new Promise((resolve) => {
+    this.knowledgeItemProcessingQueueMappingPromise.set(loaderTaskIntoOfSet(task), () => {
+      resolve(task.loaderDoneReturn!)
+    })
+  })
+}
+```
+
+## 并发控制的优势
+
+这种并发控制机制有几个重要优势：
+
+1. **资源使用优化**：通过限制同时处理的项目数量和总工作负载，避免系统资源过度使用
+2. **自动调节**：当任务完成时，会自动从队列中获取新任务，保持资源的高效利用
+3. **灵活性**：不同类型的任务有不同的工作负载评估，更准确地反映实际资源需求
+4. **可靠性**：通过状态管理和Promise解析机制，确保任务正确完成并通知调用者
+
+## 实际应用场景
+
+这种并发控制在处理大量数据时特别有用，例如：
+
+- 导入大型目录时，可能包含数百个文件
+- 处理大型网站地图，可能包含大量URL
+- 处理多个用户同时添加知识库项目的请求
+
+通过这种机制，系统可以平滑地处理大量请求，避免资源耗尽，同时保持良好的响应性。
+
+总结来说，KnowledgeService 实现了一个复杂而高效的任务队列系统，通过精确控制并发处理的项目数量和工作负载，确保系统在处理大量数据时保持稳定和高效。

electron-builder.yml

@@ -27,24 +27,25 @@ files:
   - '!node_modules/@tavily/core/node_modules/js-tiktoken'
   - '!node_modules/pdf-parse/lib/pdf.js/{v1.9.426,v1.10.88,v2.0.550}'
   - '!node_modules/mammoth/{mammoth.browser.js,mammoth.browser.min.js}'
-  - '!node_modules/html2canvas/dist/{html2canvas.min.js,html2canvas.esm.js}'
 asarUnpack:
   - resources/**
   - '**/*.{node,dll,metal,exp,lib}'
 win:
   executableName: Cherry Studio
-  artifactName: ${productName}-${version}-portable.${ext}
+  artifactName: ${productName}-${version}-${arch}-setup.${ext}
   target:
     - target: nsis
     - target: portable
 nsis:
-  artifactName: ${productName}-${version}-setup.${ext}
+  artifactName: ${productName}-${version}-${arch}-setup.${ext}
   shortcutName: ${productName}
   uninstallDisplayName: ${productName}
   createDesktopShortcut: always
   allowToChangeInstallationDirectory: true
   oneClick: false
   include: build/nsis-installer.nsh
+portable:
+  artifactName: ${productName}-${version}-${arch}-portable.${ext}
 mac:
   entitlementsInherit: build/entitlements.mac.plist
   notarize: false
@@ -73,16 +74,20 @@ linux:
   maintainer: electronjs.org
   category: Utility
 publish:
-  provider: generic
-  url: https://cherrystudio.ocool.online
+  # provider: generic
+  # url: https://cherrystudio.ocool.online
+  provider: github
+  repo: cherry-studio
+  owner: CherryHQ
 electronDownload:
   mirror: https://npmmirror.com/mirrors/electron/
 afterPack: scripts/after-pack.js
 afterSign: scripts/notarize.js
 releaseInfo:
   releaseNotes: |
-    Web 搜索增加更多配置选项
-    用户消息支持快速重新发送
-    知识库网址支持右键配置别名
-    支持更多类型的思考内容显示
-    知识库错误修复
+    增加对 grok-3 和 Grok-3-mini 的支持
+    助手支持使用拼音排序
+    网络搜索增加 Baidu, Google, Bing 支持（免费使用）
+    网络搜索增加 uBlacklist 订阅
+    快速面板 (QuickPanel) 进行性能优化
+    解决 mcp 依赖工具下载速度问题

electron.vite.config.ts

@@ -12,16 +12,18 @@ export default defineConfig({
     plugins: [
       externalizeDepsPlugin({
         exclude: [
-          '@llm-tools/embedjs',
-          '@llm-tools/embedjs-openai',
-          '@llm-tools/embedjs-loader-web',
-          '@llm-tools/embedjs-loader-markdown',
-          '@llm-tools/embedjs-loader-msoffice',
-          '@llm-tools/embedjs-loader-xml',
-          '@llm-tools/embedjs-loader-pdf',
-          '@llm-tools/embedjs-loader-sitemap',
-          '@llm-tools/embedjs-libsql',
-          '@llm-tools/embedjs-loader-image'
+          '@cherrystudio/embedjs',
+          '@cherrystudio/embedjs-openai',
+          '@cherrystudio/embedjs-loader-web',
+          '@cherrystudio/embedjs-loader-markdown',
+          '@cherrystudio/embedjs-loader-msoffice',
+          '@cherrystudio/embedjs-loader-xml',
+          '@cherrystudio/embedjs-loader-pdf',
+          '@cherrystudio/embedjs-loader-sitemap',
+          '@cherrystudio/embedjs-libsql',
+          '@cherrystudio/embedjs-loader-image',
+          'p-queue',
+          'webdav'
         ]
       }),
       ...visualizerPlugin('main')
@@ -40,10 +42,32 @@ export default defineConfig({
     }
   },
   preload: {
-    plugins: [externalizeDepsPlugin()]
+    plugins: [externalizeDepsPlugin()],
+    resolve: {
+      alias: {
+        '@shared': resolve('packages/shared')
+      }
+    }
   },
   renderer: {
-    plugins: [react(), ...visualizerPlugin('renderer')],
+    plugins: [
+      react({
+        babel: {
+          plugins: [
+            [
+              'styled-components',
+              {
+                displayName: true, // 开发环境下启用组件名称
+                fileName: false, // 不在类名中包含文件名
+                pure: true, // 优化性能
+                ssr: false // 不需要服务端渲染
+              }
+            ]
+          ]
+        }
+      }),
+      ...visualizerPlugin('renderer')
+    ],
     resolve: {
       alias: {
         '@renderer': resolve('src/renderer/src'),
@@ -51,7 +75,7 @@ export default defineConfig({
       }
     },
     optimizeDeps: {
-      exclude: ['chunk-PZ64DZKH.js', 'chunk-JMKENWIY.js', 'chunk-UXYB6GHG.js']
+      exclude: []
     }
   }
 })

eslint.config.mjs

@@ -0,0 +1,68 @@
+import electronConfigPrettier from '@electron-toolkit/eslint-config-prettier'
+import tseslint from '@electron-toolkit/eslint-config-ts'
+import eslint from '@eslint/js'
+import eslintReact from '@eslint-react/eslint-plugin'
+import { defineConfig } from 'eslint/config'
+import reactHooks from 'eslint-plugin-react-hooks'
+import simpleImportSort from 'eslint-plugin-simple-import-sort'
+import unusedImports from 'eslint-plugin-unused-imports'
+
+export default defineConfig([
+  eslint.configs.recommended,
+  tseslint.configs.recommended,
+  electronConfigPrettier,
+  eslintReact.configs['recommended-typescript'],
+  reactHooks.configs['recommended-latest'],
+  {
+    plugins: {
+      'simple-import-sort': simpleImportSort,
+      'unused-imports': unusedImports
+    },
+    rules: {
+      '@typescript-eslint/explicit-function-return-type': 'off',
+      '@typescript-eslint/no-explicit-any': 'off',
+      '@typescript-eslint/no-non-null-asserted-optional-chain': 'off',
+      'simple-import-sort/imports': 'error',
+      'simple-import-sort/exports': 'error',
+      'unused-imports/no-unused-imports': 'error',
+      '@eslint-react/no-prop-types': 'error',
+      'prettier/prettier': ['error', { endOfLine: 'auto' }]
+    }
+  },
+  // Configuration for ensuring compatibility with the original ESLint(8.x) rules
+  ...[
+    {
+      rules: {
+        '@typescript-eslint/no-require-imports': 'off',
+        '@typescript-eslint/no-unused-vars': ['error', { caughtErrors: 'none' }],
+        '@typescript-eslint/no-unused-expressions': 'off',
+        '@typescript-eslint/no-empty-object-type': 'off',
+        '@eslint-react/hooks-extra/no-direct-set-state-in-use-effect': 'off',
+        '@eslint-react/web-api/no-leaked-event-listener': 'off',
+        '@eslint-react/web-api/no-leaked-timeout': 'off',
+        '@eslint-react/no-unknown-property': 'off',
+        '@eslint-react/no-nested-component-definitions': 'off',
+        '@eslint-react/dom/no-dangerously-set-innerhtml': 'off',
+        '@eslint-react/no-array-index-key': 'off',
+        '@eslint-react/no-unstable-default-props': 'off',
+        '@eslint-react/no-unstable-context-value': 'off',
+        '@eslint-react/hooks-extra/prefer-use-state-lazy-initialization': 'off',
+        '@eslint-react/hooks-extra/no-unnecessary-use-prefix': 'off',
+        '@eslint-react/no-children-to-array': 'off'
+      }
+    }
+  ],
+  {
+    ignores: [
+      'node_modules/**',
+      'build/**',
+      'dist/**',
+      'out/**',
+      'local/**',
+      '.yarn/**',
+      '.gitignore',
+      'scripts/cloudflare-worker.js',
+      'src/main/integration/nutstore/sso/lib/**'
+    ]
+  }
+])

package.json

@@ -1,11 +1,11 @@
 {
   "name": "CherryStudio",
-  "version": "1.0.1",
+  "version": "1.2.2-batemo",
   "private": true,
   "description": "A powerful AI assistant for producer.",
   "main": "./out/main/index.js",
-  "author": "kangfenmao@qq.com",
-  "homepage": "https://github.com/kangfenmao/cherry-studio",
+  "author": "support@cherry-ai.com",
+  "homepage": "https://github.com/CherryHQ/cherry-studio",
   "workspaces": {
     "packages": [
       "local",
@@ -18,19 +18,14 @@
     }
   },
   "scripts": {
-    "format": "prettier --write .",
-    "lint": "eslint . --ext .js,.jsx,.cjs,.mjs,.ts,.tsx,.cts,.mts --fix",
-    "typecheck:node": "tsc --noEmit -p tsconfig.node.json --composite false",
-    "typecheck:web": "tsc --noEmit -p tsconfig.web.json --composite false",
-    "typecheck": "npm run typecheck:node && npm run typecheck:web",
     "start": "electron-vite preview",
     "dev": "electron-vite dev",
-    "build:check": "yarn typecheck",
     "build": "npm run typecheck && electron-vite build",
-    "postinstall": "electron-builder install-app-deps",
+    "build:check": "yarn test && yarn typecheck && yarn check:i18n",
     "build:unpack": "dotenv npm run build && electron-builder --dir",
     "build:win": "dotenv npm run build && electron-builder --win",
     "build:win:x64": "dotenv npm run build && electron-builder --win --x64",
+    "build:win:arm64": "dotenv npm run build && electron-builder --win --arm64",
     "build:mac": "dotenv electron-vite build && electron-builder --mac",
     "build:mac:arm64": "dotenv electron-vite build && electron-builder --mac --arm64",
     "build:mac:x64": "dotenv electron-vite build && electron-builder --mac --x64",
@@ -39,71 +34,112 @@
     "build:linux:x64": "dotenv electron-vite build && electron-builder --linux --x64",
     "build:npm": "node scripts/build-npm.js",
     "release": "node scripts/version.js",
-    "publish": "yarn release patch push",
+    "publish": "yarn build:check && yarn release patch push",
     "pulish:artifacts": "cd packages/artifacts && npm publish && cd -",
     "generate:agents": "yarn workspace @cherry-studio/database agents",
     "generate:icons": "electron-icon-builder --input=./build/logo.png --output=build",
     "analyze:renderer": "VISUALIZER_RENDERER=true yarn build",
     "analyze:main": "VISUALIZER_MAIN=true yarn build",
-    "check": "node scripts/check-i18n.js"
+    "typecheck": "npm run typecheck:node && npm run typecheck:web",
+    "typecheck:node": "tsc --noEmit -p tsconfig.node.json --composite false",
+    "typecheck:web": "tsc --noEmit -p tsconfig.web.json --composite false",
+    "check:i18n": "node scripts/check-i18n.js",
+    "test": "npx -y tsx --test src/**/*.test.ts",
+    "format": "prettier --write .",
+    "lint": "eslint . --ext .js,.jsx,.cjs,.mjs,.ts,.tsx,.cts,.mts --fix",
+    "postinstall": "electron-builder install-app-deps",
+    "prepare": "husky"
   },
   "dependencies": {
-    "@electron-toolkit/preload": "^3.0.0",
+    "@cherrystudio/embedjs": "^0.1.28",
+    "@cherrystudio/embedjs-libsql": "^0.1.28",
+    "@cherrystudio/embedjs-loader-csv": "^0.1.28",
+    "@cherrystudio/embedjs-loader-image": "^0.1.28",
+    "@cherrystudio/embedjs-loader-markdown": "^0.1.28",
+    "@cherrystudio/embedjs-loader-msoffice": "^0.1.28",
+    "@cherrystudio/embedjs-loader-pdf": "^0.1.28",
+    "@cherrystudio/embedjs-loader-sitemap": "^0.1.28",
+    "@cherrystudio/embedjs-loader-web": "^0.1.28",
+    "@cherrystudio/embedjs-loader-xml": "^0.1.28",
+    "@cherrystudio/embedjs-openai": "^0.1.28",
     "@electron-toolkit/utils": "^3.0.0",
     "@electron/notarize": "^2.5.0",
-    "@google/generative-ai": "^0.21.0",
-    "@llm-tools/embedjs": "patch:@llm-tools/embedjs@npm%3A0.1.28#~/.yarn/patches/@llm-tools-embedjs-npm-0.1.28-8e4393fa2d.patch",
-    "@llm-tools/embedjs-libsql": "^0.1.28",
-    "@llm-tools/embedjs-loader-csv": "^0.1.28",
-    "@llm-tools/embedjs-loader-markdown": "patch:@llm-tools/embedjs-loader-markdown@npm%3A0.1.28#~/.yarn/patches/@llm-tools-embedjs-loader-markdown-npm-0.1.28-81647ffac6.patch",
-    "@llm-tools/embedjs-loader-msoffice": "^0.1.28",
-    "@llm-tools/embedjs-loader-pdf": "^0.1.28",
-    "@llm-tools/embedjs-loader-sitemap": "^0.1.28",
-    "@llm-tools/embedjs-loader-web": "^0.1.28",
-    "@llm-tools/embedjs-loader-xml": "^0.1.28",
-    "@llm-tools/embedjs-openai": "^0.1.28",
+    "@google/generative-ai": "^0.24.0",
+    "@langchain/community": "^0.3.36",
+    "@mozilla/readability": "^0.6.0",
     "@notionhq/client": "^2.2.15",
+    "@strongtz/win32-arm64-msvc": "^0.4.7",
+    "@tryfabric/martian": "^1.2.4",
     "@types/react-infinite-scroll-component": "^5.0.0",
+    "@xyflow/react": "^12.4.4",
     "adm-zip": "^0.5.16",
-    "apache-arrow": "^18.1.0",
+    "async-mutex": "^0.5.0",
+    "color": "^5.0.0",
+    "d3": "^7.9.0",
+    "diff": "^7.0.0",
     "docx": "^9.0.2",
     "electron-log": "^5.1.5",
     "electron-store": "^8.2.0",
     "electron-updater": "^6.3.9",
     "electron-window-state": "^5.0.3",
-    "epub": "^1.3.0",
+    "epub": "patch:epub@npm%3A1.3.0#~/.yarn/patches/epub-npm-1.3.0-8325494ffe.patch",
+    "fast-xml-parser": "^5.0.9",
+    "fetch-socks": "^1.3.2",
     "fs-extra": "^11.2.0",
-    "html2canvas": "^1.4.1",
+    "got-scraping": "^4.1.1",
+    "jsdom": "^26.0.0",
     "markdown-it": "^14.1.0",
     "officeparser": "^4.1.1",
-    "tokenx": "^0.4.1",
-    "webdav": "4.11.4"
+    "proxy-agent": "^6.5.0",
+    "tar": "^7.4.3",
+    "tiny-pinyin": "^1.3.2",
+    "turndown": "^7.2.0",
+    "turndown-plugin-gfm": "^1.0.2",
+    "undici": "^7.4.0",
+    "webdav": "^5.8.0",
+    "zipread": "^1.3.3"
   },
   "devDependencies": {
-    "@anthropic-ai/sdk": "^0.24.3",
-    "@electron-toolkit/eslint-config-prettier": "^2.0.0",
-    "@electron-toolkit/eslint-config-ts": "^1.0.1",
+    "@agentic/exa": "^7.3.3",
+    "@agentic/searxng": "^7.3.3",
+    "@agentic/tavily": "^7.3.3",
+    "@analytics/google-analytics": "^1.1.0",
+    "@ant-design/v5-patch-for-react-19": "^1.0.3",
+    "@anthropic-ai/sdk": "^0.38.0",
+    "@electron-toolkit/eslint-config-prettier": "^3.0.0",
+    "@electron-toolkit/eslint-config-ts": "^3.0.0",
+    "@electron-toolkit/preload": "^3.0.0",
     "@electron-toolkit/tsconfig": "^1.0.1",
+    "@emotion/is-prop-valid": "^1.3.1",
+    "@eslint-react/eslint-plugin": "^1.36.1",
+    "@eslint/js": "^9.22.0",
+    "@google/genai": "^0.4.0",
     "@hello-pangea/dnd": "^16.6.0",
     "@kangfenmao/keyv-storage": "^0.1.0",
-    "@llm-tools/embedjs-loader-image": "^0.1.28",
+    "@modelcontextprotocol/sdk": "^1.9.0",
+    "@notionhq/client": "^2.2.15",
     "@reduxjs/toolkit": "^2.2.5",
     "@tavily/core": "patch:@tavily/core@npm%3A0.3.1#~/.yarn/patches/@tavily-core-npm-0.3.1-fe69bf2bea.patch",
+    "@tryfabric/martian": "^1.2.4",
     "@types/adm-zip": "^0",
+    "@types/d3": "^7",
+    "@types/diff": "^7",
     "@types/fs-extra": "^11",
     "@types/lodash": "^4.17.5",
     "@types/markdown-it": "^14",
     "@types/md5": "^2.3.5",
     "@types/node": "^18.19.9",
     "@types/pako": "^1.0.2",
-    "@types/react": "^18.2.48",
-    "@types/react-dom": "^18.2.18",
+    "@types/react": "^19.0.12",
+    "@types/react-dom": "^19.0.4",
     "@types/react-infinite-scroll-component": "^5.0.0",
     "@types/tinycolor2": "^1",
     "@vitejs/plugin-react": "^4.2.1",
+    "analytics": "^0.8.16",
     "antd": "^5.22.5",
     "applescript": "^1.0.0",
     "axios": "^1.7.3",
+    "babel-plugin-styled-components": "^2.1.4",
     "browser-image-compression": "^2.0.2",
     "dayjs": "^1.11.11",
     "dexie": "^4.0.8",
@@ -116,18 +152,24 @@
     "electron-vite": "^2.3.0",
     "emittery": "^1.0.3",
     "emoji-picker-element": "^1.22.1",
-    "eslint": "^8.56.0",
-    "eslint-plugin-react": "^7.34.3",
-    "eslint-plugin-react-hooks": "^4.6.2",
+    "eslint": "^9.22.0",
+    "eslint-plugin-react-hooks": "^5.2.0",
     "eslint-plugin-simple-import-sort": "^12.1.1",
-    "eslint-plugin-unused-imports": "^4.0.0",
+    "eslint-plugin-unused-imports": "^4.1.4",
+    "html-to-image": "^1.11.13",
+    "husky": "^9.1.7",
     "i18next": "^23.11.5",
+    "lint-staged": "^15.5.0",
     "lodash": "^4.17.21",
+    "lru-cache": "^11.1.0",
     "mime": "^4.0.4",
-    "openai": "patch:openai@npm%3A4.77.3#~/.yarn/patches/openai-npm-4.77.3-59c6d42e7a.patch",
-    "prettier": "^3.2.4",
-    "react": "^18.2.0",
-    "react-dom": "^18.2.0",
+    "npx-scope-finder": "^1.2.0",
+    "openai": "patch:openai@npm%3A4.87.3#~/.yarn/patches/openai-npm-4.87.3-2b30a7685f.patch",
+    "p-queue": "^8.1.0",
+    "prettier": "^3.5.3",
+    "rc-virtual-list": "^3.18.5",
+    "react": "^19.0.0",
+    "react-dom": "^19.0.0",
     "react-hotkeys-hook": "^4.6.1",
     "react-i18next": "^14.1.2",
     "react-infinite-scroll-component": "^6.1.0",
@@ -141,27 +183,37 @@
     "rehype-katex": "^7.0.1",
     "rehype-mathjax": "^7.0.0",
     "rehype-raw": "^7.0.0",
+    "remark-cjk-friendly": "^1.1.0",
     "remark-gfm": "^4.0.0",
     "remark-math": "^6.0.0",
     "rollup-plugin-visualizer": "^5.12.0",
     "sass": "^1.77.2",
-    "shiki": "^1.22.2",
+    "shiki": "^3.2.1",
     "string-width": "^7.2.0",
     "styled-components": "^6.1.11",
     "tinycolor2": "^1.6.0",
+    "tokenx": "^0.4.1",
     "typescript": "^5.6.2",
     "uuid": "^10.0.0",
     "vite": "^5.0.12"
   },
-  "peerDependencies": {
-    "react": "^17.0.0 || ^18.0.0",
-    "react-dom": "^17.0.0 || ^18.0.0"
-  },
   "resolutions": {
     "pdf-parse@npm:1.1.1": "patch:pdf-parse@npm%3A1.1.1#~/.yarn/patches/pdf-parse-npm-1.1.1-04a6109b2a.patch",
     "@langchain/openai@npm:^0.3.16": "patch:@langchain/openai@npm%3A0.3.16#~/.yarn/patches/@langchain-openai-npm-0.3.16-e525b59526.patch",
     "@langchain/openai@npm:>=0.1.0 <0.4.0": "patch:@langchain/openai@npm%3A0.3.16#~/.yarn/patches/@langchain-openai-npm-0.3.16-e525b59526.patch",
-    "openai@npm:^4.77.0": "patch:openai@npm%3A4.77.3#~/.yarn/patches/openai-npm-4.77.3-59c6d42e7a.patch"
+    "node-gyp": "^9.1.0",
+    "libsql@npm:^0.4.4": "patch:libsql@npm%3A0.4.7#~/.yarn/patches/libsql-npm-0.4.7-444e260fb1.patch",
+    "openai@npm:^4.77.0": "patch:openai@npm%3A4.87.3#~/.yarn/patches/openai-npm-4.87.3-2b30a7685f.patch",
+    "pkce-challenge@npm:^4.1.0": "patch:pkce-challenge@npm%3A4.1.0#~/.yarn/patches/pkce-challenge-npm-4.1.0-fbc51695a3.patch"
   },
-  "packageManager": "yarn@4.6.0"
+  "packageManager": "yarn@4.6.0",
+  "lint-staged": {
+    "*.{js,jsx,ts,tsx,cjs,mjs,cts,mts}": [
+      "prettier --write",
+      "eslint --fix"
+    ],
+    "*.{json,md,yml,yaml,css,scss,html}": [
+      "prettier --write"
+    ]
+  }
 }

packages/database/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cherry-studio/database",
-  "packageManager": "yarn@4.3.1",
+  "packageManager": "yarn@4.6.0",
   "dependencies": {
     "csv-parser": "^3.0.0",
     "sqlite3": "^5.1.7"

packages/shared/IpcChannel.ts

@@ -0,0 +1,166 @@
+export enum IpcChannel {
+  App_ClearCache = 'app:clear-cache',
+  App_SetLaunchOnBoot = 'app:set-launch-on-boot',
+  App_SetLanguage = 'app:set-language',
+  App_ShowUpdateDialog = 'app:show-update-dialog',
+  App_CheckForUpdate = 'app:check-for-update',
+  App_Reload = 'app:reload',
+  App_Info = 'app:info',
+  App_Proxy = 'app:proxy',
+  App_SetLaunchToTray = 'app:set-launch-to-tray',
+  App_SetTray = 'app:set-tray',
+  App_SetTrayOnClose = 'app:set-tray-on-close',
+  App_RestartTray = 'app:restart-tray',
+  App_SetTheme = 'app:set-theme',
+
+  App_IsBinaryExist = 'app:is-binary-exist',
+  App_GetBinaryPath = 'app:get-binary-path',
+  App_InstallUvBinary = 'app:install-uv-binary',
+  App_InstallBunBinary = 'app:install-bun-binary',
+
+  // Open
+  Open_Path = 'open:path',
+  Open_Website = 'open:website',
+
+  Minapp = 'minapp',
+
+  Config_Set = 'config:set',
+  Config_Get = 'config:get',
+
+  MiniWindow_Show = 'miniwindow:show',
+  MiniWindow_Hide = 'miniwindow:hide',
+  MiniWindow_Close = 'miniwindow:close',
+  MiniWindow_Toggle = 'miniwindow:toggle',
+  MiniWindow_SetPin = 'miniwindow:set-pin',
+
+  // Mcp
+  Mcp_RemoveServer = 'mcp:remove-server',
+  Mcp_RestartServer = 'mcp:restart-server',
+  Mcp_StopServer = 'mcp:stop-server',
+  Mcp_ListTools = 'mcp:list-tools',
+  Mcp_CallTool = 'mcp:call-tool',
+  Mcp_ListPrompts = 'mcp:list-prompts',
+  Mcp_GetPrompt = 'mcp:get-prompt',
+  Mcp_GetInstallInfo = 'mcp:get-install-info',
+  Mcp_ServersChanged = 'mcp:servers-changed',
+  Mcp_ServersUpdated = 'mcp:servers-updated',
+
+  //copilot
+  Copilot_GetAuthMessage = 'copilot:get-auth-message',
+  Copilot_GetCopilotToken = 'copilot:get-copilot-token',
+  Copilot_SaveCopilotToken = 'copilot:save-copilot-token',
+  Copilot_GetToken = 'copilot:get-token',
+  Copilot_Logout = 'copilot:logout',
+  Copilot_GetUser = 'copilot:get-user',
+
+  // obsidian
+  Obsidian_GetVaults = 'obsidian:get-vaults',
+  Obsidian_GetFiles = 'obsidian:get-files',
+
+  // nutstore
+  Nutstore_GetSsoUrl = 'nutstore:get-sso-url',
+  Nutstore_DecryptToken = 'nutstore:decrypt-token',
+  Nutstore_GetDirectoryContents = 'nutstore:get-directory-contents',
+
+  //aes
+  Aes_Encrypt = 'aes:encrypt',
+  Aes_Decrypt = 'aes:decrypt',
+
+  Gemini_UploadFile = 'gemini:upload-file',
+  Gemini_Base64File = 'gemini:base64-file',
+  Gemini_RetrieveFile = 'gemini:retrieve-file',
+  Gemini_ListFiles = 'gemini:list-files',
+  Gemini_DeleteFile = 'gemini:delete-file',
+
+  Windows_ResetMinimumSize = 'window:reset-minimum-size',
+  Windows_SetMinimumSize = 'window:set-minimum-size',
+
+  SelectionMenu_Action = 'selection-menu:action',
+
+  KnowledgeBase_Create = 'knowledge-base:create',
+  KnowledgeBase_Reset = 'knowledge-base:reset',
+  KnowledgeBase_Delete = 'knowledge-base:delete',
+  KnowledgeBase_Add = 'knowledge-base:add',
+  KnowledgeBase_Remove = 'knowledge-base:remove',
+  KnowledgeBase_Search = 'knowledge-base:search',
+  KnowledgeBase_Rerank = 'knowledge-base:rerank',
+
+  //file
+  File_Open = 'file:open',
+  File_OpenPath = 'file:openPath',
+  File_Save = 'file:save',
+  File_Select = 'file:select',
+  File_Upload = 'file:upload',
+  File_Clear = 'file:clear',
+  File_Read = 'file:read',
+  File_Delete = 'file:delete',
+  File_Get = 'file:get',
+  File_SelectFolder = 'file:selectFolder',
+  File_Create = 'file:create',
+  File_Write = 'file:write',
+  File_SaveImage = 'file:saveImage',
+  File_Base64Image = 'file:base64Image',
+  File_Download = 'file:download',
+  File_Copy = 'file:copy',
+  File_BinaryFile = 'file:binaryFile',
+
+  Fs_Read = 'fs:read',
+
+  Export_Word = 'export:word',
+
+  Shortcuts_Update = 'shortcuts:update',
+
+  // backup
+  Backup_Backup = 'backup:backup',
+  Backup_Restore = 'backup:restore',
+  Backup_BackupToWebdav = 'backup:backupToWebdav',
+  Backup_RestoreFromWebdav = 'backup:restoreFromWebdav',
+  Backup_ListWebdavFiles = 'backup:listWebdavFiles',
+  Backup_CheckConnection = 'backup:checkConnection',
+  Backup_CreateDirectory = 'backup:createDirectory',
+
+  // zip
+  Zip_Compress = 'zip:compress',
+  Zip_Decompress = 'zip:decompress',
+
+  // system
+  System_GetDeviceType = 'system:getDeviceType',
+
+  // events
+  SelectionAction = 'selection-action',
+  BackupProgress = 'backup-progress',
+  ThemeChange = 'theme:change',
+  UpdateDownloadedCancelled = 'update-downloaded-cancelled',
+  RestoreProgress = 'restore-progress',
+  UpdateError = 'update-error',
+  UpdateAvailable = 'update-available',
+  UpdateNotAvailable = 'update-not-available',
+  DownloadProgress = 'download-progress',
+  UpdateDownloaded = 'update-downloaded',
+  DownloadUpdate = 'download-update',
+
+  DirectoryProcessingPercent = 'directory-processing-percent',
+
+  FullscreenStatusChanged = 'fullscreen-status-changed',
+
+  HideMiniWindow = 'hide-mini-window',
+  ShowMiniWindow = 'show-mini-window',
+  MiniWindowReload = 'miniwindow-reload',
+
+  ReduxStateChange = 'redux-state-change',
+  ReduxStoreReady = 'redux-store-ready',
+
+  // Search Window
+  SearchWindow_Open = 'search-window:open',
+  SearchWindow_Close = 'search-window:close',
+  SearchWindow_OpenUrl = 'search-window:open-url',
+
+  // Memory File Storage
+  Memory_LoadData = 'memory:load-data',
+  Memory_SaveData = 'memory:save-data',
+  Memory_DeleteShortMemoryById = 'memory:delete-short-memory-by-id',
+
+  // Long-term Memory File Storage
+  LongTermMemory_LoadData = 'long-term-memory:load-data',
+  LongTermMemory_SaveData = 'long-term-memory:save-data'
+}

packages/shared/config/constant.ts

@@ -22,6 +22,7 @@ export const textExts = [
   '.org', // org-mode 文件
   '.wiki', // VimWiki 文件
   '.tex', // LaTeX 文件
+  '.bib', // BibTeX 文件
   '.srt', // 字幕文件
   '.xhtml', // XHTML 文件
   '.nfo', // 信息文件（主要用于场景发布）
@@ -87,7 +88,7 @@ export const textExts = [
   '.ctp', // CakePHP 视图文件
   '.cfm', // ColdFusion 标记语言文件
   '.cfc', // ColdFusion 组件文件
-  '.m', // Objective-C 源文件
+  '.m', // Objective-C 或 MATLAB 源文件
   '.mm', // Objective-C++ 源文件
   '.gradle', // Gradle 构建文件
   '.groovy', // Gradle 构建文件
@@ -102,7 +103,35 @@ export const textExts = [
   '.cxx', // C++ 源文件
   '.cppm', // C++20 模块接口文件
   '.ipp', // 模板实现文件
-  '.ixx' // C++20 模块实现文件
+  '.ixx', // C++20 模块实现文件
+  '.f90', // Fortran 90 源文件
+  '.f', // Fortran 固定格式源代码文件
+  '.f03', // Fortran 2003+ 源代码文件
+  '.ahk', // AutoHotKey 语言文件
+  '.tcl', // Tcl 脚本
+  '.do', // Questa 或 Modelsim Tcl 脚本
+  '.v', // Verilog 源文件
+  '.sv', // SystemVerilog 源文件
+  '.svh', // SystemVerilog 头文件
+  '.vhd', // VHDL 源文件
+  '.vhdl', // VHDL 源文件
+  '.lef', // Library Exchange Format
+  '.def', // Design Exchange Format
+  '.edif', // Electronic Design Interchange Format
+  '.sdf', // Standard Delay Format
+  '.sdc', // Synopsys Design Constraints
+  '.xdc', // Xilinx Design Constraints
+  '.rpt', // 报告文件
+  '.lisp', // Lisp 脚本
+  '.il', // Cadence SKILL 脚本
+  '.ils', // Cadence SKILL++ 脚本
+  '.sp', // SPICE netlist 文件
+  '.spi', // SPICE netlist 文件
+  '.cir', // SPICE netlist 文件
+  '.net', // SPICE netlist 文件
+  '.scs', // Spectre netlist 文件
+  '.asc', // LTspice netlist schematic 文件
+  '.tf' // Technology File
 ]
 
 export const ZOOM_SHORTCUTS = [
@@ -128,3 +157,8 @@ export const ZOOM_SHORTCUTS = [
     system: true
   }
 ]
+
+export const KB = 1024
+export const MB = 1024 * KB
+export const GB = 1024 * MB
+export const defaultLanguage = 'en-US'

packages/shared/config/nutstore.ts

@@ -0,0 +1 @@
+export const NUTSTORE_HOST = 'https://dav.jianguoyun.com/dav'

resources/cherry-studio/license.html

@@ -1,115 +1,106 @@
-<!doctype html>
+<!DOCTYPE html>
 <html lang="zh-CN">
 
 <head>
-  <meta charset="UTF-8" />
-  <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-  <title>CherryStudio 许可协议-ZH/EN</title>
-  <link href="https://cdn.jsdelivr.net/npm/tailwindcss@2.2.19/dist/tailwind.min.css" rel="stylesheet" />
+  <meta charset="UTF-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  <title>许可协议 | License Agreement</title>
+  <script src="https://cdn.tailwindcss.com"></script>
 </head>
 
-<body class="bg-gray-100 p-8">
-  <div class="container mx-auto bg-white p-6 rounded shadow-lg">
-    <h1 class="text-3xl font-bold mb-6 text-center">Cherry Studio 许可协议</h1>
-    <div class="mb-8">
-      <h2 class="text-2xl font-semibold mb-4">许可协议</h2>
-      <p class="mb-4">
-        本软件采用 <strong>Apache License 2.0</strong> 许可。除 Apache License 2.0 规定的条款外，您在使用 Cherry
-        Studio 时还应遵守以下附加条款：
-      </p>
-      <h3 class="text-xl font-semibold mb-2">一. 商用许可</h3>
-      <ol class="list-decimal list-inside mb-4">
-        <li><strong>免费商用</strong>：用户在不修改代码的情况下，可以免费用于商业目的。</li>
-        <li>
-          <strong>商业授权</strong>：如果您满足以下任意条件之一，需取得商业授权：
-          <ol class="list-decimal list-inside ml-4">
-            <li>对本软件进行二次修改、开发（包括但不限于修改应用名称、logo、代码以及功能）。</li>
-            <li>为企业客户提供多租户服务，且该服务支持 10 人或以上的使用。</li>
-            <li>预装或集成到硬件设备或产品中进行捆绑销售。</li>
-            <li>政府或教育机构的大规模采购项目，特别是涉及安全、数据隐私等敏感需求时。</li>
-          </ol>
-        </li>
-      </ol>
-      <h3 class="text-xl font-semibold mb-2">二. 贡献者协议</h3>
-      <ol class="list-decimal list-inside mb-4">
-        <li><strong>许可调整</strong>：生产者有权根据需要对开源协议进行调整，使其更加严格或宽松。</li>
-        <li><strong>商业用途</strong>：您贡献的代码可能会被用于商业用途，包括但不限于云业务运营。</li>
-      </ol>
-      <h3 class="text-xl font-semibold mb-2">三. 其他条款</h3>
-      <ol class="list-decimal list-inside mb-4">
-        <li>本协议条款的解释权归 Cherry Studio 开发者所有。</li>
-        <li>本协议可能根据实际情况进行更新，更新时将通过本软件通知用户。</li>
-      </ol>
-      <p class="mb-4">如有任何问题或需申请商业授权，请联系 Cherry Studio 开发团队。</p>
-      <p>
-        除上述特定条件外，其他所有权利和限制均遵循 Apache License 2.0。有关 Apache License 2.0 的详细信息，请访问
-        <a href="http://www.apache.org/licenses/LICENSE-2.0"
-          class="text-blue-500 underline">http://www.apache.org/licenses/LICENSE-2.0</a>
-      </p>
+<body class="bg-gray-50">
+  <div class="max-w-4xl mx-auto px-4 py-8">
+    <!-- 中文版本 -->
+    <div class="mb-12">
+      <h1 class="text-3xl font-bold mb-8 text-gray-900">许可协议</h1>
+
+      <p class="mb-6 text-gray-700">采用 Apache License 2.0 修改版许可，并附加以下条件：</p>
+
+      <section class="mb-8">
+        <h2 class="text-xl font-semibold mb-4 text-gray-900">一. 商用许可</h2>
+        <p class="mb-4 text-gray-700">在以下任何一种情况下，您需要联系我们并获得明确的书面商业授权后，方可继续使用 Cherry Studio 材料：</p>
+        <ol class="list-decimal pl-6 space-y-2 text-gray-700">
+          <li><strong>修改与衍生</strong>： 您对 Cherry Studio 材料进行修改或基于其进行衍生开发（包括但不限于修改应用名称、Logo、代码、功能、界面，数据等）。</li>
+          <li><strong>企业服务</strong>： 在您的企业内部，或为企业客户提供基于 Cherry Studio 的服务，且该服务支持 10 人及以上累计用户使用。</li>
+          <li><strong>硬件捆绑销售</strong>： 您将 Cherry Studio 预装或集成到硬件设备或产品中进行捆绑销售。</li>
+          <li><strong>政府或教育机构大规模采购</strong>： 您的使用场景属于政府或教育机构的大规模采购项目，特别是涉及安全、数据隐私等敏感需求时。</li>
+          <li><strong>面向公众的公有云服务</strong>：基于 Cherry Studio，提供面向公众的公有云服务。</li>
+        </ol>
+      </section>
+
+      <section class="mb-8">
+        <h2 class="text-xl font-semibold mb-4 text-gray-900">二. 贡献者协议</h2>
+        <p class="mb-4 text-gray-700">作为 Cherry Studio 的贡献者，您应当同意以下条款：</p>
+        <ol class="list-decimal pl-6 space-y-2 text-gray-700">
+          <li><strong>许可调整</strong>：生产者有权根据需要对开源协议进行调整，使其更加严格或宽松。</li>
+          <li><strong>商业用途</strong>：您贡献的代码可能会被用于商业用途，包括但不限于云业务运营。</li>
+        </ol>
+      </section>
+
+      <section class="mb-8">
+        <h2 class="text-xl font-semibold mb-4 text-gray-900">三. 其他条款</h2>
+        <ol class="list-decimal pl-6 space-y-2 text-gray-700">
+          <li>本协议条款的解释权归 Cherry Studio 开发者所有。</li>
+          <li>本协议可能根据实际情况进行更新，更新时将通过本软件通知用户。</li>
+        </ol>
+      </section>
     </div>
-    <h1 class="text-3xl font-bold mb-6 text-center">Cherry Studio License</h1>
-    <div class="mb-8">
-      <h2 class="text-2xl font-semibold mb-4">License Agreement</h2>
-      <p class="mb-4">
-        This software is licensed under the <strong>Apache License 2.0</strong>. In addition to the terms of the
-        Apache License 2.0, the following additional terms apply to the use of Cherry Studio:
-      </p>
-      <h3 class="text-xl font-semibold mb-2">I. Commercial Use License</h3>
-      <ol class="list-decimal list-inside mb-4">
-        <li>
-          <strong>Free Commercial Use</strong>: Users can use the software for commercial purposes without
-          modifying
-          the code.
-        </li>
-        <li>
-          <strong>Commercial License Required</strong>: A commercial license is required if any of the
-          following
-          conditions are met:
-          <ol class="list-decimal list-inside ml-4">
-            <li>
-              You modify, develop, or alter the software, including but not limited to changes to the
-              application
-              name, logo, code, or functionality.
-            </li>
-            <li>You provide multi-tenant services to enterprise customers with 10 or more users.</li>
-            <li>
-              You pre-install or integrate the software into hardware devices or products and bundle it
-              for sale.
-            </li>
-            <li>
-              You are engaging in large-scale procurement for government or educational institutions,
-              especially
-              involving security, data privacy, or other sensitive requirements.
-            </li>
-          </ol>
-        </li>
-      </ol>
-      <h3 class="text-xl font-semibold mb-2">II. Contributor Agreement</h3>
-      <ol class="list-decimal list-inside mb-4">
-        <li>
-          <strong>License Adjustment</strong>: The producer reserves the right to adjust the open-source
-          license as
-          needed, making it stricter or more lenient.
-        </li>
-        <li>
-          <strong>Commercial Use</strong>: Any code you contribute may be used for commercial purposes,
-          including but
-          not limited to cloud business operations.
-        </li>
-      </ol>
-      <h3 class="text-xl font-semibold mb-2">III. Other Terms</h3>
-      <ol class="list-decimal list-inside mb-4">
-        <li>The interpretation of these terms is subject to the discretion of Cherry Studio developers.</li>
-        <li>These terms may be updated, and users will be notified through the software when changes occur.</li>
-      </ol>
-      <p class="mb-4">
-        For any questions or to request a commercial license, please contact the Cherry Studio development team.
-      </p>
-      <p>
-        Apart from the specific conditions mentioned above, all other rights and restrictions follow the Apache
-        License 2.0. Detailed information about the Apache License 2.0 can be found at
+
+    <hr class="my-12 border-gray-300">
+
+    <!-- English Version -->
+    <div>
+      <h1 class="text-3xl font-bold mb-8 text-gray-900">License Agreement</h1>
+
+      <p class="mb-6 text-gray-700">This software is licensed under a modified version of the Apache License 2.0, with
+        the following additional conditions.</p>
+
+      <section class="mb-8">
+        <h2 class="text-xl font-semibold mb-4 text-gray-900">I. Commercial Licensing</h2>
+        <p class="mb-4 text-gray-700">You must contact us and obtain explicit written commercial authorization to
+          continue using Cherry Studio materials under any of the following circumstances:</p>
+        <ol class="list-decimal pl-6 space-y-2 text-gray-700">
+          <li><strong>Modifications and Derivatives:</strong> You modify Cherry Studio materials or perform derivative
+            development based on them (including but not limited to changing the application's name, logo, code,
+            functionality, user interface, data, etc.).</li>
+          <li><strong>Enterprise Services:</strong> You use Cherry Studio internally within your enterprise, or you
+            provide Cherry Studio-based services for enterprise customers, and such services support cumulative usage by
+            10 or more users.</li>
+          <li><strong>Hardware Bundling and Sales:</strong> You pre-install or integrate Cherry Studio into hardware
+            devices or products for bundled sale.</li>
+          <li><strong>Large-scale Procurement by Government or Educational Institutions:</strong> Your usage scenario
+            involves large-scale procurement projects by government or educational institutions, especially in cases
+            involving sensitive requirements such as security and data privacy.</li>
+          <li><strong>Public Cloud Services:</strong> You provide public cloud-based product services utilizing Cherry
+            Studio.</li>
+        </ol>
+      </section>
+
+      <section class="mb-8">
+        <h2 class="text-xl font-semibold mb-4 text-gray-900">II. Contributor Agreement</h2>
+        <p class="mb-4 text-gray-700">As a contributor to Cherry Studio, you must agree to the following terms:</p>
+        <ol class="list-decimal pl-6 space-y-2 text-gray-700">
+          <li><strong>License Adjustments:</strong> The producer reserves the right to adjust the open-source license as
+            necessary, making it more strict or permissive.</li>
+          <li><strong>Commercial Usage:</strong> Your contributed code may be used commercially, including but not
+            limited to cloud business operations.</li>
+        </ol>
+      </section>
+
+      <section class="mb-8">
+        <h2 class="text-xl font-semibold mb-4 text-gray-900">III. Other Terms</h2>
+        <ol class="list-decimal pl-6 space-y-2 text-gray-700">
+          <li>Cherry Studio developers reserve the right of final interpretation of these agreement terms.</li>
+          <li>This agreement may be updated according to practical circumstances, and users will be notified of updates
+            through this software.</li>
+        </ol>
+      </section>
+
+      <p class="mt-8 text-gray-700">
+        Other than these specific conditions, all remaining rights and restrictions follow the Apache License 2.0. For
+        more detailed information regarding Apache License 2.0, please visit
         <a href="http://www.apache.org/licenses/LICENSE-2.0"
-          class="text-blue-500 underline">http://www.apache.org/licenses/LICENSE-2.0</a>
+          class="text-blue-600 hover:underline">http://www.apache.org/licenses/LICENSE-2.0</a>.
       </p>
     </div>
   </div>

resources/cherry-studio/releases.html

@@ -1,7 +1,6 @@
 <!doctype html>
 <html lang="en">
-
-<head>
+  <head>
     <meta charset="UTF-8" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>Github Releases Timeline</title>
@@ -9,194 +8,201 @@
     <script src="https://unpkg.com/vue@3/dist/vue.global.prod.js"></script>
     <script src="https://cdn.jsdelivr.net/npm/markdown-it@13.0.1/dist/markdown-it.min.js"></script>
     <script src="https://cdn.jsdelivr.net/npm/@tailwindcss/typography@0.5.10/dist/typography.min.css"></script>
-</head>
+  </head>
 
-<body id="app">
+  <body id="app">
     <div :class="isDark ? 'dark-bg' : 'bg'" class="min-h-screen">
-        <div class="max-w-3xl mx-auto py-12 px-4">
-            <h1 class="text-3xl font-bold mb-8" :class="isDark ? 'text-white' : 'text-gray-900'">Release Timeline</h1>
+      <div class="max-w-3xl mx-auto py-12 px-4">
+        <h1 class="text-3xl font-bold mb-8" :class="isDark ? 'text-white' : 'text-gray-900'">Release Timeline</h1>
 
-            <!-- Loading状态 -->
-            <div v-if="loading" class="text-center py-8">
-                <div class="inline-block animate-spin rounded-full h-8 w-8 border-4"
-                    :class="isDark ? 'border-gray-700 border-t-blue-500' : 'border-gray-300 border-t-blue-500'"></div>
-            </div>
-
-            <!-- Error 状态 -->
-            <div v-else-if="error" class="text-red-500 text-center py-8">{{ error }}</div>
-
-            <!-- Release 列表 -->
-            <div v-else class="space-y-8">
-                <div v-for="release in releases" :key="release.id" class="relative pl-8"
-                    :class="isDark ? 'border-l-2 border-gray-700' : 'border-l-2 border-gray-200'">
-                    <div class="absolute -left-2 top-0 w-4 h-4 rounded-full bg-green-500"></div>
-                    <div class="rounded-lg shadow-sm p-6 transition-shadow"
-                        :class="isDark ? 'bg-black hover:shadow-md hover:shadow-black' : 'bg-white hover:shadow-md'">
-                        <div class="flex items-start justify-between mb-4">
-                            <div>
-                                <h2 class="text-xl font-semibold" :class="isDark ? 'text-white' : 'text-gray-900'">
-                                    {{ release.name || release.tag_name }}
-                                </h2>
-                                <p class="text-sm mt-1" :class="isDark ? 'text-gray-400' : 'text-gray-500'">
-                                    {{ formatDate(release.published_at) }}
-                                </p>
-                            </div>
-                            <span class="inline-flex items-center px-3 py-1 rounded-full text-sm font-medium"
-                                :class="isDark ? 'bg-green-900 text-green-200' : 'bg-green-100 text-green-800'">
-                                {{ release.tag_name }}
-                            </span>
-                        </div>
-                        <div class="prose" :class="isDark ? 'text-gray-300 dark-prose' : 'text-gray-600'"
-                            v-html="renderMarkdown(release.body)"></div>
-                    </div>
-                </div>
-            </div>
+        <!-- Loading状态 -->
+        <div v-if="loading" class="text-center py-8">
+          <div
+            class="inline-block animate-spin rounded-full h-8 w-8 border-4"
+            :class="isDark ? 'border-gray-700 border-t-blue-500' : 'border-gray-300 border-t-blue-500'"></div>
         </div>
+
+        <!-- Error 状态 -->
+        <div v-else-if="error" class="text-red-500 text-center py-8">{{ error }}</div>
+
+        <!-- Release 列表 -->
+        <div v-else class="space-y-8">
+          <div
+            v-for="release in releases"
+            :key="release.id"
+            class="relative pl-8"
+            :class="isDark ? 'border-l-2 border-gray-700' : 'border-l-2 border-gray-200'">
+            <div class="absolute -left-2 top-0 w-4 h-4 rounded-full bg-green-500"></div>
+            <div
+              class="rounded-lg shadow-sm p-6 transition-shadow"
+              :class="isDark ? 'bg-black hover:shadow-md hover:shadow-black' : 'bg-white hover:shadow-md'">
+              <div class="flex items-start justify-between mb-4">
+                <div>
+                  <h2 class="text-xl font-semibold" :class="isDark ? 'text-white' : 'text-gray-900'">
+                    {{ release.name || release.tag_name }}
+                  </h2>
+                  <p class="text-sm mt-1" :class="isDark ? 'text-gray-400' : 'text-gray-500'">
+                    {{ formatDate(release.published_at) }}
+                  </p>
+                </div>
+                <span
+                  class="inline-flex items-center px-3 py-1 rounded-full text-sm font-medium"
+                  :class="isDark ? 'bg-green-900 text-green-200' : 'bg-green-100 text-green-800'">
+                  {{ release.tag_name }}
+                </span>
+              </div>
+              <div
+                class="prose"
+                :class="isDark ? 'text-gray-300 dark-prose' : 'text-gray-600'"
+                v-html="renderMarkdown(release.body)"></div>
+            </div>
+          </div>
+        </div>
+      </div>
     </div>
 
     <script>
-        const md = window.markdownit({
-            breaks: true,
-            linkify: true
-        })
+      const md = window.markdownit({
+        breaks: true,
+        linkify: true
+      })
 
-        const { createApp } = Vue
+      const { createApp } = Vue
 
-        createApp({
-            data() {
-                return {
-                    releases: [],
-                    loading: true,
-                    error: null,
-                    isDark: false
-                }
-            },
-            methods: {
-                async fetchReleases() {
-                    try {
-                        this.loading = true
-                        this.error = null
-                        const response = await fetch('https://api.github.com/repos/kangfenmao/cherry-studio/releases')
-                        if (!response.ok) {
-                            throw new Error('Failed to fetch releases')
-                        }
-                        this.releases = await response.json()
-                    } catch (err) {
-                        this.error = 'Error loading releases: ' + err.message
-                    } finally {
-                        this.loading = false
-                    }
-                },
-                formatDate(dateString) {
-                    return new Date(dateString).toLocaleDateString('en-US', {
-                        year: 'numeric',
-                        month: 'long',
-                        day: 'numeric'
-                    })
-                },
-                renderMarkdown(content) {
-                    if (!content) return ''
-                    return md.render(content)
-                },
-                initTheme() {
-                    // 从 URL 参数获取主题设置
-                    const url = new URL(window.location.href)
-                    const theme = url.searchParams.get('theme')
-                    this.isDark = theme === 'dark'
-                }
-            },
-            mounted() {
-                this.initTheme()
-                this.fetchReleases()
+      createApp({
+        data() {
+          return {
+            releases: [],
+            loading: true,
+            error: null,
+            isDark: false
+          }
+        },
+        methods: {
+          async fetchReleases() {
+            try {
+              this.loading = true
+              this.error = null
+              const response = await fetch('https://api.github.com/repos/kangfenmao/cherry-studio/releases')
+              if (!response.ok) {
+                throw new Error('Failed to fetch releases')
+              }
+              this.releases = await response.json()
+            } catch (err) {
+              this.error = 'Error loading releases: ' + err.message
+            } finally {
+              this.loading = false
             }
-        }).mount('#app')
+          },
+          formatDate(dateString) {
+            return new Date(dateString).toLocaleDateString('en-US', {
+              year: 'numeric',
+              month: 'long',
+              day: 'numeric'
+            })
+          },
+          renderMarkdown(content) {
+            if (!content) return ''
+            return md.render(content)
+          },
+          initTheme() {
+            // 从 URL 参数获取主题设置
+            const url = new URL(window.location.href)
+            const theme = url.searchParams.get('theme')
+            this.isDark = theme === 'dark'
+          }
+        },
+        mounted() {
+          this.initTheme()
+          this.fetchReleases()
+        }
+      }).mount('#app')
     </script>
 
     <style>
-        /* 基础的 Markdown 样式 */
-        .prose {
-            line-height: 1.6;
-        }
+      /* 基础的 Markdown 样式 */
+      .prose {
+        line-height: 1.6;
+      }
 
-        .prose h1 {
-            font-size: 1.5em;
-            margin: 1em 0;
-        }
+      .prose h1 {
+        font-size: 1.5em;
+        margin: 1em 0;
+      }
 
-        .prose h2 {
-            font-size: 1.3em;
-            margin: 0.8em 0;
-        }
+      .prose h2 {
+        font-size: 1.3em;
+        margin: 0.8em 0;
+      }
 
-        .prose h3 {
-            font-size: 1.1em;
-            margin: 0.6em 0;
-        }
+      .prose h3 {
+        font-size: 1.1em;
+        margin: 0.6em 0;
+      }
 
-        .prose ul {
-            list-style-type: disc;
-            margin-left: 1.5em;
-            margin-bottom: 1em;
-        }
+      .prose ul {
+        list-style-type: disc;
+        margin-left: 1.5em;
+        margin-bottom: 1em;
+      }
 
-        .prose ol {
-            list-style-type: decimal;
-            margin-left: 1.5em;
-            margin-bottom: 1em;
-        }
+      .prose ol {
+        list-style-type: decimal;
+        margin-left: 1.5em;
+        margin-bottom: 1em;
+      }
 
-        .prose code {
-            padding: 0.2em 0.4em;
-            border-radius: 0.2em;
-            font-size: 0.9em;
-        }
+      .prose code {
+        padding: 0.2em 0.4em;
+        border-radius: 0.2em;
+        font-size: 0.9em;
+      }
 
-        .dark .prose code {
-            background-color: #1f2937;
-        }
+      .dark .prose code {
+        background-color: #1f2937;
+      }
 
-        .prose code {
-            background-color: #f3f4f6;
-        }
+      .prose code {
+        background-color: #f3f4f6;
+      }
 
-        .prose pre code {
-            display: block;
-            padding: 1em;
-            overflow-x: auto;
-        }
+      .prose pre code {
+        display: block;
+        padding: 1em;
+        overflow-x: auto;
+      }
 
-        .prose a {
-            color: #3b82f6;
-            text-decoration: underline;
-        }
+      .prose a {
+        color: #3b82f6;
+        text-decoration: underline;
+      }
 
-        .dark .prose a {
-            color: #60a5fa;
-        }
+      .dark .prose a {
+        color: #60a5fa;
+      }
 
-        .prose blockquote {
-            border-left: 4px solid #e5e7eb;
-            padding-left: 1em;
-            margin: 1em 0;
-        }
+      .prose blockquote {
+        border-left: 4px solid #e5e7eb;
+        padding-left: 1em;
+        margin: 1em 0;
+      }
 
-        .dark .prose blockquote {
-            border-left-color: #374151;
-            color: #9ca3af;
-        }
+      .dark .prose blockquote {
+        border-left-color: #374151;
+        color: #9ca3af;
+      }
 
-        .dark .prose {
-            color: #e5e7eb;
-        }
+      .dark .prose {
+        color: #e5e7eb;
+      }
 
-        .dark-bg {
-            background-color: #151515;
-        }
+      .dark-bg {
+        background-color: #151515;
+      }
 
-        .bg {
-            background-color: #f2f2f2;
-        }
+      .bg {
+        background-color: #f2f2f2;
+      }
     </style>
-</body>
-
-</html>
+  </body>
+</html>

resources/graphrag.html

@@ -1,68 +0,0 @@
-<head>
-  <style>
-    body {
-      margin: 0;
-    }
-  </style>
-  <script src="https://unpkg.com/3d-force-graph"></script>
-</head>
-<body>
-  <div id="3d-graph"></div>
-  <script src="./js/bridge.js"></script>
-  <script type="module">
-    import { getQueryParam } from './js/utils.js'
-
-    const apiUrl = getQueryParam('apiUrl')
-    const modelId = getQueryParam('modelId')
-    const jsonUrl = `${apiUrl}/v1/global_graph/${modelId}`
-
-    const infoCard = document.createElement('div')
-    infoCard.style.position = 'fixed'
-    infoCard.style.backgroundColor = 'rgba(255, 255, 255, 0.9)'
-    infoCard.style.padding = '8px'
-    infoCard.style.borderRadius = '4px'
-    infoCard.style.boxShadow = '0 2px 5px rgba(0,0,0,0.2)'
-    infoCard.style.fontSize = '12px'
-    infoCard.style.maxWidth = '200px'
-    infoCard.style.display = 'none'
-    infoCard.style.zIndex = '1000'
-    document.body.appendChild(infoCard)
-
-    document.addEventListener('mousemove', (event) => {
-      infoCard.style.left = `${event.clientX + 10}px`
-      infoCard.style.top = `${event.clientY + 10}px`
-    })
-
-    const elem = document.getElementById('3d-graph')
-    const Graph = ForceGraph3D()(elem)
-      .jsonUrl(jsonUrl)
-      .nodeAutoColorBy((node) => node.properties.type || 'default')
-      .nodeVal((node) => node.properties.degree)
-      .linkWidth((link) => link.properties.weight)
-      .onNodeHover((node) => {
-        if (node) {
-          infoCard.innerHTML = `
-            <div style="font-weight: bold; margin-bottom: 4px; color: #333;">
-              ${node.properties.title}
-            </div>
-            <div style="color: #666;">
-              ${node.properties.description}
-            </div>`
-          infoCard.style.display = 'block'
-        } else {
-          infoCard.style.display = 'none'
-        }
-      })
-      .onNodeClick((node) => {
-        const url = `${apiUrl}/v1/references/${modelId}/entities/${node.properties.human_readable_id}`
-        window.api.minApp({
-          url,
-          windowOptions: {
-            title: node.properties.title,
-            width: 500,
-            height: 800
-          }
-        })
-      })
-  </script>
-</body>

resources/scripts/download.js

@@ -0,0 +1,35 @@
+const https = require('https')
+const fs = require('fs')
+
+/**
+ * Downloads a file from a URL with redirect handling
+ * @param {string} url The URL to download from
+ * @param {string} destinationPath The path to save the file to
+ * @returns {Promise<void>} Promise that resolves when download is complete
+ */
+async function downloadWithRedirects(url, destinationPath) {
+  return new Promise((resolve, reject) => {
+    const request = (url) => {
+      https
+        .get(url, (response) => {
+          if (response.statusCode == 301 || response.statusCode == 302) {
+            request(response.headers.location)
+            return
+          }
+          if (response.statusCode !== 200) {
+            reject(new Error(`Download failed: ${response.statusCode} ${response.statusMessage}`))
+            return
+          }
+          const file = fs.createWriteStream(destinationPath)
+          response.pipe(file)
+          file.on('finish', () => resolve())
+        })
+        .on('error', (err) => {
+          reject(err)
+        })
+    }
+    request(url)
+  })
+}
+
+module.exports = { downloadWithRedirects }

resources/scripts/install-bun.js

@@ -0,0 +1,171 @@
+const fs = require('fs')
+const path = require('path')
+const os = require('os')
+const { execSync } = require('child_process')
+const AdmZip = require('adm-zip')
+const { downloadWithRedirects } = require('./download')
+
+// Base URL for downloading bun binaries
+const BUN_RELEASE_BASE_URL = 'https://gitcode.com/CherryHQ/bun/releases/download'
+const DEFAULT_BUN_VERSION = '1.2.9' // Default fallback version
+
+// Mapping of platform+arch to binary package name
+const BUN_PACKAGES = {
+  'darwin-arm64': 'bun-darwin-aarch64.zip',
+  'darwin-x64': 'bun-darwin-x64.zip',
+  'win32-x64': 'bun-windows-x64.zip',
+  'win32-x64-baseline': 'bun-windows-x64-baseline.zip',
+  'linux-x64': 'bun-linux-x64.zip',
+  'linux-x64-baseline': 'bun-linux-x64-baseline.zip',
+  'linux-arm64': 'bun-linux-aarch64.zip',
+  // MUSL variants
+  'linux-musl-x64': 'bun-linux-x64-musl.zip',
+  'linux-musl-x64-baseline': 'bun-linux-x64-musl-baseline.zip',
+  'linux-musl-arm64': 'bun-linux-aarch64-musl.zip'
+}
+
+/**
+ * Downloads and extracts the bun binary for the specified platform and architecture
+ * @param {string} platform Platform to download for (e.g., 'darwin', 'win32', 'linux')
+ * @param {string} arch Architecture to download for (e.g., 'x64', 'arm64')
+ * @param {string} version Version of bun to download
+ * @param {boolean} isMusl Whether to use MUSL variant for Linux
+ * @param {boolean} isBaseline Whether to use baseline variant
+ */
+async function downloadBunBinary(platform, arch, version = DEFAULT_BUN_VERSION, isMusl = false, isBaseline = false) {
+  let platformKey = isMusl ? `${platform}-musl-${arch}` : `${platform}-${arch}`
+  if (isBaseline) {
+    platformKey += '-baseline'
+  }
+  const packageName = BUN_PACKAGES[platformKey]
+
+  if (!packageName) {
+    console.error(`No binary available for ${platformKey}`)
+    return false
+  }
+
+  // Create output directory structure
+  const binDir = path.join(os.homedir(), '.cherrystudio', 'bin')
+  // Ensure directories exist
+  fs.mkdirSync(binDir, { recursive: true })
+
+  // Download URL for the specific binary
+  const downloadUrl = `${BUN_RELEASE_BASE_URL}/bun-v${version}/${packageName}`
+  const tempdir = os.tmpdir()
+  // Create a temporary file for the downloaded binary
+  const tempFilename = path.join(tempdir, packageName)
+
+  try {
+    console.log(`Downloading bun ${version} for ${platformKey}...`)
+    console.log(`URL: ${downloadUrl}`)
+
+    // Use the new download function
+    await downloadWithRedirects(downloadUrl, tempFilename)
+
+    // Extract the zip file using adm-zip
+    console.log(`Extracting ${packageName} to ${binDir}...`)
+    const zip = new AdmZip(tempFilename)
+    zip.extractAllTo(tempdir, true)
+
+    // Move files using Node.js fs
+    const sourceDir = path.join(tempdir, packageName.split('.')[0])
+    const files = fs.readdirSync(sourceDir)
+
+    for (const file of files) {
+      const sourcePath = path.join(sourceDir, file)
+      const destPath = path.join(binDir, file)
+
+      fs.copyFileSync(sourcePath, destPath)
+      fs.unlinkSync(sourcePath)
+
+      // Set executable permissions for non-Windows platforms
+      if (platform !== 'win32') {
+        try {
+          // 755 permission: rwxr-xr-x
+          fs.chmodSync(destPath, '755')
+        } catch (error) {
+          console.warn(`Warning: Failed to set executable permissions: ${error.message}`)
+        }
+      }
+    }
+
+    // Clean up
+    fs.unlinkSync(tempFilename)
+    fs.rmSync(sourceDir, { recursive: true })
+
+    console.log(`Successfully installed bun ${version} for ${platformKey}`)
+    return true
+  } catch (error) {
+    console.error(`Error installing bun for ${platformKey}: ${error.message}`)
+    // Clean up temporary file if it exists
+    if (fs.existsSync(tempFilename)) {
+      fs.unlinkSync(tempFilename)
+    }
+
+    // Check if binDir is empty and remove it if so
+    try {
+      const files = fs.readdirSync(binDir)
+      if (files.length === 0) {
+        fs.rmSync(binDir, { recursive: true })
+        console.log(`Removed empty directory: ${binDir}`)
+      }
+    } catch (cleanupError) {
+      console.warn(`Warning: Failed to clean up directory: ${cleanupError.message}`)
+    }
+
+    return false
+  }
+}
+
+/**
+ * Detects current platform and architecture
+ */
+function detectPlatformAndArch() {
+  const platform = os.platform()
+  const arch = os.arch()
+  const isMusl = platform === 'linux' && detectIsMusl()
+  const isBaseline = platform === 'win32'
+
+  return { platform, arch, isMusl, isBaseline }
+}
+
+/**
+ * Attempts to detect if running on MUSL libc
+ */
+function detectIsMusl() {
+  try {
+    // Simple check for Alpine Linux which uses MUSL
+    const output = execSync('cat /etc/os-release').toString()
+    return output.toLowerCase().includes('alpine')
+  } catch (error) {
+    return false
+  }
+}
+
+/**
+ * Main function to install bun
+ */
+async function installBun() {
+  // Get the latest version if no specific version is provided
+  const version = DEFAULT_BUN_VERSION
+  console.log(`Using bun version: ${version}`)
+
+  const { platform, arch, isMusl, isBaseline } = detectPlatformAndArch()
+
+  console.log(
+    `Installing bun ${version} for ${platform}-${arch}${isMusl ? ' (MUSL)' : ''}${isBaseline ? ' (baseline)' : ''}...`
+  )
+
+  await downloadBunBinary(platform, arch, version, isMusl, isBaseline)
+}
+
+// Run the installation
+installBun()
+  .then(() => {
+    console.log('Installation successful')
+    process.exit(0)
+  })
+  .catch((error) => {
+    console.error('Installation failed:', error)
+    process.exit(1)
+  })

resources/scripts/install-uv.js

@@ -0,0 +1,181 @@
+const fs = require('fs')
+const path = require('path')
+const os = require('os')
+const { execSync } = require('child_process')
+const tar = require('tar')
+const AdmZip = require('adm-zip')
+const { downloadWithRedirects } = require('./download')
+
+// Base URL for downloading uv binaries
+const UV_RELEASE_BASE_URL = 'https://gitcode.com/CherryHQ/uv/releases/download'
+const DEFAULT_UV_VERSION = '0.6.14'
+
+// Mapping of platform+arch to binary package name
+const UV_PACKAGES = {
+  'darwin-arm64': 'uv-aarch64-apple-darwin.tar.gz',
+  'darwin-x64': 'uv-x86_64-apple-darwin.tar.gz',
+  'win32-arm64': 'uv-aarch64-pc-windows-msvc.zip',
+  'win32-ia32': 'uv-i686-pc-windows-msvc.zip',
+  'win32-x64': 'uv-x86_64-pc-windows-msvc.zip',
+  'linux-arm64': 'uv-aarch64-unknown-linux-gnu.tar.gz',
+  'linux-ia32': 'uv-i686-unknown-linux-gnu.tar.gz',
+  'linux-ppc64': 'uv-powerpc64-unknown-linux-gnu.tar.gz',
+  'linux-ppc64le': 'uv-powerpc64le-unknown-linux-gnu.tar.gz',
+  'linux-s390x': 'uv-s390x-unknown-linux-gnu.tar.gz',
+  'linux-x64': 'uv-x86_64-unknown-linux-gnu.tar.gz',
+  'linux-armv7l': 'uv-armv7-unknown-linux-gnueabihf.tar.gz',
+  // MUSL variants
+  'linux-musl-arm64': 'uv-aarch64-unknown-linux-musl.tar.gz',
+  'linux-musl-ia32': 'uv-i686-unknown-linux-musl.tar.gz',
+  'linux-musl-x64': 'uv-x86_64-unknown-linux-musl.tar.gz',
+  'linux-musl-armv6l': 'uv-arm-unknown-linux-musleabihf.tar.gz',
+  'linux-musl-armv7l': 'uv-armv7-unknown-linux-musleabihf.tar.gz'
+}
+
+/**
+ * Downloads and extracts the uv binary for the specified platform and architecture
+ * @param {string} platform Platform to download for (e.g., 'darwin', 'win32', 'linux')
+ * @param {string} arch Architecture to download for (e.g., 'x64', 'arm64')
+ * @param {string} version Version of uv to download
+ * @param {boolean} isMusl Whether to use MUSL variant for Linux
+ */
+async function downloadUvBinary(platform, arch, version = DEFAULT_UV_VERSION, isMusl = false) {
+  const platformKey = isMusl ? `${platform}-musl-${arch}` : `${platform}-${arch}`
+  const packageName = UV_PACKAGES[platformKey]
+
+  if (!packageName) {
+    console.error(`No binary available for ${platformKey}`)
+    return false
+  }
+
+  // Create output directory structure
+  const binDir = path.join(os.homedir(), '.cherrystudio', 'bin')
+  // Ensure directories exist
+  fs.mkdirSync(binDir, { recursive: true })
+
+  // Download URL for the specific binary
+  const downloadUrl = `${UV_RELEASE_BASE_URL}/${version}/${packageName}`
+  const tempdir = os.tmpdir()
+  const tempFilename = path.join(tempdir, packageName)
+
+  try {
+    console.log(`Downloading uv ${version} for ${platformKey}...`)
+    console.log(`URL: ${downloadUrl}`)
+
+    await downloadWithRedirects(downloadUrl, tempFilename)
+
+    console.log(`Extracting ${packageName} to ${binDir}...`)
+
+    // 根据文件扩展名选择解压方法
+    if (packageName.endsWith('.zip')) {
+      // 使用 adm-zip 处理 zip 文件
+      const zip = new AdmZip(tempFilename)
+      zip.extractAllTo(binDir, true)
+      fs.unlinkSync(tempFilename)
+      console.log(`Successfully installed uv ${version} for ${platform}-${arch}`)
+      return true
+    } else {
+      // tar.gz 文件的处理保持不变
+      await tar.x({
+        file: tempFilename,
+        cwd: tempdir,
+        z: true
+      })
+
+      // Move files using Node.js fs
+      const sourceDir = path.join(tempdir, packageName.split('.')[0])
+      const files = fs.readdirSync(sourceDir)
+      for (const file of files) {
+        const sourcePath = path.join(sourceDir, file)
+        const destPath = path.join(binDir, file)
+        fs.copyFileSync(sourcePath, destPath)
+        fs.unlinkSync(sourcePath)
+
+        // Set executable permissions for non-Windows platforms
+        if (platform !== 'win32') {
+          try {
+            fs.chmodSync(destPath, '755')
+          } catch (error) {
+            console.warn(`Warning: Failed to set executable permissions: ${error.message}`)
+          }
+        }
+      }
+
+      // Clean up
+      fs.unlinkSync(tempFilename)
+      fs.rmSync(sourceDir, { recursive: true })
+    }
+
+    console.log(`Successfully installed uv ${version} for ${platform}-${arch}`)
+    return true
+  } catch (error) {
+    console.error(`Error installing uv for ${platformKey}: ${error.message}`)
+
+    if (fs.existsSync(tempFilename)) {
+      fs.unlinkSync(tempFilename)
+    }
+
+    // Check if binDir is empty and remove it if so
+    try {
+      const files = fs.readdirSync(binDir)
+      if (files.length === 0) {
+        fs.rmSync(binDir, { recursive: true })
+        console.log(`Removed empty directory: ${binDir}`)
+      }
+    } catch (cleanupError) {
+      console.warn(`Warning: Failed to clean up directory: ${cleanupError.message}`)
+    }
+
+    return false
+  }
+}
+
+/**
+ * Detects current platform and architecture
+ */
+function detectPlatformAndArch() {
+  const platform = os.platform()
+  const arch = os.arch()
+  const isMusl = platform === 'linux' && detectIsMusl()
+
+  return { platform, arch, isMusl }
+}
+
+/**
+ * Attempts to detect if running on MUSL libc
+ */
+function detectIsMusl() {
+  try {
+    // Simple check for Alpine Linux which uses MUSL
+    const output = execSync('cat /etc/os-release').toString()
+    return output.toLowerCase().includes('alpine')
+  } catch (error) {
+    return false
+  }
+}
+
+/**
+ * Main function to install uv
+ */
+async function installUv() {
+  // Get the latest version if no specific version is provided
+  const version = DEFAULT_UV_VERSION
+  console.log(`Using uv version: ${version}`)
+
+  const { platform, arch, isMusl } = detectPlatformAndArch()
+
+  console.log(`Installing uv ${version} for ${platform}-${arch}${isMusl ? ' (MUSL)' : ''}...`)
+
+  await downloadUvBinary(platform, arch, version, isMusl)
+}
+
+// Run the installation
+installUv()
+  .then(() => {
+    console.log('Installation successful')
+    process.exit(0)
+  })
+  .catch((error) => {
+    console.error('Installation failed:', error)
+    process.exit(1)
+  })

scripts/after-pack.js

@@ -18,28 +18,48 @@ exports.default = async function (context) {
       'node_modules'
     )
 
-    removeDifferentArchNodeFiles(node_modules_path, '@libsql', arch === Arch.arm64 ? ['darwin-arm64'] : ['darwin-x64'])
+    keepPackageNodeFiles(node_modules_path, '@libsql', arch === Arch.arm64 ? ['darwin-arm64'] : ['darwin-x64'])
   }
 
   if (platform === 'linux') {
     const node_modules_path = path.join(context.appOutDir, 'resources', 'app.asar.unpacked', 'node_modules')
     const _arch = arch === Arch.arm64 ? ['linux-arm64-gnu', 'linux-arm64-musl'] : ['linux-x64-gnu', 'linux-x64-musl']
-    removeDifferentArchNodeFiles(node_modules_path, '@libsql', _arch)
+    keepPackageNodeFiles(node_modules_path, '@libsql', _arch)
   }
 
   if (platform === 'windows') {
     const node_modules_path = path.join(context.appOutDir, 'resources', 'app.asar.unpacked', 'node_modules')
-    removeDifferentArchNodeFiles(node_modules_path, '@libsql', ['win32-x64-msvc'])
+    if (arch === Arch.arm64) {
+      keepPackageNodeFiles(node_modules_path, '@strongtz', ['win32-arm64-msvc'])
+      keepPackageNodeFiles(node_modules_path, '@libsql', ['win32-arm64-msvc'])
+    }
+    if (arch === Arch.x64) {
+      keepPackageNodeFiles(node_modules_path, '@strongtz', ['win32-x64-msvc'])
+      keepPackageNodeFiles(node_modules_path, '@libsql', ['win32-x64-msvc'])
+    }
   }
 }
 
-function removeDifferentArchNodeFiles(nodeModulesPath, packageName, arch) {
+/**
+ * 使用指定架构的 node_modules 文件
+ * @param {*} nodeModulesPath
+ * @param {*} packageName
+ * @param {*} arch
+ * @returns
+ */
+function keepPackageNodeFiles(nodeModulesPath, packageName, arch) {
   const modulePath = path.join(nodeModulesPath, packageName)
+
+  if (!fs.existsSync(modulePath)) {
+    console.log(`[After Pack] Directory does not exist: ${modulePath}`)
+    return
+  }
+
   const dirs = fs.readdirSync(modulePath)
   dirs
     .filter((dir) => !arch.includes(dir))
     .forEach((dir) => {
       fs.rmSync(path.join(modulePath, dir), { recursive: true, force: true })
-      console.log(`Removed dir: ${dir}`, arch)
+      console.log(`[After Pack] Removed dir: ${dir}`, arch)
     })
 }

scripts/build-npm.js

@@ -33,6 +33,10 @@ async function downloadNpm(platform) {
       '@libsql/win32-x64-msvc',
       'https://registry.npmjs.org/@libsql/win32-x64-msvc/-/win32-x64-msvc-0.4.7.tgz'
     )
+    downloadNpmPackage(
+      '@strongtz/win32-arm64-msvc',
+      'https://registry.npmjs.org/@strongtz/win32-arm64-msvc/-/win32-arm64-msvc-0.4.7.tgz'
+    )
   }
 }
 

scripts/update-i18n.ts

@@ -0,0 +1,130 @@
+/**
+ * OCOOL_API_KEY=sk-abcxxxxxxxxxxxxxxxxxxxxxxx123 ts-node scripts/update-i18n.ts
+ */
+
+// OCOOL API KEY
+const OCOOL_API_KEY = process.env.OCOOL_API_KEY
+
+const INDEX = [
+  //         语言的名称          代码            用来翻译的模型
+  { name: 'France', code: 'fr-fr', model: 'qwen2.5-32b-instruct' },
+  { name: 'Spanish', code: 'es-es', model: 'qwen2.5-32b-instruct' },
+  { name: 'Portuguese', code: 'pt-pt', model: 'qwen2.5-72b-instruct' },
+  { name: 'Greek', code: 'el-gr', model: 'qwen-turbo' }
+]
+
+const fs = require('fs')
+import OpenAI from 'openai'
+
+const zh = JSON.parse(fs.readFileSync('src/renderer/src/i18n/locales/zh-cn.json', 'utf8')) as object
+
+const openai = new OpenAI({
+  apiKey: OCOOL_API_KEY,
+  baseURL: 'https://one.ocoolai.com/v1'
+})
+
+// 递归遍历翻译
+async function translate(zh: object, obj: object, target: string, model: string, updateFile) {
+  const texts: { [key: string]: string } = {}
+  for (const e in zh) {
+    if (typeof zh[e] == 'object') {
+      // 遍历下一层
+      if (!obj[e] || typeof obj[e] != 'object') obj[e] = {}
+      await translate(zh[e], obj[e], target, model, updateFile)
+    } else {
+      // 加入到本层待翻译列表
+      if (!obj[e] || typeof obj[e] != 'string') {
+        texts[e] = zh[e]
+      }
+    }
+  }
+  if (Object.keys(texts).length > 0) {
+    const completion = await openai.chat.completions.create({
+      model: model,
+      response_format: { type: 'json_object' },
+      messages: [
+        {
+          role: 'user',
+          content: `
+You are a robot specifically designed for translation tasks. As a model that has been extensively fine-tuned on Russian language corpora, you are proficient in using the Russian language.
+Now, please output the translation based on the input content. The input will include both Chinese and English key values, and you should output the corresponding key values in the Russian language.
+When translating, ensure that no key value is omitted, and maintain the accuracy and fluency of the translation. Pay attention to the capitalization rules in the output to match the source text, and especially pay attention to whether to capitalize the first letter of each word except for prepositions. For strings containing \`{{value}}\`, ensure that the format is not disrupted.
+Output in JSON.
+######################################################
+INPUT
+######################################################
+${JSON.stringify({
+  confirm: '确定要备份数据吗？',
+  select_model: '选择模型',
+  title: '文件',
+  deeply_thought: '已深度思考（用时 {{secounds}} 秒）'
+})}
+######################################################
+MAKE SURE TO OUTPUT IN Russian. DO NOT OUTPUT IN UNSPECIFIED LANGUAGE.
+######################################################
+                `
+        },
+        {
+          role: 'assistant',
+          content: JSON.stringify({
+            confirm: 'Подтвердите резервное копирование данных?',
+            select_model: 'Выберите Модель',
+            title: 'Файл',
+            deeply_thought: 'Глубоко продумано (заняло {{seconds}} секунд)'
+          })
+        },
+        {
+          role: 'user',
+          content: `
+You are a robot specifically designed for translation tasks. As a model that has been extensively fine-tuned on ${target} language corpora, you are proficient in using the ${target} language.
+Now, please output the translation based on the input content. The input will include both Chinese and English key values, and you should output the corresponding key values in the ${target} language.
+When translating, ensure that no key value is omitted, and maintain the accuracy and fluency of the translation. Pay attention to the capitalization rules in the output to match the source text, and especially pay attention to whether to capitalize the first letter of each word except for prepositions. For strings containing \`{{value}}\`, ensure that the format is not disrupted.
+Output in JSON.
+######################################################
+INPUT
+######################################################
+${JSON.stringify(texts)}
+######################################################
+MAKE SURE TO OUTPUT IN ${target}. DO NOT OUTPUT IN UNSPECIFIED LANGUAGE.
+######################################################
+                `
+        }
+      ]
+    })
+    // 添加翻译后的键值，并打印错译漏译内容
+    try {
+      const result = JSON.parse(completion.choices[0].message.content!)
+      for (const e in texts) {
+        if (result[e] && typeof result[e] === 'string') {
+          obj[e] = result[e]
+        } else {
+          console.log('[warning]', `missing value "${e}" in ${target} translation`)
+        }
+      }
+    } catch (e) {
+      console.log('[error]', e)
+      for (const e in texts) {
+        console.log('[warning]', `missing value "${e}" in ${target} translation`)
+      }
+    }
+  }
+  // 删除多余的键值
+  for (const e in obj) {
+    if (!zh[e]) {
+      delete obj[e]
+    }
+  }
+  // 更新文件
+  updateFile()
+}
+
+;(async () => {
+  for (const { name, code, model } of INDEX) {
+    const obj = fs.existsSync(`src/renderer/src/i18n/translate/${code}.json`)
+      ? JSON.parse(fs.readFileSync(`src/renderer/src/i18n/translate/${code}.json`, 'utf8'))
+      : {}
+    await translate(zh, obj, name, model, () => {
+      fs.writeFileSync(`src/renderer/src/i18n/translate/${code}.json`, JSON.stringify(obj, null, 2), 'utf8')
+    })
+  }
+})()

scripts/utils.js

@@ -22,7 +22,8 @@ function downloadNpmPackage(packageName, url) {
     console.log(`Extracting ${filename}...`)
     execSync(`tar -xvf ${filename}`)
     execSync(`rm -rf ${filename}`)
-    execSync(`mv package ${targetDir}`)
+    execSync(`mkdir -p ${targetDir}`)
+    execSync(`mv package/* ${targetDir}/`)
   } catch (error) {
     console.error(`Error processing ${packageName}: ${error.message}`)
     if (fs.existsSync(filename)) {

src/main/config.ts

@@ -12,12 +12,12 @@ export const DATA_PATH = getDataPath()
 
 export const titleBarOverlayDark = {
   height: 40,
-  color: '#00000000',
+  color: 'rgba(0,0,0,0)',
   symbolColor: '#ffffff'
 }
 
 export const titleBarOverlayLight = {
   height: 40,
-  color: '#00000000',
+  color: 'rgba(255,255,255,0)',
   symbolColor: '#000000'
 }

src/main/constant.ts

@@ -1,3 +1,4 @@
 export const isMac = process.platform === 'darwin'
 export const isWin = process.platform === 'win32'
 export const isLinux = process.platform === 'linux'
+export const isDev = process.env.NODE_ENV === 'development'

src/main/embeddings/Embeddings.ts

@@ -0,0 +1,24 @@
+import type { BaseEmbeddings } from '@cherrystudio/embedjs-interfaces'
+import { KnowledgeBaseParams } from '@types'
+
+import EmbeddingsFactory from './EmbeddingsFactory'
+
+export default class Embeddings {
+  private sdk: BaseEmbeddings
+  constructor({ model, apiKey, apiVersion, baseURL, dimensions }: KnowledgeBaseParams) {
+    this.sdk = EmbeddingsFactory.create({ model, apiKey, apiVersion, baseURL, dimensions } as KnowledgeBaseParams)
+  }
+  public async init(): Promise<void> {
+    return this.sdk.init()
+  }
+  public async getDimensions(): Promise<number> {
+    return this.sdk.getDimensions()
+  }
+  public async embedDocuments(texts: string[]): Promise<number[][]> {
+    return this.sdk.embedDocuments(texts)
+  }
+
+  public async embedQuery(text: string): Promise<number[]> {
+    return this.sdk.embedQuery(text)
+  }
+}

src/main/embeddings/EmbeddingsFactory.ts

@@ -0,0 +1,38 @@
+import type { BaseEmbeddings } from '@cherrystudio/embedjs-interfaces'
+import { OpenAiEmbeddings } from '@cherrystudio/embedjs-openai'
+import { AzureOpenAiEmbeddings } from '@cherrystudio/embedjs-openai/src/azure-openai-embeddings'
+import { getInstanceName } from '@main/utils'
+import { KnowledgeBaseParams } from '@types'
+
+import VoyageEmbeddings from './VoyageEmbeddings'
+
+export default class EmbeddingsFactory {
+  static create({ model, apiKey, apiVersion, baseURL, dimensions }: KnowledgeBaseParams): BaseEmbeddings {
+    const batchSize = 10
+    if (model.includes('voyage')) {
+      return new VoyageEmbeddings({
+        modelName: model,
+        apiKey,
+        outputDimension: dimensions,
+        batchSize: 8
+      })
+    }
+    if (apiVersion !== undefined) {
+      return new AzureOpenAiEmbeddings({
+        azureOpenAIApiKey: apiKey,
+        azureOpenAIApiVersion: apiVersion,
+        azureOpenAIApiDeploymentName: model,
+        azureOpenAIApiInstanceName: getInstanceName(baseURL),
+        dimensions,
+        batchSize
+      })
+    }
+    return new OpenAiEmbeddings({
+      model,
+      apiKey,
+      dimensions,
+      batchSize,
+      configuration: { baseURL }
+    })
+  }
+}

src/main/embeddings/VoyageEmbeddings.ts

@@ -0,0 +1,30 @@
+import { BaseEmbeddings } from '@cherrystudio/embedjs-interfaces'
+import { VoyageEmbeddings as _VoyageEmbeddings } from '@langchain/community/embeddings/voyage'
+
+export default class VoyageEmbeddings extends BaseEmbeddings {
+  private model: _VoyageEmbeddings
+  constructor(private readonly configuration?: ConstructorParameters<typeof _VoyageEmbeddings>[0]) {
+    super()
+    if (!this.configuration) this.configuration = {}
+    if (!this.configuration.modelName) this.configuration.modelName = 'voyage-3'
+
+    if (!this.configuration.outputDimension) {
+      throw new Error('You need to pass in the optional dimensions parameter for this model')
+    }
+    this.model = new _VoyageEmbeddings(this.configuration)
+  }
+  override async getDimensions(): Promise<number> {
+    if (!this.configuration?.outputDimension) {
+      throw new Error('You need to pass in the optional dimensions parameter for this model')
+    }
+    return this.configuration?.outputDimension
+  }
+
+  override async embedDocuments(texts: string[]): Promise<number[][]> {
+    return this.model.embedDocuments(texts)
+  }
+
+  override async embedQuery(text: string): Promise<number[]> {
+    return this.model.embedQuery(text)
+  }
+}

src/main/index.ts

@@ -1,12 +1,19 @@
+import './services/MemoryFileService'
+
 import { electronApp, optimizer } from '@electron-toolkit/utils'
-import { app } from 'electron'
-import installExtension, { REDUX_DEVTOOLS } from 'electron-devtools-installer'
+import { replaceDevtoolsFont } from '@main/utils/windowUtil'
+import { IpcChannel } from '@shared/IpcChannel'
+import { app, ipcMain } from 'electron'
+import installExtension, { REACT_DEVELOPER_TOOLS, REDUX_DEVTOOLS } from 'electron-devtools-installer'
+import Logger from 'electron-log'
 
 import { registerIpc } from './ipc'
+import { configManager } from './services/ConfigManager'
+import mcpService from './services/MCPService'
+import { CHERRY_STUDIO_PROTOCOL, handleProtocolUrl, registerProtocolClient } from './services/ProtocolClient'
 import { registerShortcuts } from './services/ShortcutService'
 import { TrayService } from './services/TrayService'
 import { windowService } from './services/WindowService'
-import { updateUserDataPath } from './utils/upgrade'
 
 // Check for single instance lock
 if (!app.requestSingleInstanceLock()) {
@@ -18,30 +25,15 @@ if (!app.requestSingleInstanceLock()) {
   // Some APIs can only be used after this event occurs.
 
   app.whenReady().then(async () => {
-    await updateUserDataPath()
-
-    // Register custom protocol
-    if (!app.isDefaultProtocolClient('cherrystudio')) {
-      app.setAsDefaultProtocolClient('cherrystudio')
-    }
-
-    // Handle protocol open
-    app.on('open-url', (event, url) => {
-      event.preventDefault()
-      const parsedUrl = new URL(url)
-      if (parsedUrl.pathname === 'siliconflow.oauth.login') {
-        const code = parsedUrl.searchParams.get('code')
-        if (code) {
-          // Handle the OAuth code here
-          console.log('OAuth code received:', code)
-          // You can send this code to your renderer process via IPC if needed
-        }
-      }
-    })
-
     // Set app user model id for windows
     electronApp.setAppUserModelId(import.meta.env.VITE_MAIN_BUNDLE_ID || 'com.kangfenmao.CherryStudio')
 
+    // Mac: Hide dock icon before window creation when launch to tray is set
+    const isLaunchToTray = configManager.getLaunchToTray()
+    if (isLaunchToTray) {
+      app.dock?.hide()
+    }
+
     const mainWindow = windowService.createMainWindow()
     new TrayService()
 
@@ -53,20 +45,47 @@ if (!app.requestSingleInstanceLock()) {
         windowService.showMainWindow()
       }
     })
+
     registerShortcuts(mainWindow)
 
     registerIpc(mainWindow, app)
 
+    replaceDevtoolsFont(mainWindow)
+
     if (process.env.NODE_ENV === 'development') {
-      installExtension(REDUX_DEVTOOLS)
+      installExtension([REDUX_DEVTOOLS, REACT_DEVELOPER_TOOLS])
         .then((name) => console.log(`Added Extension:  ${name}`))
         .catch((err) => console.log('An error occurred: ', err))
     }
+    ipcMain.handle(IpcChannel.System_GetDeviceType, () => {
+      return process.platform === 'darwin' ? 'mac' : process.platform === 'win32' ? 'windows' : 'linux'
+    })
+  })
+
+  registerProtocolClient(app)
+
+  // macOS specific: handle protocol when app is already running
+  app.on('open-url', (event, url) => {
+    event.preventDefault()
+    handleProtocolUrl(url)
+  })
+
+  registerProtocolClient(app)
+
+  // macOS specific: handle protocol when app is already running
+  app.on('open-url', (event, url) => {
+    event.preventDefault()
+    handleProtocolUrl(url)
   })
 
   // Listen for second instance
-  app.on('second-instance', () => {
+  app.on('second-instance', (_event, argv) => {
     windowService.showMainWindow()
+
+    // Protocol handler for Windows/Linux
+    // The commandLine is an array of strings where the last item might be the URL
+    const url = argv.find((arg) => arg.startsWith(CHERRY_STUDIO_PROTOCOL + '://'))
+    if (url) handleProtocolUrl(url)
   })
 
   app.on('browser-window-created', (_, window) => {
@@ -77,6 +96,15 @@ if (!app.requestSingleInstanceLock()) {
     app.isQuitting = true
   })
 
+  app.on('will-quit', async () => {
+    // event.preventDefault()
+    try {
+      await mcpService.cleanup()
+    } catch (error) {
+      Logger.error('Error cleaning up MCP service:', error)
+    }
+  })
+
   // In this file you can include the rest of your app"s specific main process
   // code. You can also put them in separate files and require them here.
 }

src/main/integration/nutstore/sso/lib/index.d.ts

@@ -0,0 +1,8 @@
+declare function decrypt(app: string, s: string): string
+
+interface Secret {
+  app: string
+}
+declare function createOAuthUrl(secret: Secret): string
+
+export { type Secret, createOAuthUrl, decrypt }

src/main/ipc.ts

@@ -1,82 +1,138 @@
-import fs from 'node:fs'
-import path from 'node:path'
+import './services/MemoryFileService'
 
+import fs from 'node:fs'
+
+import { isMac, isWin } from '@main/constant'
+import { getBinaryPath, isBinaryExists, runInstallScript } from '@main/utils/process'
+import { IpcChannel } from '@shared/IpcChannel'
 import { Shortcut, ThemeMode } from '@types'
-import { BrowserWindow, ipcMain, ProxyConfig, session, shell } from 'electron'
+import { BrowserWindow, ipcMain, session, shell } from 'electron'
 import log from 'electron-log'
 
 import { titleBarOverlayDark, titleBarOverlayLight } from './config'
 import AppUpdater from './services/AppUpdater'
 import BackupManager from './services/BackupManager'
 import { configManager } from './services/ConfigManager'
+import CopilotService from './services/CopilotService'
 import { ExportService } from './services/ExportService'
 import FileService from './services/FileService'
 import FileStorage from './services/FileStorage'
 import { GeminiService } from './services/GeminiService'
 import KnowledgeService from './services/KnowledgeService'
+import mcpService from './services/MCPService'
+import { memoryFileService } from './services/MemoryFileService'
+import * as NutstoreService from './services/NutstoreService'
+import ObsidianVaultService from './services/ObsidianVaultService'
+import { ProxyConfig, proxyManager } from './services/ProxyManager'
+import { searchService } from './services/SearchService'
 import { registerShortcuts, unregisterAllShortcuts } from './services/ShortcutService'
 import { TrayService } from './services/TrayService'
 import { windowService } from './services/WindowService'
 import { getResourcePath } from './utils'
-import { decrypt } from './utils/aes'
-import { encrypt } from './utils/aes'
+import { decrypt, encrypt } from './utils/aes'
+import { getConfigDir, getFilesDir } from './utils/file'
 import { compress, decompress } from './utils/zip'
 
 const fileManager = new FileStorage()
 const backupManager = new BackupManager()
 const exportService = new ExportService(fileManager)
+const obsidianVaultService = new ObsidianVaultService()
 
 export function registerIpc(mainWindow: BrowserWindow, app: Electron.App) {
-  const { autoUpdater } = new AppUpdater(mainWindow)
+  const appUpdater = new AppUpdater(mainWindow)
 
-  ipcMain.handle('app:info', () => ({
+  ipcMain.handle(IpcChannel.App_Info, () => ({
     version: app.getVersion(),
     isPackaged: app.isPackaged,
     appPath: app.getAppPath(),
-    filesPath: path.join(app.getPath('userData'), 'Data', 'Files'),
+    filesPath: getFilesDir(),
+    configPath: getConfigDir(),
     appDataPath: app.getPath('userData'),
     resourcesPath: getResourcePath(),
     logsPath: log.transports.file.getFile().path
   }))
 
-  ipcMain.handle('app:proxy', async (_, proxy: string) => {
-    const sessions = [session.defaultSession, session.fromPartition('persist:webview')]
-    const proxyConfig: ProxyConfig = proxy === 'system' ? { mode: 'system' } : proxy ? { proxyRules: proxy } : {}
-    await Promise.all(sessions.map((session) => session.setProxy(proxyConfig)))
+  ipcMain.handle(IpcChannel.App_Proxy, async (_, proxy: string) => {
+    let proxyConfig: ProxyConfig
+
+    if (proxy === 'system') {
+      proxyConfig = { mode: 'system' }
+    } else if (proxy) {
+      proxyConfig = { mode: 'custom', url: proxy }
+    } else {
+      proxyConfig = { mode: 'none' }
+    }
+
+    await proxyManager.configureProxy(proxyConfig)
   })
 
-  ipcMain.handle('app:reload', () => mainWindow.reload())
-  ipcMain.handle('open:website', (_, url: string) => shell.openExternal(url))
+  ipcMain.handle(IpcChannel.App_Reload, () => mainWindow.reload())
+  ipcMain.handle(IpcChannel.Open_Website, (_, url: string) => shell.openExternal(url))
+
+  // Update
+  ipcMain.handle(IpcChannel.App_ShowUpdateDialog, () => appUpdater.showUpdateDialog(mainWindow))
 
   // language
-  ipcMain.handle('app:set-language', (_, language) => {
+  ipcMain.handle(IpcChannel.App_SetLanguage, (_, language) => {
     configManager.setLanguage(language)
   })
 
+  // launch on boot
+  ipcMain.handle(IpcChannel.App_SetLaunchOnBoot, (_, openAtLogin: boolean) => {
+    // Set login item settings for windows and mac
+    // linux is not supported because it requires more file operations
+    if (isWin || isMac) {
+      app.setLoginItemSettings({ openAtLogin })
+    }
+  })
+
+  // launch to tray
+  ipcMain.handle(IpcChannel.App_SetLaunchToTray, (_, isActive: boolean) => {
+    configManager.setLaunchToTray(isActive)
+  })
+
   // tray
-  ipcMain.handle('app:set-tray', (_, isActive: boolean) => {
+  ipcMain.handle(IpcChannel.App_SetTray, (_, isActive: boolean) => {
     configManager.setTray(isActive)
   })
 
-  ipcMain.handle('app:restart-tray', () => TrayService.getInstance().restartTray())
+  // to tray on close
+  ipcMain.handle(IpcChannel.App_SetTrayOnClose, (_, isActive: boolean) => {
+    configManager.setTrayOnClose(isActive)
+  })
 
-  ipcMain.handle('config:set', (_, key: string, value: any) => {
+  ipcMain.handle(IpcChannel.App_RestartTray, () => TrayService.getInstance().restartTray())
+
+  ipcMain.handle(IpcChannel.Config_Set, (_, key: string, value: any) => {
     configManager.set(key, value)
   })
 
-  ipcMain.handle('config:get', (_, key: string) => {
+  ipcMain.handle(IpcChannel.Config_Get, (_, key: string) => {
     return configManager.get(key)
   })
 
   // theme
-  ipcMain.handle('app:set-theme', (_, theme: ThemeMode) => {
+  ipcMain.handle(IpcChannel.App_SetTheme, (event, theme: ThemeMode) => {
+    if (theme === configManager.getTheme()) return
+
     configManager.setTheme(theme)
+
+    // should sync theme change to all windows
+    const senderWindowId = event.sender.id
+    const windows = BrowserWindow.getAllWindows()
+    // 向其他窗口广播主题变化
+    windows.forEach((win) => {
+      if (win.webContents.id !== senderWindowId) {
+        win.webContents.send(IpcChannel.ThemeChange, theme)
+      }
+    })
+
     mainWindow?.setTitleBarOverlay &&
       mainWindow.setTitleBarOverlay(theme === 'dark' ? titleBarOverlayDark : titleBarOverlayLight)
   })
 
   // clear cache
-  ipcMain.handle('app:clear-cache', async () => {
+  ipcMain.handle(IpcChannel.App_ClearCache, async () => {
     const sessions = [session.defaultSession, session.fromPartition('persist:webview')]
 
     try {
@@ -98,68 +154,59 @@ export function registerIpc(mainWindow: BrowserWindow, app: Electron.App) {
   })
 
   // check for update
-  ipcMain.handle('app:check-for-update', async () => {
-    const update = await autoUpdater.checkForUpdates()
+  ipcMain.handle(IpcChannel.App_CheckForUpdate, async () => {
+    const update = await appUpdater.autoUpdater.checkForUpdates()
     return {
-      currentVersion: autoUpdater.currentVersion,
+      currentVersion: appUpdater.autoUpdater.currentVersion,
       updateInfo: update?.updateInfo
     }
   })
 
   // zip
-  ipcMain.handle('zip:compress', (_, text: string) => compress(text))
-  ipcMain.handle('zip:decompress', (_, text: Buffer) => decompress(text))
+  ipcMain.handle(IpcChannel.Zip_Compress, (_, text: string) => compress(text))
+  ipcMain.handle(IpcChannel.Zip_Decompress, (_, text: Buffer) => decompress(text))
 
   // backup
-  ipcMain.handle('backup:backup', backupManager.backup)
-  ipcMain.handle('backup:restore', backupManager.restore)
-  ipcMain.handle('backup:backupToWebdav', backupManager.backupToWebdav)
-  ipcMain.handle('backup:restoreFromWebdav', backupManager.restoreFromWebdav)
+  ipcMain.handle(IpcChannel.Backup_Backup, backupManager.backup)
+  ipcMain.handle(IpcChannel.Backup_Restore, backupManager.restore)
+  ipcMain.handle(IpcChannel.Backup_BackupToWebdav, backupManager.backupToWebdav)
+  ipcMain.handle(IpcChannel.Backup_RestoreFromWebdav, backupManager.restoreFromWebdav)
+  ipcMain.handle(IpcChannel.Backup_ListWebdavFiles, backupManager.listWebdavFiles)
+  ipcMain.handle(IpcChannel.Backup_CheckConnection, backupManager.checkConnection)
+  ipcMain.handle(IpcChannel.Backup_CreateDirectory, backupManager.createDirectory)
 
   // file
-  ipcMain.handle('file:open', fileManager.open)
-  ipcMain.handle('file:openPath', fileManager.openPath)
-  ipcMain.handle('file:save', fileManager.save)
-  ipcMain.handle('file:select', fileManager.selectFile)
-  ipcMain.handle('file:upload', fileManager.uploadFile)
-  ipcMain.handle('file:clear', fileManager.clear)
-  ipcMain.handle('file:read', fileManager.readFile)
-  ipcMain.handle('file:delete', fileManager.deleteFile)
-  ipcMain.handle('file:get', fileManager.getFile)
-  ipcMain.handle('file:selectFolder', fileManager.selectFolder)
-  ipcMain.handle('file:create', fileManager.createTempFile)
-  ipcMain.handle('file:write', fileManager.writeFile)
-  ipcMain.handle('file:saveImage', fileManager.saveImage)
-  ipcMain.handle('file:base64Image', fileManager.base64Image)
-  ipcMain.handle('file:download', fileManager.downloadFile)
-  ipcMain.handle('file:copy', fileManager.copyFile)
-  ipcMain.handle('file:binaryFile', fileManager.binaryFile)
+  ipcMain.handle(IpcChannel.File_Open, fileManager.open)
+  ipcMain.handle(IpcChannel.File_OpenPath, fileManager.openPath)
+  ipcMain.handle(IpcChannel.File_Save, fileManager.save)
+  ipcMain.handle(IpcChannel.File_Select, fileManager.selectFile)
+  ipcMain.handle(IpcChannel.File_Upload, fileManager.uploadFile)
+  ipcMain.handle(IpcChannel.File_Clear, fileManager.clear)
+  ipcMain.handle(IpcChannel.File_Read, fileManager.readFile)
+  ipcMain.handle(IpcChannel.File_Delete, fileManager.deleteFile)
+  ipcMain.handle(IpcChannel.File_Get, fileManager.getFile)
+  ipcMain.handle(IpcChannel.File_SelectFolder, fileManager.selectFolder)
+  ipcMain.handle(IpcChannel.File_Create, fileManager.createTempFile)
+  ipcMain.handle(IpcChannel.File_Write, fileManager.writeFile)
+  ipcMain.handle(IpcChannel.File_SaveImage, fileManager.saveImage)
+  ipcMain.handle(IpcChannel.File_Base64Image, fileManager.base64Image)
+  ipcMain.handle(IpcChannel.File_Download, fileManager.downloadFile)
+  ipcMain.handle(IpcChannel.File_Copy, fileManager.copyFile)
+  ipcMain.handle(IpcChannel.File_BinaryFile, fileManager.binaryFile)
 
   // fs
-  ipcMain.handle('fs:read', FileService.readFile)
-
-  // minapp
-  ipcMain.handle('minapp', (_, args) => {
-    windowService.createMinappWindow({
-      url: args.url,
-      parent: mainWindow,
-      windowOptions: {
-        ...mainWindow.getBounds(),
-        ...args.windowOptions
-      }
-    })
-  })
+  ipcMain.handle(IpcChannel.Fs_Read, FileService.readFile)
 
   // export
-  ipcMain.handle('export:word', exportService.exportToWord)
+  ipcMain.handle(IpcChannel.Export_Word, exportService.exportToWord)
 
   // open path
-  ipcMain.handle('open:path', async (_, path: string) => {
+  ipcMain.handle(IpcChannel.Open_Path, async (_, path: string) => {
     await shell.openPath(path)
   })
 
   // shortcuts
-  ipcMain.handle('shortcuts:update', (_, shortcuts: Shortcut[]) => {
+  ipcMain.handle(IpcChannel.Shortcuts_Update, (_, shortcuts: Shortcut[]) => {
     configManager.setShortcuts(shortcuts)
     // Refresh shortcuts registration
     if (mainWindow) {
@@ -169,19 +216,20 @@ export function registerIpc(mainWindow: BrowserWindow, app: Electron.App) {
   })
 
   // knowledge base
-  ipcMain.handle('knowledge-base:create', KnowledgeService.create)
-  ipcMain.handle('knowledge-base:reset', KnowledgeService.reset)
-  ipcMain.handle('knowledge-base:delete', KnowledgeService.delete)
-  ipcMain.handle('knowledge-base:add', KnowledgeService.add)
-  ipcMain.handle('knowledge-base:remove', KnowledgeService.remove)
-  ipcMain.handle('knowledge-base:search', KnowledgeService.search)
+  ipcMain.handle(IpcChannel.KnowledgeBase_Create, KnowledgeService.create)
+  ipcMain.handle(IpcChannel.KnowledgeBase_Reset, KnowledgeService.reset)
+  ipcMain.handle(IpcChannel.KnowledgeBase_Delete, KnowledgeService.delete)
+  ipcMain.handle(IpcChannel.KnowledgeBase_Add, KnowledgeService.add)
+  ipcMain.handle(IpcChannel.KnowledgeBase_Remove, KnowledgeService.remove)
+  ipcMain.handle(IpcChannel.KnowledgeBase_Search, KnowledgeService.search)
+  ipcMain.handle(IpcChannel.KnowledgeBase_Rerank, KnowledgeService.rerank)
 
   // window
-  ipcMain.handle('window:set-minimum-size', (_, width: number, height: number) => {
+  ipcMain.handle(IpcChannel.Windows_SetMinimumSize, (_, width: number, height: number) => {
     mainWindow?.setMinimumSize(width, height)
   })
 
-  ipcMain.handle('window:reset-minimum-size', () => {
+  ipcMain.handle(IpcChannel.Windows_ResetMinimumSize, () => {
     mainWindow?.setMinimumSize(1080, 600)
     const [width, height] = mainWindow?.getSize() ?? [1080, 600]
     if (width < 1080) {
@@ -190,21 +238,91 @@ export function registerIpc(mainWindow: BrowserWindow, app: Electron.App) {
   })
 
   // gemini
-  ipcMain.handle('gemini:upload-file', GeminiService.uploadFile)
-  ipcMain.handle('gemini:base64-file', GeminiService.base64File)
-  ipcMain.handle('gemini:retrieve-file', GeminiService.retrieveFile)
-  ipcMain.handle('gemini:list-files', GeminiService.listFiles)
-  ipcMain.handle('gemini:delete-file', GeminiService.deleteFile)
+  ipcMain.handle(IpcChannel.Gemini_UploadFile, GeminiService.uploadFile)
+  ipcMain.handle(IpcChannel.Gemini_Base64File, GeminiService.base64File)
+  ipcMain.handle(IpcChannel.Gemini_RetrieveFile, GeminiService.retrieveFile)
+  ipcMain.handle(IpcChannel.Gemini_ListFiles, GeminiService.listFiles)
+  ipcMain.handle(IpcChannel.Gemini_DeleteFile, GeminiService.deleteFile)
 
   // mini window
-  ipcMain.handle('miniwindow:show', () => windowService.showMiniWindow())
-  ipcMain.handle('miniwindow:hide', () => windowService.hideMiniWindow())
-  ipcMain.handle('miniwindow:close', () => windowService.closeMiniWindow())
-  ipcMain.handle('miniwindow:toggle', () => windowService.toggleMiniWindow())
+  ipcMain.handle(IpcChannel.MiniWindow_Show, () => windowService.showMiniWindow())
+  ipcMain.handle(IpcChannel.MiniWindow_Hide, () => windowService.hideMiniWindow())
+  ipcMain.handle(IpcChannel.MiniWindow_Close, () => windowService.closeMiniWindow())
+  ipcMain.handle(IpcChannel.MiniWindow_Toggle, () => windowService.toggleMiniWindow())
+  ipcMain.handle(IpcChannel.MiniWindow_SetPin, (_, isPinned) => windowService.setPinMiniWindow(isPinned))
 
   // aes
-  ipcMain.handle('aes:encrypt', (_, text: string, secretKey: string, iv: string) => encrypt(text, secretKey, iv))
-  ipcMain.handle('aes:decrypt', (_, encryptedData: string, iv: string, secretKey: string) =>
+  ipcMain.handle(IpcChannel.Aes_Encrypt, (_, text: string, secretKey: string, iv: string) =>
+    encrypt(text, secretKey, iv)
+  )
+  ipcMain.handle(IpcChannel.Aes_Decrypt, (_, encryptedData: string, iv: string, secretKey: string) =>
     decrypt(encryptedData, iv, secretKey)
   )
+
+  // Register MCP handlers
+  ipcMain.handle(IpcChannel.Mcp_RemoveServer, mcpService.removeServer)
+  ipcMain.handle(IpcChannel.Mcp_RestartServer, mcpService.restartServer)
+  ipcMain.handle(IpcChannel.Mcp_StopServer, mcpService.stopServer)
+  ipcMain.handle(IpcChannel.Mcp_ListTools, mcpService.listTools)
+  ipcMain.handle(IpcChannel.Mcp_CallTool, mcpService.callTool)
+  ipcMain.handle(IpcChannel.Mcp_ListPrompts, mcpService.listPrompts)
+  ipcMain.handle(IpcChannel.Mcp_GetPrompt, mcpService.getPrompt)
+  ipcMain.handle(IpcChannel.Mcp_GetInstallInfo, mcpService.getInstallInfo)
+
+  ipcMain.handle(IpcChannel.App_IsBinaryExist, (_, name: string) => isBinaryExists(name))
+  ipcMain.handle(IpcChannel.App_GetBinaryPath, (_, name: string) => getBinaryPath(name))
+  ipcMain.handle(IpcChannel.App_InstallUvBinary, () => runInstallScript('install-uv.js'))
+  ipcMain.handle(IpcChannel.App_InstallBunBinary, () => runInstallScript('install-bun.js'))
+
+  //copilot
+  ipcMain.handle(IpcChannel.Copilot_GetAuthMessage, CopilotService.getAuthMessage)
+  ipcMain.handle(IpcChannel.Copilot_GetCopilotToken, CopilotService.getCopilotToken)
+  ipcMain.handle(IpcChannel.Copilot_SaveCopilotToken, CopilotService.saveCopilotToken)
+  ipcMain.handle(IpcChannel.Copilot_GetToken, CopilotService.getToken)
+  ipcMain.handle(IpcChannel.Copilot_Logout, CopilotService.logout)
+  ipcMain.handle(IpcChannel.Copilot_GetUser, CopilotService.getUser)
+
+  // Obsidian service
+  ipcMain.handle(IpcChannel.Obsidian_GetVaults, () => {
+    return obsidianVaultService.getVaults()
+  })
+
+  ipcMain.handle(IpcChannel.Obsidian_GetFiles, (_event, vaultName) => {
+    return obsidianVaultService.getFilesByVaultName(vaultName)
+  })
+
+  // nutstore
+  ipcMain.handle(IpcChannel.Nutstore_GetSsoUrl, NutstoreService.getNutstoreSSOUrl)
+  ipcMain.handle(IpcChannel.Nutstore_DecryptToken, (_, token: string) => NutstoreService.decryptToken(token))
+  ipcMain.handle(IpcChannel.Nutstore_GetDirectoryContents, (_, token: string, path: string) =>
+    NutstoreService.getDirectoryContents(token, path)
+  )
+
+  // search window
+  ipcMain.handle(IpcChannel.SearchWindow_Open, async (_, uid: string) => {
+    await searchService.openSearchWindow(uid)
+  })
+  ipcMain.handle(IpcChannel.SearchWindow_Close, async (_, uid: string) => {
+    await searchService.closeSearchWindow(uid)
+  })
+  ipcMain.handle(IpcChannel.SearchWindow_OpenUrl, async (_, uid: string, url: string) => {
+    return await searchService.openUrlInSearchWindow(uid, url)
+  })
+
+  // memory
+  ipcMain.handle(IpcChannel.Memory_LoadData, async () => {
+    return await memoryFileService.loadData()
+  })
+  ipcMain.handle(IpcChannel.Memory_SaveData, async (_, data, forceOverwrite = false) => {
+    return await memoryFileService.saveData(data, forceOverwrite)
+  })
+  ipcMain.handle(IpcChannel.Memory_DeleteShortMemoryById, async (_, id) => {
+    return await memoryFileService.deleteShortMemoryById(id)
+  })
+  ipcMain.handle(IpcChannel.LongTermMemory_LoadData, async () => {
+    return await memoryFileService.loadLongTermData()
+  })
+  ipcMain.handle(IpcChannel.LongTermMemory_SaveData, async (_, data, forceOverwrite = false) => {
+    return await memoryFileService.saveLongTermData(data, forceOverwrite)
+  })
 }

src/main/loader/draftsExportLoader.ts

@@ -1,6 +1,6 @@
 import * as fs from 'node:fs'
 
-import { JsonLoader } from '@llm-tools/embedjs'
+import { JsonLoader } from '@cherrystudio/embedjs'
 
 /**
  * Drafts 应用导出的笔记文件加载器

src/main/loader/epubLoader.ts

@@ -1,9 +1,11 @@
+import { BaseLoader } from '@cherrystudio/embedjs-interfaces'
+import { cleanString } from '@cherrystudio/embedjs-utils'
 import { RecursiveCharacterTextSplitter } from '@langchain/textsplitters'
-import { BaseLoader } from '@llm-tools/embedjs-interfaces'
-import { cleanString } from '@llm-tools/embedjs-utils'
+import { getTempDir } from '@main/utils/file'
 import Logger from 'electron-log'
 import EPub from 'epub'
 import * as fs from 'fs'
+import path from 'path'
 
 /**
  * epub 加载器的配置选项
@@ -157,7 +159,9 @@ export class EpubLoader extends BaseLoader<Record<string, string | number | bool
         throw new Error('No content found in epub file')
       }
 
-      const chapterTexts: string[] = []
+      // 使用临时文件而不是内存数组
+      const tempFilePath = path.join(getTempDir(), `epub-${Date.now()}.txt`)
+      const writeStream = fs.createWriteStream(tempFilePath)
 
       // 遍历所有章节
       for (const chapter of chapters) {
@@ -175,15 +179,31 @@ export class EpubLoader extends BaseLoader<Record<string, string | number | bool
             .trim() // 移除首尾空白
 
           if (text) {
-            chapterTexts.push(text)
+            // 直接写入文件
+            writeStream.write(text + '\n\n')
           }
         } catch (error) {
           Logger.error(`[EpubLoader] Error processing chapter ${chapter.id}:`, error)
         }
       }
 
-      // 使用双换行符连接所有章节文本
-      this.extractedText = chapterTexts.join('\n\n')
+      // 关闭写入流
+      writeStream.end()
+
+      // 等待写入完成
+      await new Promise<void>((resolve, reject) => {
+        writeStream.on('finish', resolve)
+        writeStream.on('error', reject)
+      })
+
+      // 从临时文件读取内容
+      this.extractedText = fs.readFileSync(tempFilePath, 'utf-8')
+
+      // 删除临时文件
+      fs.unlinkSync(tempFilePath)
+
+      // 只添加一条完成日志
+      Logger.info(`[EpubLoader] 电子书 ${this.metadata?.title || path.basename(this.filePath)} 处理完成`)
     } catch (error) {
       Logger.error('[EpubLoader] Error in extractTextFromEpub:', error)
       throw error

src/main/loader/index.ts

@@ -1,8 +1,8 @@
 import * as fs from 'node:fs'
 
-import { JsonLoader, LocalPathLoader, RAGApplication, TextLoader } from '@llm-tools/embedjs'
-import type { AddLoaderReturn } from '@llm-tools/embedjs-interfaces'
-import { WebLoader } from '@llm-tools/embedjs-loader-web'
+import { JsonLoader, LocalPathLoader, RAGApplication, TextLoader } from '@cherrystudio/embedjs'
+import type { AddLoaderReturn } from '@cherrystudio/embedjs-interfaces'
+import { WebLoader } from '@cherrystudio/embedjs-loader-web'
 import { LoaderReturn } from '@shared/config/types'
 import { FileType, KnowledgeBaseParams } from '@types'
 import Logger from 'electron-log'
@@ -11,8 +11,30 @@ import { DraftsExportLoader } from './draftsExportLoader'
 import { EpubLoader } from './epubLoader'
 import { OdLoader, OdType } from './odLoader'
 
-// embedjs内置loader类型
-const commonExts = ['.pdf', '.csv', '.docx', '.pptx', '.xlsx', '.md']
+// 文件扩展名到加载器类型的映射
+const FILE_LOADER_MAP: Record<string, string> = {
+  // 内置类型
+  '.pdf': 'common',
+  '.csv': 'common',
+  '.docx': 'common',
+  '.pptx': 'common',
+  '.xlsx': 'common',
+  '.md': 'common',
+  // OD类型
+  '.odt': 'od',
+  '.ods': 'od',
+  '.odp': 'od',
+  // epub类型
+  '.epub': 'epub',
+  // Drafts类型
+  '.draftsexport': 'drafts',
+  // HTML类型
+  '.html': 'html',
+  '.htm': 'html',
+  // JSON类型
+  '.json': 'json'
+  // 其他类型默认为文本类型
+}
 
 export async function addOdLoader(
   ragApplication: RAGApplication,
@@ -46,110 +68,87 @@ export async function addFileLoader(
   base: KnowledgeBaseParams,
   forceReload: boolean
 ): Promise<LoaderReturn> {
-  // 内置类型
-  if (commonExts.includes(file.ext)) {
-    const loaderReturn = await ragApplication.addLoader(
-      // @ts-ignore LocalPathLoader
-      new LocalPathLoader({ path: file.path, chunkSize: base.chunkSize, chunkOverlap: base.chunkOverlap }) as any,
-      forceReload
-    )
-    return {
-      entriesAdded: loaderReturn.entriesAdded,
-      uniqueId: loaderReturn.uniqueId,
-      uniqueIds: [loaderReturn.uniqueId],
-      loaderType: loaderReturn.loaderType
-    } as LoaderReturn
-  }
+  // 获取文件类型，如果没有匹配则默认为文本类型
+  const loaderType = FILE_LOADER_MAP[file.ext.toLowerCase()] || 'text'
+  let loaderReturn: AddLoaderReturn
 
-  // 自定义类型
-  if (['.odt', '.ods', '.odp'].includes(file.ext)) {
-    const loaderReturn = await addOdLoader(ragApplication, file, base, forceReload)
-    return {
-      entriesAdded: loaderReturn.entriesAdded,
-      uniqueId: loaderReturn.uniqueId,
-      uniqueIds: [loaderReturn.uniqueId],
-      loaderType: loaderReturn.loaderType
-    } as LoaderReturn
-  }
+  // JSON类型处理
+  let jsonObject = {}
+  let jsonParsed = true
+  Logger.info(`[KnowledgeBase] processing file ${file.path} as ${loaderType} type`)
+  switch (loaderType) {
+    case 'common':
+      // 内置类型处理
+      loaderReturn = await ragApplication.addLoader(
+        new LocalPathLoader({
+          path: file.path,
+          chunkSize: base.chunkSize,
+          chunkOverlap: base.chunkOverlap
+        }) as any,
+        forceReload
+      )
+      break
 
-  // epub 文件处理
-  if (file.ext === '.epub') {
-    const loaderReturn = await ragApplication.addLoader(
-      new EpubLoader({
-        filePath: file.path,
-        chunkSize: base.chunkSize ?? 1000,
-        chunkOverlap: base.chunkOverlap ?? 200
-      }) as any,
-      forceReload
-    )
-    return {
-      entriesAdded: loaderReturn.entriesAdded,
-      uniqueId: loaderReturn.uniqueId,
-      uniqueIds: [loaderReturn.uniqueId],
-      loaderType: loaderReturn.loaderType
-    } as LoaderReturn
-  }
+    case 'od':
+      // OD类型处理
+      loaderReturn = await addOdLoader(ragApplication, file, base, forceReload)
+      break
+    case 'epub':
+      // epub类型处理
+      loaderReturn = await ragApplication.addLoader(
+        new EpubLoader({
+          filePath: file.path,
+          chunkSize: base.chunkSize ?? 1000,
+          chunkOverlap: base.chunkOverlap ?? 200
+        }) as any,
+        forceReload
+      )
+      break
 
-  // DraftsExport类型 (file.ext会自动转换成小写)
-  if (['.draftsexport'].includes(file.ext)) {
-    const loaderReturn = await ragApplication.addLoader(new DraftsExportLoader(file.path) as any, forceReload)
-    return {
-      entriesAdded: loaderReturn.entriesAdded,
-      uniqueId: loaderReturn.uniqueId,
-      uniqueIds: [loaderReturn.uniqueId],
-      loaderType: loaderReturn.loaderType
-    }
-  }
+    case 'drafts':
+      // Drafts类型处理
+      loaderReturn = await ragApplication.addLoader(new DraftsExportLoader(file.path) as any, forceReload)
+      break
 
-  const fileContent = fs.readFileSync(file.path, 'utf-8')
+    case 'html':
+      // HTML类型处理
+      loaderReturn = await ragApplication.addLoader(
+        new WebLoader({
+          urlOrContent: fs.readFileSync(file.path, 'utf-8'),
+          chunkSize: base.chunkSize,
+          chunkOverlap: base.chunkOverlap
+        }) as any,
+        forceReload
+      )
+      break
 
-  // HTML类型
-  if (['.html', '.htm'].includes(file.ext)) {
-    const loaderReturn = await ragApplication.addLoader(
-      new WebLoader({
-        urlOrContent: fileContent,
-        chunkSize: base.chunkSize,
-        chunkOverlap: base.chunkOverlap
-      }) as any,
-      forceReload
-    )
-    return {
-      entriesAdded: loaderReturn.entriesAdded,
-      uniqueId: loaderReturn.uniqueId,
-      uniqueIds: [loaderReturn.uniqueId],
-      loaderType: loaderReturn.loaderType
-    }
-  }
-
-  // JSON类型
-  if (['.json'].includes(file.ext)) {
-    let jsonObject = {}
-    let jsonParsed = true
-    try {
-      jsonObject = JSON.parse(fileContent)
-    } catch (error) {
-      jsonParsed = false
-      Logger.warn('[KnowledgeBase] failed parsing json file, failling back to text processing:', file.path, error)
-    }
-    if (jsonParsed) {
-      const loaderReturn = await ragApplication.addLoader(new JsonLoader({ object: jsonObject }))
-      return {
-        entriesAdded: loaderReturn.entriesAdded,
-        uniqueId: loaderReturn.uniqueId,
-        uniqueIds: [loaderReturn.uniqueId],
-        loaderType: loaderReturn.loaderType
+    case 'json':
+      try {
+        jsonObject = JSON.parse(fs.readFileSync(file.path, 'utf-8'))
+      } catch (error) {
+        jsonParsed = false
+        Logger.warn('[KnowledgeBase] failed parsing json file, falling back to text processing:', file.path, error)
       }
-    }
+
+      if (jsonParsed) {
+        loaderReturn = await ragApplication.addLoader(new JsonLoader({ object: jsonObject }), forceReload)
+        break
+      }
+    // fallthrough - JSON 解析失败时作为文本处理
+    default:
+      // 文本类型处理（默认）
+      // 如果是其他文本类型且尚未读取文件，则读取文件
+      loaderReturn = await ragApplication.addLoader(
+        new TextLoader({
+          text: fs.readFileSync(file.path, 'utf-8'),
+          chunkSize: base.chunkSize,
+          chunkOverlap: base.chunkOverlap
+        }) as any,
+        forceReload
+      )
+      break
   }
 
-  // 文本类型
-  const loaderReturn = await ragApplication.addLoader(
-    new TextLoader({ text: fileContent, chunkSize: base.chunkSize, chunkOverlap: base.chunkOverlap }) as any,
-    forceReload
-  )
-
-  Logger.info('[KnowledgeBase] processing file', file.path)
-
   return {
     entriesAdded: loaderReturn.entriesAdded,
     uniqueId: loaderReturn.uniqueId,

src/main/loader/odLoader.ts

@@ -1,6 +1,6 @@
+import { BaseLoader } from '@cherrystudio/embedjs-interfaces'
+import { cleanString } from '@cherrystudio/embedjs-utils'
 import { RecursiveCharacterTextSplitter } from '@langchain/textsplitters'
-import { BaseLoader } from '@llm-tools/embedjs-interfaces'
-import { cleanString } from '@llm-tools/embedjs-utils'
 import md5 from 'md5'
 import { OfficeParserConfig, parseOfficeAsync } from 'officeparser'
 

src/main/mcpServers/brave-search.ts

@@ -0,0 +1,374 @@
+// Brave Search MCP Server
+// port https://github.com/modelcontextprotocol/servers/blob/main/src/brave-search/index.ts
+
+import { Server } from '@modelcontextprotocol/sdk/server/index.js'
+import { CallToolRequestSchema, ListToolsRequestSchema, Tool } from '@modelcontextprotocol/sdk/types.js'
+
+const WEB_SEARCH_TOOL: Tool = {
+  name: 'brave_web_search',
+  description:
+    'Performs a web search using the Brave Search API, ideal for general queries, news, articles, and online content. ' +
+    'Use this for broad information gathering, recent events, or when you need diverse web sources. ' +
+    'Supports pagination, content filtering, and freshness controls. ' +
+    'Maximum 20 results per request, with offset for pagination. ',
+  inputSchema: {
+    type: 'object',
+    properties: {
+      query: {
+        type: 'string',
+        description: 'Search query (max 400 chars, 50 words)'
+      },
+      count: {
+        type: 'number',
+        description: 'Number of results (1-20, default 10)',
+        default: 10
+      },
+      offset: {
+        type: 'number',
+        description: 'Pagination offset (max 9, default 0)',
+        default: 0
+      }
+    },
+    required: ['query']
+  }
+}
+
+const LOCAL_SEARCH_TOOL: Tool = {
+  name: 'brave_local_search',
+  description:
+    "Searches for local businesses and places using Brave's Local Search API. " +
+    'Best for queries related to physical locations, businesses, restaurants, services, etc. ' +
+    'Returns detailed information including:\n' +
+    '- Business names and addresses\n' +
+    '- Ratings and review counts\n' +
+    '- Phone numbers and opening hours\n' +
+    "Use this when the query implies 'near me' or mentions specific locations. " +
+    'Automatically falls back to web search if no local results are found.',
+  inputSchema: {
+    type: 'object',
+    properties: {
+      query: {
+        type: 'string',
+        description: "Local search query (e.g. 'pizza near Central Park')"
+      },
+      count: {
+        type: 'number',
+        description: 'Number of results (1-20, default 5)',
+        default: 5
+      }
+    },
+    required: ['query']
+  }
+}
+
+const RATE_LIMIT = {
+  perSecond: 1,
+  perMonth: 15000
+}
+
+const requestCount = {
+  second: 0,
+  month: 0,
+  lastReset: Date.now()
+}
+
+function checkRateLimit() {
+  const now = Date.now()
+  if (now - requestCount.lastReset > 1000) {
+    requestCount.second = 0
+    requestCount.lastReset = now
+  }
+  if (requestCount.second >= RATE_LIMIT.perSecond || requestCount.month >= RATE_LIMIT.perMonth) {
+    throw new Error('Rate limit exceeded')
+  }
+  requestCount.second++
+  requestCount.month++
+}
+
+interface BraveWeb {
+  web?: {
+    results?: Array<{
+      title: string
+      description: string
+      url: string
+      language?: string
+      published?: string
+      rank?: number
+    }>
+  }
+  locations?: {
+    results?: Array<{
+      id: string // Required by API
+      title?: string
+    }>
+  }
+}
+
+interface BraveLocation {
+  id: string
+  name: string
+  address: {
+    streetAddress?: string
+    addressLocality?: string
+    addressRegion?: string
+    postalCode?: string
+  }
+  coordinates?: {
+    latitude: number
+    longitude: number
+  }
+  phone?: string
+  rating?: {
+    ratingValue?: number
+    ratingCount?: number
+  }
+  openingHours?: string[]
+  priceRange?: string
+}
+
+interface BravePoiResponse {
+  results: BraveLocation[]
+}
+
+interface BraveDescription {
+  descriptions: { [id: string]: string }
+}
+
+function isBraveWebSearchArgs(args: unknown): args is { query: string; count?: number } {
+  return (
+    typeof args === 'object' &&
+    args !== null &&
+    'query' in args &&
+    typeof (args as { query: string }).query === 'string'
+  )
+}
+
+function isBraveLocalSearchArgs(args: unknown): args is { query: string; count?: number } {
+  return (
+    typeof args === 'object' &&
+    args !== null &&
+    'query' in args &&
+    typeof (args as { query: string }).query === 'string'
+  )
+}
+
+async function performWebSearch(apiKey: string, query: string, count: number = 10, offset: number = 0) {
+  checkRateLimit()
+  const url = new URL('https://api.search.brave.com/res/v1/web/search')
+  url.searchParams.set('q', query)
+  url.searchParams.set('count', Math.min(count, 20).toString()) // API limit
+  url.searchParams.set('offset', offset.toString())
+
+  const response = await fetch(url, {
+    headers: {
+      Accept: 'application/json',
+      'Accept-Encoding': 'gzip',
+      'X-Subscription-Token': apiKey
+    }
+  })
+
+  if (!response.ok) {
+    throw new Error(`Brave API error: ${response.status} ${response.statusText}\n${await response.text()}`)
+  }
+
+  const data = (await response.json()) as BraveWeb
+
+  // Extract just web results
+  const results = (data.web?.results || []).map((result) => ({
+    title: result.title || '',
+    description: result.description || '',
+    url: result.url || ''
+  }))
+
+  return results.map((r) => `Title: ${r.title}\nDescription: ${r.description}\nURL: ${r.url}`).join('\n\n')
+}
+
+async function performLocalSearch(apiKey: string, query: string, count: number = 5) {
+  checkRateLimit()
+  // Initial search to get location IDs
+  const webUrl = new URL('https://api.search.brave.com/res/v1/web/search')
+  webUrl.searchParams.set('q', query)
+  webUrl.searchParams.set('search_lang', 'en')
+  webUrl.searchParams.set('result_filter', 'locations')
+  webUrl.searchParams.set('count', Math.min(count, 20).toString())
+
+  const webResponse = await fetch(webUrl, {
+    headers: {
+      Accept: 'application/json',
+      'Accept-Encoding': 'gzip',
+      'X-Subscription-Token': apiKey
+    }
+  })
+
+  if (!webResponse.ok) {
+    throw new Error(`Brave API error: ${webResponse.status} ${webResponse.statusText}\n${await webResponse.text()}`)
+  }
+
+  const webData = (await webResponse.json()) as BraveWeb
+  const locationIds =
+    webData.locations?.results?.filter((r): r is { id: string; title?: string } => r.id != null).map((r) => r.id) || []
+
+  if (locationIds.length === 0) {
+    return performWebSearch(apiKey, query, count) // Fallback to web search
+  }
+
+  // Get POI details and descriptions in parallel
+  const [poisData, descriptionsData] = await Promise.all([
+    getPoisData(apiKey, locationIds),
+    getDescriptionsData(apiKey, locationIds)
+  ])
+
+  return formatLocalResults(poisData, descriptionsData)
+}
+
+async function getPoisData(apiKey: string, ids: string[]): Promise<BravePoiResponse> {
+  checkRateLimit()
+  const url = new URL('https://api.search.brave.com/res/v1/local/pois')
+  ids.filter(Boolean).forEach((id) => url.searchParams.append('ids', id))
+  const response = await fetch(url, {
+    headers: {
+      Accept: 'application/json',
+      'Accept-Encoding': 'gzip',
+      'X-Subscription-Token': apiKey
+    }
+  })
+
+  if (!response.ok) {
+    throw new Error(`Brave API error: ${response.status} ${response.statusText}\n${await response.text()}`)
+  }
+
+  const poisResponse = (await response.json()) as BravePoiResponse
+  return poisResponse
+}
+
+async function getDescriptionsData(apiKey: string, ids: string[]): Promise<BraveDescription> {
+  checkRateLimit()
+  const url = new URL('https://api.search.brave.com/res/v1/local/descriptions')
+  ids.filter(Boolean).forEach((id) => url.searchParams.append('ids', id))
+  const response = await fetch(url, {
+    headers: {
+      Accept: 'application/json',
+      'Accept-Encoding': 'gzip',
+      'X-Subscription-Token': apiKey
+    }
+  })
+
+  if (!response.ok) {
+    throw new Error(`Brave API error: ${response.status} ${response.statusText}\n${await response.text()}`)
+  }
+
+  const descriptionsData = (await response.json()) as BraveDescription
+  return descriptionsData
+}
+
+function formatLocalResults(poisData: BravePoiResponse, descData: BraveDescription): string {
+  return (
+    (poisData.results || [])
+      .map((poi) => {
+        const address =
+          [
+            poi.address?.streetAddress ?? '',
+            poi.address?.addressLocality ?? '',
+            poi.address?.addressRegion ?? '',
+            poi.address?.postalCode ?? ''
+          ]
+            .filter((part) => part !== '')
+            .join(', ') || 'N/A'
+
+        return `Name: ${poi.name}
+Address: ${address}
+Phone: ${poi.phone || 'N/A'}
+Rating: ${poi.rating?.ratingValue ?? 'N/A'} (${poi.rating?.ratingCount ?? 0} reviews)
+Price Range: ${poi.priceRange || 'N/A'}
+Hours: ${(poi.openingHours || []).join(', ') || 'N/A'}
+Description: ${descData.descriptions[poi.id] || 'No description available'}
+`
+      })
+      .join('\n---\n') || 'No local results found'
+  )
+}
+
+class BraveSearchServer {
+  public server: Server
+  private apiKey: string
+
+  constructor(apiKey: string) {
+    if (!apiKey) {
+      throw new Error('BRAVE_API_KEY is required for Brave Search MCP server')
+    }
+    this.apiKey = apiKey
+    this.server = new Server(
+      {
+        name: 'brave-search-server',
+        version: '0.1.0'
+      },
+      {
+        capabilities: {
+          tools: {}
+        }
+      }
+    )
+    this.initialize()
+  }
+
+  initialize() {
+    // Tool handlers
+    this.server.setRequestHandler(ListToolsRequestSchema, async () => ({
+      tools: [WEB_SEARCH_TOOL, LOCAL_SEARCH_TOOL]
+    }))
+
+    this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
+      try {
+        const { name, arguments: args } = request.params
+
+        if (!args) {
+          throw new Error('No arguments provided')
+        }
+
+        switch (name) {
+          case 'brave_web_search': {
+            if (!isBraveWebSearchArgs(args)) {
+              throw new Error('Invalid arguments for brave_web_search')
+            }
+            const { query, count = 10 } = args
+            const results = await performWebSearch(this.apiKey, query, count)
+            return {
+              content: [{ type: 'text', text: results }],
+              isError: false
+            }
+          }
+
+          case 'brave_local_search': {
+            if (!isBraveLocalSearchArgs(args)) {
+              throw new Error('Invalid arguments for brave_local_search')
+            }
+            const { query, count = 5 } = args
+            const results = await performLocalSearch(this.apiKey, query, count)
+            return {
+              content: [{ type: 'text', text: results }],
+              isError: false
+            }
+          }
+
+          default:
+            return {
+              content: [{ type: 'text', text: `Unknown tool: ${name}` }],
+              isError: true
+            }
+        }
+      } catch (error) {
+        return {
+          content: [
+            {
+              type: 'text',
+              text: `Error: ${error instanceof Error ? error.message : String(error)}`
+            }
+          ],
+          isError: true
+        }
+      }
+    })
+  }
+}
+
+export default BraveSearchServer

src/main/mcpServers/factory.ts

@@ -0,0 +1,37 @@
+import { Server } from '@modelcontextprotocol/sdk/server/index.js'
+import Logger from 'electron-log'
+
+import BraveSearchServer from './brave-search'
+import FetchServer from './fetch'
+import FileSystemServer from './filesystem'
+import MemoryServer from './memory'
+import ThinkingServer from './sequentialthinking'
+import SimpleRememberServer from './simpleremember'
+
+export function createInMemoryMCPServer(name: string, args: string[] = [], envs: Record<string, string> = {}): Server {
+  Logger.info(`[MCP] Creating in-memory MCP server: ${name} with args: ${args} and envs: ${JSON.stringify(envs)}`)
+  switch (name) {
+    case '@cherry/memory': {
+      const envPath = envs.MEMORY_FILE_PATH
+      return new MemoryServer(envPath).server
+    }
+    case '@cherry/sequentialthinking': {
+      return new ThinkingServer().server
+    }
+    case '@cherry/brave-search': {
+      return new BraveSearchServer(envs.BRAVE_API_KEY).server
+    }
+    case '@cherry/fetch': {
+      return new FetchServer().server
+    }
+    case '@cherry/filesystem': {
+      return new FileSystemServer(args).server
+    }
+    case '@cherry/simpleremember': {
+      const envPath = envs.SIMPLEREMEMBER_FILE_PATH
+      return new SimpleRememberServer(envPath).server
+    }
+    default:
+      throw new Error(`Unknown in-memory MCP server: ${name}`)
+  }
+}

src/main/mcpServers/fetch.ts

@@ -0,0 +1,236 @@
+// port https://github.com/zcaceres/fetch-mcp/blob/main/src/index.ts
+
+import { Server } from '@modelcontextprotocol/sdk/server/index.js'
+import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js'
+import { JSDOM } from 'jsdom'
+import TurndownService from 'turndown'
+import { z } from 'zod'
+
+export const RequestPayloadSchema = z.object({
+  url: z.string().url(),
+  headers: z.record(z.string()).optional()
+})
+
+export type RequestPayload = z.infer<typeof RequestPayloadSchema>
+
+export class Fetcher {
+  private static async _fetch({ url, headers }: RequestPayload): Promise<Response> {
+    try {
+      const response = await fetch(url, {
+        headers: {
+          'User-Agent':
+            'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
+          ...headers
+        }
+      })
+
+      if (!response.ok) {
+        throw new Error(`HTTP error: ${response.status}`)
+      }
+      return response
+    } catch (e: unknown) {
+      if (e instanceof Error) {
+        throw new Error(`Failed to fetch ${url}: ${e.message}`)
+      } else {
+        throw new Error(`Failed to fetch ${url}: Unknown error`)
+      }
+    }
+  }
+
+  static async html(requestPayload: RequestPayload) {
+    try {
+      const response = await this._fetch(requestPayload)
+      const html = await response.text()
+      return { content: [{ type: 'text', text: html }], isError: false }
+    } catch (error) {
+      return {
+        content: [{ type: 'text', text: (error as Error).message }],
+        isError: true
+      }
+    }
+  }
+
+  static async json(requestPayload: RequestPayload) {
+    try {
+      const response = await this._fetch(requestPayload)
+      const json = await response.json()
+      return {
+        content: [{ type: 'text', text: JSON.stringify(json) }],
+        isError: false
+      }
+    } catch (error) {
+      return {
+        content: [{ type: 'text', text: (error as Error).message }],
+        isError: true
+      }
+    }
+  }
+
+  static async txt(requestPayload: RequestPayload) {
+    try {
+      const response = await this._fetch(requestPayload)
+      const html = await response.text()
+
+      const dom = new JSDOM(html)
+      const document = dom.window.document
+
+      const scripts = document.getElementsByTagName('script')
+      const styles = document.getElementsByTagName('style')
+      Array.from(scripts).forEach((script: any) => script.remove())
+      Array.from(styles).forEach((style: any) => style.remove())
+
+      const text = document.body.textContent || ''
+
+      const normalizedText = text.replace(/\s+/g, ' ').trim()
+
+      return {
+        content: [{ type: 'text', text: normalizedText }],
+        isError: false
+      }
+    } catch (error) {
+      return {
+        content: [{ type: 'text', text: (error as Error).message }],
+        isError: true
+      }
+    }
+  }
+
+  static async markdown(requestPayload: RequestPayload) {
+    try {
+      const response = await this._fetch(requestPayload)
+      const html = await response.text()
+      const turndownService = new TurndownService()
+      const markdown = turndownService.turndown(html)
+      return { content: [{ type: 'text', text: markdown }], isError: false }
+    } catch (error) {
+      return {
+        content: [{ type: 'text', text: (error as Error).message }],
+        isError: true
+      }
+    }
+  }
+}
+
+const server = new Server(
+  {
+    name: 'zcaceres/fetch',
+    version: '0.1.0'
+  },
+  {
+    capabilities: {
+      resources: {},
+      tools: {}
+    }
+  }
+)
+
+server.setRequestHandler(ListToolsRequestSchema, async () => {
+  return {
+    tools: [
+      {
+        name: 'fetch_html',
+        description: 'Fetch a website and return the content as HTML',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            url: {
+              type: 'string',
+              description: 'URL of the website to fetch'
+            },
+            headers: {
+              type: 'object',
+              description: 'Optional headers to include in the request'
+            }
+          },
+          required: ['url']
+        }
+      },
+      {
+        name: 'fetch_markdown',
+        description: 'Fetch a website and return the content as Markdown',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            url: {
+              type: 'string',
+              description: 'URL of the website to fetch'
+            },
+            headers: {
+              type: 'object',
+              description: 'Optional headers to include in the request'
+            }
+          },
+          required: ['url']
+        }
+      },
+      {
+        name: 'fetch_txt',
+        description: 'Fetch a website, return the content as plain text (no HTML)',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            url: {
+              type: 'string',
+              description: 'URL of the website to fetch'
+            },
+            headers: {
+              type: 'object',
+              description: 'Optional headers to include in the request'
+            }
+          },
+          required: ['url']
+        }
+      },
+      {
+        name: 'fetch_json',
+        description: 'Fetch a JSON file from a URL',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            url: {
+              type: 'string',
+              description: 'URL of the JSON to fetch'
+            },
+            headers: {
+              type: 'object',
+              description: 'Optional headers to include in the request'
+            }
+          },
+          required: ['url']
+        }
+      }
+    ]
+  }
+})
+
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+  const { arguments: args } = request.params
+
+  const validatedArgs = RequestPayloadSchema.parse(args)
+
+  if (request.params.name === 'fetch_html') {
+    const fetchResult = await Fetcher.html(validatedArgs)
+    return fetchResult
+  }
+  if (request.params.name === 'fetch_json') {
+    const fetchResult = await Fetcher.json(validatedArgs)
+    return fetchResult
+  }
+  if (request.params.name === 'fetch_txt') {
+    const fetchResult = await Fetcher.txt(validatedArgs)
+    return fetchResult
+  }
+  if (request.params.name === 'fetch_markdown') {
+    const fetchResult = await Fetcher.markdown(validatedArgs)
+    return fetchResult
+  }
+  throw new Error('Tool not found')
+})
+
+class FetchServer {
+  public server: Server
+  constructor() {
+    this.server = server
+  }
+}
+export default FetchServer

src/main/mcpServers/filesystem.ts

@@ -0,0 +1,655 @@
+// port https://github.com/modelcontextprotocol/servers/blob/main/src/filesystem/index.ts
+
+import { Server } from '@modelcontextprotocol/sdk/server/index.js'
+import { CallToolRequestSchema, ListToolsRequestSchema, ToolSchema } from '@modelcontextprotocol/sdk/types.js'
+import { createTwoFilesPatch } from 'diff'
+import fs from 'fs/promises'
+import { minimatch } from 'minimatch'
+import os from 'os'
+import path from 'path'
+import { z } from 'zod'
+import { zodToJsonSchema } from 'zod-to-json-schema'
+
+// Normalize all paths consistently
+function normalizePath(p: string): string {
+  return path.normalize(p)
+}
+
+function expandHome(filepath: string): string {
+  if (filepath.startsWith('~/') || filepath === '~') {
+    return path.join(os.homedir(), filepath.slice(1))
+  }
+  return filepath
+}
+
+// Security utilities
+async function validatePath(allowedDirectories: string[], requestedPath: string): Promise<string> {
+  const expandedPath = expandHome(requestedPath)
+  const absolute = path.isAbsolute(expandedPath)
+    ? path.resolve(expandedPath)
+    : path.resolve(process.cwd(), expandedPath)
+
+  const normalizedRequested = normalizePath(absolute)
+
+  // Check if path is within allowed directories
+  const isAllowed = allowedDirectories.some((dir) => normalizedRequested.startsWith(dir))
+  if (!isAllowed) {
+    throw new Error(
+      `Access denied - path outside allowed directories: ${absolute} not in ${allowedDirectories.join(', ')}`
+    )
+  }
+
+  // Handle symlinks by checking their real path
+  try {
+    const realPath = await fs.realpath(absolute)
+    const normalizedReal = normalizePath(realPath)
+    const isRealPathAllowed = allowedDirectories.some((dir) => normalizedReal.startsWith(dir))
+    if (!isRealPathAllowed) {
+      throw new Error('Access denied - symlink target outside allowed directories')
+    }
+    return realPath
+  } catch (error) {
+    // For new files that don't exist yet, verify parent directory
+    const parentDir = path.dirname(absolute)
+    try {
+      const realParentPath = await fs.realpath(parentDir)
+      const normalizedParent = normalizePath(realParentPath)
+      const isParentAllowed = allowedDirectories.some((dir) => normalizedParent.startsWith(dir))
+      if (!isParentAllowed) {
+        throw new Error('Access denied - parent directory outside allowed directories')
+      }
+      return absolute
+    } catch {
+      throw new Error(`Parent directory does not exist: ${parentDir}`)
+    }
+  }
+}
+
+// Schema definitions
+const ReadFileArgsSchema = z.object({
+  path: z.string()
+})
+
+const ReadMultipleFilesArgsSchema = z.object({
+  paths: z.array(z.string())
+})
+
+const WriteFileArgsSchema = z.object({
+  path: z.string(),
+  content: z.string()
+})
+
+const EditOperation = z.object({
+  oldText: z.string().describe('Text to search for - must match exactly'),
+  newText: z.string().describe('Text to replace with')
+})
+
+const EditFileArgsSchema = z.object({
+  path: z.string(),
+  edits: z.array(EditOperation),
+  dryRun: z.boolean().default(false).describe('Preview changes using git-style diff format')
+})
+
+const CreateDirectoryArgsSchema = z.object({
+  path: z.string()
+})
+
+const ListDirectoryArgsSchema = z.object({
+  path: z.string()
+})
+
+const DirectoryTreeArgsSchema = z.object({
+  path: z.string()
+})
+
+const MoveFileArgsSchema = z.object({
+  source: z.string(),
+  destination: z.string()
+})
+
+const SearchFilesArgsSchema = z.object({
+  path: z.string(),
+  pattern: z.string(),
+  excludePatterns: z.array(z.string()).optional().default([])
+})
+
+const GetFileInfoArgsSchema = z.object({
+  path: z.string()
+})
+
+// eslint-disable-next-line @typescript-eslint/no-unused-vars
+const ToolInputSchema = ToolSchema.shape.inputSchema
+type ToolInput = z.infer<typeof ToolInputSchema>
+
+interface FileInfo {
+  size: number
+  created: Date
+  modified: Date
+  accessed: Date
+  isDirectory: boolean
+  isFile: boolean
+  permissions: string
+}
+
+// Tool implementations
+async function getFileStats(filePath: string): Promise<FileInfo> {
+  const stats = await fs.stat(filePath)
+  return {
+    size: stats.size,
+    created: stats.birthtime,
+    modified: stats.mtime,
+    accessed: stats.atime,
+    isDirectory: stats.isDirectory(),
+    isFile: stats.isFile(),
+    permissions: stats.mode.toString(8).slice(-3)
+  }
+}
+
+async function searchFiles(
+  allowedDirectories: string[],
+  rootPath: string,
+  pattern: string,
+  excludePatterns: string[] = []
+): Promise<string[]> {
+  const results: string[] = []
+
+  async function search(currentPath: string) {
+    const entries = await fs.readdir(currentPath, { withFileTypes: true })
+
+    for (const entry of entries) {
+      const fullPath = path.join(currentPath, entry.name)
+
+      try {
+        // Validate each path before processing
+        await validatePath(allowedDirectories, fullPath)
+
+        // Check if path matches any exclude pattern
+        const relativePath = path.relative(rootPath, fullPath)
+        const shouldExclude = excludePatterns.some((pattern) => {
+          const globPattern = pattern.includes('*') ? pattern : `**/${pattern}/**`
+          return minimatch(relativePath, globPattern, { dot: true })
+        })
+
+        if (shouldExclude) {
+          continue
+        }
+
+        if (entry.name.toLowerCase().includes(pattern.toLowerCase())) {
+          results.push(fullPath)
+        }
+
+        if (entry.isDirectory()) {
+          await search(fullPath)
+        }
+      } catch (error) {
+        // Skip invalid paths during search
+        continue
+      }
+    }
+  }
+
+  await search(rootPath)
+  return results
+}
+
+// file editing and diffing utilities
+function normalizeLineEndings(text: string): string {
+  return text.replace(/\r\n/g, '\n')
+}
+
+function createUnifiedDiff(originalContent: string, newContent: string, filepath: string = 'file'): string {
+  // Ensure consistent line endings for diff
+  const normalizedOriginal = normalizeLineEndings(originalContent)
+  const normalizedNew = normalizeLineEndings(newContent)
+
+  return createTwoFilesPatch(filepath, filepath, normalizedOriginal, normalizedNew, 'original', 'modified')
+}
+
+async function applyFileEdits(
+  filePath: string,
+  edits: Array<{ oldText: string; newText: string }>,
+  dryRun = false
+): Promise<string> {
+  // Read file content and normalize line endings
+  const content = normalizeLineEndings(await fs.readFile(filePath, 'utf-8'))
+
+  // Apply edits sequentially
+  let modifiedContent = content
+  for (const edit of edits) {
+    const normalizedOld = normalizeLineEndings(edit.oldText)
+    const normalizedNew = normalizeLineEndings(edit.newText)
+
+    // If exact match exists, use it
+    if (modifiedContent.includes(normalizedOld)) {
+      modifiedContent = modifiedContent.replace(normalizedOld, normalizedNew)
+      continue
+    }
+
+    // Otherwise, try line-by-line matching with flexibility for whitespace
+    const oldLines = normalizedOld.split('\n')
+    const contentLines = modifiedContent.split('\n')
+    let matchFound = false
+
+    for (let i = 0; i <= contentLines.length - oldLines.length; i++) {
+      const potentialMatch = contentLines.slice(i, i + oldLines.length)
+
+      // Compare lines with normalized whitespace
+      const isMatch = oldLines.every((oldLine, j) => {
+        const contentLine = potentialMatch[j]
+        return oldLine.trim() === contentLine.trim()
+      })
+
+      if (isMatch) {
+        // Preserve original indentation of first line
+        const originalIndent = contentLines[i].match(/^\s*/)?.[0] || ''
+        const newLines = normalizedNew.split('\n').map((line, j) => {
+          if (j === 0) return originalIndent + line.trimStart()
+          // For subsequent lines, try to preserve relative indentation
+          const oldIndent = oldLines[j]?.match(/^\s*/)?.[0] || ''
+          const newIndent = line.match(/^\s*/)?.[0] || ''
+          if (oldIndent && newIndent) {
+            const relativeIndent = newIndent.length - oldIndent.length
+            return originalIndent + ' '.repeat(Math.max(0, relativeIndent)) + line.trimStart()
+          }
+          return line
+        })
+
+        contentLines.splice(i, oldLines.length, ...newLines)
+        modifiedContent = contentLines.join('\n')
+        matchFound = true
+        break
+      }
+    }
+
+    if (!matchFound) {
+      throw new Error(`Could not find exact match for edit:\n${edit.oldText}`)
+    }
+  }
+
+  // Create unified diff
+  const diff = createUnifiedDiff(content, modifiedContent, filePath)
+
+  // Format diff with appropriate number of backticks
+  let numBackticks = 3
+  while (diff.includes('`'.repeat(numBackticks))) {
+    numBackticks++
+  }
+  const formattedDiff = `${'`'.repeat(numBackticks)}diff\n${diff}${'`'.repeat(numBackticks)}\n\n`
+
+  if (!dryRun) {
+    await fs.writeFile(filePath, modifiedContent, 'utf-8')
+  }
+
+  return formattedDiff
+}
+
+class FileSystemServer {
+  public server: Server
+  private allowedDirectories: string[]
+  constructor(allowedDirs: string[]) {
+    if (!Array.isArray(allowedDirs) || allowedDirs.length === 0) {
+      throw new Error('No allowed directories provided, please specify at least one directory in args')
+    }
+
+    this.allowedDirectories = allowedDirs.map((dir) => normalizePath(path.resolve(expandHome(dir))))
+
+    // Validate that all directories exist and are accessible
+    this.validateDirs().catch((error) => {
+      console.error('Error validating allowed directories:', error)
+      throw new Error(`Error validating allowed directories: ${error}`)
+    })
+
+    this.server = new Server(
+      {
+        name: 'secure-filesystem-server',
+        version: '0.2.0'
+      },
+      {
+        capabilities: {
+          tools: {}
+        }
+      }
+    )
+    this.initialize()
+  }
+
+  async validateDirs() {
+    // Validate that all directories exist and are accessible
+    await Promise.all(
+      this.allowedDirectories.map(async (dir) => {
+        try {
+          const stats = await fs.stat(expandHome(dir))
+          if (!stats.isDirectory()) {
+            console.error(`Error: ${dir} is not a directory`)
+            throw new Error(`Error: ${dir} is not a directory`)
+          }
+        } catch (error: any) {
+          console.error(`Error accessing directory ${dir}:`, error)
+          throw new Error(`Error accessing directory ${dir}:`, error)
+        }
+      })
+    )
+  }
+
+  initialize() {
+    // Tool handlers
+    this.server.setRequestHandler(ListToolsRequestSchema, async () => {
+      return {
+        tools: [
+          {
+            name: 'read_file',
+            description:
+              'Read the complete contents of a file from the file system. ' +
+              'Handles various text encodings and provides detailed error messages ' +
+              'if the file cannot be read. Use this tool when you need to examine ' +
+              'the contents of a single file. Only works within allowed directories.',
+            inputSchema: zodToJsonSchema(ReadFileArgsSchema) as ToolInput
+          },
+          {
+            name: 'read_multiple_files',
+            description:
+              'Read the contents of multiple files simultaneously. This is more ' +
+              'efficient than reading files one by one when you need to analyze ' +
+              "or compare multiple files. Each file's content is returned with its " +
+              "path as a reference. Failed reads for individual files won't stop " +
+              'the entire operation. Only works within allowed directories.',
+            inputSchema: zodToJsonSchema(ReadMultipleFilesArgsSchema) as ToolInput
+          },
+          {
+            name: 'write_file',
+            description:
+              'Create a new file or completely overwrite an existing file with new content. ' +
+              'Use with caution as it will overwrite existing files without warning. ' +
+              'Handles text content with proper encoding. Only works within allowed directories.',
+            inputSchema: zodToJsonSchema(WriteFileArgsSchema) as ToolInput
+          },
+          {
+            name: 'edit_file',
+            description:
+              'Make line-based edits to a text file. Each edit replaces exact line sequences ' +
+              'with new content. Returns a git-style diff showing the changes made. ' +
+              'Only works within allowed directories.',
+            inputSchema: zodToJsonSchema(EditFileArgsSchema) as ToolInput
+          },
+          {
+            name: 'create_directory',
+            description:
+              'Create a new directory or ensure a directory exists. Can create multiple ' +
+              'nested directories in one operation. If the directory already exists, ' +
+              'this operation will succeed silently. Perfect for setting up directory ' +
+              'structures for projects or ensuring required paths exist. Only works within allowed directories.',
+            inputSchema: zodToJsonSchema(CreateDirectoryArgsSchema) as ToolInput
+          },
+          {
+            name: 'list_directory',
+            description:
+              'Get a detailed listing of all files and directories in a specified path. ' +
+              'Results clearly distinguish between files and directories with [FILE] and [DIR] ' +
+              'prefixes. This tool is essential for understanding directory structure and ' +
+              'finding specific files within a directory. Only works within allowed directories.',
+            inputSchema: zodToJsonSchema(ListDirectoryArgsSchema) as ToolInput
+          },
+          {
+            name: 'directory_tree',
+            description:
+              'Get a recursive tree view of files and directories as a JSON structure. ' +
+              "Each entry includes 'name', 'type' (file/directory), and 'children' for directories. " +
+              'Files have no children array, while directories always have a children array (which may be empty). ' +
+              'The output is formatted with 2-space indentation for readability. Only works within allowed directories.',
+            inputSchema: zodToJsonSchema(DirectoryTreeArgsSchema) as ToolInput
+          },
+          {
+            name: 'move_file',
+            description:
+              'Move or rename files and directories. Can move files between directories ' +
+              'and rename them in a single operation. If the destination exists, the ' +
+              'operation will fail. Works across different directories and can be used ' +
+              'for simple renaming within the same directory. Both source and destination must be within allowed directories.',
+            inputSchema: zodToJsonSchema(MoveFileArgsSchema) as ToolInput
+          },
+          {
+            name: 'search_files',
+            description:
+              'Recursively search for files and directories matching a pattern. ' +
+              'Searches through all subdirectories from the starting path. The search ' +
+              'is case-insensitive and matches partial names. Returns full paths to all ' +
+              "matching items. Great for finding files when you don't know their exact location. " +
+              'Only searches within allowed directories.',
+            inputSchema: zodToJsonSchema(SearchFilesArgsSchema) as ToolInput
+          },
+          {
+            name: 'get_file_info',
+            description:
+              'Retrieve detailed metadata about a file or directory. Returns comprehensive ' +
+              'information including size, creation time, last modified time, permissions, ' +
+              'and type. This tool is perfect for understanding file characteristics ' +
+              'without reading the actual content. Only works within allowed directories.',
+            inputSchema: zodToJsonSchema(GetFileInfoArgsSchema) as ToolInput
+          },
+          {
+            name: 'list_allowed_directories',
+            description:
+              'Returns the list of directories that this server is allowed to access. ' +
+              'Use this to understand which directories are available before trying to access files.',
+            inputSchema: {
+              type: 'object',
+              properties: {},
+              required: []
+            }
+          }
+        ]
+      }
+    })
+
+    this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
+      try {
+        const { name, arguments: args } = request.params
+
+        switch (name) {
+          case 'read_file': {
+            const parsed = ReadFileArgsSchema.safeParse(args)
+            if (!parsed.success) {
+              throw new Error(`Invalid arguments for read_file: ${parsed.error}`)
+            }
+            const validPath = await validatePath(this.allowedDirectories, parsed.data.path)
+            const content = await fs.readFile(validPath, 'utf-8')
+            return {
+              content: [{ type: 'text', text: content }]
+            }
+          }
+
+          case 'read_multiple_files': {
+            const parsed = ReadMultipleFilesArgsSchema.safeParse(args)
+            if (!parsed.success) {
+              throw new Error(`Invalid arguments for read_multiple_files: ${parsed.error}`)
+            }
+            const results = await Promise.all(
+              parsed.data.paths.map(async (filePath: string) => {
+                try {
+                  const validPath = await validatePath(this.allowedDirectories, filePath)
+                  const content = await fs.readFile(validPath, 'utf-8')
+                  return `${filePath}:\n${content}\n`
+                } catch (error) {
+                  const errorMessage = error instanceof Error ? error.message : String(error)
+                  return `${filePath}: Error - ${errorMessage}`
+                }
+              })
+            )
+            return {
+              content: [{ type: 'text', text: results.join('\n---\n') }]
+            }
+          }
+
+          case 'write_file': {
+            const parsed = WriteFileArgsSchema.safeParse(args)
+            if (!parsed.success) {
+              throw new Error(`Invalid arguments for write_file: ${parsed.error}`)
+            }
+            const validPath = await validatePath(this.allowedDirectories, parsed.data.path)
+            await fs.writeFile(validPath, parsed.data.content, 'utf-8')
+            return {
+              content: [{ type: 'text', text: `Successfully wrote to ${parsed.data.path}` }]
+            }
+          }
+
+          case 'edit_file': {
+            const parsed = EditFileArgsSchema.safeParse(args)
+            if (!parsed.success) {
+              throw new Error(`Invalid arguments for edit_file: ${parsed.error}`)
+            }
+            const validPath = await validatePath(this.allowedDirectories, parsed.data.path)
+            const result = await applyFileEdits(validPath, parsed.data.edits, parsed.data.dryRun)
+            return {
+              content: [{ type: 'text', text: result }]
+            }
+          }
+
+          case 'create_directory': {
+            const parsed = CreateDirectoryArgsSchema.safeParse(args)
+            if (!parsed.success) {
+              throw new Error(`Invalid arguments for create_directory: ${parsed.error}`)
+            }
+            const validPath = await validatePath(this.allowedDirectories, parsed.data.path)
+            await fs.mkdir(validPath, { recursive: true })
+            return {
+              content: [{ type: 'text', text: `Successfully created directory ${parsed.data.path}` }]
+            }
+          }
+
+          case 'list_directory': {
+            const parsed = ListDirectoryArgsSchema.safeParse(args)
+            if (!parsed.success) {
+              throw new Error(`Invalid arguments for list_directory: ${parsed.error}`)
+            }
+            const validPath = await validatePath(this.allowedDirectories, parsed.data.path)
+            const entries = await fs.readdir(validPath, { withFileTypes: true })
+            const formatted = entries
+              .map((entry) => `${entry.isDirectory() ? '[DIR]' : '[FILE]'} ${entry.name}`)
+              .join('\n')
+            return {
+              content: [{ type: 'text', text: formatted }]
+            }
+          }
+
+          case 'directory_tree': {
+            const parsed = DirectoryTreeArgsSchema.safeParse(args)
+            if (!parsed.success) {
+              throw new Error(`Invalid arguments for directory_tree: ${parsed.error}`)
+            }
+
+            interface TreeEntry {
+              name: string
+              type: 'file' | 'directory'
+              children?: TreeEntry[]
+            }
+
+            async function buildTree(allowedDirectories: string[], currentPath: string): Promise<TreeEntry[]> {
+              const validPath = await validatePath(allowedDirectories, currentPath)
+              const entries = await fs.readdir(validPath, { withFileTypes: true })
+              const result: TreeEntry[] = []
+
+              for (const entry of entries) {
+                const entryData: TreeEntry = {
+                  name: entry.name,
+                  type: entry.isDirectory() ? 'directory' : 'file'
+                }
+
+                if (entry.isDirectory()) {
+                  const subPath = path.join(currentPath, entry.name)
+                  entryData.children = await buildTree(allowedDirectories, subPath)
+                }
+
+                result.push(entryData)
+              }
+
+              return result
+            }
+
+            const treeData = await buildTree(this.allowedDirectories, parsed.data.path)
+            return {
+              content: [
+                {
+                  type: 'text',
+                  text: JSON.stringify(treeData, null, 2)
+                }
+              ]
+            }
+          }
+
+          case 'move_file': {
+            const parsed = MoveFileArgsSchema.safeParse(args)
+            if (!parsed.success) {
+              throw new Error(`Invalid arguments for move_file: ${parsed.error}`)
+            }
+            const validSourcePath = await validatePath(this.allowedDirectories, parsed.data.source)
+            const validDestPath = await validatePath(this.allowedDirectories, parsed.data.destination)
+            await fs.rename(validSourcePath, validDestPath)
+            return {
+              content: [
+                { type: 'text', text: `Successfully moved ${parsed.data.source} to ${parsed.data.destination}` }
+              ]
+            }
+          }
+
+          case 'search_files': {
+            const parsed = SearchFilesArgsSchema.safeParse(args)
+            if (!parsed.success) {
+              throw new Error(`Invalid arguments for search_files: ${parsed.error}`)
+            }
+            const validPath = await validatePath(this.allowedDirectories, parsed.data.path)
+            const results = await searchFiles(
+              this.allowedDirectories,
+              validPath,
+              parsed.data.pattern,
+              parsed.data.excludePatterns
+            )
+            return {
+              content: [{ type: 'text', text: results.length > 0 ? results.join('\n') : 'No matches found' }]
+            }
+          }
+
+          case 'get_file_info': {
+            const parsed = GetFileInfoArgsSchema.safeParse(args)
+            if (!parsed.success) {
+              throw new Error(`Invalid arguments for get_file_info: ${parsed.error}`)
+            }
+            const validPath = await validatePath(this.allowedDirectories, parsed.data.path)
+            const info = await getFileStats(validPath)
+            return {
+              content: [
+                {
+                  type: 'text',
+                  text: Object.entries(info)
+                    .map(([key, value]) => `${key}: ${value}`)
+                    .join('\n')
+                }
+              ]
+            }
+          }
+
+          case 'list_allowed_directories': {
+            return {
+              content: [
+                {
+                  type: 'text',
+                  text: `Allowed directories:\n${this.allowedDirectories.join('\n')}`
+                }
+              ]
+            }
+          }
+
+          default:
+            throw new Error(`Unknown tool: ${name}`)
+        }
+      } catch (error) {
+        const errorMessage = error instanceof Error ? error.message : String(error)
+        return {
+          content: [{ type: 'text', text: `Error: ${errorMessage}` }],
+          isError: true
+        }
+      }
+    })
+  }
+}
+
+export default FileSystemServer

src/main/mcpServers/memory.ts

@@ -0,0 +1,700 @@
+import { getConfigDir } from '@main/utils/file'
+import { Server } from '@modelcontextprotocol/sdk/server/index.js'
+import { CallToolRequestSchema, ErrorCode, ListToolsRequestSchema, McpError } from '@modelcontextprotocol/sdk/types.js'
+import { Mutex } from 'async-mutex' // 引入 Mutex
+import { promises as fs } from 'fs'
+import path from 'path'
+
+// Define memory file path
+const defaultMemoryPath = path.join(getConfigDir(), 'memory.json')
+
+// Interfaces remain the same
+interface Entity {
+  name: string
+  entityType: string
+  observations: string[]
+}
+
+interface Relation {
+  from: string
+  to: string
+  relationType: string
+}
+
+// Structure for storing the graph in memory and in the file
+interface KnowledgeGraph {
+  entities: Entity[]
+  relations: Relation[]
+}
+
+// The KnowledgeGraphManager class contains all operations to interact with the knowledge graph
+class KnowledgeGraphManager {
+  private memoryPath: string
+  private entities: Map<string, Entity> // Use Map for efficient entity lookup
+  private relations: Set<string> // Store stringified relations for easy Set operations
+  private fileMutex: Mutex // Mutex for file writing
+
+  private constructor(memoryPath: string) {
+    this.memoryPath = memoryPath
+    this.entities = new Map<string, Entity>()
+    this.relations = new Set<string>()
+    this.fileMutex = new Mutex()
+  }
+
+  // Static async factory method for initialization
+  public static async create(memoryPath: string): Promise<KnowledgeGraphManager> {
+    const manager = new KnowledgeGraphManager(memoryPath)
+    await manager._ensureMemoryPathExists()
+    await manager._loadGraphFromDisk()
+    return manager
+  }
+
+  private async _ensureMemoryPathExists(): Promise<void> {
+    try {
+      const directory = path.dirname(this.memoryPath)
+      await fs.mkdir(directory, { recursive: true })
+      try {
+        await fs.access(this.memoryPath)
+      } catch (error) {
+        // File doesn't exist, create an empty file with initial structure
+        await fs.writeFile(this.memoryPath, JSON.stringify({ entities: [], relations: [] }, null, 2))
+      }
+    } catch (error) {
+      console.error('Failed to ensure memory path exists:', error)
+      // Propagate the error or handle it more gracefully depending on requirements
+      throw new McpError(
+        ErrorCode.InternalError,
+        `Failed to ensure memory path: ${error instanceof Error ? error.message : String(error)}`
+      )
+    }
+  }
+
+  // Load graph from disk into memory (called once during initialization)
+  private async _loadGraphFromDisk(): Promise<void> {
+    try {
+      const data = await fs.readFile(this.memoryPath, 'utf-8')
+      // Handle empty file case
+      if (data.trim() === '') {
+        this.entities = new Map()
+        this.relations = new Set()
+        // Optionally write the initial empty structure back
+        await this._persistGraph()
+        return
+      }
+      const graph: KnowledgeGraph = JSON.parse(data)
+      this.entities.clear()
+      this.relations.clear()
+      graph.entities.forEach((entity) => this.entities.set(entity.name, entity))
+      graph.relations.forEach((relation) => this.relations.add(this._serializeRelation(relation)))
+    } catch (error) {
+      if (error instanceof Error && 'code' in error && (error as any).code === 'ENOENT') {
+        // File doesn't exist (should have been created by _ensureMemoryPathExists, but handle defensively)
+        this.entities = new Map()
+        this.relations = new Set()
+        await this._persistGraph() // Create the file with empty structure
+      } else if (error instanceof SyntaxError) {
+        console.error('Failed to parse memory.json, initializing with empty graph:', error)
+        // If JSON is invalid, start fresh and overwrite the corrupted file
+        this.entities = new Map()
+        this.relations = new Set()
+        await this._persistGraph()
+      } else {
+        console.error('Failed to load knowledge graph from disk:', error)
+        throw new McpError(
+          ErrorCode.InternalError,
+          `Failed to load graph: ${error instanceof Error ? error.message : String(error)}`
+        )
+      }
+    }
+  }
+
+  // Persist the current in-memory graph to disk using a mutex
+  private async _persistGraph(): Promise<void> {
+    const release = await this.fileMutex.acquire()
+    try {
+      const graphData: KnowledgeGraph = {
+        entities: Array.from(this.entities.values()),
+        relations: Array.from(this.relations).map((rStr) => this._deserializeRelation(rStr))
+      }
+      await fs.writeFile(this.memoryPath, JSON.stringify(graphData, null, 2))
+    } catch (error) {
+      console.error('Failed to save knowledge graph:', error)
+      // Decide how to handle write errors - potentially retry or notify
+      throw new McpError(
+        ErrorCode.InternalError,
+        `Failed to save graph: ${error instanceof Error ? error.message : String(error)}`
+      )
+    } finally {
+      release()
+    }
+  }
+
+  // Helper to consistently serialize relations for Set storage
+  private _serializeRelation(relation: Relation): string {
+    // Simple serialization, ensure order doesn't matter if properties are consistent
+    return JSON.stringify({ from: relation.from, to: relation.to, relationType: relation.relationType })
+  }
+
+  // Helper to deserialize relations from Set storage
+  private _deserializeRelation(relationStr: string): Relation {
+    return JSON.parse(relationStr) as Relation
+  }
+
+  async createEntities(entities: Entity[]): Promise<Entity[]> {
+    const newEntities: Entity[] = []
+    entities.forEach((entity) => {
+      if (!this.entities.has(entity.name)) {
+        // Ensure observations is always an array
+        const newEntity = { ...entity, observations: Array.isArray(entity.observations) ? entity.observations : [] }
+        this.entities.set(entity.name, newEntity)
+        newEntities.push(newEntity)
+      }
+    })
+    if (newEntities.length > 0) {
+      await this._persistGraph()
+    }
+    return newEntities
+  }
+
+  async createRelations(relations: Relation[]): Promise<Relation[]> {
+    const newRelations: Relation[] = []
+    relations.forEach((relation) => {
+      // Ensure related entities exist before creating a relation
+      if (!this.entities.has(relation.from) || !this.entities.has(relation.to)) {
+        console.warn(`Skipping relation creation: Entity not found for relation ${relation.from} -> ${relation.to}`)
+        return // Skip this relation
+      }
+      const relationStr = this._serializeRelation(relation)
+      if (!this.relations.has(relationStr)) {
+        this.relations.add(relationStr)
+        newRelations.push(relation)
+      }
+    })
+    if (newRelations.length > 0) {
+      await this._persistGraph()
+    }
+    return newRelations
+  }
+
+  async addObservations(
+    observations: { entityName: string; contents: string[] }[]
+  ): Promise<{ entityName: string; addedObservations: string[] }[]> {
+    const results: { entityName: string; addedObservations: string[] }[] = []
+    let changed = false
+    observations.forEach((o) => {
+      const entity = this.entities.get(o.entityName)
+      if (!entity) {
+        // Option 1: Throw error
+        throw new McpError(ErrorCode.InvalidParams, `Entity with name ${o.entityName} not found`)
+        // Option 2: Skip and warn
+        // console.warn(`Entity with name ${o.entityName} not found when adding observations. Skipping.`);
+        // return;
+      }
+      // Ensure observations array exists
+      if (!Array.isArray(entity.observations)) {
+        entity.observations = []
+      }
+      const newObservations = o.contents.filter((content) => !entity.observations.includes(content))
+      if (newObservations.length > 0) {
+        entity.observations.push(...newObservations)
+        results.push({ entityName: o.entityName, addedObservations: newObservations })
+        changed = true
+      } else {
+        // Still include in results even if nothing was added, to confirm processing
+        results.push({ entityName: o.entityName, addedObservations: [] })
+      }
+    })
+    if (changed) {
+      await this._persistGraph()
+    }
+    return results
+  }
+
+  async deleteEntities(entityNames: string[]): Promise<void> {
+    let changed = false
+    const namesToDelete = new Set(entityNames)
+
+    // Delete entities
+    namesToDelete.forEach((name) => {
+      if (this.entities.delete(name)) {
+        changed = true
+      }
+    })
+
+    // Delete relations involving deleted entities
+    const relationsToDelete = new Set<string>()
+    this.relations.forEach((relStr) => {
+      const rel = this._deserializeRelation(relStr)
+      if (namesToDelete.has(rel.from) || namesToDelete.has(rel.to)) {
+        relationsToDelete.add(relStr)
+      }
+    })
+
+    relationsToDelete.forEach((relStr) => {
+      if (this.relations.delete(relStr)) {
+        changed = true
+      }
+    })
+
+    if (changed) {
+      await this._persistGraph()
+    }
+  }
+
+  async deleteObservations(deletions: { entityName: string; observations: string[] }[]): Promise<void> {
+    let changed = false
+    deletions.forEach((d) => {
+      const entity = this.entities.get(d.entityName)
+      if (entity && Array.isArray(entity.observations)) {
+        const initialLength = entity.observations.length
+        const observationsToDelete = new Set(d.observations)
+        entity.observations = entity.observations.filter((o) => !observationsToDelete.has(o))
+        if (entity.observations.length !== initialLength) {
+          changed = true
+        }
+      }
+    })
+    if (changed) {
+      await this._persistGraph()
+    }
+  }
+
+  async deleteRelations(relations: Relation[]): Promise<void> {
+    let changed = false
+    relations.forEach((rel) => {
+      const relStr = this._serializeRelation(rel)
+      if (this.relations.delete(relStr)) {
+        changed = true
+      }
+    })
+    if (changed) {
+      await this._persistGraph()
+    }
+  }
+
+  // Read the current state from memory
+  async readGraph(): Promise<KnowledgeGraph> {
+    // Return a deep copy to prevent external modification of the internal state
+    return JSON.parse(
+      JSON.stringify({
+        entities: Array.from(this.entities.values()),
+        relations: Array.from(this.relations).map((rStr) => this._deserializeRelation(rStr))
+      })
+    )
+  }
+
+  // Search operates on the in-memory graph
+  async searchNodes(query: string): Promise<KnowledgeGraph> {
+    const lowerCaseQuery = query.toLowerCase()
+    const filteredEntities = Array.from(this.entities.values()).filter(
+      (e) =>
+        e.name.toLowerCase().includes(lowerCaseQuery) ||
+        e.entityType.toLowerCase().includes(lowerCaseQuery) ||
+        (Array.isArray(e.observations) && e.observations.some((o) => o.toLowerCase().includes(lowerCaseQuery)))
+    )
+
+    const filteredEntityNames = new Set(filteredEntities.map((e) => e.name))
+
+    const filteredRelations = Array.from(this.relations)
+      .map((rStr) => this._deserializeRelation(rStr))
+      .filter((r) => filteredEntityNames.has(r.from) && filteredEntityNames.has(r.to))
+
+    return {
+      entities: filteredEntities,
+      relations: filteredRelations
+    }
+  }
+
+  // Open operates on the in-memory graph
+  async openNodes(names: string[]): Promise<KnowledgeGraph> {
+    const nameSet = new Set(names)
+    const filteredEntities = Array.from(this.entities.values()).filter((e) => nameSet.has(e.name))
+    const filteredEntityNames = new Set(filteredEntities.map((e) => e.name))
+
+    const filteredRelations = Array.from(this.relations)
+      .map((rStr) => this._deserializeRelation(rStr))
+      .filter((r) => filteredEntityNames.has(r.from) && filteredEntityNames.has(r.to))
+
+    return {
+      entities: filteredEntities,
+      relations: filteredRelations
+    }
+  }
+}
+
+class MemoryServer {
+  public server: Server
+  // Hold the manager instance, initialized asynchronously
+  private knowledgeGraphManager: KnowledgeGraphManager | null = null
+  private initializationPromise: Promise<void> // To track initialization
+
+  constructor(envPath: string = '') {
+    const memoryPath = envPath
+      ? path.isAbsolute(envPath)
+        ? envPath
+        : path.resolve(envPath) // Use path.resolve for relative paths based on CWD
+      : defaultMemoryPath
+
+    this.server = new Server(
+      {
+        name: 'memory-server',
+        version: '1.1.0' // Incremented version for changes
+      },
+      {
+        capabilities: {
+          tools: {}
+        }
+      }
+    )
+    // Start initialization, but don't block constructor
+    this.initializationPromise = this._initializeManager(memoryPath)
+    this.setupRequestHandlers() // Setup handlers immediately
+  }
+
+  // Private async method to handle manager initialization
+  private async _initializeManager(memoryPath: string): Promise<void> {
+    try {
+      this.knowledgeGraphManager = await KnowledgeGraphManager.create(memoryPath)
+      console.log('KnowledgeGraphManager initialized successfully.')
+    } catch (error) {
+      console.error('Failed to initialize KnowledgeGraphManager:', error)
+      // Server might be unusable, consider how to handle this state
+      // Maybe set a flag and return errors for all tool calls?
+      this.knowledgeGraphManager = null // Ensure it's null if init fails
+    }
+  }
+
+  // Ensures the manager is initialized before handling tool calls
+  private async _getManager(): Promise<KnowledgeGraphManager> {
+    await this.initializationPromise // Wait for initialization to complete
+    if (!this.knowledgeGraphManager) {
+      throw new McpError(ErrorCode.InternalError, 'Memory server failed to initialize. Cannot process requests.')
+    }
+    return this.knowledgeGraphManager
+  }
+
+  // Setup handlers (can be called from constructor)
+  setupRequestHandlers() {
+    // ListTools remains largely the same, descriptions might be updated if needed
+    this.server.setRequestHandler(ListToolsRequestSchema, async () => {
+      // Ensure manager is ready before listing tools that depend on it
+      // Although ListTools itself doesn't *call* the manager, it implies the
+      // manager is ready to handle calls for those tools.
+      try {
+        await this._getManager() // Wait for initialization before confirming tools are available
+      } catch (error) {
+        // If manager failed to init, maybe return an empty tool list or throw?
+        console.error('Cannot list tools, manager initialization failed:', error)
+        return { tools: [] } // Return empty list if server is not ready
+      }
+
+      return {
+        tools: [
+          {
+            name: 'create_entities',
+            description: 'Create multiple new entities in the knowledge graph. Skips existing entities.',
+            inputSchema: {
+              type: 'object',
+              properties: {
+                entities: {
+                  type: 'array',
+                  items: {
+                    type: 'object',
+                    properties: {
+                      name: { type: 'string', description: 'The name of the entity' },
+                      entityType: { type: 'string', description: 'The type of the entity' },
+                      observations: {
+                        type: 'array',
+                        items: { type: 'string' },
+                        description: 'An array of observation contents associated with the entity',
+                        default: [] // Add default empty array
+                      }
+                    },
+                    required: ['name', 'entityType'] // Observations are optional now on creation
+                  }
+                }
+              },
+              required: ['entities']
+            }
+          },
+          {
+            name: 'create_relations',
+            description:
+              'Create multiple new relations between EXISTING entities. Skips existing relations or relations with non-existent entities.',
+            inputSchema: {
+              type: 'object',
+              properties: {
+                relations: {
+                  type: 'array',
+                  items: {
+                    type: 'object',
+                    properties: {
+                      from: { type: 'string', description: 'The name of the entity where the relation starts' },
+                      to: { type: 'string', description: 'The name of the entity where the relation ends' },
+                      relationType: { type: 'string', description: 'The type of the relation' }
+                    },
+                    required: ['from', 'to', 'relationType']
+                  }
+                }
+              },
+              required: ['relations']
+            }
+          },
+          {
+            name: 'add_observations',
+            description: 'Add new observations to existing entities. Skips duplicate observations.',
+            inputSchema: {
+              type: 'object',
+              properties: {
+                observations: {
+                  type: 'array',
+                  items: {
+                    type: 'object',
+                    properties: {
+                      entityName: { type: 'string', description: 'The name of the entity to add the observations to' },
+                      contents: {
+                        type: 'array',
+                        items: { type: 'string' },
+                        description: 'An array of observation contents to add'
+                      }
+                    },
+                    required: ['entityName', 'contents']
+                  }
+                }
+              },
+              required: ['observations']
+            }
+          },
+          {
+            name: 'delete_entities',
+            description: 'Delete multiple entities and their associated relations.',
+            inputSchema: {
+              type: 'object',
+              properties: {
+                entityNames: {
+                  type: 'array',
+                  items: { type: 'string' },
+                  description: 'An array of entity names to delete'
+                }
+              },
+              required: ['entityNames']
+            }
+          },
+          {
+            name: 'delete_observations',
+            description: 'Delete specific observations from entities.',
+            inputSchema: {
+              type: 'object',
+              properties: {
+                deletions: {
+                  type: 'array',
+                  items: {
+                    type: 'object',
+                    properties: {
+                      entityName: { type: 'string', description: 'The name of the entity containing the observations' },
+                      observations: {
+                        type: 'array',
+                        items: { type: 'string' },
+                        description: 'An array of observations to delete'
+                      }
+                    },
+                    required: ['entityName', 'observations']
+                  }
+                }
+              },
+              required: ['deletions']
+            }
+          },
+          {
+            name: 'delete_relations',
+            description: 'Delete multiple specific relations.',
+            inputSchema: {
+              type: 'object',
+              properties: {
+                relations: {
+                  type: 'array',
+                  items: {
+                    type: 'object',
+                    properties: {
+                      from: { type: 'string', description: 'The name of the entity where the relation starts' },
+                      to: { type: 'string', description: 'The name of the entity where the relation ends' },
+                      relationType: { type: 'string', description: 'The type of the relation' }
+                    },
+                    required: ['from', 'to', 'relationType']
+                  },
+                  description: 'An array of relations to delete'
+                }
+              },
+              required: ['relations']
+            }
+          },
+          {
+            name: 'read_graph',
+            description: 'Read the entire knowledge graph from memory.',
+            inputSchema: {
+              type: 'object',
+              properties: {}
+            }
+          },
+          {
+            name: 'search_nodes',
+            description: 'Search nodes (entities and relations) in memory based on a query.',
+            inputSchema: {
+              type: 'object',
+              properties: {
+                query: {
+                  type: 'string',
+                  description: 'The search query to match against entity names, types, and observation content'
+                }
+              },
+              required: ['query']
+            }
+          },
+          {
+            name: 'open_nodes',
+            description: 'Retrieve specific entities and their connecting relations from memory by name.',
+            inputSchema: {
+              type: 'object',
+              properties: {
+                names: {
+                  type: 'array',
+                  items: { type: 'string' },
+                  description: 'An array of entity names to retrieve'
+                }
+              },
+              required: ['names']
+            }
+          }
+        ]
+      }
+    })
+
+    // CallTool handler needs to await the manager and the async methods
+    this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
+      const manager = await this._getManager() // Ensure manager is ready
+      const { name, arguments: args } = request.params
+
+      if (!args) {
+        // Use McpError for standard errors
+        throw new McpError(ErrorCode.InvalidParams, `No arguments provided for tool: ${name}`)
+      }
+
+      try {
+        switch (name) {
+          case 'create_entities':
+            // Validate args structure if necessary, though SDK might do basic validation
+            if (!args.entities || !Array.isArray(args.entities)) {
+              throw new McpError(
+                ErrorCode.InvalidParams,
+                `Invalid arguments for ${name}: 'entities' array is required.`
+              )
+            }
+            return {
+              content: [
+                { type: 'text', text: JSON.stringify(await manager.createEntities(args.entities as Entity[]), null, 2) }
+              ]
+            }
+          case 'create_relations':
+            if (!args.relations || !Array.isArray(args.relations)) {
+              throw new McpError(
+                ErrorCode.InvalidParams,
+                `Invalid arguments for ${name}: 'relations' array is required.`
+              )
+            }
+            return {
+              content: [
+                {
+                  type: 'text',
+                  text: JSON.stringify(await manager.createRelations(args.relations as Relation[]), null, 2)
+                }
+              ]
+            }
+          case 'add_observations':
+            if (!args.observations || !Array.isArray(args.observations)) {
+              throw new McpError(
+                ErrorCode.InvalidParams,
+                `Invalid arguments for ${name}: 'observations' array is required.`
+              )
+            }
+            return {
+              content: [
+                {
+                  type: 'text',
+                  text: JSON.stringify(
+                    await manager.addObservations(args.observations as { entityName: string; contents: string[] }[]),
+                    null,
+                    2
+                  )
+                }
+              ]
+            }
+          case 'delete_entities':
+            if (!args.entityNames || !Array.isArray(args.entityNames)) {
+              throw new McpError(
+                ErrorCode.InvalidParams,
+                `Invalid arguments for ${name}: 'entityNames' array is required.`
+              )
+            }
+            await manager.deleteEntities(args.entityNames as string[])
+            return { content: [{ type: 'text', text: 'Entities deleted successfully' }] }
+          case 'delete_observations':
+            if (!args.deletions || !Array.isArray(args.deletions)) {
+              throw new McpError(
+                ErrorCode.InvalidParams,
+                `Invalid arguments for ${name}: 'deletions' array is required.`
+              )
+            }
+            await manager.deleteObservations(args.deletions as { entityName: string; observations: string[] }[])
+            return { content: [{ type: 'text', text: 'Observations deleted successfully' }] }
+          case 'delete_relations':
+            if (!args.relations || !Array.isArray(args.relations)) {
+              throw new McpError(
+                ErrorCode.InvalidParams,
+                `Invalid arguments for ${name}: 'relations' array is required.`
+              )
+            }
+            await manager.deleteRelations(args.relations as Relation[])
+            return { content: [{ type: 'text', text: 'Relations deleted successfully' }] }
+          case 'read_graph':
+            // No arguments expected or needed for read_graph based on original schema
+            return {
+              content: [{ type: 'text', text: JSON.stringify(await manager.readGraph(), null, 2) }]
+            }
+          case 'search_nodes':
+            if (typeof args.query !== 'string') {
+              throw new McpError(ErrorCode.InvalidParams, `Invalid arguments for ${name}: 'query' string is required.`)
+            }
+            return {
+              content: [
+                { type: 'text', text: JSON.stringify(await manager.searchNodes(args.query as string), null, 2) }
+              ]
+            }
+          case 'open_nodes':
+            if (!args.names || !Array.isArray(args.names)) {
+              throw new McpError(ErrorCode.InvalidParams, `Invalid arguments for ${name}: 'names' array is required.`)
+            }
+            return {
+              content: [
+                { type: 'text', text: JSON.stringify(await manager.openNodes(args.names as string[]), null, 2) }
+              ]
+            }
+          default:
+            throw new McpError(ErrorCode.MethodNotFound, `Unknown tool: ${name}`)
+        }
+      } catch (error) {
+        // Catch errors from manager methods (like entity not found) or other issues
+        if (error instanceof McpError) {
+          throw error // Re-throw McpErrors directly
+        }
+        console.error(`Error executing tool ${name}:`, error)
+        // Throw a generic internal error for unexpected issues
+        throw new McpError(
+          ErrorCode.InternalError,
+          `Error executing tool ${name}: ${error instanceof Error ? error.message : String(error)}`
+        )
+      }
+    })
+  }
+}
+
+export default MemoryServer

src/main/mcpServers/sequentialthinking.ts

@@ -0,0 +1,289 @@
+// Sequential Thinking MCP Server
+// port https://github.com/modelcontextprotocol/servers/blob/main/src/sequentialthinking/index.ts
+
+import { Server } from '@modelcontextprotocol/sdk/server/index.js'
+import { CallToolRequestSchema, ListToolsRequestSchema, Tool } from '@modelcontextprotocol/sdk/types.js'
+// Fixed chalk import for ESM
+import chalk from 'chalk'
+
+interface ThoughtData {
+  thought: string
+  thoughtNumber: number
+  totalThoughts: number
+  isRevision?: boolean
+  revisesThought?: number
+  branchFromThought?: number
+  branchId?: string
+  needsMoreThoughts?: boolean
+  nextThoughtNeeded: boolean
+}
+
+class SequentialThinkingServer {
+  private thoughtHistory: ThoughtData[] = []
+  private branches: Record<string, ThoughtData[]> = {}
+
+  private validateThoughtData(input: unknown): ThoughtData {
+    const data = input as Record<string, unknown>
+
+    if (!data.thought || typeof data.thought !== 'string') {
+      throw new Error('Invalid thought: must be a string')
+    }
+    if (!data.thoughtNumber || typeof data.thoughtNumber !== 'number') {
+      throw new Error('Invalid thoughtNumber: must be a number')
+    }
+    if (!data.totalThoughts || typeof data.totalThoughts !== 'number') {
+      throw new Error('Invalid totalThoughts: must be a number')
+    }
+    if (typeof data.nextThoughtNeeded !== 'boolean') {
+      throw new Error('Invalid nextThoughtNeeded: must be a boolean')
+    }
+
+    return {
+      thought: data.thought,
+      thoughtNumber: data.thoughtNumber,
+      totalThoughts: data.totalThoughts,
+      nextThoughtNeeded: data.nextThoughtNeeded,
+      isRevision: data.isRevision as boolean | undefined,
+      revisesThought: data.revisesThought as number | undefined,
+      branchFromThought: data.branchFromThought as number | undefined,
+      branchId: data.branchId as string | undefined,
+      needsMoreThoughts: data.needsMoreThoughts as boolean | undefined
+    }
+  }
+
+  private formatThought(thoughtData: ThoughtData): string {
+    const { thoughtNumber, totalThoughts, thought, isRevision, revisesThought, branchFromThought, branchId } =
+      thoughtData
+
+    let prefix = ''
+    let context = ''
+
+    if (isRevision) {
+      prefix = chalk.yellow('🔄 Revision')
+      context = ` (revising thought ${revisesThought})`
+    } else if (branchFromThought) {
+      prefix = chalk.green('🌿 Branch')
+      context = ` (from thought ${branchFromThought}, ID: ${branchId})`
+    } else {
+      prefix = chalk.blue('💭 Thought')
+      context = ''
+    }
+
+    const header = `${prefix} ${thoughtNumber}/${totalThoughts}${context}`
+    const border = '─'.repeat(Math.max(header.length, thought.length) + 4)
+
+    return `
+┌${border}┐
+│ ${header} │
+├${border}┤
+│ ${thought.padEnd(border.length - 2)} │
+└${border}┘`
+  }
+
+  public processThought(input: unknown): { content: Array<{ type: string; text: string }>; isError?: boolean } {
+    try {
+      const validatedInput = this.validateThoughtData(input)
+
+      if (validatedInput.thoughtNumber > validatedInput.totalThoughts) {
+        validatedInput.totalThoughts = validatedInput.thoughtNumber
+      }
+
+      this.thoughtHistory.push(validatedInput)
+
+      if (validatedInput.branchFromThought && validatedInput.branchId) {
+        if (!this.branches[validatedInput.branchId]) {
+          this.branches[validatedInput.branchId] = []
+        }
+        this.branches[validatedInput.branchId].push(validatedInput)
+      }
+
+      const formattedThought = this.formatThought(validatedInput)
+      console.error(formattedThought)
+
+      return {
+        content: [
+          {
+            type: 'text',
+            text: JSON.stringify(
+              {
+                thoughtNumber: validatedInput.thoughtNumber,
+                totalThoughts: validatedInput.totalThoughts,
+                nextThoughtNeeded: validatedInput.nextThoughtNeeded,
+                branches: Object.keys(this.branches),
+                thoughtHistoryLength: this.thoughtHistory.length
+              },
+              null,
+              2
+            )
+          }
+        ]
+      }
+    } catch (error) {
+      return {
+        content: [
+          {
+            type: 'text',
+            text: JSON.stringify(
+              {
+                error: error instanceof Error ? error.message : String(error),
+                status: 'failed'
+              },
+              null,
+              2
+            )
+          }
+        ],
+        isError: true
+      }
+    }
+  }
+}
+
+const SEQUENTIAL_THINKING_TOOL: Tool = {
+  name: 'sequentialthinking',
+  description: `A detailed tool for dynamic and reflective problem-solving through thoughts.
+This tool helps analyze problems through a flexible thinking process that can adapt and evolve.
+Each thought can build on, question, or revise previous insights as understanding deepens.
+
+When to use this tool:
+- Breaking down complex problems into steps
+- Planning and design with room for revision
+- Analysis that might need course correction
+- Problems where the full scope might not be clear initially
+- Problems that require a multi-step solution
+- Tasks that need to maintain context over multiple steps
+- Situations where irrelevant information needs to be filtered out
+
+Key features:
+- You can adjust total_thoughts up or down as you progress
+- You can question or revise previous thoughts
+- You can add more thoughts even after reaching what seemed like the end
+- You can express uncertainty and explore alternative approaches
+- Not every thought needs to build linearly - you can branch or backtrack
+- Generates a solution hypothesis
+- Verifies the hypothesis based on the Chain of Thought steps
+- Repeats the process until satisfied
+- Provides a correct answer
+
+Parameters explained:
+- thought: Your current thinking step, which can include:
+* Regular analytical steps
+* Revisions of previous thoughts
+* Questions about previous decisions
+* Realizations about needing more analysis
+* Changes in approach
+* Hypothesis generation
+* Hypothesis verification
+- next_thought_needed: True if you need more thinking, even if at what seemed like the end
+- thought_number: Current number in sequence (can go beyond initial total if needed)
+- total_thoughts: Current estimate of thoughts needed (can be adjusted up/down)
+- is_revision: A boolean indicating if this thought revises previous thinking
+- revises_thought: If is_revision is true, which thought number is being reconsidered
+- branch_from_thought: If branching, which thought number is the branching point
+- branch_id: Identifier for the current branch (if any)
+- needs_more_thoughts: If reaching end but realizing more thoughts needed
+
+You should:
+1. Start with an initial estimate of needed thoughts, but be ready to adjust
+2. Feel free to question or revise previous thoughts
+3. Don't hesitate to add more thoughts if needed, even at the "end"
+4. Express uncertainty when present
+5. Mark thoughts that revise previous thinking or branch into new paths
+6. Ignore information that is irrelevant to the current step
+7. Generate a solution hypothesis when appropriate
+8. Verify the hypothesis based on the Chain of Thought steps
+9. Repeat the process until satisfied with the solution
+10. Provide a single, ideally correct answer as the final output
+11. Only set next_thought_needed to false when truly done and a satisfactory answer is reached`,
+  inputSchema: {
+    type: 'object',
+    properties: {
+      thought: {
+        type: 'string',
+        description: 'Your current thinking step'
+      },
+      nextThoughtNeeded: {
+        type: 'boolean',
+        description: 'Whether another thought step is needed'
+      },
+      thoughtNumber: {
+        type: 'integer',
+        description: 'Current thought number',
+        minimum: 1
+      },
+      totalThoughts: {
+        type: 'integer',
+        description: 'Estimated total thoughts needed',
+        minimum: 1
+      },
+      isRevision: {
+        type: 'boolean',
+        description: 'Whether this revises previous thinking'
+      },
+      revisesThought: {
+        type: 'integer',
+        description: 'Which thought is being reconsidered',
+        minimum: 1
+      },
+      branchFromThought: {
+        type: 'integer',
+        description: 'Branching point thought number',
+        minimum: 1
+      },
+      branchId: {
+        type: 'string',
+        description: 'Branch identifier'
+      },
+      needsMoreThoughts: {
+        type: 'boolean',
+        description: 'If more thoughts are needed'
+      }
+    },
+    required: ['thought', 'nextThoughtNeeded', 'thoughtNumber', 'totalThoughts']
+  }
+}
+
+class ThinkingServer {
+  public server: Server
+  private thinkingServer: SequentialThinkingServer
+
+  constructor() {
+    this.thinkingServer = new SequentialThinkingServer()
+    this.server = new Server(
+      {
+        name: 'sequential-thinking-server',
+        version: '0.2.0'
+      },
+      {
+        capabilities: {
+          tools: {}
+        }
+      }
+    )
+    this.initialize()
+  }
+
+  initialize() {
+    this.server.setRequestHandler(ListToolsRequestSchema, async () => ({
+      tools: [SEQUENTIAL_THINKING_TOOL]
+    }))
+
+    this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
+      if (request.params.name === 'sequentialthinking') {
+        return this.thinkingServer.processThought(request.params.arguments)
+      }
+
+      return {
+        content: [
+          {
+            type: 'text',
+            text: `Unknown tool: ${request.params.name}`
+          }
+        ],
+        isError: true
+      }
+    })
+  }
+}
+
+export default ThinkingServer

src/main/mcpServers/simpleremember.ts

@@ -0,0 +1,321 @@
+// src/main/mcpServers/simpleremember.ts
+import { getConfigDir } from '@main/utils/file'
+import { Server } from '@modelcontextprotocol/sdk/server/index.js'
+import {
+  CallToolRequestSchema,
+  ErrorCode,
+  ListPromptsRequestSchema,
+  ListToolsRequestSchema,
+  McpError
+} from '@modelcontextprotocol/sdk/types.js'
+import { Mutex } from 'async-mutex'
+import { promises as fs } from 'fs'
+import path from 'path'
+
+// 定义记忆文件路径
+const defaultMemoryPath = path.join(getConfigDir(), 'simpleremember.json')
+
+// 记忆项接口
+interface Memory {
+  content: string
+  createdAt: string
+}
+
+// 记忆存储结构
+interface MemoryStorage {
+  memories: Memory[]
+}
+
+class SimpleRememberManager {
+  private memoryPath: string
+  private memories: Memory[] = []
+  private fileMutex: Mutex = new Mutex()
+
+  constructor(memoryPath: string) {
+    this.memoryPath = memoryPath
+  }
+
+  // 静态工厂方法用于初始化
+  public static async create(memoryPath: string): Promise<SimpleRememberManager> {
+    const manager = new SimpleRememberManager(memoryPath)
+    await manager._ensureMemoryPathExists()
+    await manager._loadMemoriesFromDisk()
+    return manager
+  }
+
+  // 确保记忆文件存在
+  private async _ensureMemoryPathExists(): Promise<void> {
+    try {
+      const directory = path.dirname(this.memoryPath)
+      await fs.mkdir(directory, { recursive: true })
+      try {
+        await fs.access(this.memoryPath)
+      } catch (error) {
+        // 文件不存在，创建一个空文件
+        await fs.writeFile(this.memoryPath, JSON.stringify({ memories: [] }, null, 2))
+      }
+    } catch (error) {
+      console.error('Failed to ensure memory path exists:', error)
+      throw new McpError(
+        ErrorCode.InternalError,
+        `Failed to ensure memory path: ${error instanceof Error ? error.message : String(error)}`
+      )
+    }
+  }
+
+  // 从磁盘加载记忆
+  private async _loadMemoriesFromDisk(): Promise<void> {
+    try {
+      const data = await fs.readFile(this.memoryPath, 'utf-8')
+      // 处理空文件情况
+      if (data.trim() === '') {
+        this.memories = []
+        await this._persistMemories()
+        return
+      }
+      const storage: MemoryStorage = JSON.parse(data)
+      this.memories = storage.memories || []
+    } catch (error) {
+      if (error instanceof Error && 'code' in error && (error as any).code === 'ENOENT') {
+        this.memories = []
+        await this._persistMemories()
+      } else if (error instanceof SyntaxError) {
+        console.error('Failed to parse simpleremember.json, initializing with empty memories:', error)
+        this.memories = []
+        await this._persistMemories()
+      } else {
+        console.error('Unexpected error loading memories:', error)
+        throw new McpError(
+          ErrorCode.InternalError,
+          `Failed to load memories: ${error instanceof Error ? error.message : String(error)}`
+        )
+      }
+    }
+  }
+
+  // 将记忆持久化到磁盘
+  private async _persistMemories(): Promise<void> {
+    const release = await this.fileMutex.acquire()
+    try {
+      const storage: MemoryStorage = {
+        memories: this.memories
+      }
+      await fs.writeFile(this.memoryPath, JSON.stringify(storage, null, 2))
+    } catch (error) {
+      console.error('Failed to save memories:', error)
+      throw new McpError(
+        ErrorCode.InternalError,
+        `Failed to save memories: ${error instanceof Error ? error.message : String(error)}`
+      )
+    } finally {
+      release()
+    }
+  }
+
+  // 添加新记忆
+  async remember(memory: string): Promise<Memory> {
+    const newMemory: Memory = {
+      content: memory,
+      createdAt: new Date().toISOString()
+    }
+    this.memories.push(newMemory)
+    await this._persistMemories()
+    return newMemory
+  }
+
+  // 获取所有记忆
+  async getAllMemories(): Promise<Memory[]> {
+    return [...this.memories]
+  }
+
+  // 获取记忆 - 这个方法会被get_memories工具调用
+  async get_memories(): Promise<Memory[]> {
+    return this.getAllMemories()
+  }
+}
+
+// 定义工具 - 按照MCP规范定义工具
+const REMEMBER_TOOL = {
+  name: 'remember',
+  description:
+    '用于记忆长期有用信息的工具。这个工具会自动应用记忆，无需显式调用。只用于存储长期有用的信息，不适合临时信息。',
+  inputSchema: {
+    type: 'object',
+    properties: {
+      memory: {
+        type: 'string',
+        description: '要记住的简洁(1句话)记忆内容'
+      }
+    },
+    required: ['memory']
+  }
+}
+
+const GET_MEMORIES_TOOL = {
+  name: 'get_memories',
+  description: '获取所有已存储的记忆',
+  inputSchema: {
+    type: 'object',
+    properties: {}
+  }
+}
+
+// 添加日志以便调试
+console.log('[SimpleRemember] Defined tools:', { REMEMBER_TOOL, GET_MEMORIES_TOOL })
+
+class SimpleRememberServer {
+  public server: Server
+  private simpleRememberManager: SimpleRememberManager | null = null
+  private initializationPromise: Promise<void>
+
+  constructor(envPath: string = '') {
+    const memoryPath = envPath ? (path.isAbsolute(envPath) ? envPath : path.resolve(envPath)) : defaultMemoryPath
+
+    console.log('[SimpleRemember] Creating server with memory path:', memoryPath)
+
+    // 初始化服务器
+    this.server = new Server(
+      {
+        name: 'simple-remember-server',
+        version: '1.0.0'
+      },
+      {
+        capabilities: {
+          tools: {
+            // 按照MCP规范声明工具能力
+            listChanged: true
+          },
+          // 添加空的prompts能力，表示支持提示词功能但没有实际的提示词
+          prompts: {}
+        }
+      }
+    )
+
+    console.log('[SimpleRemember] Server initialized with tools capability')
+
+    // 手动添加工具到服务器的工具列表中
+    console.log('[SimpleRemember] Adding tools to server')
+
+    // 先设置请求处理程序，再初始化管理器
+    this.setupRequestHandlers()
+    this.initializationPromise = this._initializeManager(memoryPath)
+
+    console.log('[SimpleRemember] Server initialization complete')
+    // 打印工具信息以确认它们已注册
+    console.log('[SimpleRemember] Tools registered:', [REMEMBER_TOOL.name, GET_MEMORIES_TOOL.name])
+  }
+
+  private async _initializeManager(memoryPath: string): Promise<void> {
+    try {
+      this.simpleRememberManager = await SimpleRememberManager.create(memoryPath)
+      console.log('SimpleRememberManager initialized successfully.')
+    } catch (error) {
+      console.error('Failed to initialize SimpleRememberManager:', error)
+      this.simpleRememberManager = null
+    }
+  }
+
+  private async _getManager(): Promise<SimpleRememberManager> {
+    if (!this.simpleRememberManager) {
+      await this.initializationPromise
+      if (!this.simpleRememberManager) {
+        throw new McpError(ErrorCode.InternalError, 'SimpleRememberManager is not initialized')
+      }
+    }
+    return this.simpleRememberManager
+  }
+
+  setupRequestHandlers() {
+    // 添加对prompts/list请求的处理
+    this.server.setRequestHandler(ListPromptsRequestSchema, async (request) => {
+      console.log('[SimpleRemember] Listing prompts request received', request)
+
+      // 返回空的提示词列表
+      return {
+        prompts: []
+      }
+    })
+
+    this.server.setRequestHandler(ListToolsRequestSchema, async (request) => {
+      // 直接返回工具列表，不需要等待管理器初始化
+      console.log('[SimpleRemember] Listing tools request received', request)
+
+      // 打印工具定义以确保它们存在
+      console.log('[SimpleRemember] REMEMBER_TOOL:', JSON.stringify(REMEMBER_TOOL))
+      console.log('[SimpleRemember] GET_MEMORIES_TOOL:', JSON.stringify(GET_MEMORIES_TOOL))
+
+      const toolsList = [REMEMBER_TOOL, GET_MEMORIES_TOOL]
+      console.log('[SimpleRemember] Returning tools:', JSON.stringify(toolsList))
+
+      // 按照MCP规范返回工具列表
+      return {
+        tools: toolsList
+        // 如果有分页，可以添加nextCursor
+        // nextCursor: "next-page-cursor"
+      }
+    })
+
+    this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
+      const { name, arguments: args } = request.params
+
+      console.log(`[SimpleRemember] Received tool call: ${name}`, args)
+
+      try {
+        const manager = await this._getManager()
+
+        if (name === 'remember') {
+          if (!args || typeof args.memory !== 'string') {
+            console.error(`[SimpleRemember] Invalid arguments for ${name}:`, args)
+            throw new McpError(ErrorCode.InvalidParams, `Invalid arguments for ${name}: 'memory' string is required.`)
+          }
+          console.log(`[SimpleRemember] Remembering: "${args.memory}"`)
+          const result = await manager.remember(args.memory)
+          console.log(`[SimpleRemember] Memory saved successfully:`, result)
+          // 按照MCP规范返回工具调用结果
+          return {
+            content: [
+              {
+                type: 'text',
+                text: `记忆已保存: "${args.memory}"`
+              }
+            ],
+            isError: false
+          }
+        }
+
+        if (name === 'get_memories') {
+          console.log(`[SimpleRemember] Getting all memories`)
+          const memories = await manager.get_memories()
+          console.log(`[SimpleRemember] Retrieved ${memories.length} memories`)
+          // 按照MCP规范返回工具调用结果
+          return {
+            content: [
+              {
+                type: 'text',
+                text: JSON.stringify(memories, null, 2)
+              }
+            ],
+            isError: false
+          }
+        }
+
+        console.error(`[SimpleRemember] Unknown tool: ${name}`)
+        throw new McpError(ErrorCode.MethodNotFound, `Unknown tool: ${name}`)
+      } catch (error) {
+        console.error(`[SimpleRemember] Error handling tool call ${name}:`, error)
+        // 按照MCP规范返回工具调用错误
+        return {
+          content: [
+            {
+              type: 'text',
+              text: error instanceof Error ? error.message : String(error)
+            }
+          ],
+          isError: true
+        }
+      }
+    })
+  }
+}
+
+export default SimpleRememberServer

src/main/reranker/BaseReranker.ts

@@ -0,0 +1,77 @@
+import type { ExtractChunkData } from '@cherrystudio/embedjs-interfaces'
+import { KnowledgeBaseParams } from '@types'
+
+export default abstract class BaseReranker {
+  protected base: KnowledgeBaseParams
+
+  constructor(base: KnowledgeBaseParams) {
+    if (!base.rerankModel) {
+      throw new Error('Rerank model is required')
+    }
+    this.base = base
+  }
+
+  abstract rerank(query: string, searchResults: ExtractChunkData[]): Promise<ExtractChunkData[]>
+
+  /**
+   * Get Rerank Request Url
+   */
+  protected getRerankUrl() {
+    let baseURL = this.base?.rerankBaseURL?.endsWith('/')
+      ? this.base.rerankBaseURL.slice(0, -1)
+      : this.base.rerankBaseURL
+    // 必须携带/v1，否则会404
+    if (baseURL && !baseURL.endsWith('/v1')) {
+      baseURL = `${baseURL}/v1`
+    }
+
+    return `${baseURL}/rerank`
+  }
+
+  /**
+   * Get Rerank Result
+   * @param searchResults
+   * @param rerankResults
+   * @protected
+   */
+  protected getRerankResult(
+    searchResults: ExtractChunkData[],
+    rerankResults: Array<{
+      index: number
+      relevance_score: number
+    }>
+  ) {
+    const resultMap = new Map(rerankResults.map((result) => [result.index, result.relevance_score || 0]))
+
+    return searchResults
+      .map((doc: ExtractChunkData, index: number) => {
+        const score = resultMap.get(index)
+        if (score === undefined) return undefined
+
+        return {
+          ...doc,
+          score
+        }
+      })
+      .filter((doc): doc is ExtractChunkData => doc !== undefined)
+      .sort((a, b) => b.score - a.score)
+  }
+
+  public defaultHeaders() {
+    return {
+      Authorization: `Bearer ${this.base.rerankApiKey}`,
+      'Content-Type': 'application/json'
+    }
+  }
+
+  protected formatErrorMessage(url: string, error: any, requestBody: any) {
+    const errorDetails = {
+      url: url,
+      message: error.message,
+      status: error.response?.status,
+      statusText: error.response?.statusText,
+      requestBody: requestBody
+    }
+    return JSON.stringify(errorDetails, null, 2)
+  }
+}

src/main/reranker/DefaultReranker.ts

@@ -0,0 +1,14 @@
+import type { ExtractChunkData } from '@cherrystudio/embedjs-interfaces'
+import { KnowledgeBaseParams } from '@types'
+
+import BaseReranker from './BaseReranker'
+
+export default class DefaultReranker extends BaseReranker {
+  constructor(base: KnowledgeBaseParams) {
+    super(base)
+  }
+
+  async rerank(): Promise<ExtractChunkData[]> {
+    throw new Error('Method not implemented.')
+  }
+}

src/main/reranker/JinaReranker.ts

@@ -0,0 +1,33 @@
+import { ExtractChunkData } from '@cherrystudio/embedjs-interfaces'
+import { KnowledgeBaseParams } from '@types'
+import axios from 'axios'
+
+import BaseReranker from './BaseReranker'
+
+export default class JinaReranker extends BaseReranker {
+  constructor(base: KnowledgeBaseParams) {
+    super(base)
+  }
+
+  public rerank = async (query: string, searchResults: ExtractChunkData[]): Promise<ExtractChunkData[]> => {
+    const url = this.getRerankUrl()
+
+    const requestBody = {
+      model: this.base.rerankModel,
+      query,
+      documents: searchResults.map((doc) => doc.pageContent),
+      top_n: this.base.topN
+    }
+
+    try {
+      const { data } = await axios.post(url, requestBody, { headers: this.defaultHeaders() })
+
+      const rerankResults = data.results
+      return this.getRerankResult(searchResults, rerankResults)
+    } catch (error: any) {
+      const errorDetails = this.formatErrorMessage(url, error, requestBody)
+      console.error('Jina Reranker API Error:', errorDetails)
+      throw new Error(`重排序请求失败: ${error.message}\n请求详情: ${errorDetails}`)
+    }
+  }
+}

src/main/reranker/Reranker.ts

@@ -0,0 +1,15 @@
+import type { ExtractChunkData } from '@cherrystudio/embedjs-interfaces'
+import { KnowledgeBaseParams } from '@types'
+
+import BaseReranker from './BaseReranker'
+import RerankerFactory from './RerankerFactory'
+
+export default class Reranker {
+  private sdk: BaseReranker
+  constructor(base: KnowledgeBaseParams) {
+    this.sdk = RerankerFactory.create(base)
+  }
+  public async rerank(query: string, searchResults: ExtractChunkData[]): Promise<ExtractChunkData[]> {
+    return this.sdk.rerank(query, searchResults)
+  }
+}

src/main/reranker/RerankerFactory.ts

@@ -0,0 +1,20 @@
+import { KnowledgeBaseParams } from '@types'
+
+import BaseReranker from './BaseReranker'
+import DefaultReranker from './DefaultReranker'
+import JinaReranker from './JinaReranker'
+import SiliconFlowReranker from './SiliconFlowReranker'
+import VoyageReranker from './VoyageReranker'
+
+export default class RerankerFactory {
+  static create(base: KnowledgeBaseParams): BaseReranker {
+    if (base.rerankModelProvider === 'silicon') {
+      return new SiliconFlowReranker(base)
+    } else if (base.rerankModelProvider === 'jina') {
+      return new JinaReranker(base)
+    } else if (base.rerankModelProvider === 'voyageai') {
+      return new VoyageReranker(base)
+    }
+    return new DefaultReranker(base)
+  }
+}

src/main/reranker/SiliconFlowReranker.ts

@@ -0,0 +1,36 @@
+import type { ExtractChunkData } from '@cherrystudio/embedjs-interfaces'
+import { KnowledgeBaseParams } from '@types'
+import axios from 'axios'
+
+import BaseReranker from './BaseReranker'
+
+export default class SiliconFlowReranker extends BaseReranker {
+  constructor(base: KnowledgeBaseParams) {
+    super(base)
+  }
+
+  public rerank = async (query: string, searchResults: ExtractChunkData[]): Promise<ExtractChunkData[]> => {
+    const url = this.getRerankUrl()
+
+    const requestBody = {
+      model: this.base.rerankModel,
+      query,
+      documents: searchResults.map((doc) => doc.pageContent),
+      top_n: this.base.topN,
+      max_chunks_per_doc: this.base.chunkSize,
+      overlap_tokens: this.base.chunkOverlap
+    }
+
+    try {
+      const { data } = await axios.post(url, requestBody, { headers: this.defaultHeaders() })
+
+      const rerankResults = data.results
+      return this.getRerankResult(searchResults, rerankResults)
+    } catch (error: any) {
+      const errorDetails = this.formatErrorMessage(url, error, requestBody)
+
+      console.error('SiliconFlow Reranker API 错误:', errorDetails)
+      throw new Error(`重排序请求失败: ${error.message}\n请求详情: ${errorDetails}`)
+    }
+  }
+}

src/main/reranker/VoyageReranker.ts

@@ -0,0 +1,40 @@
+import { ExtractChunkData } from '@cherrystudio/embedjs-interfaces'
+import { KnowledgeBaseParams } from '@types'
+import axios from 'axios'
+
+import BaseReranker from './BaseReranker'
+
+export default class VoyageReranker extends BaseReranker {
+  constructor(base: KnowledgeBaseParams) {
+    super(base)
+  }
+
+  public rerank = async (query: string, searchResults: ExtractChunkData[]): Promise<ExtractChunkData[]> => {
+    const url = this.getRerankUrl()
+
+    const requestBody = {
+      model: this.base.rerankModel,
+      query,
+      documents: searchResults.map((doc) => doc.pageContent),
+      top_k: this.base.topN,
+      return_documents: false,
+      truncation: true
+    }
+
+    try {
+      const { data } = await axios.post(url, requestBody, {
+        headers: {
+          ...this.defaultHeaders()
+        }
+      })
+
+      const rerankResults = data.data
+      return this.getRerankResult(searchResults, rerankResults)
+    } catch (error: any) {
+      const errorDetails = this.formatErrorMessage(url, error, requestBody)
+
+      console.error('Voyage Reranker API Error:', errorDetails)
+      throw new Error(`重排序请求失败: ${error.message}\n请求详情: ${errorDetails}`)
+    }
+  }
+}

src/main/services/AppUpdater.ts

@@ -1,11 +1,14 @@
+import { IpcChannel } from '@shared/IpcChannel'
+import { UpdateInfo } from 'builder-util-runtime'
 import { app, BrowserWindow, dialog } from 'electron'
 import logger from 'electron-log'
-import { AppUpdater as _AppUpdater, autoUpdater, UpdateInfo } from 'electron-updater'
+import { AppUpdater as _AppUpdater, autoUpdater } from 'electron-updater'
 
 import icon from '../../../build/icon.png?asset'
 
 export default class AppUpdater {
   autoUpdater: _AppUpdater = autoUpdater
+  private releaseInfo: UpdateInfo | undefined
 
   constructor(mainWindow: BrowserWindow) {
     logger.transports.file.level = 'info'
@@ -16,55 +19,66 @@ export default class AppUpdater {
 
     // 检测下载错误
     autoUpdater.on('error', (error) => {
-      logger.error('更新异常', error)
-      mainWindow.webContents.send('update-error', error)
+      // 简单记录错误信息和时间戳
+      logger.error('更新异常', {
+        message: error.message,
+        stack: error.stack,
+        time: new Date().toISOString()
+      })
+      mainWindow.webContents.send(IpcChannel.UpdateError, error)
     })
 
     autoUpdater.on('update-available', (releaseInfo: UpdateInfo) => {
       logger.info('检测到新版本', releaseInfo)
-      mainWindow.webContents.send('update-available', releaseInfo)
+      mainWindow.webContents.send(IpcChannel.UpdateAvailable, releaseInfo)
     })
 
     // 检测到不需要更新时
     autoUpdater.on('update-not-available', () => {
-      mainWindow.webContents.send('update-not-available')
+      mainWindow.webContents.send(IpcChannel.UpdateNotAvailable)
     })
 
     // 更新下载进度
     autoUpdater.on('download-progress', (progress) => {
-      mainWindow.webContents.send('download-progress', progress)
+      mainWindow.webContents.send(IpcChannel.DownloadProgress, progress)
     })
 
     // 当需要更新的内容下载完成后
     autoUpdater.on('update-downloaded', (releaseInfo: UpdateInfo) => {
-      mainWindow.webContents.send('update-downloaded')
-
-      logger.info('下载完成，询问用户是否更新', releaseInfo)
-
-      dialog
-        .showMessageBox({
-          type: 'info',
-          title: '安装更新',
-          icon,
-          message: `新版本 ${releaseInfo.version} 已准备就绪`,
-          detail: this.formatReleaseNotes(releaseInfo.releaseNotes),
-          buttons: ['稍后安装', '立即安装'],
-          defaultId: 1,
-          cancelId: 0
-        })
-        .then(({ response }) => {
-          if (response === 1) {
-            app.isQuitting = true
-            setImmediate(() => autoUpdater.quitAndInstall())
-          } else {
-            mainWindow.webContents.send('update-downloaded-cancelled')
-          }
-        })
+      mainWindow.webContents.send(IpcChannel.UpdateDownloaded, releaseInfo)
+      this.releaseInfo = releaseInfo
+      logger.info('下载完成', releaseInfo)
     })
 
     this.autoUpdater = autoUpdater
   }
 
+  public async showUpdateDialog(mainWindow: BrowserWindow) {
+    if (!this.releaseInfo) {
+      return
+    }
+
+    dialog
+      .showMessageBox({
+        type: 'info',
+        title: '安装更新',
+        icon,
+        message: `新版本 ${this.releaseInfo.version} 已准备就绪`,
+        detail: this.formatReleaseNotes(this.releaseInfo.releaseNotes),
+        buttons: ['稍后安装', '立即安装'],
+        defaultId: 1,
+        cancelId: 0
+      })
+      .then(({ response }) => {
+        if (response === 1) {
+          app.isQuitting = true
+          setImmediate(() => autoUpdater.quitAndInstall())
+        } else {
+          mainWindow.webContents.send(IpcChannel.UpdateDownloadedCancelled)
+        }
+      })
+  }
+
   private formatReleaseNotes(releaseNotes: string | ReleaseNoteInfo[] | null | undefined): string {
     if (!releaseNotes) {
       return '暂无更新说明'

src/main/services/BackupManager.ts

@@ -1,45 +1,51 @@
+import { IpcChannel } from '@shared/IpcChannel'
 import { WebDavConfig } from '@types'
 import AdmZip from 'adm-zip'
+import { exec } from 'child_process'
 import { app } from 'electron'
 import Logger from 'electron-log'
 import * as fs from 'fs-extra'
 import * as path from 'path'
+import { createClient, CreateDirectoryOptions, FileStat } from 'webdav'
 
+import { getConfigDir } from '../utils/file'
 import WebDav from './WebDav'
-import { exec } from 'child_process'
+import { windowService } from './WindowService'
 
 class BackupManager {
   private tempDir = path.join(app.getPath('temp'), 'cherry-studio', 'backup', 'temp')
   private backupDir = path.join(app.getPath('temp'), 'cherry-studio', 'backup')
 
   constructor() {
+    this.checkConnection = this.checkConnection.bind(this)
     this.backup = this.backup.bind(this)
     this.restore = this.restore.bind(this)
     this.backupToWebdav = this.backupToWebdav.bind(this)
     this.restoreFromWebdav = this.restoreFromWebdav.bind(this)
+    this.listWebdavFiles = this.listWebdavFiles.bind(this)
   }
 
   private async setWritableRecursive(dirPath: string): Promise<void> {
     try {
-      const items = await fs.readdir(dirPath, { withFileTypes: true });
+      const items = await fs.readdir(dirPath, { withFileTypes: true })
 
       for (const item of items) {
-        const fullPath = path.join(dirPath, item.name);
+        const fullPath = path.join(dirPath, item.name)
 
         // 先处理子目录
         if (item.isDirectory()) {
-          await this.setWritableRecursive(fullPath);
+          await this.setWritableRecursive(fullPath)
         }
 
         // 统一设置权限（Windows需要特殊处理）
-        await this.forceSetWritable(fullPath);
+        await this.forceSetWritable(fullPath)
       }
 
       // 确保根目录权限
-      await this.forceSetWritable(dirPath);
+      await this.forceSetWritable(dirPath)
     } catch (error) {
-      Logger.error(`权限设置失败：${dirPath}`, error);
-      throw error;
+      Logger.error(`权限设置失败：${dirPath}`, error)
+      throw error
     }
   }
 
@@ -48,20 +54,20 @@ class BackupManager {
     try {
       // Windows系统需要先取消只读属性
       if (process.platform === 'win32') {
-        await fs.chmod(targetPath, 0o666); // Windows会忽略权限位但能移除只读
+        await fs.chmod(targetPath, 0o666) // Windows会忽略权限位但能移除只读
       } else {
-        const stats = await fs.stat(targetPath);
-        const mode = stats.isDirectory() ? 0o777 : 0o666;
-        await fs.chmod(targetPath, mode);
+        const stats = await fs.stat(targetPath)
+        const mode = stats.isDirectory() ? 0o777 : 0o666
+        await fs.chmod(targetPath, mode)
       }
 
       // 双重保险：使用文件属性命令（Windows专用）
       if (process.platform === 'win32') {
-        await exec(`attrib -R "${targetPath}" /L /D`);
+        await exec(`attrib -R "${targetPath}" /L /D`)
       }
     } catch (error) {
       if ((error as NodeJS.ErrnoException).code !== 'ENOENT') {
-        Logger.warn(`权限设置警告：${targetPath}`, error);
+        Logger.warn(`权限设置警告：${targetPath}`, error)
       }
     }
   }
@@ -72,18 +78,65 @@ class BackupManager {
     data: string,
     destinationPath: string = this.backupDir
   ): Promise<string> {
+    const mainWindow = windowService.getMainWindow()
+
+    const onProgress = (processData: { stage: string; progress: number; total: number }) => {
+      mainWindow?.webContents.send(IpcChannel.BackupProgress, processData)
+      Logger.log('[BackupManager] backup progress', processData)
+    }
+
     try {
       await fs.ensureDir(this.tempDir)
+      onProgress({ stage: 'preparing', progress: 0, total: 100 })
 
-      // 将 data 写入临时文件
+      // 使用流的方式写入 data.json
       const tempDataPath = path.join(this.tempDir, 'data.json')
-      await fs.writeFile(tempDataPath, data)
+      await new Promise<void>((resolve, reject) => {
+        const writeStream = fs.createWriteStream(tempDataPath)
+        writeStream.write(data)
+        writeStream.end()
+
+        writeStream.on('finish', () => resolve())
+        writeStream.on('error', (error) => reject(error))
+      })
+      onProgress({ stage: 'writing_data', progress: 20, total: 100 })
 
       // 复制 Data 目录到临时目录
       const sourcePath = path.join(app.getPath('userData'), 'Data')
       const tempDataDir = path.join(this.tempDir, 'Data')
-      await fs.copy(sourcePath, tempDataDir)
+
+      // 获取源目录总大小
+      const totalSize = await this.getDirSize(sourcePath)
+      let copiedSize = 0
+
+      // 使用流式复制
+      await this.copyDirWithProgress(sourcePath, tempDataDir, (size) => {
+        copiedSize += size
+        const progress = Math.min(70, 20 + Math.floor((copiedSize / totalSize) * 50))
+        onProgress({ stage: 'copying_files', progress, total: 100 })
+      })
+
+      // 复制记忆数据文件
+      const configDir = getConfigDir()
+      const memoryDataPath = path.join(configDir, 'memory-data.json')
+      const tempConfigDir = path.join(this.tempDir, 'Config')
+      const tempMemoryDataPath = path.join(tempConfigDir, 'memory-data.json')
+
+      // 确保目录存在
+      await fs.ensureDir(tempConfigDir)
+
+      // 如果记忆数据文件存在，则复制
+      if (await fs.pathExists(memoryDataPath)) {
+        await fs.copy(memoryDataPath, tempMemoryDataPath)
+        Logger.log('[BackupManager] Memory data file copied')
+        onProgress({ stage: 'copying_memory_data', progress: 75, total: 100 })
+      } else {
+        Logger.log('[BackupManager] Memory data file not found, skipping')
+        onProgress({ stage: 'copying_memory_data', progress: 75, total: 100 })
+      }
+
       await this.setWritableRecursive(tempDataDir)
+      onProgress({ stage: 'compressing', progress: 80, total: 100 })
 
       // 使用 adm-zip 创建压缩文件
       const zip = new AdmZip()
@@ -93,45 +146,86 @@ class BackupManager {
 
       // 清理临时目录
       await fs.remove(this.tempDir)
+      onProgress({ stage: 'completed', progress: 100, total: 100 })
 
-      Logger.log('Backup completed successfully')
+      Logger.log('[BackupManager] Backup completed successfully')
       return backupedFilePath
     } catch (error) {
-      Logger.error('Backup failed:', error)
+      Logger.error('[BackupManager] Backup failed:', error)
       throw error
     }
   }
 
   async restore(_: Electron.IpcMainInvokeEvent, backupPath: string): Promise<string> {
+    const mainWindow = windowService.getMainWindow()
+
+    const onProgress = (processData: { stage: string; progress: number; total: number }) => {
+      mainWindow?.webContents.send(IpcChannel.RestoreProgress, processData)
+      Logger.log('[BackupManager] restore progress', processData)
+    }
+
     try {
       // 创建临时目录
       await fs.ensureDir(this.tempDir)
+      onProgress({ stage: 'preparing', progress: 0, total: 100 })
 
       Logger.log('[backup] step 1: unzip backup file', this.tempDir)
-
       // 使用 adm-zip 解压
       const zip = new AdmZip(backupPath)
       zip.extractAllTo(this.tempDir, true) // true 表示覆盖已存在的文件
+      onProgress({ stage: 'extracting', progress: 20, total: 100 })
 
       Logger.log('[backup] step 2: read data.json')
-
       // 读取 data.json
       const dataPath = path.join(this.tempDir, 'data.json')
       const data = await fs.readFile(dataPath, 'utf-8')
+      onProgress({ stage: 'reading_data', progress: 40, total: 100 })
 
       Logger.log('[backup] step 3: restore Data directory')
-
       // 恢复 Data 目录
       const sourcePath = path.join(this.tempDir, 'Data')
       const destPath = path.join(app.getPath('userData'), 'Data')
+
+      // 获取源目录总大小
+      const totalSize = await this.getDirSize(sourcePath)
+      let copiedSize = 0
+
       await this.setWritableRecursive(destPath)
       await fs.remove(destPath)
-      await fs.copy(sourcePath, destPath)
 
-      Logger.log('[backup] step 4: clean up temp directory')
+      // 使用流式复制
+      await this.copyDirWithProgress(sourcePath, destPath, (size) => {
+        copiedSize += size
+        const progress = Math.min(80, 40 + Math.floor((copiedSize / totalSize) * 40))
+        onProgress({ stage: 'copying_files', progress, total: 100 })
+      })
 
+      // 恢复记忆数据文件
+      Logger.log('[backup] step 4: restore memory data file')
+      const tempConfigDir = path.join(this.tempDir, 'Config')
+      const tempMemoryDataPath = path.join(tempConfigDir, 'memory-data.json')
+
+      if (await fs.pathExists(tempMemoryDataPath)) {
+        const configDir = getConfigDir()
+        const memoryDataPath = path.join(configDir, 'memory-data.json')
+
+        // 确保目录存在
+        await fs.ensureDir(configDir)
+
+        // 复制记忆数据文件
+        await fs.copy(tempMemoryDataPath, memoryDataPath)
+        Logger.log('[backup] Memory data file restored')
+        onProgress({ stage: 'restoring_memory_data', progress: 90, total: 100 })
+      } else {
+        Logger.log('[backup] Memory data file not found in backup, skipping')
+        onProgress({ stage: 'restoring_memory_data', progress: 90, total: 100 })
+      }
+
+      Logger.log('[backup] step 5: clean up temp directory')
       // 清理临时目录
+      await this.setWritableRecursive(this.tempDir)
       await fs.remove(this.tempDir)
+      onProgress({ stage: 'completed', progress: 100, total: 100 })
 
       Logger.log('[backup] step 5: Restore completed successfully')
 
@@ -144,7 +238,7 @@ class BackupManager {
   }
 
   async backupToWebdav(_: Electron.IpcMainInvokeEvent, data: string, webdavConfig: WebDavConfig) {
-    const filename = 'cherry-studio.backup.zip'
+    const filename = webdavConfig.fileName || 'cherry-studio.backup.zip'
     const backupedFilePath = await this.backup(_, filename, data)
     const webdavClient = new WebDav(webdavConfig)
     return await webdavClient.putFileContents(filename, fs.createReadStream(backupedFilePath), {
@@ -153,18 +247,108 @@ class BackupManager {
   }
 
   async restoreFromWebdav(_: Electron.IpcMainInvokeEvent, webdavConfig: WebDavConfig) {
-    const filename = 'cherry-studio.backup.zip'
+    const filename = webdavConfig.fileName || 'cherry-studio.backup.zip'
     const webdavClient = new WebDav(webdavConfig)
-    const retrievedFile = await webdavClient.getFileContents(filename)
-    const backupedFilePath = path.join(this.backupDir, filename)
+    try {
+      const retrievedFile = await webdavClient.getFileContents(filename)
+      const backupedFilePath = path.join(this.backupDir, filename)
 
-    if (!fs.existsSync(this.backupDir)) {
-      fs.mkdirSync(this.backupDir, { recursive: true })
+      if (!fs.existsSync(this.backupDir)) {
+        fs.mkdirSync(this.backupDir, { recursive: true })
+      }
+
+      // 使用流的方式写入文件
+      await new Promise<void>((resolve, reject) => {
+        const writeStream = fs.createWriteStream(backupedFilePath)
+        writeStream.write(retrievedFile as Buffer)
+        writeStream.end()
+
+        writeStream.on('finish', () => resolve())
+        writeStream.on('error', (error) => reject(error))
+      })
+
+      return await this.restore(_, backupedFilePath)
+    } catch (error: any) {
+      Logger.error('[backup] Failed to restore from WebDAV:', error)
+      throw new Error(error.message || 'Failed to restore backup file')
     }
+  }
 
-    await fs.writeFileSync(backupedFilePath, retrievedFile as Buffer)
+  listWebdavFiles = async (_: Electron.IpcMainInvokeEvent, config: WebDavConfig) => {
+    try {
+      const client = createClient(config.webdavHost, {
+        username: config.webdavUser,
+        password: config.webdavPass
+      })
 
-    return await this.restore(_, backupedFilePath)
+      const response = await client.getDirectoryContents(config.webdavPath)
+      const files = Array.isArray(response) ? response : response.data
+
+      return files
+        .filter((file: FileStat) => file.type === 'file' && file.basename.endsWith('.zip'))
+        .map((file: FileStat) => ({
+          fileName: file.basename,
+          modifiedTime: file.lastmod,
+          size: file.size
+        }))
+        .sort((a, b) => new Date(b.modifiedTime).getTime() - new Date(a.modifiedTime).getTime())
+    } catch (error: any) {
+      Logger.error('Failed to list WebDAV files:', error)
+      throw new Error(error.message || 'Failed to list backup files')
+    }
+  }
+
+  private async getDirSize(dirPath: string): Promise<number> {
+    let size = 0
+    const items = await fs.readdir(dirPath, { withFileTypes: true })
+
+    for (const item of items) {
+      const fullPath = path.join(dirPath, item.name)
+      if (item.isDirectory()) {
+        size += await this.getDirSize(fullPath)
+      } else {
+        const stats = await fs.stat(fullPath)
+        size += stats.size
+      }
+    }
+    return size
+  }
+
+  private async copyDirWithProgress(
+    source: string,
+    destination: string,
+    onProgress: (size: number) => void
+  ): Promise<void> {
+    const items = await fs.readdir(source, { withFileTypes: true })
+
+    for (const item of items) {
+      const sourcePath = path.join(source, item.name)
+      const destPath = path.join(destination, item.name)
+
+      if (item.isDirectory()) {
+        await fs.ensureDir(destPath)
+        await this.copyDirWithProgress(sourcePath, destPath, onProgress)
+      } else {
+        const stats = await fs.stat(sourcePath)
+        await fs.copy(sourcePath, destPath)
+        onProgress(stats.size)
+      }
+    }
+  }
+
+  async checkConnection(_: Electron.IpcMainInvokeEvent, webdavConfig: WebDavConfig) {
+    const webdavClient = new WebDav(webdavConfig)
+    return await webdavClient.checkConnection()
+  }
+
+  async createDirectory(
+    _: Electron.IpcMainInvokeEvent,
+    webdavConfig: WebDavConfig,
+    path: string,
+    options?: CreateDirectoryOptions
+  ) {
+    const webdavClient = new WebDav(webdavConfig)
+    return await webdavClient.createDirectory(path, options)
   }
 }
 

src/main/services/ClipboardMonitor.ts

@@ -98,7 +98,7 @@ export default class ClipboardMonitor {
   private handleTextSelected(text: string) {
     if (!text) return
 
-    console.debug('[ClipboardMonitor] handleTextSelected', text)
+    console.log('[ClipboardMonitor] handleTextSelected', text)
 
     windowService.setLastSelectedText(text)
 

src/main/services/ConfigManager.ts

@@ -1,10 +1,22 @@
-import { ZOOM_SHORTCUTS } from '@shared/config/constant'
+import { defaultLanguage, ZOOM_SHORTCUTS } from '@shared/config/constant'
 import { LanguageVarious, Shortcut, ThemeMode } from '@types'
 import { app } from 'electron'
 import Store from 'electron-store'
 
 import { locales } from '../utils/locales'
 
+enum ConfigKeys {
+  Language = 'language',
+  Theme = 'theme',
+  LaunchToTray = 'launchToTray',
+  Tray = 'tray',
+  TrayOnClose = 'trayOnClose',
+  ZoomFactor = 'ZoomFactor',
+  Shortcuts = 'shortcuts',
+  ClickTrayToShowQuickAssistant = 'clickTrayToShowQuickAssistant',
+  EnableQuickAssistant = 'enableQuickAssistant'
+}
+
 export class ConfigManager {
   private store: Store
   private subscribers: Map<string, Array<(newValue: any) => void>> = new Map()
@@ -14,38 +26,54 @@ export class ConfigManager {
   }
 
   getLanguage(): LanguageVarious {
-    const locale = Object.keys(locales).includes(app.getLocale()) ? app.getLocale() : 'en-US'
-    return this.store.get('language', locale) as LanguageVarious
+    const locale = Object.keys(locales).includes(app.getLocale()) ? app.getLocale() : defaultLanguage
+    return this.get(ConfigKeys.Language, locale) as LanguageVarious
   }
 
   setLanguage(theme: LanguageVarious) {
-    this.store.set('language', theme)
+    this.set(ConfigKeys.Language, theme)
   }
 
   getTheme(): ThemeMode {
-    return this.store.get('theme', ThemeMode.light) as ThemeMode
+    return this.get(ConfigKeys.Theme, ThemeMode.light)
   }
 
   setTheme(theme: ThemeMode) {
-    this.store.set('theme', theme)
+    this.set(ConfigKeys.Theme, theme)
+  }
+
+  getLaunchToTray(): boolean {
+    return !!this.get(ConfigKeys.LaunchToTray, false)
+  }
+
+  setLaunchToTray(value: boolean) {
+    this.set(ConfigKeys.LaunchToTray, value)
   }
 
   getTray(): boolean {
-    return !!this.store.get('tray', true)
+    return !!this.get(ConfigKeys.Tray, true)
   }
 
   setTray(value: boolean) {
-    this.store.set('tray', value)
-    this.notifySubscribers('tray', value)
+    this.set(ConfigKeys.Tray, value)
+    this.notifySubscribers(ConfigKeys.Tray, value)
+  }
+
+  getTrayOnClose(): boolean {
+    return !!this.get(ConfigKeys.TrayOnClose, true)
+  }
+
+  setTrayOnClose(value: boolean) {
+    this.set(ConfigKeys.TrayOnClose, value)
   }
 
   getZoomFactor(): number {
-    return this.store.get('zoomFactor', 1) as number
+    return this.get<number>(ConfigKeys.ZoomFactor, 1)
   }
 
   setZoomFactor(factor: number) {
-    this.store.set('zoomFactor', factor)
-    this.notifySubscribers('zoomFactor', factor)
+    this.set(ConfigKeys.ZoomFactor, factor)
+    this.notifySubscribers(ConfigKeys.ZoomFactor, factor)
   }
 
   subscribe<T>(key: string, callback: (newValue: T) => void) {
@@ -73,39 +101,39 @@ export class ConfigManager {
   }
 
   getShortcuts() {
-    return this.store.get('shortcuts', ZOOM_SHORTCUTS) as Shortcut[] | []
+    return this.get(ConfigKeys.Shortcuts, ZOOM_SHORTCUTS) as Shortcut[] | []
   }
 
   setShortcuts(shortcuts: Shortcut[]) {
-    this.store.set(
-      'shortcuts',
+    this.set(
+      ConfigKeys.Shortcuts,
       shortcuts.filter((shortcut) => shortcut.system)
     )
-    this.notifySubscribers('shortcuts', shortcuts)
+    this.notifySubscribers(ConfigKeys.Shortcuts, shortcuts)
   }
 
   getClickTrayToShowQuickAssistant(): boolean {
-    return this.store.get('clickTrayToShowQuickAssistant', false) as boolean
+    return this.get<boolean>(ConfigKeys.ClickTrayToShowQuickAssistant, false)
   }
 
   setClickTrayToShowQuickAssistant(value: boolean) {
-    this.store.set('clickTrayToShowQuickAssistant', value)
+    this.set(ConfigKeys.ClickTrayToShowQuickAssistant, value)
   }
 
   getEnableQuickAssistant(): boolean {
-    return this.store.get('enableQuickAssistant', false) as boolean
+    return this.get(ConfigKeys.EnableQuickAssistant, false)
   }
 
   setEnableQuickAssistant(value: boolean) {
-    this.store.set('enableQuickAssistant', value)
+    this.set(ConfigKeys.EnableQuickAssistant, value)
   }
 
-  set(key: string, value: any) {
+  set(key: string, value: unknown) {
     this.store.set(key, value)
   }
 
-  get(key: string) {
-    return this.store.get(key)
+  get<T>(key: string, defaultValue?: T) {
+    return this.store.get(key, defaultValue) as T
   }
 }
 

src/main/services/CopilotService.ts

@@ -0,0 +1,247 @@
+import axios, { AxiosRequestConfig } from 'axios'
+import { app, safeStorage } from 'electron'
+import fs from 'fs/promises'
+import path from 'path'
+
+// 配置常量，集中管理
+const CONFIG = {
+  GITHUB_CLIENT_ID: 'Iv1.b507a08c87ecfe98',
+  POLLING: {
+    MAX_ATTEMPTS: 8,
+    INITIAL_DELAY_MS: 1000,
+    MAX_DELAY_MS: 16000 // 最大延迟16秒
+  },
+  DEFAULT_HEADERS: {
+    accept: 'application/json',
+    'editor-version': 'Neovim/0.6.1',
+    'editor-plugin-version': 'copilot.vim/1.16.0',
+    'content-type': 'application/json',
+    'user-agent': 'GithubCopilot/1.155.0',
+    'accept-encoding': 'gzip,deflate,br'
+  },
+  // API端点集中管理
+  API_URLS: {
+    GITHUB_USER: 'https://api.github.com/user',
+    GITHUB_DEVICE_CODE: 'https://github.com/login/device/code',
+    GITHUB_ACCESS_TOKEN: 'https://github.com/login/oauth/access_token',
+    COPILOT_TOKEN: 'https://api.github.com/copilot_internal/v2/token'
+  }
+}
+
+// 接口定义移到顶部，便于查阅
+interface UserResponse {
+  login: string
+  avatar: string
+}
+
+interface AuthResponse {
+  device_code: string
+  user_code: string
+  verification_uri: string
+}
+
+interface TokenResponse {
+  access_token: string
+}
+
+interface CopilotTokenResponse {
+  token: string
+}
+
+// 自定义错误类，统一错误处理
+class CopilotServiceError extends Error {
+  constructor(
+    message: string,
+    public readonly cause?: unknown
+  ) {
+    super(message)
+    this.name = 'CopilotServiceError'
+  }
+}
+
+class CopilotService {
+  private readonly tokenFilePath: string
+  private headers: Record<string, string>
+
+  constructor() {
+    this.tokenFilePath = path.join(app.getPath('userData'), '.copilot_token')
+    this.headers = { ...CONFIG.DEFAULT_HEADERS }
+  }
+
+  /**
+   * 设置自定义请求头
+   */
+  private updateHeaders = (headers?: Record<string, string>): void => {
+    if (headers && Object.keys(headers).length > 0) {
+      this.headers = { ...headers }
+    }
+  }
+
+  /**
+   * 获取GitHub登录信息
+   */
+  public getUser = async (_: Electron.IpcMainInvokeEvent, token: string): Promise<UserResponse> => {
+    try {
+      const config: AxiosRequestConfig = {
+        headers: {
+          Connection: 'keep-alive',
+          'user-agent': 'Visual Studio Code (desktop)',
+          'Sec-Fetch-Site': 'none',
+          'Sec-Fetch-Mode': 'no-cors',
+          'Sec-Fetch-Dest': 'empty',
+          authorization: `token ${token}`
+        }
+      }
+
+      const response = await axios.get(CONFIG.API_URLS.GITHUB_USER, config)
+      return {
+        login: response.data.login,
+        avatar: response.data.avatar_url
+      }
+    } catch (error) {
+      console.error('Failed to get user information:', error)
+      throw new CopilotServiceError('无法获取GitHub用户信息', error)
+    }
+  }
+
+  /**
+   * 获取GitHub设备授权信息
+   */
+  public getAuthMessage = async (
+    _: Electron.IpcMainInvokeEvent,
+    headers?: Record<string, string>
+  ): Promise<AuthResponse> => {
+    try {
+      this.updateHeaders(headers)
+
+      const response = await axios.post<AuthResponse>(
+        CONFIG.API_URLS.GITHUB_DEVICE_CODE,
+        {
+          client_id: CONFIG.GITHUB_CLIENT_ID,
+          scope: 'read:user'
+        },
+        { headers: this.headers }
+      )
+
+      return response.data
+    } catch (error) {
+      console.error('Failed to get auth message:', error)
+      throw new CopilotServiceError('无法获取GitHub授权信息', error)
+    }
+  }
+
+  /**
+   * 使用设备码获取访问令牌 - 优化轮询逻辑
+   */
+  public getCopilotToken = async (
+    _: Electron.IpcMainInvokeEvent,
+    device_code: string,
+    headers?: Record<string, string>
+  ): Promise<TokenResponse> => {
+    this.updateHeaders(headers)
+
+    let currentDelay = CONFIG.POLLING.INITIAL_DELAY_MS
+
+    for (let attempt = 0; attempt < CONFIG.POLLING.MAX_ATTEMPTS; attempt++) {
+      await this.delay(currentDelay)
+
+      try {
+        const response = await axios.post<TokenResponse>(
+          CONFIG.API_URLS.GITHUB_ACCESS_TOKEN,
+          {
+            client_id: CONFIG.GITHUB_CLIENT_ID,
+            device_code,
+            grant_type: 'urn:ietf:params:oauth:grant-type:device_code'
+          },
+          { headers: this.headers }
+        )
+
+        const { access_token } = response.data
+        if (access_token) {
+          return { access_token }
+        }
+      } catch (error) {
+        // 指数退避策略
+        currentDelay = Math.min(currentDelay * 2, CONFIG.POLLING.MAX_DELAY_MS)
+
+        // 仅在最后一次尝试失败时记录详细错误
+        const isLastAttempt = attempt === CONFIG.POLLING.MAX_ATTEMPTS - 1
+        if (isLastAttempt) {
+          console.error(`Token polling failed after ${CONFIG.POLLING.MAX_ATTEMPTS} attempts:`, error)
+        }
+      }
+    }
+
+    throw new CopilotServiceError('获取访问令牌超时，请重试')
+  }
+
+  /**
+   * 保存Copilot令牌到本地文件
+   */
+  public saveCopilotToken = async (_: Electron.IpcMainInvokeEvent, token: string): Promise<void> => {
+    try {
+      const encryptedToken = safeStorage.encryptString(token)
+      await fs.writeFile(this.tokenFilePath, encryptedToken)
+    } catch (error) {
+      console.error('Failed to save token:', error)
+      throw new CopilotServiceError('无法保存访问令牌', error)
+    }
+  }
+
+  /**
+   * 从本地文件读取令牌并获取Copilot令牌
+   */
+  public getToken = async (
+    _: Electron.IpcMainInvokeEvent,
+    headers?: Record<string, string>
+  ): Promise<CopilotTokenResponse> => {
+    try {
+      this.updateHeaders(headers)
+
+      const encryptedToken = await fs.readFile(this.tokenFilePath)
+      const access_token = safeStorage.decryptString(Buffer.from(encryptedToken))
+
+      const config: AxiosRequestConfig = {
+        headers: {
+          ...this.headers,
+          authorization: `token ${access_token}`
+        }
+      }
+
+      const response = await axios.get<CopilotTokenResponse>(CONFIG.API_URLS.COPILOT_TOKEN, config)
+
+      return response.data
+    } catch (error) {
+      console.error('Failed to get Copilot token:', error)
+      throw new CopilotServiceError('无法获取Copilot令牌，请重新授权', error)
+    }
+  }
+
+  /**
+   * 退出登录，删除本地token文件
+   */
+  public logout = async (): Promise<void> => {
+    try {
+      try {
+        await fs.access(this.tokenFilePath)
+        await fs.unlink(this.tokenFilePath)
+        console.log('Successfully logged out from Copilot')
+      } catch (error) {
+        // 文件不存在不是错误，只是记录一下
+        console.log('Token file not found, nothing to delete')
+      }
+    } catch (error) {
+      console.error('Failed to logout:', error)
+      throw new CopilotServiceError('无法完成退出登录操作', error)
+    }
+  }
+
+  /**
+   * 辅助方法：延迟执行
+   */
+  private delay = (ms: number): Promise<void> => {
+    return new Promise((resolve) => setTimeout(resolve, ms))
+  }
+}
+
+export default new CopilotService()

src/main/services/ExportService.ts

@@ -1,7 +1,22 @@
 /* eslint-disable no-case-declarations */
 // ExportService
 
-import { AlignmentType, BorderStyle, Document, HeadingLevel, Packer, Paragraph, ShadingType, TextRun } from 'docx'
+import {
+  AlignmentType,
+  BorderStyle,
+  Document,
+  ExternalHyperlink,
+  HeadingLevel,
+  Packer,
+  Paragraph,
+  ShadingType,
+  Table,
+  TableCell,
+  TableRow,
+  TextRun,
+  VerticalAlign,
+  WidthType
+} from 'docx'
 import { dialog } from 'electron'
 import Logger from 'electron-log'
 import MarkdownIt from 'markdown-it'
@@ -21,13 +36,54 @@ export class ExportService {
     const tokens = this.md.parse(markdown, {})
     const elements: any[] = []
     let listLevel = 0
+    let currentTable: Table | null = null
+    let currentRowCells: TableCell[] = []
+    let isHeaderRow = false
+    let tableColumnCount = 0
+    let tableRows: TableRow[] = [] // Store rows temporarily
 
-    const processInlineTokens = (tokens: any[]): TextRun[] => {
-      const runs: TextRun[] = []
-      for (const token of tokens) {
+    const processInlineTokens = (tokens: any[], isHeaderRow: boolean): (TextRun | ExternalHyperlink)[] => {
+      const runs: (TextRun | ExternalHyperlink)[] = []
+      let linkText = ''
+      let linkUrl = ''
+      let insideLink = false
+
+      for (let i = 0; i < tokens.length; i++) {
+        const token = tokens[i]
         switch (token.type) {
+          case 'link_open':
+            insideLink = true
+            linkUrl = token.attrs.find((attr: [string, string]) => attr[0] === 'href')[1]
+            linkText = tokens[i + 1].content
+            i += 1
+            break
+          case 'link_close':
+            if (insideLink && linkUrl && linkText) {
+              // Handle any accumulated link text with the ExternalHyperlink
+              runs.push(
+                new ExternalHyperlink({
+                  children: [
+                    new TextRun({
+                      text: linkText,
+                      style: 'Hyperlink',
+                      color: '0000FF',
+                      underline: {
+                        type: 'single'
+                      }
+                    })
+                  ],
+                  link: linkUrl
+                })
+              )
+
+              // Reset link variables
+              linkText = ''
+              linkUrl = ''
+              insideLink = false
+            }
+            break
           case 'text':
-            runs.push(new TextRun(token.content))
+            runs.push(new TextRun({ text: token.content, bold: isHeaderRow ? true : false }))
             break
           case 'strong':
             runs.push(new TextRun({ text: token.content, bold: true }))
@@ -45,7 +101,6 @@ export class ExportService {
 
     for (let i = 0; i < tokens.length; i++) {
       const token = tokens[i]
-
       switch (token.type) {
         case 'heading_open':
           // 获取标题级别 (h1 -> h6)
@@ -68,7 +123,7 @@ export class ExportService {
           const inlineTokens = tokens[i + 1].children || []
           elements.push(
             new Paragraph({
-              children: processInlineTokens(inlineTokens),
+              children: processInlineTokens(inlineTokens, false),
               spacing: {
                 before: 120,
                 after: 120
@@ -93,7 +148,7 @@ export class ExportService {
               children: [
                 new TextRun({ text: '•', bold: true }),
                 new TextRun({ text: '\t' }),
-                ...processInlineTokens(itemInlineTokens)
+                ...processInlineTokens(itemInlineTokens, false)
               ],
               indent: {
                 left: listLevel * 720
@@ -171,6 +226,116 @@ export class ExportService {
           )
           i += 3
           break
+
+        // 表格处理
+        case 'table_open':
+          tableRows = [] // Reset table rows for new table
+          break
+
+        case 'thead_open':
+          isHeaderRow = true
+          break
+
+        case 'tbody_open':
+          isHeaderRow = false
+          break
+
+        case 'tr_open':
+          currentRowCells = []
+          break
+
+        case 'tr_close':
+          const row = new TableRow({
+            children: currentRowCells,
+            tableHeader: isHeaderRow
+          })
+          tableRows.push(row)
+          // 计算表格有多少列（针对第一行）
+          if (tableColumnCount === 0) {
+            tableColumnCount = currentRowCells.length
+          }
+          break
+
+        case 'th_open':
+        case 'td_open':
+          const isFirstColumn = currentRowCells.length === 0 // 判断是否是第一列
+          const borders = {
+            top: {
+              style: BorderStyle.NONE
+            },
+            bottom: isHeaderRow
+              ? {
+                  style: BorderStyle.SINGLE,
+                  size: 0.5,
+                  color: '000000'
+                }
+              : {
+                  style: BorderStyle.NONE
+                },
+            left: {
+              style: BorderStyle.NONE
+            },
+            right: {
+              style: BorderStyle.NONE
+            }
+          }
+          const cellContent = tokens[i + 1]
+          const cellOptions = {
+            children: [
+              new Paragraph({
+                children: cellContent.children
+                  ? processInlineTokens(cellContent.children, isHeaderRow || isFirstColumn)
+                  : [new TextRun({ text: cellContent.content || '', bold: isHeaderRow || isFirstColumn })],
+                alignment: AlignmentType.CENTER
+              })
+            ],
+            verticalAlign: VerticalAlign.CENTER,
+            borders: borders
+          }
+          currentRowCells.push(new TableCell(cellOptions))
+          i += 2 // 跳过内容和结束标记
+          break
+        case 'table_close':
+          // Create table with the collected rows - avoid using protected properties
+          // Create the table with all rows
+          currentTable = new Table({
+            width: {
+              size: 100,
+              type: WidthType.PERCENTAGE
+            },
+            rows: tableRows,
+            borders: {
+              top: {
+                style: BorderStyle.SINGLE,
+                size: 1,
+                color: '000000'
+              },
+              bottom: {
+                style: BorderStyle.SINGLE,
+                size: 1,
+                color: '000000'
+              },
+              left: {
+                style: BorderStyle.NONE
+              },
+              right: {
+                style: BorderStyle.NONE
+              },
+              insideHorizontal: {
+                style: BorderStyle.NONE
+              },
+              insideVertical: {
+                style: BorderStyle.NONE
+              }
+            }
+          })
+          elements.push(currentTable)
+          currentTable = null
+          tableColumnCount = 0
+          tableRows = []
+          currentRowCells = []
+          isHeaderRow = false
+          break
       }
     }
 

src/main/services/FileStorage.ts

@@ -1,9 +1,8 @@
-import { getFileType } from '@main/utils/file'
-import { documentExts, imageExts } from '@shared/config/constant'
+import { getFilesDir, getFileType, getTempDir } from '@main/utils/file'
+import { documentExts, imageExts, MB } from '@shared/config/constant'
 import { FileType } from '@types'
 import * as crypto from 'crypto'
 import {
-  app,
   dialog,
   OpenDialogOptions,
   OpenDialogReturnValue,
@@ -21,8 +20,8 @@ import { chdir } from 'process'
 import { v4 as uuidv4 } from 'uuid'
 
 class FileStorage {
-  private storageDir = path.join(app.getPath('userData'), 'Data', 'Files')
-  private tempDir = path.join(app.getPath('temp'), 'CherryStudio')
+  private storageDir = getFilesDir()
+  private tempDir = getTempDir()
 
   constructor() {
     this.initStorageDir()
@@ -70,7 +69,7 @@ class FileStorage {
             origin_name: file,
             name: file + ext,
             path: storedFilePath,
-            created_at: storedStats.birthtime,
+            created_at: storedStats.birthtime.toISOString(),
             size: storedStats.size,
             ext,
             type: getFileType(ext),
@@ -109,7 +108,7 @@ class FileStorage {
         origin_name: path.basename(filePath),
         name: path.basename(filePath),
         path: filePath,
-        created_at: stats.birthtime,
+        created_at: stats.birthtime.toISOString(),
         size: stats.size,
         ext: ext,
         type: fileType,
@@ -123,7 +122,7 @@ class FileStorage {
   private async compressImage(sourcePath: string, destPath: string): Promise<void> {
     try {
       const stats = fs.statSync(sourcePath)
-      const fileSizeInMB = stats.size / (1024 * 1024)
+      const fileSizeInMB = stats.size / MB
 
       // 如果图片大于1MB才进行压缩
       if (fileSizeInMB > 1) {
@@ -174,7 +173,7 @@ class FileStorage {
       origin_name,
       name: uuid + ext,
       path: destPath,
-      created_at: stats.birthtime,
+      created_at: stats.birthtime.toISOString(),
       size: stats.size,
       ext: ext,
       type: fileType,
@@ -198,7 +197,7 @@ class FileStorage {
       origin_name: path.basename(filePath),
       name: path.basename(filePath),
       path: filePath,
-      created_at: stats.birthtime,
+      created_at: stats.birthtime.toISOString(),
       size: stats.size,
       ext: ext,
       type: fileType,
@@ -255,7 +254,8 @@ class FileStorage {
     const filePath = path.join(this.storageDir, id)
     const data = await fs.promises.readFile(filePath)
     const base64 = data.toString('base64')
-    const mime = `image/${path.extname(filePath).slice(1)}`
+    const ext = path.extname(filePath).slice(1) == 'jpg' ? 'jpeg' : path.extname(filePath).slice(1)
+    const mime = `image/${ext}`
     return {
       mime,
       base64,
@@ -271,12 +271,12 @@ class FileStorage {
   }
 
   public clear = async (): Promise<void> => {
-    await fs.promises.rmdir(this.storageDir, { recursive: true })
+    await fs.promises.rm(this.storageDir, { recursive: true })
     await this.initStorageDir()
   }
 
   public clearTemp = async (): Promise<void> => {
-    await fs.promises.rmdir(this.tempDir, { recursive: true })
+    await fs.promises.rm(this.tempDir, { recursive: true })
     await fs.promises.mkdir(this.tempDir, { recursive: true })
   }
 
@@ -416,7 +416,7 @@ class FileStorage {
         origin_name: filename,
         name: uuid + ext,
         path: destPath,
-        created_at: stats.birthtime,
+        created_at: stats.birthtime.toISOString(),
         size: stats.size,
         ext: ext,
         type: fileType,

src/main/services/KnowledgeService.ts

@@ -1,24 +1,101 @@
+/**
+ * Knowledge Service - Manages knowledge bases using RAG (Retrieval-Augmented Generation)
+ *
+ * This service handles creation, management, and querying of knowledge bases from various sources
+ * including files, directories, URLs, sitemaps, and notes.
+ *
+ * Features:
+ * - Concurrent task processing with workload management
+ * - Multiple data source support
+ * - Vector database integration
+ *
+ * For detailed documentation, see:
+ * @see {@link ../../../docs/technical/KnowledgeService.md}
+ */
+
 import * as fs from 'node:fs'
 import path from 'node:path'
 
-import { RAGApplication, RAGApplicationBuilder, TextLoader } from '@llm-tools/embedjs'
-import type { ExtractChunkData } from '@llm-tools/embedjs-interfaces'
-import { LibSqlDb } from '@llm-tools/embedjs-libsql'
-import { SitemapLoader } from '@llm-tools/embedjs-loader-sitemap'
-import { WebLoader } from '@llm-tools/embedjs-loader-web'
-import { AzureOpenAiEmbeddings, OpenAiEmbeddings } from '@llm-tools/embedjs-openai'
+import { RAGApplication, RAGApplicationBuilder, TextLoader } from '@cherrystudio/embedjs'
+import type { ExtractChunkData } from '@cherrystudio/embedjs-interfaces'
+import { LibSqlDb } from '@cherrystudio/embedjs-libsql'
+import { SitemapLoader } from '@cherrystudio/embedjs-loader-sitemap'
+import { WebLoader } from '@cherrystudio/embedjs-loader-web'
+import Embeddings from '@main/embeddings/Embeddings'
 import { addFileLoader } from '@main/loader'
-import { getInstanceName } from '@main/utils'
+import Reranker from '@main/reranker/Reranker'
+import { windowService } from '@main/services/WindowService'
 import { getAllFiles } from '@main/utils/file'
+import { MB } from '@shared/config/constant'
 import type { LoaderReturn } from '@shared/config/types'
+import { IpcChannel } from '@shared/IpcChannel'
 import { FileType, KnowledgeBaseParams, KnowledgeItem } from '@types'
 import { app } from 'electron'
+import Logger from 'electron-log'
 import { v4 as uuidv4 } from 'uuid'
 
-import { windowService } from './WindowService'
+export interface KnowledgeBaseAddItemOptions {
+  base: KnowledgeBaseParams
+  item: KnowledgeItem
+  forceReload?: boolean
+}
+
+interface KnowledgeBaseAddItemOptionsNonNullableAttribute {
+  base: KnowledgeBaseParams
+  item: KnowledgeItem
+  forceReload: boolean
+}
+
+interface EvaluateTaskWorkload {
+  workload: number
+}
+
+type LoaderDoneReturn = LoaderReturn | null
+
+enum LoaderTaskItemState {
+  PENDING,
+  PROCESSING,
+  DONE
+}
+
+interface LoaderTaskItem {
+  state: LoaderTaskItemState
+  task: () => Promise<unknown>
+  evaluateTaskWorkload: EvaluateTaskWorkload
+}
+
+interface LoaderTask {
+  loaderTasks: LoaderTaskItem[]
+  loaderDoneReturn: LoaderDoneReturn
+}
+
+interface LoaderTaskOfSet {
+  loaderTasks: Set<LoaderTaskItem>
+  loaderDoneReturn: LoaderDoneReturn
+}
+
+interface QueueTaskItem {
+  taskPromise: () => Promise<unknown>
+  resolve: () => void
+  evaluateTaskWorkload: EvaluateTaskWorkload
+}
+
+const loaderTaskIntoOfSet = (loaderTask: LoaderTask): LoaderTaskOfSet => {
+  return {
+    loaderTasks: new Set(loaderTask.loaderTasks),
+    loaderDoneReturn: loaderTask.loaderDoneReturn
+  }
+}
 
 class KnowledgeService {
   private storageDir = path.join(app.getPath('userData'), 'Data', 'KnowledgeBase')
+  // Byte based
+  private workload = 0
+  private processingItemCount = 0
+  private knowledgeItemProcessingQueueMappingPromise: Map<LoaderTaskOfSet, () => void> = new Map()
+  private static MAXIMUM_WORKLOAD = 80 * MB
+  private static MAXIMUM_PROCESSING_ITEM_COUNT = 30
+  private static ERROR_LOADER_RETURN: LoaderReturn = { entriesAdded: 0, uniqueId: '', uniqueIds: [''], loaderType: '' }
 
   constructor() {
     this.initStorageDir()
@@ -38,29 +115,20 @@ class KnowledgeService {
     baseURL,
     dimensions
   }: KnowledgeBaseParams): Promise<RAGApplication> => {
-    const batchSize = 10
-    return new RAGApplicationBuilder()
-      .setModel('NO_MODEL')
-      .setEmbeddingModel(
-        apiVersion
-          ? new AzureOpenAiEmbeddings({
-              azureOpenAIApiKey: apiKey,
-              azureOpenAIApiVersion: apiVersion,
-              azureOpenAIApiDeploymentName: model,
-              azureOpenAIApiInstanceName: getInstanceName(baseURL),
-              dimensions,
-              batchSize
-            })
-          : new OpenAiEmbeddings({
-              model,
-              apiKey,
-              configuration: { baseURL },
-              dimensions,
-              batchSize
-            })
-      )
-      .setVectorDatabase(new LibSqlDb({ path: path.join(this.storageDir, id) }))
-      .build()
+    let ragApplication: RAGApplication
+    const embeddings = new Embeddings({ model, apiKey, apiVersion, baseURL, dimensions } as KnowledgeBaseParams)
+    try {
+      ragApplication = await new RAGApplicationBuilder()
+        .setModel('NO_MODEL')
+        .setEmbeddingModel(embeddings)
+        .setVectorDatabase(new LibSqlDb({ path: path.join(this.storageDir, id) }))
+        .build()
+    } catch (e) {
+      Logger.error(e)
+      throw new Error(`Failed to create RAGApplication: ${e}`)
+    }
+
+    return ragApplication
   }
 
   public create = async (_: Electron.IpcMainInvokeEvent, base: KnowledgeBaseParams): Promise<void> => {
@@ -79,100 +147,311 @@ class KnowledgeService {
     }
   }
 
-  public add = async (
-    _: Electron.IpcMainInvokeEvent,
-    { base, item, forceReload = false }: { base: KnowledgeBaseParams; item: KnowledgeItem; forceReload: boolean }
-  ): Promise<LoaderReturn> => {
-    const ragApplication = await this.getRagApplication(base)
+  private maximumLoad() {
+    return (
+      this.processingItemCount >= KnowledgeService.MAXIMUM_PROCESSING_ITEM_COUNT ||
+      this.workload >= KnowledgeService.MAXIMUM_WORKLOAD
+    )
+  }
+
+  private fileTask(
+    ragApplication: RAGApplication,
+    options: KnowledgeBaseAddItemOptionsNonNullableAttribute
+  ): LoaderTask {
+    const { base, item, forceReload } = options
+    const file = item.content as FileType
+
+    const loaderTask: LoaderTask = {
+      loaderTasks: [
+        {
+          state: LoaderTaskItemState.PENDING,
+          task: () =>
+            addFileLoader(ragApplication, file, base, forceReload)
+              .then((result) => {
+                loaderTask.loaderDoneReturn = result
+                return result
+              })
+              .catch((err) => {
+                Logger.error(err)
+                return KnowledgeService.ERROR_LOADER_RETURN
+              }),
+          evaluateTaskWorkload: { workload: file.size }
+        }
+      ],
+      loaderDoneReturn: null
+    }
+
+    return loaderTask
+  }
+
+  private directoryTask(
+    ragApplication: RAGApplication,
+    options: KnowledgeBaseAddItemOptionsNonNullableAttribute
+  ): LoaderTask {
+    const { base, item, forceReload } = options
+    const directory = item.content as string
+    const files = getAllFiles(directory)
+    const totalFiles = files.length
+    let processedFiles = 0
 
     const sendDirectoryProcessingPercent = (totalFiles: number, processedFiles: number) => {
       const mainWindow = windowService.getMainWindow()
-      mainWindow?.webContents.send('directory-processing-percent', {
+      mainWindow?.webContents.send(IpcChannel.DirectoryProcessingPercent, {
         itemId: item.id,
         percent: (processedFiles / totalFiles) * 100
       })
     }
 
-    if (item.type === 'directory') {
-      const directory = item.content as string
-      const files = getAllFiles(directory)
-      const totalFiles = files.length
-      let processedFiles = 0
-
-      const loaderPromises = files.map(async (file) => {
-        const result = await addFileLoader(ragApplication, file, base, forceReload)
-        processedFiles++
-        sendDirectoryProcessingPercent(totalFiles, processedFiles)
-        return result
+    const loaderDoneReturn: LoaderDoneReturn = {
+      entriesAdded: 0,
+      uniqueId: `DirectoryLoader_${uuidv4()}`,
+      uniqueIds: [],
+      loaderType: 'DirectoryLoader'
+    }
+    const loaderTasks: LoaderTaskItem[] = []
+    for (const file of files) {
+      loaderTasks.push({
+        state: LoaderTaskItemState.PENDING,
+        task: () =>
+          addFileLoader(ragApplication, file, base, forceReload)
+            .then((result) => {
+              loaderDoneReturn.entriesAdded += 1
+              processedFiles += 1
+              sendDirectoryProcessingPercent(totalFiles, processedFiles)
+              loaderDoneReturn.uniqueIds.push(result.uniqueId)
+              return result
+            })
+            .catch((err) => {
+              Logger.error(err)
+              return KnowledgeService.ERROR_LOADER_RETURN
+            }),
+        evaluateTaskWorkload: { workload: file.size }
       })
-
-      const loaderResults = await Promise.allSettled(loaderPromises)
-      // @ts-ignore uniqueId
-      const uniqueIds = loaderResults
-        .filter((result) => result.status === 'fulfilled')
-        .map((result) => result.value.uniqueId)
-
-      return {
-        entriesAdded: loaderResults.length,
-        uniqueId: `DirectoryLoader_${uuidv4()}`,
-        uniqueIds,
-        loaderType: 'DirectoryLoader'
-      } as LoaderReturn
     }
 
-    if (item.type === 'url') {
-      const content = item.content as string
-      if (content.startsWith('http')) {
-        const loaderReturn = await ragApplication.addLoader(
-          new WebLoader({ urlOrContent: content, chunkSize: base.chunkSize, chunkOverlap: base.chunkOverlap }) as any,
-          forceReload
-        )
-        return {
-          entriesAdded: loaderReturn.entriesAdded,
-          uniqueId: loaderReturn.uniqueId,
-          uniqueIds: [loaderReturn.uniqueId],
-          loaderType: loaderReturn.loaderType
-        } as LoaderReturn
+    return {
+      loaderTasks,
+      loaderDoneReturn
+    }
+  }
+
+  private urlTask(
+    ragApplication: RAGApplication,
+    options: KnowledgeBaseAddItemOptionsNonNullableAttribute
+  ): LoaderTask {
+    const { base, item, forceReload } = options
+    const content = item.content as string
+
+    const loaderTask: LoaderTask = {
+      loaderTasks: [
+        {
+          state: LoaderTaskItemState.PENDING,
+          task: () => {
+            const loaderReturn = ragApplication.addLoader(
+              new WebLoader({
+                urlOrContent: content,
+                chunkSize: base.chunkSize,
+                chunkOverlap: base.chunkOverlap
+              }),
+              forceReload
+            ) as Promise<LoaderReturn>
+
+            return loaderReturn
+              .then((result) => {
+                const { entriesAdded, uniqueId, loaderType } = result
+                loaderTask.loaderDoneReturn = {
+                  entriesAdded: entriesAdded,
+                  uniqueId: uniqueId,
+                  uniqueIds: [uniqueId],
+                  loaderType: loaderType
+                }
+                return result
+              })
+              .catch((err) => {
+                Logger.error(err)
+                return KnowledgeService.ERROR_LOADER_RETURN
+              })
+          },
+          evaluateTaskWorkload: { workload: 2 * MB }
+        }
+      ],
+      loaderDoneReturn: null
+    }
+    return loaderTask
+  }
+
+  private sitemapTask(
+    ragApplication: RAGApplication,
+    options: KnowledgeBaseAddItemOptionsNonNullableAttribute
+  ): LoaderTask {
+    const { base, item, forceReload } = options
+    const content = item.content as string
+
+    const loaderTask: LoaderTask = {
+      loaderTasks: [
+        {
+          state: LoaderTaskItemState.PENDING,
+          task: () =>
+            ragApplication
+              .addLoader(
+                new SitemapLoader({ url: content, chunkSize: base.chunkSize, chunkOverlap: base.chunkOverlap }) as any,
+                forceReload
+              )
+              .then((result) => {
+                const { entriesAdded, uniqueId, loaderType } = result
+                loaderTask.loaderDoneReturn = {
+                  entriesAdded: entriesAdded,
+                  uniqueId: uniqueId,
+                  uniqueIds: [uniqueId],
+                  loaderType: loaderType
+                }
+                return result
+              })
+              .catch((err) => {
+                Logger.error(err)
+                return KnowledgeService.ERROR_LOADER_RETURN
+              }),
+          evaluateTaskWorkload: { workload: 20 * MB }
+        }
+      ],
+      loaderDoneReturn: null
+    }
+    return loaderTask
+  }
+
+  private noteTask(
+    ragApplication: RAGApplication,
+    options: KnowledgeBaseAddItemOptionsNonNullableAttribute
+  ): LoaderTask {
+    const { base, item, forceReload } = options
+    const content = item.content as string
+
+    const encoder = new TextEncoder()
+    const contentBytes = encoder.encode(content)
+    const loaderTask: LoaderTask = {
+      loaderTasks: [
+        {
+          state: LoaderTaskItemState.PENDING,
+          task: () => {
+            const loaderReturn = ragApplication.addLoader(
+              new TextLoader({ text: content, chunkSize: base.chunkSize, chunkOverlap: base.chunkOverlap }),
+              forceReload
+            ) as Promise<LoaderReturn>
+
+            return loaderReturn
+              .then(({ entriesAdded, uniqueId, loaderType }) => {
+                loaderTask.loaderDoneReturn = {
+                  entriesAdded: entriesAdded,
+                  uniqueId: uniqueId,
+                  uniqueIds: [uniqueId],
+                  loaderType: loaderType
+                }
+              })
+              .catch((err) => {
+                Logger.error(err)
+                return KnowledgeService.ERROR_LOADER_RETURN
+              })
+          },
+          evaluateTaskWorkload: { workload: contentBytes.length }
+        }
+      ],
+      loaderDoneReturn: null
+    }
+    return loaderTask
+  }
+
+  private processingQueueHandle() {
+    const getSubtasksUntilMaximumLoad = (): QueueTaskItem[] => {
+      const queueTaskList: QueueTaskItem[] = []
+      that: for (const [task, resolve] of this.knowledgeItemProcessingQueueMappingPromise) {
+        for (const item of task.loaderTasks) {
+          if (this.maximumLoad()) {
+            break that
+          }
+
+          const { state, task: taskPromise, evaluateTaskWorkload } = item
+
+          if (state !== LoaderTaskItemState.PENDING) {
+            continue
+          }
+
+          const { workload } = evaluateTaskWorkload
+          this.workload += workload
+          this.processingItemCount += 1
+          item.state = LoaderTaskItemState.PROCESSING
+          queueTaskList.push({
+            taskPromise: () =>
+              taskPromise().then(() => {
+                this.workload -= workload
+                this.processingItemCount -= 1
+                task.loaderTasks.delete(item)
+                if (task.loaderTasks.size === 0) {
+                  this.knowledgeItemProcessingQueueMappingPromise.delete(task)
+                  resolve()
+                }
+                this.processingQueueHandle()
+              }),
+            resolve: () => {},
+            evaluateTaskWorkload
+          })
+        }
       }
+      return queueTaskList
     }
-
-    if (item.type === 'sitemap') {
-      const content = item.content as string
-      // @ts-ignore loader type
-      const loaderReturn = await ragApplication.addLoader(
-        new SitemapLoader({ url: content, chunkSize: base.chunkSize, chunkOverlap: base.chunkOverlap }) as any,
-        forceReload
-      )
-      return {
-        entriesAdded: loaderReturn.entriesAdded,
-        uniqueId: loaderReturn.uniqueId,
-        uniqueIds: [loaderReturn.uniqueId],
-        loaderType: loaderReturn.loaderType
-      } as LoaderReturn
+    const subTasks = getSubtasksUntilMaximumLoad()
+    if (subTasks.length > 0) {
+      const subTaskPromises = subTasks.map(({ taskPromise }) => taskPromise())
+      Promise.all(subTaskPromises).then(() => {
+        subTasks.forEach(({ resolve }) => resolve())
+      })
     }
+  }
 
-    if (item.type === 'note') {
-      const content = item.content as string
-      console.debug('chunkSize', base.chunkSize)
-      const loaderReturn = await ragApplication.addLoader(
-        new TextLoader({ text: content, chunkSize: base.chunkSize, chunkOverlap: base.chunkOverlap }),
-        forceReload
-      )
-      return {
-        entriesAdded: loaderReturn.entriesAdded,
-        uniqueId: loaderReturn.uniqueId,
-        uniqueIds: [loaderReturn.uniqueId],
-        loaderType: loaderReturn.loaderType
-      } as LoaderReturn
-    }
+  private appendProcessingQueue(task: LoaderTask): Promise<LoaderReturn> {
+    return new Promise((resolve) => {
+      this.knowledgeItemProcessingQueueMappingPromise.set(loaderTaskIntoOfSet(task), () => {
+        resolve(task.loaderDoneReturn!)
+      })
+    })
+  }
 
-    if (item.type === 'file') {
-      const file = item.content as FileType
+  public add = (_: Electron.IpcMainInvokeEvent, options: KnowledgeBaseAddItemOptions): Promise<LoaderReturn> => {
+    return new Promise((resolve) => {
+      const { base, item, forceReload = false } = options
+      const optionsNonNullableAttribute = { base, item, forceReload }
+      this.getRagApplication(base)
+        .then((ragApplication) => {
+          const task = (() => {
+            switch (item.type) {
+              case 'file':
+                return this.fileTask(ragApplication, optionsNonNullableAttribute)
+              case 'directory':
+                return this.directoryTask(ragApplication, optionsNonNullableAttribute)
+              case 'url':
+                return this.urlTask(ragApplication, optionsNonNullableAttribute)
+              case 'sitemap':
+                return this.sitemapTask(ragApplication, optionsNonNullableAttribute)
+              case 'note':
+                return this.noteTask(ragApplication, optionsNonNullableAttribute)
+              default:
+                return null
+            }
+          })()
 
-      return await addFileLoader(ragApplication, file, base, forceReload)
-    }
-
-    return { entriesAdded: 0, uniqueId: '', uniqueIds: [''], loaderType: '' }
+          if (task) {
+            this.appendProcessingQueue(task).then(() => {
+              resolve(task.loaderDoneReturn!)
+            })
+            this.processingQueueHandle()
+          } else {
+            resolve(KnowledgeService.ERROR_LOADER_RETURN)
+          }
+        })
+        .catch((err) => {
+          Logger.error(err)
+          resolve(KnowledgeService.ERROR_LOADER_RETURN)
+        })
+    })
   }
 
   public remove = async (
@@ -180,7 +459,7 @@ class KnowledgeService {
     { uniqueId, uniqueIds, base }: { uniqueId: string; uniqueIds: string[]; base: KnowledgeBaseParams }
   ): Promise<void> => {
     const ragApplication = await this.getRagApplication(base)
-    console.debug(`[ KnowledgeService Remove Item UniqueId: ${uniqueId}]`)
+    console.log(`[ KnowledgeService Remove Item UniqueId: ${uniqueId}]`)
     for (const id of uniqueIds) {
       await ragApplication.deleteLoader(id)
     }
@@ -193,6 +472,16 @@ class KnowledgeService {
     const ragApplication = await this.getRagApplication(base)
     return await ragApplication.search(search)
   }
+
+  public rerank = async (
+    _: Electron.IpcMainInvokeEvent,
+    { search, base, results }: { search: string; base: KnowledgeBaseParams; results: ExtractChunkData[] }
+  ): Promise<ExtractChunkData[]> => {
+    if (results.length === 0) {
+      return results
+    }
+    return await new Reranker(base).rerank(search, results)
+  }
 }
 
 export default new KnowledgeService()

src/main/services/MCPService.ts

@@ -0,0 +1,454 @@
+import os from 'node:os'
+import path from 'node:path'
+
+import { isLinux, isMac, isWin } from '@main/constant'
+import { createInMemoryMCPServer } from '@main/mcpServers/factory'
+import { makeSureDirExists } from '@main/utils'
+import { getBinaryName, getBinaryPath } from '@main/utils/process'
+import { Client } from '@modelcontextprotocol/sdk/client/index.js'
+import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js'
+import { getDefaultEnvironment, StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js'
+import { InMemoryTransport } from '@modelcontextprotocol/sdk/inMemory'
+import { nanoid } from '@reduxjs/toolkit'
+import { GetMCPPromptResponse, MCPPrompt, MCPServer, MCPTool } from '@types'
+import { app } from 'electron'
+import Logger from 'electron-log'
+
+import { CacheService } from './CacheService'
+import { StreamableHTTPClientTransport, type StreamableHTTPClientTransportOptions } from './MCPStreamableHttpClient'
+
+// Generic type for caching wrapped functions
+type CachedFunction<T extends unknown[], R> = (...args: T) => Promise<R>
+
+/**
+ * Higher-order function to add caching capability to any async function
+ * @param fn The original function to be wrapped with caching
+ * @param getCacheKey Function to generate a cache key from the function arguments
+ * @param ttl Time to live for the cache entry in milliseconds
+ * @param logPrefix Prefix for log messages
+ * @returns The wrapped function with caching capability
+ */
+function withCache<T extends unknown[], R>(
+  fn: (...args: T) => Promise<R>,
+  getCacheKey: (...args: T) => string,
+  ttl: number,
+  logPrefix: string
+): CachedFunction<T, R> {
+  return async (...args: T): Promise<R> => {
+    const cacheKey = getCacheKey(...args)
+
+    if (CacheService.has(cacheKey)) {
+      Logger.info(`${logPrefix} loaded from cache`)
+      const cachedData = CacheService.get<R>(cacheKey)
+      if (cachedData) {
+        return cachedData
+      }
+    }
+
+    const result = await fn(...args)
+    CacheService.set(cacheKey, result, ttl)
+    return result
+  }
+}
+
+class McpService {
+  private clients: Map<string, Client> = new Map()
+
+  private getServerKey(server: MCPServer): string {
+    return JSON.stringify({
+      baseUrl: server.baseUrl,
+      command: server.command,
+      args: server.args,
+      registryUrl: server.registryUrl,
+      env: server.env,
+      id: server.id
+    })
+  }
+
+  constructor() {
+    this.initClient = this.initClient.bind(this)
+    this.listTools = this.listTools.bind(this)
+    this.callTool = this.callTool.bind(this)
+    this.listPrompts = this.listPrompts.bind(this)
+    this.getPrompt = this.getPrompt.bind(this)
+    this.closeClient = this.closeClient.bind(this)
+    this.removeServer = this.removeServer.bind(this)
+    this.restartServer = this.restartServer.bind(this)
+    this.stopServer = this.stopServer.bind(this)
+    this.cleanup = this.cleanup.bind(this)
+  }
+
+  async initClient(server: MCPServer): Promise<Client> {
+    const serverKey = this.getServerKey(server)
+
+    // Check if we already have a client for this server configuration
+    const existingClient = this.clients.get(serverKey)
+    if (existingClient) {
+      try {
+        // Check if the existing client is still connected
+        const pingResult = await existingClient.ping()
+        Logger.info(`[MCP] Ping result for ${server.name}:`, pingResult)
+        // If the ping fails, remove the client from the cache
+        // and create a new one
+        if (!pingResult) {
+          this.clients.delete(serverKey)
+        } else {
+          return existingClient
+        }
+      } catch (error) {
+        Logger.error(`[MCP] Error pinging server ${server.name}:`, error)
+        this.clients.delete(serverKey)
+      }
+    }
+    // Create new client instance for each connection
+    const client = new Client({ name: 'Cherry Studio', version: app.getVersion() }, { capabilities: {} })
+
+    const args = [...(server.args || [])]
+
+    let transport: StdioClientTransport | SSEClientTransport | InMemoryTransport | StreamableHTTPClientTransport
+
+    try {
+      // Create appropriate transport based on configuration
+      if (server.type === 'inMemory') {
+        Logger.info(`[MCP] Using in-memory transport for server: ${server.name}`)
+        const [clientTransport, serverTransport] = InMemoryTransport.createLinkedPair()
+        // start the in-memory server with the given name and environment variables
+        const inMemoryServer = createInMemoryMCPServer(server.name, args, server.env || {})
+        try {
+          await inMemoryServer.connect(serverTransport)
+          Logger.info(`[MCP] In-memory server started: ${server.name}`)
+        } catch (error) {
+          Logger.error(`[MCP] Error starting in-memory server: ${error}`)
+          throw new Error(`Failed to start in-memory server: ${error}`)
+        }
+        // set the client transport to the client
+        transport = clientTransport
+      } else if (server.baseUrl) {
+        if (server.type === 'streamableHttp') {
+          transport = new StreamableHTTPClientTransport(
+            new URL(server.baseUrl!),
+            {} as StreamableHTTPClientTransportOptions
+          )
+        } else if (server.type === 'sse') {
+          transport = new SSEClientTransport(new URL(server.baseUrl!))
+        } else {
+          throw new Error('Invalid server type')
+        }
+      } else if (server.command) {
+        let cmd = server.command
+
+        if (server.command === 'npx' || server.command === 'bun' || server.command === 'bunx') {
+          cmd = await getBinaryPath('bun')
+          Logger.info(`[MCP] Using command: ${cmd}`)
+
+          // add -x to args if args exist
+          if (args && args.length > 0) {
+            if (!args.includes('-y')) {
+              !args.includes('-y') && args.unshift('-y')
+            }
+            if (!args.includes('x')) {
+              args.unshift('x')
+            }
+          }
+          if (server.registryUrl) {
+            server.env = {
+              ...server.env,
+              NPM_CONFIG_REGISTRY: server.registryUrl
+            }
+
+            // if the server name is mcp-auto-install, use the mcp-registry.json file in the bin directory
+            if (server.name.includes('mcp-auto-install')) {
+              const binPath = await getBinaryPath()
+              makeSureDirExists(binPath)
+              server.env.MCP_REGISTRY_PATH = path.join(binPath, '..', 'config', 'mcp-registry.json')
+            }
+          }
+        } else if (server.command === 'uvx' || server.command === 'uv') {
+          cmd = await getBinaryPath(server.command)
+          if (server.registryUrl) {
+            server.env = {
+              ...server.env,
+              UV_DEFAULT_INDEX: server.registryUrl,
+              PIP_INDEX_URL: server.registryUrl
+            }
+          }
+        }
+
+        Logger.info(`[MCP] Starting server with command: ${cmd} ${args ? args.join(' ') : ''}`)
+        // Logger.info(`[MCP] Environment variables for server:`, server.env)
+
+        transport = new StdioClientTransport({
+          command: cmd,
+          args,
+          env: {
+            ...getDefaultEnvironment(),
+            PATH: this.getEnhancedPath(process.env.PATH || ''),
+            ...server.env
+          },
+          stderr: 'pipe'
+        })
+        transport.stderr?.on('data', (data) =>
+          Logger.info(`[MCP] Stdio stderr for server: ${server.name} `, data.toString())
+        )
+      } else {
+        throw new Error('Either baseUrl or command must be provided')
+      }
+
+      await client.connect(transport)
+
+      // Store the new client in the cache
+      this.clients.set(serverKey, client)
+
+      Logger.info(`[MCP] Activated server: ${server.name}`)
+      return client
+    } catch (error: any) {
+      Logger.error(`[MCP] Error activating server ${server.name}:`, error)
+      throw error
+    }
+  }
+
+  async closeClient(serverKey: string) {
+    const client = this.clients.get(serverKey)
+    if (client) {
+      // Remove the client from the cache
+      await client.close()
+      Logger.info(`[MCP] Closed server: ${serverKey}`)
+      this.clients.delete(serverKey)
+      CacheService.remove(`mcp:list_tool:${serverKey}`)
+      Logger.info(`[MCP] Cleared cache for server: ${serverKey}`)
+    } else {
+      Logger.warn(`[MCP] No client found for server: ${serverKey}`)
+    }
+  }
+
+  async stopServer(_: Electron.IpcMainInvokeEvent, server: MCPServer) {
+    const serverKey = this.getServerKey(server)
+    Logger.info(`[MCP] Stopping server: ${server.name}`)
+    await this.closeClient(serverKey)
+  }
+
+  async removeServer(_: Electron.IpcMainInvokeEvent, server: MCPServer) {
+    const serverKey = this.getServerKey(server)
+    const existingClient = this.clients.get(serverKey)
+    if (existingClient) {
+      await this.closeClient(serverKey)
+    }
+  }
+
+  async restartServer(_: Electron.IpcMainInvokeEvent, server: MCPServer) {
+    Logger.info(`[MCP] Restarting server: ${server.name}`)
+    const serverKey = this.getServerKey(server)
+    await this.closeClient(serverKey)
+    await this.initClient(server)
+  }
+
+  async cleanup() {
+    for (const [key] of this.clients) {
+      try {
+        await this.closeClient(key)
+      } catch (error) {
+        Logger.error(`[MCP] Failed to close client: ${error}`)
+      }
+    }
+  }
+
+  private async listToolsImpl(server: MCPServer): Promise<MCPTool[]> {
+    Logger.info(`[MCP] Listing tools for server: ${server.name}`)
+    const client = await this.initClient(server)
+    try {
+      const { tools } = await client.listTools()
+      const serverTools: MCPTool[] = []
+      tools.map((tool: any) => {
+        const serverTool: MCPTool = {
+          ...tool,
+          id: `f${nanoid()}`,
+          serverId: server.id,
+          serverName: server.name
+        }
+        serverTools.push(serverTool)
+      })
+      return serverTools
+    } catch (error) {
+      Logger.error(`[MCP] Failed to list tools for server: ${server.name}`, error)
+      return []
+    }
+  }
+
+  async listTools(_: Electron.IpcMainInvokeEvent, server: MCPServer) {
+    const cachedListTools = withCache<[MCPServer], MCPTool[]>(
+      this.listToolsImpl.bind(this),
+      (server) => {
+        const serverKey = this.getServerKey(server)
+        return `mcp:list_tool:${serverKey}`
+      },
+      5 * 60 * 1000, // 5 minutes TTL
+      `[MCP] Tools from ${server.name}`
+    )
+
+    return cachedListTools(server)
+  }
+
+  /**
+   * Call a tool on an MCP server
+   */
+  public async callTool(
+    _: Electron.IpcMainInvokeEvent,
+    { server, name, args }: { server: MCPServer; name: string; args: any }
+  ): Promise<any> {
+    try {
+      Logger.info('[MCP] Calling:', server.name, name, args)
+      const client = await this.initClient(server)
+      const result = await client.callTool({ name, arguments: args })
+      return result
+    } catch (error) {
+      Logger.error(`[MCP] Error calling tool ${name} on ${server.name}:`, error)
+      throw error
+    }
+  }
+
+  public async getInstallInfo() {
+    const dir = path.join(os.homedir(), '.cherrystudio', 'bin')
+    const uvName = await getBinaryName('uv')
+    const bunName = await getBinaryName('bun')
+    const uvPath = path.join(dir, uvName)
+    const bunPath = path.join(dir, bunName)
+    return { dir, uvPath, bunPath }
+  }
+
+  /**
+   * List prompts available on an MCP server
+   */
+  private async listPromptsImpl(server: MCPServer): Promise<MCPPrompt[]> {
+    Logger.info(`[MCP] Listing prompts for server: ${server.name}`)
+    const client = await this.initClient(server)
+    try {
+      const { prompts } = await client.listPrompts()
+      const serverPrompts = prompts.map((prompt: any) => ({
+        ...prompt,
+        id: `p${nanoid()}`,
+        serverId: server.id,
+        serverName: server.name
+      }))
+      return serverPrompts
+    } catch (error) {
+      Logger.error(`[MCP] Failed to list prompts for server: ${server.name}`, error)
+      return []
+    }
+  }
+
+  /**
+   * List prompts available on an MCP server with caching
+   */
+  public async listPrompts(_: Electron.IpcMainInvokeEvent, server: MCPServer): Promise<MCPPrompt[]> {
+    const cachedListPrompts = withCache<[MCPServer], MCPPrompt[]>(
+      this.listPromptsImpl.bind(this),
+      (server) => {
+        const serverKey = this.getServerKey(server)
+        return `mcp:list_prompts:${serverKey}`
+      },
+      60 * 60 * 1000, // 60 minutes TTL
+      `[MCP] Prompts from ${server.name}`
+    )
+    return cachedListPrompts(server)
+  }
+
+  /**
+   * Get a specific prompt from an MCP server (implementation)
+   */
+  private async getPromptImpl(
+    server: MCPServer,
+    name: string,
+    args?: Record<string, any>
+  ): Promise<GetMCPPromptResponse> {
+    Logger.info(`[MCP] Getting prompt ${name} from server: ${server.name}`)
+    const client = await this.initClient(server)
+    return await client.getPrompt({ name, arguments: args })
+  }
+
+  /**
+   * Get a specific prompt from an MCP server with caching
+   */
+  public async getPrompt(
+    _: Electron.IpcMainInvokeEvent,
+    { server, name, args }: { server: MCPServer; name: string; args?: Record<string, any> }
+  ): Promise<GetMCPPromptResponse> {
+    const cachedGetPrompt = withCache<[MCPServer, string, Record<string, any> | undefined], GetMCPPromptResponse>(
+      this.getPromptImpl.bind(this),
+      (server, name, args) => {
+        const serverKey = this.getServerKey(server)
+        const argsKey = args ? JSON.stringify(args) : 'no-args'
+        return `mcp:get_prompt:${serverKey}:${name}:${argsKey}`
+      },
+      30 * 60 * 1000, // 30 minutes TTL
+      `[MCP] Prompt ${name} from ${server.name}`
+    )
+    return await cachedGetPrompt(server, name, args)
+  }
+
+  /**
+   * Get enhanced PATH including common tool locations
+   */
+  private getEnhancedPath(originalPath: string): string {
+    // 将原始 PATH 按分隔符分割成数组
+    const pathSeparator = process.platform === 'win32' ? ';' : ':'
+    const existingPaths = new Set(originalPath.split(pathSeparator).filter(Boolean))
+    const homeDir = process.env.HOME || process.env.USERPROFILE || ''
+
+    // 定义要添加的新路径
+    const newPaths: string[] = []
+
+    if (isMac) {
+      newPaths.push(
+        '/bin',
+        '/usr/bin',
+        '/usr/local/bin',
+        '/usr/local/sbin',
+        '/opt/homebrew/bin',
+        '/opt/homebrew/sbin',
+        '/usr/local/opt/node/bin',
+        `${homeDir}/.nvm/current/bin`,
+        `${homeDir}/.npm-global/bin`,
+        `${homeDir}/.yarn/bin`,
+        `${homeDir}/.cargo/bin`,
+        `${homeDir}/.cherrystudio/bin`,
+        '/opt/local/bin'
+      )
+    }
+
+    if (isLinux) {
+      newPaths.push(
+        '/bin',
+        '/usr/bin',
+        '/usr/local/bin',
+        `${homeDir}/.nvm/current/bin`,
+        `${homeDir}/.npm-global/bin`,
+        `${homeDir}/.yarn/bin`,
+        `${homeDir}/.cargo/bin`,
+        `${homeDir}/.cherrystudio/bin`,
+        '/snap/bin'
+      )
+    }
+
+    if (isWin) {
+      newPaths.push(
+        `${process.env.APPDATA}\\npm`,
+        `${homeDir}\\AppData\\Local\\Yarn\\bin`,
+        `${homeDir}\\.cargo\\bin`,
+        `${homeDir}\\.cherrystudio\\bin`
+      )
+    }
+
+    // 只添加不存在的路径
+    newPaths.forEach((path) => {
+      if (path && !existingPaths.has(path)) {
+        existingPaths.add(path)
+      }
+    })
+
+    // 转换回字符串
+    return Array.from(existingPaths).join(pathSeparator)
+  }
+}
+
+const mcpService = new McpService()
+export default mcpService

src/main/services/MCPStreamableHttpClient.ts

@@ -0,0 +1,394 @@
+import { auth, AuthResult, OAuthClientProvider, UnauthorizedError } from '@modelcontextprotocol/sdk/client/auth.js'
+import { Transport } from '@modelcontextprotocol/sdk/shared/transport.js'
+import { JSONRPCMessage, JSONRPCMessageSchema } from '@modelcontextprotocol/sdk/types.js'
+
+export class StreamableHTTPError extends Error {
+  constructor(
+    public readonly code: number | undefined,
+    message: string | undefined,
+    public readonly event: ErrorEvent
+  ) {
+    super(`Streamable HTTP error: ${message}`)
+  }
+}
+
+/**
+ * Configuration options for the `StreamableHTTPClientTransport`.
+ */
+export type StreamableHTTPClientTransportOptions = {
+  /**
+   * An OAuth client provider to use for authentication.
+   *
+   * When an `authProvider` is specified and the connection is started:
+   * 1. The connection is attempted with any existing access token from the `authProvider`.
+   * 2. If the access token has expired, the `authProvider` is used to refresh the token.
+   * 3. If token refresh fails or no access token exists, and auth is required, `OAuthClientProvider.redirectToAuthorization` is called, and an `UnauthorizedError` will be thrown from `connect`/`start`.
+   *
+   * After the user has finished authorizing via their user agent, and is redirected back to the MCP client application, call `StreamableHTTPClientTransport.finishAuth` with the authorization code before retrying the connection.
+   *
+   * If an `authProvider` is not provided, and auth is required, an `UnauthorizedError` will be thrown.
+   *
+   * `UnauthorizedError` might also be thrown when sending any message over the transport, indicating that the session has expired, and needs to be re-authed and reconnected.
+   */
+  authProvider?: OAuthClientProvider
+
+  /**
+   * Customizes HTTP requests to the server.
+   */
+  requestInit?: RequestInit
+}
+
+/**
+ * Client transport for Streamable HTTP: this implements the MCP Streamable HTTP transport specification.
+ * It will connect to a server using HTTP POST for sending messages and HTTP GET with Server-Sent Events
+ * for receiving messages.
+ */
+export class StreamableHTTPClientTransport implements Transport {
+  private _activeStreams: Map<string, ReadableStreamDefaultReader<Uint8Array>> = new Map()
+  private _abortController?: AbortController
+  private _url: URL
+  private _requestInit?: RequestInit
+  private _authProvider?: OAuthClientProvider
+  private _sessionId?: string
+  private _lastEventId?: string
+
+  onclose?: () => void
+  onerror?: (error: Error) => void
+  onmessage?: (message: JSONRPCMessage) => void
+
+  constructor(url: URL, opts?: StreamableHTTPClientTransportOptions) {
+    this._url = url
+    this._requestInit = opts?.requestInit
+    this._authProvider = opts?.authProvider
+  }
+
+  private async _authThenStart(): Promise<void> {
+    if (!this._authProvider) {
+      throw new UnauthorizedError('No auth provider')
+    }
+
+    let result: AuthResult
+    try {
+      result = await auth(this._authProvider, { serverUrl: this._url })
+    } catch (error) {
+      this.onerror?.(error as Error)
+      throw error
+    }
+
+    if (result !== 'AUTHORIZED') {
+      throw new UnauthorizedError()
+    }
+
+    return await this._startOrAuth()
+  }
+
+  private async _commonHeaders(): Promise<HeadersInit> {
+    const headers: HeadersInit = {}
+    if (this._authProvider) {
+      const tokens = await this._authProvider.tokens()
+      if (tokens) {
+        headers['Authorization'] = `Bearer ${tokens.access_token}`
+      }
+    }
+
+    if (this._sessionId) {
+      headers['mcp-session-id'] = this._sessionId
+    }
+
+    return headers
+  }
+
+  private async _startOrAuth(): Promise<void> {
+    try {
+      // Try to open an initial SSE stream with GET to listen for server messages
+      // This is optional according to the spec - server may not support it
+      const commonHeaders = await this._commonHeaders()
+      const headers = new Headers(commonHeaders)
+      headers.set('Accept', 'text/event-stream')
+
+      // Include Last-Event-ID header for resumable streams
+      if (this._lastEventId) {
+        headers.set('last-event-id', this._lastEventId)
+      }
+
+      // 删除可能存在的HTTP/2伪头部
+      if (headers.has(':path')) {
+        headers.delete(':path')
+      }
+      if (headers.has(':method')) {
+        headers.delete(':method')
+      }
+      if (headers.has(':authority')) {
+        headers.delete(':authority')
+      }
+      if (headers.has(':scheme')) {
+        headers.delete(':scheme')
+      }
+
+      const response = await fetch(this._url, {
+        method: 'GET',
+        headers,
+        signal: this._abortController?.signal
+      })
+
+      if (response.status === 405) {
+        // Server doesn't support GET for SSE, which is allowed by the spec
+        // We'll rely on SSE responses to POST requests for communication
+        return
+      }
+
+      if (!response.ok) {
+        if (response.status === 401 && this._authProvider) {
+          // Need to authenticate
+          return await this._authThenStart()
+        }
+
+        const error = new Error(`Failed to open SSE stream: ${response.status} ${response.statusText}`)
+        this.onerror?.(error)
+        throw error
+      }
+
+      // Successful connection, handle the SSE stream as a standalone listener
+      const streamId = `initial-${Date.now()}`
+      this._handleSseStream(response.body, streamId)
+    } catch (error) {
+      this.onerror?.(error as Error)
+      throw error
+    }
+  }
+
+  async start() {
+    if (this._activeStreams.size > 0) {
+      throw new Error(
+        'StreamableHTTPClientTransport already started! If using Client class, note that connect() calls start() automatically.'
+      )
+    }
+
+    this._abortController = new AbortController()
+    return await this._startOrAuth()
+  }
+
+  /**
+   * Call this method after the user has finished authorizing via their user agent and is redirected back to the MCP client application. This will exchange the authorization code for an access token, enabling the next connection attempt to successfully auth.
+   */
+  async finishAuth(authorizationCode: string): Promise<void> {
+    if (!this._authProvider) {
+      throw new UnauthorizedError('No auth provider')
+    }
+
+    const result = await auth(this._authProvider, { serverUrl: this._url, authorizationCode })
+    if (result !== 'AUTHORIZED') {
+      throw new UnauthorizedError('Failed to authorize')
+    }
+  }
+
+  async close(): Promise<void> {
+    // Close all active streams
+    for (const reader of this._activeStreams.values()) {
+      try {
+        reader.cancel()
+      } catch (error) {
+        this.onerror?.(error as Error)
+      }
+    }
+    this._activeStreams.clear()
+
+    // Abort any pending requests
+    this._abortController?.abort()
+
+    // If we have a session ID, send a DELETE request to explicitly terminate the session
+    if (this._sessionId) {
+      try {
+        const commonHeaders = await this._commonHeaders()
+        const response = await fetch(this._url, {
+          method: 'DELETE',
+          headers: commonHeaders,
+          signal: this._abortController?.signal
+        })
+
+        if (!response.ok) {
+          // Server might respond with 405 if it doesn't support explicit session termination
+          // We don't throw an error in that case
+          if (response.status !== 405) {
+            const text = await response.text().catch(() => null)
+            throw new Error(`Error terminating session (HTTP ${response.status}): ${text}`)
+          }
+        }
+      } catch (error) {
+        // We still want to invoke onclose even if the session termination fails
+        this.onerror?.(error as Error)
+      }
+    }
+
+    this.onclose?.()
+  }
+
+  async send(message: JSONRPCMessage | JSONRPCMessage[]): Promise<void> {
+    try {
+      const commonHeaders = await this._commonHeaders()
+      const headers = new Headers({ ...commonHeaders, ...this._requestInit?.headers })
+      headers.set('content-type', 'application/json')
+      headers.set('accept', 'application/json, text/event-stream')
+
+      // 添加错误处理，确保不使用HTTP/2伪头部
+      // 删除可能存在的HTTP/2伪头部
+      if (headers.has(':path')) {
+        headers.delete(':path')
+      }
+      if (headers.has(':method')) {
+        headers.delete(':method')
+      }
+      if (headers.has(':authority')) {
+        headers.delete(':authority')
+      }
+      if (headers.has(':scheme')) {
+        headers.delete(':scheme')
+      }
+
+      const init = {
+        ...this._requestInit,
+        method: 'POST',
+        headers,
+        body: JSON.stringify(message),
+        signal: this._abortController?.signal
+      }
+
+      const response = await fetch(this._url, init)
+
+      // Handle session ID received during initialization
+      const sessionId = response.headers.get('mcp-session-id')
+      if (sessionId) {
+        this._sessionId = sessionId
+      }
+
+      if (!response.ok) {
+        if (response.status === 401 && this._authProvider) {
+          const result = await auth(this._authProvider, { serverUrl: this._url })
+          if (result !== 'AUTHORIZED') {
+            throw new UnauthorizedError()
+          }
+
+          // Purposely _not_ awaited, so we don't call onerror twice
+          return this.send(message)
+        }
+
+        const text = await response.text().catch(() => null)
+        throw new Error(`Error POSTing to endpoint (HTTP ${response.status}): ${text}`)
+      }
+
+      // If the response is 202 Accepted, there's no body to process
+      if (response.status === 202) {
+        return
+      }
+
+      // Get original message(s) for detecting request IDs
+      const messages = Array.isArray(message) ? message : [message]
+
+      // Extract IDs from request messages for tracking responses
+      const requestIds = messages
+        .filter((msg) => 'method' in msg && 'id' in msg)
+        .map((msg) => ('id' in msg ? msg.id : undefined))
+        .filter((id) => id !== undefined)
+
+      // If we have request IDs and an SSE response, create a unique stream ID
+      const hasRequests = requestIds.length > 0
+
+      // Check the response type
+      const contentType = response.headers.get('content-type')
+
+      if (hasRequests) {
+        if (contentType?.includes('text/event-stream')) {
+          // For streaming responses, create a unique stream ID based on request IDs
+          const streamId = `req-${requestIds.join('-')}-${Date.now()}`
+          this._handleSseStream(response.body, streamId)
+        } else if (contentType?.includes('application/json')) {
+          // For non-streaming servers, we might get direct JSON responses
+          const data = await response.json()
+          const responseMessages = Array.isArray(data)
+            ? data.map((msg) => JSONRPCMessageSchema.parse(msg))
+            : [JSONRPCMessageSchema.parse(data)]
+
+          for (const msg of responseMessages) {
+            this.onmessage?.(msg)
+          }
+        }
+      }
+    } catch (error) {
+      this.onerror?.(error as Error)
+      throw error
+    }
+  }
+
+  private _handleSseStream(stream: ReadableStream<Uint8Array> | null, streamId: string): void {
+    if (!stream) {
+      return
+    }
+
+    // Set up stream handling for server-sent events
+    const reader = stream.getReader()
+    this._activeStreams.set(streamId, reader)
+    const decoder = new TextDecoder()
+    let buffer = ''
+
+    const processStream = async () => {
+      try {
+        while (true) {
+          const { done, value } = await reader.read()
+          if (done) {
+            // Stream closed by server
+            this._activeStreams.delete(streamId)
+            break
+          }
+
+          buffer += decoder.decode(value, { stream: true })
+
+          // Process SSE messages in the buffer
+          const events = buffer.split('\n\n')
+          buffer = events.pop() || ''
+
+          for (const event of events) {
+            const lines = event.split('\n')
+            let id: string | undefined
+            let eventType: string | undefined
+            let data: string | undefined
+
+            // Parse SSE message according to the format
+            for (const line of lines) {
+              if (line.startsWith('id:')) {
+                id = line.slice(3).trim()
+              } else if (line.startsWith('event:')) {
+                eventType = line.slice(6).trim()
+              } else if (line.startsWith('data:')) {
+                data = line.slice(5).trim()
+              }
+            }
+
+            // Update last event ID if provided by server
+            // As per spec: the ID MUST be globally unique across all streams within that session
+            if (id) {
+              this._lastEventId = id
+            }
+
+            // Handle message event
+            if (data) {
+              // Default event type is 'message' per SSE spec if not specified
+              if (!eventType || eventType === 'message') {
+                try {
+                  const message = JSONRPCMessageSchema.parse(JSON.parse(data))
+                  this.onmessage?.(message)
+                } catch (error) {
+                  this.onerror?.(error as Error)
+                }
+              }
+            }
+          }
+        }
+      } catch (error) {
+        this._activeStreams.delete(streamId)
+        this.onerror?.(error as Error)
+      }
+    }
+
+    processStream()
+  }
+}

src/main/services/MemoryFileService.ts

@@ -0,0 +1,310 @@
+import log from 'electron-log'
+import { promises as fs } from 'fs'
+import path from 'path'
+
+import { getConfigDir } from '../utils/file'
+
+// 定义记忆文件路径
+const memoryDataPath = path.join(getConfigDir(), 'memory-data.json')
+// 定义长期记忆文件路径
+const longTermMemoryDataPath = path.join(getConfigDir(), 'long-term-memory-data.json')
+
+export class MemoryFileService {
+  constructor() {
+    this.registerIpcHandlers()
+  }
+
+  async loadData() {
+    try {
+      // 确保配置目录存在
+      const configDir = path.dirname(memoryDataPath)
+      try {
+        await fs.mkdir(configDir, { recursive: true })
+      } catch (mkdirError) {
+        log.warn('Failed to create config directory, it may already exist:', mkdirError)
+      }
+
+      // 检查文件是否存在
+      try {
+        await fs.access(memoryDataPath)
+      } catch (accessError) {
+        // 文件不存在，创建默认文件
+        log.info('Memory data file does not exist, creating default file')
+        const defaultData = {
+          memoryLists: [
+            {
+              id: 'default',
+              name: '默认列表',
+              isActive: true
+            }
+          ],
+          shortMemories: [],
+          analyzeModel: 'gpt-3.5-turbo',
+          shortMemoryAnalyzeModel: 'gpt-3.5-turbo',
+          historicalContextAnalyzeModel: 'gpt-3.5-turbo',
+          vectorizeModel: 'gpt-3.5-turbo'
+        }
+        await fs.writeFile(memoryDataPath, JSON.stringify(defaultData, null, 2))
+        return defaultData
+      }
+
+      // 读取文件
+      const data = await fs.readFile(memoryDataPath, 'utf-8')
+      const parsedData = JSON.parse(data)
+      log.info('Memory data loaded successfully')
+      return parsedData
+    } catch (error) {
+      log.error('Failed to load memory data:', error)
+      return null
+    }
+  }
+
+  async saveData(data: any, forceOverwrite: boolean = false) {
+    try {
+      // 确保配置目录存在
+      const configDir = path.dirname(memoryDataPath)
+      try {
+        await fs.mkdir(configDir, { recursive: true })
+      } catch (mkdirError) {
+        log.warn('Failed to create config directory, it may already exist:', mkdirError)
+      }
+
+      // 如果强制覆盖，直接使用传入的数据
+      if (forceOverwrite) {
+        log.info('Force overwrite enabled for short memory data, using provided data directly')
+
+        // 确保数据包含必要的字段
+        const defaultData = {
+          memoryLists: [],
+          shortMemories: [],
+          analyzeModel: '',
+          shortMemoryAnalyzeModel: '',
+          historicalContextAnalyzeModel: '',
+          vectorizeModel: ''
+        }
+
+        // 合并默认数据和传入的数据，确保数据结构完整
+        const completeData = { ...defaultData, ...data }
+
+        // 保存数据
+        await fs.writeFile(memoryDataPath, JSON.stringify(completeData, null, 2))
+        log.info('Memory data saved successfully (force overwrite)')
+        return true
+      }
+
+      // 尝试读取现有数据并合并
+      let existingData = {}
+      try {
+        await fs.access(memoryDataPath)
+        const fileContent = await fs.readFile(memoryDataPath, 'utf-8')
+        existingData = JSON.parse(fileContent)
+        log.info('Existing memory data loaded for merging')
+      } catch (readError) {
+        log.warn('No existing memory data found or failed to read:', readError)
+        // 如果文件不存在或读取失败，使用空对象
+      }
+
+      // 合并数据，注意数组的处理
+      const mergedData = { ...existingData }
+
+      // 处理每个属性
+      Object.entries(data).forEach(([key, value]) => {
+        // 如果是数组属性，需要特殊处理
+        if (Array.isArray(value) && Array.isArray(mergedData[key])) {
+          // 对于 shortMemories 和 memories，直接使用传入的数组，完全替换现有的记忆
+          if (key === 'shortMemories' || key === 'memories') {
+            mergedData[key] = value
+            log.info(`Replacing ${key} array with provided data`)
+          } else {
+            // 其他数组属性，使用新值
+            mergedData[key] = value
+          }
+        } else {
+          // 非数组属性，直接使用新值
+          mergedData[key] = value
+        }
+      })
+
+      // 保存合并后的数据
+      await fs.writeFile(memoryDataPath, JSON.stringify(mergedData, null, 2))
+      log.info('Memory data saved successfully')
+      return true
+    } catch (error) {
+      log.error('Failed to save memory data:', error)
+      return false
+    }
+  }
+
+  async loadLongTermData() {
+    try {
+      // 确保配置目录存在
+      const configDir = path.dirname(longTermMemoryDataPath)
+      try {
+        await fs.mkdir(configDir, { recursive: true })
+      } catch (mkdirError) {
+        log.warn('Failed to create config directory, it may already exist:', mkdirError)
+      }
+
+      // 检查文件是否存在
+      try {
+        await fs.access(longTermMemoryDataPath)
+      } catch (accessError) {
+        // 文件不存在，创建默认文件
+        log.info('Long-term memory data file does not exist, creating default file')
+        const now = new Date().toISOString()
+        const defaultData = {
+          memoryLists: [
+            {
+              id: 'default',
+              name: '默认列表',
+              isActive: true,
+              createdAt: now,
+              updatedAt: now
+            }
+          ],
+          memories: [],
+          currentListId: 'default',
+          analyzeModel: 'gpt-3.5-turbo'
+        }
+        await fs.writeFile(longTermMemoryDataPath, JSON.stringify(defaultData, null, 2))
+        return defaultData
+      }
+
+      // 读取文件
+      const data = await fs.readFile(longTermMemoryDataPath, 'utf-8')
+      const parsedData = JSON.parse(data)
+      log.info('Long-term memory data loaded successfully')
+      return parsedData
+    } catch (error) {
+      log.error('Failed to load long-term memory data:', error)
+      return null
+    }
+  }
+
+  async saveLongTermData(data: any, forceOverwrite: boolean = false) {
+    try {
+      // 确保配置目录存在
+      const configDir = path.dirname(longTermMemoryDataPath)
+      try {
+        await fs.mkdir(configDir, { recursive: true })
+      } catch (mkdirError) {
+        log.warn('Failed to create config directory, it may already exist:', mkdirError)
+      }
+
+      // 如果强制覆盖，直接使用传入的数据
+      if (forceOverwrite) {
+        log.info('Force overwrite enabled, using provided data directly')
+
+        // 确保数据包含必要的字段
+        const defaultData = {
+          memoryLists: [],
+          memories: [],
+          currentListId: '',
+          analyzeModel: ''
+        }
+
+        // 合并默认数据和传入的数据，确保数据结构完整
+        const completeData = { ...defaultData, ...data }
+
+        // 保存数据
+        await fs.writeFile(longTermMemoryDataPath, JSON.stringify(completeData, null, 2))
+        log.info('Long-term memory data saved successfully (force overwrite)')
+        return true
+      }
+
+      // 尝试读取现有数据并合并
+      let existingData = {}
+      try {
+        await fs.access(longTermMemoryDataPath)
+        const fileContent = await fs.readFile(longTermMemoryDataPath, 'utf-8')
+        existingData = JSON.parse(fileContent)
+        log.info('Existing long-term memory data loaded for merging')
+      } catch (readError) {
+        log.warn('No existing long-term memory data found or failed to read:', readError)
+        // 如果文件不存在或读取失败，使用空对象
+      }
+
+      // 合并数据，注意数组的处理
+      const mergedData = { ...existingData }
+
+      // 处理每个属性
+      Object.entries(data).forEach(([key, value]) => {
+        // 如果是数组属性，需要特殊处理
+        if (Array.isArray(value) && Array.isArray(mergedData[key])) {
+          // 对于 memories 和 shortMemories，直接使用传入的数组，完全替换现有的记忆
+          if (key === 'memories' || key === 'shortMemories') {
+            mergedData[key] = value
+            log.info(`Replacing ${key} array with provided data`)
+          } else {
+            // 其他数组属性，使用新值
+            mergedData[key] = value
+          }
+        } else {
+          // 非数组属性，直接使用新值
+          mergedData[key] = value
+        }
+      })
+
+      // 保存合并后的数据
+      await fs.writeFile(longTermMemoryDataPath, JSON.stringify(mergedData, null, 2))
+      log.info('Long-term memory data saved successfully')
+      return true
+    } catch (error) {
+      log.error('Failed to save long-term memory data:', error)
+      return false
+    }
+  }
+
+  /**
+   * 删除指定ID的短期记忆
+   * @param id 要删除的短期记忆ID
+   * @returns 是否成功删除
+   */
+  async deleteShortMemoryById(id: string) {
+    try {
+      // 检查文件是否存在
+      try {
+        await fs.access(memoryDataPath)
+      } catch (accessError) {
+        log.error('Memory data file does not exist, cannot delete memory')
+        return false
+      }
+
+      // 读取文件
+      const fileContent = await fs.readFile(memoryDataPath, 'utf-8')
+      const data = JSON.parse(fileContent)
+
+      // 检查shortMemories数组是否存在
+      if (!data.shortMemories || !Array.isArray(data.shortMemories)) {
+        log.error('No shortMemories array found in memory data file')
+        return false
+      }
+
+      // 过滤掉要删除的记忆
+      const originalLength = data.shortMemories.length
+      data.shortMemories = data.shortMemories.filter((memory: any) => memory.id !== id)
+
+      // 如果长度没变，说明没有找到要删除的记忆
+      if (data.shortMemories.length === originalLength) {
+        log.warn(`Short memory with ID ${id} not found, nothing to delete`)
+        return false
+      }
+
+      // 写回文件
+      await fs.writeFile(memoryDataPath, JSON.stringify(data, null, 2))
+      log.info(`Successfully deleted short memory with ID ${id}`)
+      return true
+    } catch (error) {
+      log.error('Failed to delete short memory:', error)
+      return false
+    }
+  }
+
+  private registerIpcHandlers() {
+    // 注册处理函数已移至ipc.ts文件中
+    // 这里不需要重复注册
+  }
+}
+
+// 创建并导出MemoryFileService实例
+export const memoryFileService = new MemoryFileService()

src/main/services/NutstoreService.ts

@@ -0,0 +1,134 @@
+import path from 'node:path'
+
+import { NUTSTORE_HOST } from '@shared/config/nutstore'
+import { XMLParser } from 'fast-xml-parser'
+import { isNil, partial } from 'lodash'
+import { type FileStat } from 'webdav'
+
+interface OAuthResponse {
+  username: string
+  userid: string
+  access_token: string
+}
+
+interface WebDAVResponse {
+  multistatus: {
+    response: Array<{
+      href: string
+      propstat: {
+        prop: {
+          displayname: string
+          resourcetype: { collection?: any }
+          getlastmodified?: string
+          getcontentlength?: string
+          getcontenttype?: string
+        }
+        status: string
+      }
+    }>
+  }
+}
+
+export async function getNutstoreSSOUrl() {
+  const { createOAuthUrl } = await import('../integration/nutstore/sso/lib')
+
+  const url = createOAuthUrl({
+    app: 'cherrystudio'
+  })
+  return url
+}
+
+export async function decryptToken(token: string) {
+  const { decrypt } = await import('../integration/nutstore/sso/lib')
+  try {
+    const decrypted = decrypt('cherrystudio', token)
+    return JSON.parse(decrypted) as OAuthResponse
+  } catch (error) {
+    console.error('解密失败:', error)
+    return null
+  }
+}
+
+export async function getDirectoryContents(token: string, target: string): Promise<FileStat[]> {
+  const contents: FileStat[] = []
+  if (!target.startsWith('/')) {
+    target = '/' + target
+  }
+
+  let currentUrl = `${NUTSTORE_HOST}${target}`
+
+  while (true) {
+    const response = await fetch(currentUrl, {
+      method: 'PROPFIND',
+      headers: {
+        Authorization: `Basic ${token}`,
+        'Content-Type': 'application/xml',
+        Depth: '1'
+      },
+      body: `<?xml version="1.0" encoding="utf-8"?>
+        <propfind xmlns="DAV:">
+          <prop>
+            <displayname/>
+            <resourcetype/>
+            <getlastmodified/>
+            <getcontentlength/>
+            <getcontenttype/>
+          </prop>
+        </propfind>`
+    })
+
+    const text = await response.text()
+
+    const result = parseXml<WebDAVResponse>(text)
+    const items = Array.isArray(result.multistatus.response)
+      ? result.multistatus.response
+      : [result.multistatus.response]
+
+    // 跳过第一个条目（当前目录）
+    contents.push(...items.slice(1).map(partial(convertToFileStat, '/dav')))
+
+    const linkHeader = response.headers['link'] || response.headers['Link']
+    if (!linkHeader) {
+      break
+    }
+
+    const nextLink = extractNextLink(linkHeader)
+    if (!nextLink) {
+      break
+    }
+
+    currentUrl = decodeURI(nextLink)
+  }
+
+  return contents
+}
+
+function extractNextLink(linkHeader: string): string | null {
+  const matches = linkHeader.match(/<([^>]+)>;\s*rel="next"/)
+  return matches ? matches[1] : null
+}
+
+function convertToFileStat(serverBase: string, item: WebDAVResponse['multistatus']['response'][number]): FileStat {
+  const props = item.propstat.prop
+  const isDir = !isNil(props.resourcetype?.collection)
+  const href = decodeURIComponent(item.href)
+  const filename = serverBase === '/' ? href : path.posix.join('/', href.replace(serverBase, ''))
+
+  return {
+    filename: filename.endsWith('/') ? filename.slice(0, -1) : filename,
+    basename: path.basename(filename),
+    lastmod: props.getlastmodified || '',
+    size: props.getcontentlength ? parseInt(props.getcontentlength, 10) : 0,
+    type: isDir ? 'directory' : 'file',
+    etag: null,
+    mime: props.getcontenttype
+  }
+}
+
+function parseXml<T>(xml: string) {
+  const parser = new XMLParser({
+    attributeNamePrefix: '',
+    removeNSPrefix: true
+  })
+  return parser.parse(xml) as T
+}