langgraph-api 0.2.54__py3-none-any.whl → 0.2.56__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.2.54"
+__version__ = "0.2.56"

langgraph_api/queue_entrypoint.py

@@ -10,11 +10,13 @@ if not (
     truststore.inject_into_ssl()  # noqa: F401
 
 import asyncio
+import contextlib
 import http.server
 import json
 import logging.config
 import os
 import pathlib
+import signal
 
 import structlog
 import uvloop
@@ -60,25 +62,39 @@ async def entrypoint():
 
     lg_logging.set_logging_context({"entrypoint": "python-queue"})
     tasks: set[asyncio.Task] = set()
-    # start simple http server for health checks
     tasks.add(asyncio.create_task(healthcheck_server()))
-    # start queue and associated tasks
     async with lifespan(None, with_cron_scheduler=False, taskset=tasks):
-        # run forever, error if any tasks fail
-        try:
-            await asyncio.gather(*tasks)
-        except asyncio.CancelledError:
-            await logger.awarning("Queue entrypoint cancelled", exc_info=True)
+        await asyncio.gather(*tasks)
+
+
+async def main():
+    """Run the queue entrypoint and shut down gracefully on SIGTERM/SIGINT."""
+    loop = asyncio.get_running_loop()
+    stop_event = asyncio.Event()
+
+    def _handle_signal() -> None:
+        logger.warning("Received termination signal, initiating graceful shutdown")
+        stop_event.set()
+
+    try:
+        loop.add_signal_handler(signal.SIGTERM, _handle_signal)
+    except (NotImplementedError, RuntimeError):
+        signal.signal(signal.SIGTERM, lambda *_: _handle_signal())
+
+    entry_task = asyncio.create_task(entrypoint())
+    await stop_event.wait()
+
+    logger.warning("Cancelling queue entrypoint task")
+    entry_task.cancel()
+    with contextlib.suppress(asyncio.CancelledError):
+        await entry_task
 
 
 if __name__ == "__main__":
-    # set up logging
     with open(pathlib.Path(__file__).parent.parent / "logging.json") as file:
         loaded_config = json.load(file)
         logging.config.dictConfig(loaded_config)
 
-    # set up uvloop
     uvloop.install()
 
-    # run the entrypoint
-    asyncio.run(entrypoint())
+    asyncio.run(main())

{langgraph_api-0.2.54.dist-info → langgraph_api-0.2.56.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.2.54
+Version: 0.2.56
 Author-email: Nuno Campos <nuno@langchain.dev>, Will Fu-Hinthorn <will@langchain.dev>
 License: Elastic-2.0
 License-File: LICENSE

{langgraph_api-0.2.54.dist-info → langgraph_api-0.2.56.dist-info}/RECORD

@@ -1,4 +1,4 @@
-langgraph_api/__init__.py,sha256=oNewG1bC0xUC2UKGSu-zMR1_CZwFYXnhulfuiJ3G4hw,23
+langgraph_api/__init__.py,sha256=yuhEwVPZG_uwEXVduMxOuyth-d2aFtmKNM7h6XPNHtM,23
 langgraph_api/asgi_transport.py,sha256=eqifhHxNnxvI7jJqrY1_8RjL4Fp9NdN4prEub2FWBt8,5091
 langgraph_api/asyncio.py,sha256=Odnc6mAJIGF3eFWT8Xcrg2Zam7FwzXkfCWEHaXfrzQQ,9371
 langgraph_api/cli.py,sha256=9Ou3tGDDY_VVLt5DFle8UviJdpI4ZigC5hElYvq2-To,14519
@@ -11,7 +11,7 @@ langgraph_api/http.py,sha256=gYbxxjY8aLnsXeJymcJ7G7Nj_yToOGpPYQqmZ1_ggfA,5240
 langgraph_api/logging.py,sha256=1BXELwUBY8gZeOWCYElbBu_GyHLM2jjlDVJznlekqvQ,4268
 langgraph_api/metadata.py,sha256=Gx0b6YszLRjdWLDVN8OcVgC_YYQG_nQitPfUfgQx1w8,4648
 langgraph_api/patch.py,sha256=Dgs0PXHytekX4SUL6KsjjN0hHcOtGLvv1GRGbh6PswU,1408
-langgraph_api/queue_entrypoint.py,sha256=dzEOqW4AHGjDKMlsNYBMVjHfQM7pm_49LImfa5G5ySM,2491
+langgraph_api/queue_entrypoint.py,sha256=rW4-OEsWwBsLLp6U0P-de-I0tR3gPZaeBKWB3LRrTr0,2923
 langgraph_api/route.py,sha256=4VBkJMeusfiZtLzyUaKm1HwLHTq0g15y2CRiRhM6xyA,4773
 langgraph_api/schema.py,sha256=2711t4PIBk5dky4gmMndrTRC9CVvAgH47C9FKDxhkBo,5444
 langgraph_api/serde.py,sha256=8fQXg7T7RVUqj_jgOoSOJrWVpQDW0qJKjAjSsEhPHo4,4803
@@ -86,8 +86,8 @@ langgraph_runtime/store.py,sha256=7mowndlsIroGHv3NpTSOZDJR0lCuaYMBoTnTrewjslw,11
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=wrJup7sCRlZXTRagjzGZ7474U1wma4ZzYTkkninrT6M,141875
-langgraph_api-0.2.54.dist-info/METADATA,sha256=fL0RwEb1lBCBPnjK0nwECORd51dAbiuDMRuz6YLW_Xk,3891
-langgraph_api-0.2.54.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langgraph_api-0.2.54.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
-langgraph_api-0.2.54.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.2.54.dist-info/RECORD,,
+langgraph_api-0.2.56.dist-info/METADATA,sha256=wfCknYuOUBNVrO4AM6m33ekJis6bPutLf-EWFDWpPTw,3891
+langgraph_api-0.2.56.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langgraph_api-0.2.56.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
+langgraph_api-0.2.56.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.2.56.dist-info/RECORD,,