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

langgraph_api/{grpc_ops → grpc}/generated/engine_common_pb2.pyi

@@ -1,6 +1,7 @@
-from google.protobuf import struct_pb2 as _struct_pb2
 from google.protobuf import wrappers_pb2 as _wrappers_pb2
 from google.protobuf import empty_pb2 as _empty_pb2
+from . import errors_pb2 as _errors_pb2
+from . import enum_durability_pb2 as _enum_durability_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
@@ -10,33 +11,6 @@ from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
 
 DESCRIPTOR: _descriptor.FileDescriptor
 
-class Durability(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
-    __slots__ = ()
-    UNKNOWN: _ClassVar[Durability]
-    ASYNC: _ClassVar[Durability]
-    SYNC: _ClassVar[Durability]
-    EXIT: _ClassVar[Durability]
-UNKNOWN: Durability
-ASYNC: Durability
-SYNC: Durability
-EXIT: Durability
-
-class Value(_message.Message):
-    __slots__ = ("base_value", "sends", "missing", "command")
-    BASE_VALUE_FIELD_NUMBER: _ClassVar[int]
-    SENDS_FIELD_NUMBER: _ClassVar[int]
-    MISSING_FIELD_NUMBER: _ClassVar[int]
-    COMMAND_FIELD_NUMBER: _ClassVar[int]
-    base_value: SerializedValue
-    sends: Sends
-    missing: Missing
-    command: Command
-    def __init__(self, base_value: _Optional[_Union[SerializedValue, _Mapping]] = ..., sends: _Optional[_Union[Sends, _Mapping]] = ..., missing: _Optional[_Union[Missing, _Mapping]] = ..., command: _Optional[_Union[Command, _Mapping]] = ...) -> None: ...
-
-class Missing(_message.Message):
-    __slots__ = ()
-    def __init__(self) -> None: ...
-
 class ChannelValue(_message.Message):
     __slots__ = ("serialized_value", "sends", "missing")
     SERIALIZED_VALUE_FIELD_NUMBER: _ClassVar[int]
@@ -95,10 +69,10 @@ class Channel(_message.Message):
     GET_RESULT_FIELD_NUMBER: _ClassVar[int]
     IS_AVAILABLE_RESULT_FIELD_NUMBER: _ClassVar[int]
     CHECKPOINT_RESULT_FIELD_NUMBER: _ClassVar[int]
-    get_result: Value
+    get_result: ChannelValue
     is_available_result: bool
-    checkpoint_result: Value
-    def __init__(self, get_result: _Optional[_Union[Value, _Mapping]] = ..., is_available_result: bool = ..., checkpoint_result: _Optional[_Union[Value, _Mapping]] = ...) -> None: ...
+    checkpoint_result: ChannelValue
+    def __init__(self, get_result: _Optional[_Union[ChannelValue, _Mapping]] = ..., is_available_result: bool = ..., checkpoint_result: _Optional[_Union[ChannelValue, _Mapping]] = ...) -> None: ...
 
 class Sends(_message.Message):
     __slots__ = ("sends",)
@@ -166,15 +140,11 @@ class GraphBubbleUp(_message.Message):
     __slots__ = ()
     def __init__(self) -> None: ...
 
-class UserCodeExecutionError(_message.Message):
-    __slots__ = ("error_type", "error_message", "traceback")
-    ERROR_TYPE_FIELD_NUMBER: _ClassVar[int]
-    ERROR_MESSAGE_FIELD_NUMBER: _ClassVar[int]
-    TRACEBACK_FIELD_NUMBER: _ClassVar[int]
-    error_type: str
-    error_message: str
-    traceback: str
-    def __init__(self, error_type: _Optional[str] = ..., error_message: _Optional[str] = ..., traceback: _Optional[str] = ...) -> None: ...
+class ParentCommand(_message.Message):
+    __slots__ = ("command",)
+    COMMAND_FIELD_NUMBER: _ClassVar[int]
+    command: Command
+    def __init__(self, command: _Optional[_Union[Command, _Mapping]] = ...) -> None: ...
 
 class GraphInterrupt(_message.Message):
     __slots__ = ("interrupts",)
@@ -190,21 +160,21 @@ class Interrupt(_message.Message):
     id: str
     def __init__(self, value: _Optional[_Union[SerializedValue, _Mapping]] = ..., id: _Optional[str] = ...) -> None: ...
 
-class WrappedInterrupt(_message.Message):
-    __slots__ = ("interrupt", "serialized_interrupt")
-    INTERRUPT_FIELD_NUMBER: _ClassVar[int]
-    SERIALIZED_INTERRUPT_FIELD_NUMBER: _ClassVar[int]
-    interrupt: Interrupt
-    serialized_interrupt: SerializedValue
-    def __init__(self, interrupt: _Optional[_Union[Interrupt, _Mapping]] = ..., serialized_interrupt: _Optional[_Union[SerializedValue, _Mapping]] = ...) -> None: ...
+class WrappedInterrupts(_message.Message):
+    __slots__ = ("interrupts", "serialized_interrupts")
+    INTERRUPTS_FIELD_NUMBER: _ClassVar[int]
+    SERIALIZED_INTERRUPTS_FIELD_NUMBER: _ClassVar[int]
+    interrupts: _containers.RepeatedCompositeFieldContainer[Interrupt]
+    serialized_interrupts: SerializedValue
+    def __init__(self, interrupts: _Optional[_Iterable[_Union[Interrupt, _Mapping]]] = ..., serialized_interrupts: _Optional[_Union[SerializedValue, _Mapping]] = ...) -> None: ...
 
 class Write(_message.Message):
     __slots__ = ("channel", "value")
     CHANNEL_FIELD_NUMBER: _ClassVar[int]
     VALUE_FIELD_NUMBER: _ClassVar[int]
     channel: str
-    value: Value
-    def __init__(self, channel: _Optional[str] = ..., value: _Optional[_Union[Value, _Mapping]] = ...) -> None: ...
+    value: ChannelValue
+    def __init__(self, channel: _Optional[str] = ..., value: _Optional[_Union[ChannelValue, _Mapping]] = ...) -> None: ...
 
 class PendingWrite(_message.Message):
     __slots__ = ("task_id", "channel", "value")
@@ -213,8 +183,8 @@ class PendingWrite(_message.Message):
     VALUE_FIELD_NUMBER: _ClassVar[int]
     task_id: str
     channel: str
-    value: Value
-    def __init__(self, task_id: _Optional[str] = ..., channel: _Optional[str] = ..., value: _Optional[_Union[Value, _Mapping]] = ...) -> None: ...
+    value: ChannelValue
+    def __init__(self, task_id: _Optional[str] = ..., channel: _Optional[str] = ..., value: _Optional[_Union[ChannelValue, _Mapping]] = ...) -> None: ...
 
 class ChannelVersions(_message.Message):
     __slots__ = ("channel_versions",)
@@ -230,7 +200,7 @@ class ChannelVersions(_message.Message):
     def __init__(self, channel_versions: _Optional[_Mapping[str, str]] = ...) -> None: ...
 
 class EngineRunnableConfig(_message.Message):
-    __slots__ = ("tags", "metadata_json", "run_name", "max_concurrency", "recursion_limit", "run_id", "extra_configurable_json", "extra_json", "runtime", "resuming", "task_id", "thread_id", "checkpoint_map", "checkpoint_id", "checkpoint_ns", "durability", "resume_map", "graph_id", "run_attempt", "server_run_id")
+    __slots__ = ("tags", "metadata_json", "run_name", "max_concurrency", "recursion_limit", "run_id", "extra_configurable_json", "extra_json", "runtime", "resuming", "task_id", "thread_id", "checkpoint_map", "checkpoint_id", "checkpoint_ns", "durability", "resume_map", "graph_id", "root_stream_modes", "run_attempt", "server_run_id")
     class MetadataJsonEntry(_message.Message):
         __slots__ = ("key", "value")
         KEY_FIELD_NUMBER: _ClassVar[int]
@@ -284,6 +254,7 @@ class EngineRunnableConfig(_message.Message):
     DURABILITY_FIELD_NUMBER: _ClassVar[int]
     RESUME_MAP_FIELD_NUMBER: _ClassVar[int]
     GRAPH_ID_FIELD_NUMBER: _ClassVar[int]
+    ROOT_STREAM_MODES_FIELD_NUMBER: _ClassVar[int]
     RUN_ATTEMPT_FIELD_NUMBER: _ClassVar[int]
     SERVER_RUN_ID_FIELD_NUMBER: _ClassVar[int]
     tags: _containers.RepeatedScalarFieldContainer[str]
@@ -301,81 +272,110 @@ class EngineRunnableConfig(_message.Message):
     checkpoint_map: _containers.ScalarMap[str, str]
     checkpoint_id: str
     checkpoint_ns: str
-    durability: Durability
+    durability: _enum_durability_pb2.Durability
     resume_map: _containers.MessageMap[str, SerializedValue]
     graph_id: str
+    root_stream_modes: _containers.RepeatedScalarFieldContainer[str]
     run_attempt: int
     server_run_id: str
-    def __init__(self, tags: _Optional[_Iterable[str]] = ..., metadata_json: _Optional[_Mapping[str, bytes]] = ..., run_name: _Optional[str] = ..., max_concurrency: _Optional[int] = ..., recursion_limit: _Optional[int] = ..., run_id: _Optional[str] = ..., extra_configurable_json: _Optional[_Mapping[str, bytes]] = ..., extra_json: _Optional[_Mapping[str, bytes]] = ..., runtime: _Optional[_Union[Runtime, _Mapping]] = ..., resuming: bool = ..., task_id: _Optional[str] = ..., thread_id: _Optional[str] = ..., checkpoint_map: _Optional[_Mapping[str, str]] = ..., checkpoint_id: _Optional[str] = ..., checkpoint_ns: _Optional[str] = ..., durability: _Optional[_Union[Durability, str]] = ..., resume_map: _Optional[_Mapping[str, SerializedValue]] = ..., graph_id: _Optional[str] = ..., run_attempt: _Optional[int] = ..., server_run_id: _Optional[str] = ...) -> None: ...
+    def __init__(self, tags: _Optional[_Iterable[str]] = ..., metadata_json: _Optional[_Mapping[str, bytes]] = ..., run_name: _Optional[str] = ..., max_concurrency: _Optional[int] = ..., recursion_limit: _Optional[int] = ..., run_id: _Optional[str] = ..., extra_configurable_json: _Optional[_Mapping[str, bytes]] = ..., extra_json: _Optional[_Mapping[str, bytes]] = ..., runtime: _Optional[_Union[Runtime, _Mapping]] = ..., resuming: bool = ..., task_id: _Optional[str] = ..., thread_id: _Optional[str] = ..., checkpoint_map: _Optional[_Mapping[str, str]] = ..., checkpoint_id: _Optional[str] = ..., checkpoint_ns: _Optional[str] = ..., durability: _Optional[_Union[_enum_durability_pb2.Durability, str]] = ..., resume_map: _Optional[_Mapping[str, SerializedValue]] = ..., graph_id: _Optional[str] = ..., root_stream_modes: _Optional[_Iterable[str]] = ..., run_attempt: _Optional[int] = ..., server_run_id: _Optional[str] = ...) -> None: ...
 
 class Runtime(_message.Message):
-    __slots__ = ("langgraph_context_json",)
+    __slots__ = ("langgraph_context_json", "previous")
     LANGGRAPH_CONTEXT_JSON_FIELD_NUMBER: _ClassVar[int]
+    PREVIOUS_FIELD_NUMBER: _ClassVar[int]
     langgraph_context_json: bytes
-    def __init__(self, langgraph_context_json: _Optional[bytes] = ...) -> None: ...
+    previous: ChannelValue
+    def __init__(self, langgraph_context_json: _Optional[bytes] = ..., previous: _Optional[_Union[ChannelValue, _Mapping]] = ...) -> None: ...
+
+class Subgraph(_message.Message):
+    __slots__ = ("graph_id",)
+    GRAPH_ID_FIELD_NUMBER: _ClassVar[int]
+    graph_id: str
+    def __init__(self, graph_id: _Optional[str] = ...) -> None: ...
 
 class Task(_message.Message):
-    __slots__ = ("name", "input", "writes", "config", "triggers", "id", "task_path", "writers", "graph_name", "pending_writes", "messages", "stream_subgraphs")
-    class InputEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: Value
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Value, _Mapping]] = ...) -> None: ...
+    __slots__ = ("name", "writes", "config", "triggers", "id", "task_path", "pending_writes", "stream_subgraphs", "subgraph")
     NAME_FIELD_NUMBER: _ClassVar[int]
