refactor

Author: guglielmo-san

src/a2a/client/base_client.py

@@ -456,9 +456,7 @@ async def _execute_stream_with_interceptors(
             await self._intercept_after(after_args, before_result['executed'])
 
             tracker = ClientTaskManager()
-            yield await self._format_stream_event(
-                after_args.result, tracker
-            )
+            yield await self._format_stream_event(after_args.result, tracker)
             return
 
         stream = transport_call(before_args.input, before_args.context)

src/a2a/client/client_factory.py

@@ -3,7 +3,7 @@
 import logging
 
 from collections.abc import Callable
-from typing import Any, cast
+from typing import TYPE_CHECKING, Any, cast
 
 import httpx
 
@@ -15,7 +15,6 @@
 from a2a.client.transports.base import ClientTransport
 from a2a.client.transports.jsonrpc import JsonRpcTransport
 from a2a.client.transports.rest import RestTransport
-from a2a.client.interceptors import ClientCallInterceptor
 from a2a.client.transports.tenant_decorator import TenantTransportDecorator
 from a2a.types.a2a_pb2 import (
     AgentCapabilities,
@@ -31,6 +30,10 @@
 )
 
 
+if TYPE_CHECKING:
+    from a2a.client.interceptors import ClientCallInterceptor
+
+
 try:
     from a2a.client.transports.grpc import GrpcTransport
 except ImportError:
@@ -197,9 +200,7 @@ def grpc_transport_producer(
                     ClientFactory._is_legacy_version(version)
                     and CompatGrpcTransport is not None
                 ):
-                    return CompatGrpcTransport.create(
-                        card, url, config
-                    )
+                    return CompatGrpcTransport.create(card, url, config)
 
                 if GrpcTransport is not None:
                     return GrpcTransport.create(card, url, config)

src/a2a/client/interceptors.py

@@ -12,6 +12,7 @@
     AgentCard,
 )
 
+
 @dataclass
 class BeforeArgs:
     """Arguments passed to the interceptor before a method call."""

src/a2a/client/transports/grpc.py

@@ -4,8 +4,8 @@
 from functools import wraps
 from typing import Any, NoReturn, cast
 
-from a2a.client.errors import A2AClientError, A2AClientTimeoutError
 from a2a.client.client import ClientCallContext
+from a2a.client.errors import A2AClientError, A2AClientTimeoutError
 
 
 try:

src/a2a/compat/v0_3/jsonrpc_transport.py

@@ -9,8 +9,8 @@
 
 from jsonrpc.jsonrpc2 import JSONRPC20Request, JSONRPC20Response
 
-from a2a.client.errors import A2AClientError
 from a2a.client.client import ClientCallContext
+from a2a.client.errors import A2AClientError
 from a2a.client.transports.base import ClientTransport
 from a2a.client.transports.http_helpers import (
     get_http_args,

src/a2a/compat/v0_3/rest_transport.py

@@ -8,9 +8,9 @@
 
 from google.protobuf.json_format import MessageToDict, Parse, ParseDict
 
+from a2a.client.client import ClientCallContext
 from a2a.client.errors import A2AClientError
 from a2a.client.transports.base import ClientTransport
-from a2a.client.client import ClientCallContext
 from a2a.client.transports.http_helpers import (
     get_http_args,
     send_http_request,

tests/client/test_auth_interceptor.py

@@ -149,9 +149,6 @@ async def test_in_memory_context_credential_store(
     assert await store.get_credentials(scheme_name, context) == new_credential
 
 
-
-
-
 def wrap_security_scheme(scheme: Any) -> SecurityScheme:
     """Wraps a security scheme in the correct SecurityScheme proto field."""
     if isinstance(scheme, APIKeySecurityScheme):