support arbitrary logs

Author: xzrderek

eval_protocol/adapters/fireworks_tracing.py

@@ -268,7 +268,7 @@ def __init__(
     def search_logs(self, tags: List[str], limit: int = 100, hours_back: int = 24) -> List[Dict[str, Any]]:
         """Fetch logs from Fireworks tracing gateway /logs endpoint.
 
-        Returns entries with keys: timestamp, message, severity, tags.
+        Returns entries with keys: timestamp, message, severity, tags, status, extras.
         """
         if not tags:
             raise ValueError("At least one tag is required to fetch logs")
@@ -315,6 +315,7 @@ def search_logs(self, tags: List[str], limit: int = 100, hours_back: int = 24) -
                     "severity": e.get("severity", "INFO"),
                     "tags": e.get("tags", []),
                     "status": e.get("status"),
+                    "extras": e.get("extras"),
                 }
             )
         return results

eval_protocol/log_utils/fireworks_tracing_http_handler.py

@@ -125,14 +125,16 @@ def _build_payload(self, record: logging.LogRecord, rollout_id: str) -> Dict[str
                 pass
         program = cast(Optional[str], getattr(record, "program", None)) or "eval_protocol"
 
+        extras_input = getattr(record, "extras", None)
+        extras: Dict[str, Any] = extras_input if isinstance(extras_input, dict) else {}
+        extras["logger_name"] = record.name
+        extras["level"] = record.levelname
+        extras["timestamp"] = timestamp
+
         return {
             "program": program,
             "status": self._get_status_info(record),
             "message": message,
             "tags": tags,
-            "extras": {
-                "logger_name": record.name,
-                "level": record.levelname,
-                "timestamp": timestamp,
-            },
+            "extras": extras,
         }

typescript/logging/fireworks-transport.ts

@@ -14,6 +14,7 @@ interface FireworksLogInfo extends TransformableInfo {
   status?: any;
   program?: string;
   logger_name?: string;
+  extras?: Record<string, any>;
   [key: string]: any;
 }
 
@@ -32,6 +33,7 @@ interface FireworksPayload {
     logger_name: string;
     level: string;
     timestamp: string;
+    [key: string]: any;
   };
 }
 
@@ -208,12 +210,18 @@ export class FireworksTransport extends Transport {
 
     const program = (typeof info.program === 'string' ? info.program : null) || 'eval_protocol';
 
+    const extraInput =
+      info.extras && typeof info.extras === 'object' && !Array.isArray(info.extras)
+        ? info.extras
+        : {};
+
     return {
       program,
       status: this.getStatusInfo(info),
       message,
       tags,
       extras: {
+        ...extraInput,
         logger_name: info.logger_name || 'winston',
         level: level.toUpperCase(),
         timestamp,