Skip to content

Commit 22602b3

Browse files
committed
update naming
1 parent 38f733b commit 22602b3

File tree

2 files changed

+10
-10
lines changed

2 files changed

+10
-10
lines changed

eval_protocol/log_utils/fireworks_tracing_http_handler.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -94,16 +94,16 @@ def _get_status_info(self, record: logging.LogRecord) -> Optional[Dict[str, Any]
9494
status_code = status_code.value
9595

9696
return {
97-
"status_code": status_code,
98-
"status_message": status.message,
99-
"status_details": getattr(status, "details", []),
97+
"code": status_code,
98+
"message": status.message,
99+
"details": getattr(status, "details", []),
100100
}
101101
elif isinstance(status, dict):
102102
# Dictionary representation of status
103103
return {
104-
"status_code": status.get("code"),
105-
"status_message": status.get("message"),
106-
"status_details": status.get("details", []),
104+
"code": status.get("code"),
105+
"message": status.get("message"),
106+
"details": status.get("details", []),
107107
}
108108
return None
109109

eval_protocol/pytest/remote_rollout_processor.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -151,10 +151,10 @@ def _get_status() -> Dict[str, Any]:
151151

152152
# Look for structured status dictionary in status field
153153
status_dict = latest_log.get("status")
154-
if status_dict and isinstance(status_dict, dict) and "status_code" in status_dict:
155-
status_code = status_dict.get("status_code")
156-
status_message = status_dict.get("status_message", "")
157-
status_details = status_dict.get("status_details", [])
154+
if status_dict and isinstance(status_dict, dict) and "code" in status_dict:
155+
status_code = status_dict.get("code")
156+
status_message = status_dict.get("message", "")
157+
status_details = status_dict.get("details", [])
158158

159159
logger.info(
160160
f"Found Fireworks log for rollout {row.execution_metadata.rollout_id} with status code {status_code}"

0 commit comments

Comments
 (0)