Function useAgentStreamService

src/services/ai-stream/service.ts:49–158

()

const requestIdMap = new Map<string, { serverId: string; abortController: AbortController }>()

export function useAgentStreamService() {
  return {
    runStream(
      params: AgentStreamParams,
      onChunk?: (chunk: AgentStreamChunk) => void
    ): { requestId: string; promise: Promise<AgentStreamResult> } {
      const abortController = new AbortController()
      const localRequestId = `local_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`
      requestIdMap.set(localRequestId, { serverId: '', abortController })

      const promise = new Promise<AgentStreamResult>((resolve) => {
        let resultContent = ''
        const toolsUsed: string[] = []
        let toolRounds = 0
        let lastUsage: import('./types').TokenUsage | undefined
        let hasError = false
        let lastError: AgentStreamResult['error']

        fetchSSE({
          url: '/_web/ai/agent/stream',
          body: params,
          signal: abortController.signal,
          onEvent: ({ event, data }) => {
            try {
              const parsed = JSON.parse(data)

              if (event === 'meta') {
                const entry = requestIdMap.get(localRequestId)
                if (entry) entry.serverId = parsed.requestId ?? ''
                return
              }

              const chunk = parsed as AgentStreamChunk
              onChunk?.(chunk)

              switch (chunk.type) {
                case 'content':
                  if (chunk.content) resultContent += chunk.content
                  break
                case 'tool_start':
                  if (chunk.toolName) toolsUsed.push(chunk.toolName)
                  break
                case 'status':
                  if (chunk.status) toolRounds = chunk.status.round
                  break
                case 'done':
                  if (chunk.usage) lastUsage = chunk.usage
                  break
                case 'error':
                  hasError = true
                  lastError = chunk.error as AgentStreamResult['error']
                  break
              }
            } catch {
              // skip malformed JSON
            }
          },
        })