langgraph-executor 0.0.1a0__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 -0
- langgraph_executor/common.py +395 -0
- langgraph_executor/example.py +29 -0
- langgraph_executor/execute_task.py +239 -0
- langgraph_executor/executor.py +341 -0
- langgraph_executor/extract_graph.py +178 -0
- langgraph_executor/info_logger.py +111 -0
- langgraph_executor/pb/__init__.py +0 -0
- langgraph_executor/pb/executor_pb2.py +79 -0
- langgraph_executor/pb/executor_pb2.pyi +415 -0
- langgraph_executor/pb/executor_pb2_grpc.py +321 -0
- langgraph_executor/pb/executor_pb2_grpc.pyi +150 -0
- langgraph_executor/pb/graph_pb2.py +55 -0
- langgraph_executor/pb/graph_pb2.pyi +230 -0
- langgraph_executor/pb/graph_pb2_grpc.py +24 -0
- langgraph_executor/pb/graph_pb2_grpc.pyi +17 -0
- langgraph_executor/pb/runtime_pb2.py +68 -0
- langgraph_executor/pb/runtime_pb2.pyi +364 -0
- langgraph_executor/pb/runtime_pb2_grpc.py +322 -0
- langgraph_executor/pb/runtime_pb2_grpc.pyi +151 -0
- langgraph_executor/pb/types_pb2.py +144 -0
- langgraph_executor/pb/types_pb2.pyi +1044 -0
- langgraph_executor/pb/types_pb2_grpc.py +24 -0
- langgraph_executor/pb/types_pb2_grpc.pyi +17 -0
- langgraph_executor/py.typed +0 -0
- langgraph_executor/server.py +186 -0
- langgraph_executor/setup.sh +29 -0
- langgraph_executor/stream_utils.py +96 -0
- langgraph_executor-0.0.1a0.dist-info/METADATA +14 -0
- langgraph_executor-0.0.1a0.dist-info/RECORD +31 -0
- langgraph_executor-0.0.1a0.dist-info/WHEEL +4 -0
@@ -0,0 +1,79 @@
|
|
1
|
+
# -*- coding: utf-8 -*-
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
3
|
+
# NO CHECKED-IN PROTOBUF GENCODE
|
4
|
+
# source: executor.proto
|
5
|
+
# Protobuf Python Version: 6.31.1
|
6
|
+
"""Generated protocol buffer code."""
|
7
|
+
from google.protobuf import descriptor as _descriptor
|
8
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
9
|
+
from google.protobuf import runtime_version as _runtime_version
|
10
|
+
from google.protobuf import symbol_database as _symbol_database
|
11
|
+
from google.protobuf.internal import builder as _builder
|
12
|
+
_runtime_version.ValidateProtobufRuntimeVersion(
|
13
|
+
_runtime_version.Domain.PUBLIC,
|
14
|
+
6,
|
15
|
+
31,
|
16
|
+
1,
|
17
|
+
'',
|
18
|
+
'executor.proto'
|
19
|
+
)
|
20
|
+
# @@protoc_insertion_point(imports)
|
21
|
+
|
22
|
+
_sym_db = _symbol_database.Default()
|
23
|
+
|
24
|
+
|
25
|
+
from . import types_pb2 as types__pb2
|
26
|
+
from . import graph_pb2 as graph__pb2
|
27
|
+
|
28
|
+
|
29
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0e\x65xecutor.proto\x12\x08\x65xecutor\x1a\x0btypes.proto\x1a\x0bgraph.proto\"\x13\n\x11ListGraphsRequest\")\n\x12ListGraphsResponse\x12\x13\n\x0bgraph_names\x18\x01 \x03(\t\"%\n\x0fGetGraphRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\"D\n\x10GetGraphResponse\x12\x30\n\x10graph_definition\x18\x01 \x01(\x0b\x32\x16.graph.GraphDefinition\"\x8a\x02\n\x1d\x43hannelsFromCheckpointRequest\x12\x12\n\ngraph_name\x18\x01 \x01(\t\x12\x1e\n\x05specs\x18\x02 \x01(\x0b\x32\x0f.types.Channels\x12g\n\x19\x63heckpoint_channel_values\x18\x03 \x03(\x0b\x32\x44.executor.ChannelsFromCheckpointRequest.CheckpointChannelValuesEntry\x1aL\n\x1c\x43heckpointChannelValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.types.Value:\x02\x38\x01\"C\n\x1e\x43hannelsFromCheckpointResponse\x12!\n\x08\x63hannels\x18\x01 \x01(\x0b\x32\x0f.types.Channels\"\x88\x01\n\x12\x45xecuteTaskRequest\x12)\n\x04init\x18\x01 \x01(\x0b\x32\x19.executor.ExecuteTaskInitH\x00\x12<\n\x14\x63\x61\x63he_check_response\x18\x02 \x01(\x0b\x32\x1c.executor.CacheCheckResponseH\x00\x42\t\n\x07message\"\xd6\x01\n\x0f\x45xecuteTaskInit\x12\x19\n\x04task\x18\x01 \x01(\x0b\x32\x0b.types.Task\x12!\n\x08\x63hannels\x18\x02 \x01(\x0b\x32\x0f.types.Channels\x12\x14\n\x0cstream_modes\x18\x03 \x03(\t\x12\x0c\n\x04step\x18\x04 \x01(\x05\x12\x15\n\rcheckpoint_ns\x18\x05 \x03(\t\x12\x13\n\x0boutput_keys\x18\x06 \x03(\t\x12\x13\n\x0bstream_keys\x18\x07 \x03(\t\x12\x12\n\ngraph_name\x18\x08 \x01(\t\x12\x0c\n\x04stop\x18\t \x01(\x05\"\xbd\x01\n\x13\x45xecuteTaskResponse\x12+\n\x0btask_result\x18\x01 \x01(\x0b\x32\x14.executor.TaskResultH\x00\x12\x32\n\x18message_or_message_chunk\x18\x02 \x01(\x0b\x32\x0e.types.MessageH\x00\x12:\n\x13\x63\x61\x63he_check_request\x18\x03 \x01(\x0b\x32\x1b.executor.CacheCheckRequestH\x00\x42\t\n\x07message\"^\n\nTaskResult\x12(\n\x05\x65rror\x18\x01 \x01(\x0b\x32\x14.types.ExecutorErrorH\x00\x88\x01\x01\x12\x1c\n\x06writes\x18\x02 \x03(\x0b\x32\x0c.types.WriteB\x08\n\x06_error\"\xa4\x01\n\x12\x41pplyWritesRequest\x12%\n\ncheckpoint\x18\x01 \x01(\x0b\x32\x11.types.Checkpoint\x12!\n\x08\x63hannels\x18\x02 \x01(\x0b\x32\x0f.types.Channels\x12\x1a\n\x05tasks\x18\x03 \x03(\x0b\x32\x0b.types.Task\x12\x14\n\x0cnext_version\x18\x04 \x01(\t\x12\x12\n\ngraph_name\x18\x05 \x01(\t\"b\n\x13\x41pplyWritesResponse\x12!\n\x07updates\x18\x01 \x01(\x0b\x32\x0e.types.UpdatesH\x00\x12\x1d\n\x05\x65rror\x18\x02 \x01(\x0b\x32\x0c.types.ValueH\x00\x42\t\n\x07message\"\xc8\x01\n\x17GenerateCacheKeyRequest\x12\x11\n\tnode_name\x18\x01 \x01(\t\x12\x16\n\x0einput_channels\x18\x02 \x03(\t\x12L\n\x0e\x63hannel_values\x18\x03 \x03(\x0b\x32\x34.executor.GenerateCacheKeyRequest.ChannelValuesEntry\x1a\x34\n\x12\x43hannelValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x0c:\x02\x38\x01\"-\n\x18GenerateCacheKeyResponse\x12\x11\n\tcache_key\x18\x01 \x01(\t\"Y\n\x11\x43\x61\x63heCheckRequest\x12\x17\n\x0f\x63\x61\x63he_namespace\x18\x01 \x03(\t\x12\x11\n\tcache_key\x18\x02 \x01(\t\x12\x10\n\x03ttl\x18\x03 \x01(\x05H\x00\x88\x01\x01\x42\x06\n\x04_ttl\"\'\n\x12\x43\x61\x63heCheckResponse\x12\x11\n\tcache_hit\x18\x01 \x01(\x08\x32\x83\x04\n\x11LangGraphExecutor\x12G\n\nListGraphs\x12\x1b.executor.ListGraphsRequest\x1a\x1c.executor.ListGraphsResponse\x12\x41\n\x08GetGraph\x12\x19.executor.GetGraphRequest\x1a\x1a.executor.GetGraphResponse\x12k\n\x16\x43hannelsFromCheckpoint\x12\'.executor.ChannelsFromCheckpointRequest\x1a(.executor.ChannelsFromCheckpointResponse\x12N\n\x0b\x45xecuteTask\x12\x1c.executor.ExecuteTaskRequest\x1a\x1d.executor.ExecuteTaskResponse(\x01\x30\x01\x12J\n\x0b\x41pplyWrites\x12\x1c.executor.ApplyWritesRequest\x1a\x1d.executor.ApplyWritesResponse\x12Y\n\x10GenerateCacheKey\x12!.executor.GenerateCacheKeyRequest\x1a\".executor.GenerateCacheKeyResponseB1Z/github.com/langchain-ai/langgraph-go/runtime/pbb\x06proto3')
|
30
|
+
|
31
|
+
_globals = globals()
|
32
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
33
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'executor_pb2', _globals)
|
34
|
+
if not _descriptor._USE_C_DESCRIPTORS:
|
35
|
+
_globals['DESCRIPTOR']._loaded_options = None
|
36
|
+
_globals['DESCRIPTOR']._serialized_options = b'Z/github.com/langchain-ai/langgraph-go/runtime/pb'
|
37
|
+
_globals['_CHANNELSFROMCHECKPOINTREQUEST_CHECKPOINTCHANNELVALUESENTRY']._loaded_options = None
|
38
|
+
_globals['_CHANNELSFROMCHECKPOINTREQUEST_CHECKPOINTCHANNELVALUESENTRY']._serialized_options = b'8\001'
|
39
|
+
_globals['_GENERATECACHEKEYREQUEST_CHANNELVALUESENTRY']._loaded_options = None
|
40
|
+
_globals['_GENERATECACHEKEYREQUEST_CHANNELVALUESENTRY']._serialized_options = b'8\001'
|
41
|
+
_globals['_LISTGRAPHSREQUEST']._serialized_start=54
|
42
|
+
_globals['_LISTGRAPHSREQUEST']._serialized_end=73
|
43
|
+
_globals['_LISTGRAPHSRESPONSE']._serialized_start=75
|
44
|
+
_globals['_LISTGRAPHSRESPONSE']._serialized_end=116
|
45
|
+
_globals['_GETGRAPHREQUEST']._serialized_start=118
|
46
|
+
_globals['_GETGRAPHREQUEST']._serialized_end=155
|
47
|
+
_globals['_GETGRAPHRESPONSE']._serialized_start=157
|
48
|
+
_globals['_GETGRAPHRESPONSE']._serialized_end=225
|
49
|
+
_globals['_CHANNELSFROMCHECKPOINTREQUEST']._serialized_start=228
|
50
|
+
_globals['_CHANNELSFROMCHECKPOINTREQUEST']._serialized_end=494
|
51
|
+
_globals['_CHANNELSFROMCHECKPOINTREQUEST_CHECKPOINTCHANNELVALUESENTRY']._serialized_start=418
|
52
|
+
_globals['_CHANNELSFROMCHECKPOINTREQUEST_CHECKPOINTCHANNELVALUESENTRY']._serialized_end=494
|
53
|
+
_globals['_CHANNELSFROMCHECKPOINTRESPONSE']._serialized_start=496
|
54
|
+
_globals['_CHANNELSFROMCHECKPOINTRESPONSE']._serialized_end=563
|
55
|
+
_globals['_EXECUTETASKREQUEST']._serialized_start=566
|
56
|
+
_globals['_EXECUTETASKREQUEST']._serialized_end=702
|
57
|
+
_globals['_EXECUTETASKINIT']._serialized_start=705
|
58
|
+
_globals['_EXECUTETASKINIT']._serialized_end=919
|
59
|
+
_globals['_EXECUTETASKRESPONSE']._serialized_start=922
|
60
|
+
_globals['_EXECUTETASKRESPONSE']._serialized_end=1111
|
61
|
+
_globals['_TASKRESULT']._serialized_start=1113
|
62
|
+
_globals['_TASKRESULT']._serialized_end=1207
|
63
|
+
_globals['_APPLYWRITESREQUEST']._serialized_start=1210
|
64
|
+
_globals['_APPLYWRITESREQUEST']._serialized_end=1374
|
65
|
+
_globals['_APPLYWRITESRESPONSE']._serialized_start=1376
|
66
|
+
_globals['_APPLYWRITESRESPONSE']._serialized_end=1474
|
67
|
+
_globals['_GENERATECACHEKEYREQUEST']._serialized_start=1477
|
68
|
+
_globals['_GENERATECACHEKEYREQUEST']._serialized_end=1677
|
69
|
+
_globals['_GENERATECACHEKEYREQUEST_CHANNELVALUESENTRY']._serialized_start=1625
|
70
|
+
_globals['_GENERATECACHEKEYREQUEST_CHANNELVALUESENTRY']._serialized_end=1677
|
71
|
+
_globals['_GENERATECACHEKEYRESPONSE']._serialized_start=1679
|
72
|
+
_globals['_GENERATECACHEKEYRESPONSE']._serialized_end=1724
|
73
|
+
_globals['_CACHECHECKREQUEST']._serialized_start=1726
|
74
|
+
_globals['_CACHECHECKREQUEST']._serialized_end=1815
|
75
|
+
_globals['_CACHECHECKRESPONSE']._serialized_start=1817
|
76
|
+
_globals['_CACHECHECKRESPONSE']._serialized_end=1856
|
77
|
+
_globals['_LANGGRAPHEXECUTOR']._serialized_start=1859
|
78
|
+
_globals['_LANGGRAPHEXECUTOR']._serialized_end=2374
|
79
|
+
# @@protoc_insertion_point(module_scope)
|
@@ -0,0 +1,415 @@
|
|
1
|
+
"""
|
2
|
+
@generated by mypy-protobuf. Do not edit manually!
|
3
|
+
isort:skip_file
|
4
|
+
"""
|
5
|
+
|
6
|
+
import builtins
|
7
|
+
import collections.abc
|
8
|
+
import google.protobuf.descriptor
|
9
|
+
import google.protobuf.internal.containers
|
10
|
+
import google.protobuf.message
|
11
|
+
import graph_pb2
|
12
|
+
import types_pb2
|
13
|
+
import typing
|
14
|
+
|
15
|
+
DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
|
16
|
+
|
17
|
+
@typing.final
|
18
|
+
class ListGraphsRequest(google.protobuf.message.Message):
|
19
|
+
"""ListGraphs"""
|
20
|
+
|
21
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
22
|
+
|
23
|
+
def __init__(
|
24
|
+
self,
|
25
|
+
) -> None: ...
|
26
|
+
|
27
|
+
global___ListGraphsRequest = ListGraphsRequest
|
28
|
+
|
29
|
+
@typing.final
|
30
|
+
class ListGraphsResponse(google.protobuf.message.Message):
|
31
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
32
|
+
|
33
|
+
GRAPH_NAMES_FIELD_NUMBER: builtins.int
|
34
|
+
@property
|
35
|
+
def graph_names(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
|
36
|
+
def __init__(
|
37
|
+
self,
|
38
|
+
*,
|
39
|
+
graph_names: collections.abc.Iterable[builtins.str] | None = ...,
|
40
|
+
) -> None: ...
|
41
|
+
def ClearField(self, field_name: typing.Literal["graph_names", b"graph_names"]) -> None: ...
|
42
|
+
|
43
|
+
global___ListGraphsResponse = ListGraphsResponse
|
44
|
+
|
45
|
+
@typing.final
|
46
|
+
class GetGraphRequest(google.protobuf.message.Message):
|
47
|
+
"""GetGraph"""
|
48
|
+
|
49
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
50
|
+
|
51
|
+
GRAPH_NAME_FIELD_NUMBER: builtins.int
|
52
|
+
graph_name: builtins.str
|
53
|
+
def __init__(
|
54
|
+
self,
|
55
|
+
*,
|
56
|
+
graph_name: builtins.str = ...,
|
57
|
+
) -> None: ...
|
58
|
+
def ClearField(self, field_name: typing.Literal["graph_name", b"graph_name"]) -> None: ...
|
59
|
+
|
60
|
+
global___GetGraphRequest = GetGraphRequest
|
61
|
+
|
62
|
+
@typing.final
|
63
|
+
class GetGraphResponse(google.protobuf.message.Message):
|
64
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
65
|
+
|
66
|
+
GRAPH_DEFINITION_FIELD_NUMBER: builtins.int
|
67
|
+
@property
|
68
|
+
def graph_definition(self) -> graph_pb2.GraphDefinition: ...
|
69
|
+
def __init__(
|
70
|
+
self,
|
71
|
+
*,
|
72
|
+
graph_definition: graph_pb2.GraphDefinition | None = ...,
|
73
|
+
) -> None: ...
|
74
|
+
def HasField(self, field_name: typing.Literal["graph_definition", b"graph_definition"]) -> builtins.bool: ...
|
75
|
+
def ClearField(self, field_name: typing.Literal["graph_definition", b"graph_definition"]) -> None: ...
|
76
|
+
|
77
|
+
global___GetGraphResponse = GetGraphResponse
|
78
|
+
|
79
|
+
@typing.final
|
80
|
+
class ChannelsFromCheckpointRequest(google.protobuf.message.Message):
|
81
|
+
"""GetChannelsFromCheckpoint"""
|
82
|
+
|
83
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
84
|
+
|
85
|
+
@typing.final
|
86
|
+
class CheckpointChannelValuesEntry(google.protobuf.message.Message):
|
87
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
88
|
+
|
89
|
+
KEY_FIELD_NUMBER: builtins.int
|
90
|
+
VALUE_FIELD_NUMBER: builtins.int
|
91
|
+
key: builtins.str
|
92
|
+
@property
|
93
|
+
def value(self) -> types_pb2.Value: ...
|
94
|
+
def __init__(
|
95
|
+
self,
|
96
|
+
*,
|
97
|
+
key: builtins.str = ...,
|
98
|
+
value: types_pb2.Value | None = ...,
|
99
|
+
) -> None: ...
|
100
|
+
def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ...
|
101
|
+
def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
|
102
|
+
|
103
|
+
GRAPH_NAME_FIELD_NUMBER: builtins.int
|
104
|
+
SPECS_FIELD_NUMBER: builtins.int
|
105
|
+
CHECKPOINT_CHANNEL_VALUES_FIELD_NUMBER: builtins.int
|
106
|
+
graph_name: builtins.str
|
107
|
+
@property
|
108
|
+
def specs(self) -> types_pb2.Channels: ...
|
109
|
+
@property
|
110
|
+
def checkpoint_channel_values(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, types_pb2.Value]: ...
|
111
|
+
def __init__(
|
112
|
+
self,
|
113
|
+
*,
|
114
|
+
graph_name: builtins.str = ...,
|
115
|
+
specs: types_pb2.Channels | None = ...,
|
116
|
+
checkpoint_channel_values: collections.abc.Mapping[builtins.str, types_pb2.Value] | None = ...,
|
117
|
+
) -> None: ...
|
118
|
+
def HasField(self, field_name: typing.Literal["specs", b"specs"]) -> builtins.bool: ...
|
119
|
+
def ClearField(self, field_name: typing.Literal["checkpoint_channel_values", b"checkpoint_channel_values", "graph_name", b"graph_name", "specs", b"specs"]) -> None: ...
|
120
|
+
|
121
|
+
global___ChannelsFromCheckpointRequest = ChannelsFromCheckpointRequest
|
122
|
+
|
123
|
+
@typing.final
|
124
|
+
class ChannelsFromCheckpointResponse(google.protobuf.message.Message):
|
125
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
126
|
+
|
127
|
+
CHANNELS_FIELD_NUMBER: builtins.int
|
128
|
+
@property
|
129
|
+
def channels(self) -> types_pb2.Channels: ...
|
130
|
+
def __init__(
|
131
|
+
self,
|
132
|
+
*,
|
133
|
+
channels: types_pb2.Channels | None = ...,
|
134
|
+
) -> None: ...
|
135
|
+
def HasField(self, field_name: typing.Literal["channels", b"channels"]) -> builtins.bool: ...
|
136
|
+
def ClearField(self, field_name: typing.Literal["channels", b"channels"]) -> None: ...
|
137
|
+
|
138
|
+
global___ChannelsFromCheckpointResponse = ChannelsFromCheckpointResponse
|
139
|
+
|
140
|
+
@typing.final
|
141
|
+
class ExecuteTaskRequest(google.protobuf.message.Message):
|
142
|
+
"""Execute task"""
|
143
|
+
|
144
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
145
|
+
|
146
|
+
INIT_FIELD_NUMBER: builtins.int
|
147
|
+
CACHE_CHECK_RESPONSE_FIELD_NUMBER: builtins.int
|
148
|
+
@property
|
149
|
+
def init(self) -> global___ExecuteTaskInit: ...
|
150
|
+
@property
|
151
|
+
def cache_check_response(self) -> global___CacheCheckResponse: ...
|
152
|
+
def __init__(
|
153
|
+
self,
|
154
|
+
*,
|
155
|
+
init: global___ExecuteTaskInit | None = ...,
|
156
|
+
cache_check_response: global___CacheCheckResponse | None = ...,
|
157
|
+
) -> None: ...
|
158
|
+
def HasField(self, field_name: typing.Literal["cache_check_response", b"cache_check_response", "init", b"init", "message", b"message"]) -> builtins.bool: ...
|
159
|
+
def ClearField(self, field_name: typing.Literal["cache_check_response", b"cache_check_response", "init", b"init", "message", b"message"]) -> None: ...
|
160
|
+
def WhichOneof(self, oneof_group: typing.Literal["message", b"message"]) -> typing.Literal["init", "cache_check_response"] | None: ...
|
161
|
+
|
162
|
+
global___ExecuteTaskRequest = ExecuteTaskRequest
|
163
|
+
|
164
|
+
@typing.final
|
165
|
+
class ExecuteTaskInit(google.protobuf.message.Message):
|
166
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
167
|
+
|
168
|
+
TASK_FIELD_NUMBER: builtins.int
|
169
|
+
CHANNELS_FIELD_NUMBER: builtins.int
|
170
|
+
STREAM_MODES_FIELD_NUMBER: builtins.int
|
171
|
+
STEP_FIELD_NUMBER: builtins.int
|
172
|
+
CHECKPOINT_NS_FIELD_NUMBER: builtins.int
|
173
|
+
OUTPUT_KEYS_FIELD_NUMBER: builtins.int
|
174
|
+
STREAM_KEYS_FIELD_NUMBER: builtins.int
|
175
|
+
GRAPH_NAME_FIELD_NUMBER: builtins.int
|
176
|
+
STOP_FIELD_NUMBER: builtins.int
|
177
|
+
step: builtins.int
|
178
|
+
graph_name: builtins.str
|
179
|
+
stop: builtins.int
|
180
|
+
@property
|
181
|
+
def task(self) -> types_pb2.Task: ...
|
182
|
+
@property
|
183
|
+
def channels(self) -> types_pb2.Channels: ...
|
184
|
+
@property
|
185
|
+
def stream_modes(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
|
186
|
+
@property
|
187
|
+
def checkpoint_ns(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
|
188
|
+
@property
|
189
|
+
def output_keys(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
|
190
|
+
@property
|
191
|
+
def stream_keys(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
|
192
|
+
def __init__(
|
193
|
+
self,
|
194
|
+
*,
|
195
|
+
task: types_pb2.Task | None = ...,
|
196
|
+
channels: types_pb2.Channels | None = ...,
|
197
|
+
stream_modes: collections.abc.Iterable[builtins.str] | None = ...,
|
198
|
+
step: builtins.int = ...,
|
199
|
+
checkpoint_ns: collections.abc.Iterable[builtins.str] | None = ...,
|
200
|
+
output_keys: collections.abc.Iterable[builtins.str] | None = ...,
|
201
|
+
stream_keys: collections.abc.Iterable[builtins.str] | None = ...,
|
202
|
+
graph_name: builtins.str = ...,
|
203
|
+
stop: builtins.int = ...,
|
204
|
+
) -> None: ...
|
205
|
+
def HasField(self, field_name: typing.Literal["channels", b"channels", "task", b"task"]) -> builtins.bool: ...
|
206
|
+
def ClearField(self, field_name: typing.Literal["channels", b"channels", "checkpoint_ns", b"checkpoint_ns", "graph_name", b"graph_name", "output_keys", b"output_keys", "step", b"step", "stop", b"stop", "stream_keys", b"stream_keys", "stream_modes", b"stream_modes", "task", b"task"]) -> None: ...
|
207
|
+
|
208
|
+
global___ExecuteTaskInit = ExecuteTaskInit
|
209
|
+
|
210
|
+
@typing.final
|
211
|
+
class ExecuteTaskResponse(google.protobuf.message.Message):
|
212
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
213
|
+
|
214
|
+
TASK_RESULT_FIELD_NUMBER: builtins.int
|
215
|
+
MESSAGE_OR_MESSAGE_CHUNK_FIELD_NUMBER: builtins.int
|
216
|
+
CACHE_CHECK_REQUEST_FIELD_NUMBER: builtins.int
|
217
|
+
@property
|
218
|
+
def task_result(self) -> global___TaskResult: ...
|
219
|
+
@property
|
220
|
+
def message_or_message_chunk(self) -> types_pb2.Message: ...
|
221
|
+
@property
|
222
|
+
def cache_check_request(self) -> global___CacheCheckRequest: ...
|
223
|
+
def __init__(
|
224
|
+
self,
|
225
|
+
*,
|
226
|
+
task_result: global___TaskResult | None = ...,
|
227
|
+
message_or_message_chunk: types_pb2.Message | None = ...,
|
228
|
+
cache_check_request: global___CacheCheckRequest | None = ...,
|
229
|
+
) -> None: ...
|
230
|
+
def HasField(self, field_name: typing.Literal["cache_check_request", b"cache_check_request", "message", b"message", "message_or_message_chunk", b"message_or_message_chunk", "task_result", b"task_result"]) -> builtins.bool: ...
|
231
|
+
def ClearField(self, field_name: typing.Literal["cache_check_request", b"cache_check_request", "message", b"message", "message_or_message_chunk", b"message_or_message_chunk", "task_result", b"task_result"]) -> None: ...
|
232
|
+
def WhichOneof(self, oneof_group: typing.Literal["message", b"message"]) -> typing.Literal["task_result", "message_or_message_chunk", "cache_check_request"] | None: ...
|
233
|
+
|
234
|
+
global___ExecuteTaskResponse = ExecuteTaskResponse
|
235
|
+
|
236
|
+
@typing.final
|
237
|
+
class TaskResult(google.protobuf.message.Message):
|
238
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
239
|
+
|
240
|
+
ERROR_FIELD_NUMBER: builtins.int
|
241
|
+
WRITES_FIELD_NUMBER: builtins.int
|
242
|
+
@property
|
243
|
+
def error(self) -> types_pb2.ExecutorError: ...
|
244
|
+
@property
|
245
|
+
def writes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[types_pb2.Write]: ...
|
246
|
+
def __init__(
|
247
|
+
self,
|
248
|
+
*,
|
249
|
+
error: types_pb2.ExecutorError | None = ...,
|
250
|
+
writes: collections.abc.Iterable[types_pb2.Write] | None = ...,
|
251
|
+
) -> None: ...
|
252
|
+
def HasField(self, field_name: typing.Literal["_error", b"_error", "error", b"error"]) -> builtins.bool: ...
|
253
|
+
def ClearField(self, field_name: typing.Literal["_error", b"_error", "error", b"error", "writes", b"writes"]) -> None: ...
|
254
|
+
def WhichOneof(self, oneof_group: typing.Literal["_error", b"_error"]) -> typing.Literal["error"] | None: ...
|
255
|
+
|
256
|
+
global___TaskResult = TaskResult
|
257
|
+
|
258
|
+
@typing.final
|
259
|
+
class ApplyWritesRequest(google.protobuf.message.Message):
|
260
|
+
"""Apply Writes"""
|
261
|
+
|
262
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
263
|
+
|
264
|
+
CHECKPOINT_FIELD_NUMBER: builtins.int
|
265
|
+
CHANNELS_FIELD_NUMBER: builtins.int
|
266
|
+
TASKS_FIELD_NUMBER: builtins.int
|
267
|
+
NEXT_VERSION_FIELD_NUMBER: builtins.int
|
268
|
+
GRAPH_NAME_FIELD_NUMBER: builtins.int
|
269
|
+
next_version: builtins.str
|
270
|
+
graph_name: builtins.str
|
271
|
+
@property
|
272
|
+
def checkpoint(self) -> types_pb2.Checkpoint: ...
|
273
|
+
@property
|
274
|
+
def channels(self) -> types_pb2.Channels: ...
|
275
|
+
@property
|
276
|
+
def tasks(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[types_pb2.Task]: ...
|
277
|
+
def __init__(
|
278
|
+
self,
|
279
|
+
*,
|
280
|
+
checkpoint: types_pb2.Checkpoint | None = ...,
|
281
|
+
channels: types_pb2.Channels | None = ...,
|
282
|
+
tasks: collections.abc.Iterable[types_pb2.Task] | None = ...,
|
283
|
+
next_version: builtins.str = ...,
|
284
|
+
graph_name: builtins.str = ...,
|
285
|
+
) -> None: ...
|
286
|
+
def HasField(self, field_name: typing.Literal["channels", b"channels", "checkpoint", b"checkpoint"]) -> builtins.bool: ...
|
287
|
+
def ClearField(self, field_name: typing.Literal["channels", b"channels", "checkpoint", b"checkpoint", "graph_name", b"graph_name", "next_version", b"next_version", "tasks", b"tasks"]) -> None: ...
|
288
|
+
|
289
|
+
global___ApplyWritesRequest = ApplyWritesRequest
|
290
|
+
|
291
|
+
@typing.final
|
292
|
+
class ApplyWritesResponse(google.protobuf.message.Message):
|
293
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
294
|
+
|
295
|
+
UPDATES_FIELD_NUMBER: builtins.int
|
296
|
+
ERROR_FIELD_NUMBER: builtins.int
|
297
|
+
@property
|
298
|
+
def updates(self) -> types_pb2.Updates: ...
|
299
|
+
@property
|
300
|
+
def error(self) -> types_pb2.Value: ...
|
301
|
+
def __init__(
|
302
|
+
self,
|
303
|
+
*,
|
304
|
+
updates: types_pb2.Updates | None = ...,
|
305
|
+
error: types_pb2.Value | None = ...,
|
306
|
+
) -> None: ...
|
307
|
+
def HasField(self, field_name: typing.Literal["error", b"error", "message", b"message", "updates", b"updates"]) -> builtins.bool: ...
|
308
|
+
def ClearField(self, field_name: typing.Literal["error", b"error", "message", b"message", "updates", b"updates"]) -> None: ...
|
309
|
+
def WhichOneof(self, oneof_group: typing.Literal["message", b"message"]) -> typing.Literal["updates", "error"] | None: ...
|
310
|
+
|
311
|
+
global___ApplyWritesResponse = ApplyWritesResponse
|
312
|
+
|
313
|
+
@typing.final
|
314
|
+
class GenerateCacheKeyRequest(google.protobuf.message.Message):
|
315
|
+
"""Cache key generation messages"""
|
316
|
+
|
317
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
318
|
+
|
319
|
+
@typing.final
|
320
|
+
class ChannelValuesEntry(google.protobuf.message.Message):
|
321
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
322
|
+
|
323
|
+
KEY_FIELD_NUMBER: builtins.int
|
324
|
+
VALUE_FIELD_NUMBER: builtins.int
|
325
|
+
key: builtins.str
|
326
|
+
value: builtins.bytes
|
327
|
+
def __init__(
|
328
|
+
self,
|
329
|
+
*,
|
330
|
+
key: builtins.str = ...,
|
331
|
+
value: builtins.bytes = ...,
|
332
|
+
) -> None: ...
|
333
|
+
def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
|
334
|
+
|
335
|
+
NODE_NAME_FIELD_NUMBER: builtins.int
|
336
|
+
INPUT_CHANNELS_FIELD_NUMBER: builtins.int
|
337
|
+
CHANNEL_VALUES_FIELD_NUMBER: builtins.int
|
338
|
+
node_name: builtins.str
|
339
|
+
@property
|
340
|
+
def input_channels(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
|
341
|
+
@property
|
342
|
+
def channel_values(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.bytes]:
|
343
|
+
"""Current values of input channels"""
|
344
|
+
|
345
|
+
def __init__(
|
346
|
+
self,
|
347
|
+
*,
|
348
|
+
node_name: builtins.str = ...,
|
349
|
+
input_channels: collections.abc.Iterable[builtins.str] | None = ...,
|
350
|
+
channel_values: collections.abc.Mapping[builtins.str, builtins.bytes] | None = ...,
|
351
|
+
) -> None: ...
|
352
|
+
def ClearField(self, field_name: typing.Literal["channel_values", b"channel_values", "input_channels", b"input_channels", "node_name", b"node_name"]) -> None: ...
|
353
|
+
|
354
|
+
global___GenerateCacheKeyRequest = GenerateCacheKeyRequest
|
355
|
+
|
356
|
+
@typing.final
|
357
|
+
class GenerateCacheKeyResponse(google.protobuf.message.Message):
|
358
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
359
|
+
|
360
|
+
CACHE_KEY_FIELD_NUMBER: builtins.int
|
361
|
+
cache_key: builtins.str
|
362
|
+
def __init__(
|
363
|
+
self,
|
364
|
+
*,
|
365
|
+
cache_key: builtins.str = ...,
|
366
|
+
) -> None: ...
|
367
|
+
def ClearField(self, field_name: typing.Literal["cache_key", b"cache_key"]) -> None: ...
|
368
|
+
|
369
|
+
global___GenerateCacheKeyResponse = GenerateCacheKeyResponse
|
370
|
+
|
371
|
+
@typing.final
|
372
|
+
class CacheCheckRequest(google.protobuf.message.Message):
|
373
|
+
"""Cache check messages for cache hit/miss communication"""
|
374
|
+
|
375
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
376
|
+
|
377
|
+
CACHE_NAMESPACE_FIELD_NUMBER: builtins.int
|
378
|
+
CACHE_KEY_FIELD_NUMBER: builtins.int
|
379
|
+
TTL_FIELD_NUMBER: builtins.int
|
380
|
+
cache_key: builtins.str
|
381
|
+
"""Cache key"""
|
382
|
+
ttl: builtins.int
|
383
|
+
"""TTL in seconds, optional"""
|
384
|
+
@property
|
385
|
+
def cache_namespace(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
|
386
|
+
"""Serialized namespace components"""
|
387
|
+
|
388
|
+
def __init__(
|
389
|
+
self,
|
390
|
+
*,
|
391
|
+
cache_namespace: collections.abc.Iterable[builtins.str] | None = ...,
|
392
|
+
cache_key: builtins.str = ...,
|
393
|
+
ttl: builtins.int | None = ...,
|
394
|
+
) -> None: ...
|
395
|
+
def HasField(self, field_name: typing.Literal["_ttl", b"_ttl", "ttl", b"ttl"]) -> builtins.bool: ...
|
396
|
+
def ClearField(self, field_name: typing.Literal["_ttl", b"_ttl", "cache_key", b"cache_key", "cache_namespace", b"cache_namespace", "ttl", b"ttl"]) -> None: ...
|
397
|
+
def WhichOneof(self, oneof_group: typing.Literal["_ttl", b"_ttl"]) -> typing.Literal["ttl"] | None: ...
|
398
|
+
|
399
|
+
global___CacheCheckRequest = CacheCheckRequest
|
400
|
+
|
401
|
+
@typing.final
|
402
|
+
class CacheCheckResponse(google.protobuf.message.Message):
|
403
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
404
|
+
|
405
|
+
CACHE_HIT_FIELD_NUMBER: builtins.int
|
406
|
+
cache_hit: builtins.bool
|
407
|
+
"""Whether cache hit occurred"""
|
408
|
+
def __init__(
|
409
|
+
self,
|
410
|
+
*,
|
411
|
+
cache_hit: builtins.bool = ...,
|
412
|
+
) -> None: ...
|
413
|
+
def ClearField(self, field_name: typing.Literal["cache_hit", b"cache_hit"]) -> None: ...
|
414
|
+
|
415
|
+
global___CacheCheckResponse = CacheCheckResponse
|