langgraph-api 0.2.27__tar.gz → 0.2.28__tar.gz

langgraph_api-0.2.28/.dockerignore

@@ -0,0 +1,14 @@
+.venv
+.envrc
+tests
+compose.yml
+.mypy_cache
+.pytest_cache
+.ruff_cache
+poetry.lock
+uv.lock
+Dockerfile
+Dockerfile.node
+*.sock
+node_modules
+langgraph_api/js/node_modules

langgraph_api-0.2.28/.gitignore

@@ -0,0 +1,5 @@
+.envrc
+.ipynb_checkpoints
+.langgraph-data
+*.sock
+langgraph_api.duckdb*

langgraph_api-0.2.28/Makefile

@@ -0,0 +1,114 @@
+.PHONY: build release lint format test test_watch start start-inmem start-inmem-license-oss start check-version
+
+# lint commands
+
+lint:
+	uv run ruff check .
+	uv run ruff format . --diff
+
+format:
+	uv run ruff check --fix .
+	uv run ruff format .
+
+# test commands
+
+TEST ?= "tests/integration_tests/"
+AUTH_TEST ?= "tests/integration_tests/test_custom_auth.py"
+LANGGRAPH_HTTP ?= {"disable_mcp": false}
+LANGGRAPH_AES_KEY ?= '1234567890123456'
+
+ifeq ($(LANGGRAPH_HTTP),fastapi)
+	HTTP_CONFIG := {"app": "./tests/graphs/my_router.py:app", "disable_mcp": false, "mount_prefix": "/my-cool/api"}
+else
+	HTTP_CONFIG := $(LANGGRAPH_HTTP)
+endif
+
+LANGGRAPH_STORE ?= ""
+ifeq ($(LANGGRAPH_STORE),custom)
+	STORE_CONFIG := {"path": "./tests/graphs/custom_store.py:generate_store"}
+else
+	STORE_CONFIG := {"index": {"dims": 500, "embed": "./tests/graphs/test_utils/embeddings.py:embeddings"}}
+endif
+
+REVISION ?= $(shell git rev-parse --short HEAD)
+
+test-license-oss:
+	LANGGRAPH_RUNTIME_EDITION=inmem LANGGRAPH_HTTP='$(HTTP_CONFIG)' LANGGRAPH_STORE='$(STORE_CONFIG)' REDIS_URI=_FAKE DATABASE_URI=:memory: MIGRATIONS_PATH=__inmem__ uv run pytest -v $(TEST)
+
+test-watch-oss:
+	LANGGRAPH_RUNTIME_EDITION=inmem LANGGRAPH_HTTP='$(HTTP_CONFIG)' REDIS_URI=_FAKE DATABASE_URI=:memory: MIGRATIONS_PATH=__inmem__ uv run ptw . -- -x -vv --ff --capture=no $(TEST)
+
+test: test-license-oss
+test-watch: test-watch-oss
+
+test-auth:
+	LANGGRAPH_RUNTIME_EDITION=inmem LANGGRAPH_AUTH='{"path": "./tests/graphs/fastapi_jwt_auth.py:get_current_active_user"}' REDIS_URI=_FAKE DATABASE_URI=:memory: MIGRATIONS_PATH=__inmem__ uv run pytest -v $(AUTH_TEST)
+
+test-auth-watch:
+	LANGGRAPH_RUNTIME_EDITION=inmem LANGGRAPH_AUTH='{"path": "./tests/graphs/fastapi_jwt_auth.py:get_current_active_user"}' REDIS_URI=_FAKE DATABASE_URI=:memory: MIGRATIONS_PATH=__inmem__ uv run ptw . -- -x -vv --ff --capture=no $(AUTH_TEST)
+
+# dev commands
+
+start:
+	LANGGRAPH_HTTP='$(HTTP_CONFIG)' \
+	LANGGRAPH_RUNTIME_EDITION=inmem \
+	LANGGRAPH_AES_KEY='$(LANGGRAPH_AES_KEY)' \
+	N_JOBS_PER_WORKER=2 \
+	LANGGRAPH_CONFIG='{"agent": {"configurable": {"model_name": "openai"}}}' \
+	LANGSERVE_GRAPHS='{"agent": "./tests/graphs/agent.py:graph", "single_node": "./tests/graphs/single_node.py:graph", "other": "./tests/graphs/other.py:make_graph", "weather": "./tests/graphs/weather.py:mk_weather_graph", "searchy": "./tests/graphs/searchy.py:graph", "agent_simple": "./tests/graphs/agent_simple.py:graph"}' \
+	LANGGRAPH_STORE='$(STORE_CONFIG)' \
+	LANGGRAPH_CONFIG='{"agent": {"configurable": {"model_name": "openai"}}}' \
+	LANGSMITH_LANGGRAPH_API_VARIANT=test \
+	REDIS_URI=fake \
+	DATABASE_URI=:memory: \
+	MIGRATIONS_PATH=__inmem \
+	uv run uvicorn \
+		"langgraph_api.server:app" \
+		--reload \
+		--port 9123 \
+		--reload-dir langgraph_api \
+		--no-access-log
+
+
+start-license-oss: start
+
+
+start-auth-jwt:
+	LANGGRAPH_RUNTIME_EDITION=inmem LANGGRAPH_HTTP='$(HTTP_CONFIG)' \
+	LANGGRAPH_AES_KEY='$(LANGGRAPH_AES_KEY)' \
+	N_JOBS_PER_WORKER=2 \
+	LANGSERVE_GRAPHS='{"agent": "./tests/graphs/agent.py:graph", "other": "./tests/graphs/other.py:make_graph", "weather": "./tests/graphs/weather.py:mk_weather_graph", "searchy": "./tests/graphs/searchy.py:graph", "agent_simple": "./tests/graphs/agent_simple.py:graph"}' \
+	LANGGRAPH_STORE='$(STORE_CONFIG)' \
+	LANGGRAPH_AUTH='{"path": "tests/graphs/jwt_auth.py:auth"}' \
+	LANGSMITH_LANGGRAPH_API_VARIANT=test \
+	REDIS_URI=fake \
+	DATABASE_URI=:memory: \
+	MIGRATIONS_PATH=__inmem \
+	uv run uvicorn \
+		"langgraph_api.server:app" \
+		--reload \
+		--port 9123 \
+		--reload-dir langgraph_api \
+		--no-access-log
+
+start-auth-fastapi-jwt:
+	LANGGRAPH_RUNTIME_EDITION=inmem LANGGRAPH_HTTP='$(HTTP_CONFIG)' \
+	N_JOBS_PER_WORKER=2 \
+	LANGSERVE_GRAPHS='{"agent": "./tests/graphs/agent.py:graph", "other": "./tests/graphs/other.py:make_graph", "weather": "./tests/graphs/weather.py:mk_weather_graph", "searchy": "./tests/graphs/searchy.py:graph", "agent_simple": "./tests/graphs/agent_simple.py:graph"}' \
+	LANGGRAPH_STORE='$(STORE_CONFIG)' \
+	LANGGRAPH_AUTH='{"path": "./tests/graphs/fastapi_jwt_auth.py:auth"}' \
+	LANGSMITH_LANGGRAPH_API_VARIANT=test \
+	REDIS_URI=fake \
+	DATABASE_URI=:memory: \
+	MIGRATIONS_PATH=__inmem \
+	uv run uvicorn \
+		"langgraph_api.server:app" \
+		--reload \
+		--port 9123 \
+		--reload-dir langgraph_api \
+		--no-access-log
+
+VERSION_KIND ?= patch
+
+bump-version:
+	uv run --with hatch hatch version $(VERSION_KIND)

