Formatting

Author: holtskinner

src/a2a/server/apps/__init__.py

@@ -7,9 +7,10 @@
     JSONRPCApplication,
 )
 
+
 __all__ = [
     'A2AFastAPIApplication',
     'A2AStarletteApplication',
     'CallContextBuilder',
     'JSONRPCApplication',
-]
+]

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

@@ -1,7 +1,7 @@
 """A2A JSON-RPC Applications."""
 
-from .jsonrpc_app import CallContextBuilder, JSONRPCApplication
 from .fastapi_app import A2AFastAPIApplication
+from .jsonrpc_app import CallContextBuilder, JSONRPCApplication
 from .starlette_app import A2AStarletteApplication
 
 

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

@@ -4,10 +4,11 @@
 
 from fastapi import FastAPI, Request
 
-from .jsonrpc_app import CallContextBuilder, JSONRPCApplication
 from a2a.server.request_handlers.jsonrpc_handler import RequestHandler
 from a2a.types import AgentCard
 
+from .jsonrpc_app import CallContextBuilder, JSONRPCApplication
+
 
 logger = logging.getLogger(__name__)
 
@@ -73,10 +74,13 @@ async def handle_a2a_request(request: Request):
         @app.get(agent_card_url)
         async def get_agent_card(request: Request):
             return await self._handle_get_agent_card(request)
-        
+
         if self.agent_card.supportsAuthenticatedExtendedCard:
+
             @app.get(extended_agent_card_url)
             async def get_extended_agent_card(request: Request):
-                return await self._handle_get_authenticated_extended_agent_card(request)
+                return await self._handle_get_authenticated_extended_agent_card(
+                    request
+                )
 
         return app

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

@@ -5,7 +5,7 @@
 
 from abc import ABC, abstractmethod
 from collections.abc import AsyncGenerator
-from typing import Any, Union
+from typing import Any
 
 from fastapi import FastAPI
 from pydantic import ValidationError
@@ -397,7 +397,7 @@ def build(
         agent_card_url: str = '/.well-known/agent.json',
         rpc_url: str = '/',
         **kwargs: Any,
-    ) -> Union[FastAPI, Starlette]:
+    ) -> FastAPI | Starlette:
         """Builds and returns the JSONRPC application instance.
 
         Args:

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

@@ -5,10 +5,11 @@
 from starlette.applications import Starlette
 from starlette.routing import Route
 
-from .jsonrpc_app import CallContextBuilder, JSONRPCApplication
 from a2a.server.request_handlers.jsonrpc_handler import RequestHandler
 from a2a.types import AgentCard
 
+from .jsonrpc_app import CallContextBuilder, JSONRPCApplication
+
 
 logger = logging.getLogger(__name__)
 

tests/server/test_integration.py

@@ -19,8 +19,8 @@
 from starlette.testclient import TestClient
 
 from a2a.server.apps import (
-    A2AStarletteApplication,
     A2AFastAPIApplication,
+    A2AStarletteApplication,
 )
 from a2a.types import (
     AgentCapabilities,
@@ -135,6 +135,7 @@ def handler():
 def app(agent_card: AgentCard, handler: mock.AsyncMock):
     return A2AStarletteApplication(agent_card, handler)
 
+
 @pytest.fixture
 def client(app: A2AStarletteApplication, **kwargs):
     """Create a test client with the Starlette app."""
@@ -167,6 +168,7 @@ def test_authenticated_extended_agent_card_endpoint_not_supported(
     response = client.get('/agent/authenticatedExtendedCard')
     assert response.status_code == 404  # Starlette's default for no route
 
+
 def test_authenticated_extended_agent_card_endpoint_not_supported_fastapi(
     agent_card: AgentCard, handler: mock.AsyncMock
 ):
@@ -207,6 +209,7 @@ def test_authenticated_extended_agent_card_endpoint_supported_with_specific_exte
         'Extended skill not found in served card'
     )
 
+
 def test_authenticated_extended_agent_card_endpoint_supported_with_specific_extended_card_fastapi(
     agent_card: AgentCard,
     extended_agent_card_fixture: AgentCard,
@@ -232,6 +235,7 @@ def test_authenticated_extended_agent_card_endpoint_supported_with_specific_exte
         'Extended skill not found in served card'
     )
 
+
 def test_agent_card_custom_url(
     app: A2AStarletteApplication, agent_card: AgentCard
 ):
@@ -267,6 +271,7 @@ def test_starlette_rpc_endpoint_custom_url(
     data = response.json()
     assert data['result']['id'] == 'task1'
 
+
 def test_fastapi_rpc_endpoint_custom_url(
     app: A2AFastAPIApplication, handler: mock.AsyncMock
 ):
@@ -291,6 +296,7 @@ def test_fastapi_rpc_endpoint_custom_url(
     data = response.json()
     assert data['result']['id'] == 'task1'
 
+
 def test_starlette_build_with_extra_routes(
     app: A2AStarletteApplication, agent_card: AgentCard
 ):
@@ -314,6 +320,7 @@ def custom_handler(request):
     data = response.json()
     assert data['name'] == agent_card.name
 
+
 def test_fastapi_build_with_extra_routes(
     app: A2AFastAPIApplication, agent_card: AgentCard
 ):