langgraph-executor 0.0.1a5__py3-none-any.whl → 0.0.1a7__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (25) hide show
  1. langgraph_executor/__init__.py +1 -1
  2. langgraph_executor/client/__init__.py +0 -0
  3. langgraph_executor/client/patch.py +305 -0
  4. langgraph_executor/client/utils.py +340 -0
  5. langgraph_executor/common.py +65 -16
  6. langgraph_executor/executor.py +22 -8
  7. langgraph_executor/executor_base.py +91 -3
  8. langgraph_executor/extract_graph.py +19 -26
  9. langgraph_executor/pb/executor_pb2.py +45 -43
  10. langgraph_executor/pb/executor_pb2.pyi +39 -0
  11. langgraph_executor/pb/executor_pb2_grpc.py +44 -0
  12. langgraph_executor/pb/executor_pb2_grpc.pyi +20 -0
  13. langgraph_executor/pb/graph_pb2.py +12 -12
  14. langgraph_executor/pb/graph_pb2.pyi +13 -14
  15. langgraph_executor/pb/runtime_pb2.py +49 -31
  16. langgraph_executor/pb/runtime_pb2.pyi +197 -7
  17. langgraph_executor/pb/runtime_pb2_grpc.py +229 -1
  18. langgraph_executor/pb/runtime_pb2_grpc.pyi +133 -6
  19. langgraph_executor/pb/types_pb2.py +17 -9
  20. langgraph_executor/pb/types_pb2.pyi +95 -0
  21. langgraph_executor/serde.py +13 -0
  22. {langgraph_executor-0.0.1a5.dist-info → langgraph_executor-0.0.1a7.dist-info}/METADATA +1 -1
  23. langgraph_executor-0.0.1a7.dist-info/RECORD +36 -0
  24. langgraph_executor-0.0.1a5.dist-info/RECORD +0 -32
  25. {langgraph_executor-0.0.1a5.dist-info → langgraph_executor-0.0.1a7.dist-info}/WHEEL +0 -0
langgraph_executor/pb/executor_pb2_grpc.py CHANGED
@@ -70,6 +70,11 @@ class LangGraphExecutorStub(object):
70
70
  request_serializer=executor__pb2.GenerateCacheKeyRequest.SerializeToString,
71
71
  response_deserializer=executor__pb2.GenerateCacheKeyResponse.FromString,
72
72
  _registered_method=True)
73
+ self.StateUpdate = channel.unary_unary(
74
+ '/executor.LangGraphExecutor/StateUpdate',
75
+ request_serializer=executor__pb2.StateUpdateRequest.SerializeToString,
76
+ response_deserializer=executor__pb2.TaskResult.FromString,
77
+ _registered_method=True)
73
78
 
74
79
 
75
80
  class LangGraphExecutorServicer(object):
@@ -125,6 +130,13 @@ class LangGraphExecutorServicer(object):
125
130
  context.set_details('Method not implemented!')
126
131
  raise NotImplementedError('Method not implemented!')
127
132
 
133
+ def StateUpdate(self, request, context):
134
+ """State update
135
+ """
136
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
137
+ context.set_details('Method not implemented!')
138
+ raise NotImplementedError('Method not implemented!')
139
+
128
140
 
129
141
  def add_LangGraphExecutorServicer_to_server(servicer, server):
130
142
  rpc_method_handlers = {
@@ -163,6 +175,11 @@ def add_LangGraphExecutorServicer_to_server(servicer, server):
163
175
  request_deserializer=executor__pb2.GenerateCacheKeyRequest.FromString,
164
176
  response_serializer=executor__pb2.GenerateCacheKeyResponse.SerializeToString,
165
177
  ),
178
+ 'StateUpdate': grpc.unary_unary_rpc_method_handler(
179
+ servicer.StateUpdate,
180
+ request_deserializer=executor__pb2.StateUpdateRequest.FromString,
181
+ response_serializer=executor__pb2.TaskResult.SerializeToString,
182
+ ),
166
183
  }
167
184
  generic_handler = grpc.method_handlers_generic_handler(
168
185
  'executor.LangGraphExecutor', rpc_method_handlers)
@@ -363,3 +380,30 @@ class LangGraphExecutor(object):
363
380
  timeout,
364
381
  metadata,
365
382
  _registered_method=True)
383
+
384
+ @staticmethod
385
+ def StateUpdate(request,
386
+ target,
387
+ options=(),
388
+ channel_credentials=None,
389
+ call_credentials=None,
390
+ insecure=False,
391
+ compression=None,
392
+ wait_for_ready=None,
393
+ timeout=None,
394
+ metadata=None):
395
+ return grpc.experimental.unary_unary(
396
+ request,
397
+ target,
398
+ '/executor.LangGraphExecutor/StateUpdate',
399
+ executor__pb2.StateUpdateRequest.SerializeToString,
400
+ executor__pb2.TaskResult.FromString,
401
+ options,
402
+ channel_credentials,
403
+ insecure,
404
+ call_credentials,
405
+ compression,
406
+ wait_for_ready,
407
+ timeout,
408
+ metadata,
409
+ _registered_method=True)
langgraph_executor/pb/executor_pb2_grpc.pyi CHANGED
@@ -63,6 +63,12 @@ class LangGraphExecutorStub:
63
63
  ]
64
64
  """Generate cache key for a node execution"""
65
65
 
66
+ StateUpdate: grpc.UnaryUnaryMultiCallable[
67
+ executor_pb2.StateUpdateRequest,
68
+ executor_pb2.TaskResult,
69
+ ]
70
+ """State update"""
71
+
66
72
  class LangGraphExecutorAsyncStub:
67
73
  """Main service for LangGraph runtime operations"""
68
74
 
@@ -108,6 +114,12 @@ class LangGraphExecutorAsyncStub:
108
114
  ]
109
115
  """Generate cache key for a node execution"""
110
116
 
117
+ StateUpdate: grpc.aio.UnaryUnaryMultiCallable[
118
+ executor_pb2.StateUpdateRequest,
119
+ executor_pb2.TaskResult,
120
+ ]
121
+ """State update"""
122
+
111
123
  class LangGraphExecutorServicer(metaclass=abc.ABCMeta):
112
124
  """Main service for LangGraph runtime operations"""
113
125
 
@@ -167,4 +179,12 @@ class LangGraphExecutorServicer(metaclass=abc.ABCMeta):
167
179
  ) -> typing.Union[executor_pb2.GenerateCacheKeyResponse, collections.abc.Awaitable[executor_pb2.GenerateCacheKeyResponse]]:
168
180
  """Generate cache key for a node execution"""
169
181
 
