langgraph-api 0.16.8__tar.gz → 0.16.9__tar.gz

{langgraph_api-0.16.8 → langgraph_api-0.16.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.16.8
+Version: 0.16.9
 Author-email: Will Fu-Hinthorn <will@langchain.dev>, Josh Rogers <josh@langchain.dev>, Parker Rule <parker@langchain.dev>
 License: Elastic-2.0
 License-File: LICENSE
@@ -35,7 +35,7 @@ Requires-Dist: structlog<26,>=24.1.0
 Requires-Dist: tenacity>=8.0.0
 Requires-Dist: truststore>=0.1
 Requires-Dist: uuid-utils>=0.16.2
-Requires-Dist: uvicorn>=0.26.0
+Requires-Dist: uvicorn>=0.29.0
 Requires-Dist: uvloop>=0.18.0; platform_system != 'Windows'
 Requires-Dist: watchfiles>=0.13
 Requires-Dist: websockets>=13.0

{langgraph_api-0.16.8 → langgraph_api-0.16.9}/constraints.txt

@@ -9,7 +9,7 @@ langsmith>=0.7.31
 orjson>=3.9.7
 httpx>=0.25.0
 tenacity>=8.0.0
-uvicorn>=0.26.0
+uvicorn>=0.29.0
 sse-starlette>=2.1.3,<4.0.0
 uvloop>=0.18.0
 httptools>=0.5.0

langgraph_api-0.16.9/langgraph_api/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.16.9"

{langgraph_api-0.16.8 → langgraph_api-0.16.9}/langgraph_api/asyncio.py

@@ -1,12 +1,18 @@
 import asyncio
 import concurrent.futures
 import sys
-from collections.abc import AsyncIterator, Coroutine
-from contextlib import AbstractAsyncContextManager, asynccontextmanager, suppress
+from collections.abc import AsyncIterator, Coroutine, Generator
+from contextlib import (
+    AbstractAsyncContextManager,
+    asynccontextmanager,
+    contextmanager,
+    suppress,
+)
 from functools import partial
 from typing import Any, Generic, TypeVar
 
 import structlog
+import uvicorn
 
 if sys.platform == "win32":
     _MAIN_LOOP_FACTORY = None
@@ -354,3 +360,19 @@ class AsyncQueue(Generic[T], asyncio.Queue[T]):
                     # the call.  Wake up the next in line.
                     self._wakeup_next(self._getters)
                 raise
+
+
+class NoSignalServer(uvicorn.Server):
+    """Serve until cancelled without replacing the process signal handlers."""
+
+    @contextmanager
+    def capture_signals(self) -> Generator[None, None, None]:
+        yield
+
+    async def serve(self, sockets: list | None = None) -> None:
+        try:
+            await super().serve(sockets)
+        except asyncio.CancelledError:
+            if self.started:
+                await self.shutdown(sockets)
+            raise

{langgraph_api-0.16.8 → langgraph_api-0.16.9}/langgraph_api/graph.py

@@ -679,6 +679,9 @@ async def collect_graphs_from_env(
 
 
 def _handle_exception(task: asyncio.Task) -> None:
+    if task.cancelled():
+        # Intentional stop (stop_remote_graphs); let lifespan teardown finish.
+        return
     try:
         task.result()
     except (asyncio.CancelledError, SystemExit):

{langgraph_api-0.16.8 → langgraph_api-0.16.9}/langgraph_api/js/remote.py

@@ -43,6 +43,7 @@ from starlette.routing import Route
 
 from langgraph_api import config as lg_config
 from langgraph_api import store as api_store
+from langgraph_api.asyncio import NoSignalServer
 from langgraph_api.auth.custom import DotDict, ProxyUser, connection_method
 from langgraph_api.config import LANGGRAPH_AUTH, LANGGRAPH_AUTH_TYPE
 from langgraph_api.js.base import BaseRemotePregel, RemoteInterrupt
@@ -974,7 +975,7 @@ async def run_remote_checkpointer():
         ]
     )
 
-    server = uvicorn.Server(
+    server = NoSignalServer(
         uvicorn.Config(
             remote,
             port=REMOTE_PORT,

{langgraph_api-0.16.8 → langgraph_api-0.16.9}/langgraph_api/queue_entrypoint.py

@@ -19,7 +19,7 @@ from contextlib import suppress
 
 import structlog
 
-from langgraph_api.asyncio import run_main_loop
+from langgraph_api.asyncio import NoSignalServer, run_main_loop
 from langgraph_api.utils.errors import GraphLoadError, HealthServerStartupError
 from langgraph_api.utils.network import format_hostport, normalize_host
 from langgraph_runtime import lifespan
@@ -138,15 +138,13 @@ async def health_and_metrics_server():
         access_log=False,
     )
     # Server will run indefinitely until the process is terminated
-    server = uvicorn.Server(config)
+    server = NoSignalServer(config)
 
     logger.info(
         f"Health and metrics server started at http://{format_hostport(host, port)}"
     )
     try:
-        # Use the internal serve to skip capturing signals, otherwise there's a race condition
-        # where uvicorn captures the signal and will exit early before the queue can gracefully shutdown
-        await server._serve()
+        await server.serve()
     except SystemExit as exc:
         if exc.code == 0:
             return
@@ -172,9 +170,7 @@ async def health_and_metrics_server():
         )
         raise error from exc
     except asyncio.CancelledError:
-        # Close the health server cleanly once we've exited the lifespan to make sure we respect graceful shutdown
         logger.info("Shutting down health and metrics server")
-        await server.shutdown()
 
 
 async def entrypoint(

{langgraph_api-0.16.8 → langgraph_api-0.16.9}/pyproject.toml

@@ -24,7 +24,7 @@ dependencies = [
     "langgraph>=0.4.10,<2",
     "langgraph-checkpoint>=3.0.1,<5",
     "orjson>=3.9.7",
-    "uvicorn>=0.26.0",
+    "uvicorn>=0.29.0",
     # Required for the protocol v2 WebSocket transport
     # (``WebSocketRoute`` at ``/threads/{thread_id}/stream``). Without a
     # WebSocket library uvicorn falls back to ``Unsupported upgrade

{langgraph_api-0.16.8 → langgraph_api-0.16.9}/uv.lock

@@ -794,7 +794,7 @@ requires-dist = [
     { name = "tenacity", specifier = ">=8.0.0" },
     { name = "truststore", specifier = ">=0.1" },
     { name = "uuid-utils", specifier = ">=0.16.2" },
-    { name = "uvicorn", specifier = ">=0.26.0" },
+    { name = "uvicorn", specifier = ">=0.29.0" },
     { name = "uvloop", marker = "sys_platform != 'win32'", specifier = ">=0.18.0" },
     { name = "watchfiles", specifier = ">=0.13" },
     { name = "websockets", specifier = ">=13.0" },

langgraph_api-0.16.8/langgraph_api/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.16.8"