Skip to content

Commit fd4358c

Browse files
committed
fix: preserve rollout status extras
Read rollout status extras from the top-level status response so RemoteRolloutProcessor preserves metadata that previously came from log entries. Made-with: Cursor
1 parent 8e0e0ef commit fd4358c

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

eval_protocol/adapters/fireworks_tracing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -379,7 +379,7 @@ async def async_get_status(self, session: aiohttp.ClientSession, rollout_id: str
379379
"""Fetch rollout status from the lightweight /status endpoint.
380380
381381
Returns the parsed JSON response or None if the status is not yet available.
382-
Response shape: {"rollout_id": "...", "status": {"code": ...} | null}
382+
Response shape: {"rollout_id": "...", "status": {"code": ...} | null, "extras": {...} | null}
383383
"""
384384
headers = {
385385
"Authorization": f"Bearer {self._get_api_key()}",

eval_protocol/pytest/remote_rollout_processor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ async def _process_row(row: EvaluationRow) -> EvaluationRow:
152152
details=status_details,
153153
)
154154

155-
status_extras = status.get("extras")
155+
status_extras = (status_result or {}).get("extras")
156156
if isinstance(status_extras, dict):
157157
if row.execution_metadata.extra:
158158
row.execution_metadata.extra.update(status_extras)

0 commit comments

Comments
 (0)