langgraph-api 0.12.0.dev13__py3-none-any.whl → 0.12.0.dev15__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.12.0.dev13"
+__version__ = "0.12.0.dev15"

langgraph_api/asyncio.py

@@ -103,6 +103,8 @@ async def wait_if_not_done(coro: Coroutine[Any, Any, T], done: ValueEvent) -> T:
                     await logger.ainfo("Done awaiting.", task=str(fut))
                 if e.args and isinstance(e.args[0], Exception):
                     raise e.args[0] from None
+                if isinstance(done._value, Exception):
+                    raise done._value from None
                 raise
     except ExceptionGroup as e:
         raise e.exceptions[0] from None

langgraph_api/config/__init__.py

@@ -243,6 +243,8 @@ WEBHOOKS_ENABLED = HTTP_CONFIG and HTTP_CONFIG.get("disable_webhooks")
 STORE_CONFIG = env(
     "LANGGRAPH_STORE", cast=_parse.parse_schema(StoreConfig), default=None
 )
+LANGGRAPH_STORE_BACKEND = env("LANGGRAPH_STORE_BACKEND", cast=str, default="python")
+USE_GRPC_STORE = LANGGRAPH_STORE_BACKEND == "grpc"
 
 
 def _validate_mount_prefix(mount_prefix: str | None) -> str | None:
