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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.16.7
+Version: 0.16.8
 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

langgraph_api-0.16.8/langgraph_api/__init__.py

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

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

@@ -1,5 +1,6 @@
 import asyncio
 import concurrent.futures
+import sys
 from collections.abc import AsyncIterator, Coroutine
 from contextlib import AbstractAsyncContextManager, asynccontextmanager, suppress
 from functools import partial
@@ -7,6 +8,11 @@ from typing import Any, Generic, TypeVar
 
 import structlog
 
+if sys.platform == "win32":
+    _MAIN_LOOP_FACTORY = None
+else:
+    from uvloop import new_event_loop as _MAIN_LOOP_FACTORY
+
 T = TypeVar("T")
 
 logger = structlog.stdlib.get_logger(__name__)
@@ -44,6 +50,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)

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

@@ -19,6 +19,7 @@ from contextlib import suppress
 
 import structlog
 
+from langgraph_api.asyncio import 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
@@ -309,11 +310,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.8}/pyproject.toml

@@ -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/__init__.py

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