feat: add a2a routes to existing app

Author: MadaraUchiha-314

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

@@ -48,27 +48,22 @@ def __init__(
             extended_agent_card=extended_agent_card,
             context_builder=context_builder,
         )
-
-    def build(
-        self,
+    
+    def add_routes_to_app(
+        self, 
+        app: FastAPI, 
         agent_card_url: str = '/.well-known/agent.json',
         rpc_url: str = '/',
-        extended_agent_card_url: str = '/agent/authenticatedExtendedCard',
-        **kwargs: Any,
-    ) -> FastAPI:
-        """Builds and returns the FastAPI application instance.
+        extended_agent_card_url: str = '/agent/authenticatedExtendedCard'
+    ):
+        """Adds the routes to the FastAPI application.
 
         Args:
+            app: The FastAPI application to add the routes to.
             agent_card_url: The URL for the agent card endpoint.
-            rpc_url: The URL for the A2A JSON-RPC endpoint.
+            rpc_url: The URL for the A2A JSON-RPC endpoint. 
             extended_agent_card_url: The URL for the authenticated extended agent card endpoint.
-            **kwargs: Additional keyword arguments to pass to the FastAPI constructor.
-
-        Returns:
-            A configured FastAPI application instance.
         """
-        app = FastAPI(**kwargs)
-
         @app.post(rpc_url)
         async def handle_a2a_request(request: Request) -> Response:
             return await self._handle_requests(request)
@@ -84,5 +79,26 @@ async def get_extended_agent_card(request: Request) -> Response:
                 return await self._handle_get_authenticated_extended_agent_card(
                     request
                 )
+    def build(
+        self,
+        agent_card_url: str = '/.well-known/agent.json',
+        rpc_url: str = '/',
+        extended_agent_card_url: str = '/agent/authenticatedExtendedCard',
+        **kwargs: Any,
+    ) -> FastAPI:
+        """Builds and returns the FastAPI application instance.
+
+        Args:
+            agent_card_url: The URL for the agent card endpoint.
+            rpc_url: The URL for the A2A JSON-RPC endpoint.
+            extended_agent_card_url: The URL for the authenticated extended agent card endpoint.
+            **kwargs: Additional keyword arguments to pass to the FastAPI constructor.
+
+        Returns:
+            A configured FastAPI application instance.
+        """
+        app = FastAPI(**kwargs)
+
+        self.add_routes_to_app(app, agent_card_url, rpc_url, extended_agent_card_url)
 
         return app

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

@@ -91,6 +91,28 @@ def routes(
                 )
             )
         return app_routes
+    
+    def add_routes_to_app(
+        self,
+        app: Starlette,
+        agent_card_url: str = '/.well-known/agent.json',
+        rpc_url: str = '/',
+        extended_agent_card_url: str = '/agent/authenticatedExtendedCard'
+    ):
+        """Adds the routes to the Starlette application.
+
+        Args:
+            app: The Starlette application to add the routes to.
+            agent_card_url: The URL path for the agent card endpoint.
+            rpc_url: The URL path for the A2A JSON-RPC endpoint (POST requests).
+            extended_agent_card_url: The URL for the authenticated extended agent card endpoint.
+        """
+        routes = self.routes(
+            agent_card_url=agent_card_url,
+            rpc_url=rpc_url,
+            extended_agent_card_url=extended_agent_card_url,
+        )
+        app.routes.extend(routes)
 
     def build(
         self,
@@ -110,14 +132,8 @@ def build(
         Returns:
             A configured Starlette application instance.
         """
-        app_routes = self.routes(
-            agent_card_url=agent_card_url,
-            rpc_url=rpc_url,
-            extended_agent_card_url=extended_agent_card_url,
-        )
-        if 'routes' in kwargs:
-            kwargs['routes'].extend(app_routes)
-        else:
-            kwargs['routes'] = app_routes
+        app = Starlette(**kwargs)
+
+        self.add_routes_to_app(app, agent_card_url, rpc_url, extended_agent_card_url)
 
-        return Starlette(**kwargs)
+        return app