langgraph-api 0.2.135__py3-none-any.whl → 0.2.137__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.2.135"
+__version__ = "0.2.137"

langgraph_api/api/assistants.py

@@ -82,9 +82,12 @@ def _get_configurable_jsonschema(graph: Pregel) -> dict:
                 json_schema["properties"].pop(key, None)
         # The type name of the configurable type is not preserved.
         # We'll add it back to the schema if we can.
-        if hasattr(graph, "config_type") and graph.config_type is not None:
-            if hasattr(graph.config_type, "__name__"):
-                json_schema["title"] = graph.config_type.__name__
+        if (
+            hasattr(graph, "config_type")
+            and graph.config_type is not None
+            and hasattr(graph.config_type, "__name__")
+        ):
+            json_schema["title"] = graph.config_type.__name__
         return json_schema
     # If the schema does not have a configurable field, return an empty schema.
     return {}
@@ -263,18 +266,16 @@ async def get_assistant_graph(
             drawable_graph = await graph.fetch_graph(xray=xray)
             json_graph = drawable_graph.to_json()
             for node in json_graph.get("nodes", []):
-                if data := node.get("data"):
-                    if isinstance(data, dict):
-                        data.pop("id", None)
+                if (data := node.get("data")) and isinstance(data, dict):
+                    data.pop("id", None)
             return ApiResponse(json_graph)
 
         try:
             drawable_graph = await graph.aget_graph(xray=xray)
             json_graph = drawable_graph.to_json()
             for node in json_graph.get("nodes", []):
-                if data := node.get("data"):
-                    if isinstance(data, dict):
-                        data.pop("id", None)
+                if (data := node.get("data")) and isinstance(data, dict):
+                    data.pop("id", None)
             return ApiResponse(json_graph)
         except NotImplementedError:
             raise HTTPException(

langgraph_api/api/runs.py

@@ -212,9 +212,11 @@ async def wait_run(request: ApiRequest):
             )
         ) as stream:
             async for mode, chunk, _ in stream:
-                if mode == b"values":
-                    vchunk = chunk
-                elif mode == b"updates" and b"__interrupt__" in chunk:
+                if (
+                    mode == b"values"
+                    or mode == b"updates"
+                    and b"__interrupt__" in chunk
+                ):
                     vchunk = chunk
                 elif mode == b"error":
                     vchunk = orjson.dumps({"__error__": orjson.Fragment(chunk)})
@@ -295,9 +297,11 @@ async def wait_run_stateless(request: ApiRequest):
             )
         ) as stream:
             async for mode, chunk, _ in stream:
-                if mode == b"values":
-                    vchunk = chunk
-                elif mode == b"updates" and b"__interrupt__" in chunk:
+                if (
+                    mode == b"values"
+                    or mode == b"updates"
+                    and b"__interrupt__" in chunk
+                ):
                     vchunk = chunk
                 elif mode == b"error":
                     vchunk = orjson.dumps({"__error__": orjson.Fragment(chunk)})

langgraph_api/api/store.py

@@ -134,6 +134,7 @@ async def list_namespaces(request: ApiRequest):
     payload = await request.json(StoreListNamespacesRequest)
     prefix = tuple(payload["prefix"]) if payload.get("prefix") else None
     suffix = tuple(payload["suffix"]) if payload.get("suffix") else None
+    err = None
     if prefix and (err := _validate_namespace(prefix)):
         return err
     if suffix and (err := _validate_namespace(suffix)):

langgraph_api/asyncio.py

@@ -1,7 +1,7 @@
 import asyncio
 import concurrent.futures
 from collections.abc import AsyncIterator, Coroutine
-from contextlib import AbstractAsyncContextManager
+from contextlib import AbstractAsyncContextManager, suppress
 from functools import partial
 from typing import Any, Generic, TypeVar
 
@@ -26,10 +26,8 @@ def get_event_loop() -> asyncio.AbstractEventLoop:
 
 
 async def sleep_if_not_done(delay: float, done: asyncio.Event) -> None:
-    try:
+    with suppress(TimeoutError):
         await asyncio.wait_for(done.wait(), delay)
-    except TimeoutError:
-        pass
 
 
 class ValueEvent(asyncio.Event):
@@ -95,14 +93,13 @@ PENDING_TASKS = set()
 
 
 def _create_task_done_callback(
-    ignore_exceptions: tuple[Exception, ...],
+    ignore_exceptions: tuple[type[Exception], ...],
     task: asyncio.Task | asyncio.Future,
 ) -> None:
     PENDING_TASKS.discard(task)
     try:
-        if exc := task.exception():
-            if not isinstance(exc, ignore_exceptions):
-                logger.exception("asyncio.task failed", exc_info=exc)
+        if (exc := task.exception()) and not isinstance(exc, ignore_exceptions):
+            logger.exception("asyncio.task failed", exc_info=exc)
     except asyncio.CancelledError:
         pass
 
@@ -176,16 +173,13 @@ class SimpleTaskGroup(AbstractAsyncContextManager["SimpleTaskGroup"]):
         self.taskgroup_name = f" {taskgroup_name} " if taskgroup_name else ""
 
     def _create_task_done_callback(
-        self, ignore_exceptions: tuple[Exception, ...], task: asyncio.Task
+        self, ignore_exceptions: tuple[type[Exception], ...], task: asyncio.Task
     ) -> None:
