Class AIHandler

src/main/aiHandler.ts:24–405

}

class AIHandler {
  private abortControllers = new Map<string, AbortController>()
  private requestStates = new Map<string, RequestState>()

  private initRequestState(
    requestId: string,
    event: Electron.IpcMainInvokeEvent,
    eventChannel: string
  ): void {
    this.requestStates.set(requestId, {
      event,
      eventChannel,
      fullResponse: '',
      fullReasoning: '',
      status: 'running'
    })
  }

  private cleanupRequest(requestId: string): void {
    this.abortControllers.delete(requestId)
    this.requestStates.delete(requestId)
  }

  private emitChunk(requestId: string, content: string): void {
    const state = this.requestStates.get(requestId)
    if (!state || state.status !== 'running') {
      return
    }

    state.fullResponse += content
    state.event.sender.send(state.eventChannel, {
      requestId,
      type: 'chunk',
      content
    } as AIStreamChunk)
  }

  private emitReasoning(requestId: string, reasoning: string): void {
    const state = this.requestStates.get(requestId)
    if (!state || state.status !== 'running') {
      return
    }

    state.fullReasoning += reasoning
    state.event.sender.send(state.eventChannel, {
      requestId,
      type: 'reasoning_content',
      reasoning_content: reasoning
    } as AIStreamChunk)
  }

  private finalizeComplete(requestId: string): void {
    const state = this.requestStates.get(requestId)
    if (!state || (state.status !== 'running' && state.status !== 'aborted')) {
      return
    }

    state.status = 'completed'