-    INPUT_FIELD_NUMBER: _ClassVar[int]
     WRITES_FIELD_NUMBER: _ClassVar[int]
     CONFIG_FIELD_NUMBER: _ClassVar[int]
     TRIGGERS_FIELD_NUMBER: _ClassVar[int]
     ID_FIELD_NUMBER: _ClassVar[int]
     TASK_PATH_FIELD_NUMBER: _ClassVar[int]
-    WRITERS_FIELD_NUMBER: _ClassVar[int]
-    GRAPH_NAME_FIELD_NUMBER: _ClassVar[int]
     PENDING_WRITES_FIELD_NUMBER: _ClassVar[int]
-    MESSAGES_FIELD_NUMBER: _ClassVar[int]
     STREAM_SUBGRAPHS_FIELD_NUMBER: _ClassVar[int]
+    SUBGRAPH_FIELD_NUMBER: _ClassVar[int]
     name: str
-    input: _containers.MessageMap[str, Value]
     writes: _containers.RepeatedCompositeFieldContainer[Write]
     config: EngineRunnableConfig
     triggers: _containers.RepeatedScalarFieldContainer[str]
     id: str
     task_path: _containers.RepeatedCompositeFieldContainer[PathSegment]
-    writers: _containers.RepeatedScalarFieldContainer[str]
-    graph_name: str
     pending_writes: _containers.RepeatedCompositeFieldContainer[PendingWrite]
