formatting

Author: vladkol

src/a2a/server/apps/jsonrpc/jsonrpc_app.py

@@ -2,6 +2,7 @@
 import json
 import logging
 import traceback
+
 from abc import ABC, abstractmethod
 from collections.abc import AsyncGenerator, Callable
 from typing import TYPE_CHECKING, Any
@@ -12,7 +13,7 @@
 from a2a.auth.user import User as A2AUser
 from a2a.extensions.common import (
     HTTP_EXTENSION_HEADER,
-    get_requested_extensions
+    get_requested_extensions,
 )
 from a2a.server.context import ServerCallContext
 from a2a.server.request_handlers.jsonrpc_handler import JSONRPCHandler
@@ -39,16 +40,17 @@
     SendStreamingMessageResponse,
     SetTaskPushNotificationConfigRequest,
     TaskResubscriptionRequest,
-    UnsupportedOperationError
+    UnsupportedOperationError,
 )
 from a2a.utils.constants import (
     AGENT_CARD_WELL_KNOWN_PATH,
     DEFAULT_RPC_URL,
     EXTENDED_AGENT_CARD_PATH,
-    PREV_AGENT_CARD_WELL_KNOWN_PATH
+    PREV_AGENT_CARD_WELL_KNOWN_PATH,
 )
 from a2a.utils.errors import MethodNotImplementedError
 
+
 logger = logging.getLogger(__name__)
 
 if TYPE_CHECKING:

src/a2a/server/apps/rest/rest_adapter.py

@@ -1,8 +1,10 @@
 import functools
 import logging
+
 from collections.abc import AsyncIterable, AsyncIterator, Awaitable, Callable
 from typing import TYPE_CHECKING, Any
 
+
 if TYPE_CHECKING:
     from sse_starlette.sse import EventSourceResponse
     from starlette.datastructures import URL
@@ -29,18 +31,19 @@
 
 from a2a.server.apps.jsonrpc import (
     CallContextBuilder,
-    DefaultCallContextBuilder
+    DefaultCallContextBuilder,
 )
 from a2a.server.context import ServerCallContext
 from a2a.server.request_handlers.request_handler import RequestHandler
 from a2a.server.request_handlers.rest_handler import RESTHandler
 from a2a.types import AgentCard, AuthenticatedExtendedCardNotConfiguredError
 from a2a.utils.error_handlers import (
     rest_error_handler,
-    rest_stream_error_handler
+    rest_stream_error_handler,
 )
 from a2a.utils.errors import ServerError
 
+
 logger = logging.getLogger(__name__)
 
 
@@ -148,7 +151,7 @@ async def handle_get_agent_card(
         if self.card_modifier:
             card_to_serve = self.card_modifier(card_to_serve)
         if rpc_url == card_to_serve.url:
-          self._modify_rpc_url(card_to_serve, request)
+            self._modify_rpc_url(card_to_serve, request)
 
         return card_to_serve.model_dump(mode='json', exclude_none=True)
 
@@ -185,7 +188,7 @@ async def handle_authenticated_agent_card(
             base_card = card_to_serve if card_to_serve else self.agent_card
             card_to_serve = self.extended_card_modifier(base_card, context)
         if rpc_url == card_to_serve.url:
-          self._modify_rpc_url(card_to_serve, request)
+            self._modify_rpc_url(card_to_serve, request)
 
         return card_to_serve.model_dump(mode='json', exclude_none=True)
 

tests/integration/test_client_server_integration.py

@@ -1,4 +1,5 @@
 import asyncio
+
 from collections.abc import AsyncGenerator
 from typing import NamedTuple
 from unittest.mock import ANY, AsyncMock
@@ -7,6 +8,7 @@
 import httpx
 import pytest
 import pytest_asyncio
+
 from grpc.aio import Channel
 
 from a2a.client.transports import JsonRpcTransport, RestTransport
@@ -36,6 +38,7 @@
     TransportProtocol,
 )
 
+
 # --- Test Constants ---
 
 TASK_FROM_STREAM = Task(