{langgraph_api-0.2.27 → langgraph_api-0.2.28}/PKG-INFO

@@ -1,35 +1,29 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.2.27
-Summary: 
+Version: 0.2.28
+Author-email: Nuno Campos <nuno@langchain.dev>, Will Fu-Hinthorn <will@langchain.dev>
 License: Elastic-2.0
-Author: Nuno Campos
-Author-email: nuno@langchain.dev
+License-File: LICENSE
 Requires-Python: >=3.11
-Classifier: License :: Other/Proprietary License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: cloudpickle (>=3.0.0,<4.0.0)
-Requires-Dist: cryptography (>=42.0.0,<45.0)
-Requires-Dist: httpx (>=0.25.0)
-Requires-Dist: jsonschema-rs (>=0.20.0,<0.30)
-Requires-Dist: langchain-core (>=0.2.38) ; python_version < "4.0"
-Requires-Dist: langgraph (>=0.3.27) ; python_version < "4.0"
-Requires-Dist: langgraph-checkpoint (>=2.0.23) ; python_version < "4.0"
-Requires-Dist: langgraph-runtime-inmem (>=0.0.9,<0.1)
-Requires-Dist: langgraph-sdk (>=0.1.66,<0.2.0) ; python_version < "4.0"
-Requires-Dist: langsmith (>=0.1.63)
-Requires-Dist: orjson (>=3.9.7)
-Requires-Dist: pyjwt (>=2.9.0,<3.0.0)
-Requires-Dist: sse-starlette (>=2.1.0,<2.2.0)
-Requires-Dist: starlette (>=0.38.6)
-Requires-Dist: structlog (>=24.1.0,<26)
-Requires-Dist: tenacity (>=8.0.0)
-Requires-Dist: truststore (>=0.1)
-Requires-Dist: uvicorn (>=0.26.0)
-Requires-Dist: watchfiles (>=0.13)
+Requires-Dist: cloudpickle>=3.0.0
+Requires-Dist: cryptography<45.0,>=42.0.0
+Requires-Dist: httpx>=0.25.0
+Requires-Dist: jsonschema-rs<0.30,>=0.20.0
+Requires-Dist: langchain-core>=0.2.38
+Requires-Dist: langgraph-checkpoint>=2.0.23
+Requires-Dist: langgraph-runtime-inmem<0.2,>=0.1.0
+Requires-Dist: langgraph-sdk>=0.1.66
+Requires-Dist: langgraph>=0.3.27
+Requires-Dist: langsmith>=0.1.112
+Requires-Dist: orjson>=3.9.7
+Requires-Dist: pyjwt>=2.9.0
+Requires-Dist: sse-starlette<2.2.0,>=2.1.0
+Requires-Dist: starlette>=0.38.6
+Requires-Dist: structlog<26,>=24.1.0
+Requires-Dist: tenacity>=8.0.0
+Requires-Dist: truststore>=0.1
+Requires-Dist: uvicorn>=0.26.0
+Requires-Dist: watchfiles>=0.13
 Description-Content-Type: text/markdown
 
 # LangGraph API