-    messages: _containers.RepeatedCompositeFieldContainer[ChatMessageEnvelope]
     stream_subgraphs: bool
-    def __init__(self, name: _Optional[str] = ..., input: _Optional[_Mapping[str, Value]] = ..., writes: _Optional[_Iterable[_Union[Write, _Mapping]]] = ..., config: _Optional[_Union[EngineRunnableConfig, _Mapping]] = ..., triggers: _Optional[_Iterable[str]] = ..., id: _Optional[str] = ..., task_path: _Optional[_Iterable[_Union[PathSegment, _Mapping]]] = ..., writers: _Optional[_Iterable[str]] = ..., graph_name: _Optional[str] = ..., pending_writes: _Optional[_Iterable[_Union[PendingWrite, _Mapping]]] = ..., messages: _Optional[_Iterable[_Union[ChatMessageEnvelope, _Mapping]]] = ..., stream_subgraphs: bool = ...) -> None: ...
+    subgraph: Subgraph
+    def __init__(self, name: _Optional[str] = ..., writes: _Optional[_Iterable[_Union[Write, _Mapping]]] = ..., config: _Optional[_Union[EngineRunnableConfig, _Mapping]] = ..., triggers: _Optional[_Iterable[str]] = ..., id: _Optional[str] = ..., task_path: _Optional[_Iterable[_Union[PathSegment, _Mapping]]] = ..., pending_writes: _Optional[_Iterable[_Union[PendingWrite, _Mapping]]] = ..., stream_subgraphs: bool = ..., subgraph: _Optional[_Union[Subgraph, _Mapping]] = ...) -> None: ...
+
+class TaskResult(_message.Message):
+    __slots__ = ("user_error", "interrupts", "parent_command", "writes")
+    USER_ERROR_FIELD_NUMBER: _ClassVar[int]
+    INTERRUPTS_FIELD_NUMBER: _ClassVar[int]
+    PARENT_COMMAND_FIELD_NUMBER: _ClassVar[int]
+    WRITES_FIELD_NUMBER: _ClassVar[int]
+    user_error: _errors_pb2.UserCodeExecutionError
+    interrupts: WrappedInterrupts
+    parent_command: ParentCommand
+    writes: _containers.RepeatedCompositeFieldContainer[Write]
+    def __init__(self, user_error: _Optional[_Union[_errors_pb2.UserCodeExecutionError, _Mapping]] = ..., interrupts: _Optional[_Union[WrappedInterrupts, _Mapping]] = ..., parent_command: _Optional[_Union[ParentCommand, _Mapping]] = ..., writes: _Optional[_Iterable[_Union[Write, _Mapping]]] = ...) -> None: ...
+
+class CheckpointRef(_message.Message):
+    __slots__ = ("checkpoint_id", "thread_id", "checkpoint_ns", "checkpoint_map")
+    class CheckpointMapEntry(_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: ...
+    CHECKPOINT_ID_FIELD_NUMBER: _ClassVar[int]
+    THREAD_ID_FIELD_NUMBER: _ClassVar[int]
+    CHECKPOINT_NS_FIELD_NUMBER: _ClassVar[int]
+    CHECKPOINT_MAP_FIELD_NUMBER: _ClassVar[int]
+    checkpoint_id: str
+    thread_id: str
+    checkpoint_ns: str
+    checkpoint_map: _containers.ScalarMap[str, str]
+    def __init__(self, checkpoint_id: _Optional[str] = ..., thread_id: _Optional[str] = ..., checkpoint_ns: _Optional[str] = ..., checkpoint_map: _Optional[_Mapping[str, str]] = ...) -> None: ...
 
 class PregelTaskSnapshot(_message.Message):
-    __slots__ = ("id", "name", "path", "interrupts", "state", "result", "error")
+    __slots__ = ("id", "name", "path", "interrupts", "result_json", "error", "checkpoint_ref", "state_snapshot")
     ID_FIELD_NUMBER: _ClassVar[int]
     NAME_FIELD_NUMBER: _ClassVar[int]
     PATH_FIELD_NUMBER: _ClassVar[int]
     INTERRUPTS_FIELD_NUMBER: _ClassVar[int]
-    STATE_FIELD_NUMBER: _ClassVar[int]
-    RESULT_FIELD_NUMBER: _ClassVar[int]
+    RESULT_JSON_FIELD_NUMBER: _ClassVar[int]
     ERROR_FIELD_NUMBER: _ClassVar[int]
+    CHECKPOINT_REF_FIELD_NUMBER: _ClassVar[int]
+    STATE_SNAPSHOT_FIELD_NUMBER: _ClassVar[int]
     id: str
     name: str
     path: _containers.RepeatedCompositeFieldContainer[PathSegment]
     interrupts: _containers.RepeatedCompositeFieldContainer[Interrupt]
-    state: _struct_pb2.Struct
-    result: _struct_pb2.Struct
+    result_json: bytes
     error: SerializedValue
-    def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., path: _Optional[_Iterable[_Union[PathSegment, _Mapping]]] = ..., interrupts: _Optional[_Iterable[_Union[Interrupt, _Mapping]]] = ..., state: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., result: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., error: _Optional[_Union[SerializedValue, _Mapping]] = ...) -> None: ...
+    checkpoint_ref: CheckpointRef
+    state_snapshot: StateSnapshot
+    def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., path: _Optional[_Iterable[_Union[PathSegment, _Mapping]]] = ..., interrupts: _Optional[_Iterable[_Union[Interrupt, _Mapping]]] = ..., result_json: _Optional[bytes] = ..., error: _Optional[_Union[SerializedValue, _Mapping]] = ..., checkpoint_ref: _Optional[_Union[CheckpointRef, _Mapping]] = ..., state_snapshot: _Optional[_Union[StateSnapshot, _Mapping]] = ...) -> None: ...
 
 class Checkpoint(_message.Message):
-    __slots__ = ("v", "id", "channel_values", "channel_versions", "versions_seen", "ts")
+    __slots__ = ("v", "id", "channel_values", "channel_versions", "versions_seen", "ts", "updated_channels")
     class ChannelValuesEntry(_message.Message):
         __slots__ = ("key", "value")
         KEY_FIELD_NUMBER: _ClassVar[int]
         VALUE_FIELD_NUMBER: _ClassVar[int]
         key: str
-        value: Value
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Value, _Mapping]] = ...) -> None: ...
+        value: ChannelValue
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[ChannelValue, _Mapping]] = ...) -> None: ...
     class ChannelVersionsEntry(_message.Message):
         __slots__ = ("key", "value")
         KEY_FIELD_NUMBER: _ClassVar[int]
@@ -396,13 +396,15 @@ class Checkpoint(_message.Message):
     CHANNEL_VERSIONS_FIELD_NUMBER: _ClassVar[int]
     VERSIONS_SEEN_FIELD_NUMBER: _ClassVar[int]
     TS_FIELD_NUMBER: _ClassVar[int]
+    UPDATED_CHANNELS_FIELD_NUMBER: _ClassVar[int]
     v: int
     id: str
-    channel_values: _containers.MessageMap[str, Value]
+    channel_values: _containers.MessageMap[str, ChannelValue]
     channel_versions: _containers.ScalarMap[str, str]
     versions_seen: _containers.MessageMap[str, ChannelVersions]
     ts: str
