Fix tool name extraction from ToolUseBlock messages

- Extract tool name, parameters, and ID directly from ToolUseBlock attributes instead of parsing from string
- Extract tool_use_id from ToolResultBlock to link results to tool calls
- Keeps fallback to regex-based extraction for edge cases
- Fixes issue where many tool calls showed as 'unknown' in dashboard

Author: amrit110

scripts/agent_tracer.py

@@ -278,12 +278,29 @@ async def capture_agent_stream(
                     "content": content if content else str(message),
                 }
 
-                # Extract tool info if applicable
-                tool_info = self.extract_tool_info(
-                    content if content else str(message), event_type
-                )
-                if tool_info:
-                    event.update(tool_info)
+                # Extract tool info for ToolUseBlock messages
+                if msg_class == "ToolUseBlock":
+                    # Extract directly from message attributes
+                    tool_name = getattr(message, "name", None)
+                    tool_input = getattr(message, "input", {})
+                    tool_id = getattr(message, "id", None)
+                    if tool_name:
+                        event["tool"] = tool_name
+                        event["parameters"] = tool_input
+                        if tool_id:
+                            event["tool_use_id"] = tool_id
+                elif msg_class == "ToolResultBlock":
+                    # Extract tool_use_id to link result to tool call
+                    tool_use_id = getattr(message, "tool_use_id", None)
+                    if tool_use_id:
+                        event["tool_use_id"] = tool_use_id
+                elif event_type == "TOOL_CALL":
+                    # Fallback: try to extract from content string
+                    tool_info = self.extract_tool_info(
+                        content if content else str(message), event_type
+                    )
+                    if tool_info:
+                        event.update(tool_info)
 
                 self.trace["events"].append(event)