fix(claude_agent_sdk): log ResultMessage metrices

Author: Mahhheshh

py/src/braintrust/integrations/claude_agent_sdk/test_claude_agent_sdk.py

@@ -172,7 +172,16 @@ async def calculator_handler(args):
         assert task_span.get("metadata", {}).get("num_turns") is not None
     if hasattr(result_message, "session_id"):
         assert task_span.get("metadata", {}).get("session_id") is not None
-
+    if hasattr(result_message, "stop_reason"):
+        assert task_span.get("metadata", {}).get("stop_reason") is not None
+    if hasattr(result_message, "stop_sequence"):
+        assert task_span.get("metadata", {}).get("stop_sequence") is not None
+    if hasattr(result_message, "total_cost_usd"):
+        assert task_span.get("metadata", {}).get("total_cost_usd") is not None
+    if hasattr(result_message, "duration_ms"):
+        assert task_span.get("metadata", {}).get("duration_ms") is not None
+    if hasattr(result_message, "duration_api_ms"):
+        assert task_span.get("metadata", {}).get("duration_api_ms") is not None
     llm_spans = [s for s in spans if s["span_attributes"]["type"] == SpanTypeAttribute.LLM]
     assert len(llm_spans) >= 1, f"Should have at least one LLM span, got {len(llm_spans)}"
     llm_span_ids = {span["span_id"] for span in llm_spans}

py/src/braintrust/integrations/claude_agent_sdk/tracing.py

@@ -730,6 +730,11 @@ def _handle_result(self, message: Any) -> None:
             for k, v in {
                 "num_turns": getattr(message, "num_turns", None),
                 "session_id": getattr(message, "session_id", None),
+                "stop_reason": getattr(message, "stop_reason", None),
+                "stop_sequence": getattr(message, "stop_sequence", None),
+                "total_cost_usd": getattr(message, "total_cost_usd", None),
+                "duration_ms": getattr(message, "duration_ms", None),
+                "duration_api_ms": getattr(message, "duration_api_ms", None),
             }.items()
             if v is not None
         }