Formatting

Author: holtskinner

src/a2a/client/__init__.py

@@ -7,41 +7,42 @@
     CredentialService,
     InMemoryContextCredentialStore,
 )
+from a2a.client.client import (
+    A2ACardResolver,
+    Client,
+    ClientConfig,
+    ClientEvent,
+    Consumer,
+)
+from a2a.client.client_factory import (
+    ClientFactory,
+    ClientProducer,
+    minimal_agent_card,
+)
 from a2a.client.errors import (
     A2AClientError,
     A2AClientHTTPError,
     A2AClientJSONError,
     A2AClientTimeoutError,
 )
+from a2a.client.grpc_client import (
+    GrpcClient,
+    GrpcTransportClient,
+    NewGrpcClient,
+)
+from a2a.client.helpers import create_text_message_object
 from a2a.client.jsonrpc_client import (
     JsonRpcClient,
     JsonRpcTransportClient,
     NewJsonRpcClient,
 )
-from a2a.client.grpc_client import (
-    GrpcTransportClient,
-    GrpcClient,
-    NewGrpcClient,
-)
+from a2a.client.middleware import ClientCallContext, ClientCallInterceptor
 from a2a.client.rest_client import (
-    RestTransportClient,
-    RestClient,
     NewRestfulClient,
+    RestClient,
+    RestTransportClient,
 )
-from a2a.client.helpers import create_text_message_object
-from a2a.client.middleware import ClientCallContext, ClientCallInterceptor
-from a2a.client.client import (
-    A2ACardResolver,
-    Client,
-    ClientConfig,
-    Consumer,
-    ClientEvent,
-)
-from a2a.client.client_factory import (
-    ClientFactory,
-    ClientProducer,
-    minimal_agent_card
-)
+
 
 # For backward compatability define this alias. This will be deprecated in
 # a future release.
@@ -71,32 +72,32 @@ def __init__(self, *args, **kwargs):
 
 __all__ = [
     'A2ACardResolver',
+    'A2AClient',  # for backward compatability
     'A2AClientError',
     'A2AClientHTTPError',
     'A2AClientJSONError',
     'A2AClientTimeoutError',
+    'A2AGrpcClient',  # for backward compatability
     'AuthInterceptor',
+    'Client',
     'ClientCallContext',
     'ClientCallInterceptor',
-    'Consumer',
-    'CredentialService',
-    'InMemoryContextCredentialStore',
-    'create_text_message_object',
-    'A2AClient',  # for backward compatability
-    'A2AGrpcClient', # for backward compatability
-    'Client',
+    'ClientConfig',
     'ClientEvent',
     'ClientFactory',
-    'ClientConfig',
     'ClientProducer',
-    'GrpcTransportClient',
+    'Consumer',
+    'CredentialService',
     'GrpcClient',
-    'NewGrpcClient',
+    'GrpcTransportClient',
+    'InMemoryContextCredentialStore',
     'JsonRpcClient',
     'JsonRpcTransportClient',
+    'NewGrpcClient',
     'NewJsonRpcClient',
-    'minimal_agent_card',
-    'RestTransportClient',
-    'RestClient',
     'NewRestfulClient',
+    'RestClient',
+    'RestTransportClient',
+    'create_text_message_object',
+    'minimal_agent_card',
 ]

src/a2a/client/client.py

@@ -3,31 +3,30 @@
 import logging
 
 from abc import ABC, abstractmethod
-from collections.abc import AsyncIterator
-from typing import Any, Callable, Coroutine
-from uuid import uuid4
+from collections.abc import AsyncIterator, Callable, Coroutine
+from typing import Any
 
 import httpx
 
-from httpx_sse import SSEError, aconnect_sse
 from pydantic import ValidationError
 
+
 # Attempt to import the optional module
 try:
     from grpc.aio import Channel
 except ImportError:
     # If grpc.aio is not available, define a dummy type for type checking.
     # This dummy type will only be used by type checkers.
     if TYPE_CHECKING:
+
         class Channel:  # type: ignore[no-redef]
             pass
     else:
