跳到主要内容

子问题证据汇总与引用编号分配

上一篇我们已经讲完了单个子问题的完整检索流程：多通道召回、证据闸门、RRF 融合、父块提升、Rerank、FinalTopK、观测记录，最后返回一个 SubQuestionEvidence。

这一篇回到最外层的 RagRetrievalEngine#retrieve。前面我们看到外层方法会把多个子问题并发丢出去，那等它们都跑完了，还得做几件事：等结果、统计一下哪些子问题真的查到了东西、给证据分配引用编号，然后把结果放回上下文给后面用。

retrieve 方法的最后部分​

先看这段剩下的源码，就是 retrieve 方法最后要做的事：

// 等待所有子问题完成；异常已经在 exceptionally 中被转换为空证据，因此这里 join 不会中断主流程。
List<SubQuestionEvidence> evidenceList = futures.stream()
    .map(CompletableFuture::join)
    .toList();
// acceptedCount 表示真正拿到最终候选文档的子问题数量，用于检索效果日志。
int acceptedCount = (int) evidenceList.stream().filter(item -> item.getDocuments() != null && !item.getDocuments().isEmpty()).count();
log.info("RAG 检索完成: retrievalQuestion='{}', originalSubQuestionCount={}, acceptedSubQuestionCount={}, notes={}",
    plan.getRetrievalQuestion(),
    evidenceList.size(),
    acceptedCount,
    context.getRetrievalNotes());
// 所有子问题检索结束后统一分配 referenceId，保证跨子问题重复证据引用同一个编号。
assignReferenceIds(evidenceList);
// 把子问题证据列表放回上下文，后续 Prompt 组装会从这里读取证据和引用。
context.setSubQuestionEvidenceList(evidenceList);
return context;

这段代码看着不长，但干的活挺重要。前面每个子问题都在各自线程里跑，跑完以后都会变成一个 SubQuestionEvidence。这里要做的就是把这些散落在各个线程里的结果拿过来，组装成一个完整的 RagRetrievalContext。

等待所有子问题完成​

第一步是把所有 CompletableFuture<SubQuestionEvidence> 变成真正的 SubQuestionEvidence：

// 等待所有子问题完成；异常已经在 exceptionally 中被转换为空证据，因此这里 join 不会中断主流程。
List<SubQuestionEvidence> evidenceList = futures.stream()
    .map(CompletableFuture::join)
    .toList();

这里的重点是 CompletableFuture::join。

前面创建子问题异步任务的时候，每个 future 后面都挂了 exceptionally。如果某个子问题超时、异常、通道崩了，它不会把异常继续往外抛，而是会返回一个空的 SubQuestionEvidence：

return new SubQuestionEvidence(subQuestionIndex, subQuestion, List.of(), new ArrayList<>(), List.of(), 0, 0, 0);

付费内容提示

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

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

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. 查看星球使用指导，获取完整项目讲解资料索引
解锁全部付费内容
🎁优惠