langgraph-api 0.5.4__py3-none-any.whl → 0.7.3__py3-none-any.whl

langgraph_api/{grpc_ops → grpc}/config_conversion.py

@@ -8,7 +8,10 @@ from typing import Any, cast
 import orjson
 from langchain_core.runnables.config import RunnableConfig
 
-from langgraph_api.grpc_ops.generated import engine_common_pb2
+from langgraph_api.grpc.generated import (
+    engine_common_pb2,
+    enum_durability_pb2,
+)
 
 CONFIG_KEY_SEND = "__pregel_send"
 CONFIG_KEY_READ = "__pregel_read"
@@ -25,27 +28,27 @@ CONFIG_KEY_GRAPH_ID = "graph_id"
 
 def _durability_to_proto(
     durability: str,
-) -> engine_common_pb2.Durability:
+) -> enum_durability_pb2.Durability:
     match durability:
         case "async":
-            return engine_common_pb2.Durability.ASYNC
+            return enum_durability_pb2.Durability.ASYNC
         case "sync":
-            return engine_common_pb2.Durability.SYNC
+            return enum_durability_pb2.Durability.SYNC
         case "exit":
-            return engine_common_pb2.Durability.EXIT
+            return enum_durability_pb2.Durability.EXIT
         case _:
             raise ValueError(f"invalid durability: {durability}")
 
 
 def _durability_from_proto(
-    durability: engine_common_pb2.Durability,
+    durability: enum_durability_pb2.Durability,
 ) -> str:
     match durability:
-        case engine_common_pb2.Durability.ASYNC:
+        case enum_durability_pb2.Durability.ASYNC:
             return "async"
-        case engine_common_pb2.Durability.SYNC:
+        case enum_durability_pb2.Durability.SYNC:
             return "sync"
-        case engine_common_pb2.Durability.EXIT:
+        case enum_durability_pb2.Durability.EXIT:
             return "exit"
         case _:
             raise ValueError(f"invalid durability: {durability}")
