linter

Author: guglielmo-san

src/a2a/server/routes/__init__.py

@@ -9,8 +9,8 @@
 
 
 __all__ = [
-    'create_agent_card_routes',
     'CallContextBuilder',
     'DefaultCallContextBuilder',
+    'create_agent_card_routes',
     'create_jsonrpc_routes',
 ]

src/a2a/server/routes/agent_card_routes.py

@@ -1,6 +1,6 @@
 import logging
 
-from collections.abc import Awaitable, Callable, Sequence
+from collections.abc import Awaitable, Callable
 from typing import TYPE_CHECKING, Any
 
 

src/a2a/server/routes/jsonrpc_routes.py

@@ -1,6 +1,6 @@
 import logging
 
-from collections.abc import Awaitable, Callable, Sequence
+from collections.abc import Awaitable, Callable
 from typing import TYPE_CHECKING, Any
 
 

tests/integration/cross_version/client_server/server_1_0.py

@@ -5,7 +5,7 @@
 import grpc
 
 from a2a.server.agent_execution import AgentExecutor, RequestContext
-from a2a.server.routes import AgentCardRoutes, create_jsonrpc_routes
+from a2a.server.routes import create_agent_card_routes, create_jsonrpc_routes
 from a2a.server.apps import A2ARESTFastAPIApplication
 from a2a.server.events import EventQueue
 from a2a.server.events.in_memory_queue_manager import InMemoryQueueManager
@@ -167,7 +167,7 @@ async def main_async(http_port: int, grpc_port: int):
     app = FastAPI()
     app.add_middleware(CustomLoggingMiddleware)
 
-    agent_card_routes = AgentCardRoutes(
+    agent_card_routes = create_agent_card_routes(
         agent_card=agent_card, card_url='/.well-known/agent-card.json'
     )
     jsonrpc_routes = create_jsonrpc_routes(
@@ -179,7 +179,7 @@ async def main_async(http_port: int, grpc_port: int):
     )
     app.mount(
         '/jsonrpc',
-        FastAPI(routes=jsonrpc_routes + agent_card_routes.routes),
+        FastAPI(routes=jsonrpc_routes + agent_card_routes),
     )
 
     app.mount(

tests/integration/test_client_server_integration.py

@@ -222,7 +222,9 @@ def http_base_setup(mock_request_handler: AsyncMock, agent_card: AgentCard):
 def jsonrpc_setup(http_base_setup) -> TransportSetup:
     """Sets up the JsonRpcTransport and in-memory server."""
     mock_request_handler, agent_card = http_base_setup
-    agent_card_routes = create_agent_card_routes(agent_card=agent_card, card_url='/')
+    agent_card_routes = create_agent_card_routes(
+        agent_card=agent_card, card_url='/'
+    )
     jsonrpc_routes = create_jsonrpc_routes(
         agent_card=agent_card,
         request_handler=mock_request_handler,
@@ -695,7 +697,9 @@ async def test_client_get_signed_extended_card(
         },
     )
 
-    agent_card_routes = create_agent_card_routes(agent_card=agent_card, card_url='/')
+    agent_card_routes = create_agent_card_routes(
+        agent_card=agent_card, card_url='/'
+    )
     jsonrpc_routes = create_jsonrpc_routes(
         agent_card=agent_card,
         request_handler=mock_request_handler,

tests/integration/test_end_to_end.py

@@ -194,7 +194,9 @@ def rest_setup(agent_card, base_e2e_setup) -> ClientSetup:
 @pytest.fixture
 def jsonrpc_setup(agent_card, base_e2e_setup) -> ClientSetup:
     task_store, handler = base_e2e_setup
-    agent_card_routes = create_agent_card_routes(agent_card=agent_card, card_url='/')
+    agent_card_routes = create_agent_card_routes(
+        agent_card=agent_card, card_url='/'
+    )
     jsonrpc_routes = create_jsonrpc_routes(
         agent_card=agent_card,
         request_handler=handler,

tests/integration/test_tenant.py

@@ -207,9 +207,7 @@ def server_app(self, jsonrpc_agent_card, mock_handler):
             extended_agent_card=jsonrpc_agent_card,
             rpc_url='/jsonrpc',
         )
-        app = Starlette(
-            routes=[*agent_card_routes, *jsonrpc_routes]
-        )
+        app = Starlette(routes=[*agent_card_routes, *jsonrpc_routes])
         return app
 
     @pytest.mark.asyncio

tests/integration/test_version_header.py

@@ -57,7 +57,9 @@ async def mock_on_message_send_stream(*args, **kwargs):
     handler.on_message_send_stream = mock_on_message_send_stream
 
     app = FastAPI()
-    agent_card_routes = create_agent_card_routes(agent_card=agent_card, card_url='/')
+    agent_card_routes = create_agent_card_routes(
+        agent_card=agent_card, card_url='/'
+    )
     jsonrpc_routes = create_jsonrpc_routes(
         agent_card=agent_card,
         request_handler=handler,

tests/server/routes/test_agent_card_routes.py

@@ -62,7 +62,9 @@ async def modifier(card: AgentCard) -> AgentCard:
 def test_agent_card_custom_url(agent_card):
     """Tests that custom card_url is respected."""
     custom_url = '/custom/path/agent.json'
-    routes = create_agent_card_routes(agent_card=agent_card, card_url=custom_url)
+    routes = create_agent_card_routes(
+        agent_card=agent_card, card_url=custom_url
+    )
 
     app = Starlette(routes=routes)
     client = TestClient(app)
@@ -71,5 +73,3 @@ def test_agent_card_custom_url(agent_card):
     assert client.get('/.well-known/agent-card.json').status_code == 404
     # Check that custom returns 200
     assert client.get(custom_url).status_code == 200
-
-

tests/server/routes/test_jsonrpc_routes.py

@@ -59,4 +59,3 @@ def test_jsonrpc_custom_url(agent_card, mock_handler):
     assert 'error' in resp_json
     # Method not found error from dispatcher
     assert resp_json['error']['code'] == -32601
-