-        Channel = None # At runtime, pd will be None if the import failed.
+        Channel = None  # At runtime, pd will be None if the import failed.
 
 from a2a.client.errors import (
     A2AClientHTTPError,
     A2AClientJSONError,
-    A2AClientTimeoutError,
 )
 from a2a.client.middleware import ClientCallContext, ClientCallInterceptor
 from a2a.types import (
@@ -36,14 +35,13 @@ class Channel:  # type: ignore[no-redef]
     Message,
     PushNotificationConfig,
     Task,
+    TaskArtifactUpdateEvent,
     TaskIdParams,
-    TaskQueryParams,
     TaskPushNotificationConfig,
+    TaskQueryParams,
     TaskStatusUpdateEvent,
-    TaskArtifactUpdateEvent,
 )
 from a2a.utils.constants import AGENT_CARD_WELL_KNOWN_PATH
-from a2a.utils.telemetry import SpanKind, trace_class
 
 
 logger = logging.getLogger(__name__)
@@ -136,6 +134,7 @@ async def get_agent_card(
 
         return agent_card
 
+
 @dataclasses.dataclass
 class ClientConfig:
     """Configuration class for the A2A Client Factory"""
@@ -154,7 +153,7 @@ class ClientConfig:
     grpc_channel_factory: Callable[[str], Channel] | None = None
     """Generates a grpc connection channel for a given url."""
 
-    supported_transports: list[str] =  dataclasses.field(default_factory=list)
+    supported_transports: list[str] = dataclasses.field(default_factory=list)
     """Ordered list of transports for connecting to agent
        (in order of preference). Empty implies JSONRPC only.
 
@@ -166,12 +165,15 @@ class ClientConfig:
     """Whether to use client transport preferences over server preferences.
        Recommended to use server preferences in most situations."""
 
-    accepted_outputModes: list[str] =  dataclasses.field(default_factory=list)
+    accepted_outputModes: list[str] = dataclasses.field(default_factory=list)
     """The set of accepted output modes for the client."""
 
-    push_notification_configs: list[PushNotificationConfig] = dataclasses.field(default_factory=list)
+    push_notification_configs: list[PushNotificationConfig] = dataclasses.field(
+        default_factory=list
+    )
     """Push notification callbacks to use for every request."""
 
+
 UpdateEvent = TaskStatusUpdateEvent | TaskArtifactUpdateEvent | None
 # Alias for emitted events from client
 ClientEvent = tuple[Task, UpdateEvent]
@@ -183,7 +185,6 @@ class ClientConfig:
 
 
 class Client(ABC):
-
     def __init__(
         self,
         consumers: list[Consumer] = [],
@@ -252,14 +253,11 @@ async def resubscribe(
         *,
         context: ClientCallContext | None = None,
     ) -> AsyncIterator[Task | Message]:
-        pass
         yield
 
     @abstractmethod
     async def get_card(
-        self,
-        *,
-        context: ClientCallContext | None = None
+        self, *, context: ClientCallContext | None = None
     ) -> AgentCard:
         pass
 

src/a2a/client/client_factory.py

@@ -1,33 +1,20 @@
 from __future__ import annotations
-import json
-import logging
-
-from collections.abc import AsyncGenerator
-from typing import Any, TYPE_CHECKING, Callable
-
-import httpx
 
-from httpx_sse import SSEError, aconnect_sse
-from pydantic import ValidationError
+import logging
 
-from a2a.utils import Transports
+from collections.abc import Callable
 
 from a2a.client.client import Client, ClientConfig, Consumer
-from a2a.client.jsonrpc_client import NewJsonRpcClient
 from a2a.client.grpc_client import NewGrpcClient
+from a2a.client.jsonrpc_client import NewJsonRpcClient
+from a2a.client.middleware import ClientCallInterceptor
 from a2a.client.rest_client import NewRestfulClient
-from a2a.client.errors import A2AClientHTTPError, A2AClientJSONError
-from a2a.client.middleware import ClientCallContext, ClientCallInterceptor
 from a2a.types import (
     AgentCapabilities,
     AgentCard,
-    Message,
-    Task,
-    TaskIdParams,
-    TaskQueryParams,
-    GetTaskPushNotificationConfigParams,
-    TaskPushNotificationConfig,
 )
+from a2a.utils import Transports
+
 
 logger = logging.getLogger(__name__)
 
@@ -36,11 +23,12 @@
         AgentCard | str,
         ClientConfig,
         list[Consumer],
-        list[ClientCallInterceptor]
+        list[ClientCallInterceptor],
     ],
-    Client
+    Client,
 ]
 
+
 class ClientFactory:
     """ClientFactory is used to generate the appropriate client for the agent.
 
@@ -103,9 +91,7 @@ def create(
         # Determine preferential transport
         server_set = [card.preferred_transport or 'JSONRPC']
         if card.additional_interfaces:
-            server_set.extend(
-                [x.transport for x in card.additional_interfaces]
-            )
+            server_set.extend([x.transport for x in card.additional_interfaces])
         client_set = self._config.supported_transports or ['JSONRPC']
         transport = None
         # Two options, use the client ordering or the server ordering.
@@ -130,6 +116,7 @@ def create(
             card, self._config, all_consumers, interceptors
         )
 
+
 def minimal_agent_card(url: str, transports: list[str] = []) -> AgentCard:
     """Generates a minimal card to simplify bootstrapping client creation.
 

src/a2a/client/client_task_manager.py

@@ -1,8 +1,8 @@
 import logging
 
+from a2a.client.errors import A2AClientInvalidArgsError
 from a2a.server.events.event_queue import Event
 from a2a.types import (
-    InvalidParamsError,
     Message,
     Task,
     TaskArtifactUpdateEvent,
@@ -11,7 +11,6 @@
     TaskStatusUpdateEvent,
 )
 from a2a.utils import append_artifact_to_task
-from a2a.client.errors import A2AClientInvalidArgsError
 
 
 logger = logging.getLogger(__name__)
@@ -74,7 +73,7 @@ async def save_task_event(
         if isinstance(event, Task):
             if self._current_task:
                 raise A2AClientInvalidArgsError(
-                    "Task is already set, create new manager for new tasks."
+                    'Task is already set, create new manager for new tasks.'
                 )
             await self._save_task(event)
             return event
@@ -101,7 +100,9 @@ async def save_task_event(
             )
         if isinstance(event, TaskStatusUpdateEvent):
             logger.debug(
-                'Updating task %s status to: %s', event.taskId, event.status.state
+                'Updating task %s status to: %s',
+                event.taskId,
+                event.status.state,
             )
             if event.status.message:
                 if not task.history:

src/a2a/client/grpc_client.py

@@ -17,11 +17,11 @@
     Client,
     ClientCallContext,
     ClientConfig,
-    Consumer,
     ClientEvent,
+    Consumer,
 )
-from a2a.client.middleware import ClientCallInterceptor
 from a2a.client.client_task_manager import ClientTaskManager
+from a2a.client.middleware import ClientCallInterceptor
 from a2a.grpc import a2a_pb2, a2a_pb2_grpc
 from a2a.types import (
     AgentCard,
@@ -64,8 +64,7 @@ def __init__(
         # If they don't provide an agent card, but do have a stub, lookup the
         # card from the stub.
         self._needs_extended_card = (
-            agent_card.supportsAuthenticatedExtendedCard
-            if agent_card else True
+            agent_card.supportsAuthenticatedExtendedCard if agent_card else True
         )
 
     async def send_message(
@@ -251,7 +250,7 @@ async def get_card(
         card_pb = await self.stub.GetAgentCard(
             a2a_pb2.GetAgentCardRequest(),
         )
-        card =  proto_utils.FromProto.agent_card(card_pb)
+        card = proto_utils.FromProto.agent_card(card_pb)
         self.agent_card = card
         self._needs_extended_card = False
         return card
@@ -324,7 +323,7 @@ async def send_message(
             await tracker.process(event)
             result = (
                 tracker.get_task(),
-                None if isinstance(event, Task) else event
+                None if isinstance(event, Task) else event,
             )
             await self.consume(result, self._card)
             yield result
@@ -410,7 +409,7 @@ def NewGrpcClient(
     card: AgentCard,
     config: ClientConfig,
     consumers: list[Consumer],
-    middleware: list[ClientCallInterceptor]
+    middleware: list[ClientCallInterceptor],
 ) -> Client:
     """Generator for the `GrpcClient` implementation."""
     return GrpcClient(card, config, consumers, middleware)