langgraph-api 0.0.30__py3-none-any.whl → 0.0.32__py3-none-any.whl

langgraph_api/api/assistants.py

@@ -87,7 +87,13 @@ def _graph_schemas(graph: Pregel) -> dict:
             f"Failed to get output schema for graph {graph.name} with error: `{str(e)}`"
         )
         output_schema = None
-    state_schema = _state_jsonschema(graph)
+    try:
+        state_schema = _state_jsonschema(graph)
+    except Exception as e:
+        logger.warning(
+            f"Failed to get state schema for graph {graph.name} with error: `{str(e)}`"
+        )
+        state_schema = None
     try:
         config_schema = _get_configurable_jsonschema(graph)
     except Exception as e:
@@ -184,7 +190,8 @@ async def get_assistant_graph(
             return ApiResponse(drawable_graph.to_json())
 
         try:
-            return ApiResponse(graph.get_graph(xray=xray).to_json())
+            drawable_graph = await graph.aget_graph(xray=xray)
+            return ApiResponse(drawable_graph.to_json())
         except NotImplementedError:
             raise HTTPException(
                 422, detail="The graph does not support visualization"

langgraph_api/api/runs.py

@@ -354,7 +354,7 @@ async def join_run_stream_endpoint(request: ApiRequest):
     cancel_on_disconnect = cancel_on_disconnect_str.lower() in {"true", "yes", "1"}
     validate_uuid(thread_id, "Invalid thread ID: must be a UUID")
     validate_uuid(run_id, "Invalid run ID: must be a UUID")
-    stream_mode = request.query_params.get("stream_mode")
+    stream_mode = request.query_params.get("stream_mode") or None
     return EventSourceResponse(
         Runs.Stream.join(
             run_id,

langgraph_api/api/threads.py

@@ -28,12 +28,21 @@ async def create_thread(
     if thread_id := payload.get("thread_id"):
         validate_uuid(thread_id, "Invalid thread ID: must be a UUID")
     async with connect() as conn:
+        thread_id = thread_id or str(uuid4())
         iter = await Threads.put(
             conn,
-            thread_id or uuid4(),
+            thread_id,
             metadata=payload.get("metadata"),
             if_exists=payload.get("if_exists") or "raise",
         )
+
+        if supersteps := payload.get("supersteps"):
+            await Threads.State.bulk(
+                conn,
+                config={"configurable": {"thread_id": thread_id}},
+                supersteps=supersteps,
+            )
+
     return ApiResponse(await fetchone(iter, not_found_code=409))
 
 

langgraph_api/cli.py

@@ -10,7 +10,7 @@ from collections.abc import Mapping, Sequence
 from typing_extensions import TypedDict
 
 if typing.TYPE_CHECKING:
-    from langgraph_api.config import HttpConfig
+    from langgraph_api.config import HttpConfig, StoreConfig
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -75,40 +75,6 @@ def patch_environment(**kwargs):
                 os.environ[key] = value
 
 
-class IndexConfig(TypedDict, total=False):
-    """Configuration for indexing documents for semantic search in the store."""
-
-    dims: int
-    """Number of dimensions in the embedding vectors.
-    
-    Common embedding models have the following dimensions:
-        - OpenAI text-embedding-3-large: 256, 1024, or 3072
-        - OpenAI text-embedding-3-small: 512 or 1536
-        - OpenAI text-embedding-ada-002: 1536
-        - Cohere embed-english-v3.0: 1024
-        - Cohere embed-english-light-v3.0: 384
-        - Cohere embed-multilingual-v3.0: 1024
-        - Cohere embed-multilingual-light-v3.0: 384
-    """
-
-    embed: str
-    """Either a path to an embedding model (./path/to/file.py:embedding_model)
-    or a name of an embedding model (openai:text-embedding-3-small)
-    
-    Note: LangChain is required to use the model format specification.
-    """
-
-    fields: list[str] | None
-    """Fields to extract text from for embedding generation.
-    
-    Defaults to the root ["$"], which embeds the json object as a whole.
-    """
-
-
-class StoreConfig(TypedDict, total=False):
-    index: IndexConfig
-
-
 class SecurityConfig(TypedDict, total=False):
     securitySchemes: dict
     security: list
@@ -160,9 +126,10 @@ def run_server(
     env: str | pathlib.Path | Mapping[str, str] | None = None,
     reload_includes: Sequence[str] | None = None,
     reload_excludes: Sequence[str] | None = None,
-    store: StoreConfig | None = None,
+    store: typing.Optional["StoreConfig"] = None,
     auth: AuthConfig | None = None,
     http: typing.Optional["HttpConfig"] = None,
+    studio_url: str | None = None,
     **kwargs: typing.Any,
 ):
     """Run the LangGraph API server."""
@@ -226,11 +193,11 @@ def run_server(
         ALLOW_PRIVATE_NETWORK="true",
         **(env_vars or {}),
     ):
-        studio_origin = _get_ls_origin() or "https://smith.langchain.com"
-        studio_url = f"{studio_origin}/studio/?baseUrl={local_url}"
+        studio_origin = studio_url or _get_ls_origin() or "https://smith.langchain.com"
+        full_studio_url = f"{studio_origin}/studio/?baseUrl={local_url}"
 
         def _open_browser():
-            nonlocal studio_origin, studio_url
+            nonlocal studio_origin, full_studio_url
             import time
             import urllib.request
             import webbrowser
@@ -252,7 +219,7 @@ def run_server(
                                 try:
                                     org_id = org_id_future.result(timeout=3.0)
                                     if org_id:
-                                        studio_url = f"{studio_origin}/studio/?baseUrl={local_url}&organizationId={org_id}"
+                                        full_studio_url = f"{studio_origin}/studio/?baseUrl={local_url}&organizationId={org_id}"
                                 except TimeoutError as e:
                                     thread_logger.debug(
                                         f"Failed to get organization ID: {str(e)}"
@@ -264,8 +231,8 @@ def run_server(
                                 thread_logger.info(
                                     "🎨 Opening Studio in your browser..."
                                 )
-                                thread_logger.info("URL: " + studio_url)
-                                webbrowser.open(studio_url)
+                                thread_logger.info("URL: " + full_studio_url)
+                                webbrowser.open(full_studio_url)
                                 return
                     except urllib.error.URLError:
                         pass
@@ -280,7 +247,7 @@ def run_server(
 ╩═╝┴ ┴┘└┘└─┘╚═╝┴└─┴ ┴┴  ┴ ┴
 
 - 🚀 API: \033[36m{local_url}\033[0m
-- 🎨 Studio UI: \033[36m{studio_url}\033[0m
+- 🎨 Studio UI: \033[36m{full_studio_url}\033[0m
 - 📚 API Docs: \033[36m{local_url}/docs\033[0m
 
 This in-memory server is designed for development and testing.

langgraph_api/command.py

@@ -0,0 +1,29 @@
+from langgraph.types import Command, Send
+
+from langgraph_api.schema import RunCommand
+
+
+def map_cmd(cmd: RunCommand) -> Command:
+    goto = cmd.get("goto")
+    if goto is not None and not isinstance(goto, list):
+        goto = [cmd.get("goto")]
+
+    update = cmd.get("update")
+    if isinstance(update, tuple | list) and all(
+        isinstance(t, tuple | list) and len(t) == 2 and isinstance(t[0], str)
+        for t in update
+    ):
+        update = [tuple(t) for t in update]
+
+    return Command(
+        update=update,
+        goto=(
+            [
+                it if isinstance(it, str) else Send(it["node"], it["input"])
+                for it in goto
+            ]
+            if goto
+            else None
+        ),
+        resume=cmd.get("resume"),
+    )

langgraph_api/config.py

@@ -1,5 +1,5 @@
 from os import environ, getenv
-from typing import TypedDict
+from typing import Literal, TypedDict
 
 import orjson
 from starlette.config import Config, undefined
@@ -34,6 +34,66 @@ class HttpConfig(TypedDict, total=False):
     cors: CorsConfig | None
 
 
+class IndexConfig(TypedDict, total=False):
+    """Configuration for indexing documents for semantic search in the store."""
+
+    dims: int
+    """Number of dimensions in the embedding vectors.
+    
+    Common embedding models have the following dimensions:
+        - OpenAI text-embedding-3-large: 256, 1024, or 3072
+        - OpenAI text-embedding-3-small: 512 or 1536
+        - OpenAI text-embedding-ada-002: 1536
+        - Cohere embed-english-v3.0: 1024
+        - Cohere embed-english-light-v3.0: 384
+        - Cohere embed-multilingual-v3.0: 1024
+        - Cohere embed-multilingual-light-v3.0: 384
+    """
+
+    embed: str
+    """Either a path to an embedding model (./path/to/file.py:embedding_model)
+    or a name of an embedding model (openai:text-embedding-3-small)
+    
+    Note: LangChain is required to use the model format specification.
+    """
+
+    fields: list[str] | None
+    """Fields to extract text from for embedding generation.
+    
+    Defaults to the root ["$"], which embeds the json object as a whole.
+    """
+
+
+class TTLConfig(TypedDict, total=False):
+    """Configuration for TTL (time-to-live) behavior in the store."""
+
+    refresh_on_read: bool
+    """Default behavior for refreshing TTLs on read operations (GET and SEARCH).
+    
+    If True, TTLs will be refreshed on read operations (get/search) by default.
+    This can be overridden per-operation by explicitly setting refresh_ttl.
+    Defaults to True if not configured.
+    """
+    default_ttl: float | None
+    """Default TTL (time-to-live) in minutes for new items.
+    
+    If provided, new items will expire after this many minutes after their last access.
+    The expiration timer refreshes on both read and write operations.
+    Defaults to None (no expiration).
+    """
+    sweep_interval_minutes: int | None
+    """Interval in minutes between TTL sweep operations.
+    
+    If provided, the store will periodically delete expired items based on TTL.
+    Defaults to None (no sweeping).
+    """
+
+
+class StoreConfig(TypedDict, total=False):
+    index: IndexConfig
+    ttl: TTLConfig
+
+
 # env
 
 env = Config()
@@ -55,6 +115,18 @@ STATS_INTERVAL_SECS = env("STATS_INTERVAL_SECS", cast=int, default=60)
 DATABASE_URI = env("DATABASE_URI", cast=str, default=getenv("POSTGRES_URI", undefined))
 MIGRATIONS_PATH = env("MIGRATIONS_PATH", cast=str, default="/storage/migrations")
 
+
+def _get_encryption_key(key_str: str | None):
+    if not key_str:
+        return None
+    key = key_str.encode(encoding="utf-8")
+    if len(key) not in (16, 24, 32):
+        raise ValueError("LANGGRAPH_AES_KEY must be 16, 24, or 32 bytes long.")
+    return key
+
+
+LANGGRAPH_AES_KEY = env("LANGGRAPH_AES_KEY", default=None, cast=_get_encryption_key)
+
 # redis
 REDIS_URI = env("REDIS_URI", cast=str)
 REDIS_CLUSTER = env("REDIS_CLUSTER", cast=bool, default=False)
@@ -69,7 +141,9 @@ See https://developer.chrome.com/blog/private-network-access-update-2024-03
 """
 
 HTTP_CONFIG: HttpConfig | None = env("LANGGRAPH_HTTP", cast=_parse_json, default=None)
-STORE_CONFIG: dict | None = env("LANGGRAPH_STORE", cast=_parse_json, default=None)
+STORE_CONFIG: StoreConfig | None = env(
+    "LANGGRAPH_STORE", cast=_parse_json, default=None
+)
 CORS_ALLOW_ORIGINS = env("CORS_ALLOW_ORIGINS", cast=CommaSeparatedStrings, default="*")
 if HTTP_CONFIG and HTTP_CONFIG.get("cors"):
     CORS_CONFIG = HTTP_CONFIG["cors"]
@@ -132,6 +206,13 @@ FF_CRONS_ENABLED = env("FF_CRONS_ENABLED", cast=bool, default=True)
 # auth
 
 LANGGRAPH_AUTH_TYPE = env("LANGGRAPH_AUTH_TYPE", cast=str, default="noop")
+LANGGRAPH_POSTGRES_EXTENSIONS: Literal["standard", "lite"] = env(
+    "LANGGRAPH_POSTGRES_EXTENSIONS", cast=str, default="standard"
+)
+if LANGGRAPH_POSTGRES_EXTENSIONS not in ("standard", "lite"):
+    raise ValueError(
+        f"Unknown LANGGRAPH_POSTGRES_EXTENSIONS value: {LANGGRAPH_POSTGRES_EXTENSIONS}"
+    )
 LANGGRAPH_AUTH = env("LANGGRAPH_AUTH", cast=_parse_json, default=None)
 LANGSMITH_TENANT_ID = env("LANGSMITH_TENANT_ID", cast=str, default=None)
 LANGSMITH_AUTH_VERIFY_TENANT_ID = env(

langgraph_api/js/remote.py

@@ -1,8 +1,10 @@
 import asyncio
+import logging
 import os
 import shutil
 import ssl
 from collections.abc import AsyncIterator
+from contextlib import AbstractContextManager
 from typing import Any, Literal
 
 import certifi
@@ -605,60 +607,79 @@ async def run_remote_checkpointer():
     await server.serve()
 
 
+class DisableHttpxLoggingContextManager(AbstractContextManager):
+    """
+    Disable HTTP/1.1 200 OK logs spamming stdout.
+    """
+
+    filter: logging.Filter
+
+    def filter(self, record: logging.LogRecord) -> bool:
+        return "200 OK" not in record.getMessage()
+
+    def __enter__(self):
+        logging.getLogger("httpx").addFilter(self.filter)
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        logging.getLogger("httpx").removeFilter(self.filter)
+
+
 async def wait_until_js_ready():
-    async with (
-        httpx.AsyncClient(
-            base_url=f"http://localhost:{GRAPH_PORT}",
-            limits=httpx.Limits(max_connections=1),
-            transport=httpx.AsyncHTTPTransport(verify=SSL),
-        ) as graph_client,
-        httpx.AsyncClient(
-            base_url=f"http://localhost:{REMOTE_PORT}",
-            limits=httpx.Limits(max_connections=1),
-            transport=httpx.AsyncHTTPTransport(verify=SSL),
-        ) as checkpointer_client,
-    ):
-        attempt = 0
-        while not asyncio.current_task().cancelled():
+    with DisableHttpxLoggingContextManager():
+        async with (
+            httpx.AsyncClient(
+                base_url=f"http://localhost:{GRAPH_PORT}",
+                limits=httpx.Limits(max_connections=1),
+                transport=httpx.AsyncHTTPTransport(verify=SSL),
+            ) as graph_client,
+            httpx.AsyncClient(
+                base_url=f"http://localhost:{REMOTE_PORT}",
+                limits=httpx.Limits(max_connections=1),
+                transport=httpx.AsyncHTTPTransport(verify=SSL),
+            ) as checkpointer_client,
+        ):
+            attempt = 0
+            while not asyncio.current_task().cancelled():
+                try:
+                    res = await graph_client.get("/ok")
+                    res.raise_for_status()
+                    res = await checkpointer_client.get("/ok")
+                    res.raise_for_status()
+                    return
+                except httpx.HTTPError:
+                    if attempt > 240:
+                        raise
+                    else:
+                        attempt += 1
+                        await asyncio.sleep(0.5)
+
+
+async def js_healthcheck():
+    with DisableHttpxLoggingContextManager():
+        async with (
+            httpx.AsyncClient(
+                base_url=f"http://localhost:{GRAPH_PORT}",
+                limits=httpx.Limits(max_connections=1),
+                transport=httpx.AsyncHTTPTransport(verify=SSL),
+            ) as graph_client,
+            httpx.AsyncClient(
+                base_url=f"http://localhost:{REMOTE_PORT}",
+                limits=httpx.Limits(max_connections=1),
+                transport=httpx.AsyncHTTPTransport(verify=SSL),
+            ) as checkpointer_client,
+        ):
             try:
                 res = await graph_client.get("/ok")
                 res.raise_for_status()
                 res = await checkpointer_client.get("/ok")
                 res.raise_for_status()
-                return
-            except httpx.HTTPError:
-                if attempt > 240:
-                    raise
-                else:
-                    attempt += 1
-                    await asyncio.sleep(0.5)
-
-
-async def js_healthcheck():
-    async with (
-        httpx.AsyncClient(
-            base_url=f"http://localhost:{GRAPH_PORT}",
-            limits=httpx.Limits(max_connections=1),
-            transport=httpx.AsyncHTTPTransport(verify=SSL),
-        ) as graph_client,
-        httpx.AsyncClient(
-            base_url=f"http://localhost:{REMOTE_PORT}",
-            limits=httpx.Limits(max_connections=1),
-            transport=httpx.AsyncHTTPTransport(verify=SSL),
-        ) as checkpointer_client,
-    ):
-        try:
-            res = await graph_client.get("/ok")
-            res.raise_for_status()
-            res = await checkpointer_client.get("/ok")
-            res.raise_for_status()
-            return True
-        except httpx.HTTPError as exc:
-            logger.warning(
-                "JS healthcheck failed. Either the JS server is not running or the event loop is blocked by a CPU-intensive task.",
-                error=exc,
-            )
-            raise HTTPException(
-                status_code=500,
-                detail="JS healthcheck failed. Either the JS server is not running or the event loop is blocked by a CPU-intensive task.",
-            ) from exc
+                return True
+            except httpx.HTTPError as exc:
+                logger.warning(
+                    "JS healthcheck failed. Either the JS server is not running or the event loop is blocked by a CPU-intensive task.",
+                    error=exc,
+                )
+                raise HTTPException(
+                    status_code=500,
+                    detail="JS healthcheck failed. Either the JS server is not running or the event loop is blocked by a CPU-intensive task.",
+                ) from exc

langgraph_api/js/tests/api.test.mts

@@ -765,7 +765,7 @@ describe("runs", () => {
 
   it.concurrent(
     "human in the loop - no modification",
-    { retry: 0 },
+    { retry: 3 },
     async () => {
       const assistant = await client.assistants.create({ graphId: "agent" });
       const thread = await client.threads.create();

langgraph_api/lifespan.py

@@ -12,6 +12,7 @@ from langgraph_api.metadata import metadata_loop
 from langgraph_license.validation import get_license_status, plus_features_enabled
 from langgraph_storage.database import start_pool, stop_pool
 from langgraph_storage.queue import queue
+from langgraph_storage.store import Store
 
 
 @asynccontextmanager
@@ -44,6 +45,8 @@ async def lifespan(
                 and plus_features_enabled()
             ):
                 tg.create_task(cron_scheduler())
+            if config.STORE_CONFIG and config.STORE_CONFIG.get("ttl"):
+                tg.create_task(Store().start_ttl_sweeper())
             yield
     finally:
         await stop_remote_graphs()

langgraph_api/stream.py

@@ -19,15 +19,15 @@ from langgraph.errors import (
     InvalidUpdateError,
 )
 from langgraph.pregel.debug import CheckpointPayload, TaskResultPayload
-from langgraph.types import Command, Send
 from pydantic import ValidationError
 from pydantic.v1 import ValidationError as ValidationErrorLegacy
 
 from langgraph_api.asyncio import ValueEvent, wait_if_not_done
+from langgraph_api.command import map_cmd
 from langgraph_api.graph import get_graph
 from langgraph_api.js.base import BaseRemotePregel
 from langgraph_api.metadata import HOST, PLAN, incr_nodes
-from langgraph_api.schema import Run, RunCommand, StreamMode
+from langgraph_api.schema import Run, StreamMode
 from langgraph_api.serde import json_dumpb
 from langgraph_api.utils import AsyncConnectionProto
 from langgraph_storage.checkpoint import Checkpointer
@@ -70,32 +70,6 @@ def _preprocess_debug_checkpoint(payload: CheckpointPayload | None) -> dict[str,
     return payload
 
 
-def _map_cmd(cmd: RunCommand) -> Command:
-    goto = cmd.get("goto")
-    if goto is not None and not isinstance(goto, list):
-        goto = [cmd.get("goto")]
-
-    update = cmd.get("update")
-    if isinstance(update, tuple | list) and all(
-        isinstance(t, tuple | list) and len(t) == 2 and isinstance(t[0], str)
-        for t in update
-    ):
-        update = [tuple(t) for t in update]
-
-    return Command(
-        update=update,
-        goto=(
-            [
-                it if isinstance(it, str) else Send(it["node"], it["input"])
-                for it in goto
-            ]
-            if goto
-            else None
-        ),
-        resume=cmd.get("resume"),
-    )
-
-
 async def astream_state(
     stack: AsyncExitStack,
     conn: AsyncConnectionProto,
@@ -125,7 +99,7 @@ async def astream_state(
     )
     input = kwargs.pop("input")
     if cmd := kwargs.pop("command"):
-        input = _map_cmd(cmd)
+        input = map_cmd(cmd)
     stream_mode: list[StreamMode] = kwargs.pop("stream_mode")
     feedback_keys = kwargs.pop("feedback_keys", None)
     stream_modes_set: set[StreamMode] = set(stream_mode) - {"events"}

langgraph_api/validation.py

@@ -27,7 +27,18 @@ AssistantVersionChange = jsonschema_rs.validator_for(
     openapi["components"]["schemas"]["AssistantVersionChange"]
 )
 ThreadCreate = jsonschema_rs.validator_for(
-    openapi["components"]["schemas"]["ThreadCreate"]
+    {
+        **openapi["components"]["schemas"]["ThreadCreate"],
+        "components": {
+            "schemas": {
+                "ThreadSuperstepUpdate": openapi["components"]["schemas"][
+                    "ThreadSuperstepUpdate"
+                ],
+                "Command": openapi["components"]["schemas"]["Command"],
+                "Send": openapi["components"]["schemas"]["Send"],
+            }
+        },
+    }
 )
 ThreadPatch = jsonschema_rs.validator_for(
     openapi["components"]["schemas"]["ThreadPatch"]
@@ -42,6 +53,7 @@ ThreadStateUpdate = jsonschema_rs.validator_for(
         },
     }
 )
+
 ThreadStateCheckpointRequest = jsonschema_rs.validator_for(
     {
         **openapi["components"]["schemas"]["ThreadStateCheckpointRequest"],

{langgraph_api-0.0.30.dist-info → langgraph_api-0.0.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.30
+Version: 0.0.32
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos
@@ -12,11 +12,11 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: cryptography (>=43.0.3,<44.0.0)
 Requires-Dist: httpx (>=0.25.0)
-Requires-Dist: jsonschema-rs (>=0.20.0,<0.21.0)
+Requires-Dist: jsonschema-rs (>=0.20.0,<0.30)
 Requires-Dist: langchain-core (>=0.2.38,<0.4.0)
 Requires-Dist: langgraph (>=0.2.56,<0.4.0)
-Requires-Dist: langgraph-checkpoint (>=2.0.15,<3.0)
-Requires-Dist: langgraph-sdk (>=0.1.53,<0.2.0)
+Requires-Dist: langgraph-checkpoint (>=2.0.21,<3.0)
+Requires-Dist: langgraph-sdk (>=0.1.58,<0.2.0)
 Requires-Dist: langsmith (>=0.1.63,<0.4.0)
 Requires-Dist: orjson (>=3.9.7)
 Requires-Dist: pyjwt (>=2.9.0,<3.0.0)

{langgraph_api-0.0.30.dist-info → langgraph_api-0.0.32.dist-info}/RECORD

@@ -1,12 +1,12 @@
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
 langgraph_api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/api/__init__.py,sha256=B2kHOBhKjBRdNiWPljYuJFxP__wYPCvgtgTaWYKZGTQ,5129
-langgraph_api/api/assistants.py,sha256=6K5HTUjl7lQv0sVRe4RURPrLzTW-BblUjbSYHeN2PfQ,13992
+langgraph_api/api/assistants.py,sha256=nU6tnbgdr_6Utlq0A9nw2a6xxpUM_DNuCFI42_Kcs_o,14233
 langgraph_api/api/meta.py,sha256=ifJ_Ki0Qf2DYbmY6OKlqKhLGxbt55gm0lEqH1A0cJbw,2790
 langgraph_api/api/openapi.py,sha256=f9gfmWN2AMKNUpLCpSgZuw_aeOF9jCXPdOtFT5PaTWM,10960
-langgraph_api/api/runs.py,sha256=0eS92Wl2f3LCgImkYLCjt-HyuMZjB1gTx0a9JtV2EqE,16669
+langgraph_api/api/runs.py,sha256=_RWKtmjD89ALnTk56dwo2rJwEi2oghk2Tqp0l1aCcZg,16677
 langgraph_api/api/store.py,sha256=VzAJVOwO0IxosBB7km5TTf2rhlWGyPkVz_LpvbxetVY,5437
-langgraph_api/api/threads.py,sha256=taU61XPcCEhBPCYPZcMDsgVDwwWUWJs8p-PrXFXWY48,8661
+langgraph_api/api/threads.py,sha256=meaDGF3R2bYkx0KRa_le4Ka5nOSeqlMDWtLdnEhVYSY,8930
 langgraph_api/api/ui.py,sha256=LiOZVewKOPbKEykCm30hCEaOA7vuS_Ti5hB32EEy4vw,2082
 langgraph_api/asyncio.py,sha256=ipxOGL0CuKZeHw8895ojtfoBU2fj0iJOp48uhiLAmss,7786
 langgraph_api/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -17,8 +17,9 @@ langgraph_api/auth/langsmith/client.py,sha256=eKchvAom7hdkUXauD8vHNceBDDUijrFgdT
 langgraph_api/auth/middleware.py,sha256=jU8aDSIZHdzCGdifejRF7ndHkSjBtqIHcBwFIuUdHEA,1875
 langgraph_api/auth/noop.py,sha256=Bk6Nf3p8D_iMVy_OyfPlyiJp_aEwzL-sHrbxoXpCbac,586
 langgraph_api/auth/studio_user.py,sha256=FzFQRROKDlA9JjtBuwyZvk6Mbwno5M9RVYjDO6FU3F8,186
-langgraph_api/cli.py,sha256=YIiuS-2D0fyqLd8BzJBIUAM3mmAJPtLUdPEMS6Awj9o,12752
-langgraph_api/config.py,sha256=iIekoGAQL-v--hVrCzSd2StrelzuHqb1vbpJAoHgxkU,5809
+langgraph_api/cli.py,sha256=zzlGrMgiRXITuvztQ12y26ljOgEXk93Vq2quKYhbaD8,11750
+langgraph_api/command.py,sha256=3O9v3i0OPa96ARyJ_oJbLXkfO8rPgDhLCswgO9koTFA,768
+langgraph_api/config.py,sha256=XMJEVfPe1LrmVKGM5sisQs1eZOUDyU3JwscwJOm0q_k,8572
 langgraph_api/cron_scheduler.py,sha256=9yzbbGxzNgJdIg4ZT7yu2oTwT_wRuPxD1c2sbbd52xs,2630
 langgraph_api/errors.py,sha256=Bu_i5drgNTyJcLiyrwVE_6-XrSU50BHf9TDpttki9wQ,1690
 langgraph_api/graph.py,sha256=viFyQa8-BRRNRZqNNUzLB31cB4IypEMkMucQEJQYLWY,16556
@@ -30,7 +31,7 @@ langgraph_api/js/client.mts,sha256=2dptAX8fMowV9OC4DU4khjpZUgALBLVBTu3jTQbeUJY,2
 langgraph_api/js/errors.py,sha256=Cm1TKWlUCwZReDC5AQ6SgNIVGD27Qov2xcgHyf8-GXo,361
 langgraph_api/js/global.d.ts,sha256=cLJRZfYVGmgQ6o_xFevVNNTIi918ZUdxVRnpLVSjiAY,133
 langgraph_api/js/package.json,sha256=j6DMoVgwRqWqTwdd7R1f-kvmiTUAbO3HaUhM8K64lbE,1224
-langgraph_api/js/remote.py,sha256=jUUgdskUbDR_tuhZFrJo4OewbndQ76joczBv-mZHG9k,22637
+langgraph_api/js/remote.py,sha256=g0H2x3W7kejxswhgZfFaljzl4Y7agMCnO5BdNVj1rDY,23474
 langgraph_api/js/schema.py,sha256=7idnv7URlYUdSNMBXQcw7E4SxaPxCq_Oxwnlml8q5ik,408
 langgraph_api/js/src/graph.mts,sha256=mRyMUp03Fwd5DlmNIFl3RiUCQuJ5XwmFp1AfAeKDfVc,3169
 langgraph_api/js/src/hooks.mjs,sha256=XtktgmIHlls_DsknAuwib-z7TqCm0haRoTXvnkgzMuo,601
@@ -42,7 +43,7 @@ langgraph_api/js/src/utils/importMap.mts,sha256=pX4TGOyUpuuWF82kXcxcv3-8mgusRezO
 langgraph_api/js/src/utils/pythonSchemas.mts,sha256=98IW7Z_VP7L_CHNRMb3_MsiV3BgLE2JsWQY_PQcRR3o,685
 langgraph_api/js/src/utils/serde.mts,sha256=OuyyO9btvwWd55rU_H4x91dFEJiaPxL-lL9O6Zgo908,742
 langgraph_api/js/sse.py,sha256=lsfp4nyJyA1COmlKG9e2gJnTttf_HGCB5wyH8OZBER8,4105
-langgraph_api/js/tests/api.test.mts,sha256=7B7uIiUsi4X3tN5ji7loaU4PGkoJyHhTIbqPN2V-7kY,57709
+langgraph_api/js/tests/api.test.mts,sha256=RTJb9ZMX_IhqUPcShK61LZG9v_-xzdIZ0oEmOIjbJ4c,57709
 langgraph_api/js/tests/compose-postgres.yml,sha256=wV1Kws7WwUWVIudPkB--v58MOPL9hOcV0MUK-cvNrpA,1738
 langgraph_api/js/tests/graphs/.gitignore,sha256=26J8MarZNXh7snXD5eTpV3CPFTht5Znv8dtHYCLNfkw,12
 langgraph_api/js/tests/graphs/agent.css,sha256=QgcOC0W7IBsrg4pSqqpull-WTgtULZfx_lF_5ZxLdag,23
@@ -58,7 +59,7 @@ langgraph_api/js/tests/graphs/yarn.lock,sha256=i2AAIgXA3XBLM8-oU45wgUefCSG-Tne4g
 langgraph_api/js/tests/parser.test.mts,sha256=dEC8KTqKygeb1u39ZvpPqCT4HtfPD947nLmITt2buxA,27883
 langgraph_api/js/tests/utils.mts,sha256=2kTybJ3O7Yfe1q3ehDouqV54ibXkNzsPZ_wBZLJvY-4,421
 langgraph_api/js/yarn.lock,sha256=W89dVYZMThcec08lJMcYnvEEnQK7VM5cPglvwpIdRv0,82773
-langgraph_api/lifespan.py,sha256=tcW68-Vxh10y15NDn0TJG2qAbU9RwsrQmumT5xs2HBA,2046
+langgraph_api/lifespan.py,sha256=u5Fv9I4JqfLk30p5cNqkcEqeB2vTRMGSM5WlFHlE0mU,2219
 langgraph_api/logging.py,sha256=KB1ciduIWcMFfG0q9c5_SZemGrscht1RZXLZfeJSS00,3619
 langgraph_api/metadata.py,sha256=jPLNIRxHi7taZ0g60UdOEXenkvDwoYdI11tsmHenb28,3443
 langgraph_api/middleware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -74,9 +75,9 @@ langgraph_api/serde.py,sha256=VoJ7Z1IuqrQGXFzEP1qijAITtWCrmjtVqlCRuScjXJI,3533
 langgraph_api/server.py,sha256=CiNK327zTsEpoVGeJK1JOtZHvOBYRoz0CnBTZUmsC7c,4567
 langgraph_api/sse.py,sha256=2wNodCOP2eg7a9mpSu0S3FQ0CHk2BBV_vv0UtIgJIcc,4034
 langgraph_api/state.py,sha256=8jx4IoTCOjTJuwzuXJKKFwo1VseHjNnw_CCq4x1SW14,2284
-langgraph_api/stream.py,sha256=LKXxol0y9xEsrgnbsDwyeKIPBDE-75uVz01OcLI-_QE,11976
+langgraph_api/stream.py,sha256=AgDjVxE8Tf3_TAB5td36mVy7_-9OwKcZJpHRWJdVCSM,11281
 langgraph_api/utils.py,sha256=92mSti9GfGdMRRWyESKQW5yV-75Z9icGHnIrBYvdypU,3619
-langgraph_api/validation.py,sha256=McizHlz-Ez8Jhdbc79mbPSde7GIuf2Jlbjx2yv_l6dA,4475
+langgraph_api/validation.py,sha256=LnEdfgID2Z0HP75R_hqWxCn5d5ULGcDJ3r1xaDVv6-w,4845
 langgraph_api/webhook.py,sha256=1ncwO0rIZcj-Df9sxSnFEzd1gP1bfS4okeZQS8NSRoE,1382
 langgraph_api/worker.py,sha256=7yQfZBANt1kgJDOEs5B5c3Xy65lzNMmngVbBqLs-r5s,9802
 langgraph_license/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -86,15 +87,15 @@ langgraph_storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 langgraph_storage/checkpoint.py,sha256=V4t2GwYEJdPCHbhq_4Udhlv0TWKDzlMu_rlNPdTDc50,3589
 langgraph_storage/database.py,sha256=I0AgFeJ-NSTT34vxKxQBUf1z2syFP0S8QpKCqTixrzY,5652
 langgraph_storage/inmem_stream.py,sha256=8bxkILIuFpr7P7RQ37SQAxrpRKvmbHdRB_nbfFiomlk,3263
-langgraph_storage/ops.py,sha256=6qj3zh3T7KZco_vz9FW1cI82sbrefo6u2BRmrUDXvZI,69529
+langgraph_storage/ops.py,sha256=0DhtDQllBnV495DPrwhbR_bQUg7ZrGBe0FxSZcicz5g,72469
 langgraph_storage/queue.py,sha256=UDgsUTtUMfBSRDrQ8Onis-FJO4n7KTsX6sdpbY8Hs0A,5055
 langgraph_storage/retry.py,sha256=XmldOP4e_H5s264CagJRVnQMDFcEJR_dldVR1Hm5XvM,763
-langgraph_storage/store.py,sha256=SL7VYCO1wI3O1WP20A9jD3lsxXIN990XuuyVrrj6RVY,2855
+langgraph_storage/store.py,sha256=33-J5-Xvobb9ArSa-GezP5KtfXgzWkHUHPyjRYmdw-E,2985
 langgraph_storage/ttl_dict.py,sha256=FlpEY8EANeXWKo_G5nmIotPquABZGyIJyk6HD9u6vqY,1533
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
-openapi.json,sha256=8NhoZn3Dse6Ia0i7fXCB46sdGnEg5qAhov_8of-6m0o,125282
-langgraph_api-0.0.30.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.0.30.dist-info/METADATA,sha256=ihj0r5-3nmveP3ukNwnF4GY_1R4tGhKNTFuyr4Lwt4M,4029
-langgraph_api-0.0.30.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-langgraph_api-0.0.30.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.0.30.dist-info/RECORD,,
+openapi.json,sha256=6okfuwPAKB2Hi1tup-EFza49pmN_iD2Yt5JktMALcq4,127984
+langgraph_api-0.0.32.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.0.32.dist-info/METADATA,sha256=LrWNwNTUVge4rKtDDFwJyFKdE5WRTPCLPj8MjSHsDvc,4027
+langgraph_api-0.0.32.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+langgraph_api-0.0.32.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.0.32.dist-info/RECORD,,

langgraph_storage/ops.py

@@ -22,6 +22,7 @@ from starlette.exceptions import HTTPException
 
 from langgraph_api.asyncio import SimpleTaskGroup, ValueEvent, create_task
 from langgraph_api.auth.custom import handle_event
+from langgraph_api.command import map_cmd
 from langgraph_api.errors import UserInterrupt, UserRollback
 from langgraph_api.graph import get_graph
 from langgraph_api.schema import (
@@ -1081,6 +1082,78 @@ class Threads(Authenticated):
             else:
                 raise HTTPException(status_code=400, detail="Thread has no graph ID.")
 
+        @staticmethod
+        async def bulk(
+            conn: InMemConnectionProto,
+            *,
+            config: Config,
+            supersteps: Sequence[dict],
+            ctx: Auth.types.BaseAuthContext | None = None,
+        ) -> ThreadUpdateResponse:
+            """Update a thread with a batch of state updates."""
+
+            from langgraph.pregel.types import StateUpdate
+
+            thread_id = _ensure_uuid(config["configurable"]["thread_id"])
+            filters = await Threads.handle_event(
+                ctx,
+                "update",
+                Auth.types.ThreadsUpdate(thread_id=thread_id),
+            )
+
+            thread_iter = await Threads.get(conn, thread_id, ctx=ctx)
+            thread = await fetchone(
+                thread_iter, not_found_detail=f"Thread {thread_id} not found."
+            )
+
+            thread_config = thread["config"]
+            metadata = thread["metadata"]
+
+            if not thread:
+                raise HTTPException(status_code=404, detail="Thread not found")
+
+            if not _check_filter_match(metadata, filters):
+                raise HTTPException(status_code=403, detail="Forbidden")
+
+            if graph_id := metadata.get("graph_id"):
+                config["configurable"].setdefault("graph_id", graph_id)
+                config["configurable"].setdefault("checkpoint_ns", "")
+
+                async with get_graph(
+                    graph_id, thread_config, checkpointer=Checkpointer(conn)
+                ) as graph:
+                    next_config = await graph.abulk_update_state(
+                        config,
+                        [
+                            [
+                                StateUpdate(
+                                    map_cmd(update.get("command"))
+                                    if update.get("command")
+                                    else update.get("values"),
+                                    update.get("as_node"),
+                                )
+                                for update in superstep.get("updates", [])
+                            ]
+                            for superstep in supersteps
+                        ],
+                    )
+
+                    state = await Threads.State.get(
+                        conn, config, subgraphs=False, ctx=ctx
+                    )
+
+                    # update thread values
+                    for thread in conn.store["threads"]:
+                        if thread["thread_id"] == thread_id:
+                            thread["values"] = state.values
+                            break
+
+                    return ThreadUpdateResponse(
+                        checkpoint=next_config["configurable"],
+                    )
+            else:
+                raise HTTPException(status_code=400, detail="Thread has no graph ID")
+
         @staticmethod
         async def list(
             conn: InMemConnectionProto,
@@ -1905,7 +1978,8 @@ class Crons:
         conn: InMemConnectionProto,
         ctx: Auth.types.BaseAuthContext | None = None,
     ) -> AsyncIterator[Cron]:
-        raise NotImplementedError
+        yield
+        raise NotImplementedError("The in-mem server does not implement Crons.")
 
     @staticmethod
     async def set_next_run_date(

langgraph_storage/store.py

@@ -1,3 +1,4 @@
+import asyncio
 import os
 import threading
 from collections import defaultdict
@@ -42,6 +43,9 @@ class DiskBackedInMemStore(InMemoryStore):
                 f"Unexpected error loading store {which} from {container.filename}: {str(e)}"
             ) from e
 
+    async def start_ttl_sweeper(self) -> asyncio.Task[None]:
+        return asyncio.create_task(asyncio.sleep(0))
+
     def close(self) -> None:
         self._data.close()
         self._vectors.close()

openapi.json

@@ -805,6 +805,58 @@
         }
       }
     },
+    "/threads/state/bulk": {
+      "post": {
+        "tags": [
+          "Threads"
+        ],
+        "summary": "Bulk Update Thread State",
+        "description": "Create a new thread from a batch of state updates.",
+        "operationId": "bulk_update_thread_state_post",
+        "requestBody": {
+          "content": {
+            "application/json": {
+              "schema": {
+                "$ref": "#/components/schemas/ThreadStateBulkUpdate"
+              }
+            }
+          },
+          "required": true
+        },
+        "responses": {
+          "200": {
+            "description": "Success",
+            "content": {
+              "application/json": {
+                "schema": {
+                  "$ref": "#/components/schemas/Thread"
+                }
+              }
+            }
+          },
+          "409": {
+            "description": "Conflict",
+            "content": {
+              "application/json": {
+                "schema": {
+                  "$ref": "#/components/schemas/ErrorResponse"
+                }
+              }
+            }
+          },
+          "422": {
+            "description": "Validation Error",
+            "content": {
+              "application/json": {
+                "schema": {
+                  "$ref": "#/components/schemas/ErrorResponse"
+                }
+              }
+            }
+          }
+        }
+      }
+    },
     "/threads/{thread_id}/state": {
       "get": {
         "tags": [
@@ -3906,6 +3958,19 @@
             "title": "If Exists",
             "description": "How to handle duplicate creation. Must be either 'raise' (raise error if duplicate), or 'do_nothing' (return existing thread).",
             "default": "raise"
+          },
+          "supersteps": {
+            "type": "array",
+            "items": {
+              "type": "object",
+              "properties": {
+                "updates": {
+                  "type": "array",
+                  "items": { "$ref": "#/components/schemas/ThreadSuperstepUpdate" }
+                }
+              },
+              "required": ["updates"]
+            }
           }
         },
         "type": "object",
@@ -4094,6 +4159,43 @@
         "title": "ThreadStateUpdate",
         "description": "Payload for updating the state of a thread."
       },
+      "ThreadSuperstepUpdate": {
+        "properties": {
+          "values": {
+            "anyOf": [
+              {
+                "type": "array",
+                "items": {
+                  "type": "object"
+                }
+              },
+              {
+                "type": "object"
+              },
+              {
+                "type": "null"
+              }
+            ]
+          },
+          "command": {
+            "anyOf": [
+              {
+                "$ref": "#/components/schemas/Command"
+              },
+              {
+                "type": "null"
+              }
+            ],
+            "description": "The command associated with the update."
+          },
+          "as_node": {
+            "type": "string",
+            "description": "Update the state as if this node had just executed."
+          }
+        },
+        "required": ["as_node"],
+        "type": "object"
+      },
       "ThreadStateUpdateResponse": {
         "properties": {
           "checkpoint": {