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.
- langgraph_executor/__init__.py +1 -1
- langgraph_executor/client/__init__.py +0 -0
- langgraph_executor/client/patch.py +305 -0
- langgraph_executor/client/utils.py +340 -0
- langgraph_executor/common.py +65 -16
- langgraph_executor/executor.py +22 -8
- langgraph_executor/executor_base.py +91 -3
- langgraph_executor/extract_graph.py +19 -26
- langgraph_executor/pb/executor_pb2.py +45 -43
- langgraph_executor/pb/executor_pb2.pyi +39 -0
- langgraph_executor/pb/executor_pb2_grpc.py +44 -0
- langgraph_executor/pb/executor_pb2_grpc.pyi +20 -0
- langgraph_executor/pb/graph_pb2.py +12 -12
- langgraph_executor/pb/graph_pb2.pyi +13 -14
- langgraph_executor/pb/runtime_pb2.py +49 -31
- langgraph_executor/pb/runtime_pb2.pyi +197 -7
- langgraph_executor/pb/runtime_pb2_grpc.py +229 -1
- langgraph_executor/pb/runtime_pb2_grpc.pyi +133 -6
- langgraph_executor/pb/types_pb2.py +17 -9
- langgraph_executor/pb/types_pb2.pyi +95 -0
- langgraph_executor/serde.py +13 -0
- {langgraph_executor-0.0.1a5.dist-info → langgraph_executor-0.0.1a7.dist-info}/METADATA +1 -1
- langgraph_executor-0.0.1a7.dist-info/RECORD +36 -0
- langgraph_executor-0.0.1a5.dist-info/RECORD +0 -32
- {langgraph_executor-0.0.1a5.dist-info → langgraph_executor-0.0.1a7.dist-info}/WHEEL +0 -0
@@ -28,6 +28,7 @@ if _version_not_supported:
|
|
28
28
|
|
29
29
|
class LangGraphRuntimeStub(object):
|
30
30
|
"""Runtime service for executing graphs
|
31
|
+
RUNTIME UTILS
|
31
32
|
"""
|
32
33
|
|
33
34
|
def __init__(self, channel):
|
@@ -66,10 +67,36 @@ class LangGraphRuntimeStub(object):
|
|
66
67
|
request_serializer=runtime__pb2.GetStateHistoryRequest.SerializeToString,
|
67
68
|
response_deserializer=runtime__pb2.GetStateHistoryResponse.FromString,
|
68
69
|
_registered_method=True)
|
70
|
+
self.UpdateState = channel.unary_unary(
|
71
|
+
'/runtime.LangGraphRuntime/UpdateState',
|
72
|
+
request_serializer=runtime__pb2.UpdateStateRequest.SerializeToString,
|
73
|
+
response_deserializer=runtime__pb2.UpdateStateResponse.FromString,
|
74
|
+
_registered_method=True)
|
75
|
+
self.BulkUpdateState = channel.unary_unary(
|
76
|
+
'/runtime.LangGraphRuntime/BulkUpdateState',
|
77
|
+
request_serializer=runtime__pb2.BulkUpdateStateRequest.SerializeToString,
|
78
|
+
response_deserializer=runtime__pb2.BulkUpdateStateResponse.FromString,
|
79
|
+
_registered_method=True)
|
80
|
+
self.CheckpointerList = channel.unary_unary(
|
81
|
+
'/runtime.LangGraphRuntime/CheckpointerList',
|
82
|
+
request_serializer=runtime__pb2.CheckpointerListRequest.SerializeToString,
|
83
|
+
response_deserializer=runtime__pb2.CheckpointerListResponse.FromString,
|
84
|
+
_registered_method=True)
|
85
|
+
self.CheckpointerGetTuple = channel.unary_unary(
|
86
|
+
'/runtime.LangGraphRuntime/CheckpointerGetTuple',
|
87
|
+
request_serializer=runtime__pb2.CheckpointerGetTupleRequest.SerializeToString,
|
88
|
+
response_deserializer=runtime__pb2.CheckpointerGetTupleResponse.FromString,
|
89
|
+
_registered_method=True)
|
90
|
+
self.CheckpointerDeleteThread = channel.unary_unary(
|
91
|
+
'/runtime.LangGraphRuntime/CheckpointerDeleteThread',
|
92
|
+
request_serializer=runtime__pb2.CheckpointerDeleteThreadRequest.SerializeToString,
|
93
|
+
response_deserializer=google_dot_protobuf_dot_empty__pb2.Empty.FromString,
|
94
|
+
_registered_method=True)
|
69
95
|
|
70
96
|
|
71
97
|
class LangGraphRuntimeServicer(object):
|
72
98
|
"""Runtime service for executing graphs
|
99
|
+
RUNTIME UTILS
|
73
100
|
"""
|
74
101
|
|
75
102
|
def AddExecutor(self, request, context):
|
@@ -87,7 +114,9 @@ class LangGraphRuntimeServicer(object):
|
|
87
114
|
raise NotImplementedError('Method not implemented!')
|
88
115
|
|
89
116
|
def Invoke(self, request, context):
|
90
|
-
"""
|
117
|
+
"""PREGEL METHODS
|
118
|
+
|
119
|
+
Invoke a graph synchronously
|
91
120
|
"""
|
92
121
|
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
93
122
|
context.set_details('Method not implemented!')
|
@@ -114,6 +143,44 @@ class LangGraphRuntimeServicer(object):
|
|
114
143
|
context.set_details('Method not implemented!')
|
115
144
|
raise NotImplementedError('Method not implemented!')
|
116
145
|
|
146
|
+
def UpdateState(self, request, context):
|
147
|
+
"""Update state
|
148
|
+
"""
|
149
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
150
|
+
context.set_details('Method not implemented!')
|
151
|
+
raise NotImplementedError('Method not implemented!')
|
152
|
+
|
153
|
+
def BulkUpdateState(self, request, context):
|
154
|
+
"""Bulk update state
|
155
|
+
"""
|
156
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
157
|
+
context.set_details('Method not implemented!')
|
158
|
+
raise NotImplementedError('Method not implemented!')
|
159
|
+
|
160
|
+
def CheckpointerList(self, request, context):
|
161
|
+
"""CHECKPOINTER METHODS
|
162
|
+
TODO where should these live? Ideally in separate checkpointer service.
|
163
|
+
|
164
|
+
List checkpoint tuples
|
165
|
+
"""
|
166
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
167
|
+
context.set_details('Method not implemented!')
|
168
|
+
raise NotImplementedError('Method not implemented!')
|
169
|
+
|
170
|
+
def CheckpointerGetTuple(self, request, context):
|
171
|
+
"""Get checkpoint tuple
|
172
|
+
"""
|
173
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
174
|
+
context.set_details('Method not implemented!')
|
175
|
+
raise NotImplementedError('Method not implemented!')
|
176
|
+
|
177
|
+
def CheckpointerDeleteThread(self, request, context):
|
178
|
+
"""Delete thread
|
179
|
+
"""
|
180
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
181
|
+
context.set_details('Method not implemented!')
|
182
|
+
raise NotImplementedError('Method not implemented!')
|
183
|
+
|
117
184
|
|
118
185
|
def add_LangGraphRuntimeServicer_to_server(servicer, server):
|
119
186
|
rpc_method_handlers = {
|
@@ -147,6 +214,31 @@ def add_LangGraphRuntimeServicer_to_server(servicer, server):
|
|
147
214
|
request_deserializer=runtime__pb2.GetStateHistoryRequest.FromString,
|
148
215
|
response_serializer=runtime__pb2.GetStateHistoryResponse.SerializeToString,
|
149
216
|
),
|
217
|
+
'UpdateState': grpc.unary_unary_rpc_method_handler(
|
218
|
+
servicer.UpdateState,
|
219
|
+
request_deserializer=runtime__pb2.UpdateStateRequest.FromString,
|
220
|
+
response_serializer=runtime__pb2.UpdateStateResponse.SerializeToString,
|
221
|
+
),
|
222
|
+
'BulkUpdateState': grpc.unary_unary_rpc_method_handler(
|
223
|
+
servicer.BulkUpdateState,
|
224
|
+
request_deserializer=runtime__pb2.BulkUpdateStateRequest.FromString,
|
225
|
+
response_serializer=runtime__pb2.BulkUpdateStateResponse.SerializeToString,
|
226
|
+
),
|
227
|
+
'CheckpointerList': grpc.unary_unary_rpc_method_handler(
|
228
|
+
servicer.CheckpointerList,
|
229
|
+
request_deserializer=runtime__pb2.CheckpointerListRequest.FromString,
|
230
|
+
response_serializer=runtime__pb2.CheckpointerListResponse.SerializeToString,
|
231
|
+
),
|
232
|
+
'CheckpointerGetTuple': grpc.unary_unary_rpc_method_handler(
|
233
|
+
servicer.CheckpointerGetTuple,
|
234
|
+
request_deserializer=runtime__pb2.CheckpointerGetTupleRequest.FromString,
|
235
|
+
response_serializer=runtime__pb2.CheckpointerGetTupleResponse.SerializeToString,
|
236
|
+
),
|
237
|
+
'CheckpointerDeleteThread': grpc.unary_unary_rpc_method_handler(
|
238
|
+
servicer.CheckpointerDeleteThread,
|
239
|
+
request_deserializer=runtime__pb2.CheckpointerDeleteThreadRequest.FromString,
|
240
|
+
response_serializer=google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
|
241
|
+
),
|
150
242
|
}
|
151
243
|
generic_handler = grpc.method_handlers_generic_handler(
|
152
244
|
'runtime.LangGraphRuntime', rpc_method_handlers)
|
@@ -157,6 +249,7 @@ def add_LangGraphRuntimeServicer_to_server(servicer, server):
|
|
157
249
|
# This class is part of an EXPERIMENTAL API.
|
158
250
|
class LangGraphRuntime(object):
|
159
251
|
"""Runtime service for executing graphs
|
252
|
+
RUNTIME UTILS
|
160
253
|
"""
|
161
254
|
|
162
255
|
@staticmethod
|
@@ -320,3 +413,138 @@ class LangGraphRuntime(object):
|
|
320
413
|
timeout,
|
321
414
|
metadata,
|
322
415
|
_registered_method=True)
|
416
|
+
|
417
|
+
@staticmethod
|
418
|
+
def UpdateState(request,
|
419
|
+
target,
|
420
|
+
options=(),
|
421
|
+
channel_credentials=None,
|
422
|
+
call_credentials=None,
|
423
|
+
insecure=False,
|
424
|
+
compression=None,
|
425
|
+
wait_for_ready=None,
|
426
|
+
timeout=None,
|
427
|
+
metadata=None):
|
428
|
+
return grpc.experimental.unary_unary(
|
429
|
+
request,
|
430
|
+
target,
|
431
|
+
'/runtime.LangGraphRuntime/UpdateState',
|
432
|
+
runtime__pb2.UpdateStateRequest.SerializeToString,
|
433
|
+
runtime__pb2.UpdateStateResponse.FromString,
|
434
|
+
options,
|
435
|
+
channel_credentials,
|
436
|
+
insecure,
|
437
|
+
call_credentials,
|
438
|
+
compression,
|
439
|
+
wait_for_ready,
|
440
|
+
timeout,
|
441
|
+
metadata,
|
442
|
+
_registered_method=True)
|
443
|
+
|
444
|
+
@staticmethod
|
445
|
+
def BulkUpdateState(request,
|
446
|
+
target,
|
447
|
+
options=(),
|
448
|
+
channel_credentials=None,
|
449
|
+
call_credentials=None,
|
450
|
+
insecure=False,
|
451
|
+
compression=None,
|
452
|
+
wait_for_ready=None,
|
453
|
+
timeout=None,
|
454
|
+
metadata=None):
|
455
|
+
return grpc.experimental.unary_unary(
|
456
|
+
request,
|
457
|
+
target,
|
458
|
+
'/runtime.LangGraphRuntime/BulkUpdateState',
|
459
|
+
runtime__pb2.BulkUpdateStateRequest.SerializeToString,
|
460
|
+
runtime__pb2.BulkUpdateStateResponse.FromString,
|
461
|
+
options,
|
462
|
+
channel_credentials,
|
463
|
+
insecure,
|
464
|
+
call_credentials,
|
465
|
+
compression,
|
466
|
+
wait_for_ready,
|
467
|
+
timeout,
|
468
|
+
metadata,
|
469
|
+
_registered_method=True)
|
470
|
+
|
471
|
+
@staticmethod
|
472
|
+
def CheckpointerList(request,
|
473
|
+
target,
|
474
|
+
options=(),
|
475
|
+
channel_credentials=None,
|
476
|
+
call_credentials=None,
|
477
|
+
insecure=False,
|
478
|
+
compression=None,
|
479
|
+
wait_for_ready=None,
|
480
|
+
timeout=None,
|
481
|
+
metadata=None):
|
482
|
+
return grpc.experimental.unary_unary(
|
483
|
+
request,
|
484
|
+
target,
|
485
|
+
'/runtime.LangGraphRuntime/CheckpointerList',
|
486
|
+
runtime__pb2.CheckpointerListRequest.SerializeToString,
|
487
|
+
runtime__pb2.CheckpointerListResponse.FromString,
|
488
|
+
options,
|
489
|
+
channel_credentials,
|
490
|
+
insecure,
|
491
|
+
call_credentials,
|
492
|
+
compression,
|
493
|
+
wait_for_ready,
|
494
|
+
timeout,
|
495
|
+
metadata,
|
496
|
+
_registered_method=True)
|
497
|
+
|
498
|
+
@staticmethod
|
499
|
+
def CheckpointerGetTuple(request,
|
500
|
+
target,
|
501
|
+
options=(),
|
502
|
+
channel_credentials=None,
|
503
|
+
call_credentials=None,
|
504
|
+
insecure=False,
|
505
|
+
compression=None,
|
506
|
+
wait_for_ready=None,
|
507
|
+
timeout=None,
|
508
|
+
metadata=None):
|
509
|
+
return grpc.experimental.unary_unary(
|
510
|
+
request,
|
511
|
+
target,
|
512
|
+
'/runtime.LangGraphRuntime/CheckpointerGetTuple',
|
513
|
+
runtime__pb2.CheckpointerGetTupleRequest.SerializeToString,
|
514
|
+
runtime__pb2.CheckpointerGetTupleResponse.FromString,
|
515
|
+
options,
|
516
|
+
channel_credentials,
|
517
|
+
insecure,
|
518
|
+
call_credentials,
|
519
|
+
compression,
|
520
|
+
wait_for_ready,
|
521
|
+
timeout,
|
522
|
+
metadata,
|
523
|
+
_registered_method=True)
|
524
|
+
|
525
|
+
@staticmethod
|
526
|
+
def CheckpointerDeleteThread(request,
|
527
|
+
target,
|
528
|
+
options=(),
|
529
|
+
channel_credentials=None,
|
530
|
+
call_credentials=None,
|
531
|
+
insecure=False,
|
532
|
+
compression=None,
|
533
|
+
wait_for_ready=None,
|
534
|
+
timeout=None,
|
535
|
+
metadata=None):
|
536
|
+
return grpc.experimental.unary_unary(
|
537
|
+
request,
|
538
|
+
target,
|
539
|
+
'/runtime.LangGraphRuntime/CheckpointerDeleteThread',
|
540
|
+
runtime__pb2.CheckpointerDeleteThreadRequest.SerializeToString,
|
541
|
+
google_dot_protobuf_dot_empty__pb2.Empty.FromString,
|
542
|
+
options,
|
543
|
+
channel_credentials,
|
544
|
+
insecure,
|
545
|
+
call_credentials,
|
546
|
+
compression,
|
547
|
+
wait_for_ready,
|
548
|
+
timeout,
|
549
|
+
metadata,
|
550
|
+
_registered_method=True)
|
@@ -19,7 +19,9 @@ class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext): # type:
|
|
19
19
|
...
|
20
20
|
|
21
21
|
class LangGraphRuntimeStub:
|
22
|
-
"""Runtime service for executing graphs
|
22
|
+
"""Runtime service for executing graphs
|
23
|
+
RUNTIME UTILS
|
24
|
+
"""
|
23
25
|
|
24
26
|
def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ...
|
25
27
|
AddExecutor: grpc.UnaryUnaryMultiCallable[
|
@@ -38,7 +40,10 @@ class LangGraphRuntimeStub:
|
|
38
40
|
runtime_pb2.InvokeRequest,
|
39
41
|
runtime_pb2.OutputChunk,
|
40
42
|
]
|
41
|
-
"""
|
43
|
+
"""PREGEL METHODS
|
44
|
+
|
45
|
+
Invoke a graph synchronously
|
46
|
+
"""
|
42
47
|
|
43
48
|
Stream: grpc.UnaryStreamMultiCallable[
|
44
49
|
runtime_pb2.StreamRequest,
|
@@ -58,8 +63,44 @@ class LangGraphRuntimeStub:
|
|
58
63
|
]
|
59
64
|
"""State history"""
|
60
65
|
|
66
|
+
UpdateState: grpc.UnaryUnaryMultiCallable[
|
67
|
+
runtime_pb2.UpdateStateRequest,
|
68
|
+
runtime_pb2.UpdateStateResponse,
|
69
|
+
]
|
70
|
+
"""Update state"""
|
71
|
+
|
72
|
+
BulkUpdateState: grpc.UnaryUnaryMultiCallable[
|
73
|
+
runtime_pb2.BulkUpdateStateRequest,
|
74
|
+
runtime_pb2.BulkUpdateStateResponse,
|
75
|
+
]
|
76
|
+
"""Bulk update state"""
|
77
|
+
|
78
|
+
CheckpointerList: grpc.UnaryUnaryMultiCallable[
|
79
|
+
runtime_pb2.CheckpointerListRequest,
|
80
|
+
runtime_pb2.CheckpointerListResponse,
|
81
|
+
]
|
82
|
+
"""CHECKPOINTER METHODS
|
83
|
+
TODO where should these live? Ideally in separate checkpointer service.
|
84
|
+
|
85
|
+
List checkpoint tuples
|
86
|
+
"""
|
87
|
+
|
88
|
+
CheckpointerGetTuple: grpc.UnaryUnaryMultiCallable[
|
89
|
+
runtime_pb2.CheckpointerGetTupleRequest,
|
90
|
+
runtime_pb2.CheckpointerGetTupleResponse,
|
91
|
+
]
|
92
|
+
"""Get checkpoint tuple"""
|
93
|
+
|
94
|
+
CheckpointerDeleteThread: grpc.UnaryUnaryMultiCallable[
|
95
|
+
runtime_pb2.CheckpointerDeleteThreadRequest,
|
96
|
+
google.protobuf.empty_pb2.Empty,
|
97
|
+
]
|
98
|
+
"""Delete thread"""
|
99
|
+
|
61
100
|
class LangGraphRuntimeAsyncStub:
|
62
|
-
"""Runtime service for executing graphs
|
101
|
+
"""Runtime service for executing graphs
|
102
|
+
RUNTIME UTILS
|
103
|
+
"""
|
63
104
|
|
64
105
|
AddExecutor: grpc.aio.UnaryUnaryMultiCallable[
|
65
106
|
runtime_pb2.AddExecutorRequest,
|
@@ -77,7 +118,10 @@ class LangGraphRuntimeAsyncStub:
|
|
77
118
|
runtime_pb2.InvokeRequest,
|
78
119
|
runtime_pb2.OutputChunk,
|
79
120
|
]
|
80
|
-
"""
|
121
|
+
"""PREGEL METHODS
|
122
|
+
|
123
|
+
Invoke a graph synchronously
|
124
|
+
"""
|
81
125
|
|
82
126
|
Stream: grpc.aio.UnaryStreamMultiCallable[
|
83
127
|
runtime_pb2.StreamRequest,
|
@@ -97,8 +141,44 @@ class LangGraphRuntimeAsyncStub:
|
|
97
141
|
]
|
98
142
|
"""State history"""
|
99
143
|
|
144
|
+
UpdateState: grpc.aio.UnaryUnaryMultiCallable[
|
145
|
+
runtime_pb2.UpdateStateRequest,
|
146
|
+
runtime_pb2.UpdateStateResponse,
|
147
|
+
]
|
148
|
+
"""Update state"""
|
149
|
+
|
150
|
+
BulkUpdateState: grpc.aio.UnaryUnaryMultiCallable[
|
151
|
+
runtime_pb2.BulkUpdateStateRequest,
|
152
|
+
runtime_pb2.BulkUpdateStateResponse,
|
153
|
+
]
|
154
|
+
"""Bulk update state"""
|
155
|
+
|
156
|
+
CheckpointerList: grpc.aio.UnaryUnaryMultiCallable[
|
157
|
+
runtime_pb2.CheckpointerListRequest,
|
158
|
+
runtime_pb2.CheckpointerListResponse,
|
159
|
+
]
|
160
|
+
"""CHECKPOINTER METHODS
|
161
|
+
TODO where should these live? Ideally in separate checkpointer service.
|
162
|
+
|
163
|
+
List checkpoint tuples
|
164
|
+
"""
|
165
|
+
|
166
|
+
CheckpointerGetTuple: grpc.aio.UnaryUnaryMultiCallable[
|
167
|
+
runtime_pb2.CheckpointerGetTupleRequest,
|
168
|
+
runtime_pb2.CheckpointerGetTupleResponse,
|
169
|
+
]
|
170
|
+
"""Get checkpoint tuple"""
|
171
|
+
|
172
|
+
CheckpointerDeleteThread: grpc.aio.UnaryUnaryMultiCallable[
|
173
|
+
runtime_pb2.CheckpointerDeleteThreadRequest,
|
174
|
+
google.protobuf.empty_pb2.Empty,
|
175
|
+
]
|
176
|
+
"""Delete thread"""
|
177
|
+
|
100
178
|
class LangGraphRuntimeServicer(metaclass=abc.ABCMeta):
|
101
|
-
"""Runtime service for executing graphs
|
179
|
+
"""Runtime service for executing graphs
|
180
|
+
RUNTIME UTILS
|
181
|
+
"""
|
102
182
|
|
103
183
|
@abc.abstractmethod
|
104
184
|
def AddExecutor(
|
@@ -122,7 +202,10 @@ class LangGraphRuntimeServicer(metaclass=abc.ABCMeta):
|
|
122
202
|
request: runtime_pb2.InvokeRequest,
|
123
203
|
context: _ServicerContext,
|
124
204
|
) -> typing.Union[runtime_pb2.OutputChunk, collections.abc.Awaitable[runtime_pb2.OutputChunk]]:
|
125
|
-
"""
|
205
|
+
"""PREGEL METHODS
|
206
|
+
|
207
|
+
Invoke a graph synchronously
|
208
|
+
"""
|
126
209
|
|
127
210
|
@abc.abstractmethod
|
128
211
|
def Stream(
|
@@ -148,4 +231,48 @@ class LangGraphRuntimeServicer(metaclass=abc.ABCMeta):
|
|
148
231
|
) -> typing.Union[runtime_pb2.GetStateHistoryResponse, collections.abc.Awaitable[runtime_pb2.GetStateHistoryResponse]]:
|
149
232
|
"""State history"""
|
150
233
|
|
234
|
+
@abc.abstractmethod
|
235
|
+
def UpdateState(
|
236
|
+
self,
|
237
|
+
request: runtime_pb2.UpdateStateRequest,
|
238
|
+
context: _ServicerContext,
|
239
|
+
) -> typing.Union[runtime_pb2.UpdateStateResponse, collections.abc.Awaitable[runtime_pb2.UpdateStateResponse]]:
|
240
|
+
"""Update state"""
|
241
|
+
|
242
|
+
@abc.abstractmethod
|
243
|
+
def BulkUpdateState(
|
244
|
+
self,
|
245
|
+
request: runtime_pb2.BulkUpdateStateRequest,
|
246
|
+
context: _ServicerContext,
|
247
|
+
) -> typing.Union[runtime_pb2.BulkUpdateStateResponse, collections.abc.Awaitable[runtime_pb2.BulkUpdateStateResponse]]:
|
248
|
+
"""Bulk update state"""
|
249
|
+
|
250
|
+
@abc.abstractmethod
|
251
|
+
def CheckpointerList(
|
252
|
+
self,
|
253
|
+
request: runtime_pb2.CheckpointerListRequest,
|
254
|
+
context: _ServicerContext,
|
255
|
+
) -> typing.Union[runtime_pb2.CheckpointerListResponse, collections.abc.Awaitable[runtime_pb2.CheckpointerListResponse]]:
|
256
|
+
"""CHECKPOINTER METHODS
|
257
|
+
TODO where should these live? Ideally in separate checkpointer service.
|
258
|
+
|
259
|
+
List checkpoint tuples
|
260
|
+
"""
|
261
|
+
|
262
|
+
@abc.abstractmethod
|
263
|
+
def CheckpointerGetTuple(
|
264
|
+
self,
|
265
|
+
request: runtime_pb2.CheckpointerGetTupleRequest,
|
266
|
+
context: _ServicerContext,
|
267
|
+
) -> typing.Union[runtime_pb2.CheckpointerGetTupleResponse, collections.abc.Awaitable[runtime_pb2.CheckpointerGetTupleResponse]]:
|
268
|
+
"""Get checkpoint tuple"""
|
269
|
+
|
270
|
+
@abc.abstractmethod
|
271
|
+
def CheckpointerDeleteThread(
|
272
|
+
self,
|
273
|
+
request: runtime_pb2.CheckpointerDeleteThreadRequest,
|
274
|
+
context: _ServicerContext,
|
275
|
+
) -> typing.Union[google.protobuf.empty_pb2.Empty, collections.abc.Awaitable[google.protobuf.empty_pb2.Empty]]:
|
276
|
+
"""Delete thread"""
|
277
|
+
|
151
278
|
def add_LangGraphRuntimeServicer_to_server(servicer: LangGraphRuntimeServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...
|
@@ -25,7 +25,7 @@ _sym_db = _symbol_database.Default()
|
|
25
25
|
from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
|
26
26
|
|
27
27
|
|
28
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0btypes.proto\x12\x05types\x1a\x1cgoogle/protobuf/struct.proto\"\xa8\x02\n\x05Value\x12\x30\n\x10serialized_value\x18\x01 \x01(\x0b\x32\x16.types.SerializedValue\x12,\n\nbase_value\x18\x02 \x01(\x0b\x32\x16.types.SerializedValueH\x00\x12\x1d\n\x05sends\x18\x03 \x01(\x0b\x32\x0c.types.SendsH\x00\x12!\n\x07missing\x18\x04 \x01(\x0b\x32\x0e.types.MissingH\x00\x12!\n\x07\x63ommand\x18\x05 \x01(\x0b\x32\x0e.types.CommandH\x00\x12%\n\x05\x65rror\x18\x06 \x01(\x0b\x32\x14.types.ExecutorErrorH\x00\x12(\n\x0bmessage_ids\x18\x07 \x01(\x0b\x32\x11.types.MessageIdsH\x00\x42\t\n\x07message\"\t\n\x07Missing\"0\n\x0fSerializedValue\x12\x0e\n\x06method\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x0c\"!\n\nMessageIds\x12\x13\n\x0bmessage_ids\x18\x01 \x03(\t\"|\n\x08\x43hannels\x12/\n\x08\x63hannels\x18\x01 \x03(\x0b\x32\x1d.types.Channels.ChannelsEntry\x1a?\n\rChannelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1d\n\x05value\x18\x02 \x01(\x0b\x32\x0e.types.Channel:\x02\x38\x01\"q\n\x07\x43hannel\x12 \n\nget_result\x18\x01 \x01(\x0b\x32\x0c.types.Value\x12\x1b\n\x13is_available_result\x18\x02 \x01(\x08\x12\'\n\x11\x63heckpoint_result\x18\x03 \x01(\x0b\x32\x0c.types.Value\"#\n\x05Sends\x12\x1a\n\x05sends\x18\x01 \x03(\x0b\x32\x0b.types.Send\"/\n\x04Send\x12\x0c\n\x04node\x18\x01 \x01(\t\x12\x19\n\x03\x61rg\x18\x02 \x01(\x0b\x32\x0c.types.Value\"\xcb\x01\n\x07\x43ommand\x12\x12\n\x05graph\x18\x01 \x01(\tH\x00\x88\x01\x01\x12*\n\x06update\x18\x02 \x03(\x0b\x32\x1a.types.Command.UpdateEntry\x12\x1d\n\x06resume\x18\x03 \x01(\x0b\x32\r.types.Resume\x12\x1a\n\x05gotos\x18\x04 \x03(\x0b\x32\x0b.types.Goto\x1a;\n\x0bUpdateEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.types.Value:\x02\x38\x01\x42\x08\n\x06_graph\"\\\n\x06Resume\x12\x1d\n\x05value\x18\x01 \x01(\x0b\x32\x0c.types.ValueH\x00\x12(\n\x06values\x18\x02 \x01(\x0b\x32\x16.types.InterruptValuesH\x00\x42\t\n\x07message\"\x82\x01\n\x0fInterruptValues\x12\x32\n\x06values\x18\x01 \x03(\x0b\x32\".types.InterruptValues.ValuesEntry\x1a;\n\x0bValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.types.Value:\x02\x38\x01\"T\n\x04Goto\x12$\n\tnode_name\x18\x01 \x01(\x0b\x32\x0f.types.NodeNameH\x00\x12\x1b\n\x04send\x18\x02 \x01(\x0b\x32\x0b.types.SendH\x00\x42\t\n\x07message\"\x18\n\x08NodeName\x12\x0c\n\x04name\x18\x01 \x01(\t\"\xed\x01\n\rExecutorError\x12\x30\n\x10\x65rror_serialized\x18\x01 \x01(\x0b\x32\x16.types.SerializedValue\x12/\n\x0fgraph_bubble_up\x18\x02 \x01(\x0b\x32\x14.types.GraphBubbleUpH\x00\x12\x30\n\x0fgraph_interrupt\x18\x03 \x01(\x0b\x32\x15.types.GraphInterruptH\x00\x12&\n\nbase_error\x18\x04 \x01(\x0b\x32\x10.types.BaseErrorH\x00\x12\x11\n\ttraceback\x18\x05 \x01(\tB\x0c\n\nerror_type\"\x0f\n\rGraphBubbleUp\"m\n\x0eGraphInterrupt\x12$\n\ninterrupts\x18\x01 \x03(\x0b\x32\x10.types.Interrupt\x12\x35\n\x15interrupts_serialized\x18\x02 \x01(\x0b\x32\x16.types.SerializedValue\"h\n\tBaseError\x12\x12\n\nerror_type\x18\x01 \x01(\t\x12\x15\n\rerror_message\x18\x02 \x01(\t\x12\x30\n\x10\x65rror_serialized\x18\x03 \x01(\x0b\x32\x16.types.SerializedValue\"4\n\tInterrupt\x12\x1b\n\x05value\x18\x01 \x01(\x0b\x32\x0c.types.Value\x12\n\n\x02id\x18\x02 \x01(\t\"5\n\x05Write\x12\x0f\n\x07\x63hannel\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.types.Value\"M\n\x0cPendingWrite\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0f\n\x07\x63hannel\x18\x02 \x01(\t\x12\x1b\n\x05value\x18\x03 \x01(\x0b\x32\x0c.types.Value\"\x90\x01\n\x0f\x43hannelVersions\x12\x45\n\x10\x63hannel_versions\x18\x01 \x03(\x0b\x32+.types.ChannelVersions.ChannelVersionsEntry\x1a\x36\n\x14\x43hannelVersionsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x88\x02\n\x0eRunnableConfig\x12\x0c\n\x04tags\x18\x01 \x03(\t\x12)\n\x08metadata\x18\x02 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x10\n\x08run_name\x18\x03 \x01(\t\x12\x17\n\x0fmax_concurrency\x18\x04 \x01(\x05\x12\x17\n\x0frecursion_limit\x18\x05 \x01(\x05\x12-\n\x0c\x63onfigurable\x18\x06 \x01(\x0b\x32\x17.google.protobuf.Struct\x12:\n\x15reserved_configurable\x18\x07 \x01(\x0b\x32\x1b.types.ReservedConfigurable\x12\x0e\n\x06run_id\x18\x08 \x01(\t\"3\n\x07\x43ontext\x12(\n\x07\x63ontext\x18\x01 \x01(\x0b\x32\x17.google.protobuf.Struct\"\xe3\x03\n\x14ReservedConfigurable\x12\x10\n\x08resuming\x18\x01 \x01(\x08\x12\x0f\n\x07task_id\x18\x02 \x01(\t\x12\x14\n\x0c\x64\x65\x64upe_tasks\x18\x03 \x01(\x08\x12\x15\n\rensure_latest\x18\x04 \x01(\x08\x12\x11\n\tthread_id\x18\x05 \x01(\t\x12\x46\n\x0e\x63heckpoint_map\x18\x06 \x03(\x0b\x32..types.ReservedConfigurable.CheckpointMapEntry\x12\x15\n\rcheckpoint_id\x18\x07 \x01(\t\x12\x15\n\rcheckpoint_ns\x18\x08 \x01(\t\x12(\n\x08previous\x18\t \x01(\x0b\x32\x16.types.SerializedValue\x12\x12\n\ndurability\x18\n \x01(\t\x12>\n\nresume_map\x18\x0b \x03(\x0b\x32*.types.ReservedConfigurable.ResumeMapEntry\x1a\x34\n\x12\x43heckpointMapEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a>\n\x0eResumeMapEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.types.Value:\x02\x38\x01\"\xe1\x02\n\x04Task\x12\x0c\n\x04name\x18\x01 \x01(\t\x12%\n\x05input\x18\x02 \x03(\x0b\x32\x16.types.Task.InputEntry\x12\x1c\n\x06writes\x18\x03 \x03(\x0b\x32\x0c.types.Write\x12%\n\x06\x63onfig\x18\x04 \x01(\x0b\x32\x15.types.RunnableConfig\x12\x10\n\x08triggers\x18\x05 \x03(\t\x12\n\n\x02id\x18\x06 \x01(\t\x12\x11\n\ttask_path\x18\x07 \x03(\t\x12\x0f\n\x07writers\x18\x08 \x03(\t\x12\x12\n\ngraph_name\x18\t \x01(\t\x12+\n\x0epending_writes\x18\n \x03(\x0b\x32\x13.types.PendingWrite\x12 \n\x08messages\x18\x0b \x03(\x0b\x32\x0e.types.Message\x1a:\n\nInputEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.types.Value:\x02\x38\x01\"\xc8\x01\n\nPregelTask\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x0c\n\x04path\x18\x03 \x03(\t\x12$\n\ninterrupts\x18\x04 \x03(\x0b\x32\x10.types.Interrupt\x12&\n\x05state\x18\x05 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\'\n\x06result\x18\x06 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x1b\n\x05\x65rror\x18\x07 \x01(\x0b\x32\x0c.types.Value\"\xb4\x03\n\nCheckpoint\x12\t\n\x01v\x18\x01 \x01(\x04\x12\n\n\x02id\x18\x02 \x01(\t\x12<\n\x0e\x63hannel_values\x18\x03 \x03(\x0b\x32$.types.Checkpoint.ChannelValuesEntry\x12@\n\x10\x63hannel_versions\x18\x04 \x03(\x0b\x32&.types.Checkpoint.ChannelVersionsEntry\x12:\n\rversions_seen\x18\x05 \x03(\x0b\x32#.types.Checkpoint.VersionsSeenEntry\x12\n\n\x02ts\x18\x06 \x01(\t\x1a\x42\n\x12\x43hannelValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.types.Value:\x02\x38\x01\x1a\x36\n\x14\x43hannelVersionsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1aK\n\x11VersionsSeenEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.types.ChannelVersions:\x02\x38\x01\"\x9b\x01\n\x12\x43heckpointMetadata\x12\x0e\n\x06source\x18\x01 \x01(\t\x12\x0c\n\x04step\x18\x02 \x01(\x05\x12\x37\n\x07parents\x18\x03 \x03(\x0b\x32&.types.CheckpointMetadata.ParentsEntry\x1a.\n\x0cParentsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"m\n\x07Updates\x12%\n\ncheckpoint\x18\x01 \x01(\x0b\x32\x11.types.Checkpoint\x12!\n\x08\x63hannels\x18\x02 \x01(\x0b\x32\x0f.types.Channels\x12\x18\n\x10updated_channels\x18\x03 \x03(\t\"_\n\x0bStreamChunk\x12\n\n\x02ns\x18\x01 \x03(\t\x12\x11\n\x04mode\x18\x02 \x01(\tH\x00\x88\x01\x01\x12(\n\x07payload\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructB\x07\n\x05_mode\"3\n\x07Message\x12(\n\x07payload\x18\x01 \x01(\x0b\x32\x17.google.protobuf.Struct\"\xa2\x02\n\rStateSnapshot\x12\'\n\x06values\x18\x01 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x0c\n\x04next\x18\x02 \x03(\t\x12%\n\x06\x63onfig\x18\x03 \x01(\x0b\x32\x15.types.RunnableConfig\x12)\n\x08metadata\x18\x04 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x12\n\ncreated_at\x18\x05 \x01(\t\x12,\n\rparent_config\x18\x06 \x01(\x0b\x32\x15.types.RunnableConfig\x12 \n\x05tasks\x18\x07 \x03(\x0b\x32\x11.types.PregelTask\x12$\n\ninterrupts\x18\x08 \x03(\x0b\x32\x10.types.
|
28
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0btypes.proto\x12\x05types\x1a\x1cgoogle/protobuf/struct.proto\"\xa8\x02\n\x05Value\x12\x30\n\x10serialized_value\x18\x01 \x01(\x0b\x32\x16.types.SerializedValue\x12,\n\nbase_value\x18\x02 \x01(\x0b\x32\x16.types.SerializedValueH\x00\x12\x1d\n\x05sends\x18\x03 \x01(\x0b\x32\x0c.types.SendsH\x00\x12!\n\x07missing\x18\x04 \x01(\x0b\x32\x0e.types.MissingH\x00\x12!\n\x07\x63ommand\x18\x05 \x01(\x0b\x32\x0e.types.CommandH\x00\x12%\n\x05\x65rror\x18\x06 \x01(\x0b\x32\x14.types.ExecutorErrorH\x00\x12(\n\x0bmessage_ids\x18\x07 \x01(\x0b\x32\x11.types.MessageIdsH\x00\x42\t\n\x07message\"\t\n\x07Missing\"0\n\x0fSerializedValue\x12\x0e\n\x06method\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x0c\"!\n\nMessageIds\x12\x13\n\x0bmessage_ids\x18\x01 \x03(\t\"|\n\x08\x43hannels\x12/\n\x08\x63hannels\x18\x01 \x03(\x0b\x32\x1d.types.Channels.ChannelsEntry\x1a?\n\rChannelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1d\n\x05value\x18\x02 \x01(\x0b\x32\x0e.types.Channel:\x02\x38\x01\"q\n\x07\x43hannel\x12 \n\nget_result\x18\x01 \x01(\x0b\x32\x0c.types.Value\x12\x1b\n\x13is_available_result\x18\x02 \x01(\x08\x12\'\n\x11\x63heckpoint_result\x18\x03 \x01(\x0b\x32\x0c.types.Value\"#\n\x05Sends\x12\x1a\n\x05sends\x18\x01 \x03(\x0b\x32\x0b.types.Send\"/\n\x04Send\x12\x0c\n\x04node\x18\x01 \x01(\t\x12\x19\n\x03\x61rg\x18\x02 \x01(\x0b\x32\x0c.types.Value\"\xcb\x01\n\x07\x43ommand\x12\x12\n\x05graph\x18\x01 \x01(\tH\x00\x88\x01\x01\x12*\n\x06update\x18\x02 \x03(\x0b\x32\x1a.types.Command.UpdateEntry\x12\x1d\n\x06resume\x18\x03 \x01(\x0b\x32\r.types.Resume\x12\x1a\n\x05gotos\x18\x04 \x03(\x0b\x32\x0b.types.Goto\x1a;\n\x0bUpdateEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.types.Value:\x02\x38\x01\x42\x08\n\x06_graph\"\\\n\x06Resume\x12\x1d\n\x05value\x18\x01 \x01(\x0b\x32\x0c.types.ValueH\x00\x12(\n\x06values\x18\x02 \x01(\x0b\x32\x16.types.InterruptValuesH\x00\x42\t\n\x07message\"\x82\x01\n\x0fInterruptValues\x12\x32\n\x06values\x18\x01 \x03(\x0b\x32\".types.InterruptValues.ValuesEntry\x1a;\n\x0bValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.types.Value:\x02\x38\x01\"T\n\x04Goto\x12$\n\tnode_name\x18\x01 \x01(\x0b\x32\x0f.types.NodeNameH\x00\x12\x1b\n\x04send\x18\x02 \x01(\x0b\x32\x0b.types.SendH\x00\x42\t\n\x07message\"\x18\n\x08NodeName\x12\x0c\n\x04name\x18\x01 \x01(\t\"\xed\x01\n\rExecutorError\x12\x30\n\x10\x65rror_serialized\x18\x01 \x01(\x0b\x32\x16.types.SerializedValue\x12/\n\x0fgraph_bubble_up\x18\x02 \x01(\x0b\x32\x14.types.GraphBubbleUpH\x00\x12\x30\n\x0fgraph_interrupt\x18\x03 \x01(\x0b\x32\x15.types.GraphInterruptH\x00\x12&\n\nbase_error\x18\x04 \x01(\x0b\x32\x10.types.BaseErrorH\x00\x12\x11\n\ttraceback\x18\x05 \x01(\tB\x0c\n\nerror_type\"\x0f\n\rGraphBubbleUp\"m\n\x0eGraphInterrupt\x12$\n\ninterrupts\x18\x01 \x03(\x0b\x32\x10.types.Interrupt\x12\x35\n\x15interrupts_serialized\x18\x02 \x01(\x0b\x32\x16.types.SerializedValue\"h\n\tBaseError\x12\x12\n\nerror_type\x18\x01 \x01(\t\x12\x15\n\rerror_message\x18\x02 \x01(\t\x12\x30\n\x10\x65rror_serialized\x18\x03 \x01(\x0b\x32\x16.types.SerializedValue\"4\n\tInterrupt\x12\x1b\n\x05value\x18\x01 \x01(\x0b\x32\x0c.types.Value\x12\n\n\x02id\x18\x02 \x01(\t\"5\n\x05Write\x12\x0f\n\x07\x63hannel\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.types.Value\"M\n\x0cPendingWrite\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0f\n\x07\x63hannel\x18\x02 \x01(\t\x12\x1b\n\x05value\x18\x03 \x01(\x0b\x32\x0c.types.Value\"\x90\x01\n\x0f\x43hannelVersions\x12\x45\n\x10\x63hannel_versions\x18\x01 \x03(\x0b\x32+.types.ChannelVersions.ChannelVersionsEntry\x1a\x36\n\x14\x43hannelVersionsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x88\x02\n\x0eRunnableConfig\x12\x0c\n\x04tags\x18\x01 \x03(\t\x12)\n\x08metadata\x18\x02 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x10\n\x08run_name\x18\x03 \x01(\t\x12\x17\n\x0fmax_concurrency\x18\x04 \x01(\x05\x12\x17\n\x0frecursion_limit\x18\x05 \x01(\x05\x12-\n\x0c\x63onfigurable\x18\x06 \x01(\x0b\x32\x17.google.protobuf.Struct\x12:\n\x15reserved_configurable\x18\x07 \x01(\x0b\x32\x1b.types.ReservedConfigurable\x12\x0e\n\x06run_id\x18\x08 \x01(\t\"3\n\x07\x43ontext\x12(\n\x07\x63ontext\x18\x01 \x01(\x0b\x32\x17.google.protobuf.Struct\"\xe3\x03\n\x14ReservedConfigurable\x12\x10\n\x08resuming\x18\x01 \x01(\x08\x12\x0f\n\x07task_id\x18\x02 \x01(\t\x12\x14\n\x0c\x64\x65\x64upe_tasks\x18\x03 \x01(\x08\x12\x15\n\rensure_latest\x18\x04 \x01(\x08\x12\x11\n\tthread_id\x18\x05 \x01(\t\x12\x46\n\x0e\x63heckpoint_map\x18\x06 \x03(\x0b\x32..types.ReservedConfigurable.CheckpointMapEntry\x12\x15\n\rcheckpoint_id\x18\x07 \x01(\t\x12\x15\n\rcheckpoint_ns\x18\x08 \x01(\t\x12(\n\x08previous\x18\t \x01(\x0b\x32\x16.types.SerializedValue\x12\x12\n\ndurability\x18\n \x01(\t\x12>\n\nresume_map\x18\x0b \x03(\x0b\x32*.types.ReservedConfigurable.ResumeMapEntry\x1a\x34\n\x12\x43heckpointMapEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a>\n\x0eResumeMapEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.types.Value:\x02\x38\x01\"\xe1\x02\n\x04Task\x12\x0c\n\x04name\x18\x01 \x01(\t\x12%\n\x05input\x18\x02 \x03(\x0b\x32\x16.types.Task.InputEntry\x12\x1c\n\x06writes\x18\x03 \x03(\x0b\x32\x0c.types.Write\x12%\n\x06\x63onfig\x18\x04 \x01(\x0b\x32\x15.types.RunnableConfig\x12\x10\n\x08triggers\x18\x05 \x03(\t\x12\n\n\x02id\x18\x06 \x01(\t\x12\x11\n\ttask_path\x18\x07 \x03(\t\x12\x0f\n\x07writers\x18\x08 \x03(\t\x12\x12\n\ngraph_name\x18\t \x01(\t\x12+\n\x0epending_writes\x18\n \x03(\x0b\x32\x13.types.PendingWrite\x12 \n\x08messages\x18\x0b \x03(\x0b\x32\x0e.types.Message\x1a:\n\nInputEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.types.Value:\x02\x38\x01\"\xc8\x01\n\nPregelTask\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x0c\n\x04path\x18\x03 \x03(\t\x12$\n\ninterrupts\x18\x04 \x03(\x0b\x32\x10.types.Interrupt\x12&\n\x05state\x18\x05 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\'\n\x06result\x18\x06 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x1b\n\x05\x65rror\x18\x07 \x01(\x0b\x32\x0c.types.Value\"\xb4\x03\n\nCheckpoint\x12\t\n\x01v\x18\x01 \x01(\x04\x12\n\n\x02id\x18\x02 \x01(\t\x12<\n\x0e\x63hannel_values\x18\x03 \x03(\x0b\x32$.types.Checkpoint.ChannelValuesEntry\x12@\n\x10\x63hannel_versions\x18\x04 \x03(\x0b\x32&.types.Checkpoint.ChannelVersionsEntry\x12:\n\rversions_seen\x18\x05 \x03(\x0b\x32#.types.Checkpoint.VersionsSeenEntry\x12\n\n\x02ts\x18\x06 \x01(\t\x1a\x42\n\x12\x43hannelValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.types.Value:\x02\x38\x01\x1a\x36\n\x14\x43hannelVersionsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1aK\n\x11VersionsSeenEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.types.ChannelVersions:\x02\x38\x01\"\x9b\x01\n\x12\x43heckpointMetadata\x12\x0e\n\x06source\x18\x01 \x01(\t\x12\x0c\n\x04step\x18\x02 \x01(\x05\x12\x37\n\x07parents\x18\x03 \x03(\x0b\x32&.types.CheckpointMetadata.ParentsEntry\x1a.\n\x0cParentsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xe7\x01\n\x0f\x43heckpointTuple\x12%\n\x06\x63onfig\x18\x01 \x01(\x0b\x32\x15.types.RunnableConfig\x12%\n\ncheckpoint\x18\x02 \x01(\x0b\x32\x11.types.Checkpoint\x12+\n\x08metadata\x18\x03 \x01(\x0b\x32\x19.types.CheckpointMetadata\x12,\n\rparent_config\x18\x04 \x01(\x0b\x32\x15.types.RunnableConfig\x12+\n\x0epending_writes\x18\x05 \x03(\x0b\x32\x13.types.PendingWrite\"m\n\x07Updates\x12%\n\ncheckpoint\x18\x01 \x01(\x0b\x32\x11.types.Checkpoint\x12!\n\x08\x63hannels\x18\x02 \x01(\x0b\x32\x0f.types.Channels\x12\x18\n\x10updated_channels\x18\x03 \x03(\t\"_\n\x0bStreamChunk\x12\n\n\x02ns\x18\x01 \x03(\t\x12\x11\n\x04mode\x18\x02 \x01(\tH\x00\x88\x01\x01\x12(\n\x07payload\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructB\x07\n\x05_mode\"3\n\x07Message\x12(\n\x07payload\x18\x01 \x01(\x0b\x32\x17.google.protobuf.Struct\"\xa2\x02\n\rStateSnapshot\x12\'\n\x06values\x18\x01 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x0c\n\x04next\x18\x02 \x03(\t\x12%\n\x06\x63onfig\x18\x03 \x01(\x0b\x32\x15.types.RunnableConfig\x12)\n\x08metadata\x18\x04 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x12\n\ncreated_at\x18\x05 \x01(\t\x12,\n\rparent_config\x18\x06 \x01(\x0b\x32\x15.types.RunnableConfig\x12 \n\x05tasks\x18\x07 \x03(\x0b\x32\x11.types.PregelTask\x12$\n\ninterrupts\x18\x08 \x03(\x0b\x32\x10.types.Interrupt\"o\n\x0bStateUpdate\x12\x1c\n\x06values\x18\x01 \x01(\x0b\x32\x0c.types.Value\x12\x14\n\x07\x61s_node\x18\x02 \x01(\tH\x00\x88\x01\x01\x12\x14\n\x07task_id\x18\x03 \x01(\tH\x01\x88\x01\x01\x42\n\n\x08_as_nodeB\n\n\x08_task_id\"7\n\x10SuperstepUpdates\x12#\n\x07updates\x18\x01 \x03(\x0b\x32\x12.types.StateUpdate\"2\n\rStringOrSlice\x12\x0e\n\x06values\x18\x01 \x03(\t\x12\x11\n\tis_string\x18\x02 \x01(\x08\x42\x31Z/github.com/langchain-ai/langgraph-go/runtime/pbb\x06proto3')
|
29
29
|
|
30
30
|
_globals = globals()
|
31
31
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
@@ -133,12 +133,20 @@ if not _descriptor._USE_C_DESCRIPTORS:
|
|
133
133
|
_globals['_CHECKPOINTMETADATA']._serialized_end=4088
|
134
134
|
_globals['_CHECKPOINTMETADATA_PARENTSENTRY']._serialized_start=4042
|
135
135
|
_globals['_CHECKPOINTMETADATA_PARENTSENTRY']._serialized_end=4088
|
136
|
-
_globals['
|
137
|
-
_globals['
|
138
|
-
_globals['
|
139
|
-
_globals['
|
140
|
-
_globals['
|
141
|
-
_globals['
|
142
|
-
_globals['
|
143
|
-
_globals['
|
136
|
+
_globals['_CHECKPOINTTUPLE']._serialized_start=4091
|
137
|
+
_globals['_CHECKPOINTTUPLE']._serialized_end=4322
|
138
|
+
_globals['_UPDATES']._serialized_start=4324
|
139
|
+
_globals['_UPDATES']._serialized_end=4433
|
140
|
+
_globals['_STREAMCHUNK']._serialized_start=4435
|
141
|
+
_globals['_STREAMCHUNK']._serialized_end=4530
|
142
|
+
_globals['_MESSAGE']._serialized_start=4532
|
143
|
+
_globals['_MESSAGE']._serialized_end=4583
|
144
|
+
_globals['_STATESNAPSHOT']._serialized_start=4586
|
145
|
+
_globals['_STATESNAPSHOT']._serialized_end=4876
|
146
|
+
_globals['_STATEUPDATE']._serialized_start=4878
|
147
|
+
_globals['_STATEUPDATE']._serialized_end=4989
|
148
|
+
_globals['_SUPERSTEPUPDATES']._serialized_start=4991
|
149
|
+
_globals['_SUPERSTEPUPDATES']._serialized_end=5046
|
150
|
+
_globals['_STRINGORSLICE']._serialized_start=5048
|
151
|
+
_globals['_STRINGORSLICE']._serialized_end=5098
|
144
152
|
# @@protoc_insertion_point(module_scope)
|
@@ -924,6 +924,39 @@ class CheckpointMetadata(google.protobuf.message.Message):
|
|
924
924
|
|
925
925
|
global___CheckpointMetadata = CheckpointMetadata
|
926
926
|
|
927
|
+
@typing.final
|
928
|
+
class CheckpointTuple(google.protobuf.message.Message):
|
929
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
930
|
+
|
931
|
+
CONFIG_FIELD_NUMBER: builtins.int
|
932
|
+
CHECKPOINT_FIELD_NUMBER: builtins.int
|
933
|
+
METADATA_FIELD_NUMBER: builtins.int
|
934
|
+
PARENT_CONFIG_FIELD_NUMBER: builtins.int
|
935
|
+
PENDING_WRITES_FIELD_NUMBER: builtins.int
|
936
|
+
@property
|
937
|
+
def config(self) -> global___RunnableConfig: ...
|
938
|
+
@property
|
939
|
+
def checkpoint(self) -> global___Checkpoint: ...
|
940
|
+
@property
|
941
|
+
def metadata(self) -> global___CheckpointMetadata: ...
|
942
|
+
@property
|
943
|
+
def parent_config(self) -> global___RunnableConfig: ...
|
944
|
+
@property
|
945
|
+
def pending_writes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___PendingWrite]: ...
|
946
|
+
def __init__(
|
947
|
+
self,
|
948
|
+
*,
|
949
|
+
config: global___RunnableConfig | None = ...,
|
950
|
+
checkpoint: global___Checkpoint | None = ...,
|
951
|
+
metadata: global___CheckpointMetadata | None = ...,
|
952
|
+
parent_config: global___RunnableConfig | None = ...,
|
953
|
+
pending_writes: collections.abc.Iterable[global___PendingWrite] | None = ...,
|
954
|
+
) -> None: ...
|
955
|
+
def HasField(self, field_name: typing.Literal["checkpoint", b"checkpoint", "config", b"config", "metadata", b"metadata", "parent_config", b"parent_config"]) -> builtins.bool: ...
|
956
|
+
def ClearField(self, field_name: typing.Literal["checkpoint", b"checkpoint", "config", b"config", "metadata", b"metadata", "parent_config", b"parent_config", "pending_writes", b"pending_writes"]) -> None: ...
|
957
|
+
|
958
|
+
global___CheckpointTuple = CheckpointTuple
|
959
|
+
|
927
960
|
@typing.final
|
928
961
|
class Updates(google.protobuf.message.Message):
|
929
962
|
"""Updates"""
|
@@ -1042,3 +1075,65 @@ class StateSnapshot(google.protobuf.message.Message):
|
|
1042
1075
|
def ClearField(self, field_name: typing.Literal["config", b"config", "created_at", b"created_at", "interrupts", b"interrupts", "metadata", b"metadata", "next", b"next", "parent_config", b"parent_config", "tasks", b"tasks", "values", b"values"]) -> None: ...
|
1043
1076
|
|
1044
1077
|
global___StateSnapshot = StateSnapshot
|
1078
|
+
|
1079
|
+
@typing.final
|
1080
|
+
class StateUpdate(google.protobuf.message.Message):
|
1081
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
1082
|
+
|
1083
|
+
VALUES_FIELD_NUMBER: builtins.int
|
1084
|
+
AS_NODE_FIELD_NUMBER: builtins.int
|
1085
|
+
TASK_ID_FIELD_NUMBER: builtins.int
|
1086
|
+
as_node: builtins.str
|
1087
|
+
task_id: builtins.str
|
1088
|
+
@property
|
1089
|
+
def values(self) -> global___Value: ...
|
1090
|
+
def __init__(
|
1091
|
+
self,
|
1092
|
+
*,
|
1093
|
+
values: global___Value | None = ...,
|
1094
|
+
as_node: builtins.str | None = ...,
|
1095
|
+
task_id: builtins.str | None = ...,
|
1096
|
+
) -> None: ...
|
1097
|
+
def HasField(self, field_name: typing.Literal["_as_node", b"_as_node", "_task_id", b"_task_id", "as_node", b"as_node", "task_id", b"task_id", "values", b"values"]) -> builtins.bool: ...
|
1098
|
+
def ClearField(self, field_name: typing.Literal["_as_node", b"_as_node", "_task_id", b"_task_id", "as_node", b"as_node", "task_id", b"task_id", "values", b"values"]) -> None: ...
|
1099
|
+
@typing.overload
|
1100
|
+
def WhichOneof(self, oneof_group: typing.Literal["_as_node", b"_as_node"]) -> typing.Literal["as_node"] | None: ...
|
1101
|
+
@typing.overload
|
1102
|
+
def WhichOneof(self, oneof_group: typing.Literal["_task_id", b"_task_id"]) -> typing.Literal["task_id"] | None: ...
|
1103
|
+
|
1104
|
+
global___StateUpdate = StateUpdate
|
1105
|
+
|
1106
|
+
@typing.final
|
1107
|
+
class SuperstepUpdates(google.protobuf.message.Message):
|
1108
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
1109
|
+
|
1110
|
+
UPDATES_FIELD_NUMBER: builtins.int
|
1111
|
+
@property
|
1112
|
+
def updates(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___StateUpdate]: ...
|
1113
|
+
def __init__(
|
1114
|
+
self,
|
1115
|
+
*,
|
1116
|
+
updates: collections.abc.Iterable[global___StateUpdate] | None = ...,
|
1117
|
+
) -> None: ...
|
1118
|
+
def ClearField(self, field_name: typing.Literal["updates", b"updates"]) -> None: ...
|
1119
|
+
|
1120
|
+
global___SuperstepUpdates = SuperstepUpdates
|
1121
|
+
|
1122
|
+
@typing.final
|
1123
|
+
class StringOrSlice(google.protobuf.message.Message):
|
1124
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
1125
|
+
|
1126
|
+
VALUES_FIELD_NUMBER: builtins.int
|
1127
|
+
IS_STRING_FIELD_NUMBER: builtins.int
|
1128
|
+
is_string: builtins.bool
|
1129
|
+
@property
|
1130
|
+
def values(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
|
1131
|
+
def __init__(
|
1132
|
+
self,
|
1133
|
+
*,
|
1134
|
+
values: collections.abc.Iterable[builtins.str] | None = ...,
|
1135
|
+
is_string: builtins.bool = ...,
|
1136
|
+
) -> None: ...
|
1137
|
+
def ClearField(self, field_name: typing.Literal["is_string", b"is_string", "values", b"values"]) -> None: ...
|
1138
|
+
|
1139
|
+
global___StringOrSlice = StringOrSlice
|
@@ -0,0 +1,13 @@
|
|
1
|
+
from langgraph.checkpoint.serde.base import SerializerProtocol
|
2
|
+
from langgraph.checkpoint.serde.jsonplus import JsonPlusSerializer
|
3
|
+
|
4
|
+
SERIALIZER: SerializerProtocol = JsonPlusSerializer()
|
5
|
+
|
6
|
+
|
7
|
+
def set_serializer(serializer: SerializerProtocol) -> None:
|
8
|
+
global SERIALIZER
|
9
|
+
SERIALIZER = serializer
|
10
|
+
|
11
|
+
|
12
|
+
def get_serializer() -> SerializerProtocol:
|
13
|
+
return SERIALIZER
|