Fix formatting

Author: holtskinner

src/a2a/client/__init__.py

@@ -1,12 +1,12 @@
 """Client-side components for interacting with an A2A agent."""
 
 from a2a.client.client import A2ACardResolver, A2AClient
-from a2a.client.grpc_client import A2AGrpcClient
 from a2a.client.errors import (
     A2AClientError,
     A2AClientHTTPError,
     A2AClientJSONError,
 )
+from a2a.client.grpc_client import A2AGrpcClient
 from a2a.client.helpers import create_text_message_object
 
 

src/a2a/client/grpc_client.py

@@ -1,26 +1,24 @@
-import json
 import logging
+
 from collections.abc import AsyncGenerator
-from typing import Any
-from uuid import uuid4
+
 import grpc
 
-from a2a.client.errors import A2AClientHTTPError, A2AClientJSONError
+from a2a.grpc import a2a_pb2, a2a_pb2_grpc
 from a2a.types import (
     AgentCard,
+    Message,
     MessageSendParams,
     Task,
-    TaskStatusUpdateEvent,
     TaskArtifactUpdateEvent,
-    TaskPushNotificationConfig,
     TaskIdParams,
+    TaskPushNotificationConfig,
     TaskQueryParams,
-    Message,
+    TaskStatusUpdateEvent,
 )
-from a2a.utils.telemetry import SpanKind, trace_class
 from a2a.utils import proto_utils
-from a2a.grpc import a2a_pb2_grpc
-from a2a.grpc import a2a_pb2
+from a2a.utils.telemetry import SpanKind, trace_class
+
 
 logger = logging.getLogger(__name__)
 
@@ -48,7 +46,7 @@ def __init__(
     async def send_message(
         self,
         request: MessageSendParams,
-    ) -> Task | Message :
+    ) -> Task | Message:
         """Sends a non-streaming message request to the agent.
 
         Args:
@@ -174,7 +172,7 @@ async def set_task_callback(
 
     async def get_task_callback(
         self,
-        request: TaskIdParams, # TODO: Update to a push id params
+        request: TaskIdParams,  # TODO: Update to a push id params
     ) -> TaskPushNotificationConfig:
         """Retrieves the push notification configuration for a specific task.
 

src/a2a/server/request_handlers/__init__.py

@@ -3,8 +3,8 @@
 from a2a.server.request_handlers.default_request_handler import (
     DefaultRequestHandler,
 )
-from a2a.server.request_handlers.jsonrpc_handler import JSONRPCHandler
 from a2a.server.request_handlers.grpc_handler import GrpcHandler
+from a2a.server.request_handlers.jsonrpc_handler import JSONRPCHandler
 from a2a.server.request_handlers.request_handler import RequestHandler
 from a2a.server.request_handlers.response_helpers import (
     build_error_response,
@@ -14,8 +14,8 @@
 
 __all__ = [
     'DefaultRequestHandler',
-    'JSONRPCHandler',
     'GrpcHandler',
+    'JSONRPCHandler',
     'RequestHandler',
     'build_error_response',
     'prepare_response_object',

src/a2a/server/request_handlers/default_request_handler.py

@@ -1,9 +1,9 @@
 import asyncio
 import logging
+import uuid
 
 from collections.abc import AsyncGenerator
 from typing import cast
-import uuid
 
 from a2a.server.agent_execution import (
     AgentExecutor,

src/a2a/server/request_handlers/grpc_handler.py

@@ -1,37 +1,32 @@
-import logging
-import grpc
 import contextlib
+import logging
 
-from typing import AsyncIterable
 from abc import ABC, abstractmethod
+from collections.abc import AsyncIterable
+
+import grpc
+
+import a2a.grpc.a2a_pb2_grpc as a2a_grpc
 
+from a2a import types
+from a2a.auth.user import UnauthenticatedUser
+from a2a.grpc import a2a_pb2
 from a2a.server.context import ServerCallContext
 from a2a.server.request_handlers.request_handler import RequestHandler
 from a2a.types import (
     AgentCard,
-    InternalError,
-    Message,
-    Task,
-    TaskArtifactUpdateEvent,
     TaskNotFoundError,
-    TaskPushNotificationConfig,
-    TaskStatusUpdateEvent,
 )
-from a2a import types
-from a2a.auth.user import User as A2AUser
-from a2a.auth.user import UnauthenticatedUser
-from a2a.server.context import ServerCallContext
+from a2a.utils import proto_utils
 from a2a.utils.errors import ServerError
 from a2a.utils.helpers import validate, validate_async_generator
-from a2a.utils import proto_utils
-import a2a.grpc.a2a_pb2 as a2a_pb2
-import a2a.grpc.a2a_pb2_grpc as a2a_grpc
 
 
 logger = logging.getLogger(__name__)
 
 # For now we use a trivial wrapper on the grpc context object
 
+
 class CallContextBuilder(ABC):
     """A class for building ServerCallContexts using the Starlette Request."""
 
@@ -53,7 +48,8 @@ def build(self, context: grpc.ServicerContext) -> ServerCallContext:
 
 class GrpcHandler(a2a_grpc.A2AServiceServicer):
     """Maps incoming gRPC requests to the appropriate request handler method
- and formats responses."""
+    and formats responses.
+    """
 
     def __init__(
         self,
@@ -115,7 +111,7 @@ async def SendStreamingMessage(
         """Handles the 'StreamMessage' gRPC method.
 
         Yields response objects as they are produced by the underlying handler's
- stream.
+        stream.
 
         Args:
             request: The incoming `SendMessageRequest` object.
@@ -181,7 +177,7 @@ async def TaskSubscription(
         """Handles the 'TaskSubscription' gRPC method.
 
         Yields response objects as they are produced by the underlying handler's
- stream.
+        stream.
 
         Args:
             request: The incoming `TaskSubscriptionRequest` object.
@@ -193,7 +189,8 @@ async def TaskSubscription(
         try:
             server_context = self.context_builder.build(context)
             async for event in self.request_handler.on_resubscribe_to_task(
-                proto_utils.FromProto.task_id_params(request), server_context,
+                proto_utils.FromProto.task_id_params(request),
+                server_context,
             ):
                 yield proto_utils.ToProto.stream_response(event)
         except ServerError as e:

src/a2a/utils/helpers.py

@@ -1,4 +1,5 @@
 """General utility functions for the A2A Python SDK."""
+
 import functools
 import logging
 
@@ -147,6 +148,7 @@ def wrapper(self: Any, *args, **kwargs) -> Any:
 
     return decorator
 
+
 def validate_async_generator(
     expression: Callable[[Any], bool], error_message: str | None = None
 ):
@@ -179,6 +181,7 @@ async def wrapper(self, *args, **kwargs):
 
     return decorator
 
+
 def are_modalities_compatible(
     server_output_modes: list[str] | None, client_output_modes: list[str] | None
 ) -> bool: