formatting

Author: holtskinner

src/a2a/client/__init__.py

@@ -31,7 +31,7 @@
 except ImportError as e:
     _original_error = e
     logger.debug(
-        "A2AGrpcClient not loaded. This is expected if gRPC dependencies are not installed. Error: %s",
+        'A2AGrpcClient not loaded. This is expected if gRPC dependencies are not installed. Error: %s',
         _original_error,
     )
 
@@ -40,29 +40,29 @@ class A2AGrpcClient:  # type: ignore
 
         def __init__(self, *args, **kwargs):
             raise ImportError(
-                "To use A2AGrpcClient, its dependencies must be installed. "
+                'To use A2AGrpcClient, its dependencies must be installed. '
                 'You can install them with \'pip install "a2a-sdk[grpc]"\''
             ) from _original_error
 
 
 __all__ = [
-    "A2ACardResolver",
-    "A2AClient",
-    "A2AClientError",
-    "A2AClientHTTPError",
-    "A2AClientJSONError",
-    "A2AClientTimeoutError",
-    "A2AGrpcClient",
-    "AuthInterceptor",
-    "Client",
-    "ClientCallContext",
-    "ClientCallInterceptor",
-    "ClientConfig",
-    "ClientEvent",
-    "ClientFactory",
-    "Consumer",
-    "CredentialService",
-    "InMemoryContextCredentialStore",
-    "create_text_message_object",
-    "minimal_agent_card",
+    'A2ACardResolver',
+    'A2AClient',
+    'A2AClientError',
+    'A2AClientHTTPError',
+    'A2AClientJSONError',
+    'A2AClientTimeoutError',
+    'A2AGrpcClient',
+    'AuthInterceptor',
+    'Client',
+    'ClientCallContext',
+    'ClientCallInterceptor',
+    'ClientConfig',
+    'ClientEvent',
+    'ClientFactory',
+    'Consumer',
+    'CredentialService',
+    'InMemoryContextCredentialStore',
+    'create_text_message_object',
+    'minimal_agent_card',
 ]

src/a2a/client/legacy_grpc.py

@@ -8,19 +8,17 @@
 
 
 class A2AGrpcClient(GrpcTransport):
-    """
-    [DEPRECATED] Backwards compatibility wrapper for the gRPC client.
-    """
+    """[DEPRECATED] Backwards compatibility wrapper for the gRPC client."""
 
     def __init__(
         self,
-        grpc_stub: "a2a_pb2_grpc.A2AServiceStub",
+        grpc_stub: 'a2a_pb2_grpc.A2AServiceStub',
         agent_card: AgentCard,
     ):
         warnings.warn(
-            "A2AGrpcClient is deprecated and will be removed in a future version. "
-            "Use ClientFactory to create a client with a gRPC transport.",
+            'A2AGrpcClient is deprecated and will be removed in a future version. '
+            'Use ClientFactory to create a client with a gRPC transport.',
             DeprecationWarning,
             stacklevel=2,
         )
-        super().__init__(grpc_stub, agent_card)
+        super().__init__(grpc_stub, agent_card)

src/a2a/client/transports/__init__.py

@@ -12,8 +12,8 @@
 
 
 __all__ = [
-    "ClientTransport",
-    "GrpcTransport",
-    "JsonRpcTransport",
-    "RestTransport",
+    'ClientTransport',
+    'GrpcTransport',
+    'JsonRpcTransport',
+    'RestTransport',
 ]

src/a2a/utils/proto_utils.py

@@ -19,7 +19,9 @@
 
 # Regexp patterns for matching
 _TASK_NAME_MATCH = r'tasks/([\w-]+)'
-_TASK_PUSH_CONFIG_NAME_MATCH = r'tasks/([\w-]+)/pushNotificationConfigs/([\w-]+)'
+_TASK_PUSH_CONFIG_NAME_MATCH = (
+    r'tasks/([\w-]+)/pushNotificationConfigs/([\w-]+)'
+)
 
 
 class ToProto:

tests/client/test_auth_middleware.py