-    def __init__(self, v: _Optional[int] = ..., id: _Optional[str] = ..., channel_values: _Optional[_Mapping[str, Value]] = ..., channel_versions: _Optional[_Mapping[str, str]] = ..., versions_seen: _Optional[_Mapping[str, ChannelVersions]] = ..., ts: _Optional[str] = ...) -> None: ...
+    updated_channels: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, v: _Optional[int] = ..., id: _Optional[str] = ..., channel_values: _Optional[_Mapping[str, ChannelValue]] = ..., channel_versions: _Optional[_Mapping[str, str]] = ..., versions_seen: _Optional[_Mapping[str, ChannelVersions]] = ..., ts: _Optional[str] = ..., updated_channels: _Optional[_Iterable[str]] = ...) -> None: ...
 
 class CheckpointMetadata(_message.Message):
     __slots__ = ("source", "step", "parents", "run_id")
@@ -460,14 +462,14 @@ class Updates(_message.Message):
     def __init__(self, checkpoint: _Optional[_Union[Checkpoint, _Mapping]] = ..., channels: _Optional[_Union[Channels, _Mapping]] = ..., updated_channels: _Optional[_Iterable[str]] = ...) -> None: ...
 
 class ToolCall(_message.Message):
-    __slots__ = ("name", "args", "id")
+    __slots__ = ("name", "args_json", "id")
     NAME_FIELD_NUMBER: _ClassVar[int]
-    ARGS_FIELD_NUMBER: _ClassVar[int]
+    ARGS_JSON_FIELD_NUMBER: _ClassVar[int]
     ID_FIELD_NUMBER: _ClassVar[int]
     name: str
-    args: _struct_pb2.Struct
+    args_json: bytes
     id: _wrappers_pb2.StringValue
