a2a-handler 0.1.23__tar.gz → 0.1.24__tar.gz

{a2a_handler-0.1.23 → a2a_handler-0.1.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: a2a-handler
-Version: 0.1.23
+Version: 0.1.24
 Summary: An A2A Protocol client TUI and CLI.
 Keywords: a2a,a2a-protocol,agent,agent-to-agent,ai,cli,tui
 Author: Al Duncanson

{a2a_handler-0.1.23 → a2a_handler-0.1.24}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "a2a-handler"
-version = "0.1.23"
+version = "0.1.24"
 description = "An A2A Protocol client TUI and CLI."
 readme = "README.md"
 requires-python = ">=3.11"

{a2a_handler-0.1.23 → a2a_handler-0.1.24}/src/a2a_handler/server/app.py

@@ -1,7 +1,8 @@
 """A2A application setup and middleware."""
 
 import secrets
-from collections.abc import Awaitable, Callable
+from collections.abc import AsyncIterator, Awaitable, Callable
+from contextlib import asynccontextmanager
 
 import httpx
 from a2a.server.apps import A2AStarletteApplication
@@ -195,9 +196,7 @@ def create_a2a_application(
             Middleware(APIKeyAuthMiddleware, api_key=api_key)  # type: ignore[arg-type]
         )
 
-    application = Starlette(middleware=middleware)
-
-    async def setup_a2a_routes() -> None:
+    async def setup_a2a_routes(application: Starlette) -> None:
         a2a_starlette_app = A2AStarletteApplication(
             agent_card=agent_card,
             http_handler=request_handler,
@@ -209,7 +208,14 @@ def create_a2a_application(
         await http_client.aclose()
         logger.info("HTTP client closed")
 
-    application.add_event_handler("startup", setup_a2a_routes)
-    application.add_event_handler("shutdown", cleanup_http_client)
+    @asynccontextmanager
+    async def lifespan(application: Starlette) -> AsyncIterator[None]:
+        await setup_a2a_routes(application)
+        try:
+            yield
+        finally:
+            await cleanup_http_client()
+
+    application = Starlette(middleware=middleware, lifespan=lifespan)
 
     return application