V3.5 支持Advanced模式,用户提供python命令

This commit is contained in:
2026-01-04 15:28:41 +08:00
parent e3dcfe526f
commit 5d0d849c28
22 changed files with 1355 additions and 34 deletions

7
specs/mvp/v3.5/README.md Normal file
View File

@@ -0,0 +1,7 @@
# MVP v3.5
本目录包含 v3.5 的需求与设计(精简版):
- `requirement.md`:需求补充说明(来源于讨论)
- `roadmap_v3.5.png`:架构草图(Advanced Task + Resume + IB + Serving)
- `v3.5_design.md`:详细设计方案(基于 v3.0;当前迭代仅聚焦 Advanced TaskSpec + Custom Reward,Serving/IB/Resume/多版本 verl 暂缓)

3
specs/mvp/v3.5/note.md Normal file
View File

@@ -0,0 +1,3 @@
1. node management(v3.5 引入的接口骨架:通过 SSH/平台能力管理 head/worker 节点生命周期;先做最小可用 --- 这个是干嘛的?
2.

View File

@@ -0,0 +1,40 @@
v3.5 版本是在v3.0的基础上进行功能扩展:
1. 支持自定义命令,不走固定的TaskSpec模板,用户直接提供调用verl 的python命令,如下,这个灵活度更高,需要用户自己把握文件路径,用户使用 $HOME,服务层替换为用户自己的/private/users/<user>/路径,使用$COMMON 则替换为/private/
```
PYTHONUNBUFFERED=1 python3 -m verl.trainer.main_ppo \
data.train_files=$HOME/data/gsm8k/train.parquet \
data.val_files=$HOME/data/gsm8k/test.parquet \
data.train_batch_size=256 \
data.max_prompt_length=512 \
data.max_response_length=512 \
actor_rollout_ref.model.path=Qwen/Qwen2.5-0.5B-Instruct \
actor_rollout_ref.actor.optim.lr=1e-6 \
actor_rollout_ref.actor.ppo_mini_batch_size=64 \
actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=4 \
actor_rollout_ref.rollout.name=vllm \
actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8 \
actor_rollout_ref.rollout.tensor_model_parallel_size=1 \
actor_rollout_ref.rollout.gpu_memory_utilization=0.4 \
actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=4 \
critic.optim.lr=1e-5 \
critic.model.path=Qwen/Qwen2.5-0.5B-Instruct \
critic.ppo_micro_batch_size_per_gpu=4 \
algorithm.kl_ctrl.kl_coef=0.001 \
trainer.logger=console \
trainer.val_before_train=False \
trainer.n_gpus_per_node=1 \
trainer.nnodes=1 \
trainer.save_freq=10 \
trainer.test_freq=10 \
trainer.total_epochs=15
```
2. 支持自定义的奖励函数方法,你参考 verl 项目 [text](../../../verl) 里的示例,设计方案
3. 支持codepath指定用户上传到自己user路径下的 verl版本代码
4. 断点续训:支持某个已经complete(成功或者fail或者stopped)的任务task,从最后一个保存的checkpoint 继续训练,参数应该保持不变,你确认一下是不是对应一个新的ray job,或者分析一下verl 是否已经有类似的功能支持。
5. 支持训练走NCCL,使用RoCEv2和Infiband网络,调研一些verl怎样支持,需要哪些配置。

Binary file not shown.

After

Width:  |  Height:  |  Size: 98 KiB

View File

@@ -0,0 +1,366 @@
# MVP v3.5 详细设计方案(进一步精简版,基于 v3.0)
> 背景:v3.0 已具备 WebUI + API server + 用户/任务隔离 + SFTPGo 数据管理 + Stateless Ray cluster(head + worker node pool)。
>
> v3.5 本轮 **只做 2 件事**:
> 1) Advanced Task:支持用户提交自定义训练命令(command)
> 2) Custom Reward:支持用户通过 VERL 原生 `custom_reward_function.*` 方式注入 reward(仅方式 A:用户自己写命令)
>
> 明确不做(从上一版设计中移除):(3) 自定义 verl 版本/代码路径、(4) 断点续训、(5) IB/RoCEv2 网络支持、(6) Model Serving。
---
## 0. 继承 v3.0 的不变点(重要约束)
1) **Node management 不变**
- v3.5 不新增/不修改 node management 机制;仍按 v3.0 现状运行(head 写 discovery、worker watchdog 自动 join、自愈)。
2) **Head 不跑训练**
- 所有训练/Serving driver 通过 Ray entrypoint placement 强制落在 worker(例如 `entrypoint_resources={"worker_node": 1}`)。
3) **SFTPGo 的 “common” 目录约定变更**
- 不再使用 `$COMMON` 宏。
- 在 SFTPGo 中,把共享只读资源映射到用户 home 下的固定目录(用户在 SFTP/WebClient 看到的是 `$HOME/common/...`):
- `$HOME/common/datasets` → 容器内真实路径 `/private/datasets`(只读)
- `$HOME/common/hf` → 容器内真实路径 `/private/hf`(只读)
> 这里的 `$HOME` 指:`/private/users/<user_id>`(容器内路径)。
---
## 1. v3.5 需求范围(精简后)
### 1.1 In scope
**A. Advanced TaskSpec(自定义命令)**
- 用户提交 `command`(多行 shell 或单行)
- 平台做 `$HOME` 宏替换
- 平台做 best-effort 安全检查(路径/关键参数),然后提交为 Ray job
**B. Custom Reward(仅方式 A)**
- 用户在 `command` 里显式写 hydra overrides:
- `custom_reward_function.path=...`
- `custom_reward_function.name=...`
- `custom_reward_function.reward_kwargs.*=...`(可选)
- 平台不提供结构化 reward 字段(不做方式 B),只做检查(校验 path 合法)
### 1.2 Out of scope(本轮不做)
- 自定义 verl 版本/代码路径(仍使用平台内置/公共 verl 代码快照)
- 断点续训(resume from checkpoint)
- IB/RoCEv2 网络专门支持(NCCL/RDMA env 先不引入平台)
- Model Serving(暂缓,后续单独设计迭代)
---
## 2. Advanced TaskSpec 设计
### 2.1 为什么需要 Advanced Task
v3.0 的 Basic TaskSpec(ppo/grpo/sft)通过平台模板生成固定 overrides,适合“快速跑通”。
但科研/调参场景需要更高自由度:用户希望直接写 `python3 -m verl.trainer.main_ppo ...` 并自行控制每个 override。
### 2.2 Advanced TaskSpec(建议 schema)
建议新增一种 TaskSpec 类型,通过 `kind: advanced` 区分:
```yaml
kind: advanced
# 资源(平台调度与预检查用;仍需要)
nnodes: 2
n_gpus_per_node: 4
# 自定义命令(用户负责写对 VERL 的参数/路径)
# 平台会对 $HOME 做宏替换;其余保持原样
command: |
PYTHONUNBUFFERED=1 python3 -m verl.trainer.main_ppo \
data.train_files=$HOME/datasets/gsm8k/train.parquet \
data.val_files=$HOME/datasets/gsm8k/test.parquet \
actor_rollout_ref.model.path=Qwen/Qwen2.5-0.5B-Instruct \
trainer.nnodes=2 \
trainer.n_gpus_per_node=4 \
trainer.total_epochs=1 \
trainer.save_freq=10 \
+ray_kwargs.ray_init.address=auto
```
### 2.3 `$HOME` 宏替换规则
仅支持 `$HOME`(v3.5 移除 `$COMMON`):
- `$HOME` → `/private/users/<user_id>`
用户如果要用共享数据/缓存:
- 共享数据:`$HOME/common/datasets/...`
- 共享 HF 缓存:`$HOME/common/hf/...`(通常不需要写进 command,但可用于 debug)
#### 2.3.1 重要说明:SFTPGo “virtual folder” 与训练进程看到的“真实路径”
在 SFTPGo 中,`$HOME/common/datasets` / `$HOME/common/hf` 是 **SFTP 虚拟目录映射**(virtual folder),它们映射到容器内真实路径:
- `$HOME/common/datasets` ↔ `/private/datasets`
- `$HOME/common/hf` ↔ `/private/hf`
训练进程(Ray worker 上的 python 进程)看到的是 **容器内真实文件系统**,它并不会理解 SFTPGo 的 virtual folder。
因此,为了让用户能沿用 WebClient 里看到的路径语义(写 `$HOME/common/...`),服务层在提交 Advanced command 前需要做 **路径宏映射**:
- `"$HOME/common/datasets"` → `"/private/datasets"`
- `"$HOME/common/hf"` → `"/private/hf"`
- 其余 `"$HOME"` → `"/private/users/<user_id>"`
这样用户写的 command 能在训练进程里正确读到文件。
### 2.4 服务层检查(best-effort,强约束 + 弱约束)
> 目标:在不“解析完整 shell”的前提下,尽可能避免跨用户读文件与明显错误的任务。
**强约束(必须通过,否则 400)**
1) `nnodes`、`n_gpus_per_node` 必须存在(用于队列/资源预检查/placement)
2) `command` 必须包含一个明确的 python entry:
- 建议最低要求:包含 `python3` 且包含 `-m verl.trainer.`(防止随意执行系统命令)
3) 路径隔离校验(字符串/正则级别):
- 展开 `$HOME`(含 `$HOME/common/*` 映射到 `/private/*`)后:
- 禁止出现 `/private/users/` 下 “非当前用户”的路径(例如 `/private/users/bob/...`)
- 对 `data.train_files=...`、`data.val_files=...`(若出现)做 allowlist:
- 允许(用户目录):`/private/users/<me>/datasets/...`
- 允许(共享目录):`/private/datasets/...`
- 对 `custom_reward_function.path=...`(若出现)做 allowlist:
- 允许:`/private/users/<me>/code/...`(用户自行上传)
**弱约束(warning,不阻塞)**
- 未检测到 `data.train_files=`/`data.val_files=`(可能是用户写成了别的 key 或使用了 config file)
- 未检测到 `+ray_kwargs.ray_init.address=auto`(v3.0/v3.5 推荐加,但用户可自行负责)
> 说明:Advanced command 本质上属于“内部可信用户”能力,v3.5 不做强沙箱;安全检查以 best-effort 为主。
---
## 3. Custom Reward(仅方式 A:用户自己写)
### 3.1 VERL 原生机制(本仓库 `verl/` 已调研)
VERL PPO trainer 配置里支持:
- `custom_reward_function.path`
- `custom_reward_function.name`
- `custom_reward_function.reward_kwargs`
对应实现位置:
- 配置模板:`verl/verl/trainer/config/ppo_trainer.yaml`
- 加载逻辑:`verl/verl/trainer/ppo/reward.py:get_custom_reward_fn`
- 典型 reward manager:`verl/verl/workers/reward_manager/naive.py` 会调用 `compute_score(...)`
### 3.2 用户写法(示例)
用户上传 `$HOME/code/reward.py`,在 command 里加:
```bash
custom_reward_function.path=$HOME/code/reward.py \
custom_reward_function.name=compute_score
```
函数签名建议(与 `naive` reward manager 参数对齐):
```python
def compute_score(*, data_source: str, solution_str: str, ground_truth: str, extra_info=None, **kwargs):
...
```
### 3.3 平台侧只做检查(不做字段扩展)
v3.5 限定 reward 注入方式为 “用户写 command”,平台只做:
- 展开 `$HOME`
- 若检测到 `custom_reward_function.path=`,校验 path 在 `$HOME/code/` 下
- 不尝试解析/合并 reward_kwargs(用户自己写)
---
## 4. 服务层与 SFTPGo 的映射修改(你提出的关键点)
v3.0 时代平台允许用户引用:
- `/private/common/datasets/...`
- `/private/common/hf/...`
但现在 common 以 **SFTPGo virtual folder** 的形式呈现给用户(用户看到 `$HOME/common/...`,真实路径是 `/private/...`),因此 v3.5 的服务层需要做两件事:
1) **用户侧语义(写 TaskSpec/command)**
- 共享 datasets(只读):`$HOME/common/datasets/...`
- 共享 hf cache(只读):`$HOME/common/hf/...`
2) **运行时真实路径(提交到 Ray 前展开)**
- `$HOME/common/datasets/...` → `/private/datasets/...`
- `$HOME/common/hf/...` → `/private/hf/...`
同时保留用户自有目录:
- 用户 datasets:`$HOME/datasets/...`
- 用户 models:`$HOME/models/...`
- 用户 code(reward):`$HOME/code/...`
> 这部分主要影响:
> - Advanced command 检查(allowlist)
> - WebUI/Data 页面文案(告诉用户共享数据在哪里)
> 兼容性建议:为了不影响 v3.0 期间已经习惯使用 `/private/common/datasets/...` 的用户/历史任务,
> v3.5 实现阶段建议 **同时接受**:
> - `/private/common/datasets/...`(旧路径语义,仍可读)
> - `/private/datasets/...`(真实路径语义,推荐)
> - Advanced command 里写的 `$HOME/common/datasets/...` 会先映射到 `/private/datasets/...`
---
## 5. 验收标准(精简版)
### 5.1 Advanced command
- 提交一个 Advanced PPO command(train/val 使用 `$HOME/common/datasets/...` 或 `$HOME/datasets/...`)
- 确认:
- 任务从 QUEUED → SUBMITTED/RUNNING
- driver 在 worker 上(head 不跑训练)
- 训练能正常跑至少若干 step
### 5.2 Custom reward(方式 A)
- 用户上传 `$HOME/code/reward.py`
- 在 command 中设置 `custom_reward_function.path=$HOME/code/reward.py`
- 确认训练日志出现 `using customized reward function ...`
---
## 6. 待确认问题(需要你拍板/补充)
1) Advanced command 的“强约束”是否需要更严格?
- 目前建议要求包含 `python3 -m verl.trainer.`,否则拒绝。
- 你是否允许用户跑非 verl 的命令(例如自定义评估脚本)?
2) `$HOME/common/datasets` 与 `$HOME/common/hf` 两个映射目录在平台侧是否需要“强制只读”语义?
- 例如:TaskSpec 校验允许读取但禁止写入(目前设计是 best-effort 字符串级校验)。
---
## 7. 基于现有源码的改动点分析(实现清单)
本节按当前 v3.0 已上线的源码结构(`src/mvp/py/argus/...`)逐文件列出 v3.5 需要的具体改动点,并评估对现有能力的影响面。
### 7.1 TaskSpec/模型层(解析与兼容)
**现状**
- Basic TaskSpec 由 `argus.ray.models.JobSpec.from_dict()` 解析:`src/mvp/py/argus/ray/models.py`
- API `/api/v2/tasks` 直接 `JobSpec.from_dict(obj)`,并基于字段做路径校验:`src/mvp/py/argus/service/app.py`
- Scheduler 同样假定 jobspec_yaml 能解析为 `JobSpec`:`src/mvp/py/argus/service/scheduler.py`
**v3.5 需要新增**
1) 新增 `AdvancedTaskSpec` 数据结构(建议放在 `src/mvp/py/argus/ray/models.py`):
- 必填:`kind: advanced`、`workload`(建议仍要求 ppo/grpo/sft,用于 task_id 命名与 UI 分类)、`nnodes`、`n_gpus_per_node`、`command`
- 可选:`submission_id`(由服务层 override)
2) 新增 “union 解析”:
- 新增 `parse_taskspec(obj: dict) -> Basic(JobSpec) | Advanced(AdvancedTaskSpec)`
- 兼容策略:如果没有 `kind` 字段,则 **默认按 v3.0 Basic JobSpec 解析**(保证老客户端无感)。
### 7.2 Builder 层(把 TaskSpec 转为可执行 argv)
**现状**
- `src/mvp/py/argus/ray/builders.py:build_training_argv(spec: JobSpec, ...)` 只支持模板化 PPO/GRPO/SFT。
**v3.5 需要新增**
1) 新增 `build_advanced_argv(command: str) -> list[str]`
- 推荐实现:返回 `["bash", "-lc", "<expanded_command>"]`
- 原因:用户 command 允许 `ENV=... python3 ... \` 多行以及 shell 语法,`bash -lc` 兼容性最好。
2) Driver entrypoint 复用:
- 仍通过 `argus.ray.driver_entrypoint` 执行(统一 job_dir、日志与退出码)。
### 7.3 RayJobTool 层(runtime_env 与提交)
**现状**
- `src/mvp/py/argus/ray/ray_job_tool.py:RayJobTool.submit(spec: JobSpec, ...)`:
- runtime_env 的 `PYTHONPATH` 由 `spec.code_path` 决定
- entrypoint 固定为 driver_entrypoint + builder 生成 argv
**v3.5 需要新增**
1) 扩展 submit 支持 AdvancedTaskSpec:
- 方案 A(最小侵入):新增 `submit_advanced(...)` 方法,参数为 `command` + `job_dir` + `submission_id` + `nnodes/n_gpus...`
- 方案 B(统一接口):新增内部抽象 `SubmitPlan`(包含 `runtime_env` + `entrypoint` + `artifacts`),Basic/Advanced 都生成 plan,再走同一 submit 逻辑。
2) runtime_env 的 code path:
- 因 v3.5 本轮不做“自定义 verl code_path”,建议仍固定使用公共快照(例如 `/private/common/code/verl/verl_repo`)。
- 为减少散落常量,建议在 config 增加 `ray.verl_code_path`(或 `service.verl_code_path`),RayJobTool 统一读取。
3) runtime_env 的用户代码目录(可选增强):
- VERL 的自定义 reward 函数是通过 `custom_reward_function.path` 以“文件路径”动态 import 的,理论上不依赖 `PYTHONPATH`。
- 但用户的 `reward.py` 可能会 `import` 自己目录下的其他模块;为了提升易用性,可将
`/private/users/<user>/code` 追加到 job 的 `PYTHONPATH`。
- 这需要 RayJobTool.submit/submit_advanced 能感知 `user_id`(由 Scheduler 传入),属于小改动但要注意兼容性。
### 7.4 API Server(提交校验、宏替换、spec 展示)
**现状**
- `POST /api/v2/tasks`:只支持 Basic JobSpec 且强校验 `code_path/train_file/val_file/model_id` 前缀:`src/mvp/py/argus/service/app.py`
- `/api/v2/tasks/{task_id}/spec`:返回 resolved 的 Basic JobSpec(补默认值/补 submission_id):`src/mvp/py/argus/service/app.py`
**v3.5 需要新增/修改**
1) `POST /api/v2/tasks` 分流:
- `kind != advanced`:走原 Basic 流程(兼容 v3.0)
- `kind == advanced`:走 Advanced 解析 + 校验
2) Advanced command 宏替换与映射(核心):
- 实现 `expand_command(user_id, command)`:
- 先把 `$HOME/common/datasets` → `/private/datasets`
- 再把 `$HOME/common/hf` → `/private/hf`
- 再把其余 `$HOME` → `/private/users/<user>`
- 校验使用 “展开后的 command”
3) reward 注入检查(仅方式 A):
- 若发现 `custom_reward_function.path=...`:
- 校验展开后的 path 前缀必须是 `/private/users/<me>/code/`
4) `/api/v2/tasks/{task_id}/spec`:
- 需要支持返回 AdvancedTaskSpec 的 resolved 版本:
- 展示时可选择“原始 command”(含 `$HOME`)或“展开后的 command”(建议都展示:raw + expanded)
### 7.5 Scheduler(队列与提交)
**现状**
- `src/mvp/py/argus/service/scheduler.py` 假定 jobspec_yaml 一定是 Basic JobSpec,并调用 `tool.submit(spec2, ...)`。
**v3.5 需要新增**
1) Scheduler 的 `_parse_jobspec` 替换为 `parse_taskspec`(支持 Basic/Advanced)。
2) `_submit_one` 根据 spec 类型调用:
- Basic:保持现状 `tool.submit(JobSpec, ...)`
- Advanced:调用 `tool.submit_advanced(...)`(或统一 SubmitPlan)
### 7.6 WebUI(最小改动)
**现状**
- `src/mvp/py/argus/service/ui.py` 的 New Task 页面只提供 Basic YAML 模板。
**v3.5 需要新增**
- 增加 “Advanced Task” 模板按钮:
- `kind: advanced`
- `workload: ppo|grpo|sft`(用于 UI 分类与 task_id)
- `nnodes/n_gpus_per_node`
- `command: | ...`(带中文注释)
- Data 页面文案更新:
- 明确共享目录在 `$HOME/common/datasets`、`$HOME/common/hf`(并解释会映射到 `/private/datasets`、`/private/hf`)
---
## 8. 对现有功能的兼容性影响评估
### 8.1 API/TaskSpec 兼容
- 兼容策略:**没有 `kind` 字段的 YAML 一律按 v3.0 Basic JobSpec 解析**。
- 现有脚本/客户端(提交 ppo/grpo/sft 的 YAML)无需修改。
- AdvancedTaskSpec 是新增能力,不影响既有任务状态机/DB。
### 8.2 路径策略变更的影响
风险点:v3.0 的 Basic 任务/模板大量使用 `/private/common/datasets/...`。
建议:
- v3.5 实现阶段先保持 “双栈兼容”:
- Basic 继续接受 `/private/common/datasets/...`(旧)
- 同时接受 `/private/datasets/...`(新/真实路径)
- Advanced command 允许用户写 `$HOME/common/datasets/...`,服务层展开为 `/private/datasets/...`(避免虚拟目录不可见问题)。
### 8.3 任务执行/调度兼容
- Scheduler 队列/并发控制(`max_running_tasks`)保持不变。
- 资源预检查仍只依赖 `nnodes/n_gpus_per_node`,AdvancedTaskSpec 不改变资源模型。
### 8.4 安全边界变化
- Advanced command 引入后,平台从“结构化参数”变成“执行用户命令”,安全边界变宽。
- 缓解措施(best-effort):
- 强约束要求命令包含 `python3 -m verl.trainer.`
- 基础路径隔离校验(禁止跨用户路径)
- reward 文件路径限制在 `$HOME/code`
### 8.5 数据库兼容
- DB schema 不强制变更:仍复用 `tasks.jobspec_yaml` 存储原始 YAML。
- 若后续需要更强查询/过滤,再考虑增加 `tasks.kind` 字段(可选增量迁移)。

View File

@@ -0,0 +1,200 @@
# MVP v3.5(精简版)开发计划(TDD)
> 目标:在 v3.0 已有能力基础上,仅新增两项能力:
> 1) **Advanced TaskSpec(自定义 command)**
> 2) **Custom Reward(方式 A:用户自己在 command 里写 `custom_reward_function.*`)**
>
> 设计依据:`specs/mvp/v3.5/v3.5_design.md`(本计划不再扩展 scope)。
---
## 0. 范围与约束
### 0.1 In scope
- 新增 `kind: advanced` 的 TaskSpec:用户提供 `command`,平台做 `$HOME` 宏替换与 best-effort 校验,再提交 Ray Job。
- Custom Reward:平台仅做 **reward path 校验**(方式 A),不新增结构化字段。
- `$HOME/common/*` 路径语义支持(关键):用户在 SFTPGo/WebClient 看到的路径能被训练进程正确读取。
### 0.2 Out of scope(本轮不做)
- 自定义 verl 版本/代码路径(多版本共存)
- 断点续训(resume from checkpoint)
- IB/RoCEv2/NCCL 专项支持
- Model Serving
- Node management 改造(v3.0 的 stateless head/worker/watchdog/supervisor 机制保持不变)
### 0.3 关键路径映射(必须保持一致)
> 说明:SFTPGo 的 `$HOME/common/...` 是 **virtual folder**,训练进程看不到该虚拟路径。
提交 Advanced command 前必须展开/映射:
- `$HOME/common/datasets` → `/private/datasets`(只读语义)
- `$HOME/common/hf` → `/private/hf`(只读语义)
- 其余 `$HOME` → `/private/users/<user_id>`
并且为兼容历史用法(v3.0):
- Basic TaskSpec 仍接受 `/private/common/datasets/...`、`/private/common/hf/...`(不强制迁移)。
---
## 1. 测试策略(TDD)
### 1.1 单元测试优先级
1) **解析与兼容**:`kind: advanced` 能解析;无 `kind` 仍按 Basic 解析,旧用法不破坏。
2) **宏替换正确性**:`$HOME` / `$HOME/common/*` 映射严格按约定展开。
3) **best-effort 校验**:拒绝明显危险/跨用户路径;对 reward path 做 allowlist。
4) **提交链路**:Scheduler 能识别 Advanced spec 并调用对应的提交方法,确保 submission_id/目录规范不变。
5) **WebUI/API**:New Task 模板与 `/spec` 展示完整 resolved spec(包含展开后的 command)。
### 1.2 本地运行方式
- 复用已有 `.venv`,执行:`.venv/bin/python -m pytest`
- 若环境没有 pip,使用 uv 的方式参考 v3.0 约定(不在本计划重复)。
---
## 2. 里程碑划分(每个里程碑可独立验证)
> 约定:每个里程碑先写测试(失败),再实现代码使测试通过;里程碑结束跑一遍 `pytest`。
### M1 — TaskSpec 模型与解析(兼容优先)
**目标**
- 引入 AdvancedTaskSpec 数据结构与 union parser,同时保证 v3.0 Basic 行为不变。
**新增/修改(建议位置)**
- `src/mvp/py/argus/ray/models.py`
- 新增 `AdvancedTaskSpec`
- 新增 `parse_taskspec(obj: dict) -> JobSpec | AdvancedTaskSpec`
- 兼容策略:缺省 `kind` → 走 `JobSpec.from_dict`
**测试(先写)**
- `src/mvp/py/tests/test_models.py`
- `test_parse_taskspec_basic_no_kind_compat()`
- `test_parse_taskspec_advanced_smoke()`
- `test_parse_taskspec_advanced_requires_command_nnodes_gpus()`
**验收**
- `pytest -q` 通过;旧测试不修改或仅做最小必要更新。
---
### M2 — Advanced command 展开与校验(核心能力)
**目标**
- 实现 command 展开(含 `$HOME/common/*` 映射)与 best-effort 强约束校验。
**实现点(建议新增模块)**
- `src/mvp/py/argus/service/command_expand.py`(或放在 `argus/service/validation.py`)
- `expand_advanced_command(user_id: str, command: str) -> str`
- `validate_advanced_command(user_id: str, expanded_command: str) -> None`(失败抛 `ValueError`)
**强约束(与设计文档一致)**
- 必须包含 `python3` 且包含 `-m verl.trainer.`(否则 400)
- 禁止出现 `/private/users/<other>/...`(跨用户路径)
- 若检测到 `data.train_files=`/`data.val_files=`:
- 只允许 `/private/users/<me>/datasets/...` 或 `/private/datasets/...`
- (兼容)允许 `/private/common/datasets/...`(旧路径)
- 若检测到 `custom_reward_function.path=`:
- 只允许 `/private/users/<me>/code/...`(展开后校验)
**测试(先写)**
- 新增:`src/mvp/py/tests/test_advanced_command.py`
- `test_expand_maps_home_common_datasets_to_private_datasets()`
- `test_expand_maps_home_common_hf_to_private_hf()`
- `test_expand_maps_home_to_private_users()`
- `test_validate_rejects_cross_user_paths()`
- `test_validate_requires_verl_trainer_entry()`
- `test_validate_allows_reward_path_under_user_code()`
- `test_validate_rejects_reward_path_outside_user_code()`
**验收**
- 单测覆盖映射/校验的正反例;错误信息可读(用于 API 400 detail)。
---
### M3 — Ray 提交链路支持 Advanced(Builder/Tool/Scheduler)
**目标**
- Advanced spec 能进入 scheduler 队列并提交为 Ray job(driver 仍落 worker)。
**代码改动点(建议)**
- `src/mvp/py/argus/ray/builders.py`
- 新增 `build_advanced_argv(command: str)`:返回 `["bash","-lc", expanded_command]`
- `src/mvp/py/argus/ray/ray_job_tool.py`
- 新增 `submit_advanced(...)`(或统一成内部 submit plan)
- runtime_env:继续注入公共 verl code path(本轮不支持用户自定义 verl 代码)
- 可选:把 `/private/users/<user>/code` 加入 `PYTHONPATH`,提升 reward 代码 `import` 体验
- `src/mvp/py/argus/service/scheduler.py`
- 使用 `parse_taskspec` 分流 Basic/Advanced
- Advanced 调用 `tool.submit_advanced(...)`
**测试(先写)**
- `src/mvp/py/tests/test_builders.py`
- `test_build_advanced_argv_uses_bash_lc()`
- `src/mvp/py/tests/test_scheduler.py`
- 新增一个 `kind: advanced` 的任务,断言 scheduler 调用了 `submit_advanced`
- 断言 job_dir/submission_id 规则不变(仍按 `/private/users/<user>/jobs/<sid>`)
- `src/mvp/py/tests/test_ray_job_tool.py`
- 断言 advanced 提交时 entrypoint 是 driver_entrypoint + `bash -lc ...`
**验收**
- 单测跑通;Scheduler tick 能完成 Advanced 任务从 QUEUED → SUBMITTED(mock Ray)。
---
### M4 — API & WebUI(最小功能闭环)
**目标**
- WebUI/HTTP API 能提交 Advanced Task,并在详情页看到 resolved spec(含完整 command)。
**API 改动点**
- `src/mvp/py/argus/service/app.py`
- `POST /api/v2/tasks`:支持 `kind: advanced`
- 保存 raw YAML(保持与 Basic 一致)
- 对 Advanced:展开 command + 校验(失败返回 400)
- `GET /api/v2/tasks/{task_id}/spec`:
- 返回 resolved spec(建议同时返回 raw + expanded,或 YAML 中直接给 expanded)
**WebUI 改动点**
- `src/mvp/py/argus/service/ui.py`
- New Task 页面新增 Advanced 模板(含中文注释)
- 文案强调共享目录:`$HOME/common/datasets`、`$HOME/common/hf`
**测试(先写)**
- `src/mvp/py/tests/test_app.py`
- `test_create_task_advanced_ok()`(最小 valid command)
- `test_create_task_advanced_rejects_invalid_command()`
- `test_task_spec_endpoint_includes_expanded_command()`
- `src/mvp/py/tests/test_ui.py`
- 断言页面包含 Advanced 示例块
**验收**
- `pytest` 通过;浏览器可提交 Advanced YAML 并看到 expanded command。
---
### M5 — 端到端验证(远端 argus@h1)
**目标**
- 在真实 Ray cluster + VERL 环境下验证 Advanced 与 Custom Reward(方式 A)。
**步骤(手工验收脚本化可选)**
1) 启动 v3.0/v3.5 统一的 compose + API(沿用现有 `run_all` 脚本体系)
2) 用户(如 `alice`)通过 SFTP 上传 reward 代码到:
- `$HOME/code/reward.py`(真实路径 `/private/users/alice/code/reward.py`)
3) 通过 WebUI 或 curl 提交 Advanced task:
- `command` 中包含:
- `custom_reward_function.path=$HOME/code/reward.py`
- `custom_reward_function.name=compute_score`
- `data.train_files=$HOME/common/datasets/gsm8k/train.parquet`
- `data.val_files=$HOME/common/datasets/gsm8k/test.parquet`
4) 检查:
- 任务状态从 QUEUED → RUNNING → SUCCEEDED/FAILED(有日志)
- driver 不在 head 上跑(dashboard 验证)
- 日志出现 “custom reward” 生效的提示(按 VERL 实际日志关键字确认)
5) 回归:提交 Basic ppo/grpo/sft 任务仍可运行(确保兼容性)
**验收**
- Advanced task 能跑至少若干 step,且 reward 注入生效。
- Basic 任务兼容不回退。
---
## 3. 风险点与边界(明确写进 PR/变更说明)
- Advanced command 只做 best-effort 校验,不做完整 shell AST 解析;复杂命令可能存在漏检/误判(后续可扩展)。
- `$HOME/common/*` 是“用户侧语义”,服务层必须映射到真实路径,否则训练必然 FileNotFound。
- 校验策略(强约束)如果后续要允许非 VERL 命令,需要调整规则并补测试(本轮默认拒绝)。