Updates

Author: ishymko

src/a2a/client/transports/http_helpers.py

@@ -118,9 +118,9 @@ def __init__(
         url: str,
         **kwargs: Any,
     ) -> None:
-        headers = kwargs.pop('headers', {})
-        headers['Accept'] = 'text/event-stream'
-        headers['Cache-Control'] = 'no-store'
+        headers = httpx.Headers(kwargs.pop('headers', None))
+        headers.setdefault('Accept', 'text/event-stream')
+        headers.setdefault('Cache-Control', 'no-store')
         self._request = client.build_request(
             method, url, headers=headers, **kwargs
         )

tests/client/transports/test_jsonrpc_client.py

@@ -433,7 +433,7 @@ async def test_close(self, transport, mock_httpx_client):
 
 class TestStreamingErrors:
     @pytest.mark.asyncio
-    @patch('a2a.client.transports.http_helpers.aconnect_sse')
+    @patch('a2a.client.transports.http_helpers._SSEEventSource')
     async def test_send_message_streaming_sse_error(
         self,
         mock_aconnect_sse: AsyncMock,
@@ -457,7 +457,7 @@ async def test_send_message_streaming_sse_error(
                 pass
 
     @pytest.mark.asyncio
-    @patch('a2a.client.transports.http_helpers.aconnect_sse')
+    @patch('a2a.client.transports.http_helpers._SSEEventSource')
     async def test_send_message_streaming_request_error(
         self,
         mock_aconnect_sse: AsyncMock,
@@ -483,7 +483,7 @@ async def test_send_message_streaming_request_error(
                 pass
 
     @pytest.mark.asyncio
-    @patch('a2a.client.transports.http_helpers.aconnect_sse')
+    @patch('a2a.client.transports.http_helpers._SSEEventSource')
     async def test_send_message_streaming_timeout(
         self,
         mock_aconnect_sse: AsyncMock,
@@ -560,7 +560,7 @@ async def test_extensions_added_to_request(
         )
 
     @pytest.mark.asyncio
-    @patch('a2a.client.transports.http_helpers.aconnect_sse')
+    @patch('a2a.client.transports.http_helpers._SSEEventSource')
     async def test_send_message_streaming_server_error_propagates(
         self,
         mock_aconnect_sse: AsyncMock,

tests/client/transports/test_rest_client.py

@@ -70,7 +70,7 @@ def _assert_extensions_header(mock_kwargs: dict, expected_extensions: set[str]):
 
 class TestRestTransport:
     @pytest.mark.asyncio
-    @patch('a2a.client.transports.http_helpers.aconnect_sse')
+    @patch('a2a.client.transports.http_helpers._SSEEventSource')
     async def test_send_message_streaming_timeout(
         self,
         mock_aconnect_sse: AsyncMock,
@@ -280,7 +280,7 @@ async def test_send_message_with_default_extensions(
         )
 
     @pytest.mark.asyncio
-    @patch('a2a.client.transports.http_helpers.aconnect_sse')
+    @patch('a2a.client.transports.http_helpers._SSEEventSource')
     async def test_send_message_streaming_with_new_extensions(
         self,
         mock_aconnect_sse: AsyncMock,
@@ -329,7 +329,7 @@ async def test_send_message_streaming_with_new_extensions(
         )
 
     @pytest.mark.asyncio
-    @patch('a2a.client.transports.http_helpers.aconnect_sse')
+    @patch('a2a.client.transports.http_helpers._SSEEventSource')
     async def test_send_message_streaming_server_error_propagates(
         self,
         mock_aconnect_sse: AsyncMock,
@@ -693,7 +693,7 @@ async def test_rest_get_task_prepend_empty_tenant(
         ],
     )
     @pytest.mark.asyncio
-    @patch('a2a.client.transports.http_helpers.aconnect_sse')
+    @patch('a2a.client.transports.http_helpers._SSEEventSource')
     async def test_rest_streaming_methods_prepend_tenant(  # noqa: PLR0913
         self,
         mock_aconnect_sse,