@@ -1,4 +1,5 @@
 import json
+
 from collections.abc import Callable
 from dataclasses import dataclass
 from typing import Any
@@ -17,9 +18,9 @@
     InMemoryContextCredentialStore,
 )
 from a2a.types import (
+    APIKeySecurityScheme,
     AgentCapabilities,
     AgentCard,
-    APIKeySecurityScheme,
     AuthorizationCodeOAuthFlow,
     HTTPAuthSecurityScheme,
     In,

tests/client/test_client_factory.py

@@ -6,8 +6,8 @@
 from a2a.client import ClientConfig, ClientFactory
 from a2a.client.transports import JsonRpcTransport, RestTransport
 from a2a.types import (
-    AgentCard,
     AgentCapabilities,
+    AgentCard,
     AgentInterface,
     TransportProtocol,
 )
@@ -17,10 +17,10 @@
 def base_agent_card() -> AgentCard:
     """Provides a base AgentCard for tests."""
     return AgentCard(
-        name="Test Agent",
-        description="An agent for testing.",
-        url="http://primary-url.com",
-        version="1.0.0",
+        name='Test Agent',
+        description='An agent for testing.',
+        url='http://primary-url.com',
+        version='1.0.0',
         capabilities=AgentCapabilities(),
         skills=[],
         default_input_modes=[],
@@ -33,53 +33,65 @@ def test_client_factory_selects_preferred_transport(base_agent_card: AgentCard):
     """Verify that the factory selects the preferred transport by default."""
     config = ClientConfig(
         httpx_client=httpx.AsyncClient(),
-        supported_transports=[TransportProtocol.jsonrpc, TransportProtocol.http_json],
+        supported_transports=[
+            TransportProtocol.jsonrpc,
+            TransportProtocol.http_json,
+        ],
     )
     factory = ClientFactory(config)
     client = factory.create(base_agent_card)
 
     assert isinstance(client._transport, JsonRpcTransport)
-    assert client._transport.url == "http://primary-url.com"
+    assert client._transport.url == 'http://primary-url.com'
 
 
-def test_client_factory_selects_secondary_transport_url(base_agent_card: AgentCard):
+def test_client_factory_selects_secondary_transport_url(
+    base_agent_card: AgentCard,
+):
     """Verify that the factory selects the correct URL for a secondary transport."""
     base_agent_card.additional_interfaces = [
         AgentInterface(
-            transport=TransportProtocol.http_json, url="http://secondary-url.com"
+            transport=TransportProtocol.http_json,
+            url='http://secondary-url.com',
         )
     ]
     # Client prefers REST, which is available as a secondary transport
     config = ClientConfig(
         httpx_client=httpx.AsyncClient(),
-        supported_transports=[TransportProtocol.http_json, TransportProtocol.jsonrpc],
+        supported_transports=[
+            TransportProtocol.http_json,
+            TransportProtocol.jsonrpc,
+        ],
         use_client_preference=True,
     )
     factory = ClientFactory(config)
     client = factory.create(base_agent_card)
 
     assert isinstance(client._transport, RestTransport)
-    assert client._transport.url == "http://secondary-url.com"
+    assert client._transport.url == 'http://secondary-url.com'
 
 
 def test_client_factory_server_preference(base_agent_card: AgentCard):
     """Verify that the factory respects server transport preference."""
     base_agent_card.preferred_transport = TransportProtocol.http_json
     base_agent_card.additional_interfaces = [
         AgentInterface(
-            transport=TransportProtocol.jsonrpc, url="http://secondary-url.com"
+            transport=TransportProtocol.jsonrpc, url='http://secondary-url.com'
         )
     ]
     # Client supports both, but server prefers REST
     config = ClientConfig(
         httpx_client=httpx.AsyncClient(),
-        supported_transports=[TransportProtocol.jsonrpc, TransportProtocol.http_json],
+        supported_transports=[
+            TransportProtocol.jsonrpc,
+            TransportProtocol.http_json,
+        ],
     )
     factory = ClientFactory(config)
     client = factory.create(base_agent_card)
 
     assert isinstance(client._transport, RestTransport)
-    assert client._transport.url == "http://primary-url.com"
+    assert client._transport.url == 'http://primary-url.com'
 
 
 def test_client_factory_no_compatible_transport(base_agent_card: AgentCard):
@@ -89,5 +101,5 @@ def test_client_factory_no_compatible_transport(base_agent_card: AgentCard):
         supported_transports=[TransportProtocol.grpc],
     )
     factory = ClientFactory(config)
-    with pytest.raises(ValueError, match="no compatible transports found"):
+    with pytest.raises(ValueError, match='no compatible transports found'):
         factory.create(base_agent_card)

tests/client/test_jsonrpc_client.py

@@ -1,11 +1,13 @@
 import json
+
 from collections.abc import AsyncGenerator
 from typing import Any
-from unittest.mock import ANY, AsyncMock, MagicMock, patch
+from unittest.mock import AsyncMock, MagicMock, patch
 
 import httpx
 import pytest
-from httpx_sse import EventSource, ServerSentEvent, SSEError
+
+from httpx_sse import EventSource, SSEError, ServerSentEvent
 
 from a2a.client import (
     A2ACardResolver,
@@ -27,12 +29,12 @@
     SendMessageSuccessResponse,
     Task,
     TaskIdParams,
-    TaskNotCancelableError,
     TaskPushNotificationConfig,
     TaskQueryParams,
 )
 from a2a.utils import AGENT_CARD_WELL_KNOWN_PATH
 
+
 AGENT_CARD = AgentCard(
     name='Hello World Agent',
     description='Just a hello world agent',
@@ -550,9 +552,10 @@ async def test_get_task_success(
             response = await client.get_task(request=params)
 
         assert isinstance(response, Task)
-        assert response.model_dump() == Task.model_validate(
-            MINIMAL_TASK
-        ).model_dump()
+        assert (
+            response.model_dump()
+            == Task.model_validate(MINIMAL_TASK).model_dump()
+        )
         mock_send_request.assert_called_once()
         sent_payload = mock_send_request.call_args.args[0]
         assert sent_payload['method'] == 'tasks/get'
@@ -577,9 +580,10 @@ async def test_cancel_task_success(
             response = await client.cancel_task(request=params)
 
         assert isinstance(response, Task)
-        assert response.model_dump() == Task.model_validate(
-            MINIMAL_CANCELLED_TASK
-        ).model_dump()
+        assert (
+            response.model_dump()
+            == Task.model_validate(MINIMAL_CANCELLED_TASK).model_dump()
+        )
         mock_send_request.assert_called_once()
         sent_payload = mock_send_request.call_args.args[0]
         assert sent_payload['method'] == 'tasks/cancel'
@@ -731,7 +735,9 @@ async def test_send_message_streaming_request_error(
             ]
 
     @pytest.mark.asyncio
-    async def test_get_card_no_card_provided(self, mock_httpx_client: AsyncMock):
+    async def test_get_card_no_card_provided(
+        self, mock_httpx_client: AsyncMock
+    ):
         client = JsonRpcTransport(
             httpx_client=mock_httpx_client, url=self.AGENT_URL
         )

tests/client/test_legacy_client.py

@@ -9,7 +9,6 @@
 from a2a.types import (
     AgentCapabilities,
     AgentCard,
-    GetTaskRequest,
     Message,
     MessageSendParams,
     Part,
@@ -72,26 +71,12 @@ async def test_a2a_client_send_message(
     """Tests for the legacy client compatibility layer."""
 
 
-from unittest.mock import AsyncMock, MagicMock
+from unittest.mock import AsyncMock
 
-import httpx
 import pytest
 
-from a2a.client import A2AClient, A2AGrpcClient
 from a2a.types import (
-    AgentCapabilities,
     AgentCard,
-    GetTaskRequest,
-    Message,
-    MessageSendParams,
-    Part,
-    Role,
-    SendMessageRequest,
-    Task,
-    TaskQueryParams,
-    TaskState,
-    TaskStatus,
-    TextPart,
 )