Class AIHandler

src/main/aiHandler.ts:50–405

}

class AIHandler {
  // 使用 Map 管理多个并行请求的 AbortController
  private abortControllers = new Map<string, AbortController>()

  /**
   * 准备 API 消息数组，处理 systemPrompt 和文件附件
   */
  private async prepareApiMessages(
    messages: ChatMessage[],
    modelConfig: ModelConfig
  ): Promise<ChatMessage[]> {
    const attachmentsDir = path.join(app.getPath('userData'), 'attachments')

    const apiMessages = await Promise.all(
      messages.map(async (msg) => {
        // 如果消息有附件，需要转换为多模态格式
        if (msg.attachments && msg.attachments.length > 0) {
          const contentParts: Array<{
            type: 'text' | 'image_url'
            text?: string
            image_url?: { url: string }
          }> = []

          // 添加文本内容
          if (typeof msg.content === 'string' && msg.content.trim()) {
            contentParts.push({
              type: 'text',
              text: msg.content
            })
          }

          // 添加图片附件 - 从文件系统读取
          for (const attachment of msg.attachments) {
            if (attachment.type.startsWith('image/')) {
              try {
                const fullPath = path.join(attachmentsDir, attachment.localPath)
                const buffer = await readFile(fullPath)
                const base64Content = buffer.toString('base64')

                contentParts.push({
                  type: 'image_url',
                  image_url: {
                    url: `data:${attachment.type};base64,${base64Content}`
                  }
                })
              } catch (error) {
                console.error('Failed to read attachment file:', attachment.localPath, error)
                // 跳过无法读取的附件
              }
            }
          }

          return {
            role: msg.role,
            content: contentParts
          }
        }