-    def __init__(self, name: _Optional[str] = ..., args: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., id: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
+    def __init__(self, name: _Optional[str] = ..., args_json: _Optional[bytes] = ..., id: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
 
 class ToolCallChunk(_message.Message):
     __slots__ = ("name", "args_json", "id", "index")
@@ -527,40 +529,97 @@ class UsageMetadata(_message.Message):
 
 class ResponseMetadata(_message.Message):
     __slots__ = ("data",)
+    class DataEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: bytes
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[bytes] = ...) -> None: ...
     DATA_FIELD_NUMBER: _ClassVar[int]
-    data: _struct_pb2.Struct
-    def __init__(self, data: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
+    data: _containers.ScalarMap[str, bytes]
+    def __init__(self, data: _Optional[_Mapping[str, bytes]] = ...) -> None: ...
 
 class AIFields(_message.Message):
-    __slots__ = ("usage_metadata", "response_metadata", "tool_calls", "tool_call_chunks", "invalid_tool_calls", "reasoning_content")
+    __slots__ = ("usage_metadata", "response_metadata", "tool_calls", "tool_call_chunks", "invalid_tool_calls", "chunk_position", "reasoning_content")
     USAGE_METADATA_FIELD_NUMBER: _ClassVar[int]
     RESPONSE_METADATA_FIELD_NUMBER: _ClassVar[int]
     TOOL_CALLS_FIELD_NUMBER: _ClassVar[int]
     TOOL_CALL_CHUNKS_FIELD_NUMBER: _ClassVar[int]
     INVALID_TOOL_CALLS_FIELD_NUMBER: _ClassVar[int]
+    CHUNK_POSITION_FIELD_NUMBER: _ClassVar[int]
     REASONING_CONTENT_FIELD_NUMBER: _ClassVar[int]
     usage_metadata: UsageMetadata
     response_metadata: ResponseMetadata
     tool_calls: _containers.RepeatedCompositeFieldContainer[ToolCall]
     tool_call_chunks: _containers.RepeatedCompositeFieldContainer[ToolCallChunk]
     invalid_tool_calls: _containers.RepeatedCompositeFieldContainer[InvalidToolCall]
+    chunk_position: str
     reasoning_content: _wrappers_pb2.StringValue
-    def __init__(self, usage_metadata: _Optional[_Union[UsageMetadata, _Mapping]] = ..., response_metadata: _Optional[_Union[ResponseMetadata, _Mapping]] = ..., tool_calls: _Optional[_Iterable[_Union[ToolCall, _Mapping]]] = ..., tool_call_chunks: _Optional[_Iterable[_Union[ToolCallChunk, _Mapping]]] = ..., invalid_tool_calls: _Optional[_Iterable[_Union[InvalidToolCall, _Mapping]]] = ..., reasoning_content: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
+    def __init__(self, usage_metadata: _Optional[_Union[UsageMetadata, _Mapping]] = ..., response_metadata: _Optional[_Union[ResponseMetadata, _Mapping]] = ..., tool_calls: _Optional[_Iterable[_Union[ToolCall, _Mapping]]] = ..., tool_call_chunks: _Optional[_Iterable[_Union[ToolCallChunk, _Mapping]]] = ..., invalid_tool_calls: _Optional[_Iterable[_Union[InvalidToolCall, _Mapping]]] = ..., chunk_position: _Optional[str] = ..., reasoning_content: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
 
 class ToolFields(_message.Message):
-    __slots__ = ("tool_call_id", "status")
+    __slots__ = ("tool_call_id", "status", "artifact")
     TOOL_CALL_ID_FIELD_NUMBER: _ClassVar[int]
     STATUS_FIELD_NUMBER: _ClassVar[int]
+    ARTIFACT_FIELD_NUMBER: _ClassVar[int]
     tool_call_id: _wrappers_pb2.StringValue
     status: str
-    def __init__(self, tool_call_id: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., status: _Optional[str] = ...) -> None: ...
+    artifact: SerializedValue
+    def __init__(self, tool_call_id: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., status: _Optional[str] = ..., artifact: _Optional[_Union[SerializedValue, _Mapping]] = ...) -> None: ...
 
 class HumanFields(_message.Message):
     __slots__ = ()
     def __init__(self) -> None: ...
 
+class Content(_message.Message):
+    __slots__ = ("text", "blocks")
+    TEXT_FIELD_NUMBER: _ClassVar[int]
+    BLOCKS_FIELD_NUMBER: _ClassVar[int]
+    text: str
+    blocks: ContentBlocks
+    def __init__(self, text: _Optional[str] = ..., blocks: _Optional[_Union[ContentBlocks, _Mapping]] = ...) -> None: ...
+
+class ContentBlocks(_message.Message):
+    __slots__ = ("items",)
+    ITEMS_FIELD_NUMBER: _ClassVar[int]
+    items: _containers.RepeatedCompositeFieldContainer[ContentBlock]
+    def __init__(self, items: _Optional[_Iterable[_Union[ContentBlock, _Mapping]]] = ...) -> None: ...
+
+class ContentBlock(_message.Message):
+    __slots__ = ("text", "structured")
+    TEXT_FIELD_NUMBER: _ClassVar[int]
+    STRUCTURED_FIELD_NUMBER: _ClassVar[int]
+    text: str
+    structured: StructuredBlock
+    def __init__(self, text: _Optional[str] = ..., structured: _Optional[_Union[StructuredBlock, _Mapping]] = ...) -> None: ...
+
+class StructuredBlock(_message.Message):
+    __slots__ = ("index", "type", "data_json")
+    INDEX_FIELD_NUMBER: _ClassVar[int]
+    TYPE_FIELD_NUMBER: _ClassVar[int]
+    DATA_JSON_FIELD_NUMBER: _ClassVar[int]
+    index: int
+    type: str
+    data_json: bytes
+    def __init__(self, index: _Optional[int] = ..., type: _Optional[str] = ..., data_json: _Optional[bytes] = ...) -> None: ...
+
 class ChatMessage(_message.Message):
     __slots__ = ("id", "name", "type", "content", "additional_kwargs", "ai", "tool", "human", "extensions")
+    class AdditionalKwargsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: bytes
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[bytes] = ...) -> None: ...
+    class ExtensionsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: bytes
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[bytes] = ...) -> None: ...
     ID_FIELD_NUMBER: _ClassVar[int]
     NAME_FIELD_NUMBER: _ClassVar[int]
     TYPE_FIELD_NUMBER: _ClassVar[int]
@@ -573,29 +632,31 @@ class ChatMessage(_message.Message):
     id: str
     name: str
     type: str
-    content: _struct_pb2.Value
-    additional_kwargs: _struct_pb2.Struct
+    content: Content
+    additional_kwargs: _containers.ScalarMap[str, bytes]
     ai: AIFields
     tool: ToolFields
     human: HumanFields
-    extensions: _struct_pb2.Struct
-    def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., type: _Optional[str] = ..., content: _Optional[_Union[_struct_pb2.Value, _Mapping]] = ..., additional_kwargs: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., ai: _Optional[_Union[AIFields, _Mapping]] = ..., tool: _Optional[_Union[ToolFields, _Mapping]] = ..., human: _Optional[_Union[HumanFields, _Mapping]] = ..., extensions: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
+    extensions: _containers.ScalarMap[str, bytes]
+    def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., type: _Optional[str] = ..., content: _Optional[_Union[Content, _Mapping]] = ..., additional_kwargs: _Optional[_Mapping[str, bytes]] = ..., ai: _Optional[_Union[AIFields, _Mapping]] = ..., tool: _Optional[_Union[ToolFields, _Mapping]] = ..., human: _Optional[_Union[HumanFields, _Mapping]] = ..., extensions: _Optional[_Mapping[str, bytes]] = ...) -> None: ...
 
 class ChatMessageEnvelope(_message.Message):
-    __slots__ = ("is_streaming_chunk", "namespace", "message", "metadata")
+    __slots__ = ("is_streaming_chunk", "namespace", "message", "metadata", "node_name")
     IS_STREAMING_CHUNK_FIELD_NUMBER: _ClassVar[int]
     NAMESPACE_FIELD_NUMBER: _ClassVar[int]
     MESSAGE_FIELD_NUMBER: _ClassVar[int]
     METADATA_FIELD_NUMBER: _ClassVar[int]
+    NODE_NAME_FIELD_NUMBER: _ClassVar[int]
     is_streaming_chunk: bool
     namespace: _containers.RepeatedScalarFieldContainer[str]
     message: ChatMessage
     metadata: bytes
-    def __init__(self, is_streaming_chunk: bool = ..., namespace: _Optional[_Iterable[str]] = ..., message: _Optional[_Union[ChatMessage, _Mapping]] = ..., metadata: _Optional[bytes] = ...) -> None: ...
+    node_name: str
+    def __init__(self, is_streaming_chunk: bool = ..., namespace: _Optional[_Iterable[str]] = ..., message: _Optional[_Union[ChatMessage, _Mapping]] = ..., metadata: _Optional[bytes] = ..., node_name: _Optional[str] = ...) -> None: ...
 
 class StateSnapshot(_message.Message):
-    __slots__ = ("values", "next", "config", "metadata", "created_at", "parent_config", "tasks", "interrupts")
-    VALUES_FIELD_NUMBER: _ClassVar[int]
+    __slots__ = ("values_json", "next", "config", "metadata", "created_at", "parent_config", "tasks", "interrupts")
+    VALUES_JSON_FIELD_NUMBER: _ClassVar[int]
     NEXT_FIELD_NUMBER: _ClassVar[int]
     CONFIG_FIELD_NUMBER: _ClassVar[int]
     METADATA_FIELD_NUMBER: _ClassVar[int]
@@ -603,7 +664,7 @@ class StateSnapshot(_message.Message):
     PARENT_CONFIG_FIELD_NUMBER: _ClassVar[int]
     TASKS_FIELD_NUMBER: _ClassVar[int]
     INTERRUPTS_FIELD_NUMBER: _ClassVar[int]
-    values: _struct_pb2.Struct
+    values_json: bytes
     next: _containers.RepeatedScalarFieldContainer[str]
     config: EngineRunnableConfig
     metadata: CheckpointMetadata
@@ -611,7 +672,7 @@ class StateSnapshot(_message.Message):
     parent_config: EngineRunnableConfig
     tasks: _containers.RepeatedCompositeFieldContainer[PregelTaskSnapshot]
     interrupts: _containers.RepeatedCompositeFieldContainer[Interrupt]
-    def __init__(self, values: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., next: _Optional[_Iterable[str]] = ..., config: _Optional[_Union[EngineRunnableConfig, _Mapping]] = ..., metadata: _Optional[_Union[CheckpointMetadata, _Mapping]] = ..., created_at: _Optional[str] = ..., parent_config: _Optional[_Union[EngineRunnableConfig, _Mapping]] = ..., tasks: _Optional[_Iterable[_Union[PregelTaskSnapshot, _Mapping]]] = ..., interrupts: _Optional[_Iterable[_Union[Interrupt, _Mapping]]] = ...) -> None: ...
+    def __init__(self, values_json: _Optional[bytes] = ..., next: _Optional[_Iterable[str]] = ..., config: _Optional[_Union[EngineRunnableConfig, _Mapping]] = ..., metadata: _Optional[_Union[CheckpointMetadata, _Mapping]] = ..., created_at: _Optional[str] = ..., parent_config: _Optional[_Union[EngineRunnableConfig, _Mapping]] = ..., tasks: _Optional[_Iterable[_Union[PregelTaskSnapshot, _Mapping]]] = ..., interrupts: _Optional[_Iterable[_Union[Interrupt, _Mapping]]] = ...) -> None: ...
 
 class StateUpdate(_message.Message):
     __slots__ = ("values", "as_node", "task_id")
@@ -646,3 +707,16 @@ class PathSegment(_message.Message):
     int_value: int
     bool_value: bool
     def __init__(self, string_value: _Optional[str] = ..., int_value: _Optional[int] = ..., bool_value: bool = ...) -> None: ...
+
+class StaticInterruptConfig(_message.Message):
+    __slots__ = ("all", "node_names")
+    class NodeNames(_message.Message):
+        __slots__ = ("names",)
+        NAMES_FIELD_NUMBER: _ClassVar[int]
+        names: _containers.RepeatedScalarFieldContainer[str]
+        def __init__(self, names: _Optional[_Iterable[str]] = ...) -> None: ...
+    ALL_FIELD_NUMBER: _ClassVar[int]
+    NODE_NAMES_FIELD_NUMBER: _ClassVar[int]
+    all: bool
+    node_names: StaticInterruptConfig.NodeNames
+    def __init__(self, all: bool = ..., node_names: _Optional[_Union[StaticInterruptConfig.NodeNames, _Mapping]] = ...) -> None: ...

langgraph_api/grpc/generated/enum_cancel_run_action_pb2.py

@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: enum_cancel_run_action.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,
+    '',
+    'enum_cancel_run_action.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1c\x65num_cancel_run_action.proto\x12\x13\x65numCancelRunAction*.\n\x0f\x43\x61ncelRunAction\x12\r\n\tinterrupt\x10\x00\x12\x0c\n\x08rollback\x10\x01\x42;Z9github.com/langchain-ai/langgraph-api/core/internal/enumsb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'enum_cancel_run_action_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z9github.com/langchain-ai/langgraph-api/core/internal/enums'
+  _globals['_CANCELRUNACTION']._serialized_start=53
+  _globals['_CANCELRUNACTION']._serialized_end=99
+# @@protoc_insertion_point(module_scope)

langgraph_api/grpc/generated/enum_cancel_run_action_pb2.pyi

@@ -0,0 +1,12 @@
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from typing import ClassVar as _ClassVar
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class CancelRunAction(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    interrupt: _ClassVar[CancelRunAction]
+    rollback: _ClassVar[CancelRunAction]
+interrupt: CancelRunAction
+rollback: CancelRunAction

langgraph_api/grpc/generated/enum_cancel_run_action_pb2_grpc.py

@@ -0,0 +1,24 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+
+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 enum_cancel_run_action_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}.'
+    )

