chore: pin a2a to a minor version while it is still in beta (#586)

dbschmigelski · web-flow · commit b56a4ff32e93 · 2025-08-01T09:42:35.000-04:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -96,8 +96,8 @@ sagemaker = [
 ]
 
 a2a = [
-    "a2a-sdk>=0.2.16,<1.0.0",
-    "a2a-sdk[sql]>=0.2.11,<1.0.0",
+    "a2a-sdk>=0.3.0,<0.4.0",
+    "a2a-sdk[sql]>=0.3.0,<0.4.0",
     "uvicorn>=0.34.2,<1.0.0",
     "httpx>=0.28.1,<1.0.0",
     "fastapi>=0.115.12,<1.0.0",
@@ -143,7 +143,7 @@ all = [
     "opentelemetry-exporter-otlp-proto-http>=1.30.0,<2.0.0",
 
     # a2a
-    "a2a-sdk[sql]>=0.2.11,<1.0.0",
+    "a2a-sdk[sql]>=0.3.0,<0.4.0",
     "uvicorn>=0.34.2,<1.0.0",
     "httpx>=0.28.1,<1.0.0",
     "fastapi>=0.115.12,<1.0.0",
diff --git a/src/strands/multiagent/a2a/executor.py b/src/strands/multiagent/a2a/executor.py
@@ -61,7 +61,7 @@ async def execute(
             task = new_task(context.message)  # type: ignore
             await event_queue.enqueue_event(task)
 
-        updater = TaskUpdater(event_queue, task.id, task.contextId)
+        updater = TaskUpdater(event_queue, task.id, task.context_id)
 
         try:
             await self._execute_streaming(context, updater)
diff --git a/tests/strands/multiagent/a2a/test_executor.py b/tests/strands/multiagent/a2a/test_executor.py
@@ -36,7 +36,7 @@ async def mock_stream(user_input):
     # Mock the task creation
     mock_task = MagicMock()
     mock_task.id = "test-task-id"
-    mock_task.contextId = "test-context-id"
+    mock_task.context_id = "test-context-id"
     mock_request_context.current_task = mock_task
 
     await executor.execute(mock_request_context, mock_event_queue)
@@ -65,7 +65,7 @@ async def mock_stream(user_input):
     # Mock the task creation
     mock_task = MagicMock()
     mock_task.id = "test-task-id"
-    mock_task.contextId = "test-context-id"
+    mock_task.context_id = "test-context-id"
     mock_request_context.current_task = mock_task
 
     await executor.execute(mock_request_context, mock_event_queue)
@@ -95,7 +95,7 @@ async def mock_stream(user_input):
     # Mock the task creation
     mock_task = MagicMock()
     mock_task.id = "test-task-id"
-    mock_task.contextId = "test-context-id"
+    mock_task.context_id = "test-context-id"
     mock_request_context.current_task = mock_task
 
     await executor.execute(mock_request_context, mock_event_queue)
@@ -125,7 +125,7 @@ async def mock_stream(user_input):
     # Mock the task creation
     mock_task = MagicMock()
     mock_task.id = "test-task-id"
-    mock_task.contextId = "test-context-id"
+    mock_task.context_id = "test-context-id"
     mock_request_context.current_task = mock_task
 
     await executor.execute(mock_request_context, mock_event_queue)
@@ -156,7 +156,7 @@ async def mock_stream(user_input):
     mock_request_context.current_task = None
 
     with patch("strands.multiagent.a2a.executor.new_task") as mock_new_task:
-        mock_new_task.return_value = MagicMock(id="new-task-id", contextId="new-context-id")
+        mock_new_task.return_value = MagicMock(id="new-task-id", context_id="new-context-id")
 
         await executor.execute(mock_request_context, mock_event_queue)
 
@@ -180,7 +180,7 @@ async def test_execute_streaming_mode_handles_agent_exception(
     # Mock the task creation
     mock_task = MagicMock()
     mock_task.id = "test-task-id"
-    mock_task.contextId = "test-context-id"
+    mock_task.context_id = "test-context-id"
     mock_request_context.current_task = mock_task
 
     with pytest.raises(ServerError):
@@ -210,7 +210,7 @@ async def test_handle_agent_result_with_none_result(mock_strands_agent, mock_req
     # Mock the task creation
     mock_task = MagicMock()
     mock_task.id = "test-task-id"
-    mock_task.contextId = "test-context-id"
+    mock_task.context_id = "test-context-id"
     mock_request_context.current_task = mock_task
 
     # Mock TaskUpdater
@@ -235,7 +235,7 @@ async def test_handle_agent_result_with_result_but_no_message(
     # Mock the task creation
     mock_task = MagicMock()
     mock_task.id = "test-task-id"
-    mock_task.contextId = "test-context-id"
+    mock_task.context_id = "test-context-id"
     mock_request_context.current_task = mock_task
 
     # Mock TaskUpdater
diff --git a/tests/strands/multiagent/a2a/test_server.py b/tests/strands/multiagent/a2a/test_server.py
@@ -87,8 +87,8 @@ def test_public_agent_card(mock_strands_agent):
     assert card.description == "A test agent for unit testing"
     assert card.url == "http://0.0.0.0:9000/"
     assert card.version == "0.0.1"
-    assert card.defaultInputModes == ["text"]
-    assert card.defaultOutputModes == ["text"]
+    assert card.default_input_modes == ["text"]
+    assert card.default_output_modes == ["text"]
     assert card.skills == []
     assert card.capabilities == a2a_agent.capabilities
 
