跳到主要内容

流式文本提取与输出控制

为什么需要文本提取?

ReactAgent 的输出流 Flux<NodeOutput> 里面并不全是给用户看的文本。Agent 在执行过程中会产生各种类型的输出:

  • 模型正在流式生成的文本片段
  • 模型生成完毕的完整文本
  • 工具调用的请求和响应
  • 图节点的状态变更
  • 中间推理过程的数据

我们只需要把"模型生成的文本"提取出来推送给前端,其他的都应该过滤掉。这就是 extractTextChunk 方法的职责。

extractTextChunk:输出类型分拣

private Mono<String> extractTextChunk(NodeOutput output, AtomicBoolean streamedText) {
// ReactAgent 图里可能会输出非流式节点结果;只有 StreamingOutput 才可能携带给用户的文本。
if (!(output instanceof StreamingOutput<?> streamingOutput)) {

// 非 StreamingOutput 不参与前端答案流,直接返回空 Mono。
return Mono.empty();
}

// 从 StreamingOutput 中兼容提取 message 文本、originData Message 文本或 originData 字符串。
String content = extractStreamingText(streamingOutput);
// 没有文本内容时,不向用户流输出任何片段。
if (StrUtil.isBlank(content)) {
return Mono.empty();
}

// AGENT_MODEL_STREAMING 表示模型正在流式输出增量文本,应立即返回给前端。
if (streamingOutput.getOutputType() == OutputType.AGENT_MODEL_STREAMING) {

// 标记已经收到过流式片段,后面遇到 FINISHED 时就不再重复发送完整文本。
streamedText.set(true);
// 返回当前流式文本片段。
return Mono.just(content);
}

// AGENT_MODEL_FINISHED 表示模型最终完成事件,可能携带完整文本或最终文本。
if (streamingOutput.getOutputType() == OutputType.AGENT_MODEL_FINISHED) {

// 如果前面已经流式输出过文本,完成事件通常是重复内容,跳过以避免答案重复。
if (streamedText.get()) {
return Mono.empty();
}
// 如果没有收到过流式片段,则用完成事件中的文本作为兜底输出。
return Mono.just(content);
}

// 其他输出类型可能是工具调用、节点状态或中间数据,不直接作为用户答案文本。
return Mono.empty();
}

这个方法的逻辑可以用一张决策图来表示:

决策图
决策图

付费内容提示

该文档的全部内容仅对「码力全开」项目实战&技术讲解 知识星球用户开放

加入星球,一次获得完整项目资料、全栈技术知识库和长期答疑服务。

100万+字全栈技术知识库深入讲解技术核心、数据库、中间件和分布式等内容
8套热门的实战项目持续更新的企业级项目覆盖高并发、微服务、数据中台 和 AI Agent 等方向
AI 技术知识大模型面试详解覆盖 AI 模型原理、Agent、RAG、MCP、Skills、Harness 等核心知识点
文档 + 视频两种讲解形式既能系统阅读,也能跟随视频理解核心业务

完整项目实战资料

每套项目均包含从 0 到 1 讲解文档核心业务讲解视频

从基础项目到复杂业务场景,项目资料会持续更新。

8 套项目
  • 01Nexus Agent AI 智能体
  • 02Nexus Agent Pro 完全版
  • 03黑马点评Plus
  • 04大麦
  • 05大麦Pro
  • 06大麦AI
  • 07流量切换
  • 08数据中台

加入后还能获得

进入星球后,即可享受上述所有服务,保证不会再有其他隐藏费用。

从学习、面试到项目启动,都可以继续获得支持。

  • 1 对 1 解答项目和技术问题都可以提问
  • 针对性补充没有讲清楚的内容会继续补充
  • 面试与简历指导梳理回答技巧和项目亮点
  • 中间件云环境项目依赖可以直接接入使用
  • 面试后复盘被问住的问题可以继续交流
  • 远程问题解决项目启动问题可协助排查
知识星球二维码

扫码进入知识星球

  1. 打开微信,扫描左侧二维码,加入「码力全开」项目实战&技术讲解 知识星球
  2. 查看星球使用指导,获取完整项目讲解资料索引
解锁全部付费内容
🎁优惠