add-fastapi-application

Author: martimfasantos

src/a2a/server/apps/__init__.py

@@ -1,12 +1,12 @@
-from a2a.server.apps.http_app import HttpApp
 from a2a.server.apps.default_app import DefaultA2AApplication
-from a2a.server.apps.starlette_app import A2AStarletteApplication
 from a2a.server.apps.fastapi_app import A2AFastAPIApplication
+from a2a.server.apps.http_app import HttpApp
+from a2a.server.apps.starlette_app import A2AStarletteApplication
 
 
 __all__ = [
+    'A2AFastAPIApplication',
+    'A2AStarletteApplication',
     'DefaultA2AApplication',
-    'A2AStarletteApplication', 
-    'A2AFastAPIApplication', 
-    'HttpApp'
+    'HttpApp',
 ]

src/a2a/server/apps/default_app.py

@@ -1,15 +1,15 @@
 import json
 import logging
 import traceback
-from abc import ABC, abstractmethod
 
+from abc import ABC, abstractmethod
 from collections.abc import AsyncGenerator
-from typing import Any, Optional, Union
+from typing import Any
 
+from fastapi import FastAPI
 from pydantic import ValidationError
 from sse_starlette.sse import EventSourceResponse
 from starlette.applications import Starlette
-from fastapi import FastAPI
 from starlette.requests import Request
 from starlette.responses import JSONResponse, Response
 
@@ -39,6 +39,7 @@
 )
 from a2a.utils.errors import MethodNotImplementedError
 
+
 logger = logging.getLogger(__name__)
 
 
@@ -258,7 +259,7 @@ def build(
         agent_card_url: str = '/.well-known/agent.json',
         rpc_url: str = '/',
         **kwargs: Any,
-    ) -> Union[Starlette, FastAPI]:
+    ) -> Starlette | FastAPI:
         """Builds and returns the FastAPI application instance.
 
         Args:
@@ -269,4 +270,3 @@ def build(
         Returns:
             A configured FastAPI application instance.
         """
-        pass

src/a2a/server/apps/fastapi_app.py

@@ -1,4 +1,5 @@
 import logging
+
 from typing import Any
 
 from fastapi import FastAPI, Request
@@ -7,6 +8,7 @@
 from a2a.server.request_handlers.jsonrpc_handler import RequestHandler
 from a2a.types import AgentCard
 
+
 logger = logging.getLogger(__name__)
 
 
@@ -15,7 +17,8 @@ class A2AFastAPIApplication(DefaultA2AApplication):
 
     Handles incoming JSON-RPC requests, routes them to the appropriate
     handler methods, and manages response generation including Server-Sent Events
-    (SSE)."""
+    (SSE).
+    """
 
     def __init__(self, agent_card: AgentCard, http_handler: RequestHandler):
         """Initializes the A2A FastAPI application.

src/a2a/server/apps/starlette_app.py

@@ -59,7 +59,7 @@ def routes(
                 name='agent_card',
             ),
         ]
-    
+
     def build(
         self,
         agent_card_url: str = '/.well-known/agent.json',