@@ -115,7 +118,7 @@ def _inject_configurable_into_proto(
         elif key == CONFIG_KEY_THREAD_ID:
             proto.thread_id = str(value)
         elif key == CONFIG_KEY_CHECKPOINT_MAP:
-            proto.checkpoint_map.update(cast(dict[str, str], value))
+            proto.checkpoint_map.update(cast("dict[str, str]", value))
         elif key == CONFIG_KEY_CHECKPOINT_ID:
             proto.checkpoint_id = str(value)
         elif key == CONFIG_KEY_CHECKPOINT_NS:
@@ -215,6 +218,9 @@ def _configurable_from_proto(
     if config_proto.HasField("graph_id"):
         configurable[CONFIG_KEY_GRAPH_ID] = config_proto.graph_id
 
+    if config_proto.HasField("run_id"):
+        configurable["run_id"] = config_proto.run_id
+
     if len(config_proto.checkpoint_map) > 0:
         configurable[CONFIG_KEY_CHECKPOINT_MAP] = dict(config_proto.checkpoint_map)
 

langgraph_api/grpc/generated/__init__.py

@@ -0,0 +1,29 @@
+# Generated protobuf files
+# Import enum files first to avoid circular imports
+from . import enum_cancel_run_action_pb2
+from . import enum_control_signal_pb2
+from . import enum_durability_pb2
+from . import enum_multitask_strategy_pb2
+from . import enum_run_status_pb2
+from . import enum_stream_mode_pb2
+from . import enum_thread_status_pb2
+from . import enum_thread_stream_mode_pb2
+from . import core_api_pb2
+from . import core_api_pb2_grpc
+from . import checkpointer_pb2
+from . import checkpointer_pb2_grpc
+
+__all__ = [
+    "core_api_pb2",
+    "core_api_pb2_grpc",
+    "checkpointer_pb2",
+    "checkpointer_pb2_grpc",
+    "enum_cancel_run_action_pb2",
+    "enum_control_signal_pb2",
+    "enum_durability_pb2",
+    "enum_multitask_strategy_pb2",
+    "enum_run_status_pb2",
+    "enum_stream_mode_pb2",
+    "enum_thread_status_pb2",
+    "enum_thread_stream_mode_pb2",
+]

langgraph_api/grpc/generated/checkpointer_pb2.py

@@ -0,0 +1,63 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: checkpointer.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,
+    '',
+    'checkpointer.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
+from . import engine_common_pb2 as engine__common__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x12\x63heckpointer.proto\x12\x0c\x63heckpointer\x1a\x1bgoogle/protobuf/empty.proto\x1a\x13\x65ngine-common.proto\"\x97\x02\n\nPutRequest\x12\x32\n\x06\x63onfig\x18\x01 \x01(\x0b\x32\".engineCommon.EngineRunnableConfig\x12,\n\ncheckpoint\x18\x02 \x01(\x0b\x32\x18.engineCommon.Checkpoint\x12\x32\n\x08metadata\x18\x03 \x01(\x0b\x32 .engineCommon.CheckpointMetadata\x12?\n\x0cnew_versions\x18\x04 \x03(\x0b\x32).checkpointer.PutRequest.NewVersionsEntry\x1a\x32\n\x10NewVersionsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x8f\x01\n\x10PutWritesRequest\x12\x32\n\x06\x63onfig\x18\x01 \x01(\x0b\x32\".engineCommon.EngineRunnableConfig\x12#\n\x06writes\x18\x02 \x03(\x0b\x32\x13.engineCommon.Write\x12\x0f\n\x07task_id\x18\x03 \x01(\t\x12\x11\n\ttask_path\x18\x04 \x01(\t\"\xa8\x01\n\x0bListRequest\x12\x32\n\x06\x63onfig\x18\x01 \x01(\x0b\x32\".engineCommon.EngineRunnableConfig\x12\x13\n\x0b\x66ilter_json\x18\x02 \x01(\x0c\x12\x32\n\x06\x62\x65\x66ore\x18\x03 \x01(\x0b\x32\".engineCommon.EngineRunnableConfig\x12\x12\n\x05limit\x18\x04 \x01(\x03H\x00\x88\x01\x01\x42\x08\n\x06_limit\"(\n\x13\x44\x65leteThreadRequest\x12\x11\n\tthread_id\x18\x01 \x01(\t\"\xa1\x01\n\x0cPruneRequest\x12\x12\n\nthread_ids\x18\x01 \x03(\t\x12:\n\x08strategy\x18\x02 \x01(\x0e\x32(.checkpointer.PruneRequest.PruneStrategy\"A\n\rPruneStrategy\x12\x0f\n\x0bUNSPECIFIED\x10\x00\x12\x0f\n\x0bKEEP_LATEST\x10\x01\x12\x0e\n\nDELETE_ALL\x10\x02\"E\n\x0fGetTupleRequest\x12\x32\n\x06\x63onfig\x18\x01 \x01(\x0b\x32\".engineCommon.EngineRunnableConfig\"F\n\x0bPutResponse\x12\x37\n\x0bnext_config\x18\x01 \x01(\x0b\x32\".engineCommon.EngineRunnableConfig\"H\n\x0cListResponse\x12\x38\n\x11\x63heckpoint_tuples\x18\x01 \x03(\x0b\x32\x1d.engineCommon.CheckpointTuple\"e\n\x10GetTupleResponse\x12<\n\x10\x63heckpoint_tuple\x18\x01 \x01(\x0b\x32\x1d.engineCommon.CheckpointTupleH\x00\x88\x01\x01\x42\x13\n\x11_checkpoint_tuple2\xa1\x03\n\x0c\x43heckpointer\x12:\n\x03Put\x12\x18.checkpointer.PutRequest\x1a\x19.checkpointer.PutResponse\x12\x43\n\tPutWrites\x12\x1e.checkpointer.PutWritesRequest\x1a\x16.google.protobuf.Empty\x12=\n\x04List\x12\x19.checkpointer.ListRequest\x1a\x1a.checkpointer.ListResponse\x12I\n\x08GetTuple\x12\x1d.checkpointer.GetTupleRequest\x1a\x1e.checkpointer.GetTupleResponse\x12I\n\x0c\x44\x65leteThread\x12!.checkpointer.DeleteThreadRequest\x1a\x16.google.protobuf.Empty\x12;\n\x05Prune\x12\x1a.checkpointer.PruneRequest\x1a\x16.google.protobuf.EmptyB?Z=github.com/langchain-ai/langgraph-api/core/internal/engine/pbb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'checkpointer_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['_PUTREQUEST_NEWVERSIONSENTRY']._loaded_options = None
+  _globals['_PUTREQUEST_NEWVERSIONSENTRY']._serialized_options = b'8\001'
+  _globals['_PUTREQUEST']._serialized_start=87
+  _globals['_PUTREQUEST']._serialized_end=366
+  _globals['_PUTREQUEST_NEWVERSIONSENTRY']._serialized_start=316
+  _globals['_PUTREQUEST_NEWVERSIONSENTRY']._serialized_end=366
+  _globals['_PUTWRITESREQUEST']._serialized_start=369
+  _globals['_PUTWRITESREQUEST']._serialized_end=512
+  _globals['_LISTREQUEST']._serialized_start=515
+  _globals['_LISTREQUEST']._serialized_end=683
+  _globals['_DELETETHREADREQUEST']._serialized_start=685
+  _globals['_DELETETHREADREQUEST']._serialized_end=725
+  _globals['_PRUNEREQUEST']._serialized_start=728
+  _globals['_PRUNEREQUEST']._serialized_end=889
+  _globals['_PRUNEREQUEST_PRUNESTRATEGY']._serialized_start=824
+  _globals['_PRUNEREQUEST_PRUNESTRATEGY']._serialized_end=889
+  _globals['_GETTUPLEREQUEST']._serialized_start=891
+  _globals['_GETTUPLEREQUEST']._serialized_end=960
+  _globals['_PUTRESPONSE']._serialized_start=962
+  _globals['_PUTRESPONSE']._serialized_end=1032
+  _globals['_LISTRESPONSE']._serialized_start=1034
+  _globals['_LISTRESPONSE']._serialized_end=1106
+  _globals['_GETTUPLERESPONSE']._serialized_start=1108
+  _globals['_GETTUPLERESPONSE']._serialized_end=1209
+  _globals['_CHECKPOINTER']._serialized_start=1212
+  _globals['_CHECKPOINTER']._serialized_end=1629
+# @@protoc_insertion_point(module_scope)

langgraph_api/grpc/generated/checkpointer_pb2.pyi

@@ -0,0 +1,99 @@
+from google.protobuf import empty_pb2 as _empty_pb2
+from . import engine_common_pb2 as _engine_common_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Iterable as _Iterable, Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class PutRequest(_message.Message):
+    __slots__ = ("config", "checkpoint", "metadata", "new_versions")
+    class NewVersionsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    CONFIG_FIELD_NUMBER: _ClassVar[int]
+    CHECKPOINT_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    NEW_VERSIONS_FIELD_NUMBER: _ClassVar[int]
+    config: _engine_common_pb2.EngineRunnableConfig
+    checkpoint: _engine_common_pb2.Checkpoint
+    metadata: _engine_common_pb2.CheckpointMetadata
+    new_versions: _containers.ScalarMap[str, str]
+    def __init__(self, config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., checkpoint: _Optional[_Union[_engine_common_pb2.Checkpoint, _Mapping]] = ..., metadata: _Optional[_Union[_engine_common_pb2.CheckpointMetadata, _Mapping]] = ..., new_versions: _Optional[_Mapping[str, str]] = ...) -> None: ...
+
+class PutWritesRequest(_message.Message):
+    __slots__ = ("config", "writes", "task_id", "task_path")
+    CONFIG_FIELD_NUMBER: _ClassVar[int]
+    WRITES_FIELD_NUMBER: _ClassVar[int]
+    TASK_ID_FIELD_NUMBER: _ClassVar[int]
+    TASK_PATH_FIELD_NUMBER: _ClassVar[int]
+    config: _engine_common_pb2.EngineRunnableConfig
+    writes: _containers.RepeatedCompositeFieldContainer[_engine_common_pb2.Write]
+    task_id: str
+    task_path: str
+    def __init__(self, config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., writes: _Optional[_Iterable[_Union[_engine_common_pb2.Write, _Mapping]]] = ..., task_id: _Optional[str] = ..., task_path: _Optional[str] = ...) -> None: ...
+
+class ListRequest(_message.Message):
+    __slots__ = ("config", "filter_json", "before", "limit")
+    CONFIG_FIELD_NUMBER: _ClassVar[int]
+    FILTER_JSON_FIELD_NUMBER: _ClassVar[int]
+    BEFORE_FIELD_NUMBER: _ClassVar[int]
+    LIMIT_FIELD_NUMBER: _ClassVar[int]
+    config: _engine_common_pb2.EngineRunnableConfig
+    filter_json: bytes
+    before: _engine_common_pb2.EngineRunnableConfig
+    limit: int
+    def __init__(self, config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., filter_json: _Optional[bytes] = ..., before: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., limit: _Optional[int] = ...) -> None: ...
+
+class DeleteThreadRequest(_message.Message):
+    __slots__ = ("thread_id",)
+    THREAD_ID_FIELD_NUMBER: _ClassVar[int]
+    thread_id: str
+    def __init__(self, thread_id: _Optional[str] = ...) -> None: ...
+
+class PruneRequest(_message.Message):
+    __slots__ = ("thread_ids", "strategy")
+    class PruneStrategy(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        UNSPECIFIED: _ClassVar[PruneRequest.PruneStrategy]
+        KEEP_LATEST: _ClassVar[PruneRequest.PruneStrategy]
+        DELETE_ALL: _ClassVar[PruneRequest.PruneStrategy]
+    UNSPECIFIED: PruneRequest.PruneStrategy
+    KEEP_LATEST: PruneRequest.PruneStrategy
+    DELETE_ALL: PruneRequest.PruneStrategy
+    THREAD_IDS_FIELD_NUMBER: _ClassVar[int]
+    STRATEGY_FIELD_NUMBER: _ClassVar[int]
+    thread_ids: _containers.RepeatedScalarFieldContainer[str]
+    strategy: PruneRequest.PruneStrategy
+    def __init__(self, thread_ids: _Optional[_Iterable[str]] = ..., strategy: _Optional[_Union[PruneRequest.PruneStrategy, str]] = ...) -> None: ...
+
+class GetTupleRequest(_message.Message):
+    __slots__ = ("config",)
+    CONFIG_FIELD_NUMBER: _ClassVar[int]
+    config: _engine_common_pb2.EngineRunnableConfig
+    def __init__(self, config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ...) -> None: ...
+
+class PutResponse(_message.Message):
+    __slots__ = ("next_config",)
+    NEXT_CONFIG_FIELD_NUMBER: _ClassVar[int]
+    next_config: _engine_common_pb2.EngineRunnableConfig
+    def __init__(self, next_config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ...) -> None: ...
+
+class ListResponse(_message.Message):
+    __slots__ = ("checkpoint_tuples",)
+    CHECKPOINT_TUPLES_FIELD_NUMBER: _ClassVar[int]
+    checkpoint_tuples: _containers.RepeatedCompositeFieldContainer[_engine_common_pb2.CheckpointTuple]
+    def __init__(self, checkpoint_tuples: _Optional[_Iterable[_Union[_engine_common_pb2.CheckpointTuple, _Mapping]]] = ...) -> None: ...
+
+class GetTupleResponse(_message.Message):
+    __slots__ = ("checkpoint_tuple",)
+    CHECKPOINT_TUPLE_FIELD_NUMBER: _ClassVar[int]
+    checkpoint_tuple: _engine_common_pb2.CheckpointTuple
+    def __init__(self, checkpoint_tuple: _Optional[_Union[_engine_common_pb2.CheckpointTuple, _Mapping]] = ...) -> None: ...

langgraph_api/grpc/generated/checkpointer_pb2_grpc.py

@@ -0,0 +1,329 @@
+# 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 checkpointer_pb2 as checkpointer__pb2
+from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
+
+GRPC_GENERATED_VERSION = '1.75.1'
+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},'
+        + f' but the generated code in checkpointer_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 CheckpointerStub(object):
+    """Checkpoint persistence.
+    API mostly based on the Python specification:
+    https://github.com/langchain-ai/langgraph/tree/df8becd5cfc74bb09ab2d7cbdf17c5559114b8d8/libs/checkpoint#interface
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.Put = channel.unary_unary(
+                '/checkpointer.Checkpointer/Put',
+                request_serializer=checkpointer__pb2.PutRequest.SerializeToString,
+                response_deserializer=checkpointer__pb2.PutResponse.FromString,
+                _registered_method=True)
+        self.PutWrites = channel.unary_unary(
+                '/checkpointer.Checkpointer/PutWrites',
+                request_serializer=checkpointer__pb2.PutWritesRequest.SerializeToString,
+                response_deserializer=google_dot_protobuf_dot_empty__pb2.Empty.FromString,
+                _registered_method=True)
+        self.List = channel.unary_unary(
+                '/checkpointer.Checkpointer/List',
+                request_serializer=checkpointer__pb2.ListRequest.SerializeToString,
+                response_deserializer=checkpointer__pb2.ListResponse.FromString,
+                _registered_method=True)
+        self.GetTuple = channel.unary_unary(
+                '/checkpointer.Checkpointer/GetTuple',
+                request_serializer=checkpointer__pb2.GetTupleRequest.SerializeToString,
+                response_deserializer=checkpointer__pb2.GetTupleResponse.FromString,
+                _registered_method=True)
+        self.DeleteThread = channel.unary_unary(
+                '/checkpointer.Checkpointer/DeleteThread',
+                request_serializer=checkpointer__pb2.DeleteThreadRequest.SerializeToString,
+                response_deserializer=google_dot_protobuf_dot_empty__pb2.Empty.FromString,
+                _registered_method=True)
+        self.Prune = channel.unary_unary(
+                '/checkpointer.Checkpointer/Prune',
+                request_serializer=checkpointer__pb2.PruneRequest.SerializeToString,
+                response_deserializer=google_dot_protobuf_dot_empty__pb2.Empty.FromString,
+                _registered_method=True)
+
+
+class CheckpointerServicer(object):
+    """Checkpoint persistence.
+    API mostly based on the Python specification:
+    https://github.com/langchain-ai/langgraph/tree/df8becd5cfc74bb09ab2d7cbdf17c5559114b8d8/libs/checkpoint#interface
+    """
+
+    def Put(self, request, context):
+        """Put stores a checkpoint with its configuration and metadata.
+        Returns the next config pointing at the saved checkpoint ID.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def PutWrites(self, request, context):
+        """PutWrites stores intermediate writes linked to a checkpoint (pending writes).
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def List(self, request, context):
+        """List returns checkpoints that match a given configuration and filter criteria.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetTuple(self, request, context):
+        """GetTuple fetches a checkpoint tuple for a given configuration (thread_id and checkpoint_id).
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def DeleteThread(self, request, context):
+        """DeleteThread deletes all checkpoints and writes for a thread.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Prune(self, request, context):
+        """Prune deletes checkpoints and related data for a set of threads.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_CheckpointerServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'Put': grpc.unary_unary_rpc_method_handler(
+                    servicer.Put,
+                    request_deserializer=checkpointer__pb2.PutRequest.FromString,
+                    response_serializer=checkpointer__pb2.PutResponse.SerializeToString,
+            ),
+            'PutWrites': grpc.unary_unary_rpc_method_handler(
+                    servicer.PutWrites,
+                    request_deserializer=checkpointer__pb2.PutWritesRequest.FromString,
+                    response_serializer=google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
+            ),
+            'List': grpc.unary_unary_rpc_method_handler(
+                    servicer.List,
+                    request_deserializer=checkpointer__pb2.ListRequest.FromString,
+                    response_serializer=checkpointer__pb2.ListResponse.SerializeToString,
+            ),
+            'GetTuple': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetTuple,
+                    request_deserializer=checkpointer__pb2.GetTupleRequest.FromString,
+                    response_serializer=checkpointer__pb2.GetTupleResponse.SerializeToString,
+            ),
+            'DeleteThread': grpc.unary_unary_rpc_method_handler(
+                    servicer.DeleteThread,
+                    request_deserializer=checkpointer__pb2.DeleteThreadRequest.FromString,
+                    response_serializer=google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
+            ),
+            'Prune': grpc.unary_unary_rpc_method_handler(
+                    servicer.Prune,
+                    request_deserializer=checkpointer__pb2.PruneRequest.FromString,
+                    response_serializer=google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'checkpointer.Checkpointer', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('checkpointer.Checkpointer', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class Checkpointer(object):
+    """Checkpoint persistence.
+    API mostly based on the Python specification:
+    https://github.com/langchain-ai/langgraph/tree/df8becd5cfc74bb09ab2d7cbdf17c5559114b8d8/libs/checkpoint#interface
+    """
+
+    @staticmethod
+    def Put(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,
+            '/checkpointer.Checkpointer/Put',
+            checkpointer__pb2.PutRequest.SerializeToString,
+            checkpointer__pb2.PutResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def PutWrites(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,
+            '/checkpointer.Checkpointer/PutWrites',
+            checkpointer__pb2.PutWritesRequest.SerializeToString,
+            google_dot_protobuf_dot_empty__pb2.Empty.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def List(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,
+            '/checkpointer.Checkpointer/List',
+            checkpointer__pb2.ListRequest.SerializeToString,
+            checkpointer__pb2.ListResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def GetTuple(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,
+            '/checkpointer.Checkpointer/GetTuple',
+            checkpointer__pb2.GetTupleRequest.SerializeToString,
+            checkpointer__pb2.GetTupleResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def DeleteThread(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,
+            '/checkpointer.Checkpointer/DeleteThread',
+            checkpointer__pb2.DeleteThreadRequest.SerializeToString,
+            google_dot_protobuf_dot_empty__pb2.Empty.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def Prune(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,
+            '/checkpointer.Checkpointer/Prune',
+            checkpointer__pb2.PruneRequest.SerializeToString,
+            google_dot_protobuf_dot_empty__pb2.Empty.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)