@@ -668,6 +670,7 @@ __all__ = [
     "LANGGRAPH_METRICS_ENDPOINT",
     "LANGGRAPH_METRICS_EXPORT_INTERVAL_MS",
     "LANGGRAPH_POSTGRES_EXTENSIONS",
+    "LANGGRAPH_STORE_BACKEND",
     "LANGSMITH_API_KEY",
     "LANGSMITH_AUTH_ENDPOINT",
     "LANGSMITH_AUTH_VERIFY_TENANT_ID",
@@ -715,6 +718,7 @@ __all__ = [
     "USES_STORE_TTL",
     "USES_THREAD_TTL",
     "USE_CUSTOM_CHECKPOINTER",
+    "USE_GRPC_STORE",
     "AuthConfig",
     "CheckpointerConfig",
     "CorsConfig",

langgraph_api/grpc/ops/__init__.py

@@ -39,6 +39,7 @@ GRPC_STATUS_TO_HTTP_STATUS = {
     StatusCode.INVALID_ARGUMENT: HTTPStatus.UNPROCESSABLE_ENTITY,
     StatusCode.PERMISSION_DENIED: HTTPStatus.FORBIDDEN,
     StatusCode.UNAUTHENTICATED: HTTPStatus.UNAUTHORIZED,
+    StatusCode.RESOURCE_EXHAUSTED: HTTPStatus.TOO_MANY_REQUESTS,
 }
 
 
@@ -187,19 +188,33 @@ def _handle_grpc_error(error: AioRpcError) -> None:
     Always return detail as a string here.
     """
     error_details = error.details()
+    retry_after_seconds = None
     if error_details is not None:
         try:
             details = orjson.loads(error_details)
-            error_details = orjson.dumps(details.get("message", "")).decode()
+            retry_after_seconds = details.get("retry_after_seconds")
+            message = details.get("message", "")
+            # A string message is the detail verbatim; only re-encode structured
+            # (non-string) messages as JSON.
+            error_details = (
+                message if isinstance(message, str) else orjson.dumps(message).decode()
+            )
         except orjson.JSONDecodeError:
             # error details is not json, so just retun it as is
             pass
 
+    status_code = GRPC_STATUS_TO_HTTP_STATUS.get(
+        error.code(), HTTPStatus.INTERNAL_SERVER_ERROR
+    )
+    headers = None
+    if status_code == HTTPStatus.TOO_MANY_REQUESTS and retry_after_seconds:
+        # The Go core carries Retry-After as a structured integer (seconds).
+        headers = {"Retry-After": str(int(retry_after_seconds))}
+
     raise HTTPException(
-        status_code=GRPC_STATUS_TO_HTTP_STATUS.get(
-            error.code(), HTTPStatus.INTERNAL_SERVER_ERROR
-        ),
+        status_code=status_code,
         detail=error_details,
+        headers=headers,
     )
 
 

langgraph_api/metadata.py

@@ -2,9 +2,9 @@ import asyncio
 import os
 import uuid
 from datetime import UTC, datetime
+from importlib import metadata as importlib_metadata
 from typing import Any
 
-import langgraph.version
 import orjson
 import structlog
 
@@ -58,6 +58,10 @@ else:
     HOST = "self-hosted"
 PLAN = "enterprise" if plus_features_enabled() else "developer"
 USER_API_URL = os.getenv("LANGGRAPH_API_URL", None)
+DEPENDENCY_VERSION_DISTRIBUTIONS = {
+    "langgraph.python.version": "langgraph",
+    "starlette.version": "starlette",
+}
 
 RUN_COUNTER = 0
 NODE_COUNTER = 0
@@ -80,6 +84,19 @@ if LANGSMITH_AUTH_ENDPOINT:
         )
 
 
+def _dependency_version_metadata() -> dict[str, str]:
+    tags = {}
+    for tag, distribution in DEPENDENCY_VERSION_DISTRIBUTIONS.items():
+        try:
+            tags[tag] = importlib_metadata.version(distribution)
+        except importlib_metadata.PackageNotFoundError:
+            logger.debug(
+                "Dependency package not found for metadata", package=distribution
+            )
+            tags[tag] = ""
+    return tags
+
+
 def _lang_usage_metadata() -> tuple[dict[str, str], dict[str, int]]:
     js_graph_count = sum(1 for graph_id in GRAPHS if is_js_graph(graph_id))
     py_graph_count = len(GRAPHS) - js_graph_count
@@ -130,8 +147,6 @@ async def metadata_loop() -> None:
             "No license key or control plane API key set, skipping metadata loop"
         )
         return
-    lg_version = langgraph.version.__version__
-
     if (
         LANGGRAPH_CLOUD_LICENSE_KEY
         and not LANGGRAPH_CLOUD_LICENSE_KEY.startswith("lcl_")
@@ -160,7 +175,7 @@ async def metadata_loop() -> None:
     base_tags = _ensure_strings(
         # Tag values must be strings.
         {
-            "langgraph.python.version": lg_version,
+            **_dependency_version_metadata(),
             "langgraph_api.version": __version__ or "",
             "langgraph.platform.revision": REVISION or "",
             "langgraph.platform.variant": VARIANT or "",

langgraph_api/stream.py

@@ -339,7 +339,26 @@ async def astream_state(
                 ):
                     yield event["name"], event["data"]
                 # TODO support messages-tuple for js graphs
-                if event["event"] == "on_chain_stream" and event["run_id"] == run_id:
+                # For JS (BaseRemotePregel) with subgraphs=True, subgraph events
+                # carry the subgraph's own run_id. Accept them if the chunk is a
+                # proper 3-tuple [ns, mode, chunk] — that's the sidecar's
+                # normalized format. Limit this bypass to remote graphs only;
+                # Python child events must fall through to the raw "events"
+                # fallback below when the client requests stream_mode="events".
+                _chunk_data = (
+                    event["data"].get("chunk")
+                    if event["event"] == "on_chain_stream"
+                    else None
+                )
+                if event["event"] == "on_chain_stream" and (
+                    event["run_id"] == run_id
+                    or (
+                        subgraphs
+                        and is_remote_pregel
+                        and isinstance(_chunk_data, (list, tuple))
+                        and len(_chunk_data) == 3
+                    )
+                ):
                     if subgraphs:
                         ns, mode, chunk = event["data"]["chunk"]
                     else:

{langgraph_api-0.12.0.dev13.dist-info → langgraph_api-0.12.0.dev15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.12.0.dev13
+Version: 0.12.0.dev15
 Author-email: Will Fu-Hinthorn <will@langchain.dev>, Josh Rogers <josh@langchain.dev>, Parker Rule <parker@langchain.dev>
 License: Elastic-2.0
 License-File: LICENSE
@@ -29,7 +29,7 @@ Requires-Dist: prometheus-client>=0.0.1
 Requires-Dist: protobuf<7.0.0,>=6.32.1
 Requires-Dist: pyjwt>=2.13.0
 Requires-Dist: sse-starlette<3.4.0,>=2.1.3
-Requires-Dist: starlette>=0.38.6
+Requires-Dist: starlette>=1.3.1
 Requires-Dist: structlog<26,>=24.1.0
 Requires-Dist: tenacity>=8.0.0
 Requires-Dist: truststore>=0.1

{langgraph_api-0.12.0.dev13.dist-info → langgraph_api-0.12.0.dev15.dist-info}/RECORD

@@ -1,7 +1,7 @@
-langgraph_api/__init__.py,sha256=hom8LDza4isfZKrYaGZxZ5cj1otdh2O5s3xZqMaK8zc,29
+langgraph_api/__init__.py,sha256=3hHYNZlyWnoc9JVCLo-c7OAsrbQDPhSBvtrj8CKX1z4,29
 langgraph_api/_factory_utils.py,sha256=5JsiJbg_YocVSryN2jwoZTg03-eyymlWMK6sKCmXwz0,5756
 langgraph_api/asgi_transport.py,sha256=XApY3lIWBZTMbbsl8dDJzl0cLGirmAGE0SifqZUnXvs,11896
-langgraph_api/asyncio.py,sha256=c-YE-14N7_AP1GzifsbP14XnhLsmxT2P916KXruerpI,10573
+langgraph_api/asyncio.py,sha256=smqyAoO9nIyPPQRw1IVSCh85fwoZP4PEo-5dDhxGruc,10676
 langgraph_api/cache.py,sha256=LuB3Te0UdXC8a-uEJoRHoe5XsgXulXI59f_Q6DCJNKc,13005
 langgraph_api/cli.py,sha256=ATtS9s9Cx7QNiGPJceKnMCko29A25ZA-xz39fdxmgfg,22389
 langgraph_api/command.py,sha256=d-k8h6H4ix1n7fSZ-Zb01NbSkEyqrD6cMKfDFXEIYEw,821
@@ -13,7 +13,7 @@ langgraph_api/http.py,sha256=7hPxKbj-xoAKcm7iucBpT5nM_hXOgGVCPbBsCD693Cw,6977
 langgraph_api/http_metrics.py,sha256=etxbZNmYxdb58DVLNkHP7S-N6njXPTiQh2OWKMaIZi8,5336
 langgraph_api/http_metrics_utils.py,sha256=sjxF7SYGTzY0Wz_G0dzatsYNnWr31S6ujej4JmBG2yo,866
 langgraph_api/logging.py,sha256=V1RCnqVLuMvJtrBiyMMLfaEdbS3k5A2M8Unhr4FUUdQ,6801
-langgraph_api/metadata.py,sha256=ih2et_R0prFsCzikQ4_L0j9up7t0rObAMVKyEk7ienI,9778
+langgraph_api/metadata.py,sha256=hXkF55cWxxnjAJ0MB_PhU_4NfVzpLuwU4xwFDaAgi7I,10314
 langgraph_api/metrics_collector.py,sha256=gMLHL18rJyYl985AOmu9eH7W1ttdRdkPHzeyczjCOBw,8280
 langgraph_api/metrics_otlp.py,sha256=t9oJrxfxY2O5jY4JW2gONPKoBiBuklhzCrnZvn1qTxQ,28730
 langgraph_api/otel_context.py,sha256=DWFwW4Yu88QY4W2J0IRcURR450Th9J2DupvDDkSkMBA,7166
@@ -29,7 +29,7 @@ langgraph_api/server.py,sha256=1eAZPim0Pkgh5oGS4EvW-_7Zh_82iGOZtR1rpX08FoA,11216
 langgraph_api/sse.py,sha256=cChZ7raQUHp8p5BreE_5wMBR8lFO0n7746sV8_HQOrc,4822
 langgraph_api/state.py,sha256=A6jx2OpcVmiFNBafjZ2tX1x6s_AhBk5EW2drd-vaqIc,4396
 langgraph_api/store.py,sha256=i-jVfWzDEmPESVofh1tNWbo0HMLiLz_BSKKV4QjN-dE,5027
-langgraph_api/stream.py,sha256=MGWh1NvDGPqNgVXp-3wknqpam9g9nrqaMlKMXCWwx6M,32407
+langgraph_api/stream.py,sha256=rJIGe8rEeS6QxQfsDKomUBRED6ZPIF-ALtx4D8Gw0JQ,33348
 langgraph_api/stream_v2.py,sha256=KCIsPf6gs_jrIdZ8D2VTEi5c5p5aRNpVNhMP4kzjp2g,10824
 langgraph_api/traceblock.py,sha256=GhgAtLhAQ8Z8LHtRQq3par-rfPpRBQp2Q7WBACJq1NI,677
 langgraph_api/validation.py,sha256=XyeKyt7jAICmIlT_b0J0mv2YbwIbNoe4m6zEmfk9gOA,14657
@@ -64,7 +64,7 @@ langgraph_api/auth/studio_user.py,sha256=gNCicIo6cYaHmFj2sEdsvDYkKW7NWfGXGS2tTAM
 langgraph_api/auth/langsmith/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/auth/langsmith/backend.py,sha256=Y6-VxD7zfV1jzGdjmQ66CgNa3SenLbo3d_375CcKZ9U,3770
 langgraph_api/auth/langsmith/client.py,sha256=79kwCVeHU64nsHsxWipfZhf44lM6vfs2nlfTxlJF6LU,4142
-langgraph_api/config/__init__.py,sha256=qj7HF1XmojpG6WXahlFxrMZiO8hUI6QlPNVT-32xfvs,25528
+langgraph_api/config/__init__.py,sha256=NYJWKJ-PbfeI6k5KnF3W_HWLE8EOS0rJKaQQqKTCAus,25717
 langgraph_api/config/_parse.py,sha256=RpfZgFAPJlRMv13AzGr3kAYbIrqHcgjzO8IgeboTw4A,3922
 langgraph_api/config/schemas.py,sha256=cHzVvepthpD7DDeWE2ytkOHah-iDNH7xRx9dSWUatQI,20864
 langgraph_api/encryption/__init__.py,sha256=gaCZ00CocSbqSqrDn6XJHaSp2CZCnC8qnrD9G4fbzyI,363
@@ -86,7 +86,7 @@ langgraph_api/grpc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
 langgraph_api/grpc/client.py,sha256=ypSgZWclK7OCKLGW8gD_LZwwgsgQEahJqc8mBWF8Rug,15578
 langgraph_api/grpc/server.py,sha256=Fknr6BJbdSK38wF7B1vPDplK25KB_UfMB9Fp1er6DPU,8366
 langgraph_api/grpc/generated/core_api_pb2.pyi,sha256=0PEzVh8s6rqHq52ekhqPaMI_TOsP4_xuUUv9_XS7r1w,54758
-langgraph_api/grpc/ops/__init__.py,sha256=k2Ehs9LbMgYQp5rZ0QlxVA18OWJJd64UotVrB9GwGnM,18148
+langgraph_api/grpc/ops/__init__.py,sha256=ACO-Cp7ygehNfOG1ucSH8iMw3hVXbLluxiqczG2PKTQ,18827
 langgraph_api/grpc/ops/assistants.py,sha256=00vY7Dd7BcXaVp_6UrKZIb5b4tQ8Lm5MxJcEpyVPfo0,13902
 langgraph_api/grpc/ops/cache.py,sha256=MoPAmSuOwbt2Cud1YZJm_YJeaDHPFy1IiV4m0PDImgg,1399
 langgraph_api/grpc/ops/crons.py,sha256=oGPW9qW-J4H9baX7Jsee1opsFnXfThHLHh_N47mDf9s,19596
@@ -165,14 +165,16 @@ openapi.json,sha256=0N-D4j0_3ruH6nMF4k5lnqmpV-aoB5Se-WWheCQuQQs,219552
 langgraph_grpc_common/__init__.py,sha256=sO7jFHekQxt6zZPhTPZAWLYLB_S8w7-VuJ7NyWFlFNo,97
 langgraph_grpc_common/checkpointer.py,sha256=u3ERFgqnUCpCYpOSxKR_6Jvo_tR4yDvq7qU-EL61bSg,14022
 langgraph_grpc_common/serde.py,sha256=pxIhgxtrXtwWVhAlLE2LAeSAuKk2ZHtnvAP3DRDT2Qw,6184
+langgraph_grpc_common/store.py,sha256=BIHYT_BTw2ByFAIsomu9vwNVu1kZdUEOjkJChV7rjuA,3721
 langgraph_grpc_common/conversion/__init__.py,sha256=pLsOrRtmF9YkgmCPhcfym6lYOB7PCJTRItyo1YK_AT8,208
 langgraph_grpc_common/conversion/_compat.py,sha256=MIEo69I7tfaPskunPGwBCKheksLwxpw-z-4O3hvO9og,4010
 langgraph_grpc_common/conversion/checkpoint.py,sha256=Ty81cijafeal7blzzXbYFjWpu9iXVb_spFLBZ2FvmgI,12045
 langgraph_grpc_common/conversion/config.py,sha256=HjlHRX-_GpKa1m0RWVL1PEZQCvA3w_8wKVYrtaT06do,13015
 langgraph_grpc_common/conversion/durability.py,sha256=FWhqjzVN4JrrsAL9AwsJkvIorQGNRl14w97Va-WZRk0,991
+langgraph_grpc_common/conversion/store.py,sha256=NGGZp4FiMzJXByIqRRy2qYDxlWeXBGkEN9k1aNTsVUA,6253
 langgraph_grpc_common/conversion/struct.py,sha256=yk4jpPdywzn8UosAohun21lU5D_IIglOY9rkBhiG-gA,2072
 langgraph_grpc_common/conversion/value.py,sha256=eZ9fDqKje9tKsO8A_6f0BPSI7lt94Vzuu3NNEEYLFBM,8343
-langgraph_grpc_common/proto/__init__.py,sha256=cFnJ18b62FnqKcZHz9wjKES0QetC_5twG47G1W0A2mM,1283
+langgraph_grpc_common/proto/__init__.py,sha256=eBejrrULiDdasi3_Hga_AJW8ZPMkjKyyAeW9Ey640Ks,1375
 langgraph_grpc_common/proto/checkpointer_pb2.py,sha256=c-gaVwNHm7sUzThJDzumME2M_nP4B1RjEki4ZG7-kjM,7915
 langgraph_grpc_common/proto/checkpointer_pb2.pyi,sha256=rGUxQyNFifmu8OPy0S418sCWSBQY_CVckobPXtFfZfs,20763
 langgraph_grpc_common/proto/checkpointer_pb2_grpc.py,sha256=MLhZ7M4g9gjfXZqWenk93oO2uIZSybbFoaeU8bauFPA,20176
@@ -229,8 +231,12 @@ langgraph_grpc_common/proto/errors_pb2.py,sha256=JI6x-vBK1AE7DHZ5DQwN1mZWF6C4xTR
 langgraph_grpc_common/proto/errors_pb2.pyi,sha256=rd3-BYUH8V-aO66taL7OOblaLgdrDtf1Vcd38GUoVVM,2181
 langgraph_grpc_common/proto/errors_pb2_grpc.py,sha256=2-LwQ0OPGo-NtC0269q7Fw6GPBxnTLYWq3xP5Eq0_YA,886
 langgraph_grpc_common/proto/errors_pb2_grpc.pyi,sha256=uC9Wnq6uyg488QiONpJ0ba1s_iouQCOYsjd_FDd1XUM,495
-langgraph_api-0.12.0.dev13.dist-info/METADATA,sha256=JurA5etESRZZMt2ew_YYMC9wLFzGh53TVBzR6b5l0to,4632
-langgraph_api-0.12.0.dev13.dist-info/WHEEL,sha256=mffPy8wBnZQn2VnJUU5jE99KsxaSfiyMHV9Yt0aLVxs,87
-langgraph_api-0.12.0.dev13.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
-langgraph_api-0.12.0.dev13.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.12.0.dev13.dist-info/RECORD,,
+langgraph_grpc_common/proto/store_pb2.py,sha256=5px40nIUag8p0KlE6o5vpvgfenX71BDIr3VjFfFQP6E,3406
+langgraph_grpc_common/proto/store_pb2.pyi,sha256=cZZGod9tBReUukPkjNzssEdz5OBoZ2pecXugbp-9Lmo,9755
+langgraph_grpc_common/proto/store_pb2_grpc.py,sha256=wSD65QgFkYTkPTHcf7VhyI2FQ7UH_RgIf3vQW3VYlsM,3324
+langgraph_grpc_common/proto/store_pb2_grpc.pyi,sha256=oS1h2cDoq2OjyM8xazcmc8LwRDC5oCAD0zto2QUmPQw,2027
+langgraph_api-0.12.0.dev15.dist-info/METADATA,sha256=nc5DmJfxPOxGO2rAoxSIuNyd_s0yNxlCB-M8FZGlWL8,4631
+langgraph_api-0.12.0.dev15.dist-info/WHEEL,sha256=mffPy8wBnZQn2VnJUU5jE99KsxaSfiyMHV9Yt0aLVxs,87
+langgraph_api-0.12.0.dev15.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
+langgraph_api-0.12.0.dev15.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.12.0.dev15.dist-info/RECORD,,

langgraph_grpc_common/conversion/store.py

@@ -0,0 +1,208 @@
+"""Marshal langgraph.store.base GetOp/PutOp/Item/Result to and from store.proto."""
+
+from __future__ import annotations
+
+from datetime import datetime
+from typing import TYPE_CHECKING, Any, cast
+
+import orjson
+from google.protobuf import empty_pb2
+from langgraph.store.base import (
+    GetOp,
+    IndexConfig,
+    Item,
+    Op,
+    PutOp,
+    Result,
+    ensure_embeddings,
+    get_text_at_path,
+    tokenize_path,
+)
+
+from langgraph_grpc_common.proto import store_pb2
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+    from langchain_core.embeddings import Embeddings
+
+
+def _ensure_index_config(
+    index_config: IndexConfig,
+) -> tuple[Embeddings | None, IndexConfig]:
+    # Vendored from langgraph.store.postgres.base so grpc_common_py stays
+    # backend-agnostic (no langgraph-checkpoint-postgres / psycopg dependency).
+    index_config = index_config.copy()
+    tokenized: list[tuple[str, str | list[str]]] = []
+    tot = 0
+    fields = index_config.get("fields") or ["$"]
+    if isinstance(fields, str):
+        fields = [fields]
+    if not isinstance(fields, list):
+        raise ValueError(f"Text fields must be a list or a string. Got {fields}")
+    for p in fields:
+        if p == "$":
+            tokenized.append((p, "$"))
+            tot += 1
+        else:
+            toks = tokenize_path(p)
+            tokenized.append((p, toks))
+            tot += len(toks)
+    index_config["__tokenized_fields"] = tokenized  # ty: ignore[invalid-key]
+    index_config["__estimated_num_vectors"] = tot  # ty: ignore[invalid-key]
+    embeddings = ensure_embeddings(index_config.get("embed"))
+    return embeddings, index_config
+
+
+def _ttl_to_proto_minutes(ttl: float | None) -> int:
+    if ttl is None:
+        return 0
+    if ttl != int(ttl):
+        raise ValueError(
+            f"gRPC store only supports integer-minute TTL; got fractional ttl={ttl}"
+        )
+    return int(ttl)
+
+
+def op_to_proto(
+    op: Op,
+    *,
+    vectors: Sequence[store_pb2.VectorValue] | None = None,
+) -> store_pb2.Op:
+    if isinstance(op, GetOp):
+        return store_pb2.Op(
+            get=store_pb2.GetOp(
+                namespace=list(op.namespace),
+                key=str(op.key),
+                refreshTTL=op.refresh_ttl,
+            )
+        )
+    if isinstance(op, PutOp):
+        if op.value is None:
+            raise TypeError(
+                "PutOp with value=None (delete) is not supported by gRPC store"
+            )
+        put_kwargs: dict[str, Any] = {
+            "namespace": list(op.namespace),
+            "key": str(op.key),
+            "value_json": orjson.dumps(op.value),
+            "ttlMinutes": _ttl_to_proto_minutes(op.ttl),
+        }
+        if vectors is not None:
+            put_kwargs["vectors"] = list(vectors)
+        return store_pb2.Op(put=store_pb2.PutOp(**put_kwargs))
+    raise TypeError(f"unsupported op: {type(op).__name__}")
+
+
+def op_from_proto(msg: store_pb2.Op) -> Op:
+    kind = msg.WhichOneof("op")
+    if kind == "get":
+        g = msg.get
+        return GetOp(
+            namespace=tuple(g.namespace),
+            key=g.key,
+            refresh_ttl=g.refreshTTL,
+        )
+    if kind == "put":
+        p = msg.put
+        ttl: float | None = None
+        if p.ttlMinutes:
+            ttl = float(p.ttlMinutes)
+        return PutOp(
+            namespace=tuple(p.namespace),
+            key=p.key,
+            value=orjson.loads(p.value_json),
+            ttl=ttl,
+        )
+    raise ValueError("empty Op message (no oneof set)")
+
+
+def result_to_proto(result: Result, op: Op) -> store_pb2.OpResult:
+    if isinstance(op, GetOp):
+        if result is None:
+            return store_pb2.OpResult(none=empty_pb2.Empty())
+        if not isinstance(result, Item):
+            raise TypeError(
+                f"expected Item for GetOp result, got {type(result).__name__}"
+            )
+        return store_pb2.OpResult(item=_item_to_proto(result))
+    if isinstance(op, PutOp):
+        return store_pb2.OpResult(none=empty_pb2.Empty())
+    raise TypeError(f"unsupported op for result encoding: {type(op).__name__}")
+
+
+def result_from_proto(msg: store_pb2.OpResult) -> Result:
+    kind = msg.WhichOneof("result")
+    if kind == "none":
+        return None
+    if kind == "item":
+        return _item_from_proto(msg.item)
+    raise ValueError("empty OpResult message (no oneof set)")
+
+
+def _item_to_proto(item: Item) -> store_pb2.Item:
+    return store_pb2.Item(
+        namespace=list(item.namespace),
+        key=item.key,
+        value_json=orjson.dumps(item.value),
+        created_at=_format_dt(item.created_at),
+        updated_at=_format_dt(item.updated_at),
+    )
+
+
+def _item_from_proto(msg: store_pb2.Item) -> Item:
+    return Item(
+        namespace=tuple(msg.namespace),
+        key=msg.key,
+        value=orjson.loads(msg.value_json),
+        created_at=datetime.fromisoformat(msg.created_at),
+        updated_at=datetime.fromisoformat(msg.updated_at),
+    )
+
+
+def _format_dt(dt: datetime) -> str:
+    return dt.isoformat()
+
+
+async def vectors_for_put_op(
+    op: PutOp,
+    index_config: IndexConfig,
+    embeddings: Embeddings,
+) -> list[store_pb2.VectorValue]:
+    if op.index is False or op.value is None:
+        return []
+
+    value = op.value
+    if op.index is None:
+        paths = cast(
+            "list[tuple[str, str | list[str]]]",
+            index_config["__tokenized_fields"],  # ty: ignore[invalid-key]
+        )
+    else:
+        paths = [(ix, tokenize_path(ix)) for ix in op.index]
+
+    texts: list[str] = []
+    pathnames: list[str] = []
+    for path, tokenized_path in paths:
+        extracted = get_text_at_path(value, tokenized_path)
+        for i, text in enumerate(extracted):
+            pathname = f"{path}.{i}" if len(extracted) > 1 else path
+            pathnames.append(pathname)
+            texts.append(text)
+
+    if not texts:
+        return []
+
+    vectors = await embeddings.aembed_documents(texts)
+    return [
+        store_pb2.VectorValue(valuePath=pathname, embedding=vector)
+        for pathname, vector in zip(pathnames, vectors, strict=False)
+    ]
+
+
+def normalize_index_config(
+    index: IndexConfig | None,
+) -> tuple[Embeddings | None, IndexConfig | None]:
+    if not index:
+        return None, None
+    return _ensure_index_config(index)

langgraph_grpc_common/proto/__init__.py

@@ -15,6 +15,8 @@ from . import engine_common_pb2
 from . import engine_common_pb2_grpc
 from . import checkpointer_pb2
 from . import checkpointer_pb2_grpc
+from . import store_pb2
+from . import store_pb2_grpc
 from . import encryption_pb2
 from . import encryption_pb2_grpc
 from . import errors_pb2
@@ -27,6 +29,8 @@ __all__ = [
     "engine_common_pb2_grpc",
     "checkpointer_pb2",
     "checkpointer_pb2_grpc",
+    "store_pb2",
+    "store_pb2_grpc",
     "encryption_pb2",
     "encryption_pb2_grpc",
     "errors_pb2",

langgraph_grpc_common/proto/store_pb2.py

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: store.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'store.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0bstore.proto\x12\x05store\x1a\x1bgoogle/protobuf/empty.proto\"&\n\x0c\x42\x61tchRequest\x12\x16\n\x03ops\x18\x01 \x03(\x0b\x32\t.store.Op\"1\n\rBatchResponse\x12 \n\x07results\x18\x01 \x03(\x0b\x32\x0f.store.OpResult\"D\n\x02Op\x12\x1b\n\x03get\x18\x01 \x01(\x0b\x32\x0c.store.GetOpH\x00\x12\x1b\n\x03put\x18\x02 \x01(\x0b\x32\x0c.store.PutOpH\x00\x42\x04\n\x02op\";\n\x05GetOp\x12\x11\n\tnamespace\x18\x01 \x03(\t\x12\x0b\n\x03key\x18\x02 \x01(\t\x12\x12\n\nrefreshTTL\x18\x03 \x01(\x08\"3\n\x0bVectorValue\x12\x11\n\tvaluePath\x18\x01 \x01(\t\x12\x11\n\tembedding\x18\x02 \x03(\x02\"t\n\x05PutOp\x12\x11\n\tnamespace\x18\x01 \x03(\t\x12\x0b\n\x03key\x18\x02 \x01(\t\x12\x12\n\nvalue_json\x18\x03 \x01(\x0c\x12\x12\n\nttlMinutes\x18\x04 \x01(\x05\x12#\n\x07vectors\x18\x05 \x03(\x0b\x32\x12.store.VectorValue\"Y\n\x08OpResult\x12\x1b\n\x04item\x18\x01 \x01(\x0b\x32\x0b.store.ItemH\x00\x12&\n\x04none\x18\x02 \x01(\x0b\x32\x16.google.protobuf.EmptyH\x00\x42\x08\n\x06result\"b\n\x04Item\x12\x11\n\tnamespace\x18\x01 \x03(\t\x12\x0b\n\x03key\x18\x02 \x01(\t\x12\x12\n\nvalue_json\x18\x03 \x01(\x0c\x12\x12\n\ncreated_at\x18\x04 \x01(\t\x12\x12\n\nupdated_at\x18\x05 \x01(\t2;\n\x05Store\x12\x32\n\x05\x42\x61tch\x12\x13.store.BatchRequest\x1a\x14.store.BatchResponseB?Z=github.com/langchain-ai/langgraph-api/core/internal/engine/pbb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'store_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z=github.com/langchain-ai/langgraph-api/core/internal/engine/pb'
+  _globals['_BATCHREQUEST']._serialized_start=51
+  _globals['_BATCHREQUEST']._serialized_end=89
+  _globals['_BATCHRESPONSE']._serialized_start=91
+  _globals['_BATCHRESPONSE']._serialized_end=140
+  _globals['_OP']._serialized_start=142
+  _globals['_OP']._serialized_end=210
+  _globals['_GETOP']._serialized_start=212
+  _globals['_GETOP']._serialized_end=271
+  _globals['_VECTORVALUE']._serialized_start=273
+  _globals['_VECTORVALUE']._serialized_end=324
+  _globals['_PUTOP']._serialized_start=326
+  _globals['_PUTOP']._serialized_end=442
+  _globals['_OPRESULT']._serialized_start=444
+  _globals['_OPRESULT']._serialized_end=533
+  _globals['_ITEM']._serialized_start=535
+  _globals['_ITEM']._serialized_end=633
+  _globals['_STORE']._serialized_start=635
+  _globals['_STORE']._serialized_end=694
+# @@protoc_insertion_point(module_scope)

langgraph_grpc_common/proto/store_pb2.pyi

@@ -0,0 +1,246 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+from collections import abc as _abc
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import empty_pb2 as _empty_pb2
+from google.protobuf import message as _message
+from google.protobuf.internal import containers as _containers
+import builtins as _builtins
+import sys
+import typing as _typing
+
+if sys.version_info >= (3, 11):
+    from typing import TypeAlias as _TypeAlias, Never as _Never
+else:
+    from typing_extensions import TypeAlias as _TypeAlias, Never as _Never
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+@_typing.final
+class BatchRequest(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
+
+    OPS_FIELD_NUMBER: _builtins.int
+    @_builtins.property
+    def ops(self) -> _containers.RepeatedCompositeFieldContainer[Global___Op]: ...
+    def __init__(
+        self,
+        *,
+        ops: _abc.Iterable[Global___Op] | None = ...,
+    ) -> None: ...
+    _HasFieldArgType: _TypeAlias = _Never  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal["ops", b"ops"]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
+    def WhichOneof(self, oneof_group: _Never) -> None: ...
+
+Global___BatchRequest: _TypeAlias = BatchRequest  # noqa: Y015
+
+@_typing.final
+class BatchResponse(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
+
+    RESULTS_FIELD_NUMBER: _builtins.int
+    @_builtins.property
+    def results(self) -> _containers.RepeatedCompositeFieldContainer[Global___OpResult]:
+        """Same length as BatchRequest.ops, same order."""
+
+    def __init__(
+        self,
+        *,
+        results: _abc.Iterable[Global___OpResult] | None = ...,
+    ) -> None: ...
+    _HasFieldArgType: _TypeAlias = _Never  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal["results", b"results"]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
+    def WhichOneof(self, oneof_group: _Never) -> None: ...
+
+Global___BatchResponse: _TypeAlias = BatchResponse  # noqa: Y015
+
+@_typing.final
+class Op(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
+
+    GET_FIELD_NUMBER: _builtins.int
+    PUT_FIELD_NUMBER: _builtins.int
+    @_builtins.property
+    def get(self) -> Global___GetOp: ...
+    @_builtins.property
+    def put(self) -> Global___PutOp: ...
+    def __init__(
+        self,
+        *,
+        get: Global___GetOp | None = ...,
+        put: Global___PutOp | None = ...,
+    ) -> None: ...
+    _HasFieldArgType: _TypeAlias = _typing.Literal["get", b"get", "op", b"op", "put", b"put"]  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal["get", b"get", "op", b"op", "put", b"put"]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
+    _WhichOneofReturnType_op: _TypeAlias = _typing.Literal["get", "put"]  # noqa: Y015
+    _WhichOneofArgType_op: _TypeAlias = _typing.Literal["op", b"op"]  # noqa: Y015
+    def WhichOneof(self, oneof_group: _WhichOneofArgType_op) -> _WhichOneofReturnType_op | None: ...
+
+Global___Op: _TypeAlias = Op  # noqa: Y015
+
+@_typing.final
+class GetOp(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
+
+    NAMESPACE_FIELD_NUMBER: _builtins.int
+    KEY_FIELD_NUMBER: _builtins.int
+    REFRESHTTL_FIELD_NUMBER: _builtins.int
+    key: _builtins.str
+    refreshTTL: _builtins.bool
+    @_builtins.property
+    def namespace(self) -> _containers.RepeatedScalarFieldContainer[_builtins.str]:
+        """Hierarchical namespace path, e.g. ["users", "profiles"]."""
+
+    def __init__(
+        self,
+        *,
+        namespace: _abc.Iterable[_builtins.str] | None = ...,
+        key: _builtins.str = ...,
+        refreshTTL: _builtins.bool = ...,
+    ) -> None: ...
+    _HasFieldArgType: _TypeAlias = _Never  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal["key", b"key", "namespace", b"namespace", "refreshTTL", b"refreshTTL"]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
+    def WhichOneof(self, oneof_group: _Never) -> None: ...
+
+Global___GetOp: _TypeAlias = GetOp  # noqa: Y015
+
+@_typing.final
+class VectorValue(_message.Message):
+    """Vector value for embedding support"""
+
+    DESCRIPTOR: _descriptor.Descriptor
+
+    VALUEPATH_FIELD_NUMBER: _builtins.int
+    EMBEDDING_FIELD_NUMBER: _builtins.int
+    valuePath: _builtins.str
+    """path to the value in the document"""
+    @_builtins.property
+    def embedding(self) -> _containers.RepeatedScalarFieldContainer[_builtins.float]:
+        """embedding vector"""
+
+    def __init__(
+        self,
+        *,
+        valuePath: _builtins.str = ...,
+        embedding: _abc.Iterable[_builtins.float] | None = ...,
+    ) -> None: ...
+    _HasFieldArgType: _TypeAlias = _Never  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal["embedding", b"embedding", "valuePath", b"valuePath"]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
+    def WhichOneof(self, oneof_group: _Never) -> None: ...
+
+Global___VectorValue: _TypeAlias = VectorValue  # noqa: Y015
+
+@_typing.final
+class PutOp(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
+
+    NAMESPACE_FIELD_NUMBER: _builtins.int
+    KEY_FIELD_NUMBER: _builtins.int
+    VALUE_JSON_FIELD_NUMBER: _builtins.int
+    TTLMINUTES_FIELD_NUMBER: _builtins.int
+    VECTORS_FIELD_NUMBER: _builtins.int
+    key: _builtins.str
+    value_json: _builtins.bytes
+    """JSON-encoded value (orjson)."""
+    ttlMinutes: _builtins.int
+    @_builtins.property
+    def namespace(self) -> _containers.RepeatedScalarFieldContainer[_builtins.str]: ...
+    @_builtins.property
+    def vectors(self) -> _containers.RepeatedCompositeFieldContainer[Global___VectorValue]:
+        """array of vector values for embedding support"""
+
+    def __init__(
+        self,
+        *,
+        namespace: _abc.Iterable[_builtins.str] | None = ...,
+        key: _builtins.str = ...,
+        value_json: _builtins.bytes = ...,
+        ttlMinutes: _builtins.int = ...,
+        vectors: _abc.Iterable[Global___VectorValue] | None = ...,
+    ) -> None: ...
+    _HasFieldArgType: _TypeAlias = _Never  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal["key", b"key", "namespace", b"namespace", "ttlMinutes", b"ttlMinutes", "value_json", b"value_json", "vectors", b"vectors"]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
+    def WhichOneof(self, oneof_group: _Never) -> None: ...
+
+Global___PutOp: _TypeAlias = PutOp  # noqa: Y015
+
+@_typing.final
+class OpResult(_message.Message):
+    """Result corresponds to langgraph.store.base.Result.
+    Kind is derived from the originating op:
+      GetOp -> item (or none if missing)
+      PutOp -> none
+    """
+
+    DESCRIPTOR: _descriptor.Descriptor
+
+    ITEM_FIELD_NUMBER: _builtins.int
+    NONE_FIELD_NUMBER: _builtins.int
+    @_builtins.property
+    def item(self) -> Global___Item: ...
+    @_builtins.property
+    def none(self) -> _empty_pb2.Empty: ...
+    def __init__(
+        self,
+        *,
+        item: Global___Item | None = ...,
+        none: _empty_pb2.Empty | None = ...,
+    ) -> None: ...
+    _HasFieldArgType: _TypeAlias = _typing.Literal["item", b"item", "none", b"none", "result", b"result"]  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal["item", b"item", "none", b"none", "result", b"result"]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
+    _WhichOneofReturnType_result: _TypeAlias = _typing.Literal["item", "none"]  # noqa: Y015
+    _WhichOneofArgType_result: _TypeAlias = _typing.Literal["result", b"result"]  # noqa: Y015
+    def WhichOneof(self, oneof_group: _WhichOneofArgType_result) -> _WhichOneofReturnType_result | None: ...
+
+Global___OpResult: _TypeAlias = OpResult  # noqa: Y015
+
+@_typing.final
+class Item(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
+
+    NAMESPACE_FIELD_NUMBER: _builtins.int
+    KEY_FIELD_NUMBER: _builtins.int
+    VALUE_JSON_FIELD_NUMBER: _builtins.int
+    CREATED_AT_FIELD_NUMBER: _builtins.int
+    UPDATED_AT_FIELD_NUMBER: _builtins.int
+    key: _builtins.str
+    value_json: _builtins.bytes
+    """JSON-encoded value (orjson)."""
+    created_at: _builtins.str
+    """RFC 3339 / ISO 8601 strings, matching Item.dict()."""
+    updated_at: _builtins.str
+    @_builtins.property
+    def namespace(self) -> _containers.RepeatedScalarFieldContainer[_builtins.str]: ...
+    def __init__(
+        self,
+        *,
+        namespace: _abc.Iterable[_builtins.str] | None = ...,
+        key: _builtins.str = ...,
+        value_json: _builtins.bytes = ...,
+        created_at: _builtins.str = ...,
+        updated_at: _builtins.str = ...,
+    ) -> None: ...
+    _HasFieldArgType: _TypeAlias = _Never  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal["created_at", b"created_at", "key", b"key", "namespace", b"namespace", "updated_at", b"updated_at", "value_json", b"value_json"]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
+    def WhichOneof(self, oneof_group: _Never) -> None: ...
+
+Global___Item: _TypeAlias = Item  # noqa: Y015

langgraph_grpc_common/proto/store_pb2_grpc.py

@@ -0,0 +1,101 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+from . import store_pb2 as store__pb2
+
+GRPC_GENERATED_VERSION = '1.80.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + ' but the generated code in store_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
+
+class StoreStub(object):
+    """Store is the gRPC surface of langgraph's BaseStore.
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.Batch = channel.unary_unary(
+                '/store.Store/Batch',
+                request_serializer=store__pb2.BatchRequest.SerializeToString,
+                response_deserializer=store__pb2.BatchResponse.FromString,
+                _registered_method=True)
+
+
+class StoreServicer(object):
+    """Store is the gRPC surface of langgraph's BaseStore.
+    """
+
+    def Batch(self, request, context):
+        """Batch executes a sequence of operations, returning results in input order.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_StoreServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'Batch': grpc.unary_unary_rpc_method_handler(
+                    servicer.Batch,
+                    request_deserializer=store__pb2.BatchRequest.FromString,
+                    response_serializer=store__pb2.BatchResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'store.Store', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('store.Store', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class Store(object):
+    """Store is the gRPC surface of langgraph's BaseStore.
+    """
+
+    @staticmethod
+    def Batch(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/store.Store/Batch',
+            store__pb2.BatchRequest.SerializeToString,
+            store__pb2.BatchResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

langgraph_grpc_common/proto/store_pb2_grpc.pyi

@@ -0,0 +1,58 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+from collections import abc as _abc
+from grpc import aio as _aio
+import abc as _abc_1
+import grpc as _grpc
+from . import store_pb2 as _store_pb2
+import sys
+import typing as _typing
+
+if sys.version_info >= (3, 11):
+    from typing import Self as _Self
+else:
+    from typing_extensions import Self as _Self
+
+_T = _typing.TypeVar("_T")
+
+class _MaybeAsyncIterator(_abc.AsyncIterator[_T], _abc.Iterator[_T], metaclass=_abc_1.ABCMeta): ...
+
+class _ServicerContext(_grpc.ServicerContext, _aio.ServicerContext):  # type: ignore[misc, type-arg]
+    ...
+
+GRPC_GENERATED_VERSION: str
+GRPC_VERSION: str
+
+class StoreStub:
+    """Store is the gRPC surface of langgraph's BaseStore."""
+
+    @_typing.overload
+    def __new__(cls, channel: _grpc.Channel) -> _Self: ...
+    @_typing.overload
+    def __new__(cls, channel: _aio.Channel) -> StoreAsyncStub: ...
+    Batch: _grpc.UnaryUnaryMultiCallable[_store_pb2.BatchRequest, _store_pb2.BatchResponse]
+    """Batch executes a sequence of operations, returning results in input order."""
+
+@_typing.type_check_only
+class StoreAsyncStub(StoreStub):
+    """Store is the gRPC surface of langgraph's BaseStore."""
+
+    def __init__(self, channel: _aio.Channel) -> None: ...
+    Batch: _aio.UnaryUnaryMultiCallable[_store_pb2.BatchRequest, _store_pb2.BatchResponse]  # type: ignore[assignment]
+    """Batch executes a sequence of operations, returning results in input order."""
+
+class StoreServicer(metaclass=_abc_1.ABCMeta):
+    """Store is the gRPC surface of langgraph's BaseStore."""
+
+    @_abc_1.abstractmethod
+    def Batch(
+        self,
+        request: _store_pb2.BatchRequest,
+        context: _ServicerContext,
+    ) -> _typing.Union[_store_pb2.BatchResponse, _abc.Awaitable[_store_pb2.BatchResponse]]:
+        """Batch executes a sequence of operations, returning results in input order."""
+
+def add_StoreServicer_to_server(servicer: StoreServicer, server: _typing.Union[_grpc.Server, _aio.Server]) -> None: ...

langgraph_grpc_common/store.py

@@ -0,0 +1,109 @@
+"""GrpcStore — BaseStore subclass that proxies to core-server Store."""
+
+from __future__ import annotations
+
+import asyncio
+import threading
+from typing import TYPE_CHECKING, Any, TypeVar
+
+import grpc.aio
+from langgraph.store.base import IndexConfig, Op, PutOp, Result
+from langgraph.store.base.batch import AsyncBatchedBaseStore
+
+from langgraph_grpc_common.conversion.store import (
+    normalize_index_config,
+    op_to_proto,
+    result_from_proto,
+    vectors_for_put_op,
+)
+from langgraph_grpc_common.proto import store_pb2
+from langgraph_grpc_common.proto.store_pb2_grpc import StoreStub
+
+if TYPE_CHECKING:
+    from collections.abc import Coroutine, Iterable
+
+T = TypeVar("T")
+
+
+class GrpcStore(AsyncBatchedBaseStore):
+    """BaseStore client that dispatches ``abatch`` over gRPC (get/put, TTL, vectors)."""
+
+    supports_ttl = True
+    _UNSUPPORTED = "delete, search, and list_namespaces are not implemented yet"
+
+    def __init__(
+        self,
+        address: str,
+        *,
+        index: IndexConfig | None = None,
+        ttl: dict[str, Any] | None = None,
+    ) -> None:
+        super().__init__()
+        self.ttl_config = ttl
+        self._embeddings, self._index_config = normalize_index_config(index)
+        self._address = address
+        self._channel_lock = threading.Lock()
+        self._channel: grpc.aio.Channel | None = None
+        self._stub: StoreStub | None = None
+
+    def _get_stub(self) -> StoreStub:
+        if self._stub is None:
+            with self._channel_lock:
+                if self._stub is None:
+                    # Internal loopback to core-server; matches langgraph_api.grpc.client.
+                    self._channel = grpc.aio.insecure_channel(self._address)
+                    self._stub = StoreStub(self._channel)
+        return self._stub
+
+    async def abatch(self, ops: Iterable[Op]) -> list[Result]:
+        op_list = list(ops)
+        if not op_list:
+            return []
+
+        proto_ops: list[store_pb2.Op] = []
+        for op in op_list:
+            if (
+                self._index_config is not None
+                and self._embeddings is not None
+                and isinstance(op, PutOp)
+                and op.index is not False
+            ):
+                vectors = await vectors_for_put_op(
+                    op, self._index_config, self._embeddings
+                )
+                proto_ops.append(op_to_proto(op, vectors=vectors))
+            else:
+                proto_ops.append(op_to_proto(op))
+
+        request = store_pb2.BatchRequest(ops=proto_ops)
+        stub = self._get_stub()
+        response = await stub.Batch(request)
+        if len(response.results) != len(op_list):
+            raise RuntimeError(
+                f"store-server returned {len(response.results)} results for "
+                f"{len(op_list)} ops"
+            )
+        return [result_from_proto(r) for r in response.results]
+
+    def batch(self, ops: Iterable[Op]) -> list[Result]:
+        return self._run_sync(self.abatch(ops))
+
+    def _run_sync(self, coro: Coroutine[Any, Any, T]) -> T:
+        return asyncio.run_coroutine_threadsafe(coro, self._loop).result()
+
+    async def aclose(self) -> None:
+        if self._channel is not None:
+            await self._channel.close()
+            self._channel = None
+            self._stub = None
+
+    async def adelete(self, namespace: tuple[str, ...], key: str) -> None:
+        raise NotImplementedError(self._UNSUPPORTED)
+
+    async def asearch(self, *args: Any, **kwargs: Any) -> list[Any]:
+        raise NotImplementedError(self._UNSUPPORTED)
+
+    async def alist_namespaces(
+        self, *args: Any, **kwargs: Any
+    ) -> list[tuple[str, ...]]:
+        raise NotImplementedError(self._UNSUPPORTED)