langgraph_api/grpc/generated/enum_control_signal_pb2.py

@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: enum_control_signal.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,
+    '',
+    'enum_control_signal.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x19\x65num_control_signal.proto\x12\x11\x65numControlSignal*C\n\rControlSignal\x12\x0b\n\x07unknown\x10\x00\x12\r\n\tinterrupt\x10\x01\x12\x0c\n\x08rollback\x10\x02\x12\x08\n\x04\x64one\x10\x03\x42;Z9github.com/langchain-ai/langgraph-api/core/internal/enumsb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'enum_control_signal_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z9github.com/langchain-ai/langgraph-api/core/internal/enums'
+  _globals['_CONTROLSIGNAL']._serialized_start=48
+  _globals['_CONTROLSIGNAL']._serialized_end=115
+# @@protoc_insertion_point(module_scope)

langgraph_api/grpc/generated/enum_control_signal_pb2.pyi

@@ -0,0 +1,16 @@
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from typing import ClassVar as _ClassVar
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class ControlSignal(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    unknown: _ClassVar[ControlSignal]
+    interrupt: _ClassVar[ControlSignal]
+    rollback: _ClassVar[ControlSignal]
+    done: _ClassVar[ControlSignal]
+unknown: ControlSignal
+interrupt: ControlSignal
+rollback: ControlSignal
+done: ControlSignal

langgraph_api/grpc/generated/enum_control_signal_pb2_grpc.py

@@ -0,0 +1,24 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+
+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 enum_control_signal_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}.'
+    )