@@ -132,4 +126,4 @@ Options:
 
 ## License
 
-This project is licensed under the Elastic License 2.0 - see the [LICENSE](./LICENSE) file for details.
+This project is licensed under the Elastic License 2.0 - see the [LICENSE](./LICENSE) file for details.

langgraph_api-0.2.28/benchmark/weather.js

@@ -0,0 +1,75 @@
+import http from "k6/http";
+import { check } from "k6";
+
+export const options = {
+  vus: 100,
+  duration: "30s",
+};
+
+const ORIGIN = "http://localhost:9123";
+
+export function setup() {
+  const assistant = http.post(
+    `${ORIGIN}/assistants`,
+    JSON.stringify({ graph_id: "weather" }),
+    { headers: { "Content-Type": "application/json" } }
+  );
+
+  check(assistant, { "assistant status is 200": (r) => r.status === 200 });
+
+  return assistant.json();
+}
+
+export default function main(assistant) {
+  const thread = http.post(`${ORIGIN}/threads`, JSON.stringify({}), {
+    params: { headers: { "Content-Type": "application/json" } },
+  });
+
+  check(thread, { "thread status is 200": (r) => r.status === 200 });
+
+  const wait = http.post(
+    `${ORIGIN}/threads/${thread.json().thread_id}/runs/wait`,
+    JSON.stringify({
+      input: {
+        messages: [{ role: "human", content: "SF", id: "initial-message" }],
+      },
+      assistant_id: assistant.assistant_id,
+    }),
+    { headers: { "Content-Type": "application/json" } }
+  );
+
+  check(wait, {
+    "interrupted run status is 200": (r) => r.status === 200,
+    "interrupted run has values": (r) => {
+      const json = r.json();
+      if (json.route !== "weather") return false;
+      if (json.messages.length !== 1) return false;
+      if (json.messages[0].id !== "initial-message") return false;
+      if (json.messages[0].content !== "SF") return false;
+      return true;
+    },
+  });
+
+  const continueWait = http.post(
+    `${ORIGIN}/threads/${thread.json().thread_id}/runs/wait`,
+    JSON.stringify({
+      input: null,
+      assistant_id: assistant.assistant_id,
+    }),
+    { headers: { "Content-Type": "application/json" } }
+  );
+
+  check(continueWait, {
+    "run status is 200": (r) => r.status === 200,
+    "run has values": (r) => {
+      const json = r.json();
+      if (json.route !== "weather") return false;
+      if (json.messages.length !== 2) return false;
+      if (json.messages[0].id !== "initial-message") return false;
+      if (json.messages[0].content !== "SF") return false;
+      if (json.messages[1].content !== "It's sunny in San Francisco!")
+        return false;
+      return true;
+    },
+  });
+}

