Skip to content

[DYNAMO] wire TITO client to Dynamo renderer transport#2446

Draft
AmeenP wants to merge 1 commit into
feat/dynamo-renderer-transport-configfrom
codex/dynamo-tito-client-followup
Draft

[DYNAMO] wire TITO client to Dynamo renderer transport#2446
AmeenP wants to merge 1 commit into
feat/dynamo-renderer-transport-configfrom
codex/dynamo-tito-client-followup

Conversation

@AmeenP
Copy link
Copy Markdown
Contributor

@AmeenP AmeenP commented May 8, 2026

Summary

Ports the missing prime-rl client-side TITO wiring from Biswa's tested branches on top of #2420 (feat/dynamo-renderer-transport-config). The shared renderer/verifiers pins and return_token_ids cleanup now live in #2420, so this PR only carries the TITO-specific follow-up.

  • treats openai_chat_completions_token as token-aware for client.backend = "dynamo"
  • with [DYNAMO] pass renderer transport to verifiers #2420, uses renderer_transport="dynamo" for both renderer-mode and TITO clients when using Dynamo
  • adds unit coverage for Dynamo TITO transport selection

Biswa Compatibility Notes

Included from Biswa's rl-sdk-2 where it matches the current Dynamo contract:

Not included yet:

  • Biswa's compute_teacher_logprobs dispatch that asks Dynamo chat completions for nvext.extra_fields=["prompt_logprobs"]. Dynamo #9509 currently exposes engine_data / completion token data, not nvext.prompt_logprobs, so copying that prime-rl path would require a matching Dynamo change first.

Validation

Local:

  • uv lock --check
  • uvx ruff==0.13.0 check --config=pyproject.toml src/prime_rl/utils/client.py tests/unit/utils/test_client.py
  • uvx ruff==0.13.0 format --check --config=pyproject.toml src/prime_rl/utils/client.py tests/unit/utils/test_client.py

GitHub Actions:

  • Ruff: pass
  • Slim install (prime-rl-configs only): pass
  • Unit tests: pass

Local test note:

  • uv run pytest tests/unit/utils/test_client.py is blocked on macOS by the repo's Linux-only lockfile; uv run --no-sync pytest tests/unit/utils/test_client.py exited 139 locally.

@AmeenP AmeenP force-pushed the codex/dynamo-tito-client-followup branch from cfe5428 to 4a2503d Compare May 8, 2026 10:03
@AmeenP AmeenP force-pushed the feat/dynamo-renderer-transport-config branch from 9cad934 to 360f0c1 Compare May 8, 2026 10:03
@AmeenP AmeenP force-pushed the codex/dynamo-tito-client-followup branch 2 times, most recently from 3f89cf5 to 058755e Compare May 8, 2026 10:14
@AmeenP AmeenP force-pushed the codex/dynamo-tito-client-followup branch 2 times, most recently from 1c53674 to 0448473 Compare May 15, 2026 13:15
@AmeenP AmeenP force-pushed the codex/dynamo-tito-client-followup branch from 0448473 to ff77174 Compare May 15, 2026 14:18
@AmeenP AmeenP changed the title [DYNAMO] wire TITO client to Dynamo nvext transport [DYNAMO] wire TITO client to Dynamo renderer transport May 15, 2026
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