182
+ @abc.abstractmethod
183
+ def StateUpdate(
184
+ self,
185
+ request: executor_pb2.StateUpdateRequest,
186
+ context: _ServicerContext,
187
+ ) -> typing.Union[executor_pb2.TaskResult, collections.abc.Awaitable[executor_pb2.TaskResult]]:
188
+ """State update"""
189
+
170
190
  def add_LangGraphExecutorServicer_to_server(servicer: LangGraphExecutorServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...
langgraph_executor/pb/graph_pb2.py CHANGED
@@ -26,7 +26,7 @@ from . import types_pb2 as types__pb2
26
26
  from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
27
27
 
28
28
 
29
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0bgraph.proto\x12\x05graph\x1a\x0btypes.proto\x1a\x1cgoogle/protobuf/struct.proto\"\x84\x05\n\x0fGraphDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12$\n\x05nodes\x18\x02 \x03(\x0b\x32\x15.graph.NodeDefinition\x12!\n\x08\x63hannels\x18\x03 \x01(\x0b\x32\x0f.types.Channels\x12\x16\n\x0einput_channels\x18\x04 \x03(\t\x12\x17\n\x0foutput_channels\x18\x05 \x03(\t\x12\x1e\n\x16input_channels_is_list\x18\x06 \x01(\x08\x12\x1f\n\x17output_channels_is_list\x18\x07 \x01(\x08\x12\x1e\n\x16interrupt_before_nodes\x18\x08 \x03(\t\x12\x1d\n\x15interrupt_after_nodes\x18\t \x03(\t\x12\x44\n\x10trigger_to_nodes\x18\n \x03(\x0b\x32*.graph.GraphDefinition.TriggerToNodesEntry\x12\x13\n\x0bstream_mode\x18\x0b \x03(\t\x12\x14\n\x0cstream_eager\x18\x0c \x01(\x08\x12\x17\n\x0fstream_channels\x18\r \x03(\t\x12\x14\n\x0cstep_timeout\x18\x0e \x01(\x02\x12\r\n\x05\x64\x65\x62ug\x18\x0f \x01(\x08\x12\x1b\n\x05\x63\x61\x63he\x18\x10 \x01(\x0b\x32\x0c.graph.Cache\x12(\n\x0cretry_policy\x18\x11 \x01(\x0b\x32\x12.graph.RetryPolicy\x12%\n\x06\x63onfig\x18\x12 \x01(\x0b\x32\x15.types.RunnableConfig\x1aL\n\x13TriggerToNodesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12$\n\x05value\x18\x02 \x01(\x0b\x32\x15.graph.TriggerMapping:\x02\x38\x01\"\x8c\x01\n\x0eNodeDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x10\n\x08triggers\x18\x02 \x03(\t\x12\x0f\n\x07writers\x18\x03 \x03(\t\x12\x0c\n\x04tags\x18\x04 \x03(\t\x12)\n\x08metadata\x18\x05 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x10\n\x08\x63hannels\x18\x06 \x03(\t\"{\n\x0bRetryPolicy\x12\x18\n\x10initial_interval\x18\x01 \x01(\x02\x12\x16\n\x0e\x62\x61\x63koff_factor\x18\x02 \x01(\x02\x12\x14\n\x0cmax_interval\x18\x03 \x01(\x02\x12\x14\n\x0cmax_attempts\x18\x04 \x01(\x05\x12\x0e\n\x06jitter\x18\x05 \x01(\x08\"\x1b\n\x05\x43\x61\x63he\x12\x12\n\ncache_type\x18\x01 \x01(\t\"\x1f\n\x0eTriggerMapping\x12\r\n\x05nodes\x18\x01 \x03(\tB1Z/github.com/langchain-ai/langgraph-go/runtime/pbb\x06proto3')
29
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0bgraph.proto\x12\x05graph\x1a\x0btypes.proto\x1a\x1cgoogle/protobuf/struct.proto\"\xd2\x05\n\x0fGraphDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12$\n\x05nodes\x18\x02 \x03(\x0b\x32\x15.graph.NodeDefinition\x12!\n\x08\x63hannels\x18\x03 \x01(\x0b\x32\x0f.types.Channels\x12,\n\x0einput_channels\x18\x04 \x01(\x0b\x32\x14.types.StringOrSlice\x12-\n\x0foutput_channels\x18\x05 \x01(\x0b\x32\x14.types.StringOrSlice\x12\x1e\n\x16interrupt_before_nodes\x18\x08 \x03(\t\x12\x1d\n\x15interrupt_after_nodes\x18\t \x03(\t\x12\x44\n\x10trigger_to_nodes\x18\n \x03(\x0b\x32*.graph.GraphDefinition.TriggerToNodesEntry\x12\x13\n\x0bstream_mode\x18\x0b \x03(\t\x12\x14\n\x0cstream_eager\x18\x0c \x01(\x08\x12\x32\n\x0fstream_channels\x18\r \x01(\x0b\x32\x14.types.StringOrSliceH\x00\x88\x01\x01\x12\x14\n\x0cstep_timeout\x18\x0e \x01(\x02\x12\r\n\x05\x64\x65\x62ug\x18\x0f \x01(\x08\x12\x1b\n\x05\x63\x61\x63he\x18\x10 \x01(\x0b\x32\x0c.graph.Cache\x12(\n\x0cretry_policy\x18\x11 \x01(\x0b\x32\x12.graph.RetryPolicy\x12%\n\x06\x63onfig\x18\x12 \x01(\x0b\x32\x15.types.RunnableConfig\x12\x32\n\x14stream_channels_asis\x18\x13 \x01(\x0b\x32\x14.types.StringOrSlice\x1aL\n\x13TriggerToNodesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12$\n\x05value\x18\x02 \x01(\x0b\x32\x15.graph.TriggerMapping:\x02\x38\x01\x42\x12\n\x10_stream_channels\"\x8c\x01\n\x0eNodeDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x10\n\x08triggers\x18\x02 \x03(\t\x12\x0f\n\x07writers\x18\x03 \x03(\t\x12\x0c\n\x04tags\x18\x04 \x03(\t\x12)\n\x08metadata\x18\x05 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x10\n\x08\x63hannels\x18\x06 \x03(\t\"{\n\x0bRetryPolicy\x12\x18\n\x10initial_interval\x18\x01 \x01(\x02\x12\x16\n\x0e\x62\x61\x63koff_factor\x18\x02 \x01(\x02\x12\x14\n\x0cmax_interval\x18\x03 \x01(\x02\x12\x14\n\x0cmax_attempts\x18\x04 \x01(\x05\x12\x0e\n\x06jitter\x18\x05 \x01(\x08\"\x1b\n\x05\x43\x61\x63he\x12\x12\n\ncache_type\x18\x01 \x01(\t\"\x1f\n\x0eTriggerMapping\x12\r\n\x05nodes\x18\x01 \x03(\tB1Z/github.com/langchain-ai/langgraph-go/runtime/pbb\x06proto3')
30
30
 
31
31
  _globals = globals()
32
32
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -37,15 +37,15 @@ if not _descriptor._USE_C_DESCRIPTORS:
37
37
  _globals['_GRAPHDEFINITION_TRIGGERTONODESENTRY']._loaded_options = None
38
38
  _globals['_GRAPHDEFINITION_TRIGGERTONODESENTRY']._serialized_options = b'8\001'
39
39
  _globals['_GRAPHDEFINITION']._serialized_start=66
40
- _globals['_GRAPHDEFINITION']._serialized_end=710
41
- _globals['_GRAPHDEFINITION_TRIGGERTONODESENTRY']._serialized_start=634
42
- _globals['_GRAPHDEFINITION_TRIGGERTONODESENTRY']._serialized_end=710
43
- _globals['_NODEDEFINITION']._serialized_start=713
44
- _globals['_NODEDEFINITION']._serialized_end=853
45
- _globals['_RETRYPOLICY']._serialized_start=855
46
- _globals['_RETRYPOLICY']._serialized_end=978
47
- _globals['_CACHE']._serialized_start=980
48
- _globals['_CACHE']._serialized_end=1007
49
- _globals['_TRIGGERMAPPING']._serialized_start=1009
50
- _globals['_TRIGGERMAPPING']._serialized_end=1040
40
+ _globals['_GRAPHDEFINITION']._serialized_end=788
41
+ _globals['_GRAPHDEFINITION_TRIGGERTONODESENTRY']._serialized_start=692
42
+ _globals['_GRAPHDEFINITION_TRIGGERTONODESENTRY']._serialized_end=768
43
+ _globals['_NODEDEFINITION']._serialized_start=791
44
+ _globals['_NODEDEFINITION']._serialized_end=931
45
+ _globals['_RETRYPOLICY']._serialized_start=933
46
+ _globals['_RETRYPOLICY']._serialized_end=1056
47
+ _globals['_CACHE']._serialized_start=1058
48
+ _globals['_CACHE']._serialized_end=1085
49
+ _globals['_TRIGGERMAPPING']._serialized_start=1087
50
+ _globals['_TRIGGERMAPPING']._serialized_end=1118
51
51
  # @@protoc_insertion_point(module_scope)
langgraph_executor/pb/graph_pb2.pyi CHANGED
@@ -41,8 +41,6 @@ class GraphDefinition(google.protobuf.message.Message):
41
41
  CHANNELS_FIELD_NUMBER: builtins.int
42
42
  INPUT_CHANNELS_FIELD_NUMBER: builtins.int
43
43
  OUTPUT_CHANNELS_FIELD_NUMBER: builtins.int
44
- INPUT_CHANNELS_IS_LIST_FIELD_NUMBER: builtins.int
45
- OUTPUT_CHANNELS_IS_LIST_FIELD_NUMBER: builtins.int
46
44
  INTERRUPT_BEFORE_NODES_FIELD_NUMBER: builtins.int
47
45
  INTERRUPT_AFTER_NODES_FIELD_NUMBER: builtins.int
48
46
  TRIGGER_TO_NODES_FIELD_NUMBER: builtins.int
@@ -54,9 +52,8 @@ class GraphDefinition(google.protobuf.message.Message):
54
52
  CACHE_FIELD_NUMBER: builtins.int
55
53
  RETRY_POLICY_FIELD_NUMBER: builtins.int
56
54
  CONFIG_FIELD_NUMBER: builtins.int
55
+ STREAM_CHANNELS_ASIS_FIELD_NUMBER: builtins.int
57
56
  name: builtins.str
58
- input_channels_is_list: builtins.bool
59
- output_channels_is_list: builtins.bool
60
57
  stream_eager: builtins.bool
61
58
  step_timeout: builtins.float
62
59
  debug: builtins.bool
@@ -65,9 +62,9 @@ class GraphDefinition(google.protobuf.message.Message):
65
62
  @property
66
63
  def channels(self) -> types_pb2.Channels: ...
67
64
  @property
68
- def input_channels(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
65
+ def input_channels(self) -> types_pb2.StringOrSlice: ...
69
66
  @property
70
- def output_channels(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
67
+ def output_channels(self) -> types_pb2.StringOrSlice: ...
71
68
  @property
72
69
  def interrupt_before_nodes(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
73
70
  @property
@@ -77,37 +74,39 @@ class GraphDefinition(google.protobuf.message.Message):
77
74
  @property
78
75
  def stream_mode(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
79
76
  @property
80
- def stream_channels(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
77
+ def stream_channels(self) -> types_pb2.StringOrSlice: ...
81
78
  @property
82
79
  def cache(self) -> global___Cache: ...
83
80
  @property
84
81
  def retry_policy(self) -> global___RetryPolicy: ...
85
82
  @property
86
83
  def config(self) -> types_pb2.RunnableConfig: ...
84
+ @property
85
+ def stream_channels_asis(self) -> types_pb2.StringOrSlice: ...
87
86
  def __init__(
88
87
  self,
89
88
  *,
90
89
  name: builtins.str = ...,
91
90
  nodes: collections.abc.Iterable[global___NodeDefinition] | None = ...,
92
91
  channels: types_pb2.Channels | None = ...,
93
- input_channels: collections.abc.Iterable[builtins.str] | None = ...,
94
- output_channels: collections.abc.Iterable[builtins.str] | None = ...,
95
- input_channels_is_list: builtins.bool = ...,
96
- output_channels_is_list: builtins.bool = ...,
92
+ input_channels: types_pb2.StringOrSlice | None = ...,
93
+ output_channels: types_pb2.StringOrSlice | None = ...,
97
94
  interrupt_before_nodes: collections.abc.Iterable[builtins.str] | None = ...,
98
95
  interrupt_after_nodes: collections.abc.Iterable[builtins.str] | None = ...,
99
96
  trigger_to_nodes: collections.abc.Mapping[builtins.str, global___TriggerMapping] | None = ...,
100
97
  stream_mode: collections.abc.Iterable[builtins.str] | None = ...,
101
98
  stream_eager: builtins.bool = ...,
102
- stream_channels: collections.abc.Iterable[builtins.str] | None = ...,
99
+ stream_channels: types_pb2.StringOrSlice | None = ...,
103
100
  step_timeout: builtins.float = ...,
104
101
  debug: builtins.bool = ...,
105
102
  cache: global___Cache | None = ...,
106
103
  retry_policy: global___RetryPolicy | None = ...,
107
104
  config: types_pb2.RunnableConfig | None = ...,
105
+ stream_channels_asis: types_pb2.StringOrSlice | None = ...,
108
106
  ) -> None: ...
109
- def HasField(self, field_name: typing.Literal["cache", b"cache", "channels", b"channels", "config", b"config", "retry_policy", b"retry_policy"]) -> builtins.bool: ...
110
- def ClearField(self, field_name: typing.Literal["cache", b"cache", "channels", b"channels", "config", b"config", "debug", b"debug", "input_channels", b"input_channels", "input_channels_is_list", b"input_channels_is_list", "interrupt_after_nodes", b"interrupt_after_nodes", "interrupt_before_nodes", b"interrupt_before_nodes", "name", b"name", "nodes", b"nodes", "output_channels", b"output_channels", "output_channels_is_list", b"output_channels_is_list", "retry_policy", b"retry_policy", "step_timeout", b"step_timeout", "stream_channels", b"stream_channels", "stream_eager", b"stream_eager", "stream_mode", b"stream_mode", "trigger_to_nodes", b"trigger_to_nodes"]) -> None: ...
107
+ def HasField(self, field_name: typing.Literal["_stream_channels", b"_stream_channels", "cache", b"cache", "channels", b"channels", "config", b"config", "input_channels", b"input_channels", "output_channels", b"output_channels", "retry_policy", b"retry_policy", "stream_channels", b"stream_channels", "stream_channels_asis", b"stream_channels_asis"]) -> builtins.bool: ...
108
+ def ClearField(self, field_name: typing.Literal["_stream_channels", b"_stream_channels", "cache", b"cache", "channels", b"channels", "config", b"config", "debug", b"debug", "input_channels", b"input_channels", "interrupt_after_nodes", b"interrupt_after_nodes", "interrupt_before_nodes", b"interrupt_before_nodes", "name", b"name", "nodes", b"nodes", "output_channels", b"output_channels", "retry_policy", b"retry_policy", "step_timeout", b"step_timeout", "stream_channels", b"stream_channels", "stream_channels_asis", b"stream_channels_asis", "stream_eager", b"stream_eager", "stream_mode", b"stream_mode", "trigger_to_nodes", b"trigger_to_nodes"]) -> None: ...
109
+ def WhichOneof(self, oneof_group: typing.Literal["_stream_channels", b"_stream_channels"]) -> typing.Literal["stream_channels"] | None: ...
111
110
 
112
111
  global___GraphDefinition = GraphDefinition
113
112
 
langgraph_executor/pb/runtime_pb2.py CHANGED
@@ -27,7 +27,7 @@ from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
27
27
  from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
28
28
 
29
29
 
30
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\rruntime.proto\x12\x07runtime\x1a\x0btypes.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x1cgoogle/protobuf/struct.proto\"%\n\x12\x41\x64\x64\x45xecutorRequest\x12\x0f\n\x07\x61\x64\x64ress\x18\x01 \x01(\t\"L\n\x0f\x41\x64\x64GraphRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12+\n\x0c\x63heckpointer\x18\x02 \x01(\x0b\x32\x15.runtime.Checkpointer\"}\n\x0c\x43heckpointer\x12/\n\x06memory\x18\x02 \x01(\x0b\x32\x1d.runtime.InMemoryCheckpointerH\x00\x12\x31\n\x08postgres\x18\x03 \x01(\x0b\x32\x1d.runtime.PostgresCheckpointerH\x00\x42\t\n\x07message\"\x16\n\x14InMemoryCheckpointer\",\n\x14PostgresCheckpointer\x12\x14\n\x0c\x64\x61tabase_uri\x18\x01 \x01(\t\"\xcf\x01\n\rInvokeRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12\x1b\n\x05input\x18\x02 \x01(\x0b\x32\x0c.types.Value\x12%\n\x06\x63onfig\x18\x03 \x01(\x0b\x32\x15.types.RunnableConfig\x12\'\n\x08run_opts\x18\x04 \x01(\x0b\x32\x10.runtime.RunOptsH\x00\x88\x01\x01\x12$\n\x07\x63ontext\x18\x05 \x01(\x0b\x32\x0e.types.ContextH\x01\x88\x01\x01\x42\x0b\n\t_run_optsB\n\n\x08_context\"\xcf\x01\n\rStreamRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12\x1b\n\x05input\x18\x02 \x01(\x0b\x32\x0c.types.Value\x12%\n\x06\x63onfig\x18\x03 \x01(\x0b\x32\x15.types.RunnableConfig\x12\'\n\x08run_opts\x18\x04 \x01(\x0b\x32\x10.runtime.RunOptsH\x00\x88\x01\x01\x12$\n\x07\x63ontext\x18\x05 \x01(\x0b\x32\x0e.types.ContextH\x01\x88\x01\x01\x42\x0b\n\t_run_optsB\n\n\x08_context\"\xd2\x01\n\x07RunOpts\x12\x13\n\x0boutput_keys\x18\x01 \x03(\t\x12\x18\n\x10interrupt_before\x18\x02 \x03(\t\x12\x17\n\x0finterrupt_after\x18\x03 \x03(\t\x12\x17\n\ndurability\x18\x04 \x01(\tH\x00\x88\x01\x01\x12\x12\n\x05\x64\x65\x62ug\x18\x05 \x01(\x08H\x01\x88\x01\x01\x12\x16\n\tsubgraphs\x18\x06 \x01(\x08H\x02\x88\x01\x01\x12\x13\n\x0bstream_mode\x18\x07 \x03(\tB\r\n\x0b_durabilityB\x08\n\x06_debugB\x0c\n\n_subgraphs\"1\n\tChunkList\x12$\n\x06\x63hunks\x18\x01 \x03(\x0b\x32\x14.runtime.OutputChunk\"\x86\x01\n\x0bOutputChunk\x12#\n\x05\x63hunk\x18\x01 \x01(\x0b\x32\x12.types.StreamChunkH\x00\x12\x1d\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x0c.types.ValueH\x00\x12(\n\nchunk_list\x18\x05 \x01(\x0b\x32\x12.runtime.ChunkListH\x00\x42\t\n\x07message\"_\n\x0fGetStateRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12%\n\x06\x63onfig\x18\x02 \x01(\x0b\x32\x15.types.RunnableConfig\x12\x11\n\tsubgraphs\x18\x03 \x01(\x08\"7\n\x10GetStateResponse\x12#\n\x05state\x18\x01 \x01(\x0b\x32\x14.types.StateSnapshot\"\xb2\x01\n\x16GetStateHistoryRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12%\n\x06\x63onfig\x18\x02 \x01(\x0b\x32\x15.types.RunnableConfig\x12\'\n\x06\x66ilter\x18\x03 \x01(\x0b\x32\x17.google.protobuf.Struct\x12%\n\x06\x62\x65\x66ore\x18\x04 \x01(\x0b\x32\x15.types.RunnableConfig\x12\r\n\x05limit\x18\x05 \x01(\x03\"@\n\x17GetStateHistoryResponse\x12%\n\x07history\x18\x01 \x03(\x0b\x32\x14.types.StateSnapshot2\x9d\x03\n\x10LangGraphRuntime\x12\x42\n\x0b\x41\x64\x64\x45xecutor\x12\x1b.runtime.AddExecutorRequest\x1a\x16.google.protobuf.Empty\x12<\n\x08\x41\x64\x64Graph\x12\x18.runtime.AddGraphRequest\x1a\x16.google.protobuf.Empty\x12\x36\n\x06Invoke\x12\x16.runtime.InvokeRequest\x1a\x14.runtime.OutputChunk\x12\x38\n\x06Stream\x12\x16.runtime.StreamRequest\x1a\x14.runtime.OutputChunk0\x01\x12?\n\x08GetState\x12\x18.runtime.GetStateRequest\x1a\x19.runtime.GetStateResponse\x12T\n\x0fGetStateHistory\x12\x1f.runtime.GetStateHistoryRequest\x1a .runtime.GetStateHistoryResponseB1Z/github.com/langchain-ai/langgraph-go/runtime/pbb\x06proto3')
30
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\rruntime.proto\x12\x07runtime\x1a\x0btypes.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x1cgoogle/protobuf/struct.proto\"4\n\x1f\x43heckpointerDeleteThreadRequest\x12\x11\n\tthread_id\x18\x01 \x01(\t\"D\n\x1b\x43heckpointerGetTupleRequest\x12%\n\x06\x63onfig\x18\x01 \x01(\x0b\x32\x15.types.RunnableConfig\"j\n\x1c\x43heckpointerGetTupleResponse\x12\x35\n\x10\x63heckpoint_tuple\x18\x01 \x01(\x0b\x32\x16.types.CheckpointTupleH\x00\x88\x01\x01\x42\x13\n\x11_checkpoint_tuple\"\xae\x01\n\x17\x43heckpointerListRequest\x12%\n\x06\x63onfig\x18\x01 \x01(\x0b\x32\x15.types.RunnableConfig\x12\'\n\x06\x66ilter\x18\x02 \x01(\x0b\x32\x17.google.protobuf.Struct\x12%\n\x06\x62\x65\x66ore\x18\x03 \x01(\x0b\x32\x15.types.RunnableConfig\x12\x12\n\x05limit\x18\x04 \x01(\x03H\x00\x88\x01\x01\x42\x08\n\x06_limit\"M\n\x18\x43heckpointerListResponse\x12\x31\n\x11\x63heckpoint_tuples\x18\x01 \x03(\x0b\x32\x16.types.CheckpointTuple\"s\n\x12UpdateStateRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12%\n\x06\x63onfig\x18\x02 \x01(\x0b\x32\x15.types.RunnableConfig\x12\"\n\x06update\x18\x03 \x01(\x0b\x32\x12.types.StateUpdate\"A\n\x13UpdateStateResponse\x12*\n\x0bnext_config\x18\x01 \x01(\x0b\x32\x15.types.RunnableConfig\"\x80\x01\n\x16\x42ulkUpdateStateRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12%\n\x06\x63onfig\x18\x02 \x01(\x0b\x32\x15.types.RunnableConfig\x12+\n\nsupersteps\x18\x03 \x03(\x0b\x32\x17.types.SuperstepUpdates\"E\n\x17\x42ulkUpdateStateResponse\x12*\n\x0bnext_config\x18\x01 \x01(\x0b\x32\x15.types.RunnableConfig\"%\n\x12\x41\x64\x64\x45xecutorRequest\x12\x0f\n\x07\x61\x64\x64ress\x18\x01 \x01(\t\"L\n\x0f\x41\x64\x64GraphRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12+\n\x0c\x63heckpointer\x18\x02 \x01(\x0b\x32\x15.runtime.Checkpointer\"}\n\x0c\x43heckpointer\x12/\n\x06memory\x18\x02 \x01(\x0b\x32\x1d.runtime.InMemoryCheckpointerH\x00\x12\x31\n\x08postgres\x18\x03 \x01(\x0b\x32\x1d.runtime.PostgresCheckpointerH\x00\x42\t\n\x07message\"\x16\n\x14InMemoryCheckpointer\",\n\x14PostgresCheckpointer\x12\x14\n\x0c\x64\x61tabase_uri\x18\x01 \x01(\t\"\xcf\x01\n\rInvokeRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12\x1b\n\x05input\x18\x02 \x01(\x0b\x32\x0c.types.Value\x12%\n\x06\x63onfig\x18\x03 \x01(\x0b\x32\x15.types.RunnableConfig\x12\'\n\x08run_opts\x18\x04 \x01(\x0b\x32\x10.runtime.RunOptsH\x00\x88\x01\x01\x12$\n\x07\x63ontext\x18\x05 \x01(\x0b\x32\x0e.types.ContextH\x01\x88\x01\x01\x42\x0b\n\t_run_optsB\n\n\x08_context\"\xcf\x01\n\rStreamRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12\x1b\n\x05input\x18\x02 \x01(\x0b\x32\x0c.types.Value\x12%\n\x06\x63onfig\x18\x03 \x01(\x0b\x32\x15.types.RunnableConfig\x12\'\n\x08run_opts\x18\x04 \x01(\x0b\x32\x10.runtime.RunOptsH\x00\x88\x01\x01\x12$\n\x07\x63ontext\x18\x05 \x01(\x0b\x32\x0e.types.ContextH\x01\x88\x01\x01\x42\x0b\n\t_run_optsB\n\n\x08_context\"\xfd\x01\n\x07RunOpts\x12.\n\x0boutput_keys\x18\x01 \x01(\x0b\x32\x14.types.StringOrSliceH\x00\x88\x01\x01\x12\x18\n\x10interrupt_before\x18\x02 \x03(\t\x12\x17\n\x0finterrupt_after\x18\x03 \x03(\t\x12\x17\n\ndurability\x18\x04 \x01(\tH\x01\x88\x01\x01\x12\x12\n\x05\x64\x65\x62ug\x18\x05 \x01(\x08H\x02\x88\x01\x01\x12\x16\n\tsubgraphs\x18\x06 \x01(\x08H\x03\x88\x01\x01\x12\x13\n\x0bstream_mode\x18\x07 \x03(\tB\x0e\n\x0c_output_keysB\r\n\x0b_durabilityB\x08\n\x06_debugB\x0c\n\n_subgraphs\"1\n\tChunkList\x12$\n\x06\x63hunks\x18\x01 \x03(\x0b\x32\x14.runtime.OutputChunk\"\x86\x01\n\x0bOutputChunk\x12#\n\x05\x63hunk\x18\x01 \x01(\x0b\x32\x12.types.StreamChunkH\x00\x12\x1d\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x0c.types.ValueH\x00\x12(\n\nchunk_list\x18\x05 \x01(\x0b\x32\x12.runtime.ChunkListH\x00\x42\t\n\x07message\"_\n\x0fGetStateRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12%\n\x06\x63onfig\x18\x02 \x01(\x0b\x32\x15.types.RunnableConfig\x12\x11\n\tsubgraphs\x18\x03 \x01(\x08\"7\n\x10GetStateResponse\x12#\n\x05state\x18\x01 \x01(\x0b\x32\x14.types.StateSnapshot\"\xc1\x01\n\x16GetStateHistoryRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12%\n\x06\x63onfig\x18\x02 \x01(\x0b\x32\x15.types.RunnableConfig\x12\'\n\x06\x66ilter\x18\x03 \x01(\x0b\x32\x17.google.protobuf.Struct\x12%\n\x06\x62\x65\x66ore\x18\x04 \x01(\x0b\x32\x15.types.RunnableConfig\x12\x12\n\x05limit\x18\x05 \x01(\x03H\x00\x88\x01\x01\x42\x08\n\x06_limit\"@\n\x17GetStateHistoryResponse\x12%\n\x07history\x18\x01 \x03(\x0b\x32\x14.types.StateSnapshot2\xd9\x06\n\x10LangGraphRuntime\x12\x42\n\x0b\x41\x64\x64\x45xecutor\x12\x1b.runtime.AddExecutorRequest\x1a\x16.google.protobuf.Empty\x12<\n\x08\x41\x64\x64Graph\x12\x18.runtime.AddGraphRequest\x1a\x16.google.protobuf.Empty\x12\x36\n\x06Invoke\x12\x16.runtime.InvokeRequest\x1a\x14.runtime.OutputChunk\x12\x38\n\x06Stream\x12\x16.runtime.StreamRequest\x1a\x14.runtime.OutputChunk0\x01\x12?\n\x08GetState\x12\x18.runtime.GetStateRequest\x1a\x19.runtime.GetStateResponse\x12T\n\x0fGetStateHistory\x12\x1f.runtime.GetStateHistoryRequest\x1a .runtime.GetStateHistoryResponse\x12H\n\x0bUpdateState\x12\x1b.runtime.UpdateStateRequest\x1a\x1c.runtime.UpdateStateResponse\x12T\n\x0f\x42ulkUpdateState\x12\x1f.runtime.BulkUpdateStateRequest\x1a .runtime.BulkUpdateStateResponse\x12W\n\x10\x43heckpointerList\x12 .runtime.CheckpointerListRequest\x1a!.runtime.CheckpointerListResponse\x12\x63\n\x14\x43heckpointerGetTuple\x12$.runtime.CheckpointerGetTupleRequest\x1a%.runtime.CheckpointerGetTupleResponse\x12\\\n\x18\x43heckpointerDeleteThread\x12(.runtime.CheckpointerDeleteThreadRequest\x1a\x16.google.protobuf.EmptyB1Z/github.com/langchain-ai/langgraph-go/runtime/pbb\x06proto3')
31
31
 
32
32
  _globals = globals()
33
33
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -35,34 +35,52 @@ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'runtime_pb2', _globals)
35
35
  if not _descriptor._USE_C_DESCRIPTORS:
36
36
  _globals['DESCRIPTOR']._loaded_options = None
37
37
  _globals['DESCRIPTOR']._serialized_options = b'Z/github.com/langchain-ai/langgraph-go/runtime/pb'
38
- _globals['_ADDEXECUTORREQUEST']._serialized_start=98
39
- _globals['_ADDEXECUTORREQUEST']._serialized_end=135
40
- _globals['_ADDGRAPHREQUEST']._serialized_start=137
41
- _globals['_ADDGRAPHREQUEST']._serialized_end=213
42
- _globals['_CHECKPOINTER']._serialized_start=215
43
- _globals['_CHECKPOINTER']._serialized_end=340
44
- _globals['_INMEMORYCHECKPOINTER']._serialized_start=342
45
- _globals['_INMEMORYCHECKPOINTER']._serialized_end=364
46
- _globals['_POSTGRESCHECKPOINTER']._serialized_start=366
47
- _globals['_POSTGRESCHECKPOINTER']._serialized_end=410
48
- _globals['_INVOKEREQUEST']._serialized_start=413
49
- _globals['_INVOKEREQUEST']._serialized_end=620
50
- _globals['_STREAMREQUEST']._serialized_start=623
51
- _globals['_STREAMREQUEST']._serialized_end=830
52
- _globals['_RUNOPTS']._serialized_start=833
53
- _globals['_RUNOPTS']._serialized_end=1043
54
- _globals['_CHUNKLIST']._serialized_start=1045
55
- _globals['_CHUNKLIST']._serialized_end=1094
56
- _globals['_OUTPUTCHUNK']._serialized_start=1097
57
- _globals['_OUTPUTCHUNK']._serialized_end=1231
58
- _globals['_GETSTATEREQUEST']._serialized_start=1233
59
- _globals['_GETSTATEREQUEST']._serialized_end=1328
60
- _globals['_GETSTATERESPONSE']._serialized_start=1330
61
- _globals['_GETSTATERESPONSE']._serialized_end=1385
62
- _globals['_GETSTATEHISTORYREQUEST']._serialized_start=1388
63
- _globals['_GETSTATEHISTORYREQUEST']._serialized_end=1566
64
- _globals['_GETSTATEHISTORYRESPONSE']._serialized_start=1568
65
- _globals['_GETSTATEHISTORYRESPONSE']._serialized_end=1632
66
- _globals['_LANGGRAPHRUNTIME']._serialized_start=1635
67
- _globals['_LANGGRAPHRUNTIME']._serialized_end=2048
38
+ _globals['_CHECKPOINTERDELETETHREADREQUEST']._serialized_start=98
39
+ _globals['_CHECKPOINTERDELETETHREADREQUEST']._serialized_end=150
40
+ _globals['_CHECKPOINTERGETTUPLEREQUEST']._serialized_start=152
41
+ _globals['_CHECKPOINTERGETTUPLEREQUEST']._serialized_end=220
42
+ _globals['_CHECKPOINTERGETTUPLERESPONSE']._serialized_start=222
43
+ _globals['_CHECKPOINTERGETTUPLERESPONSE']._serialized_end=328
44
+ _globals['_CHECKPOINTERLISTREQUEST']._serialized_start=331
45
+ _globals['_CHECKPOINTERLISTREQUEST']._serialized_end=505
46
+ _globals['_CHECKPOINTERLISTRESPONSE']._serialized_start=507
47
+ _globals['_CHECKPOINTERLISTRESPONSE']._serialized_end=584
48
+ _globals['_UPDATESTATEREQUEST']._serialized_start=586
49
+ _globals['_UPDATESTATEREQUEST']._serialized_end=701
50
+ _globals['_UPDATESTATERESPONSE']._serialized_start=703
51
+ _globals['_UPDATESTATERESPONSE']._serialized_end=768
52
+ _globals['_BULKUPDATESTATEREQUEST']._serialized_start=771
53
+ _globals['_BULKUPDATESTATEREQUEST']._serialized_end=899
54
+ _globals['_BULKUPDATESTATERESPONSE']._serialized_start=901
55
+ _globals['_BULKUPDATESTATERESPONSE']._serialized_end=970
56
+ _globals['_ADDEXECUTORREQUEST']._serialized_start=972
57
+ _globals['_ADDEXECUTORREQUEST']._serialized_end=1009
58
+ _globals['_ADDGRAPHREQUEST']._serialized_start=1011
59
+ _globals['_ADDGRAPHREQUEST']._serialized_end=1087
60
+ _globals['_CHECKPOINTER']._serialized_start=1089
61
+ _globals['_CHECKPOINTER']._serialized_end=1214
62
+ _globals['_INMEMORYCHECKPOINTER']._serialized_start=1216
63
+ _globals['_INMEMORYCHECKPOINTER']._serialized_end=1238
64
+ _globals['_POSTGRESCHECKPOINTER']._serialized_start=1240
65
+ _globals['_POSTGRESCHECKPOINTER']._serialized_end=1284
66
+ _globals['_INVOKEREQUEST']._serialized_start=1287
67
+ _globals['_INVOKEREQUEST']._serialized_end=1494
68
+ _globals['_STREAMREQUEST']._serialized_start=1497
69
+ _globals['_STREAMREQUEST']._serialized_end=1704
70
+ _globals['_RUNOPTS']._serialized_start=1707
71
+ _globals['_RUNOPTS']._serialized_end=1960
72
+ _globals['_CHUNKLIST']._serialized_start=1962
73
+ _globals['_CHUNKLIST']._serialized_end=2011
74
+ _globals['_OUTPUTCHUNK']._serialized_start=2014
75
+ _globals['_OUTPUTCHUNK']._serialized_end=2148
76
+ _globals['_GETSTATEREQUEST']._serialized_start=2150
77
+ _globals['_GETSTATEREQUEST']._serialized_end=2245
78
+ _globals['_GETSTATERESPONSE']._serialized_start=2247
79
+ _globals['_GETSTATERESPONSE']._serialized_end=2302
80
+ _globals['_GETSTATEHISTORYREQUEST']._serialized_start=2305
81
+ _globals['_GETSTATEHISTORYREQUEST']._serialized_end=2498
82
+ _globals['_GETSTATEHISTORYRESPONSE']._serialized_start=2500
83
+ _globals['_GETSTATEHISTORYRESPONSE']._serialized_end=2564
84
+ _globals['_LANGGRAPHRUNTIME']._serialized_start=2567
85
+ _globals['_LANGGRAPHRUNTIME']._serialized_end=3424
68
86
  # @@protoc_insertion_point(module_scope)
langgraph_executor/pb/runtime_pb2.pyi CHANGED
@@ -14,6 +14,193 @@ import typing
14
14
 
15
15
  DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
16
16
 
17
+ @typing.final
18
+ class CheckpointerDeleteThreadRequest(google.protobuf.message.Message):
19
+ """Checkpointer delete thread"""
20
+
21
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
22
+
23
+ THREAD_ID_FIELD_NUMBER: builtins.int
24
+ thread_id: builtins.str
25
+ def __init__(
26
+ self,
27
+ *,
28
+ thread_id: builtins.str = ...,
29
+ ) -> None: ...
30
+ def ClearField(self, field_name: typing.Literal["thread_id", b"thread_id"]) -> None: ...
31
+
32
+ global___CheckpointerDeleteThreadRequest = CheckpointerDeleteThreadRequest
33
+
34
+ @typing.final
35
+ class CheckpointerGetTupleRequest(google.protobuf.message.Message):
36
+ """Checkpointer Get Tuple"""
37
+
38
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
39
+
40
+ CONFIG_FIELD_NUMBER: builtins.int
41
+ @property
42
+ def config(self) -> types_pb2.RunnableConfig: ...
43
+ def __init__(
44
+ self,
45
+ *,
46
+ config: types_pb2.RunnableConfig | None = ...,
47
+ ) -> None: ...
48
+ def HasField(self, field_name: typing.Literal["config", b"config"]) -> builtins.bool: ...
49
+ def ClearField(self, field_name: typing.Literal["config", b"config"]) -> None: ...
50
+
51
+ global___CheckpointerGetTupleRequest = CheckpointerGetTupleRequest
52
+
53
+ @typing.final
54
+ class CheckpointerGetTupleResponse(google.protobuf.message.Message):
55
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
56
+
57
+ CHECKPOINT_TUPLE_FIELD_NUMBER: builtins.int
58
+ @property
59
+ def checkpoint_tuple(self) -> types_pb2.CheckpointTuple: ...
60
+ def __init__(
61
+ self,
62
+ *,
63
+ checkpoint_tuple: types_pb2.CheckpointTuple | None = ...,
64
+ ) -> None: ...
65
+ def HasField(self, field_name: typing.Literal["_checkpoint_tuple", b"_checkpoint_tuple", "checkpoint_tuple", b"checkpoint_tuple"]) -> builtins.bool: ...
66
+ def ClearField(self, field_name: typing.Literal["_checkpoint_tuple", b"_checkpoint_tuple", "checkpoint_tuple", b"checkpoint_tuple"]) -> None: ...
67
+ def WhichOneof(self, oneof_group: typing.Literal["_checkpoint_tuple", b"_checkpoint_tuple"]) -> typing.Literal["checkpoint_tuple"] | None: ...
68
+
69
+ global___CheckpointerGetTupleResponse = CheckpointerGetTupleResponse
70
+
71
+ @typing.final
72
+ class CheckpointerListRequest(google.protobuf.message.Message):
73
+ """Checkpointer List"""
74
+
75
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
76
+
77
+ CONFIG_FIELD_NUMBER: builtins.int
78
+ FILTER_FIELD_NUMBER: builtins.int
79
+ BEFORE_FIELD_NUMBER: builtins.int
80
+ LIMIT_FIELD_NUMBER: builtins.int
81
+ limit: builtins.int
82
+ @property
83
+ def config(self) -> types_pb2.RunnableConfig: ...
84
+ @property
85
+ def filter(self) -> google.protobuf.struct_pb2.Struct: ...
86
+ @property
87
+ def before(self) -> types_pb2.RunnableConfig: ...
88
+ def __init__(
89
+ self,
90
+ *,
91
+ config: types_pb2.RunnableConfig | None = ...,
92
+ filter: google.protobuf.struct_pb2.Struct | None = ...,
93
+ before: types_pb2.RunnableConfig | None = ...,
94
+ limit: builtins.int | None = ...,
95
+ ) -> None: ...
96
+ def HasField(self, field_name: typing.Literal["_limit", b"_limit", "before", b"before", "config", b"config", "filter", b"filter", "limit", b"limit"]) -> builtins.bool: ...
97
+ def ClearField(self, field_name: typing.Literal["_limit", b"_limit", "before", b"before", "config", b"config", "filter", b"filter", "limit", b"limit"]) -> None: ...
98
+ def WhichOneof(self, oneof_group: typing.Literal["_limit", b"_limit"]) -> typing.Literal["limit"] | None: ...
99
+
100
+ global___CheckpointerListRequest = CheckpointerListRequest
101
+
102
+ @typing.final
103
+ class CheckpointerListResponse(google.protobuf.message.Message):
104
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
105
+
106
+ CHECKPOINT_TUPLES_FIELD_NUMBER: builtins.int
107
+ @property
108
+ def checkpoint_tuples(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[types_pb2.CheckpointTuple]: ...
109
+ def __init__(
110
+ self,
111
+ *,
112
+ checkpoint_tuples: collections.abc.Iterable[types_pb2.CheckpointTuple] | None = ...,
113
+ ) -> None: ...
114
+ def ClearField(self, field_name: typing.Literal["checkpoint_tuples", b"checkpoint_tuples"]) -> None: ...
115
+
116
+ global___CheckpointerListResponse = CheckpointerListResponse
117
+
118
+ @typing.final
119
+ class UpdateStateRequest(google.protobuf.message.Message):
120
+ """Update State"""
121
+
122
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
123
+
124
+ GRAPH_NAME_FIELD_NUMBER: builtins.int
125
+ CONFIG_FIELD_NUMBER: builtins.int
126
+ UPDATE_FIELD_NUMBER: builtins.int
127
+ graph_name: builtins.str
128
+ @property
129
+ def config(self) -> types_pb2.RunnableConfig: ...
130
+ @property
131
+ def update(self) -> types_pb2.StateUpdate: ...
132
+ def __init__(
133
+ self,
134
+ *,
135
+ graph_name: builtins.str = ...,
136
+ config: types_pb2.RunnableConfig | None = ...,
137
+ update: types_pb2.StateUpdate | None = ...,
138
+ ) -> None: ...
139
+ def HasField(self, field_name: typing.Literal["config", b"config", "update", b"update"]) -> builtins.bool: ...
140
+ def ClearField(self, field_name: typing.Literal["config", b"config", "graph_name", b"graph_name", "update", b"update"]) -> None: ...
141
+
142
+ global___UpdateStateRequest = UpdateStateRequest
143
+
144
+ @typing.final
145
+ class UpdateStateResponse(google.protobuf.message.Message):
146
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
147
+
148
+ NEXT_CONFIG_FIELD_NUMBER: builtins.int
149
+ @property
150
+ def next_config(self) -> types_pb2.RunnableConfig: ...
151
+ def __init__(
152
+ self,
153
+ *,
154
+ next_config: types_pb2.RunnableConfig | None = ...,
155
+ ) -> None: ...
156
+ def HasField(self, field_name: typing.Literal["next_config", b"next_config"]) -> builtins.bool: ...
157
+ def ClearField(self, field_name: typing.Literal["next_config", b"next_config"]) -> None: ...
158
+
159
+ global___UpdateStateResponse = UpdateStateResponse
160
+
161
+ @typing.final
162
+ class BulkUpdateStateRequest(google.protobuf.message.Message):
163
+ """BulkUpdateState"""
164
+
165
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
166
+
167
+ GRAPH_NAME_FIELD_NUMBER: builtins.int
168
+ CONFIG_FIELD_NUMBER: builtins.int
169
+ SUPERSTEPS_FIELD_NUMBER: builtins.int
170
+ graph_name: builtins.str
171
+ @property
172
+ def config(self) -> types_pb2.RunnableConfig: ...
173
+ @property
174
+ def supersteps(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[types_pb2.SuperstepUpdates]: ...
175
+ def __init__(
176
+ self,
177
+ *,
178
+ graph_name: builtins.str = ...,
179
+ config: types_pb2.RunnableConfig | None = ...,
180
+ supersteps: collections.abc.Iterable[types_pb2.SuperstepUpdates] | None = ...,
181
+ ) -> None: ...
182
+ def HasField(self, field_name: typing.Literal["config", b"config"]) -> builtins.bool: ...
183
+ def ClearField(self, field_name: typing.Literal["config", b"config", "graph_name", b"graph_name", "supersteps", b"supersteps"]) -> None: ...
184
+
185
+ global___BulkUpdateStateRequest = BulkUpdateStateRequest
186
+
187
+ @typing.final
188
+ class BulkUpdateStateResponse(google.protobuf.message.Message):
189
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
190
+
191
+ NEXT_CONFIG_FIELD_NUMBER: builtins.int
192
+ @property
193
+ def next_config(self) -> types_pb2.RunnableConfig: ...
194
+ def __init__(
195
+ self,
196
+ *,
197
+ next_config: types_pb2.RunnableConfig | None = ...,
198
+ ) -> None: ...
199
+ def HasField(self, field_name: typing.Literal["next_config", b"next_config"]) -> builtins.bool: ...
200
+ def ClearField(self, field_name: typing.Literal["next_config", b"next_config"]) -> None: ...
201
+
202
+ global___BulkUpdateStateResponse = BulkUpdateStateResponse
203
+
17
204
  @typing.final
18
205
  class AddExecutorRequest(google.protobuf.message.Message):
19
206
  """Add executor request"""
@@ -199,7 +386,7 @@ class RunOpts(google.protobuf.message.Message):
199
386
  debug: builtins.bool
200
387
  subgraphs: builtins.bool
201
388
  @property
202
- def output_keys(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
389
+ def output_keys(self) -> types_pb2.StringOrSlice: ...
203
390
  @property
204
391
  def interrupt_before(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
205
392
  @property
@@ -209,7 +396,7 @@ class RunOpts(google.protobuf.message.Message):
209
396
  def __init__(
210
397
  self,
211
398
  *,
212
- output_keys: collections.abc.Iterable[builtins.str] | None = ...,
399
+ output_keys: types_pb2.StringOrSlice | None = ...,
213
400
  interrupt_before: collections.abc.Iterable[builtins.str] | None = ...,
214
401
  interrupt_after: collections.abc.Iterable[builtins.str] | None = ...,
215
402
  durability: builtins.str | None = ...,
@@ -217,13 +404,15 @@ class RunOpts(google.protobuf.message.Message):
217
404
  subgraphs: builtins.bool | None = ...,
218
405
  stream_mode: collections.abc.Iterable[builtins.str] | None = ...,
219
406
  ) -> None: ...
220
- def HasField(self, field_name: typing.Literal["_debug", b"_debug", "_durability", b"_durability", "_subgraphs", b"_subgraphs", "debug", b"debug", "durability", b"durability", "subgraphs", b"subgraphs"]) -> builtins.bool: ...
221
- def ClearField(self, field_name: typing.Literal["_debug", b"_debug", "_durability", b"_durability", "_subgraphs", b"_subgraphs", "debug", b"debug", "durability", b"durability", "interrupt_after", b"interrupt_after", "interrupt_before", b"interrupt_before", "output_keys", b"output_keys", "stream_mode", b"stream_mode", "subgraphs", b"subgraphs"]) -> None: ...
407
+ def HasField(self, field_name: typing.Literal["_debug", b"_debug", "_durability", b"_durability", "_output_keys", b"_output_keys", "_subgraphs", b"_subgraphs", "debug", b"debug", "durability", b"durability", "output_keys", b"output_keys", "subgraphs", b"subgraphs"]) -> builtins.bool: ...
408
+ def ClearField(self, field_name: typing.Literal["_debug", b"_debug", "_durability", b"_durability", "_output_keys", b"_output_keys", "_subgraphs", b"_subgraphs", "debug", b"debug", "durability", b"durability", "interrupt_after", b"interrupt_after", "interrupt_before", b"interrupt_before", "output_keys", b"output_keys", "stream_mode", b"stream_mode", "subgraphs", b"subgraphs"]) -> None: ...
222
409
  @typing.overload
223
410
  def WhichOneof(self, oneof_group: typing.Literal["_debug", b"_debug"]) -> typing.Literal["debug"] | None: ...
224
411
  @typing.overload
225
412
  def WhichOneof(self, oneof_group: typing.Literal["_durability", b"_durability"]) -> typing.Literal["durability"] | None: ...
226
413
  @typing.overload
414
+ def WhichOneof(self, oneof_group: typing.Literal["_output_keys", b"_output_keys"]) -> typing.Literal["output_keys"] | None: ...
415
+ @typing.overload
227
416
  def WhichOneof(self, oneof_group: typing.Literal["_subgraphs", b"_subgraphs"]) -> typing.Literal["subgraphs"] | None: ...
228
417
 
229
418
  global___RunOpts = RunOpts
@@ -340,10 +529,11 @@ class GetStateHistoryRequest(google.protobuf.message.Message):
340
529
  config: types_pb2.RunnableConfig | None = ...,
341
530
  filter: google.protobuf.struct_pb2.Struct | None = ...,
342
531
  before: types_pb2.RunnableConfig | None = ...,
343
- limit: builtins.int = ...,
532
+ limit: builtins.int | None = ...,
344
533
  ) -> None: ...
345
- def HasField(self, field_name: typing.Literal["before", b"before", "config", b"config", "filter", b"filter"]) -> builtins.bool: ...
346
- def ClearField(self, field_name: typing.Literal["before", b"before", "config", b"config", "filter", b"filter", "graph_name", b"graph_name", "limit", b"limit"]) -> None: ...
534
+ def HasField(self, field_name: typing.Literal["_limit", b"_limit", "before", b"before", "config", b"config", "filter", b"filter", "limit", b"limit"]) -> builtins.bool: ...
535
+ def ClearField(self, field_name: typing.Literal["_limit", b"_limit", "before", b"before", "config", b"config", "filter", b"filter", "graph_name", b"graph_name", "limit", b"limit"]) -> None: ...
536
+ def WhichOneof(self, oneof_group: typing.Literal["_limit", b"_limit"]) -> typing.Literal["limit"] | None: ...
347
537
 
348
538
  global___GetStateHistoryRequest = GetStateHistoryRequest
349
539