-        try:
+        with suppress(AttributeError):
             self.tasks.remove(task)
-        except AttributeError:
-            pass
         try:
-            if exc := task.exception():
-                if not isinstance(exc, ignore_exceptions):
-                    logger.exception("asyncio.task failed in task group", exc_info=exc)
+            if (exc := task.exception()) and not isinstance(exc, ignore_exceptions):
+                logger.exception("asyncio.task failed in task group", exc_info=exc)
         except asyncio.CancelledError:
             pass
 
@@ -286,13 +280,8 @@ class AsyncQueue(Generic[T], asyncio.Queue[T]):
                 await getter
             except:
                 getter.cancel()  # Just in case getter is not done yet.
-                try:
-                    # Clean self._getters from canceled getters.
+                with suppress(ValueError):
                     self._getters.remove(getter)
-                except ValueError:
-                    # The getter could be removed from self._getters by a
-                    # previous put_nowait call.
-                    pass
                 if not self.empty() and not getter.cancelled():
                     # We were woken up by put_nowait(), but can't take
                     # the call.  Wake up the next in line.

langgraph_api/auth/custom.py

@@ -239,14 +239,11 @@ def _get_custom_auth_middleware(
 
 @functools.lru_cache(maxsize=1)
 def _get_auth_instance(path: str | None = None) -> Auth | Literal["js"] | None:
-    if path is not None:
-        auth_instance = _load_auth_obj(path)
-    else:
-        auth_instance = None
+    auth_instance = _load_auth_obj(path) if path is not None else None
 
     if auth_instance == "js":
         return auth_instance
-
+    deps = None
     if auth_instance is not None and (
         deps := _get_dependencies(auth_instance._authenticate_handler)
     ):
@@ -525,6 +522,12 @@ class ProxyUser(BaseUser):
         """Proxy any other attributes to the underlying user object."""
         return getattr(self._user, name)
 
+    def __iter__(self):
+        return iter(self._user)
+
+    def __len__(self):
+        return len(self._user)
+
     def __str__(self) -> str:
         return f"{self._user}"
 

langgraph_api/auth/langsmith/backend.py

@@ -86,9 +86,10 @@ class LangsmithAuthBackend(AuthenticationBackend):
             # If tenant id verification is disabled, the bearer token requests
             # are not required to match the tenant id. Api key requests are
             # always required to match the tenant id.
-            if LANGSMITH_AUTH_VERIFY_TENANT_ID or conn.headers.get("x-api-key"):
-                if auth_dict["tenant_id"] != LANGSMITH_TENANT_ID:
-                    raise AuthenticationError("Invalid tenant ID")
+            if (
+                LANGSMITH_AUTH_VERIFY_TENANT_ID or conn.headers.get("x-api-key")
+            ) and auth_dict["tenant_id"] != LANGSMITH_TENANT_ID:
+                raise AuthenticationError("Invalid tenant ID")
 
         credentials = AuthCredentials(["authenticated"])
         user = StudioUser(auth_dict.get("user_id"), is_authenticated=True)

langgraph_api/auth/langsmith/client.py

@@ -1,5 +1,5 @@
 from collections.abc import AsyncGenerator
-from contextlib import asynccontextmanager
+from contextlib import asynccontextmanager, suppress
 from typing import Any
 
 import httpx
@@ -17,11 +17,10 @@ _client: "JsonHttpClient"
 def is_retriable_error(exception: BaseException) -> bool:
     if isinstance(exception, httpx.TransportError):
         return True
-    if isinstance(exception, httpx.HTTPStatusError):
-        if exception.response.status_code > 499:
-            return True
-
-    return False
+    return (
+        isinstance(exception, httpx.HTTPStatusError)
+        and exception.response.status_code > 499
+    )
 
 
 retry_httpx = retry(
@@ -106,10 +105,8 @@ def create_client() -> JsonHttpClient:
 async def close_auth_client() -> None:
     """Close the auth http client."""
     global _client
-    try:
+    with suppress(NameError):
         await _client.client.aclose()
-    except NameError:
-        pass
 
 
 async def initialize_auth_client() -> None:
@@ -132,10 +129,7 @@ async def auth_client() -> AsyncGenerator[JsonHttpClient, None]:
             await client.client.aclose()
     else:
         try:
-            if not _client.client.is_closed:
-                found = True
-            else:
-                found = False
+            found = bool(not _client.client.is_closed)
         except NameError:
             found = False
         if found:

langgraph_api/config.py

@@ -242,9 +242,12 @@ CORS_CONFIG: CorsConfig | None = env("CORS_CONFIG", cast=_parse_json, default=No
     }
 }
 """
-if CORS_CONFIG is not None and CORS_ALLOW_ORIGINS != "*":
-    if CORS_CONFIG.get("allow_origins") is None:
-        CORS_CONFIG["allow_origins"] = CORS_ALLOW_ORIGINS
+if (
+    CORS_CONFIG is not None
+    and CORS_ALLOW_ORIGINS != "*"
+    and CORS_CONFIG.get("allow_origins") is None
+):
+    CORS_CONFIG["allow_origins"] = CORS_ALLOW_ORIGINS
 
 # queue
 
@@ -371,7 +374,7 @@ API_VARIANT = env("LANGSMITH_LANGGRAPH_API_VARIANT", cast=str, default="")
 # UI
 UI_USE_BUNDLER = env("LANGGRAPH_UI_BUNDLER", cast=bool, default=False)
 IS_QUEUE_ENTRYPOINT = False
-
+ref_sha = None
 if not os.getenv("LANGCHAIN_REVISION_ID") and (
     ref_sha := os.getenv("LANGSMITH_LANGGRAPH_GIT_REF_SHA")
 ):

langgraph_api/cron_scheduler.py

@@ -46,12 +46,11 @@ async def cron_scheduler():
                                         cron["cron_id"],
                                     )
                                 )
-                        except Exception as e:
-                            logger.error(
+                        except Exception:
+                            logger.exception(
                                 "Error scheduling cron run cron_id={}".format(
                                     cron["cron_id"]
-                                ),
-                                exc_info=e,
+                                )
                             )
                         next_run_date = await run_in_executor(
                             None, next_cron_date, cron["schedule"], cron["now"]
@@ -63,6 +62,6 @@ async def cron_scheduler():
             await asyncio.sleep(SLEEP_TIME)
         except asyncio.CancelledError:
             raise
-        except Exception as e:
-            logger.error("Error in cron_scheduler", exc_info=e)
+        except Exception:
+            logger.exception("Error in cron_scheduler")
             await asyncio.sleep(SLEEP_TIME + random())

langgraph_api/http.py

@@ -1,4 +1,5 @@
 import asyncio
+import contextlib
 
 import httpx
 from tenacity import (
@@ -65,10 +66,8 @@ class JsonHttpClient:
                 res.raise_for_status()
         finally:
             # We don't need the response body, so we close the response
-            try:
+            with contextlib.suppress(UnboundLocalError):
                 await res.aclose()
-            except UnboundLocalError:
-                pass
 
 
 _http_client: JsonHttpClient
@@ -173,10 +172,7 @@ async def http_request(
 
     content = None
     if body is not None:
-        if isinstance(body, str):
-            content = body.encode("utf-8")
-        else:
-            content = body
+        content = body.encode("utf-8") if isinstance(body, str) else body
     elif json is not None:
         content = json_dumpb(json)
 

langgraph_api/js/remote.py

@@ -1,4 +1,5 @@
 import asyncio
+import contextlib
 import json
 import logging
 import os
@@ -7,7 +8,6 @@ import shutil
 import ssl
 from collections import deque
 from collections.abc import AsyncIterator, Callable
-from contextlib import AbstractContextManager
 from typing import Any, Literal, Self, cast
 
 import certifi
@@ -57,11 +57,10 @@ GRAPH_PORT = 5556
 GRAPH_HTTP_PORT = 5557
 SSL = ssl.create_default_context(cafile=certifi.where())
 
-if port := int(os.getenv("PORT", "8080")):
-    if port in (GRAPH_PORT, REMOTE_PORT):
-        raise ValueError(
-            f"PORT={port} is a reserved port for the JS worker. Please choose a different port."
-        )
+if (port := int(os.getenv("PORT", "8080"))) and port in (GRAPH_PORT, REMOTE_PORT):
+    raise ValueError(
+        f"PORT={port} is a reserved port for the JS worker. Please choose a different port."
+    )
 
 _client = httpx.AsyncClient(
     base_url=f"http://localhost:{GRAPH_PORT}",
@@ -379,6 +378,7 @@ async def run_js_process(paths_str: str | None, watch: bool = False):
             if False
             else ("tsx", "--import", client_preload_file, client_file)
         )
+        process = None
         try:
             process = await asyncio.create_subprocess_exec(
                 *args,
@@ -395,11 +395,10 @@ async def run_js_process(paths_str: str | None, watch: bool = False):
             raise Exception(f"JS process exited with code {code}")
         except asyncio.CancelledError:
             logger.info("Terminating JS graphs process")
-            try:
-                process.terminate()
-                await process.wait()
-            except (UnboundLocalError, ProcessLookupError):
-                pass
+            if process is not None:
+                with contextlib.suppress(ProcessLookupError):
+                    process.terminate()
+                    await process.wait()
             raise
         except Exception:
             if attempt >= 3:
@@ -424,6 +423,7 @@ async def run_js_http_process(
         client_file = os.path.join(os.path.dirname(__file__), "client.http.mts")
         args = ("tsx", "watch", client_file) if watch else ("tsx", client_file)
         pid = None
+        process = None
         try:
             process = await asyncio.create_subprocess_exec(
                 *args,
@@ -445,11 +445,10 @@ async def run_js_http_process(
 
         except asyncio.CancelledError:
             logger.info("Shutting down JS HTTP process [%d]", pid or -1)
-            try:
-                process.terminate()
-                await process.wait()
-            except (UnboundLocalError, ProcessLookupError):
-                pass
+            if process is not None:
+                with contextlib.suppress(ProcessLookupError):
+                    process.terminate()
+                    await process.wait()
             raise
         except Exception:
             if attempt >= 3:
@@ -766,7 +765,7 @@ async def run_remote_checkpointer():
     await server.serve()
 
 
-class DisableHttpxLoggingContextManager(AbstractContextManager):
+class DisableHttpxLoggingContextManager(contextlib.AbstractContextManager):
     """
     Disable HTTP/1.1 200 OK logs spamming stdout.
     """
@@ -891,11 +890,10 @@ class CustomJsAuthBackend(AuthenticationBackend):
         cache_key = None
         if self.cache_keys:
             cache_key = tuple((k, headers[k]) for k in self.cache_keys if k in headers)
-            if cache_key:
-                if self.ttl_cache is not None:
-                    cached = self.ttl_cache.get(cache_key)
-                    if cached:
-                        return cached
+            if cache_key and self.ttl_cache is not None:
+                cached = self.ttl_cache.get(cache_key)
+                if cached:
+                    return cached
 
         res = await _client.post("/auth/authenticate", headers=headers)
         data = res.json()
@@ -961,11 +959,14 @@ async def handle_js_auth_event(
     # mutate metadata in value if applicable
     # we need to preserve the identity of the object, so cannot create a new
     # dictionary, otherwise the changes will not persist
-    if isinstance(value, dict) and (updated_value := response.get("value")):
-        if isinstance(value.get("metadata"), dict) and (
-            metadata := updated_value.get("metadata")
-        ):
-            value["metadata"].update(metadata)
+    metadata = None
+    if (
+        isinstance(value, dict)
+        and (updated_value := response.get("value"))
+        and isinstance(value.get("metadata"), dict)
+        and (metadata := updated_value.get("metadata"))
+    ):
+        value["metadata"].update(metadata)
 
     return filters
 

langgraph_api/js/sse.py

@@ -1,5 +1,6 @@
 """Adapted from httpx_sse to split lines on \n, \r, \r\n per the SSE spec."""
 
+import contextlib
 from collections.abc import AsyncIterator
 
 import httpx
@@ -119,10 +120,8 @@ class SSEDecoder:
             else:
                 self._last_event_id = value.decode()
         elif fieldname == b"retry":
-            try:
+            with contextlib.suppress(TypeError, ValueError):
                 self._retry = int(value)
-            except (TypeError, ValueError):
-                pass
         else:
             pass  # Field is ignored.
 

langgraph_api/js/ui.py

@@ -1,4 +1,5 @@
 import asyncio
+import contextlib
 import os
 import shutil
 import sys
@@ -56,6 +57,7 @@ async def _start_ui_bundler_process():
         os.mkdir(UI_ROOT_DIR)
 
     pid = None
+    process = None
     try:
         process = await asyncio.create_subprocess_exec(
             npx_path,
@@ -74,11 +76,10 @@ async def _start_ui_bundler_process():
 
     except asyncio.CancelledError:
         logger.info("Shutting down UI bundler process [%d]", pid or -1)
-        try:
-            process.terminate()
-            await process.wait()
-        except (UnboundLocalError, ProcessLookupError):
-            pass
+        if process is not None:
+            with contextlib.suppress(ProcessLookupError):
+                process.terminate()
+                await process.wait()
         raise
 
 

langgraph_api/logging.py

@@ -16,8 +16,16 @@ LOG_JSON = log_env("LOG_JSON", cast=bool, default=False)
 LOG_COLOR = log_env("LOG_COLOR", cast=bool, default=True)
 LOG_LEVEL = log_env("LOG_LEVEL", cast=str, default="INFO")
 
+logger = logging.getLogger()
 logging.getLogger().setLevel(LOG_LEVEL.upper())
 logging.getLogger("psycopg").setLevel(logging.WARNING)
+if hasattr(logger, "isEnabledFor"):
+    LOG_LEVEL_DEBUG = logger.isEnabledFor(logging.DEBUG)
+elif hasattr(logger, "is_enabled_for"):
+    LOG_LEVEL_DEBUG = logger.is_enabled_for(logging.DEBUG)
+else:
+    LOG_LEVEL_DEBUG = False
+del logger
 
 worker_config = contextvars.ContextVar[dict[str, typing.Any] | None](
     "worker_config", default=None
@@ -86,6 +94,7 @@ class AddLoggingContext:
     ) -> EventDict:
         if (ctx := worker_config.get()) is not None:
             event_dict.update(ctx)
+        lgnode = None
         if (
             self.cvar is not None
             and (conf := self.cvar.get())

langgraph_api/models/run.py

@@ -1,4 +1,5 @@
 import asyncio
+import contextlib
 import time
 import urllib.parse
 import uuid
@@ -381,19 +382,18 @@ async def create_valid_run(
                 else None
             ),
             run_put_ms=int((time.time() - put_time_start) * 1_000),
+            checkpoint_id=str(checkpoint_id),
         )
         # inserted, proceed
         if multitask_strategy in ("interrupt", "rollback") and inflight_runs:
-            try:
+            with contextlib.suppress(HTTPException):
+                # if we can't find the inflight runs again, we can proceeed
                 await Runs.cancel(
                     conn,
                     [run["run_id"] for run in inflight_runs],
                     thread_id=thread_id_,
                     action=multitask_strategy,
                 )
-            except HTTPException:
-                # if we can't find the inflight runs again, we can proceeed
-                pass
         return first
     elif multitask_strategy == "reject":
         raise HTTPException(

langgraph_api/serde.py

@@ -84,11 +84,10 @@ def default(obj):
         return obj.total_seconds()
     elif isinstance(obj, Decimal):
         return decimal_encoder(obj)
-    elif isinstance(obj, uuid.UUID):
-        return str(obj)
     elif isinstance(  # noqa: UP038
         obj,
         (
+            uuid.UUID,
             IPv4Address,
             IPv4Interface,
             IPv4Network,

langgraph_api/store.py

@@ -69,6 +69,7 @@ async def _yield_store(value: Any):
 
 async def collect_store_from_env() -> None:
     global CUSTOM_STORE
+    store_path = None
     if not config.STORE_CONFIG or not (store_path := config.STORE_CONFIG.get("path")):
         return
     await logger.ainfo(

langgraph_api/stream.py

@@ -14,6 +14,7 @@ from langchain_core.messages import (
     convert_to_messages,
     message_chunk_to_message,
 )
+from langchain_core.runnables import RunnableConfig
 from langgraph.errors import (
     EmptyChannelError,
     EmptyInputError,
@@ -107,7 +108,7 @@ def _preprocess_debug_checkpoint(
         **payload,
         "checkpoint": runnable_config_to_checkpoint(payload["config"]),
         "parent_checkpoint": runnable_config_to_checkpoint(
-            payload["parent_config"] if "parent_config" in payload else None
+            payload.get("parent_config", None)
         ),
         "tasks": [_preproces_debug_checkpoint_task(t) for t in payload["tasks"]],
     }
@@ -124,7 +125,7 @@ async def astream_state(
     attempt: int,
     done: ValueEvent,
     *,
-    on_checkpoint: Callable[[CheckpointPayload], None] = lambda _: None,
+    on_checkpoint: Callable[[CheckpointPayload | None], None] = lambda _: None,
     on_task_result: Callable[[TaskResultPayload], None] = lambda _: None,
 ) -> AnyStream:
     """Stream messages from the runnable."""
@@ -136,11 +137,12 @@ async def astream_state(
     subgraphs = kwargs.get("subgraphs", False)
     temporary = kwargs.pop("temporary", False)
     context = kwargs.pop("context", None)
-    config = kwargs.pop("config")
+    config = cast(RunnableConfig, kwargs.pop("config"))
+    configurable = config["configurable"]
     stack = AsyncExitStack()
     graph = await stack.enter_async_context(
         get_graph(
-            config["configurable"]["graph_id"],
+            configurable["graph_id"],
             config,
             store=(await api_store.get_store()),
             checkpointer=None if temporary else Checkpointer(),
@@ -171,6 +173,8 @@ async def astream_state(
     if "updates" not in stream_modes_set:
         stream_modes_set.add("updates")
         only_interrupt_updates = True
+    else:
+        only_interrupt_updates = False
     # attach attempt metadata
     config["metadata"]["run_attempt"] = attempt
     # attach langgraph metadata
@@ -182,7 +186,7 @@ async def astream_state(
     # attach node counter
     is_remote_pregel = isinstance(graph, BaseRemotePregel)
     if not is_remote_pregel:
-        config["configurable"]["__pregel_node_finished"] = incr_nodes
+        configurable["__pregel_node_finished"] = incr_nodes
 
     # attach run_id to config
     # for attempts beyond the first, use a fresh, unique run_id
@@ -195,9 +199,9 @@ async def astream_state(
     yield "metadata", {"run_id": run_id, "attempt": attempt}
 
     #  is a langsmith tracing project is specified, additionally pass that in to tracing context
-    if ls_project := config["configurable"].get("__langsmith_project__"):
+    if ls_project := configurable.get("__langsmith_project__"):
         updates = None
-        if example_id := config["configurable"].get("__langsmith_example_id__"):
+        if example_id := configurable.get("__langsmith_example_id__"):
             updates = {"reference_example_id": example_id}
 
         await stack.enter_async_context(
@@ -237,13 +241,18 @@ async def astream_state(
                 event = cast(dict, event)
                 if event.get("tags") and "langsmith:hidden" in event["tags"]:
                     continue
-                if "messages" in stream_mode and isinstance(graph, BaseRemotePregel):
-                    if event["event"] == "on_custom_event" and event["name"] in (
+                if (
+                    "messages" in stream_mode
+                    and isinstance(graph, BaseRemotePregel)
+                    and event["event"] == "on_custom_event"
+                    and event["name"]
+                    in (
                         "messages/complete",
                         "messages/partial",
                         "messages/metadata",
-                    ):
-                        yield event["name"], event["data"]
+                    )
+                ):
+                    yield event["name"], event["data"]
                 # TODO support messages-tuple for js graphs
                 if event["event"] == "on_chain_stream" and event["run_id"] == run_id:
                     if subgraphs:

langgraph_api/worker.py

@@ -118,8 +118,10 @@ async def worker(
         resumable=resumable,
     )
 
-    def on_checkpoint(checkpoint_arg: CheckpointPayload):
+    def on_checkpoint(checkpoint_arg: CheckpointPayload | None):
         nonlocal checkpoint
+        if checkpoint_arg is None:
+            logger.warning("Null checkpoint received")
         checkpoint = checkpoint_arg
 
     def on_task_result(task_result: TaskResultPayload):
@@ -140,9 +142,8 @@ async def worker(
             await consume(stream, run_id, resumable, stream_modes)
         except Exception as e:
             if not isinstance(e, UserRollback | UserInterrupt):
-                logger.error(
+                logger.exception(
                     f"Run encountered an error in graph: {type(e)}({e})",
-                    exc_info=e,
                 )
             # TimeoutError is a special case where we rely on asyncio.wait_for to timeout runs
             # Convert user TimeoutErrors to a custom class so we can distinguish and later convert back

{langgraph_api-0.2.135.dist-info → langgraph_api-0.2.137.dist-info}/METADATA

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

{langgraph_api-0.2.135.dist-info → langgraph_api-0.2.137.dist-info}/RECORD

@@ -1,51 +1,51 @@
-langgraph_api/__init__.py,sha256=EI8dPBQX_KEje0C_yiwGFf1oJ65gVj2GXRmDyafg6MM,24
+langgraph_api/__init__.py,sha256=Ma7kyjCpQ7p4EHYWYV7s3pmyAmbmTKBpxJUymCIyQNo,24
 langgraph_api/asgi_transport.py,sha256=XtiLOu4WWsd-xizagBLzT5xUkxc9ZG9YqwvETBPjBFE,5161
-langgraph_api/asyncio.py,sha256=l4fVoYIcczMqC2Wrj4LTk50nKV29AXwweiehOwaeC4Y,9754
+langgraph_api/asyncio.py,sha256=mZ7G32JjrGxrlH4OMy7AKlBQo5bZt4Sm2rlrBcU-Vj8,9483
 langgraph_api/cli.py,sha256=-ruIeKi1imvS6GriOfRDZY-waV4SbWiJ0BEFAciPVYI,16330
 langgraph_api/command.py,sha256=3O9v3i0OPa96ARyJ_oJbLXkfO8rPgDhLCswgO9koTFA,768
-langgraph_api/config.py,sha256=LxWOteBq-RNyr0Dc3Tm7HwoXrn1aDdvgnKlDsXCDFMg,12111
-langgraph_api/cron_scheduler.py,sha256=CiwZ-U4gDOdG9zl9dlr7mH50USUgNB2Fvb8YTKVRBN4,2625
+langgraph_api/config.py,sha256=9wXEcxvVfPQjiR7jugomwTJ3vs5d-YoYYfgLDjRc3EQ,12133
+langgraph_api/cron_scheduler.py,sha256=25wYzEQrhPEivZrAPYOmzLPDOQa-aFogU37mTXc9TJk,2566
 langgraph_api/errors.py,sha256=zlnl3xXIwVG0oGNKKpXf1an9Rn_SBDHSyhe53hU6aLw,1858
 langgraph_api/executor_entrypoint.py,sha256=ClMyM9TB9oPisQzHqixA77Lnj_QGUg55MtQx-xku4o8,671
 langgraph_api/feature_flags.py,sha256=GjwmNjfg0Jhs3OzR2VbK2WgrRy3o5l8ibIYiUtQkDPA,363
 langgraph_api/graph.py,sha256=HTjJNQadrdi1tzJYNJ_iPIR6-zqC4-hj6YTD6zGQHYA,25072
-langgraph_api/http.py,sha256=xCeyqm9Vafx_8OaUfwlIMPZTspJQzivgcJqTc4wweaE,5704
+langgraph_api/http.py,sha256=fyK-H-0UfNy_BzuVW3aWWGvhRavmGAVMkDwDArryJ_4,5659
 langgraph_api/http_metrics.py,sha256=MU9ccXt7aBb0AJ2SWEjwtbtbJEWmeqSdx7-CI51e32o,5594
-langgraph_api/logging.py,sha256=v7TOQt_YuZ_lTQ4rp_9hE6pLtSKxObkuFxyAdHW0y5c,4862
+langgraph_api/logging.py,sha256=ZZ95dDdWDayIbH1bgwNfn0U3CQ8kDoAvDFBDACna4-A,5150
 langgraph_api/metadata.py,sha256=fVsbwxVitAj4LGVYpCcadYeIFANEaNtcx6LBxQLcTqg,6949
 langgraph_api/patch.py,sha256=iLwSd9ZWoVj6MxozMyGyMvWWbE9RIP5eZX1dpCBSlSU,1480
 langgraph_api/queue_entrypoint.py,sha256=KDLpQtBu3amZTbNHS-RGFLR0DphuVQN6kUZm3ZGLe9g,5991
 langgraph_api/route.py,sha256=EBhELuJ1He-ZYcAnR5YTImcIeDtWthDae5CHELBxPkM,5056
 langgraph_api/schema.py,sha256=6gabS4_1IeRWV5lyuDV-2i__8brXl89elAlmD5BmEII,8370
-langgraph_api/serde.py,sha256=5F4xMTRY3kNwpdkAzM48KzxFEUmVD1I3CaVWzp_syT8,6067
+langgraph_api/serde.py,sha256=3GvelKhySjlXaNqpg2GyUxU6-NEkvif7WlMF9if_EgU,6029
 langgraph_api/server.py,sha256=uCAqPgCLJ6ckslLs0i_dacSR8mzuR0Y6PkkJYk0O3bE,7196
 langgraph_api/sse.py,sha256=SLdtZmTdh5D8fbWrQjuY9HYLd2dg8Rmi6ZMmFMVc2iE,4204
 langgraph_api/state.py,sha256=5RTOShiFVnkx-o6t99_x63CGwXw_8Eb-dSTpYirP8ro,4683
-langgraph_api/store.py,sha256=_UwEzGXKMFvpnyz1DUeOmfpy2w3WhPAtAJzIh7VTRBY,4679
-langgraph_api/stream.py,sha256=P82M1yVbn1N20ZRSLb6_F1wbkfQLVU1OGEHF2ES-Nvg,18199
+langgraph_api/store.py,sha256=NIoNZojs6NbtG3VLBPQEFNttvp7XPkHAfjbQ3gY7aLY,4701
+langgraph_api/stream.py,sha256=iEApgVxJU9v58J5oKSuNe_c7ThpWf0bprgIDHUs8IzA,18397
 langgraph_api/thread_ttl.py,sha256=7H3gFlWcUiODPoaEzcwB0LR61uvcuyjD0ew_4BztB7k,1902
 langgraph_api/traceblock.py,sha256=Qq5CUdefnMDaRDnyvBSWGBClEj-f3oO7NbH6fedxOSE,630
 langgraph_api/utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/validation.py,sha256=86jftgOsMa7tkeshBw6imYe7zyUXPoVuf5Voh6dFiR8,5285
 langgraph_api/webhook.py,sha256=SvSM1rdnNtiH4q3JQYmAqJUk2Sable5xAcwOLuRhtlo,1723
-langgraph_api/worker.py,sha256=0ztx8AbggDdEjnW40Fai85S2jVGtFcNLU1kGWdN_w24,15198
+langgraph_api/worker.py,sha256=HVGyGVEYcXG-iKVgoBdFgANGxPjSs57JRl5OB4ra4nw,15267
 langgraph_api/api/__init__.py,sha256=WHy6oNLWtH1K7AxmmsU9RD-Vm6WP-Ov16xS8Ey9YCmQ,6090
-langgraph_api/api/assistants.py,sha256=ffXBUTTs6bBxDISuOs1KVfcjBJuaS8R_j5S6Omzp1i4,16848
+langgraph_api/api/assistants.py,sha256=5gVvU58Y1-EftBhCHGbEaOi_7cqGMKWhOt_GVfBC0Gg,16836
 langgraph_api/api/mcp.py,sha256=qe10ZRMN3f-Hli-9TI8nbQyWvMeBb72YB1PZVbyqBQw,14418
 langgraph_api/api/meta.py,sha256=w88TK1Wu4xOhgCfs04LBfL4pZkWhUW6QRwwAWdFby5A,4245
 langgraph_api/api/openapi.py,sha256=If-z1ckXt-Yu5bwQytK1LWyX_T7G46UtLfixgEP8hwc,11959
-langgraph_api/api/runs.py,sha256=Y52LiXsEtMOF05WhgK00g0CsYrqUUcWxVaUVCsoujtM,21760
-langgraph_api/api/store.py,sha256=TSeMiuMfrifmEnEbL0aObC2DPeseLlmZvAMaMzPgG3Y,5535
+langgraph_api/api/runs.py,sha256=AiohGTFLjWCb-oTXoNDvPMod4v6RS_ivlieoiqDmtQM,21812
+langgraph_api/api/store.py,sha256=xGcPFx4v-VxlK6HRU9uCjzCQ0v66cvc3o_PB5_g7n0Q,5550
 langgraph_api/api/threads.py,sha256=Ap5zUcYqK5GJqwEc-q4QY6qCkmbLxfMmEvQZm0MCFxk,10427
 langgraph_api/api/ui.py,sha256=_genglTUy5BMHlL0lkQypX524yFv6Z5fraIvnrxp7yE,2639
 langgraph_api/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_api/auth/custom.py,sha256=b2NOPqBFUQiFkwlfFg4agZo3YfskTZMJyolv52suCeI,22433
+langgraph_api/auth/custom.py,sha256=psETw_GpLWClBbd_ESVPRLUz9GLQ0_XNsuUDSVbtZy0,22522
 langgraph_api/auth/middleware.py,sha256=jDA4t41DUoAArEY_PNoXesIUBJ0nGhh85QzRdn5EPD0,1916
 langgraph_api/auth/noop.py,sha256=Bk6Nf3p8D_iMVy_OyfPlyiJp_aEwzL-sHrbxoXpCbac,586
 langgraph_api/auth/studio_user.py,sha256=fojJpexdIZYI1w3awiqOLSwMUiK_M_3p4mlfQI0o-BE,454
 langgraph_api/auth/langsmith/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_api/auth/langsmith/backend.py,sha256=36nQnVb9VtNvSnLiNYWAI9o9H74I-mSN2X-FrWoj0QA,3646
-langgraph_api/auth/langsmith/client.py,sha256=-KyZSTyeiMhupkPfr--nlm_ELR1ZkjM-h61eGcMG5E0,4002
+langgraph_api/auth/langsmith/backend.py,sha256=060NPoZ82J1Y23hr3XgZnxyzhQ5lZngBWPcUy0QntjY,3658
+langgraph_api/auth/langsmith/client.py,sha256=Kn9503en1tmlNtkbvqRxYSRCOUrWaVpqvxyLLb1cgzY,3908
 langgraph_api/js/.gitignore,sha256=l5yI6G_V6F1600I1IjiUKn87f4uYIrBAYU1MOyBBhg4,59
 langgraph_api/js/.prettierrc,sha256=0es3ovvyNIqIw81rPQsdt1zCQcOdBqyR_DMbFE4Ifms,19
 langgraph_api/js/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -56,12 +56,12 @@ langgraph_api/js/client.mts,sha256=gDvYiW7Qfl4re2YhZ5oNqtuvffnW_Sf7DK5aUbKB3vw,3
 langgraph_api/js/errors.py,sha256=Cm1TKWlUCwZReDC5AQ6SgNIVGD27Qov2xcgHyf8-GXo,361
 langgraph_api/js/global.d.ts,sha256=j4GhgtQSZ5_cHzjSPcHgMJ8tfBThxrH-pUOrrJGteOU,196
 langgraph_api/js/package.json,sha256=syy2fEcmTxGQVfz4P9MUTgoTxHr1MUcA1rDXemAig2U,1335
-langgraph_api/js/remote.py,sha256=x2gO12HBriCb4bFXf4lt6uqDgoOKFihy95kHFSBz7bA,38374
+langgraph_api/js/remote.py,sha256=aeszJ0HGbcL9oExyeWYHb0xLV75U3KgVSxjm3ZK_a48,38403
 langgraph_api/js/schema.py,sha256=M4fLtr50O1jck8H1hm_0W4cZOGYGdkrB7riLyCes4oY,438
-langgraph_api/js/sse.py,sha256=tVcAGVz5jOKWsESxoqm0Nk1B9yP2A7cRcVDNnR1bUv4,4144
+langgraph_api/js/sse.py,sha256=hHkbncnYnXNIbHhAWneGWYkHp4UhhhGB7-MYtDrY264,4141
 langgraph_api/js/traceblock.mts,sha256=QtGSN5VpzmGqDfbArrGXkMiONY94pMQ5CgzetT_bKYg,761
 langgraph_api/js/tsconfig.json,sha256=imCYqVnqFpaBoZPx8k1nO4slHIWBFsSlmCYhO73cpBs,341
-langgraph_api/js/ui.py,sha256=XNT8iBcyT8XmbIqSQUWd-j_00HsaWB2vRTVabwFBkik,2439
+langgraph_api/js/ui.py,sha256=l9regrvKIxLOjH5SIYE2nhr8QCKLK1Q_1pZgxdL71X4,2488
 langgraph_api/js/yarn.lock,sha256=M-XjLAvW6cz56lc-IwNPbjLw8KNIKVS_k-haRP4QmRE,84904
 langgraph_api/js/src/graph.mts,sha256=9zTQNdtanI_CFnOwNRoamoCVHHQHGbNlbm91aRxDeOc,2675
 langgraph_api/js/src/load.hooks.mjs,sha256=xNVHq75W0Lk6MUKl1pQYrx-wtQ8_neiUyI6SO-k0ecM,2235
@@ -75,7 +75,7 @@ langgraph_api/middleware/http_logger.py,sha256=2LABfhzTAUtqT8nf1ACy8cYXteatkwraB
 langgraph_api/middleware/private_network.py,sha256=eYgdyU8AzU2XJu362i1L8aSFoQRiV7_aLBPw7_EgeqI,2111
 langgraph_api/middleware/request_id.py,sha256=SDj3Yi3WvTbFQ2ewrPQBjAV8sYReOJGeIiuoHeZpR9g,1242
 langgraph_api/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_api/models/run.py,sha256=p5F7npi9TFcMUyyn81_InljCg8LE8jKoFSWVl4XtbZ4,15434
+langgraph_api/models/run.py,sha256=q99y57RqUgZgw5kNkUeezo5E9yazg-4-uxKhJR2agag,15479
 langgraph_api/tunneling/cloudflare.py,sha256=iKb6tj-VWPlDchHFjuQyep2Dpb-w2NGfJKt-WJG9LH0,3650
 langgraph_api/utils/__init__.py,sha256=kj3uCnO2Md9EEhabm331Tg4Jx9qXcxbACMh2T2P-FYw,5028
 langgraph_api/utils/cache.py,sha256=SrtIWYibbrNeZzLXLUGBFhJPkMVNQnVxR5giiYGHEfI,1810
@@ -97,8 +97,8 @@ langgraph_runtime/store.py,sha256=7mowndlsIroGHv3NpTSOZDJR0lCuaYMBoTnTrewjslw,11
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=h1LbSeGqr2Oor6vO8d3m67XJ1lHhVYVyt2ULvyhf_Ks,160215
-langgraph_api-0.2.135.dist-info/METADATA,sha256=rQuOdto6JZPKYdzkN_TJF1rK2xjaiXnQ_Pck-Aa1seQ,3892
-langgraph_api-0.2.135.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langgraph_api-0.2.135.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
-langgraph_api-0.2.135.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.2.135.dist-info/RECORD,,
+langgraph_api-0.2.137.dist-info/METADATA,sha256=TF23VjTlddkNdgJewtd8X-EQxYt2wY2w_jF_etcntSU,3892
+langgraph_api-0.2.137.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langgraph_api-0.2.137.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
+langgraph_api-0.2.137.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.2.137.dist-info/RECORD,,