langgraph_api-0.2.28/constraints.txt

@@ -0,0 +1,19 @@
+# When editing this file, also update docs pages:
+# https://github.com/langchain-ai/langgraph/blob/main/docs/docs/cloud/deployment/setup.md
+# https://github.com/langchain-ai/langgraph/blob/main/docs/docs/cloud/deployment/setup_pyproject.md
+langgraph>=0.3.27
+langgraph-sdk>=0.1.66
+langgraph-checkpoint>=2.0.23
+langchain-core>=0.2.43
+langsmith>=0.1.112
+orjson>=3.9.7,<3.10.17
+httpx>=0.25.0
+tenacity>=8.0.0
+uvicorn>=0.26.0
+sse-starlette>=2.1.0,<2.2.0
+uvloop>=0.18.0
+httptools>=0.5.0
+jsonschema-rs>=0.20.0
+structlog>=24.1.0
+cloudpickle>=3.0.0
+truststore>=0.1

langgraph_api-0.2.28/forbidden.txt

@@ -0,0 +1,2 @@
+# Block user overrides of the base api.
+langgraph-api>=0.1.0

langgraph_api-0.2.28/healthcheck.py

@@ -0,0 +1,18 @@
+import json
+import os
+import urllib.request
+
+prefix = ""
+# Override prefix if it's set in the http config
+if (http := os.environ.get("LANGGRAPH_HTTP")) and (
+    mount_prefix := json.loads(http).get("mount_prefix")
+):
+    prefix = mount_prefix
+# Override that
+if os.environ.get("MOUNT_PREFIX"):
+    prefix = os.environ["MOUNT_PREFIX"]
+
+with urllib.request.urlopen(
+    f"http://localhost:{os.environ['PORT']}{prefix}/ok"
+) as response:
+    assert response.status == 200

langgraph_api-0.2.28/langgraph_api/__init__.py

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

{langgraph_api-0.2.27 → langgraph_api-0.2.28}/langgraph_api/api/assistants.py

@@ -9,6 +9,7 @@ from starlette.exceptions import HTTPException
 from starlette.responses import Response
 from starlette.routing import BaseRoute
 
+from langgraph_api import store as api_store
 from langgraph_api.graph import get_assistant_id, get_graph
 from langgraph_api.js.base import BaseRemotePregel
 from langgraph_api.route import ApiRequest, ApiResponse, ApiRoute
@@ -25,7 +26,6 @@ from langgraph_runtime.checkpoint import Checkpointer
 from langgraph_runtime.database import connect
 from langgraph_runtime.ops import Assistants
 from langgraph_runtime.retry import retry_db
-from langgraph_runtime.store import Store
 
 logger = structlog.stdlib.get_logger(__name__)
 
@@ -194,7 +194,7 @@ async def get_assistant_graph(
             assistant["graph_id"],
             config,
             checkpointer=Checkpointer(conn),
-            store=Store(),
+            store=(await api_store.get_store()),
         ) as graph:
             xray: bool | int = False
             xray_query = request.query_params.get("xray")
@@ -240,7 +240,7 @@ async def get_assistant_subgraphs(
             assistant["graph_id"],
             config,
             checkpointer=Checkpointer(conn),
-            store=Store(),
+            store=(await api_store.get_store()),
         ) as graph:
             namespace = request.path_params.get("namespace")
 
@@ -286,7 +286,7 @@ async def get_assistant_schemas(
             assistant["graph_id"],
             config,
             checkpointer=Checkpointer(conn),
-            store=Store(),
+            store=(await api_store.get_store()),
         ) as graph:
             if isinstance(graph, BaseRemotePregel):
                 schemas = await graph.fetch_state_schema()

{langgraph_api-0.2.27 → langgraph_api-0.2.28}/langgraph_api/api/store.py

@@ -6,6 +6,7 @@ from starlette.routing import BaseRoute
 
 from langgraph_api.auth.custom import handle_event as _handle_event
 from langgraph_api.route import ApiRequest, ApiResponse, ApiRoute
