Skip to main content

为什么需要流式

想象 AI 需要 30 秒才能生成完整回答——如果等 30 秒后才一次性显示,用户体验是灾难性的。 流式响应让用户实时看到 AI 的思考过程
  • 文字逐字出现,用户能提前判断方向是否正确
  • 工具调用的参数在生成过程中就能预览
  • 长时间任务不会让用户觉得”卡死了”

BetaRawMessageStreamEvent 核心事件类型

流式 API 返回的是一系列 BetaRawMessageStreamEvent,每种事件类型对应流式响应的不同阶段(src/services/api/claude.ts):

事件处理状态机

src/services/api/claude.tsqueryModelWithStreaming() 函数的事件处理循环实现了一个基于 switch(part.type) 的状态机:

内容块类型及其增量数据

content_block_start 中的 content_block.type 决定了如何处理后续 delta: 关键设计:content_block_start 时所有文本字段初始化为空字符串,只通过 content_block_delta 累加。这是因为 SDK 有时在 start 和 delta 中重复发送相同文本。

文本 chunk 和 tool_use block 的交织

一次 AI 响应可能包含多个内容块,交替出现:
每个 content_block_stop 触发一次 yield,将完整的 AssistantMessage 推送给消费者。这意味着一个 AI 响应会产生多条 AssistantMessage——文本消息和工具调用消息交替产出。 stop_reason 要等到 message_delta 才确定(可能是 end_turntool_usemax_tokens 等),所以最后一条消息的 stop_reason回写的:

流式中的错误处理

网络断开

流式连接依赖 SSE(Server-Sent Events)。当连接中断时,系统有两层检测机制:
  1. 被动停滞检测src/services/api/claude.ts 中 stall 检测逻辑):当下一个事件到达时,计算与上一个事件的时间间隔。超过阈值(30 秒,STALL_THRESHOLD_MS = 30_000)记录为一次 stall,累积计数并写入遥测日志。这是被动检测——仅在下一个 chunk 到达时才触发,不会主动中断流。
  2. 主动空闲超时看门狗src/services/api/claude.tsSTREAM_IDLE_TIMEOUT_MS 看门狗逻辑):使用 setTimeout 设置 90 秒(可通过 CLAUDE_STREAM_IDLE_TIMEOUT_MS 环境变量覆盖)的硬性超时。如果在此期间没有收到任何事件,主动终止流并抛出错误进入重试流程。
  3. 非流式降级:作为最后手段,设置 didFallBackToNonStreaming 标志,通过 executeNonStreamingRequest() 回退到非流式请求(一次性获取完整响应)。

API 限流

当 API 返回限流错误时,系统使用 withRetry 包装器进行指数退避重试。重试逻辑考虑了:
  • 错误类型(429 限流 vs 500 服务器错误)
  • 重试次数上限
  • 退避间隔

Token 超限

两种 token 超限场景有不同的处理:

流式停滞检测

系统持续监控事件到达间隔,检测”停滞”(stall):
这是被动检测——仅在下一个 chunk 到达时才触发比较。与之互补的是 90 秒主动空闲超时看门狗(STREAM_IDLE_TIMEOUT_MS),会直接中断长时间无响应的流。

工具执行的流式反馈

BashTool 的命令执行也是流式的——通过 onProgress 回调逐行推送输出:
UI 层通过 useToolCallProgress hook 实时展示命令输出,而不是等命令完全结束。长时间运行的命令还支持自动后台化(shouldAutoBackground)。

多 Provider 适配

所有 Provider 通过统一的 Stream<BetaRawMessageStreamEvent> 抽象层屏蔽差异。上层代码(QueryEngine、REPL)不需要关心底层用的是哪个 Provider。

Provider 选择

src/utils/model/providers.ts 中的 getAPIProvider() 根据配置决定使用哪个 Provider:
每个 Provider 需要适配的细节包括:认证方式、beta header、请求参数格式、错误码映射——但这些差异在 claude.tsqueryStream() 函数中被统一处理。