feat: Enable REST transport in SUT agent

Author: Łukasz Bobiński

tck/sut_agent.py

@@ -9,7 +9,7 @@
 
 from a2a.server.agent_execution.agent_executor import AgentExecutor
 from a2a.server.agent_execution.context import RequestContext
-from a2a.server.apps import A2AStarletteApplication
+from a2a.server.apps import A2AStarletteApplication, A2ARESTFastAPIApplication
 from a2a.server.events.event_queue import EventQueue
 from a2a.server.request_handlers.default_request_handler import (
     DefaultRequestHandler,
@@ -202,6 +202,10 @@ def main() -> None:
                 'url': f'http://localhost:{http_port}{JSONRPC_URL}',
                 'transport': 'JSONRPC',
             },
+            {
+                'url': f'http://localhost:{http_port}/a2a/rest',
+                'transport': 'REST',
+            },
         ],
     )
 
@@ -210,12 +214,18 @@ def main() -> None:
         task_store=InMemoryTaskStore(),
     )
 
-    server = A2AStarletteApplication(
+    jsonrpc_server = A2AStarletteApplication(
+        agent_card=agent_card,
+        http_handler=request_handler,
+    )
+
+    rest_server = A2ARESTFastAPIApplication(
         agent_card=agent_card,
         http_handler=request_handler,
     )
 
-    app = server.build(rpc_url=JSONRPC_URL)
+    app = jsonrpc_server.build(rpc_url=JSONRPC_URL)
+    app.mount('/a2a/rest', rest_server.build())
 
     logger.info('Starting HTTP server on port %s...', http_port)
     uvicorn.run(app, host='127.0.0.1', port=http_port, log_level='info')