feat: add fadeInWithBlur animation to Tailwind CSS and update Markdown component

- Introduced a new `fadeInWithBlur` keyframe animation in Tailwind CSS for enhanced visual effects.
- Removed inline fade animation styles from the `Markdown` component to streamline rendering.
- Updated the `SmoothFade` function to utilize the new animation, improving the user experience during content transitions.

.github/workflows/delete-branch.yml

@@ -0,0 +1,22 @@
+name: Delete merged branch
+on:
+  pull_request:
+    types:
+      - closed
+
+jobs:
+  delete-branch:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: write
+    if: github.event.pull_request.merged == true && github.event.pull_request.head.repo.full_name == github.repository
+    steps:
+      - name: Delete merged branch
+        uses: actions/github-script@v7
+        with:
+          script: |
+            github.rest.git.deleteRef({
+              owner: context.repo.owner,
+              repo: context.repo.repo,
+              ref: `heads/${context.payload.pull_request.head.ref}`,
+            })

.github/workflows/nightly-build.yml

@@ -98,7 +98,7 @@ jobs:
         env:
           GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
           NODE_OPTIONS: --max-old-space-size=8192
-          MAIN_VITE_CHERRYIN_CLIENT_SECRET: ${{ secrets.MAIN_VITE_CHERRYIN_CLIENT_SECRET }}
+          MAIN_VITE_CHERRYAI_CLIENT_SECRET: ${{ secrets.MAIN_VITE_CHERRYAI_CLIENT_SECRET }}
           MAIN_VITE_MINERU_API_KEY: ${{ vars.MAIN_VITE_MINERU_API_KEY }}
           RENDERER_VITE_AIHUBMIX_SECRET: ${{ vars.RENDERER_VITE_AIHUBMIX_SECRET }}
           RENDERER_VITE_PPIO_APP_SECRET: ${{ vars.RENDERER_VITE_PPIO_APP_SECRET }}
@@ -115,7 +115,7 @@ jobs:
           APPLE_TEAM_ID: ${{ vars.APPLE_TEAM_ID }}
           GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
           NODE_OPTIONS: --max-old-space-size=8192
-          MAIN_VITE_CHERRYIN_CLIENT_SECRET: ${{ secrets.MAIN_VITE_CHERRYIN_CLIENT_SECRET }}
+          MAIN_VITE_CHERRYAI_CLIENT_SECRET: ${{ secrets.MAIN_VITE_CHERRYAI_CLIENT_SECRET }}
           MAIN_VITE_MINERU_API_KEY: ${{ vars.MAIN_VITE_MINERU_API_KEY }}
           RENDERER_VITE_AIHUBMIX_SECRET: ${{ vars.RENDERER_VITE_AIHUBMIX_SECRET }}
           RENDERER_VITE_PPIO_APP_SECRET: ${{ vars.RENDERER_VITE_PPIO_APP_SECRET }}
@@ -127,7 +127,7 @@ jobs:
         env:
           GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
           NODE_OPTIONS: --max-old-space-size=8192
-          MAIN_VITE_CHERRYIN_CLIENT_SECRET: ${{ secrets.MAIN_VITE_CHERRYIN_CLIENT_SECRET }}
+          MAIN_VITE_CHERRYAI_CLIENT_SECRET: ${{ secrets.MAIN_VITE_CHERRYAI_CLIENT_SECRET }}
           MAIN_VITE_MINERU_API_KEY: ${{ vars.MAIN_VITE_MINERU_API_KEY }}
           RENDERER_VITE_AIHUBMIX_SECRET: ${{ vars.RENDERER_VITE_AIHUBMIX_SECRET }}
           RENDERER_VITE_PPIO_APP_SECRET: ${{ vars.RENDERER_VITE_PPIO_APP_SECRET }}

.github/workflows/pr-ci.yml

@@ -10,12 +10,14 @@ on:
       - main
       - develop
       - v2
+    types: [ready_for_review, synchronize, opened]
 
 jobs:
   build:
     runs-on: ubuntu-latest
     env:
       PRCI: true
+    if: github.event.pull_request.draft == false
 
     steps:
       - name: Check out Git repository

.github/workflows/release.yml

@@ -85,7 +85,7 @@ jobs:
         env:
           GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
           NODE_OPTIONS: --max-old-space-size=8192
-          MAIN_VITE_CHERRYIN_CLIENT_SECRET: ${{ secrets.MAIN_VITE_CHERRYIN_CLIENT_SECRET }}
+          MAIN_VITE_CHERRYAI_CLIENT_SECRET: ${{ secrets.MAIN_VITE_CHERRYAI_CLIENT_SECRET }}
           MAIN_VITE_MINERU_API_KEY: ${{ vars.MAIN_VITE_MINERU_API_KEY }}
           RENDERER_VITE_AIHUBMIX_SECRET: ${{ vars.RENDERER_VITE_AIHUBMIX_SECRET }}
           RENDERER_VITE_PPIO_APP_SECRET: ${{ vars.RENDERER_VITE_PPIO_APP_SECRET }}
@@ -103,7 +103,7 @@ jobs:
           APPLE_TEAM_ID: ${{ vars.APPLE_TEAM_ID }}
           GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
           NODE_OPTIONS: --max-old-space-size=8192
-          MAIN_VITE_CHERRYIN_CLIENT_SECRET: ${{ secrets.MAIN_VITE_CHERRYIN_CLIENT_SECRET }}
+          MAIN_VITE_CHERRYAI_CLIENT_SECRET: ${{ secrets.MAIN_VITE_CHERRYAI_CLIENT_SECRET }}
           MAIN_VITE_MINERU_API_KEY: ${{ vars.MAIN_VITE_MINERU_API_KEY }}
           RENDERER_VITE_AIHUBMIX_SECRET: ${{ vars.RENDERER_VITE_AIHUBMIX_SECRET }}
           RENDERER_VITE_PPIO_APP_SECRET: ${{ vars.RENDERER_VITE_PPIO_APP_SECRET }}
@@ -115,7 +115,7 @@ jobs:
         env:
           GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
           NODE_OPTIONS: --max-old-space-size=8192
-          MAIN_VITE_CHERRYIN_CLIENT_SECRET: ${{ secrets.MAIN_VITE_CHERRYIN_CLIENT_SECRET }}
+          MAIN_VITE_CHERRYAI_CLIENT_SECRET: ${{ secrets.MAIN_VITE_CHERRYAI_CLIENT_SECRET }}
           MAIN_VITE_MINERU_API_KEY: ${{ vars.MAIN_VITE_MINERU_API_KEY }}
           RENDERER_VITE_AIHUBMIX_SECRET: ${{ vars.RENDERER_VITE_AIHUBMIX_SECRET }}
           RENDERER_VITE_PPIO_APP_SECRET: ${{ vars.RENDERER_VITE_PPIO_APP_SECRET }}

.gitignore

@@ -54,6 +54,8 @@ local
 .qwen/*
 .trae/*
 .claude-code-router/*
+.codebuddy/*
+.zed/*
 CLAUDE.local.md
 
 # vitest
@@ -69,5 +71,3 @@ playwright-report
 test-results
 
 YOUR_MEMORY_FILE_PATH
-
-.sessions/

.oxlintrc.json

@@ -15,7 +15,7 @@
     ".gitignore",
     "scripts/cloudflare-worker.js",
     "src/main/integration/nutstore/sso/lib/**",
-    "src/main/integration/cherryin/index.js",
+    "src/main/integration/cherryai/index.js",
     "src/main/integration/nutstore/sso/lib/**",
     "src/renderer/src/ui/**",
     "packages/**/dist",

.vscode/settings.json

@@ -34,10 +34,10 @@
     "*.css": "tailwindcss"
   },
   "files.eol": "\n",
-  // "i18n-ally.displayLanguage": "zh-cn", // 界面显示语言
+  "i18n-ally.displayLanguage": "zh-cn",
   "i18n-ally.enabledFrameworks": ["react-i18next", "i18next"],
   "i18n-ally.enabledParsers": ["ts", "js", "json"], // 解析语言
-  "i18n-ally.fullReloadOnChanged": true,
+  "i18n-ally.fullReloadOnChanged": true, // 界面显示语言
   "i18n-ally.keystyle": "nested", // 翻译路径格式
   "i18n-ally.localesPaths": ["src/renderer/src/i18n/locales"],
   // "i18n-ally.namespace": true, // 开启命名空间

.yarn/patches/@ai-sdk-google-npm-2.0.14-376d8b03cc.patch

@@ -0,0 +1,36 @@
+diff --git a/dist/index.mjs b/dist/index.mjs
+index 110f37ec18c98b1d55ae2b73cc716194e6f9094d..91d0f336b318833c6cee9599fe91370c0ff75323 100644
+--- a/dist/index.mjs
++++ b/dist/index.mjs
+@@ -447,7 +447,10 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
+ }
+ 
+ // src/get-model-path.ts
+-function getModelPath(modelId) {
++function getModelPath(modelId, baseURL) {
++  if (baseURL?.includes('cherryin')) {
++    return `models/${modelId}`;
++  }
+   return modelId.includes("/") ? modelId : `models/${modelId}`;
+ }
+ 
+@@ -856,7 +859,8 @@ var GoogleGenerativeAILanguageModel = class {
+       rawValue: rawResponse
+     } = await postJsonToApi2({
+       url: `${this.config.baseURL}/${getModelPath(
+-        this.modelId
++        this.modelId,
++        this.config.baseURL
+       )}:generateContent`,
+       headers: mergedHeaders,
+       body: args,
+@@ -962,7 +966,8 @@ var GoogleGenerativeAILanguageModel = class {
+     );
+     const { responseHeaders, value: response } = await postJsonToApi2({
+       url: `${this.config.baseURL}/${getModelPath(
+-        this.modelId
++        this.modelId,
++        this.config.baseURL
+       )}:streamGenerateContent?alt=sse`,
+       headers,
+       body: args,

.yarn/patches/@anthropic-ai-claude-code-npm-1.0.118-bbf4e9e59f.patch

@@ -1,18 +0,0 @@
-diff --git a/sdk.mjs b/sdk.mjs
-index e2dbafb4e2faa1bf2b6b02f0009a2b9bbf57c757..3f07a1d5c2949a246fe5414e69ab45942fa605a2 100644
---- a/sdk.mjs
-+++ b/sdk.mjs
-@@ -6355,11 +6355,11 @@ class ProcessTransport {
-         prompt,
-         additionalDirectories = [],
-         cwd,
--        executable = isRunningWithBun() ? "bun" : "node",
-+        executable = process.execPath,
-         executableArgs = [],
-         extraArgs = {},
-         pathToClaudeCodeExecutable,
--        env = { ...process.env },
-+        env = { ...process.env, ELECTRON_RUN_AS_NODE: '1' },
-         stderr,
-         customSystemPrompt,
-         appendSystemPrompt,

AGENTS.md

@@ -1 +0,0 @@
-CLAUDE.md

CLAUDE.md

@@ -1,52 +1,127 @@
-# AI Assistant Guide
+# CLAUDE.md
 
-This file provides guidance to AI coding assistants when working with code in this repository. Adherence to these guidelines is crucial for maintaining code quality and consistency.
-
-## Guiding Principles
-
-- **Clarity and Simplicity**: Write code that is easy to understand and maintain.
-- **Consistency**: Follow existing patterns and conventions in the codebase.
-- **Correctness**: Ensure code is correct, well-tested, and robust.
-- **Efficiency**: Write performant code and use resources judiciously.
-
-## MUST Follow Rules
-
-1.  **Code Search**: Use `ast-grep` for semantic code pattern searches when available. Fallback to `rg` (ripgrep) or `grep` for text-based searches.
-2.  **UI Framework**: Exclusively use **HeroUI** for all new UI components. The use of `antd` or `styled-components` is strictly **PROHIBITED**.
-3.  **Quality Assurance**: **Always** run `yarn build:check` before finalizing your work or making any commits. This ensures code quality (linting, testing, and type checking).
-4.  **Centralized Logging**: Use the `loggerService` exclusively for all application logging (info, warn, error levels) with proper context. Do not use `console.log`.
-5.  **External Research**: Leverage `subagent` for gathering external information, including latest documentation, API references, news, or web-based research. This keeps the main conversation focused on the task at hand.
-6.  **Code Reviews**: Always seek a code review from a human developer before merging significant changes. This ensures adherence to project standards and catches potential issues.
-7.  **Documentation**: Update or create documentation for any new features, modules, or significant changes to existing functionality.
+This file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.
 
 ## Development Commands
 
-- **Install**: `yarn install`
-- **Development**: `yarn dev` - Runs Electron app in development mode
-- **Debug**: `yarn debug` - Starts with debugging enabled, use chrome://inspect
-- **Build Check**: `yarn build:check` - REQUIRED before commits (lint + test + typecheck)
-- **Test**: `yarn test` - Run all tests (Vitest)
-- **Single Test**: `yarn test:main` or `yarn test:renderer`
-- **Lint**: `yarn lint` - Fix linting issues and run typecheck
+### Environment Setup
 
-## Project Architecture
+- **Prerequisites**: Node.js v22.x.x or higher, Yarn 4.9.1
+- **Setup Yarn**: `corepack enable && corepack prepare yarn@4.9.1 --activate`
+- **Install Dependencies**: `yarn install`
+- **Add New Dependencies**: `yarn add -D` for renderer-specific dependencies, `yarn add` for others.
 
-### Electron Structure
-- **Main Process** (`src/main/`): Node.js backend with services (MCP, Knowledge, Storage, etc.)
-- **Renderer Process** (`src/renderer/`): React UI with Redux state management
-- **Preload Scripts** (`src/preload/`): Secure IPC bridge
+### Development
 
-### Key Components
-- **AI Core** (`src/renderer/src/aiCore/`): Middleware pipeline for multiple AI providers.
-- **Services** (`src/main/services/`): MCPService, KnowledgeService, WindowService, etc.
-- **Build System**: Electron-Vite with experimental rolldown-vite, yarn workspaces.
-- **State Management**: Redux Toolkit (`src/renderer/src/store/`) for predictable state.
-- **UI Components**: HeroUI (`@heroui/*`) for all new UI elements.
+- **Start Development**: `yarn dev` - Runs Electron app in development mode
+- **Debug Mode**: `yarn debug` - Starts with debugging enabled, use chrome://inspect
+
+### Testing & Quality
+
+- **Run Tests**: `yarn test` - Runs all tests (Vitest)
+- **Run E2E Tests**: `yarn test:e2e` - Playwright end-to-end tests
+- **Type Check**: `yarn typecheck` - Checks TypeScript for both node and web
+- **Lint**: `yarn lint` - ESLint with auto-fix
+- **Format**: `yarn format` - Biome formatting
+
+### Build & Release
+
+- **Build**: `yarn build` - Builds for production (includes typecheck)
+- **Platform-specific builds**:
+  - Windows: `yarn build:win`
+  - macOS: `yarn build:mac`
+  - Linux: `yarn build:linux`
+
+## Architecture Overview
+
+### Electron Multi-Process Architecture
+
+- **Main Process** (`src/main/`): Node.js backend handling system integration, file operations, and services
+- **Renderer Process** (`src/renderer/`): React-based UI running in Chromium
+- **Preload Scripts** (`src/preload/`): Secure bridge between main and renderer processes
+
+### Key Architectural Components
+
+#### Main Process Services (`src/main/services/`)
+
+- **MCPService**: Model Context Protocol server management
+- **KnowledgeService**: Document processing and knowledge base management
+- **FileStorage/S3Storage/WebDav**: Multiple storage backends
+- **WindowService**: Multi-window management (main, mini, selection windows)
+- **ProxyManager**: Network proxy handling
+- **SearchService**: Full-text search capabilities
+
+#### AI Core (`src/renderer/src/aiCore/`)
+
+- **Middleware System**: Composable pipeline for AI request processing
+- **Client Factory**: Supports multiple AI providers (OpenAI, Anthropic, Gemini, etc.)
+- **Stream Processing**: Real-time response handling
+
+#### State Management (`src/renderer/src/store/`)
+
+- **Redux Toolkit**: Centralized state management
+- **Persistent Storage**: Redux-persist for data persistence
+- **Thunks**: Async actions for complex operations
+
+#### Knowledge Management
+
+- **Embeddings**: Vector search with multiple providers (OpenAI, Voyage, etc.)
+- **OCR**: Document text extraction (system OCR, Doc2x, Mineru)
+- **Preprocessing**: Document preparation pipeline
+- **Loaders**: Support for various file formats (PDF, DOCX, EPUB, etc.)
+
+### Build System
+
+- **Electron-Vite**: Development and build tooling (v4.0.0)
+- **Rolldown-Vite**: Using experimental rolldown-vite instead of standard vite
+- **Workspaces**: Monorepo structure with `packages/` directory
+- **Multiple Entry Points**: Main app, mini window, selection toolbar
+- **Styled Components**: CSS-in-JS styling with SWC optimization
+
+### Testing Strategy
+
+- **Vitest**: Unit and integration testing
+- **Playwright**: End-to-end testing
+- **Component Testing**: React Testing Library
+- **Coverage**: Available via `yarn test:coverage`
+
+### Key Patterns
+
+- **IPC Communication**: Secure main-renderer communication via preload scripts
+- **Service Layer**: Clear separation between UI and business logic
+- **Plugin Architecture**: Extensible via MCP servers and middleware
+- **Multi-language Support**: i18n with dynamic loading
+- **Theme System**: Light/dark themes with custom CSS variables
+
+### UI Design
+
+The project is in the process of migrating from antd & styled-components to HeroUI. Please use HeroUI to build UI components. The use of antd and styled-components is prohibited.
+
+HeroUI Docs: https://www.heroui.com/docs/guide/introduction
+
+## Logging Standards
+
+### Usage
 
-### Logging
 ```typescript
+// Main process
 import { loggerService } from '@logger'
 const logger = loggerService.withContext('moduleName')
-// Renderer: loggerService.initWindowSource('windowName') first
+
+// Renderer process (set window source first)
+loggerService.initWindowSource('windowName')
+const logger = loggerService.withContext('moduleName')
+
+// Logging
 logger.info('message', CONTEXT)
+logger.error('message', new Error('error'), CONTEXT)
 ```
+
+### Log Levels (highest to lowest)
+
+- `error` - Critical errors causing crash/unusable functionality
+- `warn` - Potential issues that don't affect core functionality
+- `info` - Application lifecycle and key user actions
+- `verbose` - Detailed flow information for feature tracing
+- `debug` - Development diagnostic info (not for production)
+- `silly` - Extreme debugging, low-level information

electron-builder.yml

@@ -125,16 +125,59 @@ afterSign: scripts/notarize.js
 artifactBuildCompleted: scripts/artifact-build-completed.js
 releaseInfo:
   releaseNotes: |
-    🐛 问题修复：
-    - 修复 Anthropic API URL 处理，移除尾部斜杠并添加端点路径处理
-    - 修复 MessageEditor 缺少 QuickPanelProvider 包装的问题
-    - 修复 MiniWindow 高度问题
+    <!--LANG:en-->
+    🚀 New Features:
+    - Refactored AI core engine for more efficient and stable content generation
+    - Added support for multiple AI model providers: CherryIN, AiOnly
+    - Added API server functionality for external application integration
+    - Added PaddleOCR document recognition for enhanced document processing
+    - Added Anthropic OAuth authentication support
+    - Added data storage space limit notifications
+    - Added font settings for global and code fonts customization
+    - Added auto-copy feature after translation completion
+    - Added keyboard shortcuts: rename topic, edit last message, etc.
+    - Added text attachment preview for viewing file contents in messages
+    - Added custom window control buttons (minimize, maximize, close)
+    - Support for Qwen long-text (qwen-long) and document analysis (qwen-doc) models with native file uploads
+    - Support for Qwen image recognition models (Qwen-Image)
+    - Added iFlow CLI support
+    - Converted knowledge base and web search to tool-calling approach for better flexibility
 
-    🚀 性能优化：
-    - 优化输入栏提及模型状态缓存，在渲染间保持状态
-    - 重构网络搜索参数支持模型内置搜索，新增 OpenAI Chat 和 OpenRouter 支持
+    🎨 UI Improvements & Bug Fixes:
+    - Integrated HeroUI and Tailwind CSS framework
+    - Optimized message notification styles with unified toast component
+    - Moved free models to bottom with fixed position for easier access
+    - Refactored quick panel and input bar tools for smoother operation
+    - Optimized responsive design for navbar and sidebar
+    - Improved scrollbar component with horizontal scrolling support
+    - Fixed multiple translation issues: paste handling, file processing, state management
+    - Various UI optimizations and bug fixes
+    <!--LANG:zh-CN-->
+    🚀 新功能：
+    - 重构 AI 核心引擎，提供更高效稳定的内容生成
+    - 新增多个 AI 模型提供商支持：CherryIN、AiOnly
+    - 新增 API 服务器功能，支持外部应用集成
+    - 新增 PaddleOCR 文档识别，增强文档处理能力
+    - 新增 Anthropic OAuth 认证支持
+    - 新增数据存储空间限制提醒
+    - 新增字体设置，支持全局字体和代码字体自定义
+    - 新增翻译完成后自动复制功能
+    - 新增键盘快捷键：重命名主题、编辑最后一条消息等
+    - 新增文本附件预览，可查看消息中的文件内容
+    - 新增自定义窗口控制按钮（最小化、最大化、关闭）
+    - 支持通义千问长文本（qwen-long）和文档分析（qwen-doc）模型，原生文件上传
+    - 支持通义千问图像识别模型（Qwen-Image）
+    - 新增 iFlow CLI 支持
+    - 知识库和网页搜索转换为工具调用方式，提升灵活性
+
+    🎨 界面改进与问题修复：
+    - 集成 HeroUI 和 Tailwind CSS 框架
+    - 优化消息通知样式，统一 toast 组件
+    - 免费模型移至底部固定位置，便于访问
+    - 重构快捷面板和输入栏工具，操作更流畅
+    - 优化导航栏和侧边栏响应式设计
+    - 改进滚动条组件，支持水平滚动
+    - 修复多个翻译问题：粘贴处理、文件处理、状态管理
+    - 各种界面优化和问题修复
+    <!--LANG:END-->
 
-    🔧 重构改进：
-    - 更新 HeroUIProvider 导入路径，改善上下文管理
-    - 更新依赖项和 VSCode 开发环境配置
-    - 升级 @cherrystudio/ai-core 到 v1.0.0-alpha.17

electron.vite.config.ts

@@ -84,7 +84,6 @@ export default defineConfig({
       alias: {
         '@renderer': resolve('src/renderer/src'),
         '@shared': resolve('packages/shared'),
-        '@types': resolve('src/renderer/src/types'),
         '@logger': resolve('src/renderer/src/services/LoggerService'),
         '@mcp-trace/trace-core': resolve('packages/mcp-trace/trace-core'),
         '@mcp-trace/trace-web': resolve('packages/mcp-trace/trace-web'),

eslint.config.mjs

@@ -48,6 +48,27 @@ export default defineConfig([
       '@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/**',
+      'src/main/integration/cherryai/index.js',
+      'src/main/integration/nutstore/sso/lib/**',
+      'src/renderer/src/ui/**',
+      'packages/**/dist'
+    ]
+  },
+  // turn off oxlint supported rules.
+  ...oxlint.configs['flat/eslint'],
+  ...oxlint.configs['flat/typescript'],
+  ...oxlint.configs['flat/unicorn'],
   {
     // LoggerService Custom Rules - only apply to src directory
     files: ['src/**/*.{ts,tsx,js,jsx}'],
@@ -110,25 +131,4 @@ export default defineConfig([
       'i18n/no-template-in-t': 'warn'
     }
   },
-  {
-    ignores: [
-      'node_modules/**',
-      'build/**',
-      'dist/**',
-      'out/**',
-      'local/**',
-      '.yarn/**',
-      '.gitignore',
-      'scripts/cloudflare-worker.js',
-      'src/main/integration/nutstore/sso/lib/**',
-      'src/main/integration/cherryin/index.js',
-      'src/main/integration/nutstore/sso/lib/**',
-      'src/renderer/src/ui/**',
-      'packages/**/dist'
-    ]
-  },
-  // turn off oxlint supported rules.
-  ...oxlint.configs['flat/eslint'],
-  ...oxlint.configs['flat/typescript'],
-  ...oxlint.configs['flat/unicorn']
 ])

package.json

@@ -1,6 +1,6 @@
 {
   "name": "CherryStudio",
-  "version": "1.6.0-rc.2",
+  "version": "1.6.1",
   "private": true,
   "description": "A powerful AI assistant for producer.",
   "main": "./out/main/index.js",
@@ -27,7 +27,6 @@
   "scripts": {
     "start": "electron-vite preview",
     "dev": "dotenv electron-vite dev",
-    "dev:main": "dotenv electron-vite dev --watch",
     "debug": "electron-vite -- --inspect --sourcemap --remote-debugging-port=9222",
     "build": "npm run typecheck && electron-vite build",
     "build:check": "yarn lint && yarn test",
@@ -44,18 +43,15 @@
     "release": "node scripts/version.js",
     "publish": "yarn build:check && yarn release patch push",
     "pulish:artifacts": "cd packages/artifacts && npm publish && cd -",
-    "agents:generate": "NODE_ENV='development' drizzle-kit generate --config src/main/services/agents/drizzle.config.ts",
-    "agents:push": "NODE_ENV='development' drizzle-kit push --config src/main/services/agents/drizzle.config.ts",
-    "agents:studio": "NODE_ENV='development' drizzle-kit studio --config src/main/services/agents/drizzle.config.ts",
-    "agents:drop": "NODE_ENV='development' drizzle-kit drop --config src/main/services/agents/drizzle.config.ts",
+    "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",
     "typecheck": "concurrently -n \"node,web\" -c \"cyan,magenta\" \"npm run typecheck:node\" \"npm run typecheck:web\"",
     "typecheck:node": "tsgo --noEmit -p tsconfig.node.json --composite false",
     "typecheck:web": "tsgo --noEmit -p tsconfig.web.json --composite false",
-    "check:i18n": "dotenv -e .env -- tsx scripts/check-i18n.ts",
-    "sync:i18n": "dotenv -e .env -- tsx scripts/sync-i18n.ts",
+    "check:i18n": "tsx scripts/check-i18n.ts",
+    "sync:i18n": "tsx scripts/sync-i18n.ts",
     "update:i18n": "dotenv -e .env -- tsx scripts/update-i18n.ts",
     "auto:i18n": "dotenv -e .env -- tsx scripts/auto-translate-i18n.ts",
     "update:languages": "tsx scripts/update-languages.ts",
@@ -79,15 +75,11 @@
     "release:aicore": "yarn workspace @cherrystudio/ai-core version patch --immediate && yarn workspace @cherrystudio/ai-core npm publish --access public"
   },
   "dependencies": {
-    "@anthropic-ai/claude-code": "patch:@anthropic-ai/claude-code@npm%3A1.0.118#~/.yarn/patches/@anthropic-ai-claude-code-npm-1.0.118-bbf4e9e59f.patch",
     "@libsql/client": "0.14.0",
     "@libsql/win32-x64-msvc": "^0.4.7",
     "@napi-rs/system-ocr": "patch:@napi-rs/system-ocr@npm%3A1.0.2#~/.yarn/patches/@napi-rs-system-ocr-npm-1.0.2-59e7a78e8b.patch",
     "@strongtz/win32-arm64-msvc": "^0.4.7",
-    "@types/uuid": "^10.0.0",
-    "drizzle-orm": "^0.44.5",
     "express": "^5.1.0",
-    "express-validator": "^7.2.1",
     "font-list": "^2.0.0",
     "graceful-fs": "^4.2.11",
     "jsdom": "26.1.0",
@@ -159,7 +151,6 @@
     "@opentelemetry/sdk-trace-node": "^2.0.0",
     "@opentelemetry/sdk-trace-web": "^2.0.0",
     "@playwright/test": "^1.52.0",
-    "@radix-ui/react-context-menu": "^2.2.16",
     "@reduxjs/toolkit": "^2.2.5",
     "@shikijs/markdown-it": "^3.12.0",
     "@swc/plugin-styled-components": "^8.0.4",
@@ -247,11 +238,9 @@
     "docx": "^9.0.2",
     "dompurify": "^3.2.6",
     "dotenv-cli": "^7.4.2",
-    "drizzle-kit": "^0.31.4",
     "electron": "37.4.0",
     "electron-builder": "26.0.15",
     "electron-devtools-installer": "^3.2.0",
-    "electron-reload": "^2.0.0-alpha.1",
     "electron-store": "^8.2.0",
     "electron-updater": "6.6.4",
     "electron-vite": "4.0.0",
@@ -336,7 +325,6 @@
     "string-width": "^7.2.0",
     "striptags": "^3.2.0",
     "styled-components": "^6.1.11",
-    "swr": "^2.3.6",
     "tailwindcss": "^4.1.13",
     "tar": "^7.4.3",
     "tiny-pinyin": "^1.3.2",
@@ -347,7 +335,7 @@
     "typescript": "~5.8.2",
     "undici": "6.21.2",
     "unified": "^11.0.5",
-    "uuid": "^13.0.0",
+    "uuid": "^10.0.0",
     "vite": "npm:rolldown-vite@latest",
     "vitest": "^3.2.4",
     "webdav": "^5.8.0",
@@ -380,7 +368,8 @@
     "pkce-challenge@npm:^4.1.0": "patch:pkce-challenge@npm%3A4.1.0#~/.yarn/patches/pkce-challenge-npm-4.1.0-fbc51695a3.patch",
     "undici": "6.21.2",
     "vite": "npm:rolldown-vite@latest",
-    "tesseract.js@npm:*": "patch:tesseract.js@npm%3A6.0.1#~/.yarn/patches/tesseract.js-npm-6.0.1-2562a7e46d.patch"
+    "tesseract.js@npm:*": "patch:tesseract.js@npm%3A6.0.1#~/.yarn/patches/tesseract.js-npm-6.0.1-2562a7e46d.patch",
+    "@ai-sdk/google@npm:2.0.14": "patch:@ai-sdk/google@npm%3A2.0.14#~/.yarn/patches/@ai-sdk-google-npm-2.0.14-376d8b03cc.patch"
   },
   "packageManager": "yarn@4.9.1",
   "lint-staged": {

packages/aiCore/package.json

@@ -39,7 +39,7 @@
     "@ai-sdk/anthropic": "^2.0.17",
     "@ai-sdk/azure": "^2.0.30",
     "@ai-sdk/deepseek": "^1.0.17",
-    "@ai-sdk/google": "^2.0.14",
+    "@ai-sdk/google": "patch:@ai-sdk/google@npm%3A2.0.14#~/.yarn/patches/@ai-sdk-google-npm-2.0.14-376d8b03cc.patch",
     "@ai-sdk/openai": "^2.0.30",
     "@ai-sdk/openai-compatible": "^1.0.17",
     "@ai-sdk/provider": "^2.0.0",

packages/shared/IpcChannel.ts

@@ -8,6 +8,7 @@ export enum IpcChannel {
   App_ShowUpdateDialog = 'app:show-update-dialog',
   App_CheckForUpdate = 'app:check-for-update',
   App_Reload = 'app:reload',
+  App_Quit = 'app:quit',
   App_Info = 'app:info',
   App_Proxy = 'app:proxy',
   App_SetLaunchToTray = 'app:set-launch-to-tray',
@@ -89,9 +90,6 @@ export enum IpcChannel {
   // Python
   Python_Execute = 'python:execute',
 
-  // agent messages
-  AgentMessage_PersistExchange = 'agent-message:persist-exchange',
-
   //copilot
   Copilot_GetAuthMessage = 'copilot:get-auth-message',
   Copilot_GetCopilotToken = 'copilot:get-copilot-token',
@@ -324,10 +322,14 @@ export enum IpcChannel {
 
   // CodeTools
   CodeTools_Run = 'code-tools:run',
+  CodeTools_GetAvailableTerminals = 'code-tools:get-available-terminals',
+  CodeTools_SetCustomTerminalPath = 'code-tools:set-custom-terminal-path',
+  CodeTools_GetCustomTerminalPath = 'code-tools:get-custom-terminal-path',
+  CodeTools_RemoveCustomTerminalPath = 'code-tools:remove-custom-terminal-path',
 
   // OCR
   OCR_ocr = 'ocr:ocr',
 
-  // Cherryin
-  Cherryin_GetSignature = 'cherryin:get-signature'
+  // CherryAI
+  Cherryai_GetSignature = 'cherryai:get-signature'
 }

packages/shared/config/constant.ts

@@ -219,3 +219,253 @@ export enum codeTools {
   openaiCodex = 'openai-codex',
   iFlowCli = 'iflow-cli'
 }
+
+export enum terminalApps {
+  systemDefault = 'Terminal',
+  iterm2 = 'iTerm2',
+  kitty = 'kitty',
+  alacritty = 'Alacritty',
+  wezterm = 'WezTerm',
+  ghostty = 'Ghostty',
+  tabby = 'Tabby',
+  // Windows terminals
+  windowsTerminal = 'WindowsTerminal',
+  powershell = 'PowerShell',
+  cmd = 'CMD',
+  wsl = 'WSL'
+}
+
+export interface TerminalConfig {
+  id: string
+  name: string
+  bundleId?: string
+  customPath?: string // For user-configured terminal paths on Windows
+}
+
+export interface TerminalConfigWithCommand extends TerminalConfig {
+  command: (directory: string, fullCommand: string) => { command: string; args: string[] }
+}
+
+export const MACOS_TERMINALS: TerminalConfig[] = [
+  {
+    id: terminalApps.systemDefault,
+    name: 'Terminal',
+    bundleId: 'com.apple.Terminal'
+  },
+  {
+    id: terminalApps.iterm2,
+    name: 'iTerm2',
+    bundleId: 'com.googlecode.iterm2'
+  },
+  {
+    id: terminalApps.kitty,
+    name: 'kitty',
+    bundleId: 'net.kovidgoyal.kitty'
+  },
+  {
+    id: terminalApps.alacritty,
+    name: 'Alacritty',
+    bundleId: 'org.alacritty'
+  },
+  {
+    id: terminalApps.wezterm,
+    name: 'WezTerm',
+    bundleId: 'com.github.wez.wezterm'
+  },
+  {
+    id: terminalApps.ghostty,
+    name: 'Ghostty',
+    bundleId: 'com.mitchellh.ghostty'
+  },
+  {
+    id: terminalApps.tabby,
+    name: 'Tabby',
+    bundleId: 'org.tabby'
+  }
+]
+
+export const WINDOWS_TERMINALS: TerminalConfig[] = [
+  {
+    id: terminalApps.cmd,
+    name: 'Command Prompt'
+  },
+  {
+    id: terminalApps.powershell,
+    name: 'PowerShell'
+  },
+  {
+    id: terminalApps.windowsTerminal,
+    name: 'Windows Terminal'
+  },
+  {
+    id: terminalApps.wsl,
+    name: 'WSL (Ubuntu/Debian)'
+  },
+  {
+    id: terminalApps.alacritty,
+    name: 'Alacritty'
+  },
+  {
+    id: terminalApps.wezterm,
+    name: 'WezTerm'
+  }
+]
+
+export const WINDOWS_TERMINALS_WITH_COMMANDS: TerminalConfigWithCommand[] = [
+  {
+    id: terminalApps.cmd,
+    name: 'Command Prompt',
+    command: (_: string, fullCommand: string) => ({
+      command: 'cmd',
+      args: ['/c', 'start', 'cmd', '/k', fullCommand]
+    })
+  },
+  {
+    id: terminalApps.powershell,
+    name: 'PowerShell',
+    command: (_: string, fullCommand: string) => ({
+      command: 'cmd',
+      args: ['/c', 'start', 'powershell', '-NoExit', '-Command', `& '${fullCommand}'`]
+    })
+  },
+  {
+    id: terminalApps.windowsTerminal,
+    name: 'Windows Terminal',
+    command: (_: string, fullCommand: string) => ({
+      command: 'wt',
+      args: ['cmd', '/k', fullCommand]
+    })
+  },
+  {
+    id: terminalApps.wsl,
+    name: 'WSL (Ubuntu/Debian)',
+    command: (_: string, fullCommand: string) => {
+      // Start WSL in a new window and execute the batch file from within WSL using cmd.exe
+      // The batch file will run in Windows context but output will be in WSL terminal
+      return {
+        command: 'cmd',
+        args: ['/c', 'start', 'wsl', '-e', 'bash', '-c', `cmd.exe /c '${fullCommand}' ; exec bash`]
+      }
+    }
+  },
+  {
+    id: terminalApps.alacritty,
+    name: 'Alacritty',
+    customPath: '', // Will be set by user in settings
+    command: (_: string, fullCommand: string) => ({
+      command: 'alacritty', // Will be replaced with customPath if set
+      args: ['-e', 'cmd', '/k', fullCommand]
+    })
+  },
+  {
+    id: terminalApps.wezterm,
+    name: 'WezTerm',
+    customPath: '', // Will be set by user in settings
+    command: (_: string, fullCommand: string) => ({
+      command: 'wezterm', // Will be replaced with customPath if set
+      args: ['start', 'cmd', '/k', fullCommand]
+    })
+  }
+]
+
+// Helper function to escape strings for AppleScript
+const escapeForAppleScript = (str: string): string => {
+  // In AppleScript strings, backslashes and double quotes need to be escaped
+  // When passed through osascript -e with single quotes, we need:
+  // 1. Backslash: \ -> \\
+  // 2. Double quote: " -> \"
+  return str
+    .replace(/\\/g, '\\\\') // Escape backslashes first
+    .replace(/"/g, '\\"') // Then escape double quotes
+}
+
+export const MACOS_TERMINALS_WITH_COMMANDS: TerminalConfigWithCommand[] = [
+  {
+    id: terminalApps.systemDefault,
+    name: 'Terminal',
+    bundleId: 'com.apple.Terminal',
+    command: (_directory: string, fullCommand: string) => ({
+      command: 'sh',
+      args: [
+        '-c',
+        `open -na Terminal && sleep 0.5 && osascript -e 'tell application "Terminal" to activate' -e 'tell application "Terminal" to do script "${escapeForAppleScript(fullCommand)}" in front window'`
+      ]
+    })
+  },
+  {
+    id: terminalApps.iterm2,
+    name: 'iTerm2',
+    bundleId: 'com.googlecode.iterm2',
+    command: (_directory: string, fullCommand: string) => ({
+      command: 'sh',
+      args: [
+        '-c',
+        `open -na iTerm && sleep 0.8 && osascript -e 'on waitUntilRunning()\n  repeat 50 times\n    tell application "System Events"\n      if (exists process "iTerm2") then exit repeat\n    end tell\n    delay 0.1\n  end repeat\nend waitUntilRunning\n\nwaitUntilRunning()\n\ntell application "iTerm2"\n  if (count of windows) = 0 then\n    create window with default profile\n    delay 0.3\n  else\n    tell current window\n      create tab with default profile\n    end tell\n    delay 0.3\n  end if\n  tell current session of current window to write text "${escapeForAppleScript(fullCommand)}"\n  activate\nend tell'`
+      ]
+    })
+  },
+  {
+    id: terminalApps.kitty,
+    name: 'kitty',
+    bundleId: 'net.kovidgoyal.kitty',
+    command: (_directory: string, fullCommand: string) => ({
+      command: 'sh',
+      args: [
+        '-c',
+        `cd "${_directory}" && open -na kitty --args --directory="${_directory}" sh -c "${fullCommand.replace(/\\/g, '\\\\').replace(/"/g, '\\"')}; exec \\$SHELL" && sleep 0.5 && osascript -e 'tell application "kitty" to activate'`
+      ]
+    })
+  },
+  {
+    id: terminalApps.alacritty,
+    name: 'Alacritty',
+    bundleId: 'org.alacritty',
+    command: (_directory: string, fullCommand: string) => ({
+      command: 'sh',
+      args: [
+        '-c',
+        `open -na Alacritty --args --working-directory "${_directory}" -e sh -c "${fullCommand.replace(/\\/g, '\\\\').replace(/"/g, '\\"')}; exec \\$SHELL" && sleep 0.5 && osascript -e 'tell application "Alacritty" to activate'`
+      ]
+    })
+  },
+  {
+    id: terminalApps.wezterm,
+    name: 'WezTerm',
+    bundleId: 'com.github.wez.wezterm',
+    command: (_directory: string, fullCommand: string) => ({
+      command: 'sh',
+      args: [
+        '-c',
+        `open -na WezTerm --args start --new-tab --cwd "${_directory}" -- sh -c "${fullCommand.replace(/\\/g, '\\\\').replace(/"/g, '\\"')}; exec \\$SHELL" && sleep 0.5 && osascript -e 'tell application "WezTerm" to activate'`
+      ]
+    })
+  },
+  {
+    id: terminalApps.ghostty,
+    name: 'Ghostty',
+    bundleId: 'com.mitchellh.ghostty',
+    command: (_directory: string, fullCommand: string) => ({
+      command: 'sh',
+      args: [
+        '-c',
+        `cd "${_directory}" && open -na Ghostty --args --working-directory="${_directory}" -e sh -c "${fullCommand.replace(/\\/g, '\\\\').replace(/"/g, '\\"')}; exec \\$SHELL" && sleep 0.5 && osascript -e 'tell application "Ghostty" to activate'`
+      ]
+    })
+  },
+  {
+    id: terminalApps.tabby,
+    name: 'Tabby',
+    bundleId: 'org.tabby',
+    command: (_directory: string, fullCommand: string) => ({
+      command: 'sh',
+      args: [
+        '-c',
+        `if pgrep -x "Tabby" > /dev/null; then
+          open -na Tabby --args open && sleep 0.3
+        else
+          open -na Tabby --args open && sleep 2
+        fi && osascript -e 'tell application "Tabby" to activate' -e 'set the clipboard to "${escapeForAppleScript(fullCommand)}"' -e 'tell application "System Events" to tell process "Tabby" to keystroke "v" using {command down}' -e 'tell application "System Events" to key code 36'`
+      ]
+    })
+  }
+]

resources/cherry-studio/privacy-en.html

@@ -0,0 +1,252 @@
+<!DOCTYPE html>
+<html lang="en">
+
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Privacy Policy</title>
+    <style>
+        * {
+            margin: 0;
+            padding: 0;
+            box-sizing: border-box;
+        }
+
+        body {
+            font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', 'Roboto', 'Helvetica Neue', Arial, sans-serif;
+            line-height: 1.6;
+            color: #333;
+            background: transparent;
+            margin: 0 auto;
+        }
+
+        body.dark {
+            background: transparent;
+            color: rgba(255, 255, 255, 0.85);
+        }
+
+        h1 {
+            font-size: 24px;
+            font-weight: 600;
+            margin-bottom: 20px;
+            color: #1a1a1a;
+        }
+
+        body.dark h1 {
+            color: rgba(255, 255, 255, 0.95);
+        }
+
+        h2 {
+            font-size: 18px;
+            font-weight: 600;
+            margin-top: 24px;
+            margin-bottom: 12px;
+            color: #2c2c2c;
+        }
+
+        body.dark h2 {
+            color: rgba(255, 255, 255, 0.9);
+        }
+
+        p {
+            margin: 12px 0;
+            line-height: 1.8;
+        }
+
+        body.dark p {
+            color: rgba(255, 255, 255, 0.8);
+        }
+
+        ul {
+            margin: 12px 0;
+            padding-left: 24px;
+        }
+
+        li {
+            margin: 6px 0;
+            line-height: 1.6;
+        }
+
+        body.dark li {
+            color: rgba(255, 255, 255, 0.75);
+        }
+
+        a {
+            color: #0066cc;
+            text-decoration: none;
+        }
+
+        a:hover {
+            text-decoration: underline;
+        }
+
+        body.dark a {
+            color: #4da6ff;
+        }
+
+        .footer {
+            margin-top: 40px;
+            padding-top: 20px;
+            border-top: 1px solid #e0e0e0;
+            font-size: 13px;
+            color: #666;
+        }
+
+        body.dark .footer {
+            border-top-color: rgba(255, 255, 255, 0.1);
+            color: rgba(255, 255, 255, 0.5);
+        }
+
+        .content-wrapper {
+            max-height: calc(100vh - 40px);
+            overflow-y: auto;
+            padding-right: 10px;
+            background: transparent;
+        }
+
+        /* Scrollbar styles - Light mode */
+        ::-webkit-scrollbar {
+            width: 8px;
+            height: 8px;
+        }
+
+        ::-webkit-scrollbar-track {
+            background: rgba(0, 0, 0, 0.05);
+            border-radius: 4px;
+        }
+
+        ::-webkit-scrollbar-thumb {
+            background: rgba(0, 0, 0, 0.2);
+            border-radius: 4px;
+        }
+
+        ::-webkit-scrollbar-thumb:hover {
+            background: rgba(0, 0, 0, 0.3);
+        }
+
+        /* Scrollbar styles - Dark mode */
+        body.dark ::-webkit-scrollbar-track {
+            background: rgba(255, 255, 255, 0.05);
+        }
+
+        body.dark ::-webkit-scrollbar-thumb {
+            background: rgba(255, 255, 255, 0.2);
+        }
+
+        body.dark ::-webkit-scrollbar-thumb:hover {
+            background: rgba(255, 255, 255, 0.3);
+        }
+    </style>
+    <script>
+        // Detect theme
+        document.addEventListener('DOMContentLoaded', function () {
+            const urlParams = new URLSearchParams(window.location.search);
+            const theme = urlParams.get('theme');
+            if (theme === 'dark') {
+                document.documentElement.classList.add('dark');
+                document.body.classList.add('dark');
+            }
+        });
+    </script>
+</head>
+
+<body>
+    <div class="content-wrapper">
+        <h1>Privacy Policy</h1>
+
+        <p>
+            Welcome to Cherry Studio (hereinafter referred to as "the Software" or "we"). We highly value your privacy
+            protection. This Privacy Policy explains how we process and protect your personal information and data.
+            Please read and understand this policy carefully before using the Software:
+        </p>
+
+        <h2>1. Information We Collect</h2>
+        <p>To optimize user experience and improve software quality, we may only collect the following anonymous,
+            non-personal information:</p>
+        <ul>
+            <li>Software version information</li>
+            <li>Activity and usage frequency of software features</li>
+            <li>Anonymous crash and error log information</li>
+        </ul>
+        <p>The above information is completely anonymous, does not involve any personal identity data, and cannot be
+            linked to your personal information.</p>
+
+        <h2>2. Information We Do Not Collect</h2>
+        <p>To maximize the protection of your privacy and security, we explicitly commit that we:</p>
+        <ul>
+            <li>Will not collect, save, transmit, or process model service API Key information you enter into the
+                Software</li>
+            <li>Will not collect, save, transmit, or process any conversation data generated during your use of the
+                Software, including but not limited to chat content, instruction information, knowledge base
+                information, vector data, and other custom content</li>
+            <li>Will not collect, save, transmit, or process any sensitive information that can identify personal
+                identity</li>
+        </ul>
+
+        <h2>3. Data Interaction Description</h2>
+        <p>
+            The Software uses API Keys from third-party model service providers that you apply for and configure
+            yourself to complete model calls and conversation functions. The model services you use (such as large
+            models, API interfaces, etc.) are directly provided by third-party providers of your choice. We do not
+            intervene, monitor, or interfere with the data transmission process.
+        </p>
+        <p>
+            Data interactions between you and third-party model services are governed by the privacy policies and user
+            agreements of third-party service providers. We recommend that you fully understand the privacy terms of
+            relevant service providers before use.
+        </p>
+
+        <h2>4. Local Data Security Protection</h2>
+        <p>The Software is a localized application, and all data is stored on your local device by default. We have
+            taken the following measures to ensure data security:</p>
+        <ul>
+            <li>Conversation records, configuration information, and other data are only saved on your local device</li>
+            <li>Data import/export functions are provided to facilitate your independent management and backup of data
+            </li>
+            <li>Your local data will not be uploaded to any server or cloud storage</li>
+        </ul>
+
+        <h2>5. Third-Party Services</h2>
+        <p>
+            When using the Software, you may access third-party services (such as AI model APIs, translation services,
+            etc.). The use of these third-party services is governed by their respective terms of service and privacy
+            policies. We strongly recommend that you carefully read and understand the relevant terms before use.
+        </p>
+
+        <h2>6. User Rights</h2>
+        <p>You have complete control over your data:</p>
+        <ul>
+            <li>You can view, modify, and delete all locally stored data at any time</li>
+            <li>You can choose whether to enable specific features or services</li>
+            <li>You can stop using the Software and delete all related data at any time</li>
+        </ul>
+
+        <h2>7. Children's Privacy Protection</h2>
+        <p>The Software is not intended for minors under 18 years of age. If you are a minor, please use the Software
+            under the guidance of a guardian.</p>
+
+        <h2>8. Privacy Policy Updates</h2>
+        <p>
+            We may update this Privacy Policy based on legal requirements or changes in product features. The updated
+            policy will be published in the Software and you will be notified before it takes effect. If you do not
+            agree with the updated terms, you can choose to stop using the Software.
+        </p>
+
+        <h2>9. Contact Us</h2>
+        <p>If you have any questions, suggestions, or complaints about this Privacy Policy, please contact us through
+            the following methods:</p>
+        <ul>
+            <li>
+                GitHub: <a href="https://github.com/CherryHQ/cherry-studio" target="_blank"
+                    rel="noopener noreferrer">https://github.com/CherryHQ/cherry-studio</a>
+            </li>
+            <li>Email: support@cherry-ai.com</li>
+        </ul>
+
+        <div class="footer">
+            Last Updated: December 2024
+        </div>
+    </div>
+</body>
+
+</html>

resources/cherry-studio/privacy-zh.html

@@ -0,0 +1,230 @@
+<!DOCTYPE html>
+<html lang="zh">
+
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>隐私协议</title>
+    <style>
+        * {
+            margin: 0;
+            padding: 0;
+            box-sizing: border-box;
+        }
+
+        body {
+            font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', 'Roboto', 'Helvetica Neue', Arial, sans-serif;
+            line-height: 1.6;
+            color: #333;
+            background: transparent;
+            margin: 0 auto;
+        }
+
+        body.dark {
+            background: transparent;
+            color: rgba(255, 255, 255, 0.85);
+        }
+
+        h1 {
+            font-size: 24px;
+            font-weight: 600;
+            margin-bottom: 20px;
+            color: #1a1a1a;
+        }
+
+        body.dark h1 {
+            color: rgba(255, 255, 255, 0.95);
+        }
+
+        h2 {
+            font-size: 18px;
+            font-weight: 600;
+            margin-top: 24px;
+            margin-bottom: 12px;
+            color: #2c2c2c;
+        }
+
+        body.dark h2 {
+            color: rgba(255, 255, 255, 0.9);
+        }
+
+        p {
+            margin: 12px 0;
+            line-height: 1.8;
+        }
+
+        body.dark p {
+            color: rgba(255, 255, 255, 0.8);
+        }
+
+        ul {
+            margin: 12px 0;
+            padding-left: 24px;
+        }
+
+        li {
+            margin: 6px 0;
+            line-height: 1.6;
+        }
+
+        body.dark li {
+            color: rgba(255, 255, 255, 0.75);
+        }
+
+        a {
+            color: #0066cc;
+            text-decoration: none;
+        }
+
+        a:hover {
+            text-decoration: underline;
+        }
+
+        body.dark a {
+            color: #4da6ff;
+        }
+
+        .footer {
+            margin-top: 40px;
+            padding-top: 20px;
+            border-top: 1px solid #e0e0e0;
+            font-size: 13px;
+            color: #666;
+        }
+
+        body.dark .footer {
+            border-top-color: rgba(255, 255, 255, 0.1);
+            color: rgba(255, 255, 255, 0.5);
+        }
+
+        .content-wrapper {
+            overflow-y: auto;
+            padding-right: 10px;
+            background: transparent;
+        }
+
+        /* 滚动条样式 - 亮色模式 */
+        ::-webkit-scrollbar {
+            width: 8px;
+            height: 8px;
+        }
+
+        ::-webkit-scrollbar-track {
+            background: rgba(0, 0, 0, 0.05);
+            border-radius: 4px;
+        }
+
+        ::-webkit-scrollbar-thumb {
+            background: rgba(0, 0, 0, 0.2);
+            border-radius: 4px;
+        }
+
+        ::-webkit-scrollbar-thumb:hover {
+            background: rgba(0, 0, 0, 0.3);
+        }
+
+        /* 滚动条样式 - 暗色模式 */
+        body.dark ::-webkit-scrollbar-track {
+            background: rgba(255, 255, 255, 0.05);
+        }
+
+        body.dark ::-webkit-scrollbar-thumb {
+            background: rgba(255, 255, 255, 0.2);
+        }
+
+        body.dark ::-webkit-scrollbar-thumb:hover {
+            background: rgba(255, 255, 255, 0.3);
+        }
+    </style>
+    <script>
+        // 检测主题
+        document.addEventListener('DOMContentLoaded', function () {
+            const urlParams = new URLSearchParams(window.location.search);
+            const theme = urlParams.get('theme');
+            if (theme === 'dark') {
+                document.documentElement.classList.add('dark');
+                document.body.classList.add('dark');
+            }
+        });
+    </script>
+</head>
+
+<body>
+    <div class="content-wrapper">
+        <h1>隐私协议</h1>
+
+        <p>
+            欢迎使用 Cherry Studio（以下简称"本软件"或"我们"）。我们高度重视您的隐私保护，本隐私协议将说明我们如何处理与保护您的个人信息和数据。请在使用本软件前仔细阅读并理解本协议：
+        </p>
+
+        <h2>一、我们收集的信息范围</h2>
+        <p>为了优化用户体验和提升软件质量，我们仅可能会匿名收集以下非个人化信息：</p>
+        <ul>
+            <li>软件版本信息；</li>
+            <li>软件功能的活跃度、使用频次；</li>
+            <li>匿名的崩溃、错误日志信息；</li>
+        </ul>
+        <p>上述信息完全匿名，不会涉及任何个人身份数据，也无法关联到您的个人信息。</p>
+
+        <h2>二、我们不会收集的任何信息</h2>
+        <p>为了最大限度保护您的隐私安全，我们明确承诺：</p>
+        <ul>
+            <li>不会收集、保存、传输或处理您输入到本软件中的模型服务 API Key 信息；</li>
+            <li>不会收集、保存、传输或处理您在使用本软件过程中产生的任何对话数据，包括但不限于聊天内容、指令信息、知识库信息、向量数据及其他自定义内容；</li>
+            <li>不会收集、保存、传输或处理任何可识别个人身份的敏感信息。</li>
+        </ul>
+
+        <h2>三、数据交互说明</h2>
+        <p>
+            本软件采用您自行申请并配置的第三方模型服务提供商的 API Key，以完成相关模型的调用与对话功能。您使用的模型服务（例如大模型、API 接口等）由您选择的第三方提供商直接提供，我们不会介入、监控或干扰数据传输过程。
+        </p>
+        <p>
+            您与第三方模型服务之间的数据交互受第三方服务提供商的隐私政策和用户协议约束，我们建议您在使用前充分了解相关服务商的隐私条款。
+        </p>
+
+        <h2>四、本地数据的安全保护</h2>
+        <p>本软件为本地化应用程序，所有数据默认存储在您的本地设备上。我们采取了以下措施保障数据安全：</p>
+        <ul>
+            <li>对话记录、配置信息等数据仅保存在您的本地设备中；</li>
+            <li>提供数据导入/导出功能，方便您自主管理和备份数据；</li>
+            <li>不会将您的本地数据上传至任何服务器或云端存储。</li>
+        </ul>
+
+        <h2>五、第三方服务</h2>
+        <p>
+            在使用本软件过程中，您可能会接入第三方服务（如 AI 模型 API、翻译服务等）。这些第三方服务的使用受其各自的服务条款和隐私政策约束。我们强烈建议您在使用前仔细阅读并理解相关条款。
+        </p>
+
+        <h2>六、用户权利</h2>
+        <p>您对自己的数据拥有完全的控制权：</p>
+        <ul>
+            <li>您可以随时查看、修改、删除本地存储的所有数据；</li>
+            <li>您可以选择是否启用特定功能或服务；</li>
+            <li>您可以随时停止使用本软件并删除所有相关数据。</li>
+        </ul>
+
+        <h2>七、儿童隐私保护</h2>
+        <p>本软件不面向 18 岁以下的未成年人提供服务。如果您是未成年人，请在监护人的指导下使用本软件。</p>
+
+        <h2>八、隐私政策的更新</h2>
+        <p>
+            我们可能会根据法律法规要求或产品功能的变化更新本隐私协议。更新后的协议将在软件中发布，并在生效前通知您。如果您不同意更新后的条款，您可以选择停止使用本软件。
+        </p>
+
+        <h2>九、联系我们</h2>
+        <p>如果您对本隐私协议有任何疑问、建议或投诉，请通过以下方式联系我们：</p>
+        <ul>
+            <li>
+                GitHub: <a href="https://github.com/CherryHQ/cherry-studio" target="_blank"
+                    rel="noopener noreferrer">https://github.com/CherryHQ/cherry-studio</a>
+            </li>
+            <li>Email: support@cherry-ai.com</li>
+        </ul>
+
+        <div class="footer">
+            最后更新日期：2024年12月
+        </div>
+    </div>
+</body>
+
+</html>

resources/database/drizzle/0000_confused_wendigo.sql

@@ -1,53 +0,0 @@
---> statement-breakpoint
-CREATE TABLE `migrations` (
-	`version` integer PRIMARY KEY NOT NULL,
-	`tag` text NOT NULL,
-	`executed_at` integer NOT NULL
-);
-
-CREATE TABLE `agents` (
-	`id` text PRIMARY KEY NOT NULL,
-	`type` text NOT NULL,
-	`name` text NOT NULL,
-	`description` text,
-	`accessible_paths` text,
-	`instructions` text,
-	`model` text NOT NULL,
-	`plan_model` text,
-	`small_model` text,
-	`mcps` text,
-	`allowed_tools` text,
-	`configuration` text,
-	`created_at` text NOT NULL,
-	`updated_at` text NOT NULL
-);
-
---> statement-breakpoint
-CREATE TABLE `sessions` (
-	`id` text PRIMARY KEY NOT NULL,
-	`agent_type` text NOT NULL,
-	`agent_id` text NOT NULL,
-	`name` text NOT NULL,
-	`description` text,
-	`accessible_paths` text,
-	`instructions` text,
-	`model` text NOT NULL,
-	`plan_model` text,
-	`small_model` text,
-	`mcps` text,
-	`allowed_tools` text,
-	`configuration` text,
-	`created_at` text NOT NULL,
-	`updated_at` text NOT NULL
-);
-
---> statement-breakpoint
-CREATE TABLE `session_messages` (
-	`id` integer PRIMARY KEY AUTOINCREMENT NOT NULL,
-	`session_id` text NOT NULL,
-	`role` text NOT NULL,
-	`content` text NOT NULL,
-	`metadata` text,
-	`created_at` text NOT NULL,
-	`updated_at` text NOT NULL
-);

resources/database/drizzle/0001_woozy_captain_flint.sql

@@ -1 +0,0 @@
-ALTER TABLE `session_messages` ADD `agent_session_id` text DEFAULT '';

resources/database/drizzle/meta/0000_snapshot.json

@@ -1,331 +0,0 @@
-{
-  "version": "6",
-  "dialect": "sqlite",
-  "id": "35efb412-0230-4767-9c76-7b7c4d40369f",
-  "prevId": "00000000-0000-0000-0000-000000000000",
-  "tables": {
-    "agents": {
-      "name": "agents",
-      "columns": {
-        "id": {
-          "name": "id",
-          "type": "text",
-          "primaryKey": true,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "type": {
-          "name": "type",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "name": {
-          "name": "name",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "description": {
-          "name": "description",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "accessible_paths": {
-          "name": "accessible_paths",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "instructions": {
-          "name": "instructions",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "model": {
-          "name": "model",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "plan_model": {
-          "name": "plan_model",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "small_model": {
-          "name": "small_model",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "mcps": {
-          "name": "mcps",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "allowed_tools": {
-          "name": "allowed_tools",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "configuration": {
-          "name": "configuration",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "created_at": {
-          "name": "created_at",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "updated_at": {
-          "name": "updated_at",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        }
-      },
-      "indexes": {},
-      "foreignKeys": {},
-      "compositePrimaryKeys": {},
-      "uniqueConstraints": {},
-      "checkConstraints": {}
-    },
-    "session_messages": {
-      "name": "session_messages",
-      "columns": {
-        "id": {
-          "name": "id",
-          "type": "integer",
-          "primaryKey": true,
-          "notNull": true,
-          "autoincrement": true
-        },
-        "session_id": {
-          "name": "session_id",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "role": {
-          "name": "role",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "content": {
-          "name": "content",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "metadata": {
-          "name": "metadata",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "created_at": {
-          "name": "created_at",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "updated_at": {
-          "name": "updated_at",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        }
-      },
-      "indexes": {},
-      "foreignKeys": {},
-      "compositePrimaryKeys": {},
-      "uniqueConstraints": {},
-      "checkConstraints": {}
-    },
-    "migrations": {
-      "name": "migrations",
-      "columns": {
-        "version": {
-          "name": "version",
-          "type": "integer",
-          "primaryKey": true,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "tag": {
-          "name": "tag",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "executed_at": {
-          "name": "executed_at",
-          "type": "integer",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        }
-      },
-      "indexes": {},
-      "foreignKeys": {},
-      "compositePrimaryKeys": {},
-      "uniqueConstraints": {},
-      "checkConstraints": {}
-    },
-    "sessions": {
-      "name": "sessions",
-      "columns": {
-        "id": {
-          "name": "id",
-          "type": "text",
-          "primaryKey": true,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "agent_type": {
-          "name": "agent_type",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "agent_id": {
-          "name": "agent_id",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "name": {
-          "name": "name",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "description": {
-          "name": "description",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "accessible_paths": {
-          "name": "accessible_paths",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "instructions": {
-          "name": "instructions",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "model": {
-          "name": "model",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "plan_model": {
-          "name": "plan_model",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "small_model": {
-          "name": "small_model",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "mcps": {
-          "name": "mcps",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "allowed_tools": {
-          "name": "allowed_tools",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "configuration": {
-          "name": "configuration",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "created_at": {
-          "name": "created_at",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "updated_at": {
-          "name": "updated_at",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        }
-      },
-      "indexes": {},
-      "foreignKeys": {},
-      "compositePrimaryKeys": {},
-      "uniqueConstraints": {},
-      "checkConstraints": {}
-    }
-  },
-  "views": {},
-  "enums": {},
-  "_meta": {
-    "schemas": {},
-    "tables": {},
-    "columns": {}
-  },
-  "internal": {
-    "indexes": {}
-  }
-}

resources/database/drizzle/meta/0001_snapshot.json

@@ -1,339 +0,0 @@
-{
-  "version": "6",
-  "dialect": "sqlite",
-  "id": "dabab6db-a2cd-4e96-b06e-6cb87d445a87",
-  "prevId": "35efb412-0230-4767-9c76-7b7c4d40369f",
-  "tables": {
-    "agents": {
-      "name": "agents",
-      "columns": {
-        "id": {
-          "name": "id",
-          "type": "text",
-          "primaryKey": true,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "type": {
-          "name": "type",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "name": {
-          "name": "name",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "description": {
-          "name": "description",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "accessible_paths": {
-          "name": "accessible_paths",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "instructions": {
-          "name": "instructions",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "model": {
-          "name": "model",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "plan_model": {
-          "name": "plan_model",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "small_model": {
-          "name": "small_model",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "mcps": {
-          "name": "mcps",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "allowed_tools": {
-          "name": "allowed_tools",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "configuration": {
-          "name": "configuration",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "created_at": {
-          "name": "created_at",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "updated_at": {
-          "name": "updated_at",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        }
-      },
-      "indexes": {},
-      "foreignKeys": {},
-      "compositePrimaryKeys": {},
-      "uniqueConstraints": {},
-      "checkConstraints": {}
-    },
-    "session_messages": {
-      "name": "session_messages",
-      "columns": {
-        "id": {
-          "name": "id",
-          "type": "integer",
-          "primaryKey": true,
-          "notNull": true,
-          "autoincrement": true
-        },
-        "session_id": {
-          "name": "session_id",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "role": {
-          "name": "role",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "content": {
-          "name": "content",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "agent_session_id": {
-          "name": "agent_session_id",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false,
-          "default": "''"
-        },
-        "metadata": {
-          "name": "metadata",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "created_at": {
-          "name": "created_at",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "updated_at": {
-          "name": "updated_at",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        }
-      },
-      "indexes": {},
-      "foreignKeys": {},
-      "compositePrimaryKeys": {},
-      "uniqueConstraints": {},
-      "checkConstraints": {}
-    },
-    "migrations": {
-      "name": "migrations",
-      "columns": {
-        "version": {
-          "name": "version",
-          "type": "integer",
-          "primaryKey": true,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "tag": {
-          "name": "tag",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "executed_at": {
-          "name": "executed_at",
-          "type": "integer",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        }
-      },
-      "indexes": {},
-      "foreignKeys": {},
-      "compositePrimaryKeys": {},
-      "uniqueConstraints": {},
-      "checkConstraints": {}
-    },
-    "sessions": {
-      "name": "sessions",
-      "columns": {
-        "id": {
-          "name": "id",
-          "type": "text",
-          "primaryKey": true,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "agent_type": {
-          "name": "agent_type",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "agent_id": {
-          "name": "agent_id",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "name": {
-          "name": "name",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "description": {
-          "name": "description",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "accessible_paths": {
-          "name": "accessible_paths",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "instructions": {
-          "name": "instructions",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "model": {
-          "name": "model",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "plan_model": {
-          "name": "plan_model",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "small_model": {
-          "name": "small_model",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "mcps": {
-          "name": "mcps",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "allowed_tools": {
-          "name": "allowed_tools",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "configuration": {
-          "name": "configuration",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": false,
-          "autoincrement": false
-        },
-        "created_at": {
-          "name": "created_at",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        },
-        "updated_at": {
-          "name": "updated_at",
-          "type": "text",
-          "primaryKey": false,
-          "notNull": true,
-          "autoincrement": false
-        }
-      },
-      "indexes": {},
-      "foreignKeys": {},
-      "compositePrimaryKeys": {},
-      "uniqueConstraints": {},
-      "checkConstraints": {}
-    }
-  },
-  "views": {},
-  "enums": {},
-  "_meta": {
-    "schemas": {},
-    "tables": {},
-    "columns": {}
-  },
-  "internal": {
-    "indexes": {}
-  }
-}

resources/database/drizzle/meta/_journal.json

@@ -1,20 +0,0 @@
-{
-  "version": "7",
-  "dialect": "sqlite",
-  "entries": [
-    {
-      "idx": 0,
-      "version": "6",
-      "when": 1758091173882,
-      "tag": "0000_confused_wendigo",
-      "breakpoints": true
-    },
-    {
-      "idx": 1,
-      "version": "6",
-      "when": 1758187378775,
-      "tag": "0001_woozy_captain_flint",
-      "breakpoints": true
-    }
-  ]
-}

scripts/auto-translate-i18n.ts

@@ -9,9 +9,8 @@ import * as path from 'path'
 
 const localesDir = path.join(__dirname, '../src/renderer/src/i18n/locales')
 const translateDir = path.join(__dirname, '../src/renderer/src/i18n/translate')
-const baseLocale = process.env.BASE_LOCALE ?? 'zh-cn'
+const baseLocale = 'zh-cn'
 const baseFileName = `${baseLocale}.json`
-const baseLocalePath = path.join(__dirname, '../src/renderer/src/i18n/locales', baseFileName)
 
 type I18NValue = string | { [key: string]: I18NValue }
 type I18N = { [key: string]: I18NValue }
@@ -106,9 +105,6 @@ const translateRecursively = async (originObj: I18N, systemPrompt: string): Prom
 }
 
 const main = async () => {
-  if (!fs.existsSync(baseLocalePath)) {
-    throw new Error(`${baseLocalePath} not found.`)
-  }
   const localeFiles = fs
     .readdirSync(localesDir)
     .filter((file) => file.endsWith('.json') && file !== baseFileName)

scripts/before-pack.js

@@ -35,9 +35,6 @@ const allX64 = {
   '@napi-rs/system-ocr-win32-x64-msvc': '1.0.2'
 }
 
-const claudeCodeVenderPath = '@anthropic-ai/claude-code/vendor'
-const claudeCodeVenders = ['arm64-darwin', 'arm64-linux', 'x64-darwin', 'x64-linux', 'x64-win32']
-
 const platformToArch = {
   mac: 'darwin',
   windows: 'win32',
@@ -49,6 +46,9 @@ exports.default = async function (context) {
   const archType = arch === Arch.arm64 ? 'arm64' : 'x64'
   const platform = context.packager.platform.name
 
+  const arm64Filters = Object.keys(allArm64).map((f) => '!node_modules/' + f + '/**')
+  const x64Filters = Object.keys(allX64).map((f) => '!node_modules/' + f + '/*')
+
   const downloadPackages = async (packages) => {
     console.log('downloading packages ......')
     const downloadPromises = []
@@ -67,39 +67,25 @@ exports.default = async function (context) {
     await Promise.all(downloadPromises)
   }
 
-  const changeFilters = async (filtersToExclude, filtersToInclude) => {
+  const changeFilters = async (packages, filtersToExclude, filtersToInclude) => {
+    await downloadPackages(packages)
     // remove filters for the target architecture (allow inclusion)
+
     let filters = context.packager.config.files[0].filter
     filters = filters.filter((filter) => !filtersToInclude.includes(filter))
-
     // add filters for other architectures (exclude them)
     filters.push(...filtersToExclude)
 
     context.packager.config.files[0].filter = filters
   }
 
-  await downloadPackages(arch === Arch.arm64 ? allArm64 : allX64)
-
-  const arm64Filters = Object.keys(allArm64).map((f) => '!node_modules/' + f + '/**')
-  const x64Filters = Object.keys(allX64).map((f) => '!node_modules/' + f + '/*')
-  const excludeClaudeCodeRipgrepFilters = claudeCodeVenders
-    .filter((f) => f !== `${archType}-${platformToArch[platform]}`)
-    .map((f) => '!node_modules/' + claudeCodeVenderPath + '/ripgrep/' + f + '/**')
-  const excludeClaudeCodeJBPlutins = ['!node_modules/' + claudeCodeVenderPath + '/' + 'claude-code-jetbrains-plugin']
-
-  const includeClaudeCodeFilters = [
-    '!node_modules/' + claudeCodeVenderPath + '/' + `${archType}-${platformToArch[platform]}/**`
-  ]
-
   if (arch === Arch.arm64) {
-    await changeFilters(
-      [...x64Filters, ...excludeClaudeCodeRipgrepFilters, ...excludeClaudeCodeJBPlutins],
-      [...arm64Filters, ...includeClaudeCodeFilters]
-    )
-  } else {
-    await changeFilters(
-      [...arm64Filters, ...excludeClaudeCodeRipgrepFilters, ...excludeClaudeCodeJBPlutins],
-      [...x64Filters, ...includeClaudeCodeFilters]
-    )
+    await changeFilters(allArm64, x64Filters, arm64Filters)
+    return
+  }
+
+  if (arch === Arch.x64) {
+    await changeFilters(allX64, arm64Filters, x64Filters)
+    return
   }
 }

scripts/check-i18n.ts

@@ -4,7 +4,7 @@ import * as path from 'path'
 import { sortedObjectByKeys } from './sort'
 
 const translationsDir = path.join(__dirname, '../src/renderer/src/i18n/locales')
-const baseLocale = process.env.BASE_LOCALE ?? 'zh-cn'
+const baseLocale = 'zh-cn'
 const baseFileName = `${baseLocale}.json`
 const baseFilePath = path.join(translationsDir, baseFileName)
 

scripts/sync-i18n.ts

@@ -5,7 +5,7 @@ import { sortedObjectByKeys } from './sort'
 
 const localesDir = path.join(__dirname, '../src/renderer/src/i18n/locales')
 const translateDir = path.join(__dirname, '../src/renderer/src/i18n/translate')
-const baseLocale = process.env.BASE_LOCALE ?? 'zh-cn'
+const baseLocale = 'zh-cn'
 const baseFileName = `${baseLocale}.json`
 const baseFilePath = path.join(localesDir, baseFileName)
 

src/main/apiServer/app.ts

@@ -6,10 +6,8 @@ import { v4 as uuidv4 } from 'uuid'
 import { authMiddleware } from './middleware/auth'
 import { errorHandler } from './middleware/error'
 import { setupOpenAPIDocumentation } from './middleware/openapi'
-import { agentsRoutes } from './routes/agents'
 import { chatRoutes } from './routes/chat'
 import { mcpRoutes } from './routes/mcp'
-import { messagesRoutes } from './routes/messages'
 import { modelsRoutes } from './routes/models'
 
 const logger = loggerService.withContext('ApiServer')
@@ -103,7 +101,10 @@ app.get('/', (_req, res) => {
     name: 'Cherry Studio API',
     version: '1.0.0',
     endpoints: {
-      health: 'GET /health'
+      health: 'GET /health',
+      models: 'GET /v1/models',
+      chat: 'POST /v1/chat/completions',
+      mcp: 'GET /v1/mcps'
     }
   })
 })
@@ -115,9 +116,7 @@ apiRouter.use(express.json())
 // Mount routes
 apiRouter.use('/chat', chatRoutes)
 apiRouter.use('/mcps', mcpRoutes)
-apiRouter.use('/messages', messagesRoutes)
 apiRouter.use('/models', modelsRoutes)
-apiRouter.use('/agents', agentsRoutes)
 app.use('/v1', apiRouter)
 
 // Setup OpenAPI documentation

src/main/apiServer/routes/agents/handlers/agents.ts

@@ -1,532 +0,0 @@
-import { loggerService } from '@logger'
-import { AgentModelValidationError, agentService } from '@main/services/agents'
-import { ListAgentsResponse,type ReplaceAgentRequest, type UpdateAgentRequest } from '@types'
-import { Request, Response } from 'express'
-
-import type { ValidationRequest } from '../validators/zodValidator'
-
-const logger = loggerService.withContext('ApiServerAgentsHandlers')
-
-const modelValidationErrorBody = (error: AgentModelValidationError) => ({
-  error: {
-    message: `Invalid ${error.context.field}: ${error.detail.message}`,
-    type: 'invalid_request_error',
-    code: error.detail.code
-  }
-})
-
-/**
- * @swagger
- * /v1/agents:
- *   post:
- *     summary: Create a new agent
- *     description: Creates a new autonomous agent with the specified configuration
- *     tags: [Agents]
- *     requestBody:
- *       required: true
- *       content:
- *         application/json:
- *           schema:
- *             $ref: '#/components/schemas/CreateAgentRequest'
- *     responses:
- *       201:
- *         description: Agent created successfully
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/AgentEntity'
- *       400:
- *         description: Validation error
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/Error'
- *       500:
- *         description: Internal server error
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/Error'
- */
-export const createAgent = async (req: Request, res: Response): Promise<Response> => {
-  try {
-    logger.info('Creating new agent')
-    logger.debug('Agent data:', req.body)
-
-    const agent = await agentService.createAgent(req.body)
-
-    logger.info(`Agent created successfully: ${agent.id}`)
-    return res.status(201).json(agent)
-  } catch (error: any) {
-    if (error instanceof AgentModelValidationError) {
-      logger.warn('Agent model validation error during create:', {
-        agentType: error.context.agentType,
-        field: error.context.field,
-        model: error.context.model,
-        detail: error.detail
-      })
-      return res.status(400).json(modelValidationErrorBody(error))
-    }
-
-    logger.error('Error creating agent:', error)
-    return res.status(500).json({
-      error: {
-        message: `Failed to create agent: ${error.message}`,
-        type: 'internal_error',
-        code: 'agent_creation_failed'
-      }
-    })
-  }
-}
-
-/**
- * @swagger
- * /v1/agents:
- *   get:
- *     summary: List all agents
- *     description: Retrieves a paginated list of all agents
- *     tags: [Agents]
- *     parameters:
- *       - in: query
- *         name: limit
- *         schema:
- *           type: integer
- *           minimum: 1
- *           maximum: 100
- *           default: 20
- *         description: Number of agents to return
- *       - in: query
- *         name: offset
- *         schema:
- *           type: integer
- *           minimum: 0
- *           default: 0
- *         description: Number of agents to skip
- *     responses:
- *       200:
- *         description: List of agents
- *         content:
- *           application/json:
- *             schema:
- *               type: object
- *               properties:
- *                 data:
- *                   type: array
- *                   items:
- *                     $ref: '#/components/schemas/AgentEntity'
- *                 total:
- *                   type: integer
- *                   description: Total number of agents
- *                 limit:
- *                   type: integer
- *                   description: Number of agents returned
- *                 offset:
- *                   type: integer
- *                   description: Number of agents skipped
- *       400:
- *         description: Validation error
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/Error'
- *       500:
- *         description: Internal server error
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/Error'
- */
-export const listAgents = async (req: Request, res: Response): Promise<Response> => {
-  try {
-    const limit = req.query.limit ? parseInt(req.query.limit as string) : 20
-    const offset = req.query.offset ? parseInt(req.query.offset as string) : 0
-
-    logger.info(`Listing agents with limit=${limit}, offset=${offset}`)
-
-    const result = await agentService.listAgents({ limit, offset })
-
-    logger.info(`Retrieved ${result.agents.length} agents (total: ${result.total})`)
-    return res.json({
-      data: result.agents,
-      total: result.total,
-      limit,
-      offset
-    } satisfies ListAgentsResponse)
-  } catch (error: any) {
-    logger.error('Error listing agents:', error)
-    return res.status(500).json({
-      error: {
-        message: 'Failed to list agents',
-        type: 'internal_error',
-        code: 'agent_list_failed'
-      }
-    })
-  }
-}
-
-/**
- * @swagger
- * /v1/agents/{agentId}:
- *   get:
- *     summary: Get agent by ID
- *     description: Retrieves a specific agent by its ID
- *     tags: [Agents]
- *     parameters:
- *       - in: path
- *         name: agentId
- *         required: true
- *         schema:
- *           type: string
- *         description: Agent ID
- *     responses:
- *       200:
- *         description: Agent details
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/AgentEntity'
- *       404:
- *         description: Agent not found
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/Error'
- *       500:
- *         description: Internal server error
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/Error'
- */
-export const getAgent = async (req: Request, res: Response): Promise<Response> => {
-  try {
-    const { agentId } = req.params
-    logger.info(`Getting agent: ${agentId}`)
-
-    const agent = await agentService.getAgent(agentId)
-
-    if (!agent) {
-      logger.warn(`Agent not found: ${agentId}`)
-      return res.status(404).json({
-        error: {
-          message: 'Agent not found',
-          type: 'not_found',
-          code: 'agent_not_found'
-        }
-      })
-    }
-
-    logger.info(`Agent retrieved successfully: ${agentId}`)
-    return res.json(agent)
-  } catch (error: any) {
-    logger.error('Error getting agent:', error)
-    return res.status(500).json({
-      error: {
-        message: 'Failed to get agent',
-        type: 'internal_error',
-        code: 'agent_get_failed'
-      }
-    })
-  }
-}
-
-/**
- * @swagger
- * /v1/agents/{agentId}:
- *   put:
- *     summary: Update agent
- *     description: Updates an existing agent with the provided data
- *     tags: [Agents]
- *     parameters:
- *       - in: path
- *         name: agentId
- *         required: true
- *         schema:
- *           type: string
- *         description: Agent ID
- *     requestBody:
- *       required: true
- *       content:
- *         application/json:
- *           schema:
- *             $ref: '#/components/schemas/CreateAgentRequest'
- *     responses:
- *       200:
- *         description: Agent updated successfully
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/AgentEntity'
- *       400:
- *         description: Validation error
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/Error'
- *       404:
- *         description: Agent not found
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/Error'
- *       500:
- *         description: Internal server error
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/Error'
- */
-export const updateAgent = async (req: Request, res: Response): Promise<Response> => {
-  const { agentId } = req.params
-  try {
-    logger.info(`Updating agent: ${agentId}`)
-    logger.debug('Update data:', req.body)
-
-    const { validatedBody } = req as ValidationRequest
-    const replacePayload = (validatedBody ?? {}) as ReplaceAgentRequest
-
-    const agent = await agentService.updateAgent(agentId, replacePayload, { replace: true })
-
-    if (!agent) {
-      logger.warn(`Agent not found for update: ${agentId}`)
-      return res.status(404).json({
-        error: {
-          message: 'Agent not found',
-          type: 'not_found',
-          code: 'agent_not_found'
-        }
-      })
-    }
-
-    logger.info(`Agent updated successfully: ${agentId}`)
-    return res.json(agent)
-  } catch (error: any) {
-    if (error instanceof AgentModelValidationError) {
-      logger.warn('Agent model validation error during update:', {
-        agentId,
-        agentType: error.context.agentType,
-        field: error.context.field,
-        model: error.context.model,
-        detail: error.detail
-      })
-      return res.status(400).json(modelValidationErrorBody(error))
-    }
-
-    logger.error('Error updating agent:', error)
-    return res.status(500).json({
-      error: {
-        message: 'Failed to update agent: ' + error.message,
-        type: 'internal_error',
-        code: 'agent_update_failed'
-      }
-    })
-  }
-}
-
-/**
- * @swagger
- * /v1/agents/{agentId}:
- *   patch:
- *     summary: Partially update agent
- *     description: Partially updates an existing agent with only the provided fields
- *     tags: [Agents]
- *     parameters:
- *       - in: path
- *         name: agentId
- *         required: true
- *         schema:
- *           type: string
- *         description: Agent ID
- *     requestBody:
- *       required: true
- *       content:
- *         application/json:
- *           schema:
- *             type: object
- *             properties:
- *               name:
- *                 type: string
- *                 description: Agent name
- *               description:
- *                 type: string
- *                 description: Agent description
- *               avatar:
- *                 type: string
- *                 description: Agent avatar URL
- *               instructions:
- *                 type: string
- *                 description: System prompt/instructions
- *               model:
- *                 type: string
- *                 description: Main model ID
- *               plan_model:
- *                 type: string
- *                 description: Optional planning model ID
- *               small_model:
- *                 type: string
- *                 description: Optional small/fast model ID
- *               built_in_tools:
- *                 type: array
- *                 items:
- *                   type: string
- *                 description: Built-in tool IDs
- *               mcps:
- *                 type: array
- *                 items:
- *                   type: string
- *                 description: MCP tool IDs
- *               knowledges:
- *                 type: array
- *                 items:
- *                   type: string
- *                 description: Knowledge base IDs
- *               configuration:
- *                 type: object
- *                 description: Extensible settings
- *               accessible_paths:
- *                 type: array
- *                 items:
- *                   type: string
- *                 description: Accessible directory paths
- *               permission_mode:
- *                 type: string
- *                 enum: [readOnly, acceptEdits, bypassPermissions]
- *                 description: Permission mode
- *               max_steps:
- *                 type: integer
- *                 description: Maximum steps the agent can take
- *             description: Only include the fields you want to update
- *     responses:
- *       200:
- *         description: Agent updated successfully
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/AgentEntity'
- *       400:
- *         description: Validation error
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/Error'
- *       404:
- *         description: Agent not found
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/Error'
- *       500:
- *         description: Internal server error
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/Error'
- */
-export const patchAgent = async (req: Request, res: Response): Promise<Response> => {
-  const { agentId } = req.params
-  try {
-    logger.info(`Partially updating agent: ${agentId}`)
-    logger.debug('Partial update data:', req.body)
-
-    const { validatedBody } = req as ValidationRequest
-    const updatePayload = (validatedBody ?? {}) as UpdateAgentRequest
-
-    const agent = await agentService.updateAgent(agentId, updatePayload)
-
-    if (!agent) {
-      logger.warn(`Agent not found for partial update: ${agentId}`)
-      return res.status(404).json({
-        error: {
-          message: 'Agent not found',
-          type: 'not_found',
-          code: 'agent_not_found'
-        }
-      })
-    }
-
-    logger.info(`Agent partially updated successfully: ${agentId}`)
-    return res.json(agent)
-  } catch (error: any) {
-    if (error instanceof AgentModelValidationError) {
-      logger.warn('Agent model validation error during partial update:', {
-        agentId,
-        agentType: error.context.agentType,
-        field: error.context.field,
-        model: error.context.model,
-        detail: error.detail
-      })
-      return res.status(400).json(modelValidationErrorBody(error))
-    }
-
-    logger.error('Error partially updating agent:', error)
-    return res.status(500).json({
-      error: {
-        message: `Failed to partially update agent: ${error.message}`,
-        type: 'internal_error',
-        code: 'agent_patch_failed'
-      }
-    })
-  }
-}
-
-/**
- * @swagger
- * /v1/agents/{agentId}:
- *   delete:
- *     summary: Delete agent
- *     description: Deletes an agent and all associated sessions and logs
- *     tags: [Agents]
- *     parameters:
- *       - in: path
- *         name: agentId
- *         required: true
- *         schema:
- *           type: string
- *         description: Agent ID
- *     responses:
- *       204:
- *         description: Agent deleted successfully
- *       404:
- *         description: Agent not found
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/Error'
- *       500:
- *         description: Internal server error
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/Error'
- */
-export const deleteAgent = async (req: Request, res: Response): Promise<Response> => {
-  try {
-    const { agentId } = req.params
-    logger.info(`Deleting agent: ${agentId}`)
-
-    const deleted = await agentService.deleteAgent(agentId)
-
-    if (!deleted) {
-      logger.warn(`Agent not found for deletion: ${agentId}`)
-      return res.status(404).json({
-        error: {
-          message: 'Agent not found',
-          type: 'not_found',
-          code: 'agent_not_found'
-        }
-      })
-    }
-
-    logger.info(`Agent deleted successfully: ${agentId}`)
-    return res.status(204).send()
-  } catch (error: any) {
-    logger.error('Error deleting agent:', error)
-    return res.status(500).json({
-      error: {
-        message: 'Failed to delete agent',
-        type: 'internal_error',
-        code: 'agent_delete_failed'
-      }
-    })
-  }
-}

src/main/apiServer/routes/agents/handlers/index.ts

@@ -1,3 +0,0 @@
-export * as agentHandlers from './agents'
-export * as messageHandlers from './messages'
-export * as sessionHandlers from './sessions'

src/main/apiServer/routes/agents/handlers/messages.ts

@@ -1,230 +0,0 @@
-import { loggerService } from '@logger'
-import { Request, Response } from 'express'
-
-import { agentService, sessionMessageService, sessionService } from '../../../../services/agents'
-
-const logger = loggerService.withContext('ApiServerMessagesHandlers')
-
-// Helper function to verify agent and session exist and belong together
-const verifyAgentAndSession = async (agentId: string, sessionId: string) => {
-  const agentExists = await agentService.agentExists(agentId)
-  if (!agentExists) {
-    throw { status: 404, code: 'agent_not_found', message: 'Agent not found' }
-  }
-
-  const session = await sessionService.getSession(agentId, sessionId)
-  if (!session) {
-    throw { status: 404, code: 'session_not_found', message: 'Session not found' }
-  }
-
-  if (session.agent_id !== agentId) {
-    throw { status: 404, code: 'session_not_found', message: 'Session not found for this agent' }
-  }
-
-  return session
-}
-
-export const createMessage = async (req: Request, res: Response): Promise<void> => {
-  try {
-    const { agentId, sessionId } = req.params
-
-    const session = await verifyAgentAndSession(agentId, sessionId)
-
-    const messageData = req.body
-
-    logger.info(`Creating streaming message for session: ${sessionId}`)
-    logger.debug('Streaming message data:', messageData)
-
-    // Set SSE headers
-    res.setHeader('Content-Type', 'text/event-stream')
-    res.setHeader('Cache-Control', 'no-cache')
-    res.setHeader('Connection', 'keep-alive')
-    res.setHeader('Access-Control-Allow-Origin', '*')
-    res.setHeader('Access-Control-Allow-Headers', 'Cache-Control')
-
-    const abortController = new AbortController()
-    const { stream, completion } = await sessionMessageService.createSessionMessage(
-      session,
-      messageData,
-      abortController
-    )
-    const reader = stream.getReader()
-
-    // Track stream lifecycle so we keep the SSE connection open until persistence finishes
-    let responseEnded = false
-    let streamFinished = false
-
-    const finalizeResponse = () => {
-      if (responseEnded) {
-        return
-      }
-
-      if (!streamFinished) {
-        return
-      }
-
-      responseEnded = true
-      try {
-        // res.write('data: {"type":"finish"}\n\n')
-        res.write('data: [DONE]\n\n')
-      } catch (writeError) {
-        logger.error('Error writing final sentinel to SSE stream:', { error: writeError as Error })
-      }
-      res.end()
-    }
-
-    /**
-     * Client Disconnect Detection for Server-Sent Events (SSE)
-     *
-     * We monitor multiple HTTP events to reliably detect when a client disconnects
-     * from the streaming response. This is crucial for:
-     * - Aborting long-running Claude Code processes
-     * - Cleaning up resources and preventing memory leaks
-     * - Avoiding orphaned processes
-     *
-     * Event Priority & Behavior:
-     * 1. res.on('close') - Most common for SSE client disconnects (browser tab close, curl Ctrl+C)
-     * 2. req.on('aborted') - Explicit request abortion
-     * 3. req.on('close') - Request object closure (less common with SSE)
-     *
-     * When any disconnect event fires, we:
-     * - Abort the Claude Code SDK process via abortController
-     * - Clean up event listeners to prevent memory leaks
-     * - Mark the response as ended to prevent further writes
-     */
-    const handleDisconnect = () => {
-      if (responseEnded) return
-      logger.info(`Client disconnected from streaming message for session: ${sessionId}`)
-      responseEnded = true
-      abortController.abort('Client disconnected')
-      reader.cancel('Client disconnected').catch(() => {})
-    }
-
-    req.on('close', handleDisconnect)
-    req.on('aborted', handleDisconnect)
-    res.on('close', handleDisconnect)
-
-    const pumpStream = async () => {
-      try {
-        while (!responseEnded) {
-          const { done, value } = await reader.read()
-          if (done) {
-            break
-          }
-
-          res.write(`data: ${JSON.stringify(value)}\n\n`)
-        }
-
-        streamFinished = true
-        finalizeResponse()
-      } catch (error) {
-        if (responseEnded) return
-        logger.error('Error reading agent stream:', { error })
-        try {
-          res.write(
-            `data: ${JSON.stringify({
-              type: 'error',
-              error: {
-                message: (error as Error).message || 'Stream processing error',
-                type: 'stream_error',
-                code: 'stream_processing_failed'
-              }
-            })}\n\n`
-          )
-        } catch (writeError) {
-          logger.error('Error writing stream error to SSE:', { error: writeError })
-        }
-        responseEnded = true
-        res.end()
-      }
-    }
-
-    pumpStream().catch((error) => {
-      logger.error('Pump stream failure:', { error })
-    })
-
-    completion
-      .then(() => {
-        streamFinished = true
-        finalizeResponse()
-      })
-      .catch((error) => {
-        if (responseEnded) return
-        logger.error(`Streaming message error for session: ${sessionId}:`, error)
-        try {
-          res.write(
-            `data: ${JSON.stringify({
-              type: 'error',
-              error: {
-                message: (error as { message?: string })?.message || 'Stream processing error',
-                type: 'stream_error',
-                code: 'stream_processing_failed'
-              }
-            })}\n\n`
-          )
-        } catch (writeError) {
-          logger.error('Error writing completion error to SSE stream:', { error: writeError })
-        }
-        responseEnded = true
-        res.end()
-      })
-
-    // Set a timeout to prevent hanging indefinitely
-    const timeout = setTimeout(
-      () => {
-        if (!responseEnded) {
-          logger.error(`Streaming message timeout for session: ${sessionId}`)
-          try {
-            res.write(
-              `data: ${JSON.stringify({
-                type: 'error',
-                error: {
-                  message: 'Stream timeout',
-                  type: 'timeout_error',
-                  code: 'stream_timeout'
-                }
-              })}\n\n`
-            )
-          } catch (writeError) {
-            logger.error('Error writing timeout to SSE stream:', { error: writeError })
-          }
-          abortController.abort('stream timeout')
-          reader.cancel('stream timeout').catch(() => {})
-          responseEnded = true
-          res.end()
-        }
-      },
-      10 * 60 * 1000
-    ) // 10 minutes timeout
-
-    // Clear timeout when response ends
-    res.on('close', () => clearTimeout(timeout))
-    res.on('finish', () => clearTimeout(timeout))
-  } catch (error: any) {
-    logger.error('Error in streaming message handler:', error)
-
-    // Send error as SSE if possible
-    if (!res.headersSent) {
-      res.setHeader('Content-Type', 'text/event-stream')
-      res.setHeader('Cache-Control', 'no-cache')
-      res.setHeader('Connection', 'keep-alive')
-    }
-
-    try {
-      const errorResponse = {
-        type: 'error',
-        error: {
-          message: error.status ? error.message : 'Failed to create streaming message',
-          type: error.status ? 'not_found' : 'internal_error',
-          code: error.status ? error.code : 'stream_creation_failed'
-        }
-      }
-
-      res.write(`data: ${JSON.stringify(errorResponse)}\n\n`)
-    } catch (writeError) {
-      logger.error('Error writing initial error to SSE stream:', { error: writeError })
-    }
-
-    res.end()
-  }
-}

src/main/apiServer/routes/agents/handlers/sessions.ts

@@ -1,370 +0,0 @@
-import { loggerService } from '@logger'
-import {
-  AgentModelValidationError,
-  sessionMessageService,
-  sessionService
-} from '@main/services/agents'
-import {
-  CreateSessionResponse,
-  ListAgentSessionsResponse,
-  type ReplaceSessionRequest,
-  UpdateSessionResponse
-} from '@types'
-import { Request, Response } from 'express'
-
-import type { ValidationRequest } from '../validators/zodValidator'
-
-const logger = loggerService.withContext('ApiServerSessionsHandlers')
-
-const modelValidationErrorBody = (error: AgentModelValidationError) => ({
-  error: {
-    message: `Invalid ${error.context.field}: ${error.detail.message}`,
-    type: 'invalid_request_error',
-    code: error.detail.code
-  }
-})
-
-export const createSession = async (req: Request, res: Response): Promise<Response> => {
-  const { agentId } = req.params
-  try {
-    const sessionData = req.body
-
-    logger.info(`Creating new session for agent: ${agentId}`)
-    logger.debug('Session data:', sessionData)
-
-    const session = (await sessionService.createSession(agentId, sessionData)) satisfies CreateSessionResponse
-
-    logger.info(`Session created successfully: ${session.id}`)
-    return res.status(201).json(session)
-  } catch (error: any) {
-    if (error instanceof AgentModelValidationError) {
-      logger.warn('Session model validation error during create:', {
-        agentId,
-        agentType: error.context.agentType,
-        field: error.context.field,
-        model: error.context.model,
-        detail: error.detail
-      })
-      return res.status(400).json(modelValidationErrorBody(error))
-    }
-
-    logger.error('Error creating session:', error)
-    return res.status(500).json({
-      error: {
-        message: `Failed to create session: ${error.message}`,
-        type: 'internal_error',
-        code: 'session_creation_failed'
-      }
-    })
-  }
-}
-
-export const listSessions = async (req: Request, res: Response): Promise<Response> => {
-  try {
-    const { agentId } = req.params
-    const limit = req.query.limit ? parseInt(req.query.limit as string) : 20
-    const offset = req.query.offset ? parseInt(req.query.offset as string) : 0
-    const status = req.query.status as any
-
-    logger.info(`Listing sessions for agent: ${agentId} with limit=${limit}, offset=${offset}, status=${status}`)
-
-    const result = await sessionService.listSessions(agentId, { limit, offset })
-
-    logger.info(`Retrieved ${result.sessions.length} sessions (total: ${result.total}) for agent: ${agentId}`)
-    return res.json({
-      data: result.sessions,
-      total: result.total,
-      limit,
-      offset
-    })
-  } catch (error: any) {
-    logger.error('Error listing sessions:', error)
-    return res.status(500).json({
-      error: {
-        message: 'Failed to list sessions',
-        type: 'internal_error',
-        code: 'session_list_failed'
-      }
-    })
-  }
-}
-
-export const getSession = async (req: Request, res: Response): Promise<Response> => {
-  try {
-    const { agentId, sessionId } = req.params
-    logger.info(`Getting session: ${sessionId} for agent: ${agentId}`)
-
-    const session = await sessionService.getSession(agentId, sessionId)
-
-    if (!session) {
-      logger.warn(`Session not found: ${sessionId}`)
-      return res.status(404).json({
-        error: {
-          message: 'Session not found',
-          type: 'not_found',
-          code: 'session_not_found'
-        }
-      })
-    }
-
-    // // Verify session belongs to the agent
-    //   logger.warn(`Session ${sessionId} does not belong to agent ${agentId}`)
-    //   return res.status(404).json({
-    //     error: {
-    //       message: 'Session not found for this agent',
-    //       type: 'not_found',
-    //       code: 'session_not_found'
-    //     }
-    //   })
-    // }
-
-    // Fetch session messages
-    logger.info(`Fetching messages for session: ${sessionId}`)
-    const { messages } = await sessionMessageService.listSessionMessages(sessionId)
-
-    // Add messages to session
-    const sessionWithMessages = {
-      ...session,
-      messages: messages
-    }
-
-    logger.info(`Session retrieved successfully: ${sessionId} with ${messages.length} messages`)
-    return res.json(sessionWithMessages)
-  } catch (error: any) {
-    logger.error('Error getting session:', error)
-    return res.status(500).json({
-      error: {
-        message: 'Failed to get session',
-        type: 'internal_error',
-        code: 'session_get_failed'
-      }
-    })
-  }
-}
-
-export const updateSession = async (req: Request, res: Response): Promise<Response> => {
-  const { agentId, sessionId } = req.params
-  try {
-    logger.info(`Updating session: ${sessionId} for agent: ${agentId}`)
-    logger.debug('Update data:', req.body)
-
-    // First check if session exists and belongs to agent
-    const existingSession = await sessionService.getSession(agentId, sessionId)
-    if (!existingSession || existingSession.agent_id !== agentId) {
-      logger.warn(`Session ${sessionId} not found for agent ${agentId}`)
-      return res.status(404).json({
-        error: {
-          message: 'Session not found for this agent',
-          type: 'not_found',
-          code: 'session_not_found'
-        }
-      })
-    }
-
-    const { validatedBody } = req as ValidationRequest
-    const replacePayload = (validatedBody ?? {}) as ReplaceSessionRequest
-
-    const session = await sessionService.updateSession(agentId, sessionId, replacePayload)
-
-    if (!session) {
-      logger.warn(`Session not found for update: ${sessionId}`)
-      return res.status(404).json({
-        error: {
-          message: 'Session not found',
-          type: 'not_found',
-          code: 'session_not_found'
-        }
-      })
-    }
-
-    logger.info(`Session updated successfully: ${sessionId}`)
-    return res.json(session satisfies UpdateSessionResponse)
-  } catch (error: any) {
-    if (error instanceof AgentModelValidationError) {
-      logger.warn('Session model validation error during update:', {
-        agentId,
-        sessionId,
-        agentType: error.context.agentType,
-        field: error.context.field,
-        model: error.context.model,
-        detail: error.detail
-      })
-      return res.status(400).json(modelValidationErrorBody(error))
-    }
-
-    logger.error('Error updating session:', error)
-    return res.status(500).json({
-      error: {
-        message: `Failed to update session: ${error.message}`,
-        type: 'internal_error',
-        code: 'session_update_failed'
-      }
-    })
-  }
-}
-
-export const patchSession = async (req: Request, res: Response): Promise<Response> => {
-  const { agentId, sessionId } = req.params
-  try {
-    logger.info(`Patching session: ${sessionId} for agent: ${agentId}`)
-    logger.debug('Patch data:', req.body)
-
-    // First check if session exists and belongs to agent
-    const existingSession = await sessionService.getSession(agentId, sessionId)
-    if (!existingSession || existingSession.agent_id !== agentId) {
-      logger.warn(`Session ${sessionId} not found for agent ${agentId}`)
-      return res.status(404).json({
-        error: {
-          message: 'Session not found for this agent',
-          type: 'not_found',
-          code: 'session_not_found'
-        }
-      })
-    }
-
-    const updateSession = { ...existingSession, ...req.body }
-    const session = await sessionService.updateSession(agentId, sessionId, updateSession)
-
-    if (!session) {
-      logger.warn(`Session not found for patch: ${sessionId}`)
-      return res.status(404).json({
-        error: {
-          message: 'Session not found',
-          type: 'not_found',
-          code: 'session_not_found'
-        }
-      })
-    }
-
-    logger.info(`Session patched successfully: ${sessionId}`)
-    return res.json(session)
-  } catch (error: any) {
-    if (error instanceof AgentModelValidationError) {
-      logger.warn('Session model validation error during patch:', {
-        agentId,
-        sessionId,
-        agentType: error.context.agentType,
-        field: error.context.field,
-        model: error.context.model,
-        detail: error.detail
-      })
-      return res.status(400).json(modelValidationErrorBody(error))
-    }
-
-    logger.error('Error patching session:', error)
-    return res.status(500).json({
-      error: {
-        message: `Failed to patch session, ${error.message}`,
-        type: 'internal_error',
-        code: 'session_patch_failed'
-      }
-    })
-  }
-}
-
-export const deleteSession = async (req: Request, res: Response): Promise<Response> => {
-  try {
-    const { agentId, sessionId } = req.params
-    logger.info(`Deleting session: ${sessionId} for agent: ${agentId}`)
-
-    // First check if session exists and belongs to agent
-    const existingSession = await sessionService.getSession(agentId, sessionId)
-    if (!existingSession || existingSession.agent_id !== agentId) {
-      logger.warn(`Session ${sessionId} not found for agent ${agentId}`)
-      return res.status(404).json({
-        error: {
-          message: 'Session not found for this agent',
-          type: 'not_found',
-          code: 'session_not_found'
-        }
-      })
-    }
-
-    const deleted = await sessionService.deleteSession(agentId, sessionId)
-
-    if (!deleted) {
-      logger.warn(`Session not found for deletion: ${sessionId}`)
-      return res.status(404).json({
-        error: {
-          message: 'Session not found',
-          type: 'not_found',
-          code: 'session_not_found'
-        }
-      })
-    }
-
-    logger.info(`Session deleted successfully: ${sessionId}`)
-    return res.status(204).send()
-  } catch (error: any) {
-    logger.error('Error deleting session:', error)
-    return res.status(500).json({
-      error: {
-        message: 'Failed to delete session',
-        type: 'internal_error',
-        code: 'session_delete_failed'
-      }
-    })
-  }
-}
-
-// Convenience endpoints for sessions without agent context
-export const listAllSessions = async (req: Request, res: Response): Promise<Response> => {
-  try {
-    const limit = req.query.limit ? parseInt(req.query.limit as string) : 20
-    const offset = req.query.offset ? parseInt(req.query.offset as string) : 0
-    const status = req.query.status as any
-
-    logger.info(`Listing all sessions with limit=${limit}, offset=${offset}, status=${status}`)
-
-    const result = await sessionService.listSessions(undefined, { limit, offset })
-
-    logger.info(`Retrieved ${result.sessions.length} sessions (total: ${result.total})`)
-    return res.json({
-      data: result.sessions,
-      total: result.total,
-      limit,
-      offset
-    } satisfies ListAgentSessionsResponse)
-  } catch (error: any) {
-    logger.error('Error listing all sessions:', error)
-    return res.status(500).json({
-      error: {
-        message: 'Failed to list sessions',
-        type: 'internal_error',
-        code: 'session_list_failed'
-      }
-    })
-  }
-}
-
-export const getSessionById = async (req: Request, res: Response): Promise<Response> => {
-  try {
-    const { sessionId } = req.params
-    logger.info(`Getting session: ${sessionId}`)
-
-    const session = await sessionService.getSessionById(sessionId)
-
-    if (!session) {
-      logger.warn(`Session not found: ${sessionId}`)
-      return res.status(404).json({
-        error: {
-          message: 'Session not found',
-          type: 'not_found',
-          code: 'session_not_found'
-        }
-      })
-    }
-
-    logger.info(`Session retrieved successfully: ${sessionId}`)
-    return res.json(session)
-  } catch (error: any) {
-    logger.error('Error getting session:', error)
-    return res.status(500).json({
-      error: {
-        message: 'Failed to get session',
-        type: 'internal_error',
-        code: 'session_get_failed'
-      }
-    })
-  }
-}

src/main/apiServer/routes/agents/index.ts

@@ -1,927 +0,0 @@
-import express from 'express'
-
-import { agentHandlers, messageHandlers, sessionHandlers } from './handlers'
-import { checkAgentExists, handleValidationErrors } from './middleware'
-import {
-  validateAgent,
-  validateAgentId,
-  validateAgentReplace,
-  validateAgentUpdate,
-  validatePagination,
-  validateSession,
-  validateSessionId,
-  validateSessionMessage,
-  validateSessionReplace,
-  validateSessionUpdate
-} from './validators'
-
-// Create main agents router
-const agentsRouter = express.Router()
-
-/**
- * @swagger
- * components:
- *   schemas:
- *     PermissionMode:
- *       type: string
- *       enum: [default, acceptEdits, bypassPermissions, plan]
- *       description: Permission mode for agent operations
- *
- *     AgentType:
- *       type: string
- *       enum: [claude-code]
- *       description: Type of agent
- *
- *     AgentConfiguration:
- *       type: object
- *       properties:
- *         permission_mode:
- *           $ref: '#/components/schemas/PermissionMode'
- *           default: default
- *         max_turns:
- *           type: integer
- *           default: 10
- *           description: Maximum number of interaction turns
- *       additionalProperties: true
- *
- *     AgentBase:
- *       type: object
- *       properties:
- *         name:
- *           type: string
- *           description: Agent name
- *         description:
- *           type: string
- *           description: Agent description
- *         accessible_paths:
- *           type: array
- *           items:
- *             type: string
- *           description: Array of directory paths the agent can access
- *         instructions:
- *           type: string
- *           description: System prompt/instructions
- *         model:
- *           type: string
- *           description: Main model ID
- *         plan_model:
- *           type: string
- *           description: Optional planning model ID
- *         small_model:
- *           type: string
- *           description: Optional small/fast model ID
- *         mcps:
- *           type: array
- *           items:
- *             type: string
- *           description: Array of MCP tool IDs
- *         allowed_tools:
- *           type: array
- *           items:
- *             type: string
- *           description: Array of allowed tool IDs (whitelist)
- *         configuration:
- *           $ref: '#/components/schemas/AgentConfiguration'
- *       required:
- *         - model
- *         - accessible_paths
- *
- *     AgentEntity:
- *       allOf:
- *         - $ref: '#/components/schemas/AgentBase'
- *         - type: object
- *           properties:
- *             id:
- *               type: string
- *               description: Unique agent identifier
- *             type:
- *               $ref: '#/components/schemas/AgentType'
- *             created_at:
- *               type: string
- *               format: date-time
- *               description: ISO timestamp of creation
- *             updated_at:
- *               type: string
- *               format: date-time
- *               description: ISO timestamp of last update
- *           required:
- *             - id
- *             - type
- *             - created_at
- *             - updated_at
- *     CreateAgentRequest:
- *       allOf:
- *         - $ref: '#/components/schemas/AgentBase'
- *         - type: object
- *           properties:
- *             type:
- *               $ref: '#/components/schemas/AgentType'
- *             name:
- *               type: string
- *               minLength: 1
- *               description: Agent name (required)
- *             model:
- *               type: string
- *               minLength: 1
- *               description: Main model ID (required)
- *           required:
- *             - type
- *             - name
- *             - model
- *
- *     UpdateAgentRequest:
- *       type: object
- *       properties:
- *         name:
- *           type: string
- *           description: Agent name
- *         description:
- *           type: string
- *           description: Agent description
- *         accessible_paths:
- *           type: array
- *           items:
- *             type: string
- *           description: Array of directory paths the agent can access
- *         instructions:
- *           type: string
- *           description: System prompt/instructions
- *         model:
- *           type: string
- *           description: Main model ID
- *         plan_model:
- *           type: string
- *           description: Optional planning model ID
- *         small_model:
- *           type: string
- *           description: Optional small/fast model ID
- *         mcps:
- *           type: array
- *           items:
- *             type: string
- *           description: Array of MCP tool IDs
- *         allowed_tools:
- *           type: array
- *           items:
- *             type: string
- *           description: Array of allowed tool IDs (whitelist)
- *         configuration:
- *           $ref: '#/components/schemas/AgentConfiguration'
- *       description: Partial update - all fields are optional
- *
- *     ReplaceAgentRequest:
- *       $ref: '#/components/schemas/AgentBase'
- *
- *     SessionEntity:
- *       allOf:
- *         - $ref: '#/components/schemas/AgentBase'
- *         - type: object
- *           properties:
- *             id:
- *               type: string
- *               description: Unique session identifier
- *             agent_id:
- *               type: string
- *               description: Primary agent ID for the session
- *             agent_type:
- *               $ref: '#/components/schemas/AgentType'
- *             created_at:
- *               type: string
- *               format: date-time
- *               description: ISO timestamp of creation
- *             updated_at:
- *               type: string
- *               format: date-time
- *               description: ISO timestamp of last update
- *           required:
- *             - id
- *             - agent_id
- *             - agent_type
- *             - created_at
- *             - updated_at
- *
- *     CreateSessionRequest:
- *       allOf:
- *         - $ref: '#/components/schemas/AgentBase'
- *         - type: object
- *           properties:
- *             model:
- *               type: string
- *               minLength: 1
- *               description: Main model ID (required)
- *           required:
- *             - model
- *
- *     UpdateSessionRequest:
- *       type: object
- *       properties:
- *         name:
- *           type: string
- *           description: Session name
- *         description:
- *           type: string
- *           description: Session description
- *         accessible_paths:
- *           type: array
- *           items:
- *             type: string
- *           description: Array of directory paths the agent can access
- *         instructions:
- *           type: string
- *           description: System prompt/instructions
- *         model:
- *           type: string
- *           description: Main model ID
- *         plan_model:
- *           type: string
- *           description: Optional planning model ID
- *         small_model:
- *           type: string
- *           description: Optional small/fast model ID
- *         mcps:
- *           type: array
- *           items:
- *             type: string
- *           description: Array of MCP tool IDs
- *         allowed_tools:
- *           type: array
- *           items:
- *             type: string
- *           description: Array of allowed tool IDs (whitelist)
- *         configuration:
- *           $ref: '#/components/schemas/AgentConfiguration'
- *       description: Partial update - all fields are optional
- *
- *     ReplaceSessionRequest:
- *       allOf:
- *         - $ref: '#/components/schemas/AgentBase'
- *         - type: object
- *           properties:
- *             model:
- *               type: string
- *               minLength: 1
- *               description: Main model ID (required)
- *           required:
- *             - model
- *
- *     CreateSessionMessageRequest:
- *       type: object
- *       properties:
- *         content:
- *           type: string
- *           minLength: 1
- *           description: Message content
- *       required:
- *         - content
- *
- *     PaginationQuery:
- *       type: object
- *       properties:
- *         limit:
- *           type: integer
- *           minimum: 1
- *           maximum: 100
- *           default: 20
- *           description: Number of items to return
- *         offset:
- *           type: integer
- *           minimum: 0
- *           default: 0
- *           description: Number of items to skip
- *         status:
- *           type: string
- *           enum: [idle, running, completed, failed, stopped]
- *           description: Filter by session status
- *
- *     ListAgentsResponse:
- *       type: object
- *       properties:
- *         agents:
- *           type: array
- *           items:
- *             $ref: '#/components/schemas/AgentEntity'
- *         total:
- *           type: integer
- *           description: Total number of agents
- *         limit:
- *           type: integer
- *           description: Number of items returned
- *         offset:
- *           type: integer
- *           description: Number of items skipped
- *       required:
- *         - agents
- *         - total
- *         - limit
- *         - offset
- *
- *     ListSessionsResponse:
- *       type: object
- *       properties:
- *         sessions:
- *           type: array
- *           items:
- *             $ref: '#/components/schemas/SessionEntity'
- *         total:
- *           type: integer
- *           description: Total number of sessions
- *         limit:
- *           type: integer
- *           description: Number of items returned
- *         offset:
- *           type: integer
- *           description: Number of items skipped
- *       required:
- *         - sessions
- *         - total
- *         - limit
- *         - offset
- *
- *     ErrorResponse:
- *       type: object
- *       properties:
- *         error:
- *           type: object
- *           properties:
- *             message:
- *               type: string
- *               description: Error message
- *             type:
- *               type: string
- *               description: Error type
- *             code:
- *               type: string
- *               description: Error code
- *           required:
- *             - message
- *             - type
- *             - code
- *       required:
- *         - error
- */
-
-/**
- * @swagger
- * /api/agents:
- *   post:
- *     summary: Create a new agent
- *     tags: [Agents]
- *     requestBody:
- *       required: true
- *       content:
- *         application/json:
- *           schema:
- *             $ref: '#/components/schemas/CreateAgentRequest'
- *     responses:
- *       201:
- *         description: Agent created successfully
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/AgentEntity'
- *       400:
- *         description: Invalid request body
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/ErrorResponse'
- */
-// Agent CRUD routes
-agentsRouter.post('/', validateAgent, handleValidationErrors, agentHandlers.createAgent)
-
-/**
- * @swagger
- * /api/agents:
- *   get:
- *     summary: List all agents with pagination
- *     tags: [Agents]
- *     parameters:
- *       - in: query
- *         name: limit
- *         schema:
- *           type: integer
- *           minimum: 1
- *           maximum: 100
- *           default: 20
- *         description: Number of agents to return
- *       - in: query
- *         name: offset
- *         schema:
- *           type: integer
- *           minimum: 0
- *           default: 0
- *         description: Number of agents to skip
- *       - in: query
- *         name: status
- *         schema:
- *           type: string
- *           enum: [idle, running, completed, failed, stopped]
- *         description: Filter by agent status
- *     responses:
- *       200:
- *         description: List of agents
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/ListAgentsResponse'
- */
-agentsRouter.get('/', validatePagination, handleValidationErrors, agentHandlers.listAgents)
-
-/**
- * @swagger
- * /api/agents/{agentId}:
- *   get:
- *     summary: Get agent by ID
- *     tags: [Agents]
- *     parameters:
- *       - in: path
- *         name: agentId
- *         required: true
- *         schema:
- *           type: string
- *         description: Agent ID
- *     responses:
- *       200:
- *         description: Agent details
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/AgentEntity'
- *       404:
- *         description: Agent not found
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/ErrorResponse'
- */
-agentsRouter.get('/:agentId', validateAgentId, handleValidationErrors, agentHandlers.getAgent)
-/**
- * @swagger
- * /api/agents/{agentId}:
- *   put:
- *     summary: Replace agent (full update)
- *     tags: [Agents]
- *     parameters:
- *       - in: path
- *         name: agentId
- *         required: true
- *         schema:
- *           type: string
- *         description: Agent ID
- *     requestBody:
- *       required: true
- *       content:
- *         application/json:
- *           schema:
- *             $ref: '#/components/schemas/ReplaceAgentRequest'
- *     responses:
- *       200:
- *         description: Agent updated successfully
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/AgentEntity'
- *       400:
- *         description: Invalid request body
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/ErrorResponse'
- *       404:
- *         description: Agent not found
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/ErrorResponse'
- */
-agentsRouter.put('/:agentId', validateAgentId, validateAgentReplace, handleValidationErrors, agentHandlers.updateAgent)
-/**
- * @swagger
- * /api/agents/{agentId}:
- *   patch:
- *     summary: Update agent (partial update)
- *     tags: [Agents]
- *     parameters:
- *       - in: path
- *         name: agentId
- *         required: true
- *         schema:
- *           type: string
- *         description: Agent ID
- *     requestBody:
- *       required: true
- *       content:
- *         application/json:
- *           schema:
- *             $ref: '#/components/schemas/UpdateAgentRequest'
- *     responses:
- *       200:
- *         description: Agent updated successfully
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/AgentEntity'
- *       400:
- *         description: Invalid request body
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/ErrorResponse'
- *       404:
- *         description: Agent not found
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/ErrorResponse'
- */
-agentsRouter.patch('/:agentId', validateAgentId, validateAgentUpdate, handleValidationErrors, agentHandlers.patchAgent)
-/**
- * @swagger
- * /api/agents/{agentId}:
- *   delete:
- *     summary: Delete agent
- *     tags: [Agents]
- *     parameters:
- *       - in: path
- *         name: agentId
- *         required: true
- *         schema:
- *           type: string
- *         description: Agent ID
- *     responses:
- *       204:
- *         description: Agent deleted successfully
- *       404:
- *         description: Agent not found
- *         content:
- *           application/json:
- *             schema:
- *               $ref: '#/components/schemas/ErrorResponse'
- */
-agentsRouter.delete('/:agentId', validateAgentId, handleValidationErrors, agentHandlers.deleteAgent)
-
-// Create sessions router with agent context
-const createSessionsRouter = (): express.Router => {
-  const sessionsRouter = express.Router({ mergeParams: true })
-
-  // Session CRUD routes (nested under agent)
-  /**
-   * @swagger
-   * /api/agents/{agentId}/sessions:
-   *   post:
-   *     summary: Create a new session for an agent
-   *     tags: [Sessions]
-   *     parameters:
-   *       - in: path
-   *         name: agentId
-   *         required: true
-   *         schema:
-   *           type: string
-   *         description: Agent ID
-   *     requestBody:
-   *       required: true
-   *       content:
-   *         application/json:
-   *           schema:
-   *             $ref: '#/components/schemas/CreateSessionRequest'
-   *     responses:
-   *       201:
-   *         description: Session created successfully
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/SessionEntity'
-   *       400:
-   *         description: Invalid request body
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/ErrorResponse'
-   *       404:
-   *         description: Agent not found
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/ErrorResponse'
-   */
-  sessionsRouter.post('/', validateSession, handleValidationErrors, sessionHandlers.createSession)
-
-  /**
-   * @swagger
-   * /api/agents/{agentId}/sessions:
-   *   get:
-   *     summary: List sessions for an agent
-   *     tags: [Sessions]
-   *     parameters:
-   *       - in: path
-   *         name: agentId
-   *         required: true
-   *         schema:
-   *           type: string
-   *         description: Agent ID
-   *       - in: query
-   *         name: limit
-   *         schema:
-   *           type: integer
-   *           minimum: 1
-   *           maximum: 100
-   *           default: 20
-   *         description: Number of sessions to return
-   *       - in: query
-   *         name: offset
-   *         schema:
-   *           type: integer
-   *           minimum: 0
-   *           default: 0
-   *         description: Number of sessions to skip
-   *       - in: query
-   *         name: status
-   *         schema:
-   *           type: string
-   *           enum: [idle, running, completed, failed, stopped]
-   *         description: Filter by session status
-   *     responses:
-   *       200:
-   *         description: List of sessions
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/ListSessionsResponse'
-   *       404:
-   *         description: Agent not found
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/ErrorResponse'
-   */
-  sessionsRouter.get('/', validatePagination, handleValidationErrors, sessionHandlers.listSessions)
-  /**
-   * @swagger
-   * /api/agents/{agentId}/sessions/{sessionId}:
-   *   get:
-   *     summary: Get session by ID
-   *     tags: [Sessions]
-   *     parameters:
-   *       - in: path
-   *         name: agentId
-   *         required: true
-   *         schema:
-   *           type: string
-   *         description: Agent ID
-   *       - in: path
-   *         name: sessionId
-   *         required: true
-   *         schema:
-   *           type: string
-   *         description: Session ID
-   *     responses:
-   *       200:
-   *         description: Session details
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/SessionEntity'
-   *       404:
-   *         description: Agent or session not found
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/ErrorResponse'
-   */
-  sessionsRouter.get('/:sessionId', validateSessionId, handleValidationErrors, sessionHandlers.getSession)
-  /**
-   * @swagger
-   * /api/agents/{agentId}/sessions/{sessionId}:
-   *   put:
-   *     summary: Replace session (full update)
-   *     tags: [Sessions]
-   *     parameters:
-   *       - in: path
-   *         name: agentId
-   *         required: true
-   *         schema:
-   *           type: string
-   *         description: Agent ID
-   *       - in: path
-   *         name: sessionId
-   *         required: true
-   *         schema:
-   *           type: string
-   *         description: Session ID
-   *     requestBody:
-   *       required: true
-   *       content:
-   *         application/json:
-   *           schema:
-   *             $ref: '#/components/schemas/ReplaceSessionRequest'
-   *     responses:
-   *       200:
-   *         description: Session updated successfully
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/SessionEntity'
-   *       400:
-   *         description: Invalid request body
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/ErrorResponse'
-   *       404:
-   *         description: Agent or session not found
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/ErrorResponse'
-   */
-  sessionsRouter.put(
-    '/:sessionId',
-    validateSessionId,
-    validateSessionReplace,
-    handleValidationErrors,
-    sessionHandlers.updateSession
-  )
-  /**
-   * @swagger
-   * /api/agents/{agentId}/sessions/{sessionId}:
-   *   patch:
-   *     summary: Update session (partial update)
-   *     tags: [Sessions]
-   *     parameters:
-   *       - in: path
-   *         name: agentId
-   *         required: true
-   *         schema:
-   *           type: string
-   *         description: Agent ID
-   *       - in: path
-   *         name: sessionId
-   *         required: true
-   *         schema:
-   *           type: string
-   *         description: Session ID
-   *     requestBody:
-   *       required: true
-   *       content:
-   *         application/json:
-   *           schema:
-   *             $ref: '#/components/schemas/UpdateSessionRequest'
-   *     responses:
-   *       200:
-   *         description: Session updated successfully
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/SessionEntity'
-   *       400:
-   *         description: Invalid request body
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/ErrorResponse'
-   *       404:
-   *         description: Agent or session not found
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/ErrorResponse'
-   */
-  sessionsRouter.patch(
-    '/:sessionId',
-    validateSessionId,
-    validateSessionUpdate,
-    handleValidationErrors,
-    sessionHandlers.patchSession
-  )
-  /**
-   * @swagger
-   * /api/agents/{agentId}/sessions/{sessionId}:
-   *   delete:
-   *     summary: Delete session
-   *     tags: [Sessions]
-   *     parameters:
-   *       - in: path
-   *         name: agentId
-   *         required: true
-   *         schema:
-   *           type: string
-   *         description: Agent ID
-   *       - in: path
-   *         name: sessionId
-   *         required: true
-   *         schema:
-   *           type: string
-   *         description: Session ID
-   *     responses:
-   *       204:
-   *         description: Session deleted successfully
-   *       404:
-   *         description: Agent or session not found
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/ErrorResponse'
-   */
-  sessionsRouter.delete('/:sessionId', validateSessionId, handleValidationErrors, sessionHandlers.deleteSession)
-
-  return sessionsRouter
-}
-
-// Create messages router with agent and session context
-const createMessagesRouter = (): express.Router => {
-  const messagesRouter = express.Router({ mergeParams: true })
-
-  // Message CRUD routes (nested under agent/session)
-  /**
-   * @swagger
-   * /api/agents/{agentId}/sessions/{sessionId}/messages:
-   *   post:
-   *     summary: Create a new message in a session
-   *     tags: [Messages]
-   *     parameters:
-   *       - in: path
-   *         name: agentId
-   *         required: true
-   *         schema:
-   *           type: string
-   *         description: Agent ID
-   *       - in: path
-   *         name: sessionId
-   *         required: true
-   *         schema:
-   *           type: string
-   *         description: Session ID
-   *     requestBody:
-   *       required: true
-   *       content:
-   *         application/json:
-   *           schema:
-   *             $ref: '#/components/schemas/CreateSessionMessageRequest'
-   *     responses:
-   *       201:
-   *         description: Message created successfully
-   *         content:
-   *           application/json:
-   *             schema:
-   *               type: object
-   *               properties:
-   *                 id:
-   *                   type: number
-   *                   description: Message ID
-   *                 session_id:
-   *                   type: string
-   *                   description: Session ID
-   *                 role:
-   *                   type: string
-   *                   enum: [assistant, user, system, tool]
-   *                   description: Message role
-   *                 content:
-   *                   type: object
-   *                   description: Message content (AI SDK format)
-   *                 agent_session_id:
-   *                   type: string
-   *                   description: Agent session ID for resuming
-   *                 metadata:
-   *                   type: object
-   *                   description: Additional metadata
-   *                 created_at:
-   *                   type: string
-   *                   format: date-time
-   *                 updated_at:
-   *                   type: string
-   *                   format: date-time
-   *       400:
-   *         description: Invalid request body
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/ErrorResponse'
-   *       404:
-   *         description: Agent or session not found
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/ErrorResponse'
-   */
-  messagesRouter.post('/', validateSessionMessage, handleValidationErrors, messageHandlers.createMessage)
-  return messagesRouter
-}
-
-// Mount nested resources with clear hierarchy
-const sessionsRouter = createSessionsRouter()
-const messagesRouter = createMessagesRouter()
-
-// Mount sessions under specific agent
-agentsRouter.use('/:agentId/sessions', validateAgentId, checkAgentExists, handleValidationErrors, sessionsRouter)
-
-// Mount messages under specific agent/session
-agentsRouter.use(
-  '/:agentId/sessions/:sessionId/messages',
-  validateAgentId,
-  validateSessionId,
-  handleValidationErrors,
-  messagesRouter
-)
-
-// Export main router and convenience router
-export const agentsRoutes = agentsRouter

src/main/apiServer/routes/agents/middleware/common.ts

@@ -1,41 +0,0 @@
-import { Request, Response } from 'express'
-
-import { agentService } from '../../../../services/agents'
-import { loggerService } from '../../../../services/LoggerService'
-
-const logger = loggerService.withContext('ApiServerMiddleware')
-
-// Since Zod validators handle their own errors, this is now a pass-through
-export const handleValidationErrors = (_req: Request, _res: Response, next: any): void => {
-  next()
-}
-
-// Middleware to check if agent exists
-export const checkAgentExists = async (req: Request, res: Response, next: any): Promise<void> => {
-  try {
-    const { agentId } = req.params
-    const exists = await agentService.agentExists(agentId)
-
-    if (!exists) {
-      res.status(404).json({
-        error: {
-          message: 'Agent not found',
-          type: 'not_found',
-          code: 'agent_not_found'
-        }
-      })
-      return
-    }
-
-    next()
-  } catch (error) {
-    logger.error('Error checking agent existence:', error as Error)
-    res.status(500).json({
-      error: {
-        message: 'Failed to validate agent',
-        type: 'internal_error',
-        code: 'agent_validation_failed'
-      }
-    })
-  }
-}

src/main/apiServer/routes/agents/middleware/index.ts

@@ -1 +0,0 @@
-export * from './common'

src/main/apiServer/routes/agents/validators/agents.ts

@@ -1,24 +0,0 @@
-import {
-  AgentIdParamSchema,
-  CreateAgentRequestSchema,
-  ReplaceAgentRequestSchema,
-  UpdateAgentRequestSchema
-} from '@types'
-
-import { createZodValidator } from './zodValidator'
-
-export const validateAgent = createZodValidator({
-  body: CreateAgentRequestSchema
-})
-
-export const validateAgentReplace = createZodValidator({
-  body: ReplaceAgentRequestSchema
-})
-
-export const validateAgentUpdate = createZodValidator({
-  body: UpdateAgentRequestSchema
-})
-
-export const validateAgentId = createZodValidator({
-  params: AgentIdParamSchema
-})

src/main/apiServer/routes/agents/validators/common.ts

@@ -1,7 +0,0 @@
-import { PaginationQuerySchema } from '@types'
-
-import { createZodValidator } from './zodValidator'
-
-export const validatePagination = createZodValidator({
-  query: PaginationQuerySchema
-})

src/main/apiServer/routes/agents/validators/index.ts

@@ -1,4 +0,0 @@
-export * from './agents'
-export * from './common'
-export * from './messages'
-export * from './sessions'

src/main/apiServer/routes/agents/validators/messages.ts

@@ -1,7 +0,0 @@
-import { CreateSessionMessageRequestSchema } from '@types'
-
-import { createZodValidator } from './zodValidator'
-
-export const validateSessionMessage = createZodValidator({
-  body: CreateSessionMessageRequestSchema
-})

src/main/apiServer/routes/agents/validators/sessions.ts

@@ -1,24 +0,0 @@
-import {
-  CreateSessionRequestSchema,
-  ReplaceSessionRequestSchema,
-  SessionIdParamSchema,
-  UpdateSessionRequestSchema
-} from '@types'
-
-import { createZodValidator } from './zodValidator'
-
-export const validateSession = createZodValidator({
-  body: CreateSessionRequestSchema
-})
-
-export const validateSessionReplace = createZodValidator({
-  body: ReplaceSessionRequestSchema
-})
-
-export const validateSessionUpdate = createZodValidator({
-  body: UpdateSessionRequestSchema
-})
-
-export const validateSessionId = createZodValidator({
-  params: SessionIdParamSchema
-})

src/main/apiServer/routes/agents/validators/zodValidator.ts

@@ -1,68 +0,0 @@
-import { NextFunction,Request, Response } from 'express'
-import { ZodError, ZodType } from 'zod'
-
-export interface ValidationRequest extends Request {
-  validatedBody?: any
-  validatedParams?: any
-  validatedQuery?: any
-}
-
-export interface ZodValidationConfig {
-  body?: ZodType
-  params?: ZodType
-  query?: ZodType
-}
-
-export const createZodValidator = (config: ZodValidationConfig) => {
-  return (req: ValidationRequest, res: Response, next: NextFunction): void => {
-    try {
-      if (config.body && req.body) {
-        req.validatedBody = config.body.parse(req.body)
-      }
-
-      if (config.params && req.params) {
-        req.validatedParams = config.params.parse(req.params)
-      }
-
-      if (config.query && req.query) {
-        req.validatedQuery = config.query.parse(req.query)
-      }
-
-      next()
-    } catch (error) {
-      if (error instanceof ZodError) {
-        const validationErrors = error.issues.map((err) => ({
-          type: 'field',
-          value: err.input,
-          msg: err.message,
-          path: err.path.map(p => String(p)).join('.'),
-          location: getLocationFromPath(err.path, config)
-        }))
-
-        res.status(400).json({
-          error: {
-            message: 'Validation failed',
-            type: 'validation_error',
-            details: validationErrors
-          }
-        })
-        return
-      }
-
-      res.status(500).json({
-        error: {
-          message: 'Internal validation error',
-          type: 'internal_error',
-          code: 'validation_processing_failed'
-        }
-      })
-    }
-  }
-}
-
-function getLocationFromPath(path: (string | number | symbol)[], config: ZodValidationConfig): string {
-  if (config.body && path.length > 0) return 'body'
-  if (config.params && path.length > 0) return 'params'
-  if (config.query && path.length > 0) return 'query'
-  return 'unknown'
-}

src/main/apiServer/routes/chat.ts

@@ -1,105 +1,15 @@
 import express, { Request, Response } from 'express'
+import OpenAI from 'openai'
 import { ChatCompletionCreateParams } from 'openai/resources'
 
 import { loggerService } from '../../services/LoggerService'
-import {
-  ChatCompletionModelError,
-  chatCompletionService,
-  ChatCompletionValidationError
-} from '../services/chat-completion'
+import { chatCompletionService } from '../services/chat-completion'
+import { validateModelId } from '../utils'
 
 const logger = loggerService.withContext('ApiServerChatRoutes')
 
 const router = express.Router()
 
-interface ErrorResponseBody {
-  error: {
-    message: string
-    type: string
-    code: string
-  }
-}
-
-const mapChatCompletionError = (error: unknown): { status: number; body: ErrorResponseBody } => {
-  if (error instanceof ChatCompletionValidationError) {
-    logger.warn('Chat completion validation error:', {
-      errors: error.errors
-    })
-
-    return {
-      status: 400,
-      body: {
-        error: {
-          message: error.errors.join('; '),
-          type: 'invalid_request_error',
-          code: 'validation_failed'
-        }
-      }
-    }
-  }
-
-  if (error instanceof ChatCompletionModelError) {
-    logger.warn('Chat completion model error:', error.error)
-
-    return {
-      status: 400,
-      body: {
-        error: {
-          message: error.error.message,
-          type: 'invalid_request_error',
-          code: error.error.code
-        }
-      }
-    }
-  }
-
-  if (error instanceof Error) {
-    let statusCode = 500
-    let errorType = 'server_error'
-    let errorCode = 'internal_error'
-
-    if (error.message.includes('API key') || error.message.includes('authentication')) {
-      statusCode = 401
-      errorType = 'authentication_error'
-      errorCode = 'invalid_api_key'
-    } else if (error.message.includes('rate limit') || error.message.includes('quota')) {
-      statusCode = 429
-      errorType = 'rate_limit_error'
-      errorCode = 'rate_limit_exceeded'
-    } else if (error.message.includes('timeout') || error.message.includes('connection')) {
-      statusCode = 502
-      errorType = 'server_error'
-      errorCode = 'upstream_error'
-    }
-
-    logger.error('Chat completion error:', { error })
-
-    return {
-      status: statusCode,
-      body: {
-        error: {
-          message: error.message || 'Internal server error',
-          type: errorType,
-          code: errorCode
-        }
-      }
-    }
-  }
-
-  logger.error('Chat completion unknown error:', { error })
-
-  return {
-    status: 500,
-    body: {
-      error: {
-        message: 'Internal server error',
-        type: 'server_error',
-        code: 'internal_error'
-      }
-    }
-  }
-}
-
 /**
  * @swagger
  * /v1/chat/completions:
@@ -150,7 +60,7 @@ const mapChatCompletionError = (error: unknown): { status: number; body: ErrorRe
  *                       type: integer
  *                     total_tokens:
  *                       type: integer
- *           text/event-stream:
+ *           text/plain:
  *             schema:
  *               type: string
  *               description: Server-sent events stream (when stream=true)
@@ -200,22 +110,63 @@ router.post('/completions', async (req: Request, res: Response) => {
       temperature: request.temperature
     })
 
-    const isStreaming = !!request.stream
+    // Validate request
+    const validation = chatCompletionService.validateRequest(request)
+    if (!validation.isValid) {
+      return res.status(400).json({
+        error: {
+          message: validation.errors.join('; '),
+          type: 'invalid_request_error',
+          code: 'validation_failed'
+        }
+      })
+    }
 
-    if (isStreaming) {
-      const { stream } = await chatCompletionService.processStreamingCompletion(request)
+    // Validate model ID and get provider
+    const modelValidation = await validateModelId(request.model)
+    if (!modelValidation.valid) {
+      const error = modelValidation.error!
+      logger.warn(`Model validation failed for '${request.model}':`, error)
+      return res.status(400).json({
+        error: {
+          message: error.message,
+          type: 'invalid_request_error',
+          code: error.code
+        }
+      })
+    }
 
-      res.setHeader('Content-Type', 'text/event-stream; charset=utf-8')
-      res.setHeader('Cache-Control', 'no-cache, no-transform')
+    const provider = modelValidation.provider!
+    const modelId = modelValidation.modelId!
+
+    logger.info('Model validation successful:', {
+      provider: provider.id,
+      providerType: provider.type,
+      modelId: modelId,
+      fullModelId: request.model
+    })
+
+    // Create OpenAI client
+    const client = new OpenAI({
+      baseURL: provider.apiHost,
+      apiKey: provider.apiKey
+    })
+    request.model = modelId
+
+    // Handle streaming
+    if (request.stream) {
+      const streamResponse = await client.chat.completions.create(request)
+
+      res.setHeader('Content-Type', 'text/plain; charset=utf-8')
+      res.setHeader('Cache-Control', 'no-cache')
       res.setHeader('Connection', 'keep-alive')
-      res.setHeader('X-Accel-Buffering', 'no')
-      res.flushHeaders()
 
       try {
-        for await (const chunk of stream) {
+        for await (const chunk of streamResponse as any) {
           res.write(`data: ${JSON.stringify(chunk)}\n\n`)
         }
         res.write('data: [DONE]\n\n')
+        res.end()
       } catch (streamError: any) {
         logger.error('Stream error:', streamError)
         res.write(
@@ -227,17 +178,47 @@ router.post('/completions', async (req: Request, res: Response) => {
             }
           })}\n\n`
         )
-      } finally {
         res.end()
       }
       return
     }
 
-    const { response } = await chatCompletionService.processCompletion(request)
+    // Handle non-streaming
+    const response = await client.chat.completions.create(request)
     return res.json(response)
-  } catch (error: unknown) {
-    const { status, body } = mapChatCompletionError(error)
-    return res.status(status).json(body)
+  } catch (error: any) {
+    logger.error('Chat completion error:', error)
+
+    let statusCode = 500
+    let errorType = 'server_error'
+    let errorCode = 'internal_error'
+    let errorMessage = 'Internal server error'
+
+    if (error instanceof Error) {
+      errorMessage = error.message
+
+      if (error.message.includes('API key') || error.message.includes('authentication')) {
+        statusCode = 401
+        errorType = 'authentication_error'
+        errorCode = 'invalid_api_key'
+      } else if (error.message.includes('rate limit') || error.message.includes('quota')) {
+        statusCode = 429
+        errorType = 'rate_limit_error'
+        errorCode = 'rate_limit_exceeded'
+      } else if (error.message.includes('timeout') || error.message.includes('connection')) {
+        statusCode = 502
+        errorType = 'server_error'
+        errorCode = 'upstream_error'
+      }
+    }
+
+    return res.status(statusCode).json({
+      error: {
+        message: errorMessage,
+        type: errorType,
+        code: errorCode
+      }
+    })
   }
 })
 

src/main/apiServer/routes/messages.ts

@@ -1,290 +0,0 @@
-import { MessageCreateParams } from '@anthropic-ai/sdk/resources'
-import express, { Request, Response } from 'express'
-
-import { loggerService } from '../../services/LoggerService'
-import { messagesService } from '../services/messages'
-import { validateModelId } from '../utils'
-
-const logger = loggerService.withContext('ApiServerMessagesRoutes')
-
-const router = express.Router()
-
-/**
- * @swagger
- * /v1/messages:
- *   post:
- *     summary: Create message
- *     description: Create a message response using Anthropic's API format
- *     tags: [Messages]
- *     requestBody:
- *       required: true
- *       content:
- *         application/json:
- *           schema:
- *             type: object
- *             required:
- *               - model
- *               - max_tokens
- *               - messages
- *             properties:
- *               model:
- *                 type: string
- *                 description: Model ID in format "provider:model_id"
- *                 example: "my-anthropic:claude-3-5-sonnet-20241022"
- *               max_tokens:
- *                 type: integer
- *                 minimum: 1
- *                 description: Maximum number of tokens to generate
- *                 example: 1024
- *               messages:
- *                 type: array
- *                 items:
- *                   type: object
- *                   properties:
- *                     role:
- *                       type: string
- *                       enum: [user, assistant]
- *                     content:
- *                       oneOf:
- *                         - type: string
- *                         - type: array
- *               system:
- *                 type: string
- *                 description: System message
- *               temperature:
- *                 type: number
- *                 minimum: 0
- *                 maximum: 1
- *                 description: Sampling temperature
- *               top_p:
- *                 type: number
- *                 minimum: 0
- *                 maximum: 1
- *                 description: Nucleus sampling
- *               top_k:
- *                 type: integer
- *                 minimum: 0
- *                 description: Top-k sampling
- *               stream:
- *                 type: boolean
- *                 description: Whether to stream the response
- *               tools:
- *                 type: array
- *                 description: Available tools for the model
- *     responses:
- *       200:
- *         description: Message response
- *         content:
- *           application/json:
- *             schema:
- *               type: object
- *               properties:
- *                 id:
- *                   type: string
- *                 type:
- *                   type: string
- *                   example: message
- *                 role:
- *                   type: string
- *                   example: assistant
- *                 content:
- *                   type: array
- *                   items:
- *                     type: object
- *                 model:
- *                   type: string
- *                 stop_reason:
- *                   type: string
- *                 stop_sequence:
- *                   type: string
- *                 usage:
- *                   type: object
- *                   properties:
- *                     input_tokens:
- *                       type: integer
- *                     output_tokens:
- *                       type: integer
- *           text/event-stream:
- *             schema:
- *               type: string
- *               description: Server-sent events stream (when stream=true)
- *       400:
- *         description: Bad request
- *         content:
- *           application/json:
- *             schema:
- *               type: object
- *               properties:
- *                 type:
- *                   type: string
- *                   example: error
- *                 error:
- *                   type: object
- *                   properties:
- *                     type:
- *                       type: string
- *                     message:
- *                       type: string
- *       401:
- *         description: Unauthorized
- *       429:
- *         description: Rate limit exceeded
- *       500:
- *         description: Internal server error
- */
-router.post('/', async (req: Request, res: Response) => {
-  try {
-    const request: MessageCreateParams = req.body
-
-    if (!request) {
-      return res.status(400).json({
-        type: 'error',
-        error: {
-          type: 'invalid_request_error',
-          message: 'Request body is required'
-        }
-      })
-    }
-
-    logger.info('Anthropic message request:', {
-      model: request.model,
-      messageCount: request.messages?.length || 0,
-      stream: request.stream,
-      max_tokens: request.max_tokens,
-      temperature: request.temperature
-    })
-
-    // Validate model ID and get provider
-    const modelValidation = await validateModelId(request.model)
-    if (!modelValidation.valid) {
-      const error = modelValidation.error!
-      logger.warn(`Model validation failed for '${request.model}':`, error)
-      return res.status(400).json({
-        type: 'error',
-        error: {
-          type: 'invalid_request_error',
-          message: error.message
-        }
-      })
-    }
-
-    const provider = modelValidation.provider!
-
-    // Ensure provider is Anthropic type
-    if (provider.type !== 'anthropic') {
-      return res.status(400).json({
-        type: 'error',
-        error: {
-          type: 'invalid_request_error',
-          message: `Invalid provider type '${provider.type}' for messages endpoint. Expected 'anthropic' provider.`
-        }
-      })
-    }
-
-    const modelId = modelValidation.modelId!
-    request.model = modelId
-
-    logger.info('Model validation successful:', {
-      provider: provider.id,
-      providerType: provider.type,
-      modelId: modelId,
-      fullModelId: request.model
-    })
-
-    // Validate request
-    const validation = messagesService.validateRequest(request)
-    if (!validation.isValid) {
-      return res.status(400).json({
-        type: 'error',
-        error: {
-          type: 'invalid_request_error',
-          message: validation.errors.join('; ')
-        }
-      })
-    }
-
-    // Handle streaming
-    if (request.stream) {
-      res.setHeader('Content-Type', 'text/event-stream; charset=utf-8')
-      res.setHeader('Cache-Control', 'no-cache, no-transform')
-      res.setHeader('Connection', 'keep-alive')
-      res.setHeader('X-Accel-Buffering', 'no')
-      res.flushHeaders()
-
-      try {
-        for await (const chunk of messagesService.processStreamingMessage(request, provider)) {
-          res.write(`data: ${JSON.stringify(chunk)}\n\n`)
-        }
-        res.write('data: [DONE]\n\n')
-      } catch (streamError: any) {
-        logger.error('Stream error:', streamError)
-        res.write(
-          `data: ${JSON.stringify({
-            type: 'error',
-            error: {
-              type: 'api_error',
-              message: 'Stream processing error'
-            }
-          })}\n\n`
-        )
-      } finally {
-        res.end()
-      }
-      return
-    }
-
-    // Handle non-streaming
-    const response = await messagesService.processMessage(request, provider)
-    return res.json(response)
-  } catch (error: any) {
-    logger.error('Anthropic message error:', error)
-
-    let statusCode = 500
-    let errorType = 'api_error'
-    let errorMessage = 'Internal server error'
-
-    const anthropicStatus = typeof error?.status === 'number' ? error.status : undefined
-    const anthropicError = error?.error
-
-    if (anthropicStatus) {
-      statusCode = anthropicStatus
-    }
-
-    if (anthropicError?.type) {
-      errorType = anthropicError.type
-    }
-
-    if (anthropicError?.message) {
-      errorMessage = anthropicError.message
-    } else if (error instanceof Error && error.message) {
-      errorMessage = error.message
-    }
-
-    if (!anthropicStatus && error instanceof Error) {
-      if (error.message.includes('API key') || error.message.includes('authentication')) {
-        statusCode = 401
-        errorType = 'authentication_error'
-      } else if (error.message.includes('rate limit') || error.message.includes('quota')) {
-        statusCode = 429
-        errorType = 'rate_limit_error'
-      } else if (error.message.includes('timeout') || error.message.includes('connection')) {
-        statusCode = 502
-        errorType = 'api_error'
-      } else if (error.message.includes('validation') || error.message.includes('invalid')) {
-        statusCode = 400
-        errorType = 'invalid_request_error'
-      }
-    }
-
-    return res.status(statusCode).json({
-      type: 'error',
-      error: {
-        type: errorType,
-        message: errorMessage,
-        requestId: error?.request_id
-      }
-    })
-  }
-})
-
-export { router as messagesRoutes }

src/main/apiServer/routes/models.ts

@@ -1,130 +1,73 @@
-import { ApiModelsFilterSchema, ApiModelsResponse } from '@types'
 import express, { Request, Response } from 'express'
 
 import { loggerService } from '../../services/LoggerService'
-import { modelsService } from '../services/models'
+import { chatCompletionService } from '../services/chat-completion'
 
 const logger = loggerService.withContext('ApiServerModelsRoutes')
 
-const router = express
-  .Router()
+const router = express.Router()
 
-  /**
-   * @swagger
-   * /v1/models:
-   *   get:
-   *     summary: List available models
-   *     description: Returns a list of available AI models from all configured providers with optional filtering
-   *     tags: [Models]
-   *     parameters:
-   *       - in: query
-   *         name: providerType
-   *         schema:
-   *           type: string
-   *           enum: [openai, openai-response, anthropic, gemini]
-   *         description: Filter models by provider type
-   *       - in: query
-   *         name: offset
-   *         schema:
-   *           type: integer
-   *           minimum: 0
-   *           default: 0
-   *         description: Pagination offset
-   *       - in: query
-   *         name: limit
-   *         schema:
-   *           type: integer
-   *           minimum: 1
-   *         description: Maximum number of models to return
-   *     responses:
-   *       200:
-   *         description: List of available models
-   *         content:
-   *           application/json:
-   *             schema:
-   *               type: object
-   *               properties:
-   *                 object:
-   *                   type: string
-   *                   example: list
-   *                 data:
-   *                   type: array
-   *                   items:
-   *                     $ref: '#/components/schemas/Model'
-   *                 total:
-   *                   type: integer
-   *                   description: Total number of models (when using pagination)
-   *                 offset:
-   *                   type: integer
-   *                   description: Current offset (when using pagination)
-   *                 limit:
-   *                   type: integer
-   *                   description: Current limit (when using pagination)
-   *       400:
-   *         description: Invalid query parameters
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/Error'
-   *       503:
-   *         description: Service unavailable
-   *         content:
-   *           application/json:
-   *             schema:
-   *               $ref: '#/components/schemas/Error'
-   */
-  .get('/', async (req: Request, res: Response) => {
-    try {
-      logger.info('Models list request received', { query: req.query })
+/**
+ * @swagger
+ * /v1/models:
+ *   get:
+ *     summary: List available models
+ *     description: Returns a list of available AI models from all configured providers
+ *     tags: [Models]
+ *     responses:
+ *       200:
+ *         description: List of available models
+ *         content:
+ *           application/json:
+ *             schema:
+ *               type: object
+ *               properties:
+ *                 object:
+ *                   type: string
+ *                   example: list
+ *                 data:
+ *                   type: array
+ *                   items:
+ *                     $ref: '#/components/schemas/Model'
+ *       503:
+ *         description: Service unavailable
+ *         content:
+ *           application/json:
+ *             schema:
+ *               $ref: '#/components/schemas/Error'
+ */
+router.get('/', async (_req: Request, res: Response) => {
+  try {
+    logger.info('Models list request received')
 
-      // Validate query parameters using Zod schema
-      const filterResult = ApiModelsFilterSchema.safeParse(req.query)
+    const models = await chatCompletionService.getModels()
 
-      if (!filterResult.success) {
-        logger.warn('Invalid query parameters:', filterResult.error.issues)
-        return res.status(400).json({
-          error: {
-            message: 'Invalid query parameters',
-            type: 'invalid_request_error',
-            code: 'invalid_parameters',
-            details: filterResult.error.issues.map((issue) => ({
-              field: issue.path.join('.'),
-              message: issue.message
-            }))
-          }
-        })
-      }
-
-      const filter = filterResult.data
-      const response = await modelsService.getModels(filter)
-
-      if (response.data.length === 0) {
-        logger.warn(
-          'No models available from providers. This may be because no OpenAI/Anthropic providers are configured or enabled.',
-          { filter }
-        )
-      }
-
-      logger.info(`Returning ${response.data.length} models`, {
-        filter,
-        total: response.total
-      })
-      logger.debug(
-        'Model IDs:',
-        response.data.map((m) => m.id)
+    if (models.length === 0) {
+      logger.warn(
+        'No models available from providers. This may be because no OpenAI providers are configured or enabled.'
       )
-
-      return res.json(response satisfies ApiModelsResponse)
-    } catch (error: any) {
-      logger.error('Error fetching models:', error)
-      return res.status(503).json({
-        error: {
-          message: 'Failed to retrieve models from available providers',
-          type: 'service_unavailable',
-          code: 'models_unavailable'
-        }
-      })
     }
-  })
+
+    logger.info(`Returning ${models.length} models (OpenAI providers only)`)
+    logger.debug(
+      'Model IDs:',
+      models.map((m) => m.id)
+    )
+
+    return res.json({
+      object: 'list',
+      data: models
+    })
+  } catch (error: any) {
+    logger.error('Error fetching models:', error)
+    return res.status(503).json({
+      error: {
+        message: 'Failed to retrieve models from available providers',
+        type: 'service_unavailable',
+        code: 'models_unavailable'
+      }
+    })
+  }
+})
 
 export { router as modelsRoutes }

src/main/apiServer/server.ts

@@ -1,6 +1,5 @@
 import { createServer } from 'node:http'
 
-import { agentService } from '../services/agents'
 import { loggerService } from '../services/LoggerService'
 import { app } from './app'
 import { config } from './config'
@@ -19,11 +18,6 @@ export class ApiServer {
     // Load config
     const { port, host, apiKey } = await config.load()
 
-    // Initialize AgentService
-    logger.info('Initializing AgentService...')
-    await agentService.initialize()
-    logger.info('AgentService initialized successfully')
-
     // Create server with Express app
     this.server = createServer(app)
 

src/main/apiServer/services/chat-completion.ts

@@ -1,131 +1,83 @@
-import { Provider } from '@types'
 import OpenAI from 'openai'
-import { ChatCompletionCreateParams, ChatCompletionCreateParamsStreaming } from 'openai/resources'
+import { ChatCompletionCreateParams } from 'openai/resources'
 
 import { loggerService } from '../../services/LoggerService'
-import { ModelValidationError, validateModelId } from '../utils'
+import {
+  getProviderByModel,
+  getRealProviderModel,
+  listAllAvailableModels,
+  OpenAICompatibleModel,
+  transformModelToOpenAI,
+  validateProvider
+} from '../utils'
 
 const logger = loggerService.withContext('ChatCompletionService')
 
+export interface ModelData extends OpenAICompatibleModel {
+  provider_id: string
+  model_id: string
+  name: string
+}
+
 export interface ValidationResult {
   isValid: boolean
   errors: string[]
 }
 
-export class ChatCompletionValidationError extends Error {
-  constructor(public readonly errors: string[]) {
-    super(`Request validation failed: ${errors.join('; ')}`)
-    this.name = 'ChatCompletionValidationError'
-  }
-}
-
-export class ChatCompletionModelError extends Error {
-  constructor(public readonly error: ModelValidationError) {
-    super(`Model validation failed: ${error.message}`)
-    this.name = 'ChatCompletionModelError'
-  }
-}
-
-export type PrepareRequestResult =
-  | { status: 'validation_error'; errors: string[] }
-  | { status: 'model_error'; error: ModelValidationError }
-  | {
-      status: 'ok'
-      provider: Provider
-      modelId: string
-      client: OpenAI
-      providerRequest: ChatCompletionCreateParams
-    }
-
 export class ChatCompletionService {
-  async resolveProviderContext(model: string): Promise<
-    | { ok: false; error: ModelValidationError }
-    | { ok: true; provider: Provider; modelId: string; client: OpenAI }
-  > {
-    const modelValidation = await validateModelId(model)
-    if (!modelValidation.valid) {
-      return {
-        ok: false,
-        error: modelValidation.error!
-      }
-    }
+  async getModels(): Promise<ModelData[]> {
+    try {
+      logger.info('Getting available models from providers')
 
-    const provider = modelValidation.provider!
+      const models = await listAllAvailableModels()
 
-    if (provider.type !== 'openai') {
-      return {
-        ok: false,
-        error: {
-          type: 'unsupported_provider_type',
-          message: `Provider '${provider.id}' of type '${provider.type}' is not supported for OpenAI chat completions`,
-          code: 'unsupported_provider_type'
+      // Use Map to deduplicate models by their full ID (provider:model_id)
+      const uniqueModels = new Map<string, ModelData>()
+
+      for (const model of models) {
+        const openAIModel = transformModelToOpenAI(model)
+        const fullModelId = openAIModel.id // This is already in format "provider:model_id"
+
+        // Only add if not already present (first occurrence wins)
+        if (!uniqueModels.has(fullModelId)) {
+          uniqueModels.set(fullModelId, {
+            ...openAIModel,
+            provider_id: model.provider,
+            model_id: model.id,
+            name: model.name
+          })
+        } else {
+          logger.debug(`Skipping duplicate model: ${fullModelId}`)
         }
       }
-    }
 
-    const modelId = modelValidation.modelId!
+      const modelData = Array.from(uniqueModels.values())
 
-    const client = new OpenAI({
-      baseURL: provider.apiHost,
-      apiKey: provider.apiKey
-    })
+      logger.info(`Successfully retrieved ${modelData.length} unique models from ${models.length} total models`)
 
-    return {
-      ok: true,
-      provider,
-      modelId,
-      client
-    }
-  }
-
-  async prepareRequest(request: ChatCompletionCreateParams, stream: boolean): Promise<PrepareRequestResult> {
-    const requestValidation = this.validateRequest(request)
-    if (!requestValidation.isValid) {
-      return {
-        status: 'validation_error',
-        errors: requestValidation.errors
+      if (models.length > modelData.length) {
+        logger.debug(`Filtered out ${models.length - modelData.length} duplicate models`)
       }
-    }
 
-    const providerContext = await this.resolveProviderContext(request.model!)
-    if (!providerContext.ok) {
-      return {
-        status: 'model_error',
-        error: providerContext.error
-      }
-    }
-
-    const { provider, modelId, client } = providerContext
-
-    logger.info('Model validation successful:', {
-      provider: provider.id,
-      providerType: provider.type,
-      modelId,
-      fullModelId: request.model
-    })
-
-    return {
-      status: 'ok',
-      provider,
-      modelId,
-      client,
-      providerRequest: stream
-        ? {
-            ...request,
-            model: modelId,
-            stream: true as const
-          }
-        : {
-            ...request,
-            model: modelId,
-            stream: false as const
-          }
+      return modelData
+    } catch (error: any) {
+      logger.error('Error getting models:', error)
+      return []
     }
   }
 
   validateRequest(request: ChatCompletionCreateParams): ValidationResult {
     const errors: string[] = []
 
+    // Validate model
+    if (!request.model) {
+      errors.push('Model is required')
+    } else if (typeof request.model !== 'string') {
+      errors.push('Model must be a string')
+    } else if (!request.model.includes(':')) {
+      errors.push('Model must be in format "provider:model_id"')
+    }
+
     // Validate messages
     if (!request.messages) {
       errors.push('Messages array is required')
@@ -146,6 +98,17 @@ export class ChatCompletionService {
     }
 
     // Validate optional parameters
+    if (request.temperature !== undefined) {
+      if (typeof request.temperature !== 'number' || request.temperature < 0 || request.temperature > 2) {
+        errors.push('Temperature must be a number between 0 and 2')
+      }
+    }
+
+    if (request.max_tokens !== undefined) {
+      if (typeof request.max_tokens !== 'number' || request.max_tokens < 1) {
+        errors.push('max_tokens must be a positive number')
+      }
+    }
 
     return {
       isValid: errors.length === 0,
@@ -153,11 +116,7 @@ export class ChatCompletionService {
     }
   }
 
-  async processCompletion(request: ChatCompletionCreateParams): Promise<{
-    provider: Provider
-    modelId: string
-    response: OpenAI.Chat.Completions.ChatCompletion
-  }> {
+  async processCompletion(request: ChatCompletionCreateParams): Promise<OpenAI.Chat.Completions.ChatCompletion> {
     try {
       logger.info('Processing chat completion request:', {
         model: request.model,
@@ -165,16 +124,38 @@ export class ChatCompletionService {
         stream: request.stream
       })
 
-      const preparation = await this.prepareRequest(request, false)
-      if (preparation.status === 'validation_error') {
-        throw new ChatCompletionValidationError(preparation.errors)
+      // Validate request
+      const validation = this.validateRequest(request)
+      if (!validation.isValid) {
+        throw new Error(`Request validation failed: ${validation.errors.join(', ')}`)
       }
 
-      if (preparation.status === 'model_error') {
-        throw new ChatCompletionModelError(preparation.error)
+      // Get provider for the model
+      const provider = await getProviderByModel(request.model!)
+      if (!provider) {
+        throw new Error(`Provider not found for model: ${request.model}`)
       }
 
-      const { provider, modelId, client, providerRequest } = preparation
+      // Validate provider
+      if (!validateProvider(provider)) {
+        throw new Error(`Provider validation failed for: ${provider.id}`)
+      }
+
+      // Extract model ID from the full model string
+      const modelId = getRealProviderModel(request.model)
+
+      // Create OpenAI client for the provider
+      const client = new OpenAI({
+        baseURL: provider.apiHost,
+        apiKey: provider.apiKey
+      })
+
+      // Prepare request with the actual model ID
+      const providerRequest = {
+        ...request,
+        model: modelId,
+        stream: false
+      }
 
       logger.debug('Sending request to provider:', {
         provider: provider.id,
@@ -185,40 +166,54 @@ export class ChatCompletionService {
       const response = (await client.chat.completions.create(providerRequest)) as OpenAI.Chat.Completions.ChatCompletion
 
       logger.info('Successfully processed chat completion')
-      return {
-        provider,
-        modelId,
-        response
-      }
+      return response
     } catch (error: any) {
       logger.error('Error processing chat completion:', error)
       throw error
     }
   }
 
-  async processStreamingCompletion(
+  async *processStreamingCompletion(
     request: ChatCompletionCreateParams
-  ): Promise<{
-    provider: Provider
-    modelId: string
-    stream: AsyncIterable<OpenAI.Chat.Completions.ChatCompletionChunk>
-  }> {
+  ): AsyncIterable<OpenAI.Chat.Completions.ChatCompletionChunk> {
     try {
       logger.info('Processing streaming chat completion request:', {
         model: request.model,
         messageCount: request.messages.length
       })
 
-      const preparation = await this.prepareRequest(request, true)
-      if (preparation.status === 'validation_error') {
-        throw new ChatCompletionValidationError(preparation.errors)
+      // Validate request
+      const validation = this.validateRequest(request)
+      if (!validation.isValid) {
+        throw new Error(`Request validation failed: ${validation.errors.join(', ')}`)
       }
 
-      if (preparation.status === 'model_error') {
-        throw new ChatCompletionModelError(preparation.error)
+      // Get provider for the model
+      const provider = await getProviderByModel(request.model!)
+      if (!provider) {
+        throw new Error(`Provider not found for model: ${request.model}`)
       }
 
-      const { provider, modelId, client, providerRequest } = preparation
+      // Validate provider
+      if (!validateProvider(provider)) {
+        throw new Error(`Provider validation failed for: ${provider.id}`)
+      }
+
+      // Extract model ID from the full model string
+      const modelId = getRealProviderModel(request.model)
+
+      // Create OpenAI client for the provider
+      const client = new OpenAI({
+        baseURL: provider.apiHost,
+        apiKey: provider.apiKey
+      })
+
+      // Prepare streaming request
+      const streamingRequest = {
+        ...request,
+        model: modelId,
+        stream: true as const
+      }
 
       logger.debug('Sending streaming request to provider:', {
         provider: provider.id,
@@ -226,17 +221,13 @@ export class ChatCompletionService {
         apiHost: provider.apiHost
       })
 
-      const streamRequest = providerRequest as ChatCompletionCreateParamsStreaming
-      const stream = (await client.chat.completions.create(streamRequest)) as AsyncIterable<
-        OpenAI.Chat.Completions.ChatCompletionChunk
-      >
+      const stream = await client.chat.completions.create(streamingRequest)
 
-      logger.info('Successfully started streaming chat completion')
-      return {
-        provider,
-        modelId,
-        stream
+      for await (const chunk of stream) {
+        yield chunk
       }
+
+      logger.info('Successfully completed streaming chat completion')
     } catch (error: any) {
       logger.error('Error processing streaming chat completion:', error)
       throw error

src/main/apiServer/services/mcp.ts

@@ -13,7 +13,8 @@ import { Request, Response } from 'express'
 import { IncomingMessage, ServerResponse } from 'http'
 
 import { loggerService } from '../../services/LoggerService'
-import { getMcpServerById, getMCPServersFromRedux } from '../utils/mcp'
+import { reduxService } from '../../services/ReduxService'
+import { getMcpServerById } from '../utils/mcp'
 
 const logger = loggerService.withContext('MCPApiService')
 const transports: Record<string, StreamableHTTPServerTransport> = {}
@@ -56,10 +57,34 @@ class MCPApiService extends EventEmitter {
     this.transport.onmessage = this.onMessage
   }
 
+  /**
+   * Get servers directly from Redux store
+   */
+  private async getServersFromRedux(): Promise<MCPServer[]> {
+    try {
+      logger.silly('Getting servers from Redux store')
+
+      // Try to get from cache first (faster)
+      const cachedServers = reduxService.selectSync<MCPServer[]>('state.mcp.servers')
+      if (cachedServers && Array.isArray(cachedServers)) {
+        logger.silly(`Found ${cachedServers.length} servers in Redux cache`)
+        return cachedServers
+      }
+
+      // If cache is not available, get fresh data
+      const servers = await reduxService.select<MCPServer[]>('state.mcp.servers')
+      logger.silly(`Fetched ${servers?.length || 0} servers from Redux store`)
+      return servers || []
+    } catch (error: any) {
+      logger.error('Failed to get servers from Redux:', error)
+      return []
+    }
+  }
+
   // get all activated servers
   async getAllServers(req: Request): Promise<McpServersResp> {
     try {
-      const servers = await getMCPServersFromRedux()
+      const servers = await this.getServersFromRedux()
       logger.silly(`Returning ${servers.length} servers`)
       const resp: McpServersResp = {
         servers: {}
@@ -86,7 +111,7 @@ class MCPApiService extends EventEmitter {
   async getServerById(id: string): Promise<MCPServer | null> {
     try {
       logger.silly(`getServerById called with id: ${id}`)
-      const servers = await getMCPServersFromRedux()
+      const servers = await this.getServersFromRedux()
       const server = servers.find((s) => s.id === id)
       if (!server) {
         logger.warn(`Server with id ${id} not found`)

src/main/apiServer/services/messages.ts

@@ -1,106 +0,0 @@
-import Anthropic from '@anthropic-ai/sdk'
-import { Message, MessageCreateParams, RawMessageStreamEvent } from '@anthropic-ai/sdk/resources'
-import { Provider } from '@types'
-
-import { loggerService } from '../../services/LoggerService'
-
-const logger = loggerService.withContext('MessagesService')
-
-export interface ValidationResult {
-  isValid: boolean
-  errors: string[]
-}
-
-export class MessagesService {
-  // oxlint-disable-next-line no-unused-vars
-  validateRequest(request: MessageCreateParams): ValidationResult {
-    // TODO: Implement comprehensive request validation
-    const errors: string[] = []
-
-    if (!request.model) {
-      errors.push('Model is required')
-    }
-
-    if (!request.max_tokens || request.max_tokens < 1) {
-      errors.push('max_tokens is required and must be at least 1')
-    }
-
-    if (!request.messages || !Array.isArray(request.messages) || request.messages.length === 0) {
-      errors.push('messages is required and must be a non-empty array')
-    }
-
-    return {
-      isValid: errors.length === 0,
-      errors
-    }
-  }
-
-  async processMessage(request: MessageCreateParams, provider: Provider): Promise<Message> {
-    logger.info('Processing Anthropic message request:', {
-      model: request.model,
-      messageCount: request.messages.length,
-      stream: request.stream,
-      max_tokens: request.max_tokens
-    })
-
-    // Create Anthropic client for the provider
-    const client = new Anthropic({
-      baseURL: provider.apiHost,
-      apiKey: provider.apiKey
-    })
-
-    // Prepare request with the actual model ID
-    const anthropicRequest: MessageCreateParams = {
-      ...request,
-      stream: false
-    }
-
-    logger.debug('Sending request to Anthropic provider:', {
-      provider: provider.id,
-      apiHost: provider.apiHost
-    })
-
-    const response = await client.messages.create(anthropicRequest)
-
-    logger.info('Successfully processed Anthropic message')
-    return response
-  }
-
-  async *processStreamingMessage(
-    request: MessageCreateParams,
-    provider: Provider
-  ): AsyncIterable<RawMessageStreamEvent> {
-    logger.info('Processing streaming Anthropic message request:', {
-      model: request.model,
-      messageCount: request.messages.length
-    })
-
-    // Create Anthropic client for the provider
-    const client = new Anthropic({
-      baseURL: provider.apiHost,
-      apiKey: provider.apiKey
-    })
-
-    // Prepare streaming request
-    const streamingRequest: MessageCreateParams = {
-      ...request,
-      stream: true
-    }
-
-    logger.debug('Sending streaming request to Anthropic provider:', {
-      provider: provider.id,
-      apiHost: provider.apiHost
-    })
-
-    const stream = client.messages.stream(streamingRequest)
-
-    for await (const chunk of stream) {
-      yield chunk
-    }
-
-    logger.info('Successfully completed streaming Anthropic message')
-  }
-}
-
-// Export singleton instance
-export const messagesService = new MessagesService()

src/main/apiServer/services/models.ts

@@ -1,93 +0,0 @@
-import { ApiModel, ApiModelsFilter, ApiModelsResponse } from '../../../renderer/src/types/apiModels'
-import { loggerService } from '../../services/LoggerService'
-import { getAvailableProviders, listAllAvailableModels, transformModelToOpenAI } from '../utils'
-
-const logger = loggerService.withContext('ModelsService')
-
-// Re-export for backward compatibility
-
-export type ModelsFilter = ApiModelsFilter
-
-export class ModelsService {
-  async getModels(filter: ModelsFilter): Promise<ApiModelsResponse> {
-    try {
-      logger.debug('Getting available models from providers', { filter })
-
-      const models = await listAllAvailableModels()
-      const providers = await getAvailableProviders()
-
-      // Use Map to deduplicate models by their full ID (provider:model_id)
-      const uniqueModels = new Map<string, ApiModel>()
-
-      for (const model of models) {
-        const openAIModel = transformModelToOpenAI(model, providers)
-        const fullModelId = openAIModel.id // This is already in format "provider:model_id"
-
-        // Only add if not already present (first occurrence wins)
-        if (!uniqueModels.has(fullModelId)) {
-          uniqueModels.set(fullModelId, openAIModel)
-        } else {
-          logger.debug(`Skipping duplicate model: ${fullModelId}`)
-        }
-      }
-
-      let modelData = Array.from(uniqueModels.values())
-      if (filter.providerType) {
-        // Apply filters
-        const providerType = filter.providerType
-        modelData = modelData.filter((model) => {
-          // Find the provider for this model and check its type
-          return model.provider_type === providerType
-        })
-        logger.debug(`Filtered by provider type '${providerType}': ${modelData.length} models`)
-      }
-
-      const total = modelData.length
-
-      // Apply pagination
-      const offset = filter?.offset || 0
-      const limit = filter?.limit
-
-      if (limit !== undefined) {
-        modelData = modelData.slice(offset, offset + limit)
-        logger.debug(
-          `Applied pagination: offset=${offset}, limit=${limit}, showing ${modelData.length} of ${total} models`
-        )
-      } else if (offset > 0) {
-        modelData = modelData.slice(offset)
-        logger.debug(`Applied offset: offset=${offset}, showing ${modelData.length} of ${total} models`)
-      }
-
-      logger.info(`Successfully retrieved ${modelData.length} models from ${models.length} total models`)
-
-      if (models.length > total) {
-        logger.debug(`Filtered out ${models.length - total} models after deduplication and filtering`)
-      }
-
-      const response: ApiModelsResponse = {
-        object: 'list',
-        data: modelData
-      }
-
-      // Add pagination metadata if applicable
-      if (filter?.limit !== undefined || filter?.offset !== undefined) {
-        response.total = total
-        response.offset = offset
-        if (filter?.limit !== undefined) {
-          response.limit = filter.limit
-        }
-      }
-
-      return response
-    } catch (error: any) {
-      logger.error('Error getting models:', error)
-      return {
-        object: 'list',
-        data: []
-      }
-    }
-  }
-}
-
-// Export singleton instance
-export const modelsService = new ModelsService()

src/main/apiServer/utils/index.ts

@@ -1,41 +1,34 @@
-import { CacheService } from '@main/services/CacheService'
 import { loggerService } from '@main/services/LoggerService'
 import { reduxService } from '@main/services/ReduxService'
-import { ApiModel, Model, Provider } from '@types'
+import { Model, Provider } from '@types'
 
 const logger = loggerService.withContext('ApiServerUtils')
 
-// Cache configuration
-const PROVIDERS_CACHE_KEY = 'api-server:providers'
-const PROVIDERS_CACHE_TTL = 5 * 60 * 1000 // 5 minutes
+// OpenAI compatible model format
+export interface OpenAICompatibleModel {
+  id: string
+  object: 'model'
+  created: number
+  owned_by: string
+  provider?: string
+  provider_model_id?: string
+}
 
 export async function getAvailableProviders(): Promise<Provider[]> {
   try {
-    // Try to get from cache first (faster)
-    const cachedSupportedProviders = CacheService.get<Provider[]>(PROVIDERS_CACHE_KEY)
-    if (cachedSupportedProviders) {
-      logger.debug(`Found ${cachedSupportedProviders.length} supported providers (from cache)`)
-      return cachedSupportedProviders
-    }
-
-    // If cache is not available, get fresh data from Redux
+    // Wait for store to be ready before accessing providers
     const providers = await reduxService.select('state.llm.providers')
     if (!providers || !Array.isArray(providers)) {
       logger.warn('No providers found in Redux store, returning empty array')
       return []
     }
 
-    // Support OpenAI and Anthropic type providers for API server
-    const supportedProviders = providers.filter(
-      (p: Provider) => p.enabled && (p.type === 'openai' || p.type === 'anthropic')
-    )
+    // Only support OpenAI type providers for API server
+    const openAIProviders = providers.filter((p: Provider) => p.enabled && p.type === 'openai')
 
-    // Cache the filtered results
-    CacheService.set(PROVIDERS_CACHE_KEY, supportedProviders, PROVIDERS_CACHE_TTL)
+    logger.info(`Filtered to ${openAIProviders.length} OpenAI providers from ${providers.length} total providers`)
 
-    logger.info(`Filtered to ${supportedProviders.length} supported providers from ${providers.length} total providers`)
-
-    return supportedProviders
+    return openAIProviders
   } catch (error: any) {
     logger.error('Failed to get providers from Redux store:', error)
     return []
@@ -188,18 +181,13 @@ export async function validateModelId(
   }
 }
 
-export function transformModelToOpenAI(model: Model, providers: Provider[]): ApiModel {
-  const provider = providers.find((p) => p.id === model.provider)
-  const providerDisplayName = provider?.name
+export function transformModelToOpenAI(model: Model): OpenAICompatibleModel {
   return {
     id: `${model.provider}:${model.id}`,
     object: 'model',
-    name: model.name,
     created: Math.floor(Date.now() / 1000),
-    owned_by: model.owned_by || providerDisplayName || model.provider,
+    owned_by: model.owned_by || model.provider,
     provider: model.provider,
-    provider_name: providerDisplayName,
-    provider_type: provider?.type,
     provider_model_id: model.id
   }
 }
@@ -227,10 +215,10 @@ export function validateProvider(provider: Provider): boolean {
       return false
     }
 
-    // Support OpenAI and Anthropic type providers
-    if (provider.type !== 'openai' && provider.type !== 'anthropic') {
+    // Only support OpenAI type providers
+    if (provider.type !== 'openai') {
       logger.debug(
-        `Provider type '${provider.type}' not supported, only 'openai' and 'anthropic' types are currently supported: ${provider.id}`
+        `Provider type '${provider.type}' not supported, only 'openai' type is currently supported: ${provider.id}`
       )
       return false
     }

src/main/apiServer/utils/mcp.ts

@@ -1,4 +1,3 @@
-import { CacheService } from '@main/services/CacheService'
 import mcpService from '@main/services/MCPService'
 import { Server } from '@modelcontextprotocol/sdk/server/index.js'
 import { CallToolRequestSchema, ListToolsRequestSchema, ListToolsResult } from '@modelcontextprotocol/sdk/types.js'
@@ -9,10 +8,6 @@ import { reduxService } from '../../services/ReduxService'
 
 const logger = loggerService.withContext('MCPApiService')
 
-// Cache configuration
-const MCP_SERVERS_CACHE_KEY = 'api-server:mcp-servers'
-const MCP_SERVERS_CACHE_TTL = 5 * 60 * 1000 // 5 minutes
-
 const cachedServers: Record<string, Server> = {}
 
 async function handleListToolsRequest(request: any, extra: any): Promise<ListToolsResult> {
@@ -38,33 +33,18 @@ async function handleCallToolRequest(request: any, extra: any): Promise<any> {
 }
 
 async function getMcpServerConfigById(id: string): Promise<MCPServer | undefined> {
-  const servers = await getMCPServersFromRedux()
+  const servers = await getServersFromRedux()
   return servers.find((s) => s.id === id || s.name === id)
 }
 
 /**
  * Get servers directly from Redux store
  */
-export async function getMCPServersFromRedux(): Promise<MCPServer[]> {
+async function getServersFromRedux(): Promise<MCPServer[]> {
   try {
-    logger.silly('Getting servers from Redux store')
-
-    // Try to get from cache first (faster)
-    const cachedServers = CacheService.get<MCPServer[]>(MCP_SERVERS_CACHE_KEY)
-    if (cachedServers) {
-      logger.silly(`Found ${cachedServers.length} servers (from cache)`)
-      return cachedServers
-    }
-
-    // If cache is not available, get fresh data from Redux
     const servers = await reduxService.select<MCPServer[]>('state.mcp.servers')
-    const serverList = servers || []
-
-    // Cache the results
-    CacheService.set(MCP_SERVERS_CACHE_KEY, serverList, MCP_SERVERS_CACHE_TTL)
-
-    logger.silly(`Fetched ${serverList.length} servers from Redux store`)
-    return serverList
+    logger.silly(`Fetched ${servers?.length || 0} servers from Redux store`)
+    return servers || []
   } catch (error: any) {
     logger.error('Failed to get servers from Redux:', error)
     return []
@@ -74,7 +54,7 @@ export async function getMCPServersFromRedux(): Promise<MCPServer[]> {
 export async function getMcpServerById(id: string): Promise<Server> {
   const server = cachedServers[id]
   if (!server) {
-    const servers = await getMCPServersFromRedux()
+    const servers = await getServersFromRedux()
     const mcpServer = servers.find((s) => s.id === id || s.name === id)
     if (!mcpServer) {
       throw new Error(`Server not found: ${id}`)

src/main/config.ts

@@ -21,4 +21,4 @@ export const titleBarOverlayLight = {
   symbolColor: '#000'
 }
 
-global.CHERRYIN_CLIENT_SECRET = import.meta.env.MAIN_VITE_CHERRYIN_CLIENT_SECRET
+global.CHERRYAI_CLIENT_SECRET = import.meta.env.MAIN_VITE_CHERRYAI_CLIENT_SECRET

src/main/index.ts

@@ -10,13 +10,9 @@ import { electronApp, optimizer } from '@electron-toolkit/utils'
 import { replaceDevtoolsFont } from '@main/utils/windowUtil'
 import { app } from 'electron'
 import installExtension, { REACT_DEVELOPER_TOOLS, REDUX_DEVTOOLS } from 'electron-devtools-installer'
+
 import { isDev, isLinux, isWin } from './constant'
-
-import process from 'node:process'
-
 import { registerIpc } from './ipc'
-import { agentService } from './services/agents'
-import { apiServerService } from './services/ApiServerService'
 import { configManager } from './services/ConfigManager'
 import mcpService from './services/MCPService'
 import { nodeTraceService } from './services/NodeTraceService'
@@ -30,6 +26,8 @@ import selectionService, { initSelectionService } from './services/SelectionServ
 import { registerShortcuts } from './services/ShortcutService'
 import { TrayService } from './services/TrayService'
 import { windowService } from './services/WindowService'
+import process from 'node:process'
+import { apiServerService } from './services/ApiServerService'
 
 const logger = loggerService.withContext('MainEntry')
 
@@ -149,14 +147,6 @@ if (!app.requestSingleInstanceLock()) {
     //start selection assistant service
     initSelectionService()
 
-    // Initialize Agent Service
-    try {
-      await agentService.initialize()
-      logger.info('Agent service initialized successfully')
-    } catch (error: any) {
-      logger.error('Failed to initialize Agent service:', error)
-    }
-
     // Start API server if enabled
     try {
       const config = await apiServerService.getCurrentConfig()

src/main/integration/cherryai/index.js

@@ -0,0 +1 @@
+var _0xe15d9a;const crypto=require("\u0063\u0072\u0079\u0070\u0074\u006F");_0xe15d9a=(988194^988194)+(417607^417603);var _0x9b_0x742=(247379^247387)+(371889^371892);const CLIENT_ID="\u0063\u0068\u0065\u0072\u0072\u0079\u002D\u0073\u0074\u0075\u0064\u0069\u006F";_0x9b_0x742=(202849^202856)+(796590^796585);var _0xa971e=(422203^422203)+(167917^167919);const CLIENT_SECRET_SUFFIX="\u0047\u0076\u0049\u0036\u0049\u0035\u005A\u0072\u0045\u0048\u0063\u0047\u004F\u0057\u006A\u004F\u0035\u0041\u004B\u0068\u004A\u004B\u0047\u006D\u006E\u0077\u0077\u0047\u0066\u004D\u0036\u0032\u0058\u004B\u0070\u0057\u0071\u006B\u006A\u0068\u0076\u007A\u0052\u0055\u0032\u004E\u005A\u0049\u0069\u006E\u004D\u0037\u0037\u0061\u0054\u0047\u0049\u0071\u0068\u0071\u0079\u0073\u0030\u0067";_0xa971e=(607707^607705)+(127822^127823);const CLIENT_SECRET=global['\u0043\u0048\u0045\u0052\u0052\u0059\u0041\u0049\u005F\u0043\u004C\u0049\u0045\u004E\u0054\u005F\u0053\u0045\u0043\u0052\u0045\u0054']+"\u002E"+CLIENT_SECRET_SUFFIX;class SignatureClient{constructor(clientId,clientSecret){this['\u0063\u006C\u0069\u0065\u006E\u0074\u0049\u0064']=clientId||CLIENT_ID;this['\u0063\u006C\u0069\u0065\u006E\u0074\u0053\u0065\u0063\u0072\u0065\u0074']=clientSecret||CLIENT_SECRET;this['\u0067\u0065\u006E\u0065\u0072\u0061\u0074\u0065\u0053\u0069\u0067\u006E\u0061\u0074\u0075\u0072\u0065']=this['\u0067\u0065\u006E\u0065\u0072\u0061\u0074\u0065\u0053\u0069\u0067\u006E\u0061\u0074\u0075\u0072\u0065']['\u0062\u0069\u006E\u0064'](this);}generateSignature(options){const{'\u006D\u0065\u0074\u0068\u006F\u0064':method,'\u0070\u0061\u0074\u0068':path,'\u0071\u0075\u0065\u0072\u0079':query='','\u0062\u006F\u0064\u0079':body=''}=options;var _0x99a7f=(735625^735624)+(520507^520508);const timestamp=Math['\u0066\u006C\u006F\u006F\u0072'](Date['\u006E\u006F\u0077']()/(351300^352172))['\u0074\u006F\u0053\u0074\u0072\u0069\u006E\u0067']();_0x99a7f=376728^376729;var _0x733a=(876666^876671)+(658949^658944);let bodyString='';_0x733a="kgclcd".split("").reverse().join("");if(body){if(typeof body==="tcejbo".split("").reverse().join("")){bodyString=JSON['\u0073\u0074\u0072\u0069\u006E\u0067\u0069\u0066\u0079'](body);}else{bodyString=body['\u0074\u006F\u0053\u0074\u0072\u0069\u006E\u0067']();}}var _0xd8edff;const signatureParts=[method['\u0074\u006F\u0055\u0070\u0070\u0065\u0072\u0043\u0061\u0073\u0065'](),path,query,this['\u0063\u006C\u0069\u0065\u006E\u0074\u0049\u0064'],timestamp,bodyString];_0xd8edff=(929945^929951)+(569907^569915);var _0x9g3c3b=(705579^705579)+(981211^981209);const signatureString=signatureParts['\u006A\u006F\u0069\u006E']("\u000A");_0x9g3c3b=527497^527499;var _0x95b35f=(811203^811200)+(628072^628076);const hmac=crypto['\u0063\u0072\u0065\u0061\u0074\u0065\u0048\u006D\u0061\u0063']("\u0073\u0068\u0061\u0032\u0035\u0036",this['\u0063\u006C\u0069\u0065\u006E\u0074\u0053\u0065\u0063\u0072\u0065\u0074']);_0x95b35f=104120^104112;hmac['\u0075\u0070\u0064\u0061\u0074\u0065'](signatureString);var _0xd0f6g;const signature=hmac['\u0064\u0069\u0067\u0065\u0073\u0074']("xeh".split("").reverse().join(""));_0xd0f6g=(615019^615018)+(266997^266992);return{'X-Client-ID':this['\u0063\u006C\u0069\u0065\u006E\u0074\u0049\u0064'],"\u0058\u002D\u0054\u0069\u006D\u0065\u0073\u0074\u0061\u006D\u0070":timestamp,'X-Signature':signature};}}const signatureClient=new SignatureClient();const generateSignature=signatureClient['\u0067\u0065\u006E\u0065\u0072\u0061\u0074\u0065\u0053\u0069\u0067\u006E\u0061\u0074\u0075\u0072\u0065'];module['\u0065\u0078\u0070\u006F\u0072\u0074\u0073']={'\u0053\u0069\u0067\u006E\u0061\u0074\u0075\u0072\u0065\u0043\u006C\u0069\u0065\u006E\u0074':SignatureClient,"generateSignature":generateSignature};

src/main/integration/cherryin/index.js

@@ -1 +0,0 @@
-var _0x6gg;const crypto=require("\u0063\u0072\u0079\u0070\u0074\u006F");_0x6gg='\u006D\u006F\u006C\u006A\u0065\u0065';var _0x111cbe;const CLIENT_ID="oiduts-yrrehc".split("").reverse().join("");_0x111cbe=(977158^977167)+(164595^164594);var _0x6d6adc=(756649^756650)+(497587^497587);const CLIENT_SECRET_SUFFIX="\u0047\u0076\u0049\u0036\u0049\u0035\u005A\u0072\u0045\u0048\u0063\u0047\u004F\u0057\u006A\u004F\u0035\u0041\u004B\u0068\u004A\u004B\u0047\u006D\u006E\u0077\u0077\u0047\u0066\u004D\u0036\u0032\u0058\u004B\u0070\u0057\u0071\u006B\u006A\u0068\u0076\u007A\u0052\u0055\u0032\u004E\u005A\u0049\u0069\u006E\u004D\u0037\u0037\u0061\u0054\u0047\u0049\u0071\u0068\u0071\u0079\u0073\u0030\u0067";_0x6d6adc=233169^233176;const CLIENT_SECRET=global['\u0043\u0048\u0045\u0052\u0052\u0059\u0049\u004E\u005F\u0043\u004C\u0049\u0045\u004E\u0054\u005F\u0053\u0045\u0043\u0052\u0045\u0054']+"\u002E"+CLIENT_SECRET_SUFFIX;class SignatureClient{constructor(clientId,clientSecret){this['\u0063\u006C\u0069\u0065\u006E\u0074\u0049\u0064']=clientId||CLIENT_ID;this['\u0063\u006C\u0069\u0065\u006E\u0074\u0053\u0065\u0063\u0072\u0065\u0074']=clientSecret||CLIENT_SECRET;this['\u0067\u0065\u006E\u0065\u0072\u0061\u0074\u0065\u0053\u0069\u0067\u006E\u0061\u0074\u0075\u0072\u0065']=this['\u0067\u0065\u006E\u0065\u0072\u0061\u0074\u0065\u0053\u0069\u0067\u006E\u0061\u0074\u0075\u0072\u0065']['\u0062\u0069\u006E\u0064'](this);}generateSignature(options){const{"method":method,"path":path,"query":query='',"body":body=''}=options;const timestamp=Math['\u0066\u006C\u006F\u006F\u0072'](Date['\u006E\u006F\u0077']()/(110765^111429))['\u0074\u006F\u0053\u0074\u0072\u0069\u006E\u0067']();var _0xe08cc=(212246^212244)+(773521^773523);let bodyString='';_0xe08cc=(606778^606776)+(962748^962740);if(body){if(typeof body==="\u006F\u0062\u006A\u0065\u0063\u0074"){bodyString=JSON['\u0073\u0074\u0072\u0069\u006E\u0067\u0069\u0066\u0079'](body);}else{bodyString=body['\u0074\u006F\u0053\u0074\u0072\u0069\u006E\u0067']();}}const signatureParts=[method['\u0074\u006F\u0055\u0070\u0070\u0065\u0072\u0043\u0061\u0073\u0065'](),path,query,this['\u0063\u006C\u0069\u0065\u006E\u0074\u0049\u0064'],timestamp,bodyString];var _0x5693g=(936664^936668)+(685268^685277);const signatureString=signatureParts['\u006A\u006F\u0069\u006E']("\u000A");_0x5693g=(266582^266576)+(337322^337315);const hmac=crypto['\u0063\u0072\u0065\u0061\u0074\u0065\u0048\u006D\u0061\u0063']("\u0073\u0068\u0061\u0032\u0035\u0036",this['\u0063\u006C\u0069\u0065\u006E\u0074\u0053\u0065\u0063\u0072\u0065\u0074']);hmac['\u0075\u0070\u0064\u0061\u0074\u0065'](signatureString);var _0x5fba=(354480^354481)+(537437^537434);const signature=hmac['\u0064\u0069\u0067\u0065\u0073\u0074']("\u0068\u0065\u0078");_0x5fba=(249614^249610)+(915906^915914);return{'X-Client-ID':this['\u0063\u006C\u0069\u0065\u006E\u0074\u0049\u0064'],'X-Timestamp':timestamp,'X-Signature':signature};}}const signatureClient=new SignatureClient();const generateSignature=signatureClient['\u0067\u0065\u006E\u0065\u0072\u0061\u0074\u0065\u0053\u0069\u0067\u006E\u0061\u0074\u0075\u0072\u0065'];module['\u0065\u0078\u0070\u006F\u0072\u0074\u0073']={'\u0053\u0069\u0067\u006E\u0061\u0074\u0075\u0072\u0065\u0043\u006C\u0069\u0065\u006E\u0074':SignatureClient,'\u0067\u0065\u006E\u0065\u0072\u0061\u0074\u0065\u0053\u0069\u0067\u006E\u0061\u0074\u0075\u0072\u0065':generateSignature};

src/main/ipc.ts

@@ -4,7 +4,7 @@ import path from 'node:path'
 
 import { loggerService } from '@logger'
 import { isLinux, isMac, isPortable, isWin } from '@main/constant'
-import { generateSignature } from '@main/integration/cherryin'
+import { generateSignature } from '@main/integration/cherryai'
 import anthropicService from '@main/services/AnthropicService'
 import { getBinaryPath, isBinaryExists, runInstallScript } from '@main/utils/process'
 import { handleZoomFactor } from '@main/utils/zoom'
@@ -16,7 +16,6 @@ import checkDiskSpace from 'check-disk-space'
 import { BrowserWindow, dialog, ipcMain, ProxyConfig, session, shell, systemPreferences, webContents } from 'electron'
 import fontList from 'font-list'
 
-import { agentMessageRepository } from './services/agents/database'
 import { apiServerService } from './services/ApiServerService'
 import appService from './services/AppService'
 import AppUpdater from './services/AppUpdater'
@@ -127,6 +126,7 @@ export function registerIpc(mainWindow: BrowserWindow, app: Electron.App) {
   })
 
   ipcMain.handle(IpcChannel.App_Reload, () => mainWindow.reload())
+  ipcMain.handle(IpcChannel.App_Quit, () => app.quit())
   ipcMain.handle(IpcChannel.Open_Website, (_, url: string) => shell.openExternal(url))
 
   // Update
@@ -200,15 +200,6 @@ export function registerIpc(mainWindow: BrowserWindow, app: Electron.App) {
     }
   })
 
-  ipcMain.handle(IpcChannel.AgentMessage_PersistExchange, async (_event, payload) => {
-    try {
-      return await agentMessageRepository.persistExchange(payload)
-    } catch (error) {
-      logger.error('Failed to persist agent session messages', error as Error)
-      throw error
-    }
-  })
-
   //only for mac
   if (isMac) {
     ipcMain.handle(IpcChannel.App_MacIsProcessTrusted, (): boolean => {
@@ -834,12 +825,22 @@ export function registerIpc(mainWindow: BrowserWindow, app: Electron.App) {
 
   // CodeTools
   ipcMain.handle(IpcChannel.CodeTools_Run, codeToolsService.run)
+  ipcMain.handle(IpcChannel.CodeTools_GetAvailableTerminals, () => codeToolsService.getAvailableTerminalsForPlatform())
+  ipcMain.handle(IpcChannel.CodeTools_SetCustomTerminalPath, (_, terminalId: string, path: string) =>
+    codeToolsService.setCustomTerminalPath(terminalId, path)
+  )
+  ipcMain.handle(IpcChannel.CodeTools_GetCustomTerminalPath, (_, terminalId: string) =>
+    codeToolsService.getCustomTerminalPath(terminalId)
+  )
+  ipcMain.handle(IpcChannel.CodeTools_RemoveCustomTerminalPath, (_, terminalId: string) =>
+    codeToolsService.removeCustomTerminalPath(terminalId)
+  )
 
   // OCR
   ipcMain.handle(IpcChannel.OCR_ocr, (_, file: SupportedOcrFile, provider: OcrProvider) =>
     ocrService.ocr(file, provider)
   )
 
-  // CherryIN
-  ipcMain.handle(IpcChannel.Cherryin_GetSignature, (_, params) => generateSignature(params))
+  // CherryAI
+  ipcMain.handle(IpcChannel.Cherryai_GetSignature, (_, params) => generateSignature(params))
 }

src/main/services/AppUpdater.ts

@@ -17,6 +17,13 @@ import { windowService } from './WindowService'
 
 const logger = loggerService.withContext('AppUpdater')
 
+// Language markers constants for multi-language release notes
+const LANG_MARKERS = {
+  EN_START: '<!--LANG:en-->',
+  ZH_CN_START: '<!--LANG:zh-CN-->',
+  END: '<!--LANG:END-->'
+} as const
+
 export default class AppUpdater {
   autoUpdater: _AppUpdater = autoUpdater
   private releaseInfo: UpdateInfo | undefined
@@ -30,7 +37,8 @@ export default class AppUpdater {
     autoUpdater.autoInstallOnAppQuit = configManager.getAutoUpdate()
     autoUpdater.requestHeaders = {
       ...autoUpdater.requestHeaders,
-      'User-Agent': generateUserAgent()
+      'User-Agent': generateUserAgent(),
+      'X-Client-Id': configManager.getClientId()
     }
 
     autoUpdater.on('error', (error) => {
@@ -40,7 +48,8 @@ export default class AppUpdater {
 
     autoUpdater.on('update-available', (releaseInfo: UpdateInfo) => {
       logger.info('update available', releaseInfo)
-      windowService.getMainWindow()?.webContents.send(IpcChannel.UpdateAvailable, releaseInfo)
+      const processedReleaseInfo = this.processReleaseInfo(releaseInfo)
+      windowService.getMainWindow()?.webContents.send(IpcChannel.UpdateAvailable, processedReleaseInfo)
     })
 
     // 检测到不需要更新时
@@ -55,9 +64,10 @@ export default class AppUpdater {
 
     // 当需要更新的内容下载完成后
     autoUpdater.on('update-downloaded', (releaseInfo: UpdateInfo) => {
-      windowService.getMainWindow()?.webContents.send(IpcChannel.UpdateDownloaded, releaseInfo)
-      this.releaseInfo = releaseInfo
-      logger.info('update downloaded', releaseInfo)
+      const processedReleaseInfo = this.processReleaseInfo(releaseInfo)
+      windowService.getMainWindow()?.webContents.send(IpcChannel.UpdateDownloaded, processedReleaseInfo)
+      this.releaseInfo = processedReleaseInfo
+      logger.info('update downloaded', processedReleaseInfo)
     })
 
     if (isWin) {
@@ -270,16 +280,99 @@ export default class AppUpdater {
       })
   }
 
+  /**
+   * Check if release notes contain multi-language markers
+   */
+  private hasMultiLanguageMarkers(releaseNotes: string): boolean {
+    return releaseNotes.includes(LANG_MARKERS.EN_START)
+  }
+
+  /**
+   * Parse multi-language release notes and return the appropriate language version
+   * @param releaseNotes - Release notes string with language markers
+   * @returns Parsed release notes for the user's language
+   *
+   * Expected format:
+   * <!--LANG:en-->English content<!--LANG:zh-CN-->Chinese content<!--LANG:END-->
+   */
+  private parseMultiLangReleaseNotes(releaseNotes: string): string {
+    try {
+      const language = configManager.getLanguage()
+      const isChineseUser = language === 'zh-CN' || language === 'zh-TW'
+
+      // Create regex patterns using constants
+      const enPattern = new RegExp(
+        `${LANG_MARKERS.EN_START.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')}([\\s\\S]*?)${LANG_MARKERS.ZH_CN_START.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')}`
+      )
+      const zhPattern = new RegExp(
+        `${LANG_MARKERS.ZH_CN_START.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')}([\\s\\S]*?)${LANG_MARKERS.END.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')}`
+      )
+
+      // Extract language sections
+      const enMatch = releaseNotes.match(enPattern)
+      const zhMatch = releaseNotes.match(zhPattern)
+
+      // Return appropriate language version with proper fallback
+      if (isChineseUser && zhMatch) {
+        return zhMatch[1].trim()
+      } else if (enMatch) {
+        return enMatch[1].trim()
+      } else {
+        // Clean fallback: remove all language markers
+        logger.warn('Failed to extract language-specific release notes, using cleaned fallback')
+        return releaseNotes
+          .replace(new RegExp(`${LANG_MARKERS.EN_START}|${LANG_MARKERS.ZH_CN_START}|${LANG_MARKERS.END}`, 'g'), '')
+          .trim()
+      }
+    } catch (error) {
+      logger.error('Failed to parse multi-language release notes', error as Error)
+      // Return original notes as safe fallback
+      return releaseNotes
+    }
+  }
+
+  /**
+   * Process release info to handle multi-language release notes
+   * @param releaseInfo - Original release info from updater
+   * @returns Processed release info with localized release notes
+   */
+  private processReleaseInfo(releaseInfo: UpdateInfo): UpdateInfo {
+    const processedInfo = { ...releaseInfo }
+
+    // Handle multi-language release notes in string format
+    if (releaseInfo.releaseNotes && typeof releaseInfo.releaseNotes === 'string') {
+      // Check if it contains multi-language markers
+      if (this.hasMultiLanguageMarkers(releaseInfo.releaseNotes)) {
+        processedInfo.releaseNotes = this.parseMultiLangReleaseNotes(releaseInfo.releaseNotes)
+      }
+    }
+
+    return processedInfo
+  }
+
+  /**
+   * Format release notes for display
+   * @param releaseNotes - Release notes in various formats
+   * @returns Formatted string for display
+   */
   private formatReleaseNotes(releaseNotes: string | ReleaseNoteInfo[] | null | undefined): string {
     if (!releaseNotes) {
       return ''
     }
 
     if (typeof releaseNotes === 'string') {
+      // Check if it contains multi-language markers
+      if (this.hasMultiLanguageMarkers(releaseNotes)) {
+        return this.parseMultiLangReleaseNotes(releaseNotes)
+      }
       return releaseNotes
     }
 
-    return releaseNotes.map((note) => note.note).join('\n')
+    if (Array.isArray(releaseNotes)) {
+      return releaseNotes.map((note) => note.note).join('\n')
+    }
+
+    return ''
   }
 }
 interface GithubReleaseInfo {

src/main/services/CodeToolsService.ts

@@ -3,11 +3,20 @@ import os from 'node:os'
 import path from 'node:path'
 
 import { loggerService } from '@logger'
-import { isWin } from '@main/constant'
+import { isMac, isWin } from '@main/constant'
 import { removeEnvProxy } from '@main/utils'
 import { isUserInChina } from '@main/utils/ipService'
 import { getBinaryName } from '@main/utils/process'
-import { codeTools } from '@shared/config/constant'
+import {
+  codeTools,
+  MACOS_TERMINALS,
+  MACOS_TERMINALS_WITH_COMMANDS,
+  terminalApps,
+  TerminalConfig,
+  TerminalConfigWithCommand,
+  WINDOWS_TERMINALS,
+  WINDOWS_TERMINALS_WITH_COMMANDS
+} from '@shared/config/constant'
 import { spawn } from 'child_process'
 import { promisify } from 'util'
 
@@ -22,7 +31,10 @@ interface VersionInfo {
 
 class CodeToolsService {
   private versionCache: Map<string, { version: string; timestamp: number }> = new Map()
+  private terminalsCache: { terminals: TerminalConfig[]; timestamp: number } | null = null
+  private customTerminalPaths: Map<string, string> = new Map() // Store user-configured terminal paths
   private readonly CACHE_DURATION = 1000 * 60 * 30 // 30 minutes cache
+  private readonly TERMINALS_CACHE_DURATION = 1000 * 60 * 5 // 5 minutes cache for terminals
 
   constructor() {
     this.getBunPath = this.getBunPath.bind(this)
@@ -32,6 +44,23 @@ class CodeToolsService {
     this.getVersionInfo = this.getVersionInfo.bind(this)
     this.updatePackage = this.updatePackage.bind(this)
     this.run = this.run.bind(this)
+
+    if (isMac || isWin) {
+      this.preloadTerminals()
+    }
+  }
+
+  /**
+   * Preload available terminals in background
+   */
+  private async preloadTerminals(): Promise<void> {
+    try {
+      logger.info('Preloading available terminals...')
+      await this.getAvailableTerminals()
+      logger.info('Terminal preloading completed')
+    } catch (error) {
+      logger.warn('Terminal preloading failed:', error as Error)
+    }
   }
 
   public async getBunPath() {
@@ -75,10 +104,258 @@ class CodeToolsService {
     }
   }
 
+  /**
+   * Check if a single terminal is available
+   */
+  private async checkTerminalAvailability(terminal: TerminalConfig): Promise<TerminalConfig | null> {
+    try {
+      if (isMac && terminal.bundleId) {
+        // macOS: Check if application is installed via bundle ID with timeout
+        const { stdout } = await execAsync(`mdfind "kMDItemCFBundleIdentifier == '${terminal.bundleId}'"`, {
+          timeout: 3000
+        })
+        if (stdout.trim()) {
+          return terminal
+        }
+      } else if (isWin) {
+        // Windows: Check terminal availability
+        return await this.checkWindowsTerminalAvailability(terminal)
+      } else {
+        // TODO: Check if terminal is available in linux
+        await execAsync(`which ${terminal.id}`, { timeout: 2000 })
+        return terminal
+      }
+    } catch (error) {
+      logger.debug(`Terminal ${terminal.id} not available:`, error as Error)
+    }
+    return null
+  }
+
+  /**
+   * Check Windows terminal availability (simplified - user configured paths)
+   */
+  private async checkWindowsTerminalAvailability(terminal: TerminalConfig): Promise<TerminalConfig | null> {
+    try {
+      switch (terminal.id) {
+        case terminalApps.cmd:
+          // CMD is always available on Windows
+          return terminal
+
+        case terminalApps.powershell:
+          // Check for PowerShell in PATH
+          try {
+            await execAsync('powershell -Command "Get-Host"', { timeout: 3000 })
+            return terminal
+          } catch {
+            try {
+              await execAsync('pwsh -Command "Get-Host"', { timeout: 3000 })
+              return terminal
+            } catch {
+              return null
+            }
+          }
+
+        case terminalApps.windowsTerminal:
+          // Check for Windows Terminal via where command (doesn't launch the terminal)
+          try {
+            await execAsync('where wt', { timeout: 3000 })
+            return terminal
+          } catch {
+            return null
+          }
+
+        case terminalApps.wsl:
+          // Check for WSL
+          try {
+            await execAsync('wsl --status', { timeout: 3000 })
+            return terminal
+          } catch {
+            return null
+          }
+
+        default:
+          // For other terminals (Alacritty, WezTerm), check if user has configured custom path
+          return await this.checkCustomTerminalPath(terminal)
+      }
+    } catch (error) {
+      logger.debug(`Windows terminal ${terminal.id} not available:`, error as Error)
+      return null
+    }
+  }
+
+  /**
+   * Check if user has configured custom path for terminal
+   */
+  private async checkCustomTerminalPath(terminal: TerminalConfig): Promise<TerminalConfig | null> {
+    // Check if user has configured custom path
+    const customPath = this.customTerminalPaths.get(terminal.id)
+    if (customPath && fs.existsSync(customPath)) {
+      try {
+        await execAsync(`"${customPath}" --version`, { timeout: 3000 })
+        return { ...terminal, customPath }
+      } catch {
+        return null
+      }
+    }
+
+    // Fallback to PATH check
+    try {
+      const command = terminal.id === terminalApps.alacritty ? 'alacritty' : 'wezterm'
+      await execAsync(`${command} --version`, { timeout: 3000 })
+      return terminal
+    } catch {
+      return null
+    }
+  }
+
+  /**
+   * Set custom path for a terminal (called from settings UI)
+   */
+  public setCustomTerminalPath(terminalId: string, path: string): void {
+    logger.info(`Setting custom path for terminal ${terminalId}: ${path}`)
+    this.customTerminalPaths.set(terminalId, path)
+    // Clear terminals cache to force refresh
+    this.terminalsCache = null
+  }
+
+  /**
+   * Get custom path for a terminal
+   */
+  public getCustomTerminalPath(terminalId: string): string | undefined {
+    return this.customTerminalPaths.get(terminalId)
+  }
+
+  /**
+   * Remove custom path for a terminal
+   */
+  public removeCustomTerminalPath(terminalId: string): void {
+    logger.info(`Removing custom path for terminal ${terminalId}`)
+    this.customTerminalPaths.delete(terminalId)
+    // Clear terminals cache to force refresh
+    this.terminalsCache = null
+  }
+
+  /**
+   * Get available terminals (with caching and parallel checking)
+   */
+  private async getAvailableTerminals(): Promise<TerminalConfig[]> {
+    const now = Date.now()
+
+    // Check cache first
+    if (this.terminalsCache && now - this.terminalsCache.timestamp < this.TERMINALS_CACHE_DURATION) {
+      logger.info(`Using cached terminals list (${this.terminalsCache.terminals.length} terminals)`)
+      return this.terminalsCache.terminals
+    }
+
+    logger.info('Checking available terminals in parallel...')
+    const startTime = Date.now()
+
+    // Get terminal list based on platform
+    const terminalList = isWin ? WINDOWS_TERMINALS : MACOS_TERMINALS
+
+    // Check all terminals in parallel
+    const terminalPromises = terminalList.map((terminal) => this.checkTerminalAvailability(terminal))
+
+    try {
+      // Wait for all checks to complete with a global timeout
+      const results = await Promise.allSettled(
+        terminalPromises.map((p) =>
+          Promise.race([p, new Promise((_, reject) => setTimeout(() => reject(new Error('timeout')), 5000))])
+        )
+      )
+
+      const availableTerminals: TerminalConfig[] = []
+      results.forEach((result, index) => {
+        if (result.status === 'fulfilled' && result.value) {
+          availableTerminals.push(result.value as TerminalConfig)
+        } else if (result.status === 'rejected') {
+          logger.debug(`Terminal check failed for ${MACOS_TERMINALS[index].id}:`, result.reason)
+        }
+      })
+
+      const endTime = Date.now()
+      logger.info(
+        `Terminal availability check completed in ${endTime - startTime}ms, found ${availableTerminals.length} terminals`
+      )
+
+      // Cache the results
+      this.terminalsCache = {
+        terminals: availableTerminals,
+        timestamp: now
+      }
+
+      return availableTerminals
+    } catch (error) {
+      logger.error('Error checking terminal availability:', error as Error)
+      // Return cached result if available, otherwise empty array
+      return this.terminalsCache?.terminals || []
+    }
+  }
+
+  /**
+   * Get terminal config by ID, fallback to system default
+   */
+  private async getTerminalConfig(terminalId?: string): Promise<TerminalConfigWithCommand> {
+    const availableTerminals = await this.getAvailableTerminals()
+    const terminalCommands = isWin ? WINDOWS_TERMINALS_WITH_COMMANDS : MACOS_TERMINALS_WITH_COMMANDS
+    const defaultTerminal = isWin ? terminalApps.cmd : terminalApps.systemDefault
+
+    if (terminalId) {
+      let requestedTerminal = terminalCommands.find(
+        (t) => t.id === terminalId && availableTerminals.some((at) => at.id === t.id)
+      )
+
+      if (requestedTerminal) {
+        // Apply custom path if configured
+        const customPath = this.customTerminalPaths.get(terminalId)
+        if (customPath && isWin) {
+          requestedTerminal = this.applyCustomPath(requestedTerminal, customPath)
+        }
+        return requestedTerminal
+      } else {
+        logger.warn(`Requested terminal ${terminalId} not available, falling back to system default`)
+      }
+    }
+
+    // Fallback to system default Terminal
+    const systemTerminal = terminalCommands.find(
+      (t) => t.id === defaultTerminal && availableTerminals.some((at) => at.id === t.id)
+    )
+    if (systemTerminal) {
+      return systemTerminal
+    }
+
+    // If even system Terminal is not found, return the first available
+    const firstAvailable = terminalCommands.find((t) => availableTerminals.some((at) => at.id === t.id))
+    if (firstAvailable) {
+      return firstAvailable
+    }
+
+    // Last resort fallback
+    return terminalCommands.find((t) => t.id === defaultTerminal)!
+  }
+
+  /**
+   * Apply custom path to terminal configuration
+   */
+  private applyCustomPath(terminal: TerminalConfigWithCommand, customPath: string): TerminalConfigWithCommand {
+    return {
+      ...terminal,
+      customPath,
+      command: (directory: string, fullCommand: string) => {
+        const originalCommand = terminal.command(directory, fullCommand)
+        return {
+          ...originalCommand,
+          command: customPath // Replace command with custom path
+        }
+      }
+    }
+  }
+
   private async isPackageInstalled(cliTool: string): Promise<boolean> {
     const executableName = await this.getCliExecutableName(cliTool)
     const binDir = path.join(os.homedir(), '.cherrystudio', 'bin')
-    const executablePath = path.join(binDir, executableName + (process.platform === 'win32' ? '.exe' : ''))
+    const executablePath = path.join(binDir, executableName + (isWin ? '.exe' : ''))
 
     // Ensure bin directory exists
     if (!fs.existsSync(binDir)) {
@@ -105,7 +382,7 @@ class CodeToolsService {
       try {
         const executableName = await this.getCliExecutableName(cliTool)
         const binDir = path.join(os.homedir(), '.cherrystudio', 'bin')
-        const executablePath = path.join(binDir, executableName + (process.platform === 'win32' ? '.exe' : ''))
+        const executablePath = path.join(binDir, executableName + (isWin ? '.exe' : ''))
 
         const { stdout } = await execAsync(`"${executablePath}" --version`, { timeout: 10000 })
         // Extract version number from output (format may vary by tool)
@@ -191,6 +468,17 @@ class CodeToolsService {
     }
   }
 
+  /**
+   * Get available terminals for the current platform
+   */
+  public async getAvailableTerminalsForPlatform(): Promise<TerminalConfig[]> {
+    if (isMac || isWin) {
+      return this.getAvailableTerminals()
+    }
+    // For other platforms, return empty array for now
+    return []
+  }
+
   /**
    * Update a CLI tool to the latest version
    */
@@ -202,10 +490,9 @@ class CodeToolsService {
       const bunInstallPath = path.join(os.homedir(), '.cherrystudio')
       const registryUrl = await this.getNpmRegistryUrl()
 
-      const installEnvPrefix =
-        process.platform === 'win32'
-          ? `set "BUN_INSTALL=${bunInstallPath}" && set "NPM_CONFIG_REGISTRY=${registryUrl}" &&`
-          : `export BUN_INSTALL="${bunInstallPath}" && export NPM_CONFIG_REGISTRY="${registryUrl}" &&`
+      const installEnvPrefix = isWin
+        ? `set "BUN_INSTALL=${bunInstallPath}" && set "NPM_CONFIG_REGISTRY=${registryUrl}" &&`
+        : `export BUN_INSTALL="${bunInstallPath}" && export NPM_CONFIG_REGISTRY="${registryUrl}" &&`
 
       const updateCommand = `${installEnvPrefix} "${bunPath}" install -g ${packageName}`
       logger.info(`Executing update command: ${updateCommand}`)
@@ -241,7 +528,7 @@ class CodeToolsService {
     _model: string,
     directory: string,
     env: Record<string, string>,
-    options: { autoUpdateToLatest?: boolean } = {}
+    options: { autoUpdateToLatest?: boolean; terminal?: string } = {}
   ) {
     logger.info(`Starting CLI tool launch: ${cliTool} in directory: ${directory}`)
     logger.debug(`Environment variables:`, Object.keys(env))
@@ -251,7 +538,7 @@ class CodeToolsService {
     const bunPath = await this.getBunPath()
     const executableName = await this.getCliExecutableName(cliTool)
     const binDir = path.join(os.homedir(), '.cherrystudio', 'bin')
-    const executablePath = path.join(binDir, executableName + (process.platform === 'win32' ? '.exe' : ''))
+    const executablePath = path.join(binDir, executableName + (isWin ? '.exe' : ''))
 
     logger.debug(`Package name: ${packageName}`)
     logger.debug(`Bun path: ${bunPath}`)
@@ -295,7 +582,13 @@ class CodeToolsService {
 
     // Build environment variable prefix (based on platform)
     const buildEnvPrefix = (isWindows: boolean) => {
-      if (Object.keys(env).length === 0) return ''
+      if (Object.keys(env).length === 0) {
+        logger.info('No environment variables to set')
+        return ''
+      }
+
+      logger.info('Setting environment variables:', Object.keys(env))
+      logger.info('Environment variable values:', env)
 
       if (isWindows) {
         // Windows uses set command
@@ -304,13 +597,29 @@ class CodeToolsService {
           .join(' && ')
       } else {
         // Unix-like systems use export command
-        return Object.entries(env)
-          .map(([key, value]) => `export ${key}="${value.replace(/"/g, '\\"')}"`)
+        const validEntries = Object.entries(env).filter(([key, value]) => {
+          if (!key || key.trim() === '') {
+            return false
+          }
+          if (value === undefined || value === null) {
+            return false
+          }
+          return true
+        })
+
+        const envCommands = validEntries
+          .map(([key, value]) => {
+            const sanitizedValue = String(value).replace(/\\/g, '\\\\').replace(/"/g, '\\"')
+            const exportCmd = `export ${key}="${sanitizedValue}"`
+            logger.info(`Setting env var: ${key}="${sanitizedValue}"`)
+            logger.info(`Export command: ${exportCmd}`)
+            return exportCmd
+          })
           .join(' && ')
+        return envCommands
       }
     }
 
-    // Build command to execute
     let baseCommand = isWin ? `"${executablePath}"` : `"${bunPath}" "${executablePath}"`
 
     // Add configuration parameters for OpenAI Codex
@@ -351,20 +660,20 @@ class CodeToolsService {
 
     switch (platform) {
       case 'darwin': {
-        // macOS - Use osascript to launch terminal and execute command directly, without showing startup command
+        // macOS - Support multiple terminals
         const envPrefix = buildEnvPrefix(false)
-        const command = envPrefix ? `${envPrefix} && ${baseCommand}` : baseCommand
-        // Combine directory change with the main command to ensure they execute in the same shell session
-        const fullCommand = `cd '${directory.replace(/'/g, "\\'")}' && clear && ${command}`
 
-        terminalCommand = 'osascript'
-        terminalArgs = [
-          '-e',
-          `tell application "Terminal"
-  do script "${fullCommand.replace(/"/g, '\\"')}"
-  activate
-end tell`
-        ]
+        const command = envPrefix ? `${envPrefix} && ${baseCommand}` : baseCommand
+
+        // Combine directory change with the main command to ensure they execute in the same shell session
+        const fullCommand = `cd "${directory.replace(/"/g, '\\"')}" && clear && ${command}`
+
+        const terminalConfig = await this.getTerminalConfig(options.terminal)
+        logger.info(`Using terminal: ${terminalConfig.name} (${terminalConfig.id})`)
+
+        const { command: cmd, args } = terminalConfig.command(directory, fullCommand)
+        terminalCommand = cmd
+        terminalArgs = args
         break
       }
       case 'win32': {
@@ -424,9 +733,23 @@ end tell`
           throw new Error(`Failed to create launch script: ${error}`)
         }
 
-        // Launch bat file - Use safest start syntax, no title parameter
-        terminalCommand = 'cmd'
-        terminalArgs = ['/c', 'start', batFilePath]
+        // Use selected terminal configuration
+        const terminalConfig = await this.getTerminalConfig(options.terminal)
+        logger.info(`Using terminal: ${terminalConfig.name} (${terminalConfig.id})`)
+
+        // Get command and args from terminal configuration
+        // Pass the bat file path as the command to execute
+        const fullCommand = batFilePath
+        const { command: cmd, args } = terminalConfig.command(directory, fullCommand)
+
+        // Override if it's a custom terminal with a custom path
+        if (terminalConfig.customPath) {
+          terminalCommand = terminalConfig.customPath
+          terminalArgs = args
+        } else {
+          terminalCommand = cmd
+          terminalArgs = args
+        }
 
         // Set cleanup task (delete temp file after 5 minutes)
         setTimeout(() => {

src/main/services/ConfigManager.ts

@@ -2,6 +2,7 @@ import { defaultLanguage, UpgradeChannel, ZOOM_SHORTCUTS } from '@shared/config/
 import { LanguageVarious, Shortcut, ThemeMode } from '@types'
 import { app } from 'electron'
 import Store from 'electron-store'
+import { v4 as uuidv4 } from 'uuid'
 
 import { locales } from '../utils/locales'
 
@@ -27,7 +28,8 @@ export enum ConfigKeys {
   SelectionAssistantFilterList = 'selectionAssistantFilterList',
   DisableHardwareAcceleration = 'disableHardwareAcceleration',
   Proxy = 'proxy',
-  EnableDeveloperMode = 'enableDeveloperMode'
+  EnableDeveloperMode = 'enableDeveloperMode',
+  ClientId = 'clientId'
 }
 
 export class ConfigManager {
@@ -241,6 +243,17 @@ export class ConfigManager {
     this.set(ConfigKeys.EnableDeveloperMode, value)
   }
 
+  getClientId(): string {
+    let clientId = this.get<string>(ConfigKeys.ClientId)
+
+    if (!clientId) {
+      clientId = uuidv4()
+      this.set(ConfigKeys.ClientId, clientId)
+    }
+
+    return clientId
+  }
+
   set(key: string, value: unknown, isNotify: boolean = false) {
     this.store.set(key, value)
     isNotify && this.notifySubscribers(key, value)

src/main/services/WindowService.ts

@@ -256,7 +256,7 @@ export class WindowService {
 
   private setupWebContentsHandlers(mainWindow: BrowserWindow) {
     mainWindow.webContents.on('will-navigate', (event, url) => {
-      if (url.includes('localhost:5173')) {
+      if (url.includes('localhost:517')) {
         return
       }
 
@@ -275,7 +275,8 @@ export class WindowService {
         'https://aihubmix.com/topup',
         'https://aihubmix.com/statistics',
         'https://dash.302.ai/sso/login',
-        'https://dash.302.ai/charge'
+        'https://dash.302.ai/charge',
+        'https://www.aiionly.com/login'
       ]
 
       if (oauthProviderUrls.some((link) => url.startsWith(link))) {

src/main/services/__tests__/AppUpdater.test.ts

@@ -0,0 +1,319 @@
+import { UpdateInfo } from 'builder-util-runtime'
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+
+// Mock dependencies
+vi.mock('@logger', () => ({
+  loggerService: {
+    withContext: () => ({
+      info: vi.fn(),
+      error: vi.fn(),
+      warn: vi.fn()
+    })
+  }
+}))
+
+vi.mock('../ConfigManager', () => ({
+  configManager: {
+    getLanguage: vi.fn(),
+    getAutoUpdate: vi.fn(() => false),
+    getTestPlan: vi.fn(() => false),
+    getTestChannel: vi.fn(),
+    getClientId: vi.fn(() => 'test-client-id')
+  }
+}))
+
+vi.mock('../WindowService', () => ({
+  windowService: {
+    getMainWindow: vi.fn()
+  }
+}))
+
+vi.mock('@main/constant', () => ({
+  isWin: false
+}))
+
+vi.mock('@main/utils/ipService', () => ({
+  getIpCountry: vi.fn(() => 'US')
+}))
+
+vi.mock('@main/utils/locales', () => ({
+  locales: {
+    en: { translation: { update: {} } },
+    'zh-CN': { translation: { update: {} } }
+  }
+}))
+
+vi.mock('@main/utils/systemInfo', () => ({
+  generateUserAgent: vi.fn(() => 'test-user-agent')
+}))
+
+vi.mock('electron', () => ({
+  app: {
+    isPackaged: true,
+    getVersion: vi.fn(() => '1.0.0'),
+    getPath: vi.fn(() => '/test/path')
+  },
+  dialog: {
+    showMessageBox: vi.fn()
+  },
+  BrowserWindow: vi.fn(),
+  net: {
+    fetch: vi.fn()
+  }
+}))
+
+vi.mock('electron-updater', () => ({
+  autoUpdater: {
+    logger: null,
+    forceDevUpdateConfig: false,
+    autoDownload: false,
+    autoInstallOnAppQuit: false,
+    requestHeaders: {},
+    on: vi.fn(),
+    setFeedURL: vi.fn(),
+    checkForUpdates: vi.fn(),
+    downloadUpdate: vi.fn(),
+    quitAndInstall: vi.fn(),
+    channel: '',
+    allowDowngrade: false,
+    disableDifferentialDownload: false,
+    currentVersion: '1.0.0'
+  },
+  Logger: vi.fn(),
+  NsisUpdater: vi.fn(),
+  AppUpdater: vi.fn()
+}))
+
+// Import after mocks
+import AppUpdater from '../AppUpdater'
+import { configManager } from '../ConfigManager'
+
+describe('AppUpdater', () => {
+  let appUpdater: AppUpdater
+
+  beforeEach(() => {
+    vi.clearAllMocks()
+    appUpdater = new AppUpdater()
+  })
+
+  describe('parseMultiLangReleaseNotes', () => {
+    const sampleReleaseNotes = `<!--LANG:en-->
+🚀 New Features:
+- Feature A
+- Feature B
+
+🎨 UI Improvements:
+- Improvement A
+<!--LANG:zh-CN-->
+🚀 新功能：
+- 功能 A
+- 功能 B
+
+🎨 界面改进：
+- 改进 A
+<!--LANG:END-->`
+
+    it('should return Chinese notes for zh-CN users', () => {
+      vi.mocked(configManager.getLanguage).mockReturnValue('zh-CN')
+
+      const result = (appUpdater as any).parseMultiLangReleaseNotes(sampleReleaseNotes)
+
+      expect(result).toContain('新功能')
+      expect(result).toContain('功能 A')
+      expect(result).not.toContain('New Features')
+    })
+
+    it('should return Chinese notes for zh-TW users', () => {
+      vi.mocked(configManager.getLanguage).mockReturnValue('zh-TW')
+
+      const result = (appUpdater as any).parseMultiLangReleaseNotes(sampleReleaseNotes)
+
+      expect(result).toContain('新功能')
+      expect(result).toContain('功能 A')
+      expect(result).not.toContain('New Features')
+    })
+
+    it('should return English notes for non-Chinese users', () => {
+      vi.mocked(configManager.getLanguage).mockReturnValue('en-US')
+
+      const result = (appUpdater as any).parseMultiLangReleaseNotes(sampleReleaseNotes)
+
+      expect(result).toContain('New Features')
+      expect(result).toContain('Feature A')
+      expect(result).not.toContain('新功能')
+    })
+
+    it('should return English notes for other language users', () => {
+      vi.mocked(configManager.getLanguage).mockReturnValue('ru-RU')
+
+      const result = (appUpdater as any).parseMultiLangReleaseNotes(sampleReleaseNotes)
+
+      expect(result).toContain('New Features')
+      expect(result).not.toContain('新功能')
+    })
+
+    it('should handle missing language sections gracefully', () => {
+      const malformedNotes = 'Simple release notes without markers'
+
+      const result = (appUpdater as any).parseMultiLangReleaseNotes(malformedNotes)
+
+      expect(result).toBe('Simple release notes without markers')
+    })
+
+    it('should handle malformed markers', () => {
+      const malformedNotes = `<!--LANG:en-->English only`
+      vi.mocked(configManager.getLanguage).mockReturnValue('zh-CN')
+
+      const result = (appUpdater as any).parseMultiLangReleaseNotes(malformedNotes)
+
+      // Should clean up markers and return cleaned content
+      expect(result).toContain('English only')
+      expect(result).not.toContain('<!--LANG:')
+    })
+
+    it('should handle empty release notes', () => {
+      const result = (appUpdater as any).parseMultiLangReleaseNotes('')
+
+      expect(result).toBe('')
+    })
+
+    it('should handle errors gracefully', () => {
+      // Force an error by mocking configManager to throw
+      vi.mocked(configManager.getLanguage).mockImplementation(() => {
+        throw new Error('Test error')
+      })
+
+      const result = (appUpdater as any).parseMultiLangReleaseNotes(sampleReleaseNotes)
+
+      // Should return original notes as fallback
+      expect(result).toBe(sampleReleaseNotes)
+    })
+  })
+
+  describe('hasMultiLanguageMarkers', () => {
+    it('should return true when markers are present', () => {
+      const notes = '<!--LANG:en-->Test'
+
+      const result = (appUpdater as any).hasMultiLanguageMarkers(notes)
+
+      expect(result).toBe(true)
+    })
+
+    it('should return false when no markers are present', () => {
+      const notes = 'Simple text without markers'
+
+      const result = (appUpdater as any).hasMultiLanguageMarkers(notes)
+
+      expect(result).toBe(false)
+    })
+  })
+
+  describe('processReleaseInfo', () => {
+    it('should process multi-language release notes in string format', () => {
+      vi.mocked(configManager.getLanguage).mockReturnValue('zh-CN')
+
+      const releaseInfo = {
+        version: '1.0.0',
+        files: [],
+        path: '',
+        sha512: '',
+        releaseDate: new Date().toISOString(),
+        releaseNotes: `<!--LANG:en-->English notes<!--LANG:zh-CN-->中文说明<!--LANG:END-->`
+      } as UpdateInfo
+
+      const result = (appUpdater as any).processReleaseInfo(releaseInfo)
+
+      expect(result.releaseNotes).toBe('中文说明')
+    })
+
+    it('should not process release notes without markers', () => {
+      const releaseInfo = {
+        version: '1.0.0',
+        files: [],
+        path: '',
+        sha512: '',
+        releaseDate: new Date().toISOString(),
+        releaseNotes: 'Simple release notes'
+      } as UpdateInfo
+
+      const result = (appUpdater as any).processReleaseInfo(releaseInfo)
+
+      expect(result.releaseNotes).toBe('Simple release notes')
+    })
+
+    it('should handle array format release notes', () => {
+      const releaseInfo = {
+        version: '1.0.0',
+        files: [],
+        path: '',
+        sha512: '',
+        releaseDate: new Date().toISOString(),
+        releaseNotes: [
+          { version: '1.0.0', note: 'Note 1' },
+          { version: '1.0.1', note: 'Note 2' }
+        ]
+      } as UpdateInfo
+
+      const result = (appUpdater as any).processReleaseInfo(releaseInfo)
+
+      expect(result.releaseNotes).toEqual(releaseInfo.releaseNotes)
+    })
+
+    it('should handle null release notes', () => {
+      const releaseInfo = {
+        version: '1.0.0',
+        files: [],
+        path: '',
+        sha512: '',
+        releaseDate: new Date().toISOString(),
+        releaseNotes: null
+      } as UpdateInfo
+
+      const result = (appUpdater as any).processReleaseInfo(releaseInfo)
+
+      expect(result.releaseNotes).toBeNull()
+    })
+  })
+
+  describe('formatReleaseNotes', () => {
+    it('should format string release notes with markers', () => {
+      vi.mocked(configManager.getLanguage).mockReturnValue('en-US')
+      const notes = `<!--LANG:en-->English<!--LANG:zh-CN-->中文<!--LANG:END-->`
+
+      const result = (appUpdater as any).formatReleaseNotes(notes)
+
+      expect(result).toBe('English')
+    })
+
+    it('should format string release notes without markers', () => {
+      const notes = 'Simple notes'
+
+      const result = (appUpdater as any).formatReleaseNotes(notes)
+
+      expect(result).toBe('Simple notes')
+    })
+
+    it('should format array release notes', () => {
+      const notes = [
+        { version: '1.0.0', note: 'Note 1' },
+        { version: '1.0.1', note: 'Note 2' }
+      ]
+
+      const result = (appUpdater as any).formatReleaseNotes(notes)
+
+      expect(result).toBe('Note 1\nNote 2')
+    })
+
+    it('should handle null release notes', () => {
+      const result = (appUpdater as any).formatReleaseNotes(null)
+
+      expect(result).toBe('')
+    })
+
+    it('should handle undefined release notes', () => {
+      const result = (appUpdater as any).formatReleaseNotes(undefined)
+
+      expect(result).toBe('')
+    })
+  })
+})

src/main/services/agents/AGENT_MESSAGE_ARCHITECTURE.md

@@ -1,341 +0,0 @@
-# Agent Message Architecture Design Document
-
-## Overview
-
-This document describes the architecture for handling agent messages in Cherry Studio, including how agent-specific messages are generated, transformed to AI SDK format, stored, and sent to the UI. The system is designed to be agent-agnostic, allowing multiple agent types (Claude Code, OpenAI, etc.) to integrate seamlessly.
-
-## Core Design Principles
-
-1. **Agent Agnosticism**: The core message handling system should work with any agent type without modification
-2. **Data Preservation**: All raw agent data must be preserved alongside transformed UI-friendly formats
-3. **Streaming First**: Support real-time streaming of agent responses to the UI
-4. **Type Safety**: Strong TypeScript interfaces ensure consistency across the pipeline
-
-## Architecture Components
-
-### 1. Agent Service Layer
-
-Each agent (e.g., ClaudeCodeService) implements the `AgentServiceInterface`:
-
-```typescript
-interface AgentServiceInterface {
-  invoke(prompt: string, cwd: string, sessionId?: string, options?: any): AgentStream
-}
-```
-
-#### Responsibilities:
-- Spawn and manage agent-specific processes (e.g., Claude Code CLI)
-- Parse agent-specific output formats (e.g., SDKMessage for Claude Code)
-- Transform agent messages to AI SDK format
-- Emit standardized `AgentStreamEvent` objects
-
-### 2. Agent Stream Events
-
-The standardized event interface that all agents emit:
-
-```typescript
-interface AgentStreamEvent {
-  type: 'chunk' | 'error' | 'complete'
-  chunk?: UIMessageChunk              // AI SDK format for UI
-  rawAgentMessage?: any               // Agent-specific raw message
-  error?: Error
-  agentResult?: any                   // Complete agent-specific result
-}
-```
-
-### 3. Session Message Service
-
-The `SessionMessageService` acts as the orchestration layer:
-
-#### Responsibilities:
-- Manages session lifecycle and persistence
-- Collects streaming chunks and raw agent messages
-- Stores structured data in the database
-- Forwards events to the API layer
-
-### 4. Database Storage
-
-Session messages are stored with complete structured data:
-
-```typescript
-interface SessionMessageContent {
-  aiSDKChunks: UIMessageChunk[]      // UI-friendly format
-  rawAgentMessages: any[]            // Original agent messages
-  agentResult?: any                  // Complete agent result
-  agentType: string                  // Agent identifier
-}
-```
-
-## Data Flow
-
-```mermaid
-graph TD
-    A[User Input] --> B[API Handler]
-    B --> C[SessionMessageService]
-    C --> D[Agent Service]
-    D --> E[Agent Process]
-    E --> F[Raw Agent Output]
-    F --> G[Transform to AI SDK]
-    G --> H[Emit AgentStreamEvent]
-    H --> I[SessionMessageService]
-    I --> J[Store in Database]
-    I --> K[Forward to Client]
-    K --> L[UI Rendering]
-```
-
-## Message Transformation Process
-
-### Step 1: Raw Agent Message Generation
-
-Each agent generates messages in its native format:
-
-**Claude Code Example:**
-```typescript
-// SDKMessage from Claude Code CLI
-{
-  type: 'assistant',
-  uuid: 'msg_123',
-  session_id: 'session_456',
-  message: {
-    role: 'assistant',
-    content: [
-      { type: 'text', text: 'Hello, I can help...' },
-      { type: 'tool_use', id: 'tool_1', name: 'read_file', input: {...} }
-    ]
-  }
-}
-```
-
-### Step 2: Transformation to AI SDK Format
-
-The agent service transforms native messages to AI SDK `UIMessageChunk`:
-
-```typescript
-// In ClaudeCodeService
-const emitChunks = (sdkMessage: SDKMessage) => {
-  // Transform to AI SDK format
-  const chunks = transformSDKMessageToUIChunk(sdkMessage)
-
-  for (const chunk of chunks) {
-    stream.emit('data', {
-      type: 'chunk',
-      chunk,                        // AI SDK format
-      rawAgentMessage: sdkMessage   // Preserve original
-    })
-  }
-}
-```
-
-**Transformed AI SDK Chunk:**
-```typescript
-{
-  type: 'text-delta',
-  id: 'msg_123',
-  delta: 'Hello, I can help...',
-  providerMetadata: {
-    claudeCode: {
-      originalSDKMessage: {...},
-      uuid: 'msg_123',
-      session_id: 'session_456'
-    }
-  }
-}
-```
-
-### Step 3: Session Message Processing
-
-The SessionMessageService collects and processes events:
-
-```typescript
-// Collect streaming data
-const streamedChunks: UIMessageChunk[] = []
-const rawAgentMessages: any[] = []
-
-claudeStream.on('data', async (event: AgentStreamEvent) => {
-  switch (event.type) {
-    case 'chunk':
-      streamedChunks.push(event.chunk)
-      if (event.rawAgentMessage) {
-        rawAgentMessages.push(event.rawAgentMessage)
-      }
-      // Forward to client
-      sessionStream.emit('data', { type: 'chunk', chunk: event.chunk })
-      break
-
-    case 'complete':
-      // Store complete structured data
-      const content = {
-        aiSDKChunks: streamedChunks,
-        rawAgentMessages: rawAgentMessages,
-        agentResult: event.agentResult,
-        agentType: event.agentResult?.agentType || 'unknown'
-      }
-      // Save to database...
-      break
-  }
-})
-```
-
-### Step 4: Client Streaming
-
-The API handler converts events to Server-Sent Events (SSE):
-
-```typescript
-// In API handler
-messageStream.on('data', (event: any) => {
-  switch (event.type) {
-    case 'chunk':
-      // Send AI SDK chunk as SSE
-      res.write(`data: ${JSON.stringify(event.chunk)}\n\n`)
-      break
-    case 'complete':
-      res.write('data: [DONE]\n\n')
-      res.end()
-      break
-  }
-})
-```
-
-## Adding New Agent Types
-
-To add support for a new agent (e.g., OpenAI):
-
-### 1. Create Agent Service
-
-```typescript
-class OpenAIService implements AgentServiceInterface {
-  invokeStream(prompt: string, cwd: string, sessionId?: string, options?: any): AgentStream {
-    const stream = new OpenAIStream()
-
-    // Call OpenAI API
-    const openaiResponse = await openai.chat.completions.create({
-      messages: [{ role: 'user', content: prompt }],
-      stream: true
-    })
-
-    // Transform OpenAI format to AI SDK
-    for await (const chunk of openaiResponse) {
-      const aiSDKChunk = transformOpenAIToAISDK(chunk)
-      stream.emit('data', {
-        type: 'chunk',
-        chunk: aiSDKChunk,
-        rawAgentMessage: chunk  // Preserve OpenAI format
-      })
-    }
-
-    return stream
-  }
-}
-```
-
-### 2. Create Transform Function
-
-```typescript
-function transformOpenAIToAISDK(openaiChunk: OpenAIChunk): UIMessageChunk {
-  return {
-    type: 'text-delta',
-    id: openaiChunk.id,
-    delta: openaiChunk.choices[0].delta.content,
-    providerMetadata: {
-      openai: {
-        original: openaiChunk,
-        model: openaiChunk.model
-      }
-    }
-  }
-}
-```
-
-### 3. Register Agent Type
-
-Update the agent type enum and factory:
-
-```typescript
-export type AgentType = 'claude-code' | 'openai' | 'anthropic-api'
-
-function createAgentService(type: AgentType): AgentServiceInterface {
-  switch (type) {
-    case 'claude-code':
-      return new ClaudeCodeService()
-    case 'openai':
-      return new OpenAIService()
-    // ...
-  }
-}
-```
-
-## Benefits of This Architecture
-
-1. **Extensibility**: Easy to add new agent types without modifying core logic
-2. **Data Integrity**: Raw agent data is never lost during transformation
-3. **Debugging**: Complete message history available for troubleshooting
-4. **Performance**: Streaming support for real-time responses
-5. **Type Safety**: Strong interfaces prevent runtime errors
-6. **UI Consistency**: All agents provide data in standard AI SDK format
-
-## Key Interfaces Reference
-
-### AgentStreamEvent
-```typescript
-interface AgentStreamEvent {
-  type: 'chunk' | 'error' | 'complete'
-  chunk?: UIMessageChunk
-  rawAgentMessage?: any
-  error?: Error
-  agentResult?: any
-}
-```
-
-### SessionMessageEntity
-```typescript
-interface SessionMessageEntity {
-  id: number
-  session_id: string
-  parent_id?: number
-  role: 'user' | 'assistant' | 'system' | 'tool'
-  type: string
-  content: string | SessionMessageContent
-  metadata?: Record<string, any>
-  created_at: string
-  updated_at: string
-}
-```
-
-### SessionMessageContent
-```typescript
-interface SessionMessageContent {
-  aiSDKChunks: UIMessageChunk[]
-  rawAgentMessages: any[]
-  agentResult?: any
-  agentType: string
-}
-```
-
-## Testing Strategy
-
-### Unit Tests
-- Test each transform function independently
-- Verify event emission sequences
-- Validate data structure preservation
-
-### Integration Tests
-- Test complete flow from input to database
-- Verify streaming behavior
-- Test error handling and recovery
-
-### Agent-Specific Tests
-- Validate agent-specific transformations
-- Test edge cases for each agent type
-- Verify metadata preservation
-
-## Future Enhancements
-
-1. **Message Replay**: Ability to replay sessions from stored raw messages
-2. **Format Migration**: Tools to migrate between agent formats
-3. **Analytics**: Aggregate metrics from raw agent data
-4. **Caching**: Cache transformed chunks for performance
-5. **Compression**: Compress raw messages for storage efficiency
-
-## Conclusion
-
-This architecture provides a robust, extensible foundation for handling messages from multiple AI agents while maintaining data integrity and providing a consistent interface for the UI. The separation of concerns between agent-specific logic and core message handling ensures the system can evolve to support new agents and features without breaking existing functionality.

src/main/services/agents/BaseService.ts

@@ -1,311 +0,0 @@
-import { type Client, createClient } from '@libsql/client'
-import { loggerService } from '@logger'
-import { ModelValidationError, validateModelId } from '@main/apiServer/utils'
-import { AgentType, objectKeys, Provider } from '@types'
-import { drizzle, type LibSQLDatabase } from 'drizzle-orm/libsql'
-import fs from 'fs'
-import path from 'path'
-
-import { MigrationService } from './database/MigrationService'
-import * as schema from './database/schema'
-import { dbPath } from './drizzle.config'
-import { AgentModelField, AgentModelValidationError } from './errors'
-
-const logger = loggerService.withContext('BaseService')
-
-/**
- * Base service class providing shared database connection and utilities
- * for all agent-related services.
- *
- * Features:
- * - Programmatic schema management (no CLI dependencies)
- * - Automatic table creation and migration
- * - Schema version tracking and compatibility checks
- * - Transaction-based operations for safety
- * - Development vs production mode handling
- * - Connection retry logic with exponential backoff
- */
-export abstract class BaseService {
-  protected static client: Client | null = null
-  protected static db: LibSQLDatabase<typeof schema> | null = null
-  protected static isInitialized = false
-  protected static initializationPromise: Promise<void> | null = null
-  protected jsonFields: string[] = ['built_in_tools', 'mcps', 'configuration', 'accessible_paths', 'allowed_tools']
-
-  /**
-   * Initialize database with retry logic and proper error handling
-   */
-  protected static async initialize(): Promise<void> {
-    // Return existing initialization if in progress
-    if (BaseService.initializationPromise) {
-      return BaseService.initializationPromise
-    }
-
-    if (BaseService.isInitialized) {
-      return
-    }
-
-    BaseService.initializationPromise = BaseService.performInitialization()
-    return BaseService.initializationPromise
-  }
-
-  private static async performInitialization(): Promise<void> {
-    const maxRetries = 3
-    let lastError: Error
-
-    for (let attempt = 1; attempt <= maxRetries; attempt++) {
-      try {
-        logger.info(`Initializing Agent database at: ${dbPath} (attempt ${attempt}/${maxRetries})`)
-
-        // Ensure the database directory exists
-        const dbDir = path.dirname(dbPath)
-        if (!fs.existsSync(dbDir)) {
-          logger.info(`Creating database directory: ${dbDir}`)
-          fs.mkdirSync(dbDir, { recursive: true })
-        }
-
-        BaseService.client = createClient({
-          url: `file:${dbPath}`
-        })
-
-        BaseService.db = drizzle(BaseService.client, { schema })
-
-        // Run database migrations
-        const migrationService = new MigrationService(BaseService.db, BaseService.client)
-        await migrationService.runMigrations()
-
-        BaseService.isInitialized = true
-        logger.info('Agent database initialized successfully')
-        return
-      } catch (error) {
-        lastError = error as Error
-        logger.warn(`Database initialization attempt ${attempt} failed:`, lastError)
-
-        // Clean up on failure
-        if (BaseService.client) {
-          try {
-            BaseService.client.close()
-          } catch (closeError) {
-            logger.warn('Failed to close client during cleanup:', closeError as Error)
-          }
-        }
-        BaseService.client = null
-        BaseService.db = null
-
-        // Wait before retrying (exponential backoff)
-        if (attempt < maxRetries) {
-          const delay = Math.pow(2, attempt) * 1000 // 2s, 4s, 8s
-          logger.info(`Retrying in ${delay}ms...`)
-          await new Promise((resolve) => setTimeout(resolve, delay))
-        }
-      }
-    }
-
-    // All retries failed
-    BaseService.initializationPromise = null
-    logger.error('Failed to initialize Agent database after all retries:', lastError!)
-    throw lastError!
-  }
-
-  protected ensureInitialized(): void {
-    if (!BaseService.isInitialized || !BaseService.db || !BaseService.client) {
-      throw new Error('Database not initialized. Call initialize() first.')
-    }
-  }
-
-  protected get database(): LibSQLDatabase<typeof schema> {
-    this.ensureInitialized()
-    return BaseService.db!
-  }
-
-  protected get rawClient(): Client {
-    this.ensureInitialized()
-    return BaseService.client!
-  }
-
-  protected serializeJsonFields(data: any): any {
-    const serialized = { ...data }
-
-    for (const field of this.jsonFields) {
-      if (serialized[field] !== undefined) {
-        serialized[field] =
-          Array.isArray(serialized[field]) || typeof serialized[field] === 'object'
-            ? JSON.stringify(serialized[field])
-            : serialized[field]
-      }
-    }
-
-    return serialized
-  }
-
-  protected deserializeJsonFields(data: any): any {
-    if (!data) return data
-
-    const deserialized = { ...data }
-
-    for (const field of this.jsonFields) {
-      if (deserialized[field] && typeof deserialized[field] === 'string') {
-        try {
-          deserialized[field] = JSON.parse(deserialized[field])
-        } catch (error) {
-          logger.warn(`Failed to parse JSON field ${field}:`, error as Error)
-        }
-      }
-    }
-
-    // convert null from db to undefined to satisfy type definition
-    for (const key of objectKeys(data)) {
-      if (deserialized[key] === null) {
-        deserialized[key] = undefined
-      }
-    }
-
-    return deserialized
-  }
-
-  /**
-   * Validate, normalize, and ensure filesystem access for a set of absolute paths.
-   *
-   * - Requires every entry to be an absolute path and throws if not.
-   * - Normalizes each path and deduplicates while preserving order.
-   * - Creates missing directories (or parent directories for file-like paths).
-   */
-  protected ensurePathsExist(paths?: string[]): string[] {
-    if (!paths?.length) {
-      return []
-    }
-
-    const sanitizedPaths: string[] = []
-    const seenPaths = new Set<string>()
-
-    for (const rawPath of paths) {
-      if (!rawPath) {
-        continue
-      }
-
-      if (!path.isAbsolute(rawPath)) {
-        throw new Error(`Accessible path must be absolute: ${rawPath}`)
-      }
-
-      // Normalize to provide consistent values to downstream consumers.
-      const resolvedPath = path.normalize(rawPath)
-
-      let stats: fs.Stats | null = null
-      try {
-        // Attempt to stat the path to understand whether it already exists and if it is a file.
-        if (fs.existsSync(resolvedPath)) {
-          stats = fs.statSync(resolvedPath)
-        }
-      } catch (error) {
-        logger.warn('Failed to inspect accessible path', {
-          path: rawPath,
-          error: error instanceof Error ? error.message : String(error)
-        })
-      }
-
-      const looksLikeFile =
-        (stats && stats.isFile()) || (!stats && path.extname(resolvedPath) !== '' && !resolvedPath.endsWith(path.sep))
-
-      // For file-like targets create the parent directory; otherwise ensure the directory itself.
-      const directoryToEnsure = looksLikeFile ? path.dirname(resolvedPath) : resolvedPath
-
-      if (!fs.existsSync(directoryToEnsure)) {
-        try {
-          fs.mkdirSync(directoryToEnsure, { recursive: true })
-        } catch (error) {
-          logger.error('Failed to create accessible path directory', {
-            path: directoryToEnsure,
-            error: error instanceof Error ? error.message : String(error)
-          })
-          throw error
-        }
-      }
-
-      // Preserve the first occurrence only to avoid duplicates while keeping caller order stable.
-      if (!seenPaths.has(resolvedPath)) {
-        seenPaths.add(resolvedPath)
-        sanitizedPaths.push(resolvedPath)
-      }
-    }
-
-    return sanitizedPaths
-  }
-
-  /**
-   * Force re-initialization (for development/testing)
-   */
-  protected async validateAgentModels(
-    agentType: AgentType,
-    models: Partial<Record<AgentModelField, string | undefined>>
-  ): Promise<void> {
-    const entries = Object.entries(models) as [AgentModelField, string | undefined][]
-    if (entries.length === 0) {
-      return
-    }
-
-    for (const [field, rawValue] of entries) {
-      if (rawValue === undefined || rawValue === null) {
-        continue
-      }
-
-      const modelValue = rawValue
-      const validation = await validateModelId(modelValue)
-
-      if (!validation.valid || !validation.provider) {
-        const detail: ModelValidationError = validation.error ?? {
-          type: 'invalid_format',
-          message: 'Unknown model validation error',
-          code: 'validation_error'
-        }
-
-        throw new AgentModelValidationError({ agentType, field, model: modelValue }, detail)
-      }
-
-      if (!validation.provider.apiKey) {
-        throw new AgentModelValidationError(
-          { agentType, field, model: modelValue },
-          {
-            type: 'invalid_format',
-            message: `Provider '${validation.provider.id}' is missing an API key`,
-            code: 'provider_api_key_missing'
-          }
-        )
-      }
-
-      // different agent types may have different provider requirements
-      const agentTypeProviderRequirements: Record<AgentType, Provider['type']> = {
-        'claude-code': 'anthropic'
-      }
-      for (const [ak, pk] of Object.entries(agentTypeProviderRequirements)) {
-        if (agentType === ak && validation.provider.type !== pk) {
-          throw new AgentModelValidationError(
-            { agentType, field, model: modelValue },
-            {
-              type: 'unsupported_provider_type',
-              message: `Provider type '${validation.provider.type}' is not supported for agent type '${agentType}'. Expected '${pk}'`,
-              code: 'unsupported_provider_type'
-            }
-          )
-        }
-      }
-    }
-  }
-
-  static async reinitialize(): Promise<void> {
-    BaseService.isInitialized = false
-    BaseService.initializationPromise = null
-
-    if (BaseService.client) {
-      try {
-        BaseService.client.close()
-      } catch (error) {
-        logger.warn('Failed to close client during reinitialize:', error as Error)
-      }
-    }
-
-    BaseService.client = null
-    BaseService.db = null
-
-    await BaseService.initialize()
-  }
-}

src/main/services/agents/README.md

@@ -1,81 +0,0 @@
-# Agents Service
-
-Simplified Drizzle ORM implementation for agent and session management in Cherry Studio.
-
-## Features
-
-- **Native Drizzle migrations** - Uses built-in migrate() function
-- **Zero CLI dependencies** in production
-- **Auto-initialization** with retry logic
-- **Full TypeScript** type safety
-- **Model validation** to ensure models exist and provider configuration matches the agent type
-
-## Schema
-
-- `agents.schema.ts` - Agent definitions
-- `sessions.schema.ts` - Session and message tables
-- `migrations.schema.ts` - Migration tracking
-
-## Usage
-
-```typescript
-import { agentService } from './services'
-
-// Create agent - fully typed
-const agent = await agentService.createAgent({
-  type: 'custom',
-  name: 'My Agent',
-  model: 'anthropic:claude-3-5-sonnet-20241022'
-})
-```
-
-## Model Validation
-
-- Model identifiers must use the `provider:model_id` format (for example `anthropic:claude-3-5-sonnet-20241022`).
-- `model`, `plan_model`, and `small_model` are validated against the configured providers before the database is touched.
-- Invalid configurations return a `400 invalid_request_error` response and the create/update operation is aborted.
-
-## Development Commands
-
-```bash
-# Apply schema changes
-yarn agents:generate
-
-# Quick development sync
-yarn agents:push
-
-# Database tools
-yarn agents:studio    # Open Drizzle Studio
-yarn agents:health    # Health check
-yarn agents:drop      # Reset database
-```
-
-## Workflow
-
-1. **Edit schema** in `/database/schema/`
-2. **Generate migration** with `yarn agents:generate`
-3. **Test changes** with `yarn agents:health`
-4. **Deploy** - migrations apply automatically
-
-## Services
-
-- `AgentService` - Agent CRUD operations
-- `SessionService` - Session management
-- `SessionMessageService` - Message logging
-- `BaseService` - Database utilities
-- `schemaSyncer` - Migration handler
-
-## Troubleshooting
-
-```bash
-# Check status
-yarn agents:health
-
-# Apply migrations
-yarn agents:migrate
-
-# Reset completely
-yarn agents:reset --yes
-```
-
-The simplified migration system reduced complexity from 463 to ~30 lines while maintaining all functionality through Drizzle's native migration system.

src/main/services/agents/TODO.md

@@ -1,35 +0,0 @@
-# Agents Service Refactor TODO (interface-level)
-
-- [x] **SessionMessageService.createSessionMessage**
-  - Replace the current `EventEmitter` that emits `UIMessageChunk` with a readable stream of `TextStreamPart` objects (same shape produced by `/api/messages` in `messageThunk`).
-  - Update `startSessionMessageStream` to call a new adapter (`claudeToTextStreamPart(chunk)`) that maps Claude Code chunk payloads to `{ type: 'text-delta' | 'tool-call' | ... }` parts used by `AiSdkToChunkAdapter`.
-  - Add a secondary return value (promise) resolving to the persisted `ModelMessage[]` once streaming completes, so the renderer thunk can await save confirmation.
-
-- [x] **main -> renderer transport**
-  - Update the existing SSE handler in `src/main/apiServer/routes/agents/handlers/messages.ts` (e.g., `createMessage`) to forward the new `TextStreamPart` stream over HTTP, preserving the current agent endpoint contract.
-  - Keep abort handling compatible with the current HTTP server (honor `AbortController` on the request to terminate the stream).
-
-- [x] **renderer thunk integration**
-  - Introduce a thin IPC contract (e.g., `AgentMessagePersistence`) surfaced by `src/main/services/agents/database/index.ts` so the renderer thunk can request session-message writes without going through `SessionMessageService`.
-    - Define explicit entry points on the main side:
-      - `persistUserMessage({ sessionId, agentSessionId, payload, createdAt?, metadata? })`
-      - `persistAssistantMessage({ sessionId, agentSessionId, payload, createdAt?, metadata? })`
-      - `persistExchange({ sessionId, agentSessionId, user, assistant })` which runs the above in a single transaction and returns both records.
-    - Export these helpers via an `agentMessageRepository` object so both IPC handlers and legacy services share the same persistence path.
-    - Normalize persisted payloads to `{ message, blocks }` matching the renderer schema instead of AI-SDK `ModelMessage` chunks.
-  - Extend `messageThunk.sendMessage` to call the agent transport when the topic corresponds to a session, pipe chunks through `createStreamProcessor` + `AiSdkToChunkAdapter`, and invoke the new persistence interface once streaming resolves.
-  - Replace `useSession().createSessionMessage` optimistic insert with dispatching the thunk so Redux/Dexie persistence happens via the shared save helpers.
-
-- [x] **persistence alignment**
-  - Remove `persistUserMessage` / `persistAssistantMessage` calls from `SessionMessageService`; instead expose a `SessionMessageRepository` in `main` that the thunk invokes via existing Dexie helpers.
-  - On renderer side, persist agent exchanges via IPC after streaming completes, storing `{ message, blocks }` payloads while skipping Dexie writes for agent sessions so the single source of truth remains `session_messages`.
-
-- [x] **Blocks renderer**
-  - Replace `AgentSessionMessages` simple `<div>` render with the shared `Blocks` component (`src/renderer/src/pages/home/Messages/Blocks`) wired to the Redux store.
-  - Adjust `useSession` to only fetch metadata (e.g., session info) and rely on store selectors for message list.
-
-- [x] **API client clean-up**
-  - Remove `AgentApiClient.createMessage` direct POST once thunk is in place; calls should go through renderer thunk -> stream -> final persistence.
-
-- [ ] **Regression tests**
-  - Add integration test to assert agent sessions render incremental text the same way as standard assistant messages.

src/main/services/agents/database/MigrationService.ts

@@ -1,161 +0,0 @@
-import { type Client } from '@libsql/client'
-import { loggerService } from '@logger'
-import { getResourcePath } from '@main/utils'
-import { type LibSQLDatabase } from 'drizzle-orm/libsql'
-import fs from 'fs'
-import path from 'path'
-
-import * as schema from './schema'
-import { migrations, type NewMigration } from './schema/migrations.schema'
-
-const logger = loggerService.withContext('MigrationService')
-
-interface MigrationJournal {
-  version: string
-  dialect: string
-  entries: Array<{
-    idx: number
-    version: string
-    when: number
-    tag: string
-    breakpoints: boolean
-  }>
-}
-
-export class MigrationService {
-  private db: LibSQLDatabase<typeof schema>
-  private client: Client
-  private migrationDir: string
-
-  constructor(db: LibSQLDatabase<typeof schema>, client: Client) {
-    this.db = db
-    this.client = client
-    this.migrationDir = path.join(getResourcePath(), 'database', 'drizzle')
-  }
-
-  async runMigrations(): Promise<void> {
-    try {
-      logger.info('Starting migration check...')
-
-      const hasMigrationsTable = await this.migrationsTableExists()
-
-      if (!hasMigrationsTable) {
-        logger.info('Migrations table not found; assuming fresh database state')
-      }
-
-      // Read migration journal
-      const journal = await this.readMigrationJournal()
-      if (!journal.entries.length) {
-        logger.info('No migrations found in journal')
-        return
-      }
-
-      // Get applied migrations
-      const appliedMigrations = hasMigrationsTable ? await this.getAppliedMigrations() : []
-      const appliedVersions = new Set(appliedMigrations.map((m) => Number(m.version)))
-
-      const latestAppliedVersion = appliedMigrations.reduce(
-        (max, migration) => Math.max(max, Number(migration.version)),
-        0
-      )
-      const latestJournalVersion = journal.entries.reduce((max, entry) => Math.max(max, entry.idx), 0)
-
-      logger.info(`Latest applied migration: v${latestAppliedVersion}, latest available: v${latestJournalVersion}`)
-
-      // Find pending migrations (compare journal idx with stored version, which is the same value)
-      const pendingMigrations = journal.entries
-        .filter((entry) => !appliedVersions.has(entry.idx))
-        .sort((a, b) => a.idx - b.idx)
-
-      if (pendingMigrations.length === 0) {
-        logger.info('Database is up to date')
-        return
-      }
-
-      logger.info(`Found ${pendingMigrations.length} pending migrations`)
-
-      // Execute pending migrations
-      for (const migration of pendingMigrations) {
-        await this.executeMigration(migration)
-      }
-
-      logger.info('All migrations completed successfully')
-    } catch (error) {
-      logger.error('Migration failed:', { error })
-      throw error
-    }
-  }
-
-  private async migrationsTableExists(): Promise<boolean> {
-    try {
-      const table = await this.client.execute(`SELECT name FROM sqlite_master WHERE type='table' AND name='migrations'`)
-      return table.rows.length > 0
-    } catch (error) {
-      logger.error('Failed to check migrations table status:', { error })
-      throw error
-    }
-  }
-
-  private async readMigrationJournal(): Promise<MigrationJournal> {
-    const journalPath = path.join(this.migrationDir, 'meta', '_journal.json')
-
-    if (!fs.existsSync(journalPath)) {
-      logger.warn('Migration journal not found:', { journalPath })
-      return { version: '7', dialect: 'sqlite', entries: [] }
-    }
-
-    try {
-      const journalContent = fs.readFileSync(journalPath, 'utf-8')
-      return JSON.parse(journalContent)
-    } catch (error) {
-      logger.error('Failed to read migration journal:', { error })
-      throw error
-    }
-  }
-
-  private async getAppliedMigrations(): Promise<schema.Migration[]> {
-    try {
-      return await this.db.select().from(migrations)
-    } catch (error) {
-      // This should not happen since we ensure the table exists in runMigrations()
-      logger.error('Failed to query applied migrations:', { error })
-      throw error
-    }
-  }
-
-  private async executeMigration(migration: MigrationJournal['entries'][0]): Promise<void> {
-    const sqlFilePath = path.join(this.migrationDir, `${migration.tag}.sql`)
-
-    if (!fs.existsSync(sqlFilePath)) {
-      throw new Error(`Migration SQL file not found: ${sqlFilePath}`)
-    }
-
-    try {
-      logger.info(`Executing migration ${migration.tag}...`)
-      const startTime = Date.now()
-
-      // Read and execute SQL
-      const sqlContent = fs.readFileSync(sqlFilePath, 'utf-8')
-      await this.client.executeMultiple(sqlContent)
-
-      // Record migration as applied (store journal idx as version for tracking)
-      const newMigration: NewMigration = {
-        version: migration.idx,
-        tag: migration.tag,
-        executedAt: Date.now()
-      }
-
-      if (!(await this.migrationsTableExists())) {
-        throw new Error('Migrations table missing after executing migration; cannot record progress')
-      }
-
-      await this.db.insert(migrations).values(newMigration)
-
-      const executionTime = Date.now() - startTime
-      logger.info(`Migration ${migration.tag} completed in ${executionTime}ms`)
-    } catch (error) {
-      logger.error(`Migration ${migration.tag} failed:`, { error })
-      throw error
-    }
-  }
-}

src/main/services/agents/database/index.ts

@@ -1,14 +0,0 @@
-/**
- * Database Module
- *
- * This module provides centralized access to Drizzle ORM schemas
- * for type-safe database operations.
- *
- * Schema evolution is handled by Drizzle Kit migrations.
- */
-
-// Drizzle ORM schemas
-export * from './schema'
-
-// Repository helpers
-export * from './sessionMessageRepository'

src/main/services/agents/database/schema/agents.schema.ts

@@ -1,35 +0,0 @@
-/**
- * Drizzle ORM schema for agents table
- */
-
-import { index, sqliteTable, text } from 'drizzle-orm/sqlite-core'
-
-export const agentsTable = sqliteTable('agents', {
-  id: text('id').primaryKey(),
-  type: text('type').notNull(),
-  name: text('name').notNull(),
-  description: text('description'),
-  accessible_paths: text('accessible_paths'), // JSON array of directory paths the agent can access
-
-  instructions: text('instructions'),
-
-  model: text('model').notNull(), // Main model ID (required)
-  plan_model: text('plan_model'), // Optional plan/thinking model ID
-  small_model: text('small_model'), // Optional small/fast model ID
-
-  mcps: text('mcps'), // JSON array of MCP tool IDs
-  allowed_tools: text('allowed_tools'), // JSON array of allowed tool IDs (whitelist)
-
-  configuration: text('configuration'), // JSON, extensible settings
-
-  created_at: text('created_at').notNull(),
-  updated_at: text('updated_at').notNull()
-})
-
-// Indexes for agents table
-export const agentsNameIdx = index('idx_agents_name').on(agentsTable.name)
-export const agentsTypeIdx = index('idx_agents_type').on(agentsTable.type)
-export const agentsCreatedAtIdx = index('idx_agents_created_at').on(agentsTable.created_at)
-
-export type AgentRow = typeof agentsTable.$inferSelect
-export type InsertAgentRow = typeof agentsTable.$inferInsert

src/main/services/agents/database/schema/index.ts

@@ -1,8 +0,0 @@
-/**
- * Drizzle ORM schema exports
- */
-
-export * from './agents.schema'
-export * from './messages.schema'
-export * from './migrations.schema'
-export * from './sessions.schema'

src/main/services/agents/database/schema/messages.schema.ts

@@ -1,30 +0,0 @@
-import { foreignKey, index, integer, sqliteTable, text } from 'drizzle-orm/sqlite-core'
-
-import { sessionsTable } from './sessions.schema'
-
-// session_messages table to log all messages, thoughts, actions, observations in a session
-export const sessionMessagesTable = sqliteTable('session_messages', {
-  id: integer('id').primaryKey({ autoIncrement: true }),
-  session_id: text('session_id').notNull(),
-  role: text('role').notNull(), // 'user', 'agent', 'system', 'tool'
-  content: text('content').notNull(), // JSON structured data
-  agent_session_id: text('agent_session_id').default(''),
-  metadata: text('metadata'), // JSON metadata (optional)
-  created_at: text('created_at').notNull(),
-  updated_at: text('updated_at').notNull()
-})
-
-// Indexes for session_messages table
-export const sessionMessagesSessionIdIdx = index('idx_session_messages_session_id').on(sessionMessagesTable.session_id)
-export const sessionMessagesCreatedAtIdx = index('idx_session_messages_created_at').on(sessionMessagesTable.created_at)
-export const sessionMessagesUpdatedAtIdx = index('idx_session_messages_updated_at').on(sessionMessagesTable.updated_at)
-
-// Foreign keys for session_messages table
-export const sessionMessagesFkSession = foreignKey({
-  columns: [sessionMessagesTable.session_id],
-  foreignColumns: [sessionsTable.id],
-  name: 'fk_session_messages_session_id'
-}).onDelete('cascade')
-
-export type SessionMessageRow = typeof sessionMessagesTable.$inferSelect
-export type InsertSessionMessageRow = typeof sessionMessagesTable.$inferInsert

src/main/services/agents/database/schema/migrations.schema.ts

@@ -1,14 +0,0 @@
-/**
- * Migration tracking schema
- */
-
-import { integer, sqliteTable, text } from 'drizzle-orm/sqlite-core'
-
-export const migrations = sqliteTable('migrations', {
-  version: integer('version').primaryKey(),
-  tag: text('tag').notNull(),
-  executedAt: integer('executed_at').notNull()
-})
-
-export type Migration = typeof migrations.$inferSelect
-export type NewMigration = typeof migrations.$inferInsert

src/main/services/agents/database/schema/sessions.schema.ts

@@ -1,45 +0,0 @@
-/**
- * Drizzle ORM schema for sessions and session_logs tables
- */
-
-import { foreignKey, index, sqliteTable, text } from 'drizzle-orm/sqlite-core'
-
-import { agentsTable } from './agents.schema'
-
-export const sessionsTable = sqliteTable('sessions', {
-  id: text('id').primaryKey(),
-  agent_type: text('agent_type').notNull(),
-  agent_id: text('agent_id').notNull(), // Primary agent ID for the session
-  name: text('name').notNull(),
-  description: text('description'),
-  accessible_paths: text('accessible_paths'), // JSON array of directory paths the agent can access
-
-  instructions: text('instructions'),
-
-  model: text('model').notNull(), // Main model ID (required)
-  plan_model: text('plan_model'), // Optional plan/thinking model ID
-  small_model: text('small_model'), // Optional small/fast model ID
-
-  mcps: text('mcps'), // JSON array of MCP tool IDs
-  allowed_tools: text('allowed_tools'), // JSON array of allowed tool IDs (whitelist)
-
-  configuration: text('configuration'), // JSON, extensible settings
-
-  created_at: text('created_at').notNull(),
-  updated_at: text('updated_at').notNull()
-})
-
-// Foreign keys for sessions table
-export const sessionsFkAgent = foreignKey({
-  columns: [sessionsTable.agent_id],
-  foreignColumns: [agentsTable.id],
-  name: 'fk_session_agent_id'
-}).onDelete('cascade')
-
-// Indexes for sessions table
-export const sessionsCreatedAtIdx = index('idx_sessions_created_at').on(sessionsTable.created_at)
-export const sessionsMainAgentIdIdx = index('idx_sessions_agent_id').on(sessionsTable.agent_id)
-export const sessionsModelIdx = index('idx_sessions_model').on(sessionsTable.model)
-
-export type SessionRow = typeof sessionsTable.$inferSelect
-export type InsertSessionRow = typeof sessionsTable.$inferInsert

src/main/services/agents/database/sessionMessageRepository.ts

@@ -1,181 +0,0 @@
-import { loggerService } from '@logger'
-import type {
-  AgentMessageAssistantPersistPayload,
-  AgentMessagePersistExchangePayload,
-  AgentMessagePersistExchangeResult,
-  AgentMessageUserPersistPayload,
-  AgentPersistedMessage,
-  AgentSessionMessageEntity
-} from '@types'
-
-import { BaseService } from '../BaseService'
-import type { InsertSessionMessageRow } from './schema'
-import { sessionMessagesTable } from './schema'
-
-const logger = loggerService.withContext('AgentMessageRepository')
-
-type TxClient = any
-
-export type PersistUserMessageParams = AgentMessageUserPersistPayload & {
-  sessionId: string
-  agentSessionId?: string
-  tx?: TxClient
-}
-
-export type PersistAssistantMessageParams = AgentMessageAssistantPersistPayload & {
-  sessionId: string
-  agentSessionId: string
-  tx?: TxClient
-}
-
-type PersistExchangeParams = AgentMessagePersistExchangePayload & {
-  tx?: TxClient
-}
-
-type PersistExchangeResult = AgentMessagePersistExchangeResult
-
-class AgentMessageRepository extends BaseService {
-  private static instance: AgentMessageRepository | null = null
-
-  static getInstance(): AgentMessageRepository {
-    if (!AgentMessageRepository.instance) {
-      AgentMessageRepository.instance = new AgentMessageRepository()
-    }
-
-    return AgentMessageRepository.instance
-  }
-
-  private serializeMessage(payload: AgentPersistedMessage): string {
-    return JSON.stringify(payload)
-  }
-
-  private serializeMetadata(metadata?: Record<string, unknown>): string | undefined {
-    if (!metadata) {
-      return undefined
-    }
-
-    try {
-      return JSON.stringify(metadata)
-    } catch (error) {
-      logger.warn('Failed to serialize session message metadata', error as Error)
-      return undefined
-    }
-  }
-
-  private deserialize(row: any): AgentSessionMessageEntity {
-    if (!row) return row
-
-    const deserialized = { ...row }
-
-    if (typeof deserialized.content === 'string') {
-      try {
-        deserialized.content = JSON.parse(deserialized.content)
-      } catch (error) {
-        logger.warn('Failed to parse session message content JSON', error as Error)
-      }
-    }
-
-    if (typeof deserialized.metadata === 'string') {
-      try {
-        deserialized.metadata = JSON.parse(deserialized.metadata)
-      } catch (error) {
-        logger.warn('Failed to parse session message metadata JSON', error as Error)
-      }
-    }
-
-    return deserialized
-  }
-
-  private getWriter(tx?: TxClient): TxClient {
-    return tx ?? this.database
-  }
-
-  async persistUserMessage(params: PersistUserMessageParams): Promise<AgentSessionMessageEntity> {
-    await AgentMessageRepository.initialize()
-    this.ensureInitialized()
-
-    const writer = this.getWriter(params.tx)
-    const now = params.createdAt ?? params.payload.message.createdAt ?? new Date().toISOString()
-
-    const insertData: InsertSessionMessageRow = {
-      session_id: params.sessionId,
-      role: params.payload.message.role,
-      content: this.serializeMessage(params.payload),
-      agent_session_id: params.agentSessionId ?? '',
-      metadata: this.serializeMetadata(params.metadata),
-      created_at: now,
-      updated_at: now
-    }
-
-    const [saved] = await writer.insert(sessionMessagesTable).values(insertData).returning()
-
-    return this.deserialize(saved)
-  }
-
-  async persistAssistantMessage(params: PersistAssistantMessageParams): Promise<AgentSessionMessageEntity> {
-    await AgentMessageRepository.initialize()
-    this.ensureInitialized()
-
-    const writer = this.getWriter(params.tx)
-    const now = params.createdAt ?? params.payload.message.createdAt ?? new Date().toISOString()
-
-    const insertData: InsertSessionMessageRow = {
-      session_id: params.sessionId,
-      role: params.payload.message.role,
-      content: this.serializeMessage(params.payload),
-      agent_session_id: params.agentSessionId,
-      metadata: this.serializeMetadata(params.metadata),
-      created_at: now,
-      updated_at: now
-    }
-
-    const [saved] = await writer.insert(sessionMessagesTable).values(insertData).returning()
-
-    return this.deserialize(saved)
-  }
-
-  async persistExchange(params: PersistExchangeParams): Promise<PersistExchangeResult> {
-    await AgentMessageRepository.initialize()
-    this.ensureInitialized()
-
-    const { sessionId, agentSessionId, user, assistant } = params
-
-    const result = await this.database.transaction(async (tx) => {
-      const exchangeResult: PersistExchangeResult = {}
-
-      if (user?.payload) {
-        if (!user.payload.message?.role) {
-          throw new Error('User message payload missing role')
-        }
-        exchangeResult.userMessage = await this.persistUserMessage({
-          sessionId,
-          agentSessionId,
-          payload: user.payload,
-          metadata: user.metadata,
-          createdAt: user.createdAt,
-          tx
-        })
-      }
-
-      if (assistant?.payload) {
-        if (!assistant.payload.message?.role) {
-          throw new Error('Assistant message payload missing role')
-        }
-        exchangeResult.assistantMessage = await this.persistAssistantMessage({
-          sessionId,
-          agentSessionId,
-          payload: assistant.payload,
-          metadata: assistant.metadata,
-          createdAt: assistant.createdAt,
-          tx
-        })
-      }
-
-      return exchangeResult
-    })
-
-    return result
-  }
-}
-
-export const agentMessageRepository = AgentMessageRepository.getInstance()

src/main/services/agents/drizzle.config.ts

@@ -1,31 +0,0 @@
-/**
- * Drizzle Kit configuration for agents database
- */
-
-import os from 'node:os'
-import path from 'node:path'
-
-import { defineConfig } from 'drizzle-kit'
-import { app } from 'electron'
-
-function getDbPath() {
-  if (process.env.NODE_ENV === 'development') {
-    return path.join(os.homedir(), '.cherrystudio', 'data', 'agents.db')
-  }
-  return path.join(app.getPath('userData'), 'agents.db')
-}
-
-const resolvedDbPath = getDbPath()
-
-export const dbPath = resolvedDbPath
-
-export default defineConfig({
-  dialect: 'sqlite',
-  schema: './src/main/services/agents/database/schema/index.ts',
-  out: './resources/database/drizzle',
-  dbCredentials: {
-    url: `file:${resolvedDbPath}`
-  },
-  verbose: true,
-  strict: true
-})

src/main/services/agents/errors.ts

@@ -1,23 +0,0 @@
-import { ModelValidationError } from '@main/apiServer/utils'
-import { AgentType } from '@types'
-
-export type AgentModelField = 'model' | 'plan_model' | 'small_model'
-
-export interface AgentModelValidationContext {
-  agentType: AgentType
-  field: AgentModelField
-  model?: string
-}
-
-export class AgentModelValidationError extends Error {
-  readonly context: AgentModelValidationContext
-  readonly detail: ModelValidationError
-
-  constructor(context: AgentModelValidationContext, detail: ModelValidationError) {
-    super(`Validation failed for ${context.agentType}.${context.field}: ${detail.message}`)
-    this.name = 'AgentModelValidationError'
-    this.context = context
-    this.detail = detail
-  }
-}
-

src/main/services/agents/index.ts

@@ -1,25 +0,0 @@
-/**
- * Agents Service Module
- *
- * This module provides a complete autonomous agent management system with:
- * - Agent lifecycle management (CRUD operations)
- * - Session handling with conversation history
- * - Comprehensive logging and audit trails
- * - Database operations with Drizzle ORM and migration support
- * - RESTful API endpoints for external integration
- */
-
-// === Core Services ===
-// Main service classes and singleton instances
-export * from './services'
-
-// === Error Types ===
-export { type AgentModelField, AgentModelValidationError } from './errors'
-
-// === Base Infrastructure ===
-// Shared database utilities and base service class
-export { BaseService } from './BaseService'
-
-// === Database Layer ===
-// Drizzle ORM schemas, migrations, and database utilities
-export * as Database from './database'

src/main/services/agents/interfaces/AgentStreamInterface.ts

@@ -1,31 +0,0 @@
-// Agent-agnostic streaming interface
-// This interface should be implemented by all agent services
-
-import { EventEmitter } from 'node:events'
-
-import { GetAgentSessionResponse } from '@types'
-import type { TextStreamPart } from 'ai'
-
-// Generic agent stream event that works with any agent type
-export interface AgentStreamEvent {
-  type: 'chunk' | 'error' | 'complete' | 'cancelled'
-  chunk?: TextStreamPart<any> // Standard AI SDK chunk for UI consumption
-  error?: Error
-}
-
-// Agent stream interface that all agents should implement
-export interface AgentStream extends EventEmitter {
-  emit(event: 'data', data: AgentStreamEvent): boolean
-  on(event: 'data', listener: (data: AgentStreamEvent) => void): this
-  once(event: 'data', listener: (data: AgentStreamEvent) => void): this
-}
-
-// Base agent service interface
-export interface AgentServiceInterface {
-  invoke(
-    prompt: string,
-    session: GetAgentSessionResponse,
-    abortController: AbortController,
-    lastAgentSessionId?: string
-  ): Promise<AgentStream>
-}

src/main/services/agents/services/AgentService.ts

@@ -1,201 +0,0 @@
-import path from 'node:path'
-
-import { getDataPath } from '@main/utils'
-import {
-  AgentBaseSchema,
-  AgentEntity,
-  CreateAgentRequest,
-  CreateAgentResponse,
-  GetAgentResponse,
-  ListOptions,
-  UpdateAgentRequest,
-  UpdateAgentResponse
-} from '@types'
-import { count, eq } from 'drizzle-orm'
-
-import { BaseService } from '../BaseService'
-import { type AgentRow, agentsTable, type InsertAgentRow } from '../database/schema'
-import { AgentModelField } from '../errors'
-import { builtinTools } from './claudecode/tools'
-
-export class AgentService extends BaseService {
-  private static instance: AgentService | null = null
-  private readonly modelFields: AgentModelField[] = ['model', 'plan_model', 'small_model']
-
-  static getInstance(): AgentService {
-    if (!AgentService.instance) {
-      AgentService.instance = new AgentService()
-    }
-    return AgentService.instance
-  }
-
-  async initialize(): Promise<void> {
-    await BaseService.initialize()
-  }
-
-  // Agent Methods
-  async createAgent(req: CreateAgentRequest): Promise<CreateAgentResponse> {
-    this.ensureInitialized()
-
-    const id = `agent_${Date.now()}_${Math.random().toString(36).substring(2, 11)}`
-    const now = new Date().toISOString()
-
-    if (!req.accessible_paths || req.accessible_paths.length === 0) {
-      const defaultPath = path.join(getDataPath(), 'agents', id)
-      req.accessible_paths = [defaultPath]
-    }
-
-    if (req.accessible_paths !== undefined) {
-      req.accessible_paths = this.ensurePathsExist(req.accessible_paths)
-    }
-
-    await this.validateAgentModels(req.type, {
-      model: req.model,
-      plan_model: req.plan_model,
-      small_model: req.small_model
-    })
-
-    const serializedReq = this.serializeJsonFields(req)
-
-    const insertData: InsertAgentRow = {
-      id,
-      type: req.type,
-      name: req.name || 'New Agent',
-      description: req.description,
-      instructions: req.instructions || 'You are a helpful assistant.',
-      model: req.model,
-      plan_model: req.plan_model,
-      small_model: req.small_model,
-      configuration: serializedReq.configuration,
-      accessible_paths: serializedReq.accessible_paths,
-      created_at: now,
-      updated_at: now
-    }
-
-    await this.database.insert(agentsTable).values(insertData)
-    const result = await this.database.select().from(agentsTable).where(eq(agentsTable.id, id)).limit(1)
-    if (!result[0]) {
-      throw new Error('Failed to create agent')
-    }
-
-    const agent = this.deserializeJsonFields(result[0]) as AgentEntity
-    return agent
-  }
-
-  async getAgent(id: string): Promise<GetAgentResponse | null> {
-    this.ensureInitialized()
-
-    const result = await this.database.select().from(agentsTable).where(eq(agentsTable.id, id)).limit(1)
-
-    if (!result[0]) {
-      return null
-    }
-
-    const agent = this.deserializeJsonFields(result[0]) as GetAgentResponse
-    if (agent.type === 'claude-code') {
-      agent.built_in_tools = builtinTools
-    }
-
-    return agent
-  }
-
-  async listAgents(options: ListOptions = {}): Promise<{ agents: AgentEntity[]; total: number }> {
-    this.ensureInitialized() // Build query with pagination
-
-    const totalResult = await this.database.select({ count: count() }).from(agentsTable)
-
-    const baseQuery = this.database.select().from(agentsTable).orderBy(agentsTable.created_at)
-
-    const result =
-      options.limit !== undefined
-        ? options.offset !== undefined
-          ? await baseQuery.limit(options.limit).offset(options.offset)
-          : await baseQuery.limit(options.limit)
-        : await baseQuery
-
-    const agents = result.map((row) => this.deserializeJsonFields(row)) as GetAgentResponse[]
-
-    agents.forEach((agent) => {
-      if (agent.type === 'claude-code') {
-        agent.built_in_tools = builtinTools
-      }
-    })
-
-    return { agents, total: totalResult[0].count }
-  }
-
-  async updateAgent(
-    id: string,
-    updates: UpdateAgentRequest,
-    options: { replace?: boolean } = {}
-  ): Promise<UpdateAgentResponse | null> {
-    this.ensureInitialized()
-
-    // Check if agent exists
-    const existing = await this.getAgent(id)
-    if (!existing) {
-      return null
-    }
-
-    const now = new Date().toISOString()
-
-    if (updates.accessible_paths !== undefined) {
-      updates.accessible_paths = this.ensurePathsExist(updates.accessible_paths)
-    }
-
-    const modelUpdates: Partial<Record<AgentModelField, string | undefined>> = {}
-    for (const field of this.modelFields) {
-      if (Object.prototype.hasOwnProperty.call(updates, field)) {
-        modelUpdates[field] = updates[field as keyof UpdateAgentRequest] as string | undefined
-      }
-    }
-
-    if (Object.keys(modelUpdates).length > 0) {
-      await this.validateAgentModels(existing.type, modelUpdates)
-    }
-
-    const serializedUpdates = this.serializeJsonFields(updates)
-
-    const updateData: Partial<AgentRow> = {
-      updated_at: now
-    }
-    const replaceableFields = Object.keys(AgentBaseSchema.shape) as (keyof AgentRow)[]
-    const shouldReplace = options.replace ?? false
-
-    for (const field of replaceableFields) {
-      if (shouldReplace || Object.prototype.hasOwnProperty.call(serializedUpdates, field)) {
-        if (Object.prototype.hasOwnProperty.call(serializedUpdates, field)) {
-          const value = serializedUpdates[field as keyof typeof serializedUpdates]
-          ;(updateData as Record<string, unknown>)[field] = value ?? null
-        } else if (shouldReplace) {
-          ;(updateData as Record<string, unknown>)[field] = null
-        }
-      }
-    }
-
-    await this.database.update(agentsTable).set(updateData).where(eq(agentsTable.id, id))
-    return await this.getAgent(id)
-  }
-
-  async deleteAgent(id: string): Promise<boolean> {
-    this.ensureInitialized()
-
-    const result = await this.database.delete(agentsTable).where(eq(agentsTable.id, id))
-
-    return result.rowsAffected > 0
-  }
-
-  async agentExists(id: string): Promise<boolean> {
-    this.ensureInitialized()
-
-    const result = await this.database
-      .select({ id: agentsTable.id })
-      .from(agentsTable)
-      .where(eq(agentsTable.id, id))
-      .limit(1)
-
-    return result.length > 0
-  }
-}
-
-export const agentService = AgentService.getInstance()

src/main/services/agents/services/SessionMessageService.ts

@@ -1,329 +0,0 @@
-import { loggerService } from '@logger'
-import type {
-  AgentSessionMessageEntity,
-  CreateSessionMessageRequest,
-  GetAgentSessionResponse,
-  ListOptions
-} from '@types'
-import { ModelMessage, TextStreamPart } from 'ai'
-import { desc, eq } from 'drizzle-orm'
-
-import { BaseService } from '../BaseService'
-import { sessionMessagesTable } from '../database/schema'
-import { AgentStreamEvent } from '../interfaces/AgentStreamInterface'
-import ClaudeCodeService from './claudecode'
-
-const logger = loggerService.withContext('SessionMessageService')
-
-type SessionStreamResult = {
-  stream: ReadableStream<TextStreamPart<Record<string, any>>>
-  completion: Promise<{
-    userMessage?: AgentSessionMessageEntity
-    assistantMessage?: AgentSessionMessageEntity
-  }>
-}
-
-// Ensure errors emitted through SSE are serializable
-function serializeError(error: unknown): { message: string; name?: string; stack?: string } {
-  if (error instanceof Error) {
-    return {
-      message: error.message,
-      name: error.name,
-      stack: error.stack
-    }
-  }
-
-  if (typeof error === 'string') {
-    return { message: error }
-  }
-
-  return {
-    message: 'Unknown error'
-  }
-}
-
-class TextStreamAccumulator {
-  private textBuffer = ''
-  private totalText = ''
-  private readonly toolCalls = new Map<string, { toolName?: string; input?: unknown }>()
-  private readonly toolResults = new Map<string, unknown>()
-
-  add(part: TextStreamPart<Record<string, any>>): void {
-    switch (part.type) {
-      case 'text-start':
-        this.textBuffer = ''
-        break
-      case 'text-delta':
-        if (part.text) {
-          this.textBuffer += part.text
-        }
-        break
-      case 'text-end': {
-        const blockText = (part.providerMetadata?.text?.value as string | undefined) ?? this.textBuffer
-        if (blockText) {
-          this.totalText += blockText
-        }
-        this.textBuffer = ''
-        break
-      }
-      case 'tool-call':
-        if (part.toolCallId) {
-          this.toolCalls.set(part.toolCallId, {
-            toolName: part.toolName,
-            input: part.input ?? part.args ?? part.providerMetadata?.raw?.input
-          })
-        }
-        break
-      case 'tool-result':
-        if (part.toolCallId) {
-          this.toolResults.set(part.toolCallId, part.output ?? part.result ?? part.providerMetadata?.raw)
-        }
-        break
-      default:
-        break
-    }
-  }
-
-  toModelMessage(role: ModelMessage['role'] = 'assistant'): ModelMessage {
-    const content = this.totalText || this.textBuffer || ''
-
-    const toolInvocations = Array.from(this.toolCalls.entries()).map(([toolCallId, info]) => ({
-      toolCallId,
-      toolName: info.toolName,
-      args: info.input,
-      result: this.toolResults.get(toolCallId)
-    }))
-
-    const message: Record<string, unknown> = {
-      role,
-      content
-    }
-
-    if (toolInvocations.length > 0) {
-      message.toolInvocations = toolInvocations
-    }
-
-    return message as ModelMessage
-  }
-}
-
-export class SessionMessageService extends BaseService {
-  private static instance: SessionMessageService | null = null
-  private cc: ClaudeCodeService = new ClaudeCodeService()
-
-  static getInstance(): SessionMessageService {
-    if (!SessionMessageService.instance) {
-      SessionMessageService.instance = new SessionMessageService()
-    }
-    return SessionMessageService.instance
-  }
-
-  async initialize(): Promise<void> {
-    await BaseService.initialize()
-  }
-
-  async sessionMessageExists(id: number): Promise<boolean> {
-    this.ensureInitialized()
-
-    const result = await this.database
-      .select({ id: sessionMessagesTable.id })
-      .from(sessionMessagesTable)
-      .where(eq(sessionMessagesTable.id, id))
-      .limit(1)
-
-    return result.length > 0
-  }
-
-  async listSessionMessages(
-    sessionId: string,
-    options: ListOptions = {}
-  ): Promise<{ messages: AgentSessionMessageEntity[] }> {
-    this.ensureInitialized()
-
-    // Get messages with pagination
-    const baseQuery = this.database
-      .select()
-      .from(sessionMessagesTable)
-      .where(eq(sessionMessagesTable.session_id, sessionId))
-      .orderBy(sessionMessagesTable.created_at)
-
-    const result =
-      options.limit !== undefined
-        ? options.offset !== undefined
-          ? await baseQuery.limit(options.limit).offset(options.offset)
-          : await baseQuery.limit(options.limit)
-        : await baseQuery
-
-    const messages = result.map((row) => this.deserializeSessionMessage(row)) as AgentSessionMessageEntity[]
-
-    return { messages }
-  }
-
-  async createSessionMessage(
-    session: GetAgentSessionResponse,
-    messageData: CreateSessionMessageRequest,
-    abortController: AbortController
-  ): Promise<SessionStreamResult> {
-    this.ensureInitialized()
-
-    return await this.startSessionMessageStream(session, messageData, abortController)
-  }
-
-  private async startSessionMessageStream(
-    session: GetAgentSessionResponse,
-    req: CreateSessionMessageRequest,
-    abortController: AbortController
-  ): Promise<SessionStreamResult> {
-    const agentSessionId = await this.getLastAgentSessionId(session.id)
-    let newAgentSessionId = ''
-    logger.debug('Session Message stream message data:', { message: req, session_id: agentSessionId })
-
-    if (session.agent_type !== 'claude-code') {
-      // TODO: Implement support for other agent types
-      logger.error('Unsupported agent type for streaming:', { agent_type: session.agent_type })
-      throw new Error('Unsupported agent type for streaming')
-    }
-
-    const claudeStream = await this.cc.invoke(req.content, session, abortController, agentSessionId)
-    const accumulator = new TextStreamAccumulator()
-
-    let resolveCompletion!: (value: {
-      userMessage?: AgentSessionMessageEntity
-      assistantMessage?: AgentSessionMessageEntity
-    }) => void
-    let rejectCompletion!: (reason?: unknown) => void
-
-    const completion = new Promise<{
-      userMessage?: AgentSessionMessageEntity
-      assistantMessage?: AgentSessionMessageEntity
-    }>((resolve, reject) => {
-      resolveCompletion = resolve
-      rejectCompletion = reject
-    })
-
-    let finished = false
-
-    const cleanup = () => {
-      if (finished) return
-      finished = true
-      claudeStream.removeAllListeners()
-    }
-
-    const stream = new ReadableStream<TextStreamPart<Record<string, any>>>({
-      start: (controller) => {
-        claudeStream.on('data', async (event: AgentStreamEvent) => {
-          if (finished) return
-          try {
-            switch (event.type) {
-              case 'chunk': {
-                const chunk = event.chunk as TextStreamPart<Record<string, any>> | undefined
-                if (!chunk) {
-                  logger.warn('Received agent chunk event without chunk payload')
-                  return
-                }
-
-                if (chunk.type === 'start' && chunk.messageId) {
-                  newAgentSessionId = chunk.messageId
-                }
-
-                accumulator.add(chunk)
-                controller.enqueue(chunk)
-                break
-              }
-
-              case 'error': {
-                const stderrMessage = (event as any)?.data?.stderr as string | undefined
-                const underlyingError = event.error ?? (stderrMessage ? new Error(stderrMessage) : undefined)
-                cleanup()
-                const streamError = underlyingError ?? new Error('Stream error')
-                controller.error(streamError)
-                rejectCompletion(serializeError(streamError))
-                break
-              }
-
-              case 'complete': {
-                cleanup()
-                controller.close()
-                resolveCompletion({})
-                break
-              }
-
-              case 'cancelled': {
-                cleanup()
-                controller.close()
-                resolveCompletion({})
-                break
-              }
-
-              default:
-                logger.warn('Unknown event type from Claude Code service:', {
-                  type: event.type
-                })
-                break
-            }
-          } catch (error) {
-            cleanup()
-            controller.error(error)
-            rejectCompletion(serializeError(error))
-          }
-        })
-      },
-      cancel: (reason) => {
-        cleanup()
-        abortController.abort(typeof reason === 'string' ? reason : 'stream cancelled')
-        resolveCompletion({})
-      }
-    })
-
-    return { stream, completion }
-  }
-
-  private async getLastAgentSessionId(sessionId: string): Promise<string> {
-    this.ensureInitialized()
-
-    try {
-      const result = await this.database
-        .select({ agent_session_id: sessionMessagesTable.agent_session_id })
-        .from(sessionMessagesTable)
-        .where(eq(sessionMessagesTable.session_id, sessionId))
-        .orderBy(desc(sessionMessagesTable.created_at))
-        .limit(1)
-
-      return result[0]?.agent_session_id || ''
-    } catch (error) {
-      logger.error('Failed to get last agent session ID', {
-        sessionId,
-        error
-      })
-      return ''
-    }
-  }
-
-  private deserializeSessionMessage(data: any): AgentSessionMessageEntity {
-    if (!data) return data
-
-    const deserialized = { ...data }
-
-    // Parse content JSON
-    if (deserialized.content && typeof deserialized.content === 'string') {
-      try {
-        deserialized.content = JSON.parse(deserialized.content)
-      } catch (error) {
-        logger.warn(`Failed to parse content JSON:`, error as Error)
-      }
-    }
-
-    // Parse metadata JSON
-    if (deserialized.metadata && typeof deserialized.metadata === 'string') {
-      try {
-        deserialized.metadata = JSON.parse(deserialized.metadata)
-      } catch (error) {
-        logger.warn(`Failed to parse metadata JSON:`, error as Error)
-      }
-    }
-
-    return deserialized
-  }
-}
-
-export const sessionMessageService = SessionMessageService.getInstance()

src/main/services/agents/services/SessionService.ts

@@ -1,241 +0,0 @@
-import {
-  AgentBaseSchema,
-  type AgentEntity,
-  type AgentSessionEntity,
-  type CreateSessionRequest,
-  type CreateSessionResponse,
-  type GetAgentSessionResponse,
-  type ListOptions,
-  type UpdateSessionRequest,
-  UpdateSessionResponse
-} from '@types'
-import { and, count, eq, type SQL } from 'drizzle-orm'
-
-import { BaseService } from '../BaseService'
-import { agentsTable, type InsertSessionRow, type SessionRow, sessionsTable } from '../database/schema'
-import { AgentModelField } from '../errors'
-
-export class SessionService extends BaseService {
-  private static instance: SessionService | null = null
-  private readonly modelFields: AgentModelField[] = ['model', 'plan_model', 'small_model']
-
-  static getInstance(): SessionService {
-    if (!SessionService.instance) {
-      SessionService.instance = new SessionService()
-    }
-    return SessionService.instance
-  }
-
-  async initialize(): Promise<void> {
-    await BaseService.initialize()
-  }
-
-  async createSession(agentId: string, req: CreateSessionRequest): Promise<CreateSessionResponse> {
-    this.ensureInitialized()
-
-    // Validate agent exists - we'll need to import AgentService for this check
-    // For now, we'll skip this validation to avoid circular dependencies
-    // The database foreign key constraint will handle this
-
-    const agents = await this.database.select().from(agentsTable).where(eq(agentsTable.id, agentId)).limit(1)
-    if (!agents[0]) {
-      throw new Error('Agent not found')
-    }
-    const agent = this.deserializeJsonFields(agents[0]) as AgentEntity
-
-    const id = `session_${Date.now()}_${Math.random().toString(36).substring(2, 11)}`
-    const now = new Date().toISOString()
-
-    // inherit configuration from agent by default, can be overridden by sessionData
-    const sessionData: Partial<CreateSessionRequest> = {
-      ...agent,
-      ...req
-    }
-
-    await this.validateAgentModels(agent.type, {
-      model: sessionData.model,
-      plan_model: sessionData.plan_model,
-      small_model: sessionData.small_model
-    })
-
-    if (sessionData.accessible_paths !== undefined) {
-      sessionData.accessible_paths = this.ensurePathsExist(sessionData.accessible_paths)
-    }
-
-    const serializedData = this.serializeJsonFields(sessionData)
-
-    const insertData: InsertSessionRow = {
-      id,
-      agent_id: agentId,
-      agent_type: agent.type,
-      name: serializedData.name || null,
-      description: serializedData.description || null,
-      accessible_paths: serializedData.accessible_paths || null,
-      instructions: serializedData.instructions || null,
-      model: serializedData.model || null,
-      plan_model: serializedData.plan_model || null,
-      small_model: serializedData.small_model || null,
-      mcps: serializedData.mcps || null,
-      configuration: serializedData.configuration || null,
-      created_at: now,
-      updated_at: now
-    }
-
-    await this.database.insert(sessionsTable).values(insertData)
-
-    const result = await this.database.select().from(sessionsTable).where(eq(sessionsTable.id, id)).limit(1)
-
-    if (!result[0]) {
-      throw new Error('Failed to create session')
-    }
-
-    return this.deserializeJsonFields(result[0]) as AgentSessionEntity
-  }
-
-  async getSession(agentId: string, id: string): Promise<GetAgentSessionResponse | null> {
-    this.ensureInitialized()
-
-    const result = await this.database
-      .select()
-      .from(sessionsTable)
-      .where(and(eq(sessionsTable.id, id), eq(sessionsTable.agent_id, agentId)))
-      .limit(1)
-
-    if (!result[0]) {
-      return null
-    }
-
-    const session = this.deserializeJsonFields(result[0]) as GetAgentSessionResponse
-
-    return session
-  }
-
-  async getSessionById(id: string): Promise<GetAgentSessionResponse | null> {
-    this.ensureInitialized()
-
-    const result = await this.database.select().from(sessionsTable).where(eq(sessionsTable.id, id)).limit(1)
-
-    if (!result[0]) {
-      return null
-    }
-
-    const session = this.deserializeJsonFields(result[0]) as GetAgentSessionResponse
-
-    return session
-  }
-
-  async listSessions(
-    agentId?: string,
-    options: ListOptions = {}
-  ): Promise<{ sessions: AgentSessionEntity[]; total: number }> {
-    this.ensureInitialized()
-
-    // Build where conditions
-    const whereConditions: SQL[] = []
-    if (agentId) {
-      whereConditions.push(eq(sessionsTable.agent_id, agentId))
-    }
-
-    const whereClause =
-      whereConditions.length > 1
-        ? and(...whereConditions)
-        : whereConditions.length === 1
-          ? whereConditions[0]
-          : undefined
-
-    // Get total count
-    const totalResult = await this.database.select({ count: count() }).from(sessionsTable).where(whereClause)
-
-    const total = totalResult[0].count
-
-    // Build list query with pagination
-    const baseQuery = this.database.select().from(sessionsTable).where(whereClause).orderBy(sessionsTable.created_at)
-
-    const result =
-      options.limit !== undefined
-        ? options.offset !== undefined
-          ? await baseQuery.limit(options.limit).offset(options.offset)
-          : await baseQuery.limit(options.limit)
-        : await baseQuery
-
-    const sessions = result.map((row) => this.deserializeJsonFields(row)) as GetAgentSessionResponse[]
-
-    return { sessions, total }
-  }
-
-  async updateSession(
-    agentId: string,
-    id: string,
-    updates: UpdateSessionRequest
-  ): Promise<UpdateSessionResponse | null> {
-    this.ensureInitialized()
-
-    // Check if session exists
-    const existing = await this.getSession(agentId, id)
-    if (!existing) {
-      return null
-    }
-
-    // Validate agent exists if changing main_agent_id
-    // We'll skip this validation for now to avoid circular dependencies
-
-    const now = new Date().toISOString()
-
-    if (updates.accessible_paths !== undefined) {
-      updates.accessible_paths = this.ensurePathsExist(updates.accessible_paths)
-    }
-
-    const modelUpdates: Partial<Record<AgentModelField, string | undefined>> = {}
-    for (const field of this.modelFields) {
-      if (Object.prototype.hasOwnProperty.call(updates, field)) {
-        modelUpdates[field] = updates[field as keyof UpdateSessionRequest] as string | undefined
-      }
-    }
-
-    if (Object.keys(modelUpdates).length > 0) {
-      await this.validateAgentModels(existing.agent_type, modelUpdates)
-    }
-
-    const serializedUpdates = this.serializeJsonFields(updates)
-
-    const updateData: Partial<SessionRow> = {
-      updated_at: now
-    }
-    const replaceableFields = Object.keys(AgentBaseSchema.shape) as (keyof SessionRow)[]
-
-    for (const field of replaceableFields) {
-      if (Object.prototype.hasOwnProperty.call(serializedUpdates, field)) {
-        const value = serializedUpdates[field as keyof typeof serializedUpdates]
-        ;(updateData as Record<string, unknown>)[field] = value ?? null
-      }
-    }
-
-    await this.database.update(sessionsTable).set(updateData).where(eq(sessionsTable.id, id))
-
-    return await this.getSession(agentId, id)
-  }
-
-  async deleteSession(agentId: string, id: string): Promise<boolean> {
-    this.ensureInitialized()
-
-    const result = await this.database
-      .delete(sessionsTable)
-      .where(and(eq(sessionsTable.id, id), eq(sessionsTable.agent_id, agentId)))
-
-    return result.rowsAffected > 0
-  }
-
-  async sessionExists(agentId: string, id: string): Promise<boolean> {
-    this.ensureInitialized()
-
-    const result = await this.database
-      .select({ id: sessionsTable.id })
-      .from(sessionsTable)
-      .where(and(eq(sessionsTable.id, id), eq(sessionsTable.agent_id, agentId)))
-      .limit(1)
-
-    return result.length > 0
-  }
-}
-
-export const sessionService = SessionService.getInstance()

src/main/services/agents/services/claudecode/index.ts

@@ -1,221 +0,0 @@
-// src/main/services/agents/services/claudecode/index.ts
-import { EventEmitter } from 'node:events'
-import { createRequire } from 'node:module'
-
-import { McpHttpServerConfig, Options, query, SDKMessage } from '@anthropic-ai/claude-code'
-import { loggerService } from '@logger'
-import { config as apiConfigService } from '@main/apiServer/config'
-import { validateModelId } from '@main/apiServer/utils'
-
-import { GetAgentSessionResponse } from '../..'
-import { AgentServiceInterface, AgentStream, AgentStreamEvent } from '../../interfaces/AgentStreamInterface'
-import { transformSDKMessageToStreamParts } from './transform'
-
-const require_ = createRequire(import.meta.url)
-const logger = loggerService.withContext('ClaudeCodeService')
-
-class ClaudeCodeStream extends EventEmitter implements AgentStream {
-  declare emit: (event: 'data', data: AgentStreamEvent) => boolean
-  declare on: (event: 'data', listener: (data: AgentStreamEvent) => void) => this
-  declare once: (event: 'data', listener: (data: AgentStreamEvent) => void) => this
-}
-
-class ClaudeCodeService implements AgentServiceInterface {
-  private claudeExecutablePath: string
-
-  constructor() {
-    // Resolve Claude Code CLI robustly (works in dev and in asar)
-    this.claudeExecutablePath = require_.resolve('@anthropic-ai/claude-code/cli.js')
-  }
-
-  async invoke(
-    prompt: string,
-    session: GetAgentSessionResponse,
-    abortController: AbortController,
-    lastAgentSessionId?: string
-  ): Promise<AgentStream> {
-    const aiStream = new ClaudeCodeStream()
-
-    // Validate session accessible paths and make sure it exists as a directory
-    const cwd = session.accessible_paths[0]
-    if (!cwd) {
-      aiStream.emit('data', {
-        type: 'error',
-        error: new Error('No accessible paths defined for the agent session')
-      })
-      return aiStream
-    }
-
-    // Validate model info
-    const modelInfo = await validateModelId(session.model)
-    if (!modelInfo.valid) {
-      aiStream.emit('data', {
-        type: 'error',
-        error: new Error(`Invalid model ID '${session.model}': ${JSON.stringify(modelInfo.error)}`)
-      })
-      return aiStream
-    }
-    if (modelInfo.provider?.type !== 'anthropic' || modelInfo.provider.apiKey === '') {
-      aiStream.emit('data', {
-        type: 'error',
-        error: new Error(`Invalid provider type '${modelInfo.provider?.type}'. Expected 'anthropic' provider type.`)
-      })
-      return aiStream
-    }
-
-    // TODO: use cherry studio api server config instead of direct provider config to provide more flexibility (e.g. custom headers, proxy, statistics, etc).
-    const apiConfig = await apiConfigService.get()
-    // process.env.ANTHROPIC_AUTH_TOKEN = apiConfig.apiKey
-    // process.env.ANTHROPIC_BASE_URL = `http://${apiConfig.host}:${apiConfig.port}`
-    process.env.ANTHROPIC_AUTH_TOKEN = modelInfo.provider.apiKey
-    process.env.ANTHROPIC_BASE_URL = modelInfo.provider.apiHost
-
-    // Build SDK options from parameters
-    const options: Options = {
-      abortController,
-      cwd,
-      pathToClaudeCodeExecutable: this.claudeExecutablePath,
-      stderr: (chunk: string) => {
-        logger.info('claude stderr', { chunk })
-      },
-      appendSystemPrompt: session.instructions,
-      permissionMode: session.configuration?.permission_mode,
-      maxTurns: session.configuration?.max_turns
-    }
-
-    if (session.accessible_paths.length > 1) {
-      options.additionalDirectories = session.accessible_paths.slice(1)
-    }
-
-    if (session.mcps && session.mcps.length > 0) {
-      // mcp configs
-      const mcpList: Record<string, McpHttpServerConfig> = {}
-      for (const mcpId of session.mcps) {
-        mcpList[mcpId] = {
-          type: 'http',
-          url: `http://${apiConfig.host}:${apiConfig.port}/v1/mcps/${mcpId}/mcp`,
-          headers: {
-            Authorization: `Bearer ${apiConfig.apiKey}`
-          }
-        }
-      }
-      options.mcpServers = mcpList
-      options.strictMcpConfig = true
-    }
-
-    if (lastAgentSessionId) {
-      options.resume = lastAgentSessionId
-    }
-
-    logger.info('Starting Claude Code SDK query', {
-      prompt,
-      options
-    })
-
-    // Start async processing
-    this.processSDKQuery(prompt, options, aiStream)
-
-    return aiStream
-  }
-
-  private async *userMessages(prompt: string) {
-    {
-      yield {
-        type: 'user' as const,
-        parent_tool_use_id: null,
-        session_id: '',
-        message: {
-          role: 'user' as const,
-          content: prompt
-        }
-      }
-    }
-  }
-
-  /**
-   * Process SDK query and emit stream events
-   */
-  private async processSDKQuery(prompt: string, options: Options, stream: ClaudeCodeStream): Promise<void> {
-    const jsonOutput: SDKMessage[] = []
-    let hasCompleted = false
-    const startTime = Date.now()
-
-    try {
-      // Process streaming responses using SDK query
-      for await (const message of query({
-        prompt: this.userMessages(prompt),
-        options
-      })) {
-        if (hasCompleted) break
-
-        jsonOutput.push(message)
-        logger.silly('claude response', { message })
-        if (message.type === 'assistant' || message.type === 'user') {
-          logger.silly('message content', {
-            message: JSON.stringify({ role: message.message.role, content: message.message.content })
-          })
-        }
-
-        // Transform SDKMessage to UIMessageChunks
-        const chunks = transformSDKMessageToStreamParts(message)
-        for (const chunk of chunks) {
-          stream.emit('data', {
-            type: 'chunk',
-            chunk
-          })
-        }
-      }
-
-      // Successfully completed
-      hasCompleted = true
-      const duration = Date.now() - startTime
-
-      logger.debug('SDK query completed successfully', {
-        duration,
-        messageCount: jsonOutput.length
-      })
-
-      // Emit completion event
-      stream.emit('data', {
-        type: 'complete'
-      })
-    } catch (error) {
-      if (hasCompleted) return
-      hasCompleted = true
-
-      const duration = Date.now() - startTime
-
-      // Check if this is an abort error
-      const errorObj = error as any
-      const isAborted =
-        errorObj?.name === 'AbortError' ||
-        errorObj?.message?.includes('aborted') ||
-        options.abortController?.signal.aborted
-
-      if (isAborted) {
-        logger.info('SDK query aborted by client disconnect', { duration })
-        // Simply cleanup and return - don't emit error events
-        stream.emit('data', {
-          type: 'cancelled',
-          error: new Error('Request aborted by client')
-        })
-        return
-      }
-
-      // Original error handling for non-abort errors
-      logger.error('SDK query error:', {
-        error: errorObj instanceof Error ? errorObj.message : String(errorObj),
-        duration,
-        messageCount: jsonOutput.length
-      })
-
-      // Emit error event
-      stream.emit('data', {
-        type: 'error',
-        error: errorObj instanceof Error ? errorObj : new Error(String(errorObj))
-      })
-    }
-  }
-}
-
-export default ClaudeCodeService

src/main/services/agents/services/claudecode/map-claude-code-finish-reason.ts

@@ -1,34 +0,0 @@
-// ported from https://github.com/ben-vargas/ai-sdk-provider-claude-code/blob/main/src/map-claude-code-finish-reason.ts#L22
-import type { LanguageModelV2FinishReason } from '@ai-sdk/provider'
-
-/**
- * Maps Claude Code SDK result subtypes to AI SDK finish reasons.
- *
- * @param subtype - The result subtype from Claude Code SDK
- * @returns The corresponding AI SDK finish reason
- *
- * @example
- * ```typescript
- * const finishReason = mapClaudeCodeFinishReason('error_max_turns');
- * // Returns: 'length'
- * ```
- *
- * @remarks
- * Mappings:
- * - 'success' -> 'stop' (normal completion)
- * - 'error_max_turns' -> 'length' (hit turn limit)
- * - 'error_during_execution' -> 'error' (execution error)
- * - default -> 'stop' (unknown subtypes treated as normal completion)
- */
-export function mapClaudeCodeFinishReason(subtype?: string): LanguageModelV2FinishReason {
-  switch (subtype) {
-    case 'success':
-      return 'stop'
-    case 'error_max_turns':
-      return 'length'
-    case 'error_during_execution':
-      return 'error'
-    default:
-      return 'stop'
-  }
-}

src/main/services/agents/services/claudecode/tools.ts

@@ -1,48 +0,0 @@
-import { Tool } from '@types'
-
-// https://docs.anthropic.com/en/docs/claude-code/settings#tools-available-to-claude
-export const builtinTools: Tool[] = [
-  { id: 'Bash', name: 'Bash', description: 'Executes shell commands in your environment', requirePermissions: true },
-  { id: 'Edit', name: 'Edit', description: 'Makes targeted edits to specific files', requirePermissions: true },
-  { id: 'Glob', name: 'Glob', description: 'Finds files based on pattern matching', requirePermissions: false },
-  { id: 'Grep', name: 'Grep', description: 'Searches for patterns in file contents', requirePermissions: false },
-  {
-    id: 'MultiEdit',
-    name: 'MultiEdit',
-    description: 'Performs multiple edits on a single file atomically',
-    requirePermissions: true
-  },
-  {
-    id: 'NotebookEdit',
-    name: 'NotebookEdit',
-    description: 'Modifies Jupyter notebook cells',
-    requirePermissions: true
-  },
-  {
-    id: 'NotebookRead',
-    name: 'NotebookRead',
-    description: 'Reads and displays Jupyter notebook contents',
-    requirePermissions: false
-  },
-  { id: 'Read', name: 'Read', description: 'Reads the contents of files', requirePermissions: false },
-  {
-    id: 'Task',
-    name: 'Task',
-    description: 'Runs a sub-agent to handle complex, multi-step tasks',
-    requirePermissions: false
-  },
-  {
-    id: 'TodoWrite',
-    name: 'TodoWrite',
-    description: 'Creates and manages structured task lists',
-    requirePermissions: false
-  },
-  { id: 'WebFetch', name: 'WebFetch', description: 'Fetches content from a specified URL', requirePermissions: true },
-  {
-    id: 'WebSearch',
-    name: 'WebSearch',
-    description: 'Performs web searches with domain filtering',
-    requirePermissions: true
-  },
-  { id: 'Write', name: 'Write', description: 'Creates or overwrites files', requirePermissions: true }
-]

src/main/services/agents/services/claudecode/transform.ts

@@ -1,354 +0,0 @@
-// This file is used to transform claude code json response to aisdk streaming format
-
-import type { LanguageModelV2Usage } from '@ai-sdk/provider'
-import { SDKMessage } from '@anthropic-ai/claude-code'
-import { loggerService } from '@logger'
-import type { ProviderMetadata, TextStreamPart } from 'ai'
-import { v4 as uuidv4 } from 'uuid'
-
-import { mapClaudeCodeFinishReason } from './map-claude-code-finish-reason'
-
-const logger = loggerService.withContext('ClaudeCodeTransform')
-
-type AgentStreamPart = TextStreamPart<Record<string, any>>
-
-const contentBlockState = new Map<
-  string,
-  {
-    type: 'text' | 'tool-call'
-    toolCallId?: string
-    toolName?: string
-    input?: string
-  }
->()
-
-// Helper function to generate unique IDs for text blocks
-const generateMessageId = (): string => `msg_${uuidv4().replace(/-/g, '')}`
-
-// Main transform function
-export function transformSDKMessageToStreamParts(sdkMessage: SDKMessage): AgentStreamPart[] {
-  const chunks: AgentStreamPart[] = []
-  logger.debug('Transforming SDKMessage to stream parts', sdkMessage)
-  switch (sdkMessage.type) {
-    case 'assistant':
-    case 'user':
-      chunks.push(...handleUserOrAssistantMessage(sdkMessage))
-      break
-
-    case 'stream_event':
-      chunks.push(...handleStreamEvent(sdkMessage))
-      break
-
-    case 'system':
-      chunks.push(...handleSystemMessage(sdkMessage))
-      break
-
-    case 'result':
-      chunks.push(...handleResultMessage(sdkMessage))
-      break
-
-    default:
-      logger.warn('Unknown SDKMessage type:', { type: (sdkMessage as any).type })
-      break
-  }
-
-  return chunks
-}
-
-const sdkMessageToProviderMetadata = (message: SDKMessage): ProviderMetadata => {
-  return {
-    anthropic: {
-      uuid: message.uuid || generateMessageId(),
-      session_id: message.session_id
-    },
-    raw: message as Record<string, any>
-  }
-}
-
-function generateTextChunks(id: string, text: string, message: SDKMessage): AgentStreamPart[] {
-  const providerMetadata = sdkMessageToProviderMetadata(message)
-  return [
-    {
-      type: 'text-start',
-      id,
-      providerMetadata
-    },
-    {
-      type: 'text-delta',
-      id,
-      text,
-      providerMetadata
-    },
-    {
-      type: 'text-end',
-      id,
-      providerMetadata: {
-        ...providerMetadata,
-        text: {
-          value: text
-        }
-      }
-    }
-  ]
-}
-
-function handleUserOrAssistantMessage(message: Extract<SDKMessage, { type: 'assistant' | 'user' }>): AgentStreamPart[] {
-  const chunks: AgentStreamPart[] = []
-  const messageId = message.uuid?.toString() || generateMessageId()
-
-  // handle normal text content
-  if (typeof message.message.content === 'string') {
-    const textContent = message.message.content
-    if (textContent) {
-      chunks.push(...generateTextChunks(messageId, textContent, message))
-    }
-  } else if (Array.isArray(message.message.content)) {
-    for (const block of message.message.content) {
-      switch (block.type) {
-        case 'text':
-          chunks.push(...generateTextChunks(messageId, block.text, message))
-          break
-        case 'tool_use':
-          chunks.push({
-            type: 'tool-call',
-            toolCallId: block.id,
-            toolName: block.name,
-            input: block.input,
-            providerExecuted: true,
-            providerMetadata: sdkMessageToProviderMetadata(message)
-          })
-          break
-        case 'tool_result':
-          // chunks.push({
-          //   type: 'tool-result',
-          //   toolCallId: block.tool_use_id,
-          //   output: block.content,
-          //   providerMetadata: sdkMessageToProviderMetadata(message)
-          // })
-          break
-        default:
-          logger.warn('Unknown content block type in user/assistant message:', {
-            type: block.type
-          })
-          break
-      }
-    }
-  }
-
-  return chunks
-}
-
-// Handle stream events (real-time streaming)
-function handleStreamEvent(message: Extract<SDKMessage, { type: 'stream_event' }>): AgentStreamPart[] {
-  const chunks: AgentStreamPart[] = []
-  const event = message.event
-  const blockKey = `${message.uuid ?? message.session_id ?? 'session'}:${event.index}`
-
-  switch (event.type) {
-    case 'message_start':
-      // No specific UI chunk needed for message start in this protocol
-      break
-
-    case 'content_block_start':
-      const contentBlockType = event.content_block.type
-      switch (contentBlockType) {
-        case 'text': {
-          contentBlockState.set(blockKey, { type: 'text' })
-          chunks.push({
-            type: 'text-start',
-            id: String(event.index),
-            providerMetadata: {
-              ...sdkMessageToProviderMetadata(message),
-              anthropic: {
-                uuid: message.uuid,
-                session_id: message.session_id,
-                content_block_index: event.index
-              }
-            }
-          })
-          break
-        }
-        case 'tool_use': {
-          contentBlockState.set(blockKey, {
-            type: 'tool-call',
-            toolCallId: event.content_block.id,
-            toolName: event.content_block.name,
-            input: ''
-          })
-          chunks.push({
-            type: 'tool-call',
-            toolCallId: event.content_block.id,
-            toolName: event.content_block.name,
-            input: event.content_block.input,
-            providerExecuted: true,
-            providerMetadata: sdkMessageToProviderMetadata(message)
-          })
-          break
-        }
-      }
-      break
-    case 'content_block_delta':
-      switch (event.delta.type) {
-        case 'text_delta': {
-          chunks.push({
-            type: 'text-delta',
-            id: String(event.index),
-            text: event.delta.text,
-            providerMetadata: {
-              ...sdkMessageToProviderMetadata(message),
-              anthropic: {
-                uuid: message.uuid,
-                session_id: message.session_id,
-                content_block_index: event.index
-              }
-            }
-          })
-          break
-        }
-        // case 'thinking_delta': {
-        //   chunks.push({
-        //     type: 'reasoning-delta',
-        //     id: String(event.index),
-        //     text: event.delta.thinking,
-        //   });
-        //   break
-        // }
-        // case 'signature_delta': {
-        //   if (blockType === 'thinking') {
-        //     chunks.push({
-        //       type: 'reasoning-delta',
-        //       id: String(event.index),
-        //       text: '',
-        //       providerMetadata: {
-        //         ...sdkMessageToProviderMetadata(message),
-        //         anthropic: {
-        //           uuid: message.uuid,
-        //           session_id: message.session_id,
-        //           content_block_index: event.index,
-        //           signature: event.delta.signature
-        //         }
-        //       }
-        //     })
-        //   }
-        //   break
-        // }
-        case 'input_json_delta': {
-          const contentBlock = contentBlockState.get(blockKey)
-          if (contentBlock && contentBlock.type === 'tool-call') {
-            contentBlockState.set(blockKey, {
-              ...contentBlock,
-              input: `${contentBlock.input ?? ''}${event.delta.partial_json ?? ''}`
-            })
-          }
-          break
-        }
-      }
-      break
-
-    case 'content_block_stop': {
-      const contentBlock = contentBlockState.get(blockKey)
-      if (contentBlock?.type === 'text') {
-        chunks.push({
-          type: 'text-end',
-          id: String(event.index)
-        })
-      }
-      contentBlockState.delete(blockKey)
-    }
-
-    case 'message_delta':
-      // Handle usage updates or other message-level deltas
-      break
-
-    case 'message_stop':
-      // This could signal the end of the message
-      break
-
-    default:
-      logger.warn('Unknown stream event type:', { type: (event as any).type })
-      break
-  }
-
-  return chunks
-}
-
-// Handle system messages
-function handleSystemMessage(message: Extract<SDKMessage, { type: 'system' }>): AgentStreamPart[] {
-  const chunks: AgentStreamPart[] = []
-  logger.debug('Received system message', {
-    subtype: message.subtype
-  })
-  switch (message.subtype) {
-    case 'init': {
-      chunks.push({
-        type: 'start'
-      })
-    }
-  }
-  return []
-}
-
-// Handle result messages (completion with usage stats)
-function handleResultMessage(message: Extract<SDKMessage, { type: 'result' }>): AgentStreamPart[] {
-  const chunks: AgentStreamPart[] = []
-
-  let usage: LanguageModelV2Usage | undefined
-  if ('usage' in message) {
-    usage = {
-      inputTokens:
-        (message.usage.cache_creation_input_tokens ?? 0) +
-        (message.usage.cache_read_input_tokens ?? 0) +
-        (message.usage.input_tokens ?? 0),
-      outputTokens: message.usage.output_tokens ?? 0,
-      totalTokens:
-        (message.usage.cache_creation_input_tokens ?? 0) +
-        (message.usage.cache_read_input_tokens ?? 0) +
-        (message.usage.input_tokens ?? 0) +
-        (message.usage.output_tokens ?? 0)
-    }
-  }
-  if (message.subtype === 'success') {
-    chunks.push({
-      type: 'finish',
-      totalUsage: usage,
-      finishReason: mapClaudeCodeFinishReason(message.subtype),
-      providerMetadata: {
-        ...sdkMessageToProviderMetadata(message),
-        usage: message.usage,
-        durationMs: message.duration_ms,
-        costUsd: message.total_cost_usd,
-        raw: message
-      }
-    } as AgentStreamPart)
-  } else {
-    chunks.push({
-      type: 'error',
-      error: {
-        message: `${message.subtype}: Process failed after ${message.num_turns} turns`
-      }
-    } as AgentStreamPart)
-  }
-  return chunks
-}
-
-// Convenience function to transform a stream of SDKMessages
-export function* transformSDKMessageStream(sdkMessages: SDKMessage[]): Generator<AgentStreamPart> {
-  for (const sdkMessage of sdkMessages) {
-    const chunks = transformSDKMessageToStreamParts(sdkMessage)
-    for (const chunk of chunks) {
-      yield chunk
-    }
-  }
-}
-
-// Async version for async iterables
-export async function* transformSDKMessageStreamAsync(
-  sdkMessages: AsyncIterable<SDKMessage>
-): AsyncGenerator<AgentStreamPart> {
-  for await (const sdkMessage of sdkMessages) {
-    const chunks = transformSDKMessageToStreamParts(sdkMessage)
-    for (const chunk of chunks) {
-      yield chunk
-    }
-  }
-}

src/main/services/agents/services/index.ts

@@ -1,26 +0,0 @@
-/**
- * Agent Services Module
- *
- * This module provides service classes for managing agents, sessions, and session messages.
- * All services extend BaseService and provide database operations with proper error handling.
- */
-
-// Service classes
-export { AgentService } from './AgentService'
-export { SessionMessageService } from './SessionMessageService'
-export { SessionService } from './SessionService'
-
-// Service instances (singletons)
-export { agentService } from './AgentService'
-export { sessionMessageService } from './SessionMessageService'
-export { sessionService } from './SessionService'
-
-// Type definitions for service requests and responses
-export type { AgentEntity, AgentSessionEntity, CreateAgentRequest, UpdateAgentRequest } from '@types'
-export type {
-  AgentSessionMessageEntity,
-  CreateSessionRequest,
-  GetAgentSessionResponse,
-  ListOptions as SessionListOptions,
-  UpdateSessionRequest
-} from '@types'

src/main/utils/ocr.ts

@@ -1,9 +1,8 @@
 import { ImageFileMetadata } from '@types'
 import { readFile } from 'fs/promises'
+import sharp from 'sharp'
 
 const preprocessImage = async (buffer: Buffer): Promise<Buffer> => {
-  // Delayed loading: The Sharp module is only loaded when the OCR functionality is actually needed, not at app startup
-  const sharp = (await import('sharp')).default
   return sharp(buffer)
     .grayscale() // 转为灰度
     .normalize()

src/preload/index.ts

@@ -1,7 +1,7 @@
 import { electronAPI } from '@electron-toolkit/preload'
 import { SpanEntity, TokenUsage } from '@mcp-trace/trace-core'
 import { SpanContext } from '@opentelemetry/api'
-import { UpgradeChannel } from '@shared/config/constant'
+import { TerminalConfig, UpgradeChannel } from '@shared/config/constant'
 import type { LogLevel, LogSourceWithContext } from '@shared/config/logger'
 import type { FileChangeEvent } from '@shared/config/types'
 import { IpcChannel } from '@shared/IpcChannel'
@@ -47,6 +47,7 @@ const api = {
   getDiskInfo: (directoryPath: string): Promise<{ free: number; size: number } | null> =>
     ipcRenderer.invoke(IpcChannel.App_GetDiskInfo, directoryPath),
   reload: () => ipcRenderer.invoke(IpcChannel.App_Reload),
+  quit: () => ipcRenderer.invoke(IpcChannel.App_Quit),
   setProxy: (proxy: string | undefined, bypassRules?: string) =>
     ipcRenderer.invoke(IpcChannel.App_Proxy, proxy, bypassRules),
   checkForUpdate: () => ipcRenderer.invoke(IpcChannel.App_CheckForUpdate),
@@ -439,16 +440,24 @@ const api = {
       model: string,
       directory: string,
       env: Record<string, string>,
-      options?: { autoUpdateToLatest?: boolean }
-    ) => ipcRenderer.invoke(IpcChannel.CodeTools_Run, cliTool, model, directory, env, options)
+      options?: { autoUpdateToLatest?: boolean; terminal?: string }
+    ) => ipcRenderer.invoke(IpcChannel.CodeTools_Run, cliTool, model, directory, env, options),
+    getAvailableTerminals: (): Promise<TerminalConfig[]> =>
+      ipcRenderer.invoke(IpcChannel.CodeTools_GetAvailableTerminals),
+    setCustomTerminalPath: (terminalId: string, path: string): Promise<void> =>
+      ipcRenderer.invoke(IpcChannel.CodeTools_SetCustomTerminalPath, terminalId, path),
+    getCustomTerminalPath: (terminalId: string): Promise<string | undefined> =>
+      ipcRenderer.invoke(IpcChannel.CodeTools_GetCustomTerminalPath, terminalId),
+    removeCustomTerminalPath: (terminalId: string): Promise<void> =>
+      ipcRenderer.invoke(IpcChannel.CodeTools_RemoveCustomTerminalPath, terminalId)
   },
   ocr: {
     ocr: (file: SupportedOcrFile, provider: OcrProvider): Promise<OcrResult> =>
       ipcRenderer.invoke(IpcChannel.OCR_ocr, file, provider)
   },
-  cherryin: {
+  cherryai: {
     generateSignature: (params: { method: string; path: string; query: string; body: Record<string, any> }) =>
-      ipcRenderer.invoke(IpcChannel.Cherryin_GetSignature, params)
+      ipcRenderer.invoke(IpcChannel.Cherryai_GetSignature, params)
   },
   windowControls: {
     minimize: (): Promise<void> => ipcRenderer.invoke(IpcChannel.Windows_Minimize),

src/renderer/src/Router.tsx

@@ -8,7 +8,7 @@ import { ErrorBoundary } from './components/ErrorBoundary'
 import TabsContainer from './components/Tab/TabContainer'
 import NavigationHandler from './handler/NavigationHandler'
 import { useNavbarPosition } from './hooks/useSettings'
-import AssistantPresetsPage from './pages/assistantPresets/AssistantPresetsPage'
+import AgentsPage from './pages/agents/AgentsPage'
 import CodeToolsPage from './pages/code/CodeToolsPage'
 import FilesPage from './pages/files/FilesPage'
 import HomePage from './pages/home/HomePage'
@@ -29,7 +29,7 @@ const Router: FC = () => {
       <ErrorBoundary>
         <Routes>
           <Route path="/" element={<HomePage />} />
-          <Route path="/assistantPresets" element={<AssistantPresetsPage />} />
+          <Route path="/agents" element={<AgentsPage />} />
           <Route path="/paintings/*" element={<PaintingsRoutePage />} />
           <Route path="/translate" element={<TranslatePage />} />
           <Route path="/files" element={<FilesPage />} />

src/renderer/src/aiCore/chunk/AiSdkToChunkAdapter.ts

@@ -13,16 +13,6 @@ import { ToolCallChunkHandler } from './handleToolCallChunk'
 
 const logger = loggerService.withContext('AiSdkToChunkAdapter')
 
-export interface CherryStudioChunk {
-  type: 'text-delta' | 'text-complete' | 'tool-call' | 'tool-result' | 'finish' | 'error'
-  text?: string
-  toolCall?: any
-  toolResult?: any
-  finishReason?: string
-  usage?: any
-  error?: any
-}
-
 /**
  * AI SDK 到 Cherry Studio Chunk 适配器类
  * 处理 fullStream 到 Cherry Studio chunk 的转换
@@ -32,19 +22,16 @@ export class AiSdkToChunkAdapter {
   private accumulate: boolean | undefined
   private isFirstChunk = true
   private enableWebSearch: boolean = false
-  private onSessionUpdate?: (sessionId: string) => void
 
   constructor(
     private onChunk: (chunk: Chunk) => void,
     mcpTools: MCPTool[] = [],
     accumulate?: boolean,
-    enableWebSearch?: boolean,
-    onSessionUpdate?: (sessionId: string) => void
+    enableWebSearch?: boolean
   ) {
     this.toolCallHandler = new ToolCallChunkHandler(onChunk, mcpTools)
     this.accumulate = accumulate
     this.enableWebSearch = enableWebSearch || false
-    this.onSessionUpdate = onSessionUpdate
   }
 
   /**
@@ -111,15 +98,6 @@ export class AiSdkToChunkAdapter {
     chunk: TextStreamPart<any>,
     final: { text: string; reasoningContent: string; webSearchResults: AISDKWebSearchResult[]; reasoningId: string }
   ) {
-    const sessionId =
-      (chunk.providerMetadata as any)?.anthropic?.session_id ??
-      (chunk.providerMetadata as any)?.anthropic?.sessionId ??
-      (chunk.providerMetadata as any)?.raw?.session_id
-
-    if (typeof sessionId === 'string' && sessionId) {
-      this.onSessionUpdate?.(sessionId)
-    }
-
     logger.silly(`AI SDK chunk type: ${chunk.type}`, chunk)
     switch (chunk.type) {
       // === 文本相关事件 ===
@@ -192,8 +170,7 @@ export class AiSdkToChunkAdapter {
       case 'reasoning-end':
         this.onChunk({
           type: ChunkType.THINKING_COMPLETE,
-          text: (chunk.providerMetadata?.metadata?.thinking_content as string) || '',
-          thinking_millsec: (chunk.providerMetadata?.metadata?.thinking_millsec as number) || 0
+          text: (chunk.providerMetadata?.metadata?.thinking_content as string) || final.reasoningContent
         })
         final.reasoningContent = ''
         break

src/renderer/src/aiCore/chunk/handleToolCallChunk.ts

@@ -298,8 +298,29 @@ export class ToolCallChunkHandler {
         type: ChunkType.MCP_TOOL_COMPLETE,
         responses: [toolResponse]
       })
+
+      const images: string[] = []
+      for (const content of toolResponse.response?.content || []) {
+        if (content.type === 'image' && content.data) {
+          images.push(`data:${content.mimeType};base64,${content.data}`)
+        }
+      }
+
+      if (images.length) {
+        this.onChunk({
+          type: ChunkType.IMAGE_CREATED
+        })
+        this.onChunk({
+          type: ChunkType.IMAGE_COMPLETE,
+          image: {
+            type: 'base64',
+            images: images
+          }
+        })
+      }
     }
   }
+
   handleToolError(
     chunk: {
       type: 'tool-error'

src/renderer/src/aiCore/legacy/clients/ApiClientFactory.ts

@@ -1,11 +1,12 @@
 import { loggerService } from '@logger'
+import { isNewApiProvider } from '@renderer/config/providers'
 import { Provider } from '@renderer/types'
 
 import { AihubmixAPIClient } from './aihubmix/AihubmixAPIClient'
 import { AnthropicAPIClient } from './anthropic/AnthropicAPIClient'
 import { AwsBedrockAPIClient } from './aws/AwsBedrockAPIClient'
 import { BaseApiClient } from './BaseApiClient'
-import { CherryinAPIClient } from './cherryin/CherryinAPIClient'
+import { CherryAiAPIClient } from './cherryai/CherryAiAPIClient'
 import { GeminiAPIClient } from './gemini/GeminiAPIClient'
 import { VertexAPIClient } from './gemini/VertexAPIClient'
 import { NewAPIClient } from './newapi/NewAPIClient'
@@ -34,8 +35,8 @@ export class ApiClientFactory {
     let instance: BaseApiClient
 
     // 首先检查特殊的 Provider ID
-    if (provider.id === 'cherryin') {
-      instance = new CherryinAPIClient(provider) as BaseApiClient
+    if (provider.id === 'cherryai') {
+      instance = new CherryAiAPIClient(provider) as BaseApiClient
       return instance
     }
 
@@ -45,7 +46,7 @@ export class ApiClientFactory {
       return instance
     }
 
-    if (provider.id === 'new-api') {
+    if (isNewApiProvider(provider)) {
       logger.debug(`Creating NewAPIClient for provider: ${provider.id}`)
       instance = new NewAPIClient(provider) as BaseApiClient
       return instance

src/renderer/src/aiCore/legacy/clients/__tests__/ApiClientFactory.test.ts

@@ -67,7 +67,9 @@ vi.mock('@renderer/config/models', () => ({
     silicon: [],
     defaultModel: []
   },
-  isOpenAIModel: vi.fn(() => false)
+  isOpenAIModel: vi.fn(() => false),
+  glm45FlashModel: {},
+  qwen38bModel: {}
 }))
 
 describe('ApiClientFactory', () => {

src/renderer/src/aiCore/legacy/clients/__tests__/index.clientCompatibilityTypes.test.ts

@@ -35,12 +35,8 @@ vi.mock('@renderer/config/models', () => ({
   findTokenLimit: vi.fn().mockReturnValue(4096),
   isFunctionCallingModel: vi.fn().mockReturnValue(false),
   DEFAULT_MAX_TOKENS: 4096,
-  glm45FlashModel: {
-    id: 'glm-4.5-flash',
-    name: 'GLM-4.5-Flash',
-    provider: 'cherryin',
-    group: 'GLM-4.5'
-  }
+  qwen38bModel: {},
+  glm45FlashModel: {}
 }))
 
 vi.mock('@renderer/services/AssistantService', () => ({

src/renderer/src/aiCore/legacy/clients/cherryai/CherryAiAPIClient.ts

@@ -4,7 +4,7 @@ import OpenAI from 'openai'
 
 import { OpenAIAPIClient } from '../openai/OpenAIApiClient'
 
-export class CherryinAPIClient extends OpenAIAPIClient {
+export class CherryAiAPIClient extends OpenAIAPIClient {
   constructor(provider: Provider) {
     super(provider)
   }
@@ -17,7 +17,7 @@ export class CherryinAPIClient extends OpenAIAPIClient {
     options = options || {}
     options.headers = options.headers || {}
 
-    const signature = await window.api.cherryin.generateSignature({
+    const signature = await window.api.cherryai.generateSignature({
       method: 'POST',
       path: '/chat/completions',
       query: '',
@@ -34,7 +34,7 @@ export class CherryinAPIClient extends OpenAIAPIClient {
   }
 
   override getClientCompatibilityType(): string[] {
-    return ['CherryinAPIClient']
+    return ['CherryAiAPIClient']
   }
 
   public async listModels(): Promise<OpenAI.Models.Model[]> {
@@ -43,7 +43,7 @@ export class CherryinAPIClient extends OpenAIAPIClient {
     const created = Date.now()
     return models.map((id) => ({
       id,
-      owned_by: 'cherryin',
+      owned_by: 'cherryai',
       object: 'model' as const,
       created
     }))

src/renderer/src/aiCore/legacy/middleware/common/ErrorHandlerMiddleware.ts

@@ -1,6 +1,6 @@
 import { loggerService } from '@logger'
 import { isZhipuModel } from '@renderer/config/models'
-import store from '@renderer/store'
+import { getStoreProviders } from '@renderer/hooks/useStore'
 import { Chunk } from '@renderer/types/chunk'
 
 import { CompletionsParams, CompletionsResult } from '../schemas'
@@ -87,7 +87,7 @@ function handleError(error: any, params: CompletionsParams): any {
  * 2. 绘画功能（enableGenerateImage为true）使用通用错误处理
  */
 function handleZhipuError(error: any): any {
-  const provider = store.getState().llm.providers.find((p) => p.id === 'zhipu')
+  const provider = getStoreProviders().find((p) => p.id === 'zhipu')
   const logger = loggerService.withContext('handleZhipuError')
 
   // 定义错误模式映射

src/renderer/src/aiCore/middleware/AiSdkMiddlewareBuilder.ts

@@ -140,7 +140,19 @@ export function buildAiSdkMiddlewares(config: AiSdkMiddlewareConfig): LanguageMo
   return builder.build()
 }
 
-const tagNameArray = ['think', 'thought', 'reasoning']
+const tagName = {
+  reasoning: 'reasoning',
+  think: 'think',
+  thought: 'thought',
+  seedThink: 'seed:think'
+}
+
+function getReasoningTagName(modelId: string | undefined): string {
+  if (modelId?.includes('gpt-oss')) return tagName.reasoning
+  if (modelId?.includes('gemini')) return tagName.thought
+  if (modelId?.includes('seed-oss-36b')) return tagName.seedThink
+  return tagName.think
+}
 
 /**
  * 添加provider特定的中间件
@@ -156,7 +168,7 @@ function addProviderSpecificMiddlewares(builder: AiSdkMiddlewareBuilder, config:
     case 'openai':
     case 'azure-openai': {
       if (config.enableReasoning) {
-        const tagName = config.model?.id.includes('gemini') ? tagNameArray[1] : tagNameArray[0]
+        const tagName = getReasoningTagName(config.model?.id.toLowerCase())
         builder.add({
           name: 'thinking-tag-extraction',
           middleware: extractReasoningMiddleware({ tagName })
@@ -168,13 +180,6 @@ function addProviderSpecificMiddlewares(builder: AiSdkMiddlewareBuilder, config:
       // Gemini特定中间件
       break
     case 'aws-bedrock': {
-      if (config.model?.id.includes('gpt-oss')) {
-        const tagName = tagNameArray[2]
-        builder.add({
-          name: 'thinking-tag-extraction',
-          middleware: extractReasoningMiddleware({ tagName })
-        })
-      }
       break
     }
     default:

src/renderer/src/aiCore/plugins/reasoningTimePlugin.ts

@@ -7,18 +7,14 @@ export default definePlugin({
   transformStream: () => () => {
     // === 时间跟踪状态 ===
     let thinkingStartTime = 0
-    let hasStartedThinking = false
     let accumulatedThinkingContent = ''
-    let reasoningBlockId = ''
 
     return new TransformStream<TextStreamPart<ToolSet>, TextStreamPart<ToolSet>>({
       transform(chunk: TextStreamPart<ToolSet>, controller: TransformStreamDefaultController<TextStreamPart<ToolSet>>) {
         // === 处理 reasoning 类型 ===
         if (chunk.type === 'reasoning-start') {
           controller.enqueue(chunk)
-          hasStartedThinking = true
           thinkingStartTime = performance.now()
-          reasoningBlockId = chunk.id
         } else if (chunk.type === 'reasoning-delta') {
           accumulatedThinkingContent += chunk.text
           controller.enqueue({
@@ -32,21 +28,6 @@ export default definePlugin({
               }
             }
           })
-        } else if (chunk.type === 'reasoning-end' && hasStartedThinking) {
-          controller.enqueue({
-            type: 'reasoning-end',
-            id: reasoningBlockId,
-            providerMetadata: {
-              metadata: {
-                thinking_millsec: performance.now() - thinkingStartTime,
-                thinking_content: accumulatedThinkingContent
-              }
-            }
-          })
-          accumulatedThinkingContent = ''
-          hasStartedThinking = false
-          thinkingStartTime = 0
-          reasoningBlockId = ''
         } else {
           controller.enqueue(chunk)
         }

src/renderer/src/aiCore/prepareParams/parameterBuilder.ts

@@ -134,9 +134,10 @@ export async function buildStreamTextParams(
     if (aiSdkProviderId === 'google-vertex') {
       tools.google_search = vertex.tools.googleSearch({}) as ProviderDefinedTool
     } else if (aiSdkProviderId === 'google-vertex-anthropic') {
+      const blockedDomains = mapRegexToPatterns(webSearchConfig.excludeDomains)
       tools.web_search = vertexAnthropic.tools.webSearch_20250305({
         maxUses: webSearchConfig.maxResults,
-        blockedDomains: mapRegexToPatterns(webSearchConfig.excludeDomains)
+        blockedDomains: blockedDomains.length > 0 ? blockedDomains : undefined
       }) as ProviderDefinedTool
     }
   }