Federated RAG retrieval currently flattens module outputs, which makes it impossible to combine ranked…
Federated RAG retrieval currently flattens module outputs, which makes it impossible to combine ranked…: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). Extend the public RetrievalRagStage and RagContext APIs with deterministic fusion and diagnostics while preserving…
The task
Extend the public RetrievalRagStage and RagContext APIs with deterministic fusion and diagnostics while preserving concatenation as the default behavior. RetrievalRagStage must accept these keyword options: fusion_mode: Literal["concatenate", "rrf"] = "concatenate" rrf_k: float…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.