encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix bug where remote a2a agent wasn't using its a2a part converter

Browse Source 
PiperOrigin-RevId: 836399603
This commit is contained in:
Google Team Member
2025-11-24 16:18:18 -08:00
committed by Copybara-Service
parent a1c09b724b
commit 4eb2a11403
2 changed files with 34 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unittests/agents/test_remote_a2a_agent.py
+24 -3
View File
@@ -723,6 +723,7 @@ class TestRemoteA2aAgentMessageHandling:
mock_a2a_message,
self.agent.name,
self.mock_context,
self.mock_a2a_part_converter,
)
# Check that metadata was added
assert result.custom_metadata is not None
@@ -760,6 +761,7 @@ class TestRemoteA2aAgentMessageHandling:
mock_a2a_task,
self.agent.name,
self.mock_context,
self.mock_a2a_part_converter,
)
# Check the parts are not updated as Thought
assert result.content.parts[0].thought is None
@@ -864,6 +866,7 @@ class TestRemoteA2aAgentMessageHandling:
mock_a2a_task,
self.agent.name,
self.mock_context,
self.mock_a2a_part_converter,
)
# Check the parts are updated as Thought
assert result.content.parts[0].thought is True
@@ -909,6 +912,7 @@ class TestRemoteA2aAgentMessageHandling:
mock_a2a_message,
self.agent.name,
self.mock_context,
self.mock_a2a_part_converter,
)
# Check that metadata was added
assert result.custom_metadata is not None
@@ -954,6 +958,7 @@ class TestRemoteA2aAgentMessageHandling:
mock_a2a_message,
self.agent.name,
self.mock_context,
self.mock_a2a_part_converter,
)
# Check that metadata was added
assert result.custom_metadata is not None
@@ -1009,7 +1014,10 @@ class TestRemoteA2aAgentMessageHandling:
assert result == mock_event
mock_convert.assert_called_once_with(
mock_a2a_task, self.agent.name, self.mock_context
mock_a2a_task,
self.agent.name,
self.mock_context,
self.agent._a2a_part_converter,
)
# Check that metadata was added
assert result.custom_metadata is not None
@@ -1039,6 +1047,8 @@ class TestRemoteA2aAgentMessageHandlingFromFactory:
def setup_method(self):
"""Setup test fixtures."""
self.mock_a2a_part_converter = Mock()
self.agent_card = create_test_agent_card()
self.agent = RemoteA2aAgent(
name="test_agent",
@@ -1046,6 +1056,7 @@ class TestRemoteA2aAgentMessageHandlingFromFactory:
a2a_client_factory=ClientFactory(
config=ClientConfig(httpx_client=httpx.AsyncClient()),
),
a2a_part_converter=self.mock_a2a_part_converter,
)
# Mock session and context
@@ -1173,7 +1184,10 @@ class TestRemoteA2aAgentMessageHandlingFromFactory:
assert result == mock_event
mock_convert.assert_called_once_with(
mock_a2a_message, self.agent.name, self.mock_context
mock_a2a_message,
self.agent.name,
self.mock_context,
self.mock_a2a_part_converter,
)
# Check that metadata was added
assert result.custom_metadata is not None
@@ -1211,6 +1225,7 @@ class TestRemoteA2aAgentMessageHandlingFromFactory:
mock_a2a_task,
self.agent.name,
self.mock_context,
self.mock_a2a_part_converter,
)
# Check the parts are not updated as Thought
assert result.content.parts[0].thought is None
@@ -1251,6 +1266,7 @@ class TestRemoteA2aAgentMessageHandlingFromFactory:
mock_a2a_task,
self.agent.name,
self.mock_context,
self.agent._a2a_part_converter,
)
# Check the parts are updated as Thought
assert result.content.parts[0].thought is True
@@ -1296,6 +1312,7 @@ class TestRemoteA2aAgentMessageHandlingFromFactory:
mock_a2a_message,
self.agent.name,
self.mock_context,
self.agent._a2a_part_converter,
)
# Check that metadata was added
assert result.custom_metadata is not None
@@ -1341,6 +1358,7 @@ class TestRemoteA2aAgentMessageHandlingFromFactory:
mock_a2a_message,
self.agent.name,
self.mock_context,
self.agent._a2a_part_converter,
)
# Check that metadata was added
assert result.custom_metadata is not None
@@ -1396,7 +1414,10 @@ class TestRemoteA2aAgentMessageHandlingFromFactory:
assert result == mock_event
mock_convert.assert_called_once_with(
mock_a2a_task, self.agent.name, self.mock_context
mock_a2a_task,
self.agent.name,
self.mock_context,
self.agent._a2a_part_converter,
)
# Check that metadata was added
assert result.custom_metadata is not None