refactor: update AIRequest structure and streamline IPC handler setup

Author: experdot

src/main/aiHandler.ts

@@ -20,10 +20,9 @@ export interface ChatMessage {
 
 export interface AIRequest {
   requestId: string
-  config: LLMConfig
+  llmConfig: LLMConfig
   modelConfig: ModelConfig
   messages: ChatMessage[]
-  streaming?: boolean
 }
 
 export interface AIStreamChunk {
@@ -38,20 +37,7 @@ class AIHandler {
   // 使用 Map 管理多个并行请求的 AbortController
   private abortControllers = new Map<string, AbortController>()
 
-  constructor() {
-    this.setupHandlers()
-  }
-
-  private setupHandlers() {
-    ipcMain.handle('ai:send-message', (event, request: AIRequest) => this.sendMessage(request))
-    ipcMain.handle('ai:send-message-streaming', (event, request: AIRequest) =>
-      this.sendMessageStreaming(event, request)
-    )
-    ipcMain.handle('ai:stop-streaming', (event, requestId: string) => this.stopStreaming(requestId))
-    ipcMain.handle('ai:test-connection', (event, config: LLMConfig) => this.testConnection(config))
-  }
-
-  private async sendMessageStreaming(
+  public async sendMessageStreaming(
     event: Electron.IpcMainInvokeEvent,
     request: AIRequest
   ): Promise<void> {
@@ -87,19 +73,19 @@ class AIHandler {
       }
 
       const response = await fetch(
-        `${request.config.apiHost.replace(/\/$/, '')}/chat/completions`,
+        `${request.llmConfig.apiHost.replace(/\/$/, '')}/chat/completions`,
         {
           method: 'POST',
           headers: {
             'Content-Type': 'application/json',
-            Authorization: `Bearer ${request.config.apiKey}`
+            Authorization: `Bearer ${request.llmConfig.apiKey}`
           },
           body: JSON.stringify({
-            model: request.config.modelName,
+            model: request.llmConfig.modelName,
             messages: apiMessages,
-            stream: true,
             temperature: modelConfig.temperature,
-            top_p: modelConfig.topP
+            top_p: modelConfig.topP,
+            stream: true
           }),
           signal: abortController.signal
         }
@@ -231,87 +217,15 @@ class AIHandler {
     }
   }
 
-  private async stopStreaming(requestId: string): Promise<void> {
+  public async stopStreaming(requestId: string): Promise<void> {
     const abortController = this.abortControllers.get(requestId)
     if (abortController) {
       abortController.abort()
       this.abortControllers.delete(requestId)
     }
   }
 
-  private async sendMessage(
-    request: AIRequest
-  ): Promise<{ success: boolean; content?: string; reasoning_content?: string; error?: string }> {
-    try {
-      // 准备消息数组，如果有systemPrompt，插入system消息
-      const apiMessages = request.messages.map((msg) => ({
-        role: msg.role,
-        content: msg.content
-      }))
-
-      let modelConfig = request.modelConfig
-      if (!request.modelConfig) {
-        modelConfig = {
-          systemPrompt: '',
-          topP: 1,
-          temperature: 1
-        }
-      }
-
-      // 如果有systemPrompt且第一条消息不是system消息，则插入system消息
-      if (
-        modelConfig.systemPrompt &&
-        (apiMessages.length === 0 || apiMessages[0].role !== 'system')
-      ) {
-        apiMessages.unshift({
-          role: 'system',
-          content: modelConfig.systemPrompt
-        })
-      }
-
-      const response = await fetch(
-        `${request.config.apiHost.replace(/\/$/, '')}/chat/completions`,
-        {
-          method: 'POST',
-          headers: {
-            'Content-Type': 'application/json',
-            Authorization: `Bearer ${request.config.apiKey}`
-          },
-          body: JSON.stringify({
-            model: request.config.modelName,
-            messages: apiMessages,
-            temperature: modelConfig.temperature,
-            top_p: modelConfig.topP
-          })
-        }
-      )
-
-      if (!response.ok) {
-        return {
-          success: false,
-          error: `HTTP error! status: ${response.status}`
-        }
-      }
-
-      const data = await response.json()
-      const content = data.choices?.[0]?.message?.content || ''
-      const reasoning_content =
-        data.choices?.[0]?.message?.reasoning_content || data.choices?.[0]?.message?.reasoning
-
-      return {
-        success: true,
-        content,
-        reasoning_content
-      }
-    } catch (error) {
-      return {
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error'
-      }
-    }
-  }
-
-  private async testConnection(config: LLMConfig): Promise<{ success: boolean; error?: string }> {
+  public async testConnection(config: LLMConfig): Promise<{ success: boolean; error?: string }> {
     try {
       const response = await fetch(`${config.apiHost.replace(/\/$/, '')}/models`, {
         method: 'GET',
@@ -338,3 +252,11 @@ class AIHandler {
 }
 
 export const aiHandler = new AIHandler()
+
+export function setupAIHandlers() {
+  ipcMain.handle('ai:send-message-streaming', (event, request: AIRequest) =>
+    aiHandler.sendMessageStreaming(event, request)
+  )
+  ipcMain.handle('ai:stop-streaming', (event, requestId: string) => aiHandler.stopStreaming(requestId))
+  ipcMain.handle('ai:test-connection', (event, config: LLMConfig) => aiHandler.testConnection(config))
+}

src/main/autoUpdater.ts

@@ -4,7 +4,7 @@ import { is } from '@electron-toolkit/utils'
 import { join } from 'path'
 
 export function setupAutoUpdater(): void {
-  // 设置自动下载为false，我们想手动控制下载过程
+  // 设置自动下载为false，手动控制下载过程
   autoUpdater.autoDownload = false
   autoUpdater.autoInstallOnAppQuit = true
 

src/main/index.ts

@@ -1,6 +1,6 @@
 import { app, BrowserWindow } from 'electron'
 import { electronApp, optimizer } from '@electron-toolkit/utils'
-import { aiHandler } from './aiHandler'
+import { setupAIHandlers } from './aiHandler'
 import { setupAutoUpdater } from './autoUpdater'
 import { setupIpcHandlers } from './ipcHandlers'
 import { createWindow } from './window'
@@ -20,7 +20,7 @@ app.whenReady().then(() => {
   })
 
   // Initialize AI handler
-  aiHandler
+  setupAIHandlers()
 
   // Setup auto updater
   setupAutoUpdater()

src/preload/index.d.ts

@@ -6,9 +6,8 @@ declare global {
     api: {
       ai: {
         sendMessageStreaming: (request: any) => Promise<any>
-        sendMessage: (request: any) => Promise<any>
-        testConnection: (config: any) => Promise<any>
         stopStreaming: (requestId: string) => Promise<void>
+        testConnection: (config: any) => Promise<any>
         onStreamData: (requestId: string, callback: (data: any) => void) => void
         removeStreamListener: (requestId: string) => void
       }

src/preload/index.ts

@@ -6,7 +6,6 @@ const streamListeners = new Map<string, (data: any) => void>()
 
 // Custom APIs for renderer
 const api = {
-  // AI相关API
   ai: {
     sendMessageStreaming: (request: any) =>
       ipcRenderer.invoke('ai:send-message-streaming', request),

src/renderer/src/services/aiService.ts

@@ -86,7 +86,7 @@ export class AIService {
       // 发送流式请求，包含请求ID
       await window.api.ai.sendMessageStreaming({
         requestId: this.requestId,
-        config: this.llmConfig,
+        llmConfig: this.llmConfig,
         modelConfig: this.modelConfig,
         messages: messages
       })
@@ -102,33 +102,6 @@ export class AIService {
     window.api.ai.removeStreamListener(this.requestId)
   }
 
-  // 备用的非流式方法
-  async sendMessageNonStreaming(
-    messages: ChatMessage[]
-  ): Promise<{ content: string; reasoning_content?: string }> {
-    try {
-      const result = await window.api.ai.sendMessage({
-        requestId: this.requestId,
-        config: this.llmConfig,
-        modelConfig: this.modelConfig,
-        messages: messages
-      })
-
-      if (result.success) {
-        return {
-          content: result.content || '抱歉，我无法生成回复。',
-          reasoning_content: result.reasoning_content
-        }
-      } else {
-        throw new Error(result.error || 'Unknown error')
-      }
-    } catch (error) {
-      console.error('AI Service Error:', error)
-      throw error
-    }
-  }
-
-  // 测试连接
   async testConnection(): Promise<boolean> {
     try {
       const result = await window.api.ai.testConnection(this.llmConfig)