langgraph-api 0.5.2__py3-none-any.whl → 0.5.4__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.
@@ -0,0 +1,648 @@
1
+ from google.protobuf import struct_pb2 as _struct_pb2
2
+ from google.protobuf import wrappers_pb2 as _wrappers_pb2
3
+ from google.protobuf import empty_pb2 as _empty_pb2
4
+ from google.protobuf.internal import containers as _containers
5
+ from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
6
+ from google.protobuf import descriptor as _descriptor
7
+ from google.protobuf import message as _message
8
+ from collections.abc import Iterable as _Iterable, Mapping as _Mapping
9
+ from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
10
+
11
+ DESCRIPTOR: _descriptor.FileDescriptor
12
+
13
+ class Durability(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
14
+ __slots__ = ()
15
+ UNKNOWN: _ClassVar[Durability]
16
+ ASYNC: _ClassVar[Durability]
17
+ SYNC: _ClassVar[Durability]
18
+ EXIT: _ClassVar[Durability]
19
+ UNKNOWN: Durability
20
+ ASYNC: Durability
21
+ SYNC: Durability
22
+ EXIT: Durability
23
+
24
+ class Value(_message.Message):
25
+ __slots__ = ("base_value", "sends", "missing", "command")
26
+ BASE_VALUE_FIELD_NUMBER: _ClassVar[int]
27
+ SENDS_FIELD_NUMBER: _ClassVar[int]
28
+ MISSING_FIELD_NUMBER: _ClassVar[int]
29
+ COMMAND_FIELD_NUMBER: _ClassVar[int]
30
+ base_value: SerializedValue
31
+ sends: Sends
32
+ missing: Missing
33
+ command: Command
34
+ 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: ...
35
+
36
+ class Missing(_message.Message):
37
+ __slots__ = ()
38
+ def __init__(self) -> None: ...
39
+
40
+ class ChannelValue(_message.Message):
41
+ __slots__ = ("serialized_value", "sends", "missing")
42
+ SERIALIZED_VALUE_FIELD_NUMBER: _ClassVar[int]
43
+ SENDS_FIELD_NUMBER: _ClassVar[int]
44
+ MISSING_FIELD_NUMBER: _ClassVar[int]
45
+ serialized_value: SerializedValue
46
+ sends: Sends
47
+ missing: _empty_pb2.Empty
48
+ def __init__(self, serialized_value: _Optional[_Union[SerializedValue, _Mapping]] = ..., sends: _Optional[_Union[Sends, _Mapping]] = ..., missing: _Optional[_Union[_empty_pb2.Empty, _Mapping]] = ...) -> None: ...
49
+
50
+ class SerializedValue(_message.Message):
51
+ __slots__ = ("encoding", "value")
52
+ ENCODING_FIELD_NUMBER: _ClassVar[int]
53
+ VALUE_FIELD_NUMBER: _ClassVar[int]
54
+ encoding: str
55
+ value: bytes
56
+ def __init__(self, encoding: _Optional[str] = ..., value: _Optional[bytes] = ...) -> None: ...
57
+
58
+ class ResponseChunk(_message.Message):
59
+ __slots__ = ("namespaces", "mode", "payload")
60
+ NAMESPACES_FIELD_NUMBER: _ClassVar[int]
61
+ MODE_FIELD_NUMBER: _ClassVar[int]
62
+ PAYLOAD_FIELD_NUMBER: _ClassVar[int]
63
+ namespaces: _containers.RepeatedScalarFieldContainer[str]
64
+ mode: str
65
+ payload: SerializedValue
66
+ def __init__(self, namespaces: _Optional[_Iterable[str]] = ..., mode: _Optional[str] = ..., payload: _Optional[_Union[SerializedValue, _Mapping]] = ...) -> None: ...
67
+
68
+ class ResponseChunkList(_message.Message):
69
+ __slots__ = ("responses",)
70
+ RESPONSES_FIELD_NUMBER: _ClassVar[int]
71
+ responses: _containers.RepeatedCompositeFieldContainer[ResponseChunk]
72
+ def __init__(self, responses: _Optional[_Iterable[_Union[ResponseChunk, _Mapping]]] = ...) -> None: ...
73
+
74
+ class MessageIds(_message.Message):
75
+ __slots__ = ("message_ids",)
76
+ MESSAGE_IDS_FIELD_NUMBER: _ClassVar[int]
77
+ message_ids: _containers.RepeatedScalarFieldContainer[str]
78
+ def __init__(self, message_ids: _Optional[_Iterable[str]] = ...) -> None: ...
79
+
80
+ class Channels(_message.Message):
81
+ __slots__ = ("channels",)
82
+ class ChannelsEntry(_message.Message):
83
+ __slots__ = ("key", "value")
84
+ KEY_FIELD_NUMBER: _ClassVar[int]
85
+ VALUE_FIELD_NUMBER: _ClassVar[int]
86
+ key: str
87
+ value: Channel
88
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Channel, _Mapping]] = ...) -> None: ...
89
+ CHANNELS_FIELD_NUMBER: _ClassVar[int]
90
+ channels: _containers.MessageMap[str, Channel]
91
+ def __init__(self, channels: _Optional[_Mapping[str, Channel]] = ...) -> None: ...
92
+
93
+ class Channel(_message.Message):
94
+ __slots__ = ("get_result", "is_available_result", "checkpoint_result")
95
+ GET_RESULT_FIELD_NUMBER: _ClassVar[int]
96
+ IS_AVAILABLE_RESULT_FIELD_NUMBER: _ClassVar[int]
97
+ CHECKPOINT_RESULT_FIELD_NUMBER: _ClassVar[int]
98
+ get_result: Value
99
+ is_available_result: bool
100
+ checkpoint_result: Value
101
+ def __init__(self, get_result: _Optional[_Union[Value, _Mapping]] = ..., is_available_result: bool = ..., checkpoint_result: _Optional[_Union[Value, _Mapping]] = ...) -> None: ...
102
+
103
+ class Sends(_message.Message):
104
+ __slots__ = ("sends",)
105
+ SENDS_FIELD_NUMBER: _ClassVar[int]
106
+ sends: _containers.RepeatedCompositeFieldContainer[Send]
107
+ def __init__(self, sends: _Optional[_Iterable[_Union[Send, _Mapping]]] = ...) -> None: ...
108
+
109
+ class Send(_message.Message):
110
+ __slots__ = ("node", "arg")
111
+ NODE_FIELD_NUMBER: _ClassVar[int]
112
+ ARG_FIELD_NUMBER: _ClassVar[int]
113
+ node: str
114
+ arg: SerializedValue
115
+ def __init__(self, node: _Optional[str] = ..., arg: _Optional[_Union[SerializedValue, _Mapping]] = ...) -> None: ...
116
+
117
+ class Command(_message.Message):
118
+ __slots__ = ("graph", "update", "resume", "gotos")
119
+ class UpdateEntry(_message.Message):
120
+ __slots__ = ("key", "value")
121
+ KEY_FIELD_NUMBER: _ClassVar[int]
122
+ VALUE_FIELD_NUMBER: _ClassVar[int]
123
+ key: str
124
+ value: SerializedValue
125
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[SerializedValue, _Mapping]] = ...) -> None: ...
126
+ GRAPH_FIELD_NUMBER: _ClassVar[int]
127
+ UPDATE_FIELD_NUMBER: _ClassVar[int]
128
+ RESUME_FIELD_NUMBER: _ClassVar[int]
129
+ GOTOS_FIELD_NUMBER: _ClassVar[int]
130
+ graph: str
131
+ update: _containers.MessageMap[str, SerializedValue]
132
+ resume: Resume
133
+ gotos: _containers.RepeatedCompositeFieldContainer[Goto]
134
+ def __init__(self, graph: _Optional[str] = ..., update: _Optional[_Mapping[str, SerializedValue]] = ..., resume: _Optional[_Union[Resume, _Mapping]] = ..., gotos: _Optional[_Iterable[_Union[Goto, _Mapping]]] = ...) -> None: ...
135
+
136
+ class Resume(_message.Message):
137
+ __slots__ = ("value", "values")
138
+ VALUE_FIELD_NUMBER: _ClassVar[int]
139
+ VALUES_FIELD_NUMBER: _ClassVar[int]
140
+ value: SerializedValue
141
+ values: InterruptValues
142
+ def __init__(self, value: _Optional[_Union[SerializedValue, _Mapping]] = ..., values: _Optional[_Union[InterruptValues, _Mapping]] = ...) -> None: ...
143
+
144
+ class InterruptValues(_message.Message):
145
+ __slots__ = ("values",)
146
+ class ValuesEntry(_message.Message):
147
+ __slots__ = ("key", "value")
148
+ KEY_FIELD_NUMBER: _ClassVar[int]
149
+ VALUE_FIELD_NUMBER: _ClassVar[int]
150
+ key: str
151
+ value: SerializedValue
152
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[SerializedValue, _Mapping]] = ...) -> None: ...
153
+ VALUES_FIELD_NUMBER: _ClassVar[int]
154
+ values: _containers.MessageMap[str, SerializedValue]
155
+ def __init__(self, values: _Optional[_Mapping[str, SerializedValue]] = ...) -> None: ...
156
+
157
+ class Goto(_message.Message):
158
+ __slots__ = ("node_name", "send")
159
+ NODE_NAME_FIELD_NUMBER: _ClassVar[int]
160
+ SEND_FIELD_NUMBER: _ClassVar[int]
161
+ node_name: str
162
+ send: Send
163
+ def __init__(self, node_name: _Optional[str] = ..., send: _Optional[_Union[Send, _Mapping]] = ...) -> None: ...
164
+
165
+ class GraphBubbleUp(_message.Message):
166
+ __slots__ = ()
167
+ def __init__(self) -> None: ...
168
+
169
+ class UserCodeExecutionError(_message.Message):
170
+ __slots__ = ("error_type", "error_message", "traceback")
171
+ ERROR_TYPE_FIELD_NUMBER: _ClassVar[int]
172
+ ERROR_MESSAGE_FIELD_NUMBER: _ClassVar[int]
173
+ TRACEBACK_FIELD_NUMBER: _ClassVar[int]
174
+ error_type: str
175
+ error_message: str
176
+ traceback: str
177
+ def __init__(self, error_type: _Optional[str] = ..., error_message: _Optional[str] = ..., traceback: _Optional[str] = ...) -> None: ...
178
+
179
+ class GraphInterrupt(_message.Message):
180
+ __slots__ = ("interrupts",)
181
+ INTERRUPTS_FIELD_NUMBER: _ClassVar[int]
182
+ interrupts: _containers.RepeatedCompositeFieldContainer[Interrupt]
183
+ def __init__(self, interrupts: _Optional[_Iterable[_Union[Interrupt, _Mapping]]] = ...) -> None: ...
184
+
185
+ class Interrupt(_message.Message):
186
+ __slots__ = ("value", "id")
187
+ VALUE_FIELD_NUMBER: _ClassVar[int]
188
+ ID_FIELD_NUMBER: _ClassVar[int]
189
+ value: SerializedValue
190
+ id: str
191
+ def __init__(self, value: _Optional[_Union[SerializedValue, _Mapping]] = ..., id: _Optional[str] = ...) -> None: ...
192
+
193
+ class WrappedInterrupt(_message.Message):
194
+ __slots__ = ("interrupt", "serialized_interrupt")
195
+ INTERRUPT_FIELD_NUMBER: _ClassVar[int]
196
+ SERIALIZED_INTERRUPT_FIELD_NUMBER: _ClassVar[int]
197
+ interrupt: Interrupt
198
+ serialized_interrupt: SerializedValue
199
+ def __init__(self, interrupt: _Optional[_Union[Interrupt, _Mapping]] = ..., serialized_interrupt: _Optional[_Union[SerializedValue, _Mapping]] = ...) -> None: ...
200
+
201
+ class Write(_message.Message):
202
+ __slots__ = ("channel", "value")
203
+ CHANNEL_FIELD_NUMBER: _ClassVar[int]
204
+ VALUE_FIELD_NUMBER: _ClassVar[int]
205
+ channel: str
206
+ value: Value
207
+ def __init__(self, channel: _Optional[str] = ..., value: _Optional[_Union[Value, _Mapping]] = ...) -> None: ...
208
+
209
+ class PendingWrite(_message.Message):
210
+ __slots__ = ("task_id", "channel", "value")
211
+ TASK_ID_FIELD_NUMBER: _ClassVar[int]
212
+ CHANNEL_FIELD_NUMBER: _ClassVar[int]
213
+ VALUE_FIELD_NUMBER: _ClassVar[int]
214
+ task_id: str
215
+ channel: str
216
+ value: Value
217
+ def __init__(self, task_id: _Optional[str] = ..., channel: _Optional[str] = ..., value: _Optional[_Union[Value, _Mapping]] = ...) -> None: ...
218
+
219
+ class ChannelVersions(_message.Message):
220
+ __slots__ = ("channel_versions",)
221
+ class ChannelVersionsEntry(_message.Message):
222
+ __slots__ = ("key", "value")
223
+ KEY_FIELD_NUMBER: _ClassVar[int]
224
+ VALUE_FIELD_NUMBER: _ClassVar[int]
225
+ key: str
226
+ value: str
227
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
228
+ CHANNEL_VERSIONS_FIELD_NUMBER: _ClassVar[int]
229
+ channel_versions: _containers.ScalarMap[str, str]
230
+ def __init__(self, channel_versions: _Optional[_Mapping[str, str]] = ...) -> None: ...
231
+
232
+ class EngineRunnableConfig(_message.Message):
233
+ __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")
234
+ class MetadataJsonEntry(_message.Message):
235
+ __slots__ = ("key", "value")
236
+ KEY_FIELD_NUMBER: _ClassVar[int]
237
+ VALUE_FIELD_NUMBER: _ClassVar[int]
238
+ key: str
239
+ value: bytes
240
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[bytes] = ...) -> None: ...
241
+ class ExtraConfigurableJsonEntry(_message.Message):
242
+ __slots__ = ("key", "value")
243
+ KEY_FIELD_NUMBER: _ClassVar[int]
244
+ VALUE_FIELD_NUMBER: _ClassVar[int]
245
+ key: str
246
+ value: bytes
247
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[bytes] = ...) -> None: ...
248
+ class ExtraJsonEntry(_message.Message):
249
+ __slots__ = ("key", "value")
250
+ KEY_FIELD_NUMBER: _ClassVar[int]
251
+ VALUE_FIELD_NUMBER: _ClassVar[int]
252
+ key: str
253
+ value: bytes
254
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[bytes] = ...) -> None: ...
255
+ class CheckpointMapEntry(_message.Message):
256
+ __slots__ = ("key", "value")
257
+ KEY_FIELD_NUMBER: _ClassVar[int]
258
+ VALUE_FIELD_NUMBER: _ClassVar[int]
259
+ key: str
260
+ value: str
261
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
262
+ class ResumeMapEntry(_message.Message):
263
+ __slots__ = ("key", "value")
264
+ KEY_FIELD_NUMBER: _ClassVar[int]
265
+ VALUE_FIELD_NUMBER: _ClassVar[int]
266
+ key: str
267
+ value: SerializedValue
268
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[SerializedValue, _Mapping]] = ...) -> None: ...
269
+ TAGS_FIELD_NUMBER: _ClassVar[int]
270
+ METADATA_JSON_FIELD_NUMBER: _ClassVar[int]
271
+ RUN_NAME_FIELD_NUMBER: _ClassVar[int]
272
+ MAX_CONCURRENCY_FIELD_NUMBER: _ClassVar[int]
273
+ RECURSION_LIMIT_FIELD_NUMBER: _ClassVar[int]
274
+ RUN_ID_FIELD_NUMBER: _ClassVar[int]
275
+ EXTRA_CONFIGURABLE_JSON_FIELD_NUMBER: _ClassVar[int]
276
+ EXTRA_JSON_FIELD_NUMBER: _ClassVar[int]
277
+ RUNTIME_FIELD_NUMBER: _ClassVar[int]
278
+ RESUMING_FIELD_NUMBER: _ClassVar[int]
279
+ TASK_ID_FIELD_NUMBER: _ClassVar[int]
280
+ THREAD_ID_FIELD_NUMBER: _ClassVar[int]
281
+ CHECKPOINT_MAP_FIELD_NUMBER: _ClassVar[int]
282
+ CHECKPOINT_ID_FIELD_NUMBER: _ClassVar[int]
283
+ CHECKPOINT_NS_FIELD_NUMBER: _ClassVar[int]
284
+ DURABILITY_FIELD_NUMBER: _ClassVar[int]
285
+ RESUME_MAP_FIELD_NUMBER: _ClassVar[int]
286
+ GRAPH_ID_FIELD_NUMBER: _ClassVar[int]
287
+ RUN_ATTEMPT_FIELD_NUMBER: _ClassVar[int]
288
+ SERVER_RUN_ID_FIELD_NUMBER: _ClassVar[int]
289
+ tags: _containers.RepeatedScalarFieldContainer[str]
290
+ metadata_json: _containers.ScalarMap[str, bytes]
291
+ run_name: str
292
+ max_concurrency: int
293
+ recursion_limit: int
294
+ run_id: str
295
+ extra_configurable_json: _containers.ScalarMap[str, bytes]
296
+ extra_json: _containers.ScalarMap[str, bytes]
297
+ runtime: Runtime
298
+ resuming: bool
299
+ task_id: str
300
+ thread_id: str
301
+ checkpoint_map: _containers.ScalarMap[str, str]
302
+ checkpoint_id: str
303
+ checkpoint_ns: str
304
+ durability: Durability
305
+ resume_map: _containers.MessageMap[str, SerializedValue]
306
+ graph_id: str
307
+ run_attempt: int
308
+ server_run_id: str
309
+ 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: ...
310
+
311
+ class Runtime(_message.Message):
312
+ __slots__ = ("langgraph_context_json",)
313
+ LANGGRAPH_CONTEXT_JSON_FIELD_NUMBER: _ClassVar[int]
314
+ langgraph_context_json: bytes
315
+ def __init__(self, langgraph_context_json: _Optional[bytes] = ...) -> None: ...
316
+
317
+ class Task(_message.Message):
318
+ __slots__ = ("name", "input", "writes", "config", "triggers", "id", "task_path", "writers", "graph_name", "pending_writes", "messages", "stream_subgraphs")
319
+ class InputEntry(_message.Message):
320
+ __slots__ = ("key", "value")
321
+ KEY_FIELD_NUMBER: _ClassVar[int]
322
+ VALUE_FIELD_NUMBER: _ClassVar[int]
323
+ key: str
324
+ value: Value
325
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Value, _Mapping]] = ...) -> None: ...
326
+ NAME_FIELD_NUMBER: _ClassVar[int]
327
+ INPUT_FIELD_NUMBER: _ClassVar[int]
328
+ WRITES_FIELD_NUMBER: _ClassVar[int]
329
+ CONFIG_FIELD_NUMBER: _ClassVar[int]
330
+ TRIGGERS_FIELD_NUMBER: _ClassVar[int]
331
+ ID_FIELD_NUMBER: _ClassVar[int]
332
+ TASK_PATH_FIELD_NUMBER: _ClassVar[int]
333
+ WRITERS_FIELD_NUMBER: _ClassVar[int]
334
+ GRAPH_NAME_FIELD_NUMBER: _ClassVar[int]
335
+ PENDING_WRITES_FIELD_NUMBER: _ClassVar[int]
336
+ MESSAGES_FIELD_NUMBER: _ClassVar[int]
337
+ STREAM_SUBGRAPHS_FIELD_NUMBER: _ClassVar[int]
338
+ name: str
339
+ input: _containers.MessageMap[str, Value]
340
+ writes: _containers.RepeatedCompositeFieldContainer[Write]
341
+ config: EngineRunnableConfig
342
+ triggers: _containers.RepeatedScalarFieldContainer[str]
343
+ id: str
344
+ task_path: _containers.RepeatedCompositeFieldContainer[PathSegment]
345
+ writers: _containers.RepeatedScalarFieldContainer[str]
346
+ graph_name: str
347
+ pending_writes: _containers.RepeatedCompositeFieldContainer[PendingWrite]
348
+ messages: _containers.RepeatedCompositeFieldContainer[ChatMessageEnvelope]
349
+ stream_subgraphs: bool
350
+ 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: ...
351
+
352
+ class PregelTaskSnapshot(_message.Message):
353
+ __slots__ = ("id", "name", "path", "interrupts", "state", "result", "error")
354
+ ID_FIELD_NUMBER: _ClassVar[int]
355
+ NAME_FIELD_NUMBER: _ClassVar[int]
356
+ PATH_FIELD_NUMBER: _ClassVar[int]
357
+ INTERRUPTS_FIELD_NUMBER: _ClassVar[int]
358
+ STATE_FIELD_NUMBER: _ClassVar[int]
359
+ RESULT_FIELD_NUMBER: _ClassVar[int]
360
+ ERROR_FIELD_NUMBER: _ClassVar[int]
361
+ id: str
362
+ name: str
363
+ path: _containers.RepeatedCompositeFieldContainer[PathSegment]
364
+ interrupts: _containers.RepeatedCompositeFieldContainer[Interrupt]
365
+ state: _struct_pb2.Struct
366
+ result: _struct_pb2.Struct
367
+ error: SerializedValue
368
+ 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: ...
369
+
370
+ class Checkpoint(_message.Message):
371
+ __slots__ = ("v", "id", "channel_values", "channel_versions", "versions_seen", "ts")
372
+ class ChannelValuesEntry(_message.Message):
373
+ __slots__ = ("key", "value")
374
+ KEY_FIELD_NUMBER: _ClassVar[int]
375
+ VALUE_FIELD_NUMBER: _ClassVar[int]
376
+ key: str
377
+ value: Value
378
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Value, _Mapping]] = ...) -> None: ...
379
+ class ChannelVersionsEntry(_message.Message):
380
+ __slots__ = ("key", "value")
381
+ KEY_FIELD_NUMBER: _ClassVar[int]
382
+ VALUE_FIELD_NUMBER: _ClassVar[int]
383
+ key: str
384
+ value: str
385
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
386
+ class VersionsSeenEntry(_message.Message):
387
+ __slots__ = ("key", "value")
388
+ KEY_FIELD_NUMBER: _ClassVar[int]
389
+ VALUE_FIELD_NUMBER: _ClassVar[int]
390
+ key: str
391
+ value: ChannelVersions
392
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[ChannelVersions, _Mapping]] = ...) -> None: ...
393
+ V_FIELD_NUMBER: _ClassVar[int]
394
+ ID_FIELD_NUMBER: _ClassVar[int]
395
+ CHANNEL_VALUES_FIELD_NUMBER: _ClassVar[int]
396
+ CHANNEL_VERSIONS_FIELD_NUMBER: _ClassVar[int]
397
+ VERSIONS_SEEN_FIELD_NUMBER: _ClassVar[int]
398
+ TS_FIELD_NUMBER: _ClassVar[int]
399
+ v: int
400
+ id: str
401
+ channel_values: _containers.MessageMap[str, Value]
402
+ channel_versions: _containers.ScalarMap[str, str]
403
+ versions_seen: _containers.MessageMap[str, ChannelVersions]
404
+ ts: str
405
+ 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: ...
406
+
407
+ class CheckpointMetadata(_message.Message):
408
+ __slots__ = ("source", "step", "parents", "run_id")
409
+ class CheckpointSource(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
410
+ __slots__ = ()
411
+ unknown: _ClassVar[CheckpointMetadata.CheckpointSource]
412
+ loop: _ClassVar[CheckpointMetadata.CheckpointSource]
413
+ input: _ClassVar[CheckpointMetadata.CheckpointSource]
414
+ update: _ClassVar[CheckpointMetadata.CheckpointSource]
415
+ fork: _ClassVar[CheckpointMetadata.CheckpointSource]
416
+ unknown: CheckpointMetadata.CheckpointSource
417
+ loop: CheckpointMetadata.CheckpointSource
418
+ input: CheckpointMetadata.CheckpointSource
419
+ update: CheckpointMetadata.CheckpointSource
420
+ fork: CheckpointMetadata.CheckpointSource
421
+ class ParentsEntry(_message.Message):
422
+ __slots__ = ("key", "value")
423
+ KEY_FIELD_NUMBER: _ClassVar[int]
424
+ VALUE_FIELD_NUMBER: _ClassVar[int]
425
+ key: str
426
+ value: str
427
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
428
+ SOURCE_FIELD_NUMBER: _ClassVar[int]
429
+ STEP_FIELD_NUMBER: _ClassVar[int]
430
+ PARENTS_FIELD_NUMBER: _ClassVar[int]
431
+ RUN_ID_FIELD_NUMBER: _ClassVar[int]
432
+ source: CheckpointMetadata.CheckpointSource
433
+ step: int
434
+ parents: _containers.ScalarMap[str, str]
435
+ run_id: str
436
+ def __init__(self, source: _Optional[_Union[CheckpointMetadata.CheckpointSource, str]] = ..., step: _Optional[int] = ..., parents: _Optional[_Mapping[str, str]] = ..., run_id: _Optional[str] = ...) -> None: ...
437
+
438
+ class CheckpointTuple(_message.Message):
439
+ __slots__ = ("config", "checkpoint", "metadata", "parent_config", "pending_writes")
440
+ CONFIG_FIELD_NUMBER: _ClassVar[int]
441
+ CHECKPOINT_FIELD_NUMBER: _ClassVar[int]
442
+ METADATA_FIELD_NUMBER: _ClassVar[int]
443
+ PARENT_CONFIG_FIELD_NUMBER: _ClassVar[int]
444
+ PENDING_WRITES_FIELD_NUMBER: _ClassVar[int]
445
+ config: EngineRunnableConfig
446
+ checkpoint: Checkpoint
447
+ metadata: CheckpointMetadata
448
+ parent_config: EngineRunnableConfig
449
+ pending_writes: _containers.RepeatedCompositeFieldContainer[PendingWrite]
450
+ def __init__(self, config: _Optional[_Union[EngineRunnableConfig, _Mapping]] = ..., checkpoint: _Optional[_Union[Checkpoint, _Mapping]] = ..., metadata: _Optional[_Union[CheckpointMetadata, _Mapping]] = ..., parent_config: _Optional[_Union[EngineRunnableConfig, _Mapping]] = ..., pending_writes: _Optional[_Iterable[_Union[PendingWrite, _Mapping]]] = ...) -> None: ...
451
+
452
+ class Updates(_message.Message):
453
+ __slots__ = ("checkpoint", "channels", "updated_channels")
454
+ CHECKPOINT_FIELD_NUMBER: _ClassVar[int]
455
+ CHANNELS_FIELD_NUMBER: _ClassVar[int]
456
+ UPDATED_CHANNELS_FIELD_NUMBER: _ClassVar[int]
457
+ checkpoint: Checkpoint
458
+ channels: Channels
459
+ updated_channels: _containers.RepeatedScalarFieldContainer[str]
460
+ def __init__(self, checkpoint: _Optional[_Union[Checkpoint, _Mapping]] = ..., channels: _Optional[_Union[Channels, _Mapping]] = ..., updated_channels: _Optional[_Iterable[str]] = ...) -> None: ...
461
+
462
+ class ToolCall(_message.Message):
463
+ __slots__ = ("name", "args", "id")
464
+ NAME_FIELD_NUMBER: _ClassVar[int]
465
+ ARGS_FIELD_NUMBER: _ClassVar[int]
466
+ ID_FIELD_NUMBER: _ClassVar[int]
467
+ name: str
468
+ args: _struct_pb2.Struct
469
+ id: _wrappers_pb2.StringValue
470
+ def __init__(self, name: _Optional[str] = ..., args: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., id: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
471
+
472
+ class ToolCallChunk(_message.Message):
473
+ __slots__ = ("name", "args_json", "id", "index")
474
+ NAME_FIELD_NUMBER: _ClassVar[int]
475
+ ARGS_JSON_FIELD_NUMBER: _ClassVar[int]
476
+ ID_FIELD_NUMBER: _ClassVar[int]
477
+ INDEX_FIELD_NUMBER: _ClassVar[int]
478
+ name: _wrappers_pb2.StringValue
479
+ args_json: _wrappers_pb2.StringValue
480
+ id: _wrappers_pb2.StringValue
481
+ index: _wrappers_pb2.Int32Value
482
+ def __init__(self, name: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., args_json: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., id: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., index: _Optional[_Union[_wrappers_pb2.Int32Value, _Mapping]] = ...) -> None: ...
483
+
484
+ class InvalidToolCall(_message.Message):
485
+ __slots__ = ("name", "args", "id", "error")
486
+ NAME_FIELD_NUMBER: _ClassVar[int]
487
+ ARGS_FIELD_NUMBER: _ClassVar[int]
488
+ ID_FIELD_NUMBER: _ClassVar[int]
489
+ ERROR_FIELD_NUMBER: _ClassVar[int]
490
+ name: _wrappers_pb2.StringValue
491
+ args: _wrappers_pb2.StringValue
492
+ id: _wrappers_pb2.StringValue
493
+ error: _wrappers_pb2.StringValue
494
+ def __init__(self, name: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., args: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., id: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., error: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
495
+
496
+ class InputTokenDetails(_message.Message):
497
+ __slots__ = ("audio", "cache_creation", "cache_read")
498
+ AUDIO_FIELD_NUMBER: _ClassVar[int]
499
+ CACHE_CREATION_FIELD_NUMBER: _ClassVar[int]
500
+ CACHE_READ_FIELD_NUMBER: _ClassVar[int]
501
+ audio: _wrappers_pb2.Int64Value
502
+ cache_creation: _wrappers_pb2.Int64Value
503
+ cache_read: _wrappers_pb2.Int64Value
504
+ def __init__(self, audio: _Optional[_Union[_wrappers_pb2.Int64Value, _Mapping]] = ..., cache_creation: _Optional[_Union[_wrappers_pb2.Int64Value, _Mapping]] = ..., cache_read: _Optional[_Union[_wrappers_pb2.Int64Value, _Mapping]] = ...) -> None: ...
505
+
506
+ class OutputTokenDetails(_message.Message):
507
+ __slots__ = ("audio", "reasoning")
508
+ AUDIO_FIELD_NUMBER: _ClassVar[int]
509
+ REASONING_FIELD_NUMBER: _ClassVar[int]
510
+ audio: _wrappers_pb2.Int64Value
511
+ reasoning: _wrappers_pb2.Int64Value
512
+ def __init__(self, audio: _Optional[_Union[_wrappers_pb2.Int64Value, _Mapping]] = ..., reasoning: _Optional[_Union[_wrappers_pb2.Int64Value, _Mapping]] = ...) -> None: ...
513
+
514
+ class UsageMetadata(_message.Message):
515
+ __slots__ = ("input_tokens", "output_tokens", "total_tokens", "input_token_details", "output_token_details")
516
+ INPUT_TOKENS_FIELD_NUMBER: _ClassVar[int]
517
+ OUTPUT_TOKENS_FIELD_NUMBER: _ClassVar[int]
518
+ TOTAL_TOKENS_FIELD_NUMBER: _ClassVar[int]
519
+ INPUT_TOKEN_DETAILS_FIELD_NUMBER: _ClassVar[int]
520
+ OUTPUT_TOKEN_DETAILS_FIELD_NUMBER: _ClassVar[int]
521
+ input_tokens: int
522
+ output_tokens: int
523
+ total_tokens: int
524
+ input_token_details: InputTokenDetails
525
+ output_token_details: OutputTokenDetails
526
+ def __init__(self, input_tokens: _Optional[int] = ..., output_tokens: _Optional[int] = ..., total_tokens: _Optional[int] = ..., input_token_details: _Optional[_Union[InputTokenDetails, _Mapping]] = ..., output_token_details: _Optional[_Union[OutputTokenDetails, _Mapping]] = ...) -> None: ...
527
+
528
+ class ResponseMetadata(_message.Message):
529
+ __slots__ = ("data",)
530
+ DATA_FIELD_NUMBER: _ClassVar[int]
531
+ data: _struct_pb2.Struct
532
+ def __init__(self, data: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
533
+
534
+ class AIFields(_message.Message):
535
+ __slots__ = ("usage_metadata", "response_metadata", "tool_calls", "tool_call_chunks", "invalid_tool_calls", "reasoning_content")
536
+ USAGE_METADATA_FIELD_NUMBER: _ClassVar[int]
537
+ RESPONSE_METADATA_FIELD_NUMBER: _ClassVar[int]
538
+ TOOL_CALLS_FIELD_NUMBER: _ClassVar[int]
539
+ TOOL_CALL_CHUNKS_FIELD_NUMBER: _ClassVar[int]
540
+ INVALID_TOOL_CALLS_FIELD_NUMBER: _ClassVar[int]
541
+ REASONING_CONTENT_FIELD_NUMBER: _ClassVar[int]
542
+ usage_metadata: UsageMetadata
543
+ response_metadata: ResponseMetadata
544
+ tool_calls: _containers.RepeatedCompositeFieldContainer[ToolCall]
545
+ tool_call_chunks: _containers.RepeatedCompositeFieldContainer[ToolCallChunk]
546
+ invalid_tool_calls: _containers.RepeatedCompositeFieldContainer[InvalidToolCall]
547
+ reasoning_content: _wrappers_pb2.StringValue
548
+ 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: ...
549
+
550
+ class ToolFields(_message.Message):
551
+ __slots__ = ("tool_call_id", "status")
552
+ TOOL_CALL_ID_FIELD_NUMBER: _ClassVar[int]
553
+ STATUS_FIELD_NUMBER: _ClassVar[int]
554
+ tool_call_id: _wrappers_pb2.StringValue
555
+ status: str
556
+ def __init__(self, tool_call_id: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., status: _Optional[str] = ...) -> None: ...
557
+
558
+ class HumanFields(_message.Message):
559
+ __slots__ = ()
560
+ def __init__(self) -> None: ...
561
+
562
+ class ChatMessage(_message.Message):
563
+ __slots__ = ("id", "name", "type", "content", "additional_kwargs", "ai", "tool", "human", "extensions")
564
+ ID_FIELD_NUMBER: _ClassVar[int]
565
+ NAME_FIELD_NUMBER: _ClassVar[int]
566
+ TYPE_FIELD_NUMBER: _ClassVar[int]
567
+ CONTENT_FIELD_NUMBER: _ClassVar[int]
568
+ ADDITIONAL_KWARGS_FIELD_NUMBER: _ClassVar[int]
569
+ AI_FIELD_NUMBER: _ClassVar[int]
570
+ TOOL_FIELD_NUMBER: _ClassVar[int]
571
+ HUMAN_FIELD_NUMBER: _ClassVar[int]
572
+ EXTENSIONS_FIELD_NUMBER: _ClassVar[int]
573
+ id: str
574
+ name: str
575
+ type: str
576
+ content: _struct_pb2.Value
577
+ additional_kwargs: _struct_pb2.Struct
578
+ ai: AIFields
579
+ tool: ToolFields
580
+ human: HumanFields
581
+ extensions: _struct_pb2.Struct
582
+ 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: ...
583
+
584
+ class ChatMessageEnvelope(_message.Message):
585
+ __slots__ = ("is_streaming_chunk", "namespace", "message", "metadata")
586
+ IS_STREAMING_CHUNK_FIELD_NUMBER: _ClassVar[int]
587
+ NAMESPACE_FIELD_NUMBER: _ClassVar[int]
588
+ MESSAGE_FIELD_NUMBER: _ClassVar[int]
589
+ METADATA_FIELD_NUMBER: _ClassVar[int]
590
+ is_streaming_chunk: bool
591
+ namespace: _containers.RepeatedScalarFieldContainer[str]
592
+ message: ChatMessage
593
+ metadata: bytes
594
+ def __init__(self, is_streaming_chunk: bool = ..., namespace: _Optional[_Iterable[str]] = ..., message: _Optional[_Union[ChatMessage, _Mapping]] = ..., metadata: _Optional[bytes] = ...) -> None: ...
595
+
596
+ class StateSnapshot(_message.Message):
597
+ __slots__ = ("values", "next", "config", "metadata", "created_at", "parent_config", "tasks", "interrupts")
598
+ VALUES_FIELD_NUMBER: _ClassVar[int]
599
+ NEXT_FIELD_NUMBER: _ClassVar[int]
600
+ CONFIG_FIELD_NUMBER: _ClassVar[int]
601
+ METADATA_FIELD_NUMBER: _ClassVar[int]
602
+ CREATED_AT_FIELD_NUMBER: _ClassVar[int]
603
+ PARENT_CONFIG_FIELD_NUMBER: _ClassVar[int]
604
+ TASKS_FIELD_NUMBER: _ClassVar[int]
605
+ INTERRUPTS_FIELD_NUMBER: _ClassVar[int]
606
+ values: _struct_pb2.Struct
607
+ next: _containers.RepeatedScalarFieldContainer[str]
608
+ config: EngineRunnableConfig
609
+ metadata: CheckpointMetadata
610
+ created_at: str
611
+ parent_config: EngineRunnableConfig
612
+ tasks: _containers.RepeatedCompositeFieldContainer[PregelTaskSnapshot]
613
+ interrupts: _containers.RepeatedCompositeFieldContainer[Interrupt]
614
+ 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: ...
615
+
616
+ class StateUpdate(_message.Message):
617
+ __slots__ = ("values", "as_node", "task_id")
618
+ VALUES_FIELD_NUMBER: _ClassVar[int]
619
+ AS_NODE_FIELD_NUMBER: _ClassVar[int]
620
+ TASK_ID_FIELD_NUMBER: _ClassVar[int]
621
+ values: SerializedValue
622
+ as_node: str
623
+ task_id: str
624
+ def __init__(self, values: _Optional[_Union[SerializedValue, _Mapping]] = ..., as_node: _Optional[str] = ..., task_id: _Optional[str] = ...) -> None: ...
625
+
626
+ class SuperstepUpdates(_message.Message):
627
+ __slots__ = ("updates",)
628
+ UPDATES_FIELD_NUMBER: _ClassVar[int]
629
+ updates: _containers.RepeatedCompositeFieldContainer[StateUpdate]
630
+ def __init__(self, updates: _Optional[_Iterable[_Union[StateUpdate, _Mapping]]] = ...) -> None: ...
631
+
632
+ class StringOrSlice(_message.Message):
633
+ __slots__ = ("values", "is_string")
634
+ VALUES_FIELD_NUMBER: _ClassVar[int]
635
+ IS_STRING_FIELD_NUMBER: _ClassVar[int]
636
+ values: _containers.RepeatedScalarFieldContainer[str]
637
+ is_string: bool
638
+ def __init__(self, values: _Optional[_Iterable[str]] = ..., is_string: bool = ...) -> None: ...
639
+
640
+ class PathSegment(_message.Message):
641
+ __slots__ = ("string_value", "int_value", "bool_value")
642
+ STRING_VALUE_FIELD_NUMBER: _ClassVar[int]
643
+ INT_VALUE_FIELD_NUMBER: _ClassVar[int]
644
+ BOOL_VALUE_FIELD_NUMBER: _ClassVar[int]
645
+ string_value: str
646
+ int_value: int
647
+ bool_value: bool
648
+ def __init__(self, string_value: _Optional[str] = ..., int_value: _Optional[int] = ..., bool_value: bool = ...) -> None: ...
@@ -0,0 +1,24 @@
1
+ # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
+ """Client and server classes corresponding to protobuf-defined services."""
3
+ import grpc
4
+ import warnings
5
+
6
+
7
+ GRPC_GENERATED_VERSION = '1.75.1'
8
+ GRPC_VERSION = grpc.__version__
9
+ _version_not_supported = False
10
+
11
+ try:
12
+ from grpc._utilities import first_version_is_lower
13
+ _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
14
+ except ImportError:
15
+ _version_not_supported = True
16
+
17
+ if _version_not_supported:
18
+ raise RuntimeError(
19
+ f'The grpc package installed is at version {GRPC_VERSION},'
20
+ + f' but the generated code in engine_common_pb2_grpc.py depends on'
21
+ + f' grpcio>={GRPC_GENERATED_VERSION}.'
22
+ + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
23
+ + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
24
+ )