feat(experimentalist): add Tau3 experimentalist workflow to docs - #999
Conversation
📝 WalkthroughWalkthroughThe PR documents an end-to-end τ-Bench optimization example, adds Tau3 Airline smoke dataset preparation and bounded Experimentalist settings, and propagates the configured or environment-provided model into the agent container. ChangesTau3 Airline smoke workflow
Sequence Diagram(s)sequenceDiagram
participant User
participant prepare-airline-smoke.sh
participant Experimentalist
participant Tau3 NOOA agent
User->>prepare-airline-smoke.sh: Prepare train and validation smoke datasets
prepare-airline-smoke.sh-->>User: Return dataset paths
User->>Experimentalist: Start bounded optimization run
Experimentalist->>Tau3 NOOA agent: Launch candidate with model environment
Tau3 NOOA agent-->>Experimentalist: Return evaluation results
Experimentalist-->>User: Write selected winner to run.json
Possibly related PRs
Suggested labels: Suggested reviewers: 🚥 Pre-merge checks | ✅ 5✅ Passed checks (5 passed)
✨ Finishing Touches📝 Generate docstrings
🧪 Generate unit tests (beta)
Comment |
There was a problem hiding this comment.
Actionable comments posted: 1
🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.
Inline comments:
In
`@plugins/nemo-experimentalist/examples/tau3-nooa-agent/prepare-airline-smoke.sh`:
- Around line 27-36: Update the dataset cache guard in prepare-airline-smoke.sh
to verify that every selected task in TRAIN_TASKS has its task.toml under
DATASET_ROOT before skipping the download. Redownload via the existing uv run
harbor download block whenever any of the five tasks is missing, while
preserving reuse only when all task files exist.
🪄 Autofix (Beta)
Fix all unresolved CodeRabbit comments on this PR:
- Push a commit to this branch (recommended)
- Create a new PR with the fixes
ℹ️ Review info
⚙️ Run configuration
Configuration used: Path: .coderabbit.yaml
Review profile: CHILL
Plan: Enterprise
Run ID: 6ed37ac1-9909-4a5c-9f06-347b3fe6b82d
📒 Files selected for processing (6)
docs/get-started/example-agent.mdxplugins/nemo-experimentalist/examples/tau3-nooa-agent/README.mdplugins/nemo-experimentalist/examples/tau3-nooa-agent/experimentalist-smoke.yamlplugins/nemo-experimentalist/examples/tau3-nooa-agent/harbor_wrapper.pyplugins/nemo-experimentalist/examples/tau3-nooa-agent/prepare-airline-smoke.shplugins/nemo-experimentalist/tests/experimentalist/test_tau3_nooa_agent.py
|
Provide a bounded official-dataset workflow that customers can run through the NeMo CLI while recording Platform traces. Signed-off-by: Nico Tonozzi <ntonozzi@nvidia.com>
0045319 to
50dd92c
Compare
There was a problem hiding this comment.
Actionable comments posted: 1
🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.
Inline comments:
In `@docs/get-started/example-agent.mdx`:
- Line 88: Correct the spelling in the agent performance sentence by changing
“evalaution” to “evaluation,” without altering the surrounding text.
🪄 Autofix (Beta)
Fix all unresolved CodeRabbit comments on this PR:
- Push a commit to this branch (recommended)
- Create a new PR with the fixes
ℹ️ Review info
⚙️ Run configuration
Configuration used: Path: .coderabbit.yaml
Review profile: CHILL
Plan: Enterprise
Run ID: 7c0cfd0c-e4c0-4629-b9f7-23447c0600dc
📒 Files selected for processing (4)
docs/get-started/example-agent.mdxplugins/nemo-experimentalist/examples/tau3-nooa-agent/experimentalist-smoke.yamlplugins/nemo-experimentalist/examples/tau3-nooa-agent/harbor_wrapper.pyplugins/nemo-experimentalist/examples/tau3-nooa-agent/prepare-airline-smoke.sh
Summary
This is not a great UX yet -- my next step will be to go back and clean up the docs and remove tau2, but this is working for now and I want to let others test it out.
Summary by CodeRabbit
New Features
Bug Fixes