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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.16.7
+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.7 → 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.7 → langgraph_api-0.16.9}/langgraph_api/asyncio.py

@@ -1,11 +1,23 @@
 import asyncio
 import concurrent.futures
-from collections.abc import AsyncIterator, Coroutine
-from contextlib import AbstractAsyncContextManager, asynccontextmanager, suppress
+import sys
+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
+else:
+    from uvloop import new_event_loop as _MAIN_LOOP_FACTORY
 
 T = TypeVar("T")
 
@@ -44,6 +56,11 @@ def get_event_loop() -> asyncio.AbstractEventLoop:
     return _MAIN_LOOP
 
 
+def run_main_loop(main: Coroutine[object, object, T]) -> T:
+    with asyncio.Runner(loop_factory=_MAIN_LOOP_FACTORY) as runner:
+        return runner.run(main)
+
+
 async def sleep_if_not_done(delay: float, done: asyncio.Event) -> None:
     with suppress(TimeoutError):
         await asyncio.wait_for(done.wait(), delay)
@@ -343,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.7 → 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.7 → 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.7 → langgraph_api-0.16.9}/langgraph_api/queue_entrypoint.py

@@ -19,6 +19,7 @@ from contextlib import suppress
 
 import structlog
 
+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
@@ -137,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
@@ -171,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(
@@ -309,11 +306,4 @@ if __name__ == "__main__":
     with open(pathlib.Path(__file__).parent.parent / "logging.json") as file:
         loaded_config = json.load(file)
         logging.config.dictConfig(loaded_config)
-    try:
-        import uvloop
-
-        uvloop.install()
-    except ImportError:
-        pass
-    # run the entrypoint
-    asyncio.run(main())
+    run_main_loop(main())

{langgraph_api-0.16.7 → 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
@@ -166,6 +166,10 @@ lint.per-file-ignores = {"tests/**" = ["S101", "B017"]}
 extend-exclude = [".venv", "dist", "build", "__pycache__", "venv", "langgraph_api/grpc/generated/**"]
 target-version = "py311"
 
+[tool.ruff.lint.flake8-tidy-imports.banned-api]
+"asyncio.set_event_loop_policy".msg = "Pass a loop_factory to asyncio.Runner instead."
+"uvloop.install".msg = "Use langgraph_api.asyncio.run_main_loop instead."
+
 [tool.ty.rules]
 possibly-missing-attribute = "error"
 possibly-missing-implicit-call = "error"

{langgraph_api-0.16.7 → 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.7/langgraph_api/__init__.py

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