架构与组件¶
Claw-R1 的组件围绕一条数据流组织:Agent 产生交互,Gateway 采集并规范化 step,DataPool 存储与筛选数据,TrainingBackend 将可训练数据供给 RL 训练后端。
Black-box Agent White-box Agent
| |
v v
Gateway Server <---- explicit Step APIs
|
v
DataPool <---- Dashboard 读取统计、step、事件、筛选状态和 prefix-tree 预览
|
v
Async Trainer ---- Parameter Synchronizer ---- Async Rollouter / vLLM
Gateway Server¶
Gateway 是面向 Agent 的 HTTP 入口,支持两种接入方式:
- 白盒 Agent:显式构造
Step,通过 API 提交到 Gateway。 - 黑盒 Agent:使用 OpenAI 兼容接口,只修改
base_url即可通过 Gateway 转发请求并采集数据。
Gateway 会把交互整理成统一的 step-level 记录,再提交给 DataPool。
DataPool¶
DataPool 是 Ray Actor,也是 Claw-R1 的数据管理核心。它负责:
- 保存 prompt IDs、response IDs、reward、metadata 等 step-level 字段。
- 维护 trajectory ID、step index、prompt group、policy version 和 channel。
- 记录 reward、curation label、trainability、tag 和 note 等筛选信息。
- 按训练后端需要返回 batch,并跟踪数据消费状态。
Agent Flow¶
Agent Flow 管理 Agent 执行生命周期。白盒 flow 适合直接控制推理与工具调用逻辑的场景;黑盒 flow 适合已有 Agent 系统,只需通过 Gateway 捕获 OpenAI 兼容请求。
Reward System¶
Reward System 用于把不同来源的质量信号转为训练可用的 reward 或筛选依据:
- rule-based reward
- discriminative reward model
- generative judge
- explicit 或 implicit human feedback
- policy version freshness
Async Training¶
异步训练将 rollout 和 training 拆成不同 Ray actor,分别使用不同 GPU 池。DataPool 作为中间层缓冲并供给训练数据,Parameter Synchronizer 负责将训练后的权重同步给 rollout 侧。
Prefix Tree Merge¶
Prefix Tree Merge 面向 multi-step agent 训练中的共享前缀问题。它通过 prefix-tree packing 减少重复 prefix 计算,并可在 Dashboard 中预览真实 DataPool step 构成的 prefix tree。
生命周期映射¶
| 阶段 | 主要组件 | Dashboard 视图 |
|---|---|---|
| 采集交互 | Gateway, Agent Flow | Collection |
| 存储表示 | DataPool | Representation |
| 评估质量 | Reward System | Curation Signals |
| 筛选样本 | DataPool curation APIs | Curation |
| 优化共享上下文 | Prefix Tree Merge preview | Optimization |
| 供给训练 | DataPool, Async Trainer | Consumption |