Python: fix(ag-ui): read tool_call_id on TOOL_CALL_RESULT events

python/packages/ag-ui/agent_framework_ag_ui/_event_converters.py

@@ -197,7 +197,7 @@ def _handle_tool_call_end(self, event: dict[str, Any]) -> ChatResponseUpdate | N
 
     def _handle_tool_call_result(self, event: dict[str, Any]) -> ChatResponseUpdate:
         """Handle TOOL_CALL_RESULT event."""
-        tool_call_id = event.get("toolCallId", "")
+        tool_call_id = self._get_tool_call_id(event) or ""
         result = event.get("result") if event.get("result") is not None else event.get("content")
 
         return ChatResponseUpdate(

python/packages/ag-ui/tests/ag_ui/test_event_converters.py

@@ -195,6 +195,20 @@ def test_tool_call_result_event(self) -> None:
         assert update.contents[0].call_id == "call_123"
         assert update.contents[0].result == '{"temperature": 22, "condition": "sunny"}'
 
+    def test_tool_call_result_event_snake_case_id(self) -> None:
+        """TOOL_CALL_RESULT accepts tool_call_id like the other tool call events."""
+        converter = AGUIEventConverter()
+        event = {
+            "type": "TOOL_CALL_RESULT",
+            "tool_call_id": "call_snake",
+            "content": "sunny",
+        }
+
+        update = converter.convert_event(event)
+
+        assert update is not None
+        assert update.contents[0].call_id == "call_snake"
+
     def test_run_finished_event(self) -> None:
         """Test conversion of RUN_FINISHED event."""
         converter = AGUIEventConverter()