Class AIService

src/renderer/src/services/aiService.ts:11–102

}

export class AIService {
  private llmConfig: LLMConfig
  private modelConfig: ModelConfig | undefined
  private requestId: string
  private isAborted: boolean = false

  constructor(llmConfig: LLMConfig, modelConfig: ModelConfig | undefined) {
    this.llmConfig = llmConfig
    this.modelConfig = modelConfig
    this.requestId = uuidv4()
  }

  // 添加getter方法来暴露requestId
  get id(): string {
    return this.requestId
  }

  async stopStreaming(): Promise<void> {
    try {
      this.isAborted = true
      await window.api.ai.stopStreaming(this.requestId)
    } catch (error) {
      console.error('Failed to stop streaming:', error)
    }
  }

  async sendMessage(messages: ChatMessage[], callbacks: StreamingResponse): Promise<void> {
    if (this.isAborted) {
      callbacks.onError(new Error('Request was aborted'))
      return
    }

    try {
      // 直接调用新接口，传递回调函数对象
      await window.api.ai.sendMessageStreaming(
        {
          requestId: this.requestId,
          llmConfig: this.llmConfig,
          modelConfig: this.modelConfig,
          messages: messages
        },
        {
          onChunk: (chunk: string) => {
            if (!this.isAborted) {
              callbacks.onChunk(chunk)
            }
          },
          onReasoning: (reasoning: string) => {
            if (!this.isAborted) {
              callbacks.onReasoning?.(reasoning)
            }
          },
          onComplete: (fullResponse: string, reasoning?: string) => {
            if (!this.isAborted) {
              callbacks.onComplete(fullResponse, reasoning)
            }
          },
          onError: (error: string) => {