nlalpha-capability 0.2.0__tar.gz → 0.3.0__tar.gz

{nlalpha_capability-0.2.0 → nlalpha_capability-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: nlalpha-capability
-Version: 0.2.0
+Version: 0.3.0
 Summary: NLAlpha capability discovery MCP server.
 Project-URL: Homepage, https://github.com/ruimarj/nlalpha-capability
 Project-URL: Repository, https://github.com/ruimarj/nlalpha-capability

{nlalpha_capability-0.2.0 → nlalpha_capability-0.3.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "nlalpha-capability"
-version = "0.2.0"
+version = "0.3.0"
 description = "NLAlpha capability discovery MCP server."
 readme = "README.md"
 license = "AGPL-3.0-or-later"

{nlalpha_capability-0.2.0 → nlalpha_capability-0.3.0}/src/nlalpha_capability/server.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import asyncio
 from contextlib import asynccontextmanager
 from dataclasses import asdict
 from importlib.metadata import version
@@ -26,6 +27,28 @@ from nlalpha.core.telemetry import configure_telemetry, shutdown, get_meter, set
 
 logger = logging.getLogger("nlalpha-capability")
 
+
+@asynccontextmanager
+async def _session_manager_lifespan(mcp):
+    stop = asyncio.Event()
+    started = asyncio.Event()
+
+    async def run():
+        async with mcp.session_manager.run():
+            started.set()
+            await stop.wait()
+
+    task = asyncio.create_task(run())
+    task.add_done_callback(lambda _: started.set())
+    await started.wait()
+    if task.done():
+        task.result()
+    try:
+        yield
+    finally:
+        stop.set()
+        await task
+
 _meter = get_meter()
 _domains_counter = _meter.create_counter("nlalpha.capability.domains.listed")
 _helpers_histogram = _meter.create_histogram("nlalpha.capability.helpers.count")
@@ -189,17 +212,19 @@ def create_app(*, allowed_hosts: list[str] | tuple[str, ...] | None = None,
     for tool in (list_capability_domains, list_capabilities_by_domain, list_family_helpers,
                  list_capability_providers, diagnose_providers, about, confess):
         mcp.add_tool(tool)
+    http_app = mcp.streamable_http_app()
+
     @asynccontextmanager
     async def lifespan(app):
         """Invoke only the explicitly injected process cleanup on application shutdown."""
-        try:
-            yield
-        finally:
-            if on_shutdown is not None:
-                on_shutdown()
-
-    sse_app = mcp.sse_app()
-    app = Starlette(routes=[Route("/health", _health), *sse_app.routes], lifespan=lifespan)
+        async with _session_manager_lifespan(mcp):
+            try:
+                yield
+            finally:
+                if on_shutdown is not None:
+                    on_shutdown()
+
+    app = Starlette(routes=[Route("/health", _health), *http_app.routes], lifespan=lifespan)
     app.state.mcp = mcp
     return app