Skip to content

APR-57: add teacher model discovery flag#745

Open
AmeenP wants to merge 1 commit into
feat/hosted-opd-clifrom
apr-57-teacher-models
Open

APR-57: add teacher model discovery flag#745
AmeenP wants to merge 1 commit into
feat/hosted-opd-clifrom
apr-57-teacher-models

Conversation

@AmeenP

@AmeenP AmeenP commented Jun 15, 2026

Copy link
Copy Markdown
Contributor

Summary

  • add a typed Hosted Training client method for GET /rft/teacher-models
  • add prime train models --teachers to list generate-capable teacher inference models for OPD/SFT-style teacher configs
  • keep default prime train models unchanged because it lists trainable student base models and capacity, not teacher scoring candidates

Stack

platform#2342 pi-inference execution
  POST /api/v1/generate
  POST /api/inference/v1/generate
          |
          v
platform#3007 pi-inference discovery
  GET /api/internal/models -> capabilities.generate
          |
          v
platform#3008 RFT consumption
  GET /api/v1/rft/teacher-models
          |
          v
prime#699 OPD config surface
  loss = "opd" + [teacher]
          |
          v
this PR CLI discovery
  prime train models --teachers

Validation

  • .venv/bin/ruff format packages/prime/src/prime_cli/api/rl.py packages/prime/src/prime_cli/commands/rl.py packages/prime/tests/test_rl_api.py packages/prime/tests/test_rl_models.py
  • .venv/bin/ruff check packages/prime/src/prime_cli/api/rl.py packages/prime/src/prime_cli/commands/rl.py packages/prime/tests/test_rl_api.py packages/prime/tests/test_rl_models.py
  • .venv/bin/python -m pytest packages/prime/tests/test_rl_api.py packages/prime/tests/test_rl_models.py (20 passed)
  • git diff --check

Linear: APR-57

@AmeenP AmeenP force-pushed the apr-57-teacher-models branch from a016654 to f1b8e59 Compare June 15, 2026 14:18
@AmeenP AmeenP changed the title APR-57: add teacher model discovery command APR-57: add teacher model discovery flag Jun 15, 2026
@AmeenP AmeenP force-pushed the feat/hosted-opd-cli branch from 45322b6 to 961bcdb Compare June 15, 2026 14:28
@AmeenP AmeenP force-pushed the apr-57-teacher-models branch from f1b8e59 to 5d46715 Compare June 15, 2026 14:29
@AmeenP AmeenP marked this pull request as ready for review June 15, 2026 14:30
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant