langgraph-api 0.5.1__py3-none-any.whl → 0.5.3__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.

Potentially problematic release.


This version of langgraph-api might be problematic. Click here for more details.

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