langgraph-api 0.12.0.dev13__tar.gz → 0.12.0.dev15__tar.gz

{langgraph_api-0.12.0.dev13 → langgraph_api-0.12.0.dev15}/Makefile

@@ -1,7 +1,7 @@
 .PHONY: build release lint format test test_watch start start-inmem start-inmem-license-oss start start-js-server build-go-server start-go-server stop-go-server check-version check-base-imports test-a2a-tck test-a2a-tck-mandatory start-test-a2a-tck start-test-a2a-tck-mandatory start-test-a2a-tck-watch test-mcp-conformance start-test-mcp-conformance list-mcp-scenarios test-auth-read-only-assistant
 
 # Environment variables
-LANGSERVE_GRAPHS_ALL = '{"agent": {"path": "./tests/graphs/agent.py:graph", "description": "agent"}, "assistant_id_runtime": "./tests/graphs/assistant_id_runtime.py:graph", "custom_lifespan": "./tests/graphs/my_router.py:graph", "single_node": "./tests/graphs/single_node.py:graph", "benchmark": "./tests/graphs/benchmark.py:graph", "config_graph": "./tests/graphs/config_graph.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", "simple_runtime": "./tests/graphs/simple_runtime.py:graph", "agent_interrupt": "./tests/graphs/agent_interrupt.py:graph", "agent_parallel_interrupt": "./tests/graphs/agent_parallel_interrupt.py:graph", "message_type_test": "./tests/graphs/message_type_test.py:graph", "remote_subgraph_parent": "./tests/graphs/remote_subgraph_parent.py:graph", "simple_remote": "./tests/graphs/simple_remote.py:graph", "nested_subgraphs": "./tests/graphs/nested_subgraphs.py:graph", "functional_fibonacci": "./tests/graphs/functional_fibonacci.py:fibonacci", "state_graph_fibonacci": "./tests/graphs/state_graph_fibonacci.py:fibonacci", "max_concurrency_graph": "./tests/graphs/max_concurrency_graph.py:graph", "unserializable_subgraph": "./tests/graphs/unserializable_subgraph.py:graph", "agent_interrupt_text": "./tests/graphs/agent_interrupt_text.py:graph", "agent_echo_stream": "./tests/graphs/agent_echo_stream.py:graph", "agent_tool_stream": "./tests/graphs/agent_tool_stream.py:graph", "runtime_graph": "./tests/graphs/runtime_graph.py:graph", "tool_call_chunk_stream": "./tests/graphs/tool_call_chunk_stream.py:graph", "ui_tool_call_stream": "./tests/graphs/ui_tool_call_stream.py:graph", "agent_metrics_stream": "./tests/graphs/agent_metrics_stream.py:graph", "agent_multimodal_stream": "./tests/graphs/agent_multimodal_stream.py:graph", "agent_bedtime_story": "./tests/graphs/agent_bedtime_story.py:graph", "delta_channel": "./tests/graphs/delta_channel.py:graph", "delta_channel_freq": "./tests/graphs/delta_channel_freq.py:graph", "delta_channel_files": "./tests/graphs/delta_channel_files.py:graph", "delta_channel_files_freq": "./tests/graphs/delta_channel_files_freq.py:graph", "delta_channel_legacy": "./tests/graphs/delta_channel_migration.py:graph_legacy", "delta_channel_delta": "./tests/graphs/delta_channel_migration.py:graph_delta", "delta_channel_delta_filter": "./tests/graphs/delta_channel_migration.py:graph_delta_filter", "delta_channel_messages_freq": "./tests/graphs/delta_channel_messages.py:graph_freq", "delta_channel_messages": "./tests/graphs/delta_channel_messages.py:graph", "delta_channel_messages_legacy": "./tests/graphs/delta_channel_messages.py:graph_legacy", "delta_channel_messages_remove": "./tests/graphs/delta_channel_messages.py:graph_remove", "delta_channel_messages_update": "./tests/graphs/delta_channel_messages.py:graph_update", "delta_channel_messages_delta": "./tests/graphs/delta_channel_messages.py:graph_delta"}'
+LANGSERVE_GRAPHS_ALL = '{"agent": {"path": "./tests/graphs/agent.py:graph", "description": "agent"}, "assistant_id_runtime": "./tests/graphs/assistant_id_runtime.py:graph", "custom_lifespan": "./tests/graphs/my_router.py:graph", "single_node": "./tests/graphs/single_node.py:graph", "benchmark": "./tests/graphs/benchmark.py:graph", "config_graph": "./tests/graphs/config_graph.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", "simple_runtime": "./tests/graphs/simple_runtime.py:graph", "agent_interrupt": "./tests/graphs/agent_interrupt.py:graph", "agent_parallel_interrupt": "./tests/graphs/agent_parallel_interrupt.py:graph", "message_type_test": "./tests/graphs/message_type_test.py:graph", "remote_subgraph_parent": "./tests/graphs/remote_subgraph_parent.py:graph", "simple_remote": "./tests/graphs/simple_remote.py:graph", "nested_subgraphs": "./tests/graphs/nested_subgraphs.py:graph", "functional_fibonacci": "./tests/graphs/functional_fibonacci.py:fibonacci", "state_graph_fibonacci": "./tests/graphs/state_graph_fibonacci.py:fibonacci", "max_concurrency_graph": "./tests/graphs/max_concurrency_graph.py:graph", "unserializable_subgraph": "./tests/graphs/unserializable_subgraph.py:graph", "agent_interrupt_text": "./tests/graphs/agent_interrupt_text.py:graph", "agent_echo_stream": "./tests/graphs/agent_echo_stream.py:graph", "agent_tool_stream": "./tests/graphs/agent_tool_stream.py:graph", "runtime_graph": "./tests/graphs/runtime_graph.py:graph", "tool_call_chunk_stream": "./tests/graphs/tool_call_chunk_stream.py:graph", "ui_tool_call_stream": "./tests/graphs/ui_tool_call_stream.py:graph", "agent_metrics_stream": "./tests/graphs/agent_metrics_stream.py:graph", "agent_multimodal_stream": "./tests/graphs/agent_multimodal_stream.py:graph", "agent_bedtime_story": "./tests/graphs/agent_bedtime_story.py:graph", "delta_channel": "./tests/graphs/delta_channel.py:graph", "delta_channel_freq": "./tests/graphs/delta_channel_freq.py:graph", "delta_channel_files": "./tests/graphs/delta_channel_files.py:graph", "delta_channel_files_freq": "./tests/graphs/delta_channel_files_freq.py:graph", "delta_channel_legacy": "./tests/graphs/delta_channel_migration.py:graph_legacy", "delta_channel_delta": "./tests/graphs/delta_channel_migration.py:graph_delta", "delta_channel_delta_filter": "./tests/graphs/delta_channel_migration.py:graph_delta_filter", "delta_channel_messages_freq": "./tests/graphs/delta_channel_messages.py:graph_freq", "delta_channel_messages": "./tests/graphs/delta_channel_messages.py:graph", "delta_channel_messages_legacy": "./tests/graphs/delta_channel_messages.py:graph_legacy", "delta_channel_messages_remove": "./tests/graphs/delta_channel_messages.py:graph_remove", "delta_channel_messages_update": "./tests/graphs/delta_channel_messages.py:graph_update", "delta_channel_messages_delta": "./tests/graphs/delta_channel_messages.py:graph_delta", "store_e2e_async": "./tests/graphs/store_e2e.py:graph_async", "store_e2e_sync": "./tests/graphs/store_e2e.py:graph_sync"}'
 LANGSERVE_GRAPHS_AUTH = '{"agent": {"path": "./tests/graphs/agent.py:graph", "description": "agent"}, "assistant_id_runtime": "./tests/graphs/assistant_id_runtime.py:graph", "config_graph": "./tests/graphs/config_graph.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", "simple_runtime": "./tests/graphs/simple_runtime.py:graph", "agent_parallel_interrupt": "./tests/graphs/agent_parallel_interrupt.py:graph", "functional_fibonacci": "./tests/graphs/functional_fibonacci.py:fibonacci", "state_graph_fibonacci": "./tests/graphs/state_graph_fibonacci.py:fibonacci", "runtime_graph": "./tests/graphs/runtime_graph.py:graph"}'
 
 # Go server management

{langgraph_api-0.12.0.dev13 → langgraph_api-0.12.0.dev15}/PKG-INFO

@@ -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.dev15/langgraph_api/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.12.0.dev15"

{langgraph_api-0.12.0.dev13 → langgraph_api-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-0.12.0.dev13 → langgraph_api-0.12.0.dev15}/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-0.12.0.dev13 → langgraph_api-0.12.0.dev15}/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-0.12.0.dev13 → langgraph_api-0.12.0.dev15}/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-0.12.0.dev13 → langgraph_api-0.12.0.dev15}/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.dev15/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_api-0.12.0.dev13 → langgraph_api-0.12.0.dev15}/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_api-0.12.0.dev15/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)