Send display config in Prime monitor registration

Author: d42me

src/prime_rl/utils/monitor/prime.py

@@ -182,10 +182,16 @@ def _register_run(self, config: PrimeMonitorConfig, run_config: BaseConfig | Non
         environments = getattr(run_config, "env", None) if run_config else None
         wandb = getattr(run_config, "wandb", None) if run_config else None
 
-        payload: dict = {
+        payload: dict[str, Any] = {
             "base_model": model.name if model else "unknown",
             "max_steps": getattr(run_config, "max_steps", None) or 0,
         }
+        if run_config:
+            payload["run_config"] = run_config.model_dump(exclude_none=True, mode="json")
+            for field in ("batch_size", "rollouts_per_example", "seq_len"):
+                value = getattr(run_config, field, None)
+                if value is not None:
+                    payload[field] = value
         if config.run_name:
             payload["name"] = config.run_name
         if team_id:

tests/unit/utils/test_prime_monitor.py

@@ -116,6 +116,67 @@ def test_rollouts_to_parquet_bytes_skips_rollouts_without_trajectory():
     assert rows[0]["sample_id"] == 0
 
 
+def test_register_run_sends_display_config_and_run_config(monkeypatch):
+    run_config_dump = {
+        "max_steps": 30,
+        "batch_size": 64,
+        "rollouts_per_example": 16,
+        "seq_len": 4096,
+    }
+    run_config = types.SimpleNamespace(
+        model=types.SimpleNamespace(name="PrimeIntellect/Qwen3-0.6B-Reverse-Text-SFT"),
+        env=[types.SimpleNamespace(id="primeintellect/reverse-text")],
+        wandb=types.SimpleNamespace(project="prime-rl-e2e"),
+        max_steps=30,
+        batch_size=64,
+        rollouts_per_example=16,
+        seq_len=4096,
+        model_dump=MagicMock(return_value=run_config_dump),
+    )
+    monitor = _make_monitor(
+        api_key="pit-test",
+        base_url="https://api.primeintellect.ai/api/v1/rft",
+        logger=MagicMock(),
+    )
+    config = types.SimpleNamespace(
+        base_url="https://api.primeintellect.ai/api/v1/rft",
+        api_key_var="PRIME_API_KEY",
+        team_id="team-1",
+        frontend_url=None,
+        run_name="e2e-run",
+    )
+    response = MagicMock(status_code=201)
+    response.json.return_value = {"run": {"id": "run-123"}}
+    post = MagicMock(return_value=response)
+    monkeypatch.setattr("prime_rl.utils.monitor.prime.httpx.post", post)
+
+    run_id = monitor._register_run(config, run_config)
+
+    assert run_id == "run-123"
+    run_config.model_dump.assert_called_once_with(exclude_none=True, mode="json")
+    post.assert_called_once_with(
+        "https://api.primeintellect.ai/api/v1/rft/external-runs",
+        headers={
+            "Authorization": "Bearer pit-test",
+            "x-api-key": "pit-test",
+            "Content-Type": "application/json",
+        },
+        json={
+            "base_model": "PrimeIntellect/Qwen3-0.6B-Reverse-Text-SFT",
+            "max_steps": 30,
+            "run_config": run_config_dump,
+            "batch_size": 64,
+            "rollouts_per_example": 16,
+            "seq_len": 4096,
+            "name": "e2e-run",
+            "team_id": "team-1",
+            "environments": [{"id": "primeintellect/reverse-text"}],
+            "wandb_project": "prime-rl-e2e",
+        },
+        timeout=30,
+    )
+
+
 def test_api_headers_include_bearer_and_x_api_key():
     monitor = _make_monitor(api_key="pit-test")