+from langgraph_api.store import get_store
 from langgraph_api.utils import get_auth_ctx
 from langgraph_api.validation import (
     StoreDeleteRequest,
@@ -14,7 +15,6 @@ from langgraph_api.validation import (
     StoreSearchRequest,
 )
 from langgraph_runtime.retry import retry_db
-from langgraph_runtime.store import Store
 
 
 def _validate_namespace(namespace: tuple[str, ...]) -> Response | None:
@@ -57,7 +57,9 @@ async def put_item(request: ApiRequest):
         "value": payload["value"],
     }
     await handle_event("put", handler_payload)
-    await Store().aput(namespace, handler_payload["key"], handler_payload["value"])
+    await (await get_store()).aput(
+        namespace, handler_payload["key"], handler_payload["value"]
+    )
     return Response(status_code=204)
 
 
@@ -75,7 +77,7 @@ async def get_item(request: ApiRequest):
         "key": key,
     }
     await handle_event("get", handler_payload)
-    result = await Store().aget(namespace, key)
+    result = await (await get_store()).aget(namespace, key)
     return ApiResponse(result.dict() if result is not None else None)
 
 
@@ -91,7 +93,9 @@ async def delete_item(request: ApiRequest):
         "key": payload["key"],
     }
     await handle_event("delete", handler_payload)
-    await Store().adelete(handler_payload["namespace"], handler_payload["key"])
+    await (await get_store()).adelete(
+        handler_payload["namespace"], handler_payload["key"]
+    )
     return Response(status_code=204)
 
 
@@ -114,7 +118,7 @@ async def search_items(request: ApiRequest):
         "query": query,
     }
     await handle_event("search", handler_payload)
-    items = await Store().asearch(
+    items = await (await get_store()).asearch(
         handler_payload["namespace"],
         filter=handler_payload["filter"],
         limit=handler_payload["limit"],
@@ -145,7 +149,7 @@ async def list_namespaces(request: ApiRequest):
         "offset": offset,
     }
     await handle_event("list_namespaces", handler_payload)
-    result = await Store().alist_namespaces(
+    result = await (await get_store()).alist_namespaces(
         prefix=handler_payload["namespace"],
         suffix=handler_payload["suffix"],
         max_depth=handler_payload["max_depth"],

{langgraph_api-0.2.27 → langgraph_api-0.2.28}/langgraph_api/config.py

@@ -109,6 +109,7 @@ class TTLConfig(TypedDict, total=False):
 
 
 class StoreConfig(TypedDict, total=False):
+    path: str
     index: IndexConfig
     ttl: TTLConfig
 

{langgraph_api-0.2.27 → langgraph_api-0.2.28}/langgraph_api/graph.py

@@ -3,7 +3,6 @@ import functools
 import glob
 import importlib.util
 import inspect
-import json
 import os
 import sys
 import warnings
@@ -14,6 +13,7 @@ from random import choice
 from typing import TYPE_CHECKING, Any, NamedTuple
 from uuid import UUID, uuid5
 
+import orjson
 import structlog
 from langchain_core.runnables.config import run_in_executor, var_child_runnable_config
 from langgraph.checkpoint.base import BaseCheckpointSaver
@@ -60,6 +60,12 @@ async def register_graph(
     if callable(graph):
         FACTORY_ACCEPTS_CONFIG[graph_id] = len(inspect.signature(graph).parameters) > 0
     async with connect() as conn:
+        graph_name = getattr(graph, "name", None) if isinstance(graph, Pregel) else None
+        assistant_name = (
+            graph_name
+            if graph_name is not None and graph_name != "LangGraph"
+            else graph_id
+        )
         await Assistants.put(
             conn,
             str(uuid5(NAMESPACE_GRAPH, graph_id)),
@@ -67,7 +73,7 @@ async def register_graph(
             metadata={"created_by": "system"},
             config=config or {},
             if_exists="do_nothing",
-            name=graph_id,
+            name=assistant_name,
             description=description,
         )
 
@@ -200,10 +206,19 @@ def _load_graph_config_from_env() -> dict | None:
     config_str = os.getenv("LANGGRAPH_CONFIG")
     if not config_str:
         return None
+    try:
+        config_per_id = orjson.loads(config_str)
+    except orjson.JSONDecodeError as e:
+        raise ValueError(
+            "Provided environment variable LANGGRAPH_CONFIG must be a valid JSON object"
+            f"\nFound: {config_str}"
+        ) from e
 
-    config_per_id = json.loads(config_str)
     if not isinstance(config_per_id, dict):
-        raise ValueError("LANGGRAPH_CONFIG must be a JSON object")
+        raise ValueError(
+            "Provided environment variable LANGGRAPH_CONFIG must be a JSON object"
+            f"\nFound: {config_str}"
+        )
 
     return config_per_id
 
@@ -218,7 +233,15 @@ async def collect_graphs_from_env(register: bool = False) -> None:
         specs = []
         # graphs-config can be either a mapping from graph id to path where the graph
         # is defined or graph id to a dictionary containing information about the graph.
-        graphs_config = json.loads(paths_str)
+        try:
+            graphs_config = orjson.loads(paths_str)
+        except orjson.JSONDecodeError as e:
+            raise ValueError(
+                "LANGSERVE_GRAPHS must be a valid JSON object."
+                f"\nFound: {paths_str}"
+                "\n The LANGSERVE_GRAPHS environment variable is typically set"
+                'from the "graphs" field in your configuration (langgraph.json) file.'
+            ) from e
 
         for key, value in graphs_config.items():
             if isinstance(value, dict) and "path" in value:

{langgraph_api-0.2.27 → langgraph_api-0.2.28}/langgraph_api/js/remote.py

@@ -39,6 +39,7 @@ from starlette.exceptions import HTTPException
 from starlette.requests import HTTPConnection, Request
 from starlette.routing import Route
 
+from langgraph_api import store as api_store
 from langgraph_api.auth.custom import DotDict, ProxyUser
 from langgraph_api.config import LANGGRAPH_AUTH_TYPE
 from langgraph_api.js.base import BaseRemotePregel
@@ -70,6 +71,12 @@ _client = httpx.AsyncClient(
 )
 
 
+def _snapshot_defaults():
+    if not hasattr(StateSnapshot, "interrupts"):
+        return {}
+    return {"interrupts": tuple()}
+
+
 def default_command(obj):
     if isinstance(obj, Send):
         return {"node": obj.node, "args": obj.arg}
@@ -251,7 +258,7 @@ class RemotePregel(BaseRemotePregel):
             item.get("parentConfig"),
             _convert_tasks(item.get("tasks", [])),
             # TODO: add handling of interrupts when multiple resumes land in JS
-            tuple(),
+            **_snapshot_defaults(),
         )
 
     async def aget_state(
@@ -473,10 +480,8 @@ def _get_passthrough_checkpointer(conn: AsyncConnectionProto):
     return checkpointer
 
 
-def _get_passthrough_store():
-    from langgraph_runtime.store import Store
-
-    return Store()
+async def _get_passthrough_store():
+    return await api_store.get_store()
 
 
 # Setup a HTTP server on top of CHECKPOINTER_SOCKET unix socket
@@ -574,7 +579,7 @@ async def run_remote_checkpointer():
             else:
                 raise ValueError(f"Unknown operation type: {op}")
 
-        store = _get_passthrough_store()
+        store = await _get_passthrough_store()
         results = await store.abatch(processed_operations)
 
         # Handle potentially undefined or non-dict results
@@ -613,7 +618,7 @@ async def run_remote_checkpointer():
 
         namespaces = namespaces_str.split(".")
 
-        store = _get_passthrough_store()
+        store = await _get_passthrough_store()
         result = await store.aget(namespaces, key)
 
         return result
@@ -626,7 +631,7 @@ async def run_remote_checkpointer():
         value = payload["value"]
         index = payload.get("index")
 
-        store = _get_passthrough_store()
+        store = await _get_passthrough_store()
         await store.aput(namespace, key, value, index=index)
 
         return {"success": True}
@@ -639,7 +644,7 @@ async def run_remote_checkpointer():
         offset = payload.get("offset", 0)
         query = payload.get("query")
 
-        store = _get_passthrough_store()
+        store = await _get_passthrough_store()
         result = await store.asearch(
             namespace_prefix, filter=filter, limit=limit, offset=offset, query=query
         )
@@ -652,7 +657,7 @@ async def run_remote_checkpointer():
         namespace = tuple(payload["namespace"])
         key = payload["key"]
 
-        store = _get_passthrough_store()
+        store = await _get_passthrough_store()
         await store.adelete(namespace, key)
 
         return {"success": True}
@@ -665,7 +670,7 @@ async def run_remote_checkpointer():
         limit = payload.get("limit", 100)
         offset = payload.get("offset", 0)
 
-        store = _get_passthrough_store()
+        store = await _get_passthrough_store()
         result = await store.alist_namespaces(
             prefix=prefix,
             suffix=suffix,