flyte 0.2.0b35__py3-none-any.whl → 0.2.0b37__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 flyte might be problematic. Click here for more details.
- flyte/_image.py +1 -1
- flyte/_internal/controllers/_local_controller.py +3 -2
- flyte/_internal/controllers/_trace.py +14 -10
- flyte/_internal/controllers/remote/_action.py +37 -7
- flyte/_internal/controllers/remote/_controller.py +43 -21
- flyte/_internal/controllers/remote/_core.py +32 -16
- flyte/_internal/controllers/remote/_informer.py +18 -7
- flyte/_internal/runtime/task_serde.py +17 -6
- flyte/_protos/common/identifier_pb2.py +23 -1
- flyte/_protos/common/identifier_pb2.pyi +28 -0
- flyte/_protos/workflow/queue_service_pb2.py +33 -29
- flyte/_protos/workflow/queue_service_pb2.pyi +34 -16
- flyte/_protos/workflow/run_definition_pb2.py +64 -71
- flyte/_protos/workflow/run_definition_pb2.pyi +44 -31
- flyte/_protos/workflow/run_logs_service_pb2.py +10 -10
- flyte/_protos/workflow/run_logs_service_pb2.pyi +3 -3
- flyte/_protos/workflow/run_service_pb2.py +54 -46
- flyte/_protos/workflow/run_service_pb2.pyi +32 -18
- flyte/_protos/workflow/run_service_pb2_grpc.py +34 -0
- flyte/_protos/workflow/state_service_pb2.py +20 -19
- flyte/_protos/workflow/state_service_pb2.pyi +13 -12
- flyte/_run.py +11 -6
- flyte/_trace.py +4 -10
- flyte/_version.py +2 -2
- flyte/migrate/__init__.py +1 -0
- flyte/migrate/dynamic.py +13 -0
- flyte/migrate/task.py +99 -0
- flyte/migrate/workflow.py +13 -0
- flyte/remote/_action.py +56 -25
- flyte/remote/_logs.py +4 -3
- flyte/remote/_run.py +5 -4
- flyte-0.2.0b37.dist-info/METADATA +371 -0
- {flyte-0.2.0b35.dist-info → flyte-0.2.0b37.dist-info}/RECORD +38 -33
- flyte-0.2.0b37.dist-info/licenses/LICENSE +201 -0
- flyte-0.2.0b35.dist-info/METADATA +0 -249
- {flyte-0.2.0b35.data → flyte-0.2.0b37.data}/scripts/runtime.py +0 -0
- {flyte-0.2.0b35.dist-info → flyte-0.2.0b37.dist-info}/WHEEL +0 -0
- {flyte-0.2.0b35.dist-info → flyte-0.2.0b37.dist-info}/entry_points.txt +0 -0
- {flyte-0.2.0b35.dist-info → flyte-0.2.0b37.dist-info}/top_level.txt +0 -0
|
@@ -1,5 +1,6 @@
|
|
|
1
1
|
from flyte._protos.common import identifier_pb2 as _identifier_pb2
|
|
2
2
|
from flyte._protos.common import identity_pb2 as _identity_pb2
|
|
3
|
+
from flyteidl.core import catalog_pb2 as _catalog_pb2
|
|
3
4
|
from flyteidl.core import execution_pb2 as _execution_pb2
|
|
4
5
|
from flyteidl.core import literals_pb2 as _literals_pb2
|
|
5
6
|
from google.protobuf import timestamp_pb2 as _timestamp_pb2
|
|
@@ -35,18 +36,6 @@ PHASE_FAILED: Phase
|
|
|
35
36
|
PHASE_ABORTED: Phase
|
|
36
37
|
PHASE_TIMED_OUT: Phase
|
|
37
38
|
|
|
38
|
-
class RunIdentifier(_message.Message):
|
|
39
|
-
__slots__ = ["org", "project", "domain", "name"]
|
|
40
|
-
ORG_FIELD_NUMBER: _ClassVar[int]
|
|
41
|
-
PROJECT_FIELD_NUMBER: _ClassVar[int]
|
|
42
|
-
DOMAIN_FIELD_NUMBER: _ClassVar[int]
|
|
43
|
-
NAME_FIELD_NUMBER: _ClassVar[int]
|
|
44
|
-
org: str
|
|
45
|
-
project: str
|
|
46
|
-
domain: str
|
|
47
|
-
name: str
|
|
48
|
-
def __init__(self, org: _Optional[str] = ..., project: _Optional[str] = ..., domain: _Optional[str] = ..., name: _Optional[str] = ...) -> None: ...
|
|
49
|
-
|
|
50
39
|
class Labels(_message.Message):
|
|
51
40
|
__slots__ = ["values"]
|
|
52
41
|
class ValuesEntry(_message.Message):
|
|
@@ -107,14 +96,6 @@ class RunDetails(_message.Message):
|
|
|
107
96
|
action: ActionDetails
|
|
108
97
|
def __init__(self, run_spec: _Optional[_Union[RunSpec, _Mapping]] = ..., action: _Optional[_Union[ActionDetails, _Mapping]] = ...) -> None: ...
|
|
109
98
|
|
|
110
|
-
class ActionIdentifier(_message.Message):
|
|
111
|
-
__slots__ = ["run", "name"]
|
|
112
|
-
RUN_FIELD_NUMBER: _ClassVar[int]
|
|
113
|
-
NAME_FIELD_NUMBER: _ClassVar[int]
|
|
114
|
-
run: RunIdentifier
|
|
115
|
-
name: str
|
|
116
|
-
def __init__(self, run: _Optional[_Union[RunIdentifier, _Mapping]] = ..., name: _Optional[str] = ...) -> None: ...
|
|
117
|
-
|
|
118
99
|
class TaskActionMetadata(_message.Message):
|
|
119
100
|
__slots__ = ["id", "task_type", "short_name"]
|
|
120
101
|
ID_FIELD_NUMBER: _ClassVar[int]
|
|
@@ -175,10 +156,10 @@ class Action(_message.Message):
|
|
|
175
156
|
ID_FIELD_NUMBER: _ClassVar[int]
|
|
176
157
|
METADATA_FIELD_NUMBER: _ClassVar[int]
|
|
177
158
|
STATUS_FIELD_NUMBER: _ClassVar[int]
|
|
178
|
-
id: ActionIdentifier
|
|
159
|
+
id: _identifier_pb2.ActionIdentifier
|
|
179
160
|
metadata: ActionMetadata
|
|
180
161
|
status: ActionStatus
|
|
181
|
-
def __init__(self, id: _Optional[_Union[ActionIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[ActionMetadata, _Mapping]] = ..., status: _Optional[_Union[ActionStatus, _Mapping]] = ...) -> None: ...
|
|
162
|
+
def __init__(self, id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[ActionMetadata, _Mapping]] = ..., status: _Optional[_Union[ActionStatus, _Mapping]] = ...) -> None: ...
|
|
182
163
|
|
|
183
164
|
class EnrichedAction(_message.Message):
|
|
184
165
|
__slots__ = ["action", "meets_filter", "children_phase_counts"]
|
|
@@ -230,17 +211,17 @@ class ActionDetails(_message.Message):
|
|
|
230
211
|
ABORT_INFO_FIELD_NUMBER: _ClassVar[int]
|
|
231
212
|
RESOLVED_TASK_SPEC_FIELD_NUMBER: _ClassVar[int]
|
|
232
213
|
ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
|
|
233
|
-
id: ActionIdentifier
|
|
214
|
+
id: _identifier_pb2.ActionIdentifier
|
|
234
215
|
metadata: ActionMetadata
|
|
235
216
|
status: ActionStatus
|
|
236
217
|
error_info: ErrorInfo
|
|
237
218
|
abort_info: AbortInfo
|
|
238
219
|
resolved_task_spec: _task_definition_pb2.TaskSpec
|
|
239
220
|
attempts: _containers.RepeatedCompositeFieldContainer[ActionAttempt]
|
|
240
|
-
def __init__(self, id: _Optional[_Union[ActionIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[ActionMetadata, _Mapping]] = ..., status: _Optional[_Union[ActionStatus, _Mapping]] = ..., error_info: _Optional[_Union[ErrorInfo, _Mapping]] = ..., abort_info: _Optional[_Union[AbortInfo, _Mapping]] = ..., resolved_task_spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ..., attempts: _Optional[_Iterable[_Union[ActionAttempt, _Mapping]]] = ...) -> None: ...
|
|
221
|
+
def __init__(self, id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[ActionMetadata, _Mapping]] = ..., status: _Optional[_Union[ActionStatus, _Mapping]] = ..., error_info: _Optional[_Union[ErrorInfo, _Mapping]] = ..., abort_info: _Optional[_Union[AbortInfo, _Mapping]] = ..., resolved_task_spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ..., attempts: _Optional[_Iterable[_Union[ActionAttempt, _Mapping]]] = ...) -> None: ...
|
|
241
222
|
|
|
242
223
|
class ActionAttempt(_message.Message):
|
|
243
|
-
__slots__ = ["phase", "start_time", "end_time", "error_info", "attempt", "log_info", "outputs", "logs_available"]
|
|
224
|
+
__slots__ = ["phase", "start_time", "end_time", "error_info", "attempt", "log_info", "outputs", "logs_available", "cache_status", "cluster_events", "phase_transitions", "cluster"]
|
|
244
225
|
PHASE_FIELD_NUMBER: _ClassVar[int]
|
|
245
226
|
START_TIME_FIELD_NUMBER: _ClassVar[int]
|
|
246
227
|
END_TIME_FIELD_NUMBER: _ClassVar[int]
|
|
@@ -249,6 +230,10 @@ class ActionAttempt(_message.Message):
|
|
|
249
230
|
LOG_INFO_FIELD_NUMBER: _ClassVar[int]
|
|
250
231
|
OUTPUTS_FIELD_NUMBER: _ClassVar[int]
|
|
251
232
|
LOGS_AVAILABLE_FIELD_NUMBER: _ClassVar[int]
|
|
233
|
+
CACHE_STATUS_FIELD_NUMBER: _ClassVar[int]
|
|
234
|
+
CLUSTER_EVENTS_FIELD_NUMBER: _ClassVar[int]
|
|
235
|
+
PHASE_TRANSITIONS_FIELD_NUMBER: _ClassVar[int]
|
|
236
|
+
CLUSTER_FIELD_NUMBER: _ClassVar[int]
|
|
252
237
|
phase: Phase
|
|
253
238
|
start_time: _timestamp_pb2.Timestamp
|
|
254
239
|
end_time: _timestamp_pb2.Timestamp
|
|
@@ -257,10 +242,32 @@ class ActionAttempt(_message.Message):
|
|
|
257
242
|
log_info: _containers.RepeatedCompositeFieldContainer[_execution_pb2.TaskLog]
|
|
258
243
|
outputs: OutputReferences
|
|
259
244
|
logs_available: bool
|
|
260
|
-
|
|
245
|
+
cache_status: _catalog_pb2.CatalogCacheStatus
|
|
246
|
+
cluster_events: _containers.RepeatedCompositeFieldContainer[ClusterEvent]
|
|
247
|
+
phase_transitions: _containers.RepeatedCompositeFieldContainer[PhaseTransition]
|
|
248
|
+
cluster: str
|
|
249
|
+
def __init__(self, phase: _Optional[_Union[Phase, str]] = ..., start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., error_info: _Optional[_Union[ErrorInfo, _Mapping]] = ..., attempt: _Optional[int] = ..., log_info: _Optional[_Iterable[_Union[_execution_pb2.TaskLog, _Mapping]]] = ..., outputs: _Optional[_Union[OutputReferences, _Mapping]] = ..., logs_available: bool = ..., cache_status: _Optional[_Union[_catalog_pb2.CatalogCacheStatus, str]] = ..., cluster_events: _Optional[_Iterable[_Union[ClusterEvent, _Mapping]]] = ..., phase_transitions: _Optional[_Iterable[_Union[PhaseTransition, _Mapping]]] = ..., cluster: _Optional[str] = ...) -> None: ...
|
|
250
|
+
|
|
251
|
+
class ClusterEvent(_message.Message):
|
|
252
|
+
__slots__ = ["occurred_at", "message"]
|
|
253
|
+
OCCURRED_AT_FIELD_NUMBER: _ClassVar[int]
|
|
254
|
+
MESSAGE_FIELD_NUMBER: _ClassVar[int]
|
|
255
|
+
occurred_at: _timestamp_pb2.Timestamp
|
|
256
|
+
message: str
|
|
257
|
+
def __init__(self, occurred_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., message: _Optional[str] = ...) -> None: ...
|
|
258
|
+
|
|
259
|
+
class PhaseTransition(_message.Message):
|
|
260
|
+
__slots__ = ["phase", "start_time", "end_time"]
|
|
261
|
+
PHASE_FIELD_NUMBER: _ClassVar[int]
|
|
262
|
+
START_TIME_FIELD_NUMBER: _ClassVar[int]
|
|
263
|
+
END_TIME_FIELD_NUMBER: _ClassVar[int]
|
|
264
|
+
phase: Phase
|
|
265
|
+
start_time: _timestamp_pb2.Timestamp
|
|
266
|
+
end_time: _timestamp_pb2.Timestamp
|
|
267
|
+
def __init__(self, phase: _Optional[_Union[Phase, str]] = ..., start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
|
|
261
268
|
|
|
262
269
|
class ActionEvent(_message.Message):
|
|
263
|
-
__slots__ = ["id", "attempt", "phase", "version", "start_time", "updated_time", "end_time", "error_info", "log_info", "log_context", "cluster", "outputs"]
|
|
270
|
+
__slots__ = ["id", "attempt", "phase", "version", "start_time", "updated_time", "end_time", "error_info", "log_info", "log_context", "cluster", "outputs", "cache_status", "cluster_events"]
|
|
264
271
|
ID_FIELD_NUMBER: _ClassVar[int]
|
|
265
272
|
ATTEMPT_FIELD_NUMBER: _ClassVar[int]
|
|
266
273
|
PHASE_FIELD_NUMBER: _ClassVar[int]
|
|
@@ -273,7 +280,9 @@ class ActionEvent(_message.Message):
|
|
|
273
280
|
LOG_CONTEXT_FIELD_NUMBER: _ClassVar[int]
|
|
274
281
|
CLUSTER_FIELD_NUMBER: _ClassVar[int]
|
|
275
282
|
OUTPUTS_FIELD_NUMBER: _ClassVar[int]
|
|
276
|
-
|
|
283
|
+
CACHE_STATUS_FIELD_NUMBER: _ClassVar[int]
|
|
284
|
+
CLUSTER_EVENTS_FIELD_NUMBER: _ClassVar[int]
|
|
285
|
+
id: _identifier_pb2.ActionIdentifier
|
|
277
286
|
attempt: int
|
|
278
287
|
phase: Phase
|
|
279
288
|
version: int
|
|
@@ -285,7 +294,9 @@ class ActionEvent(_message.Message):
|
|
|
285
294
|
log_context: _execution_pb2.LogContext
|
|
286
295
|
cluster: str
|
|
287
296
|
outputs: OutputReferences
|
|
288
|
-
|
|
297
|
+
cache_status: _catalog_pb2.CatalogCacheStatus
|
|
298
|
+
cluster_events: _containers.RepeatedCompositeFieldContainer[ClusterEvent]
|
|
299
|
+
def __init__(self, id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ..., attempt: _Optional[int] = ..., phase: _Optional[_Union[Phase, str]] = ..., version: _Optional[int] = ..., start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., updated_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., error_info: _Optional[_Union[ErrorInfo, _Mapping]] = ..., log_info: _Optional[_Iterable[_Union[_execution_pb2.TaskLog, _Mapping]]] = ..., log_context: _Optional[_Union[_execution_pb2.LogContext, _Mapping]] = ..., cluster: _Optional[str] = ..., outputs: _Optional[_Union[OutputReferences, _Mapping]] = ..., cache_status: _Optional[_Union[_catalog_pb2.CatalogCacheStatus, str]] = ..., cluster_events: _Optional[_Iterable[_Union[ClusterEvent, _Mapping]]] = ...) -> None: ...
|
|
289
300
|
|
|
290
301
|
class NamedLiteral(_message.Message):
|
|
291
302
|
__slots__ = ["name", "value"]
|
|
@@ -296,10 +307,12 @@ class NamedLiteral(_message.Message):
|
|
|
296
307
|
def __init__(self, name: _Optional[str] = ..., value: _Optional[_Union[_literals_pb2.Literal, _Mapping]] = ...) -> None: ...
|
|
297
308
|
|
|
298
309
|
class OutputReferences(_message.Message):
|
|
299
|
-
__slots__ = ["output_uri"]
|
|
310
|
+
__slots__ = ["output_uri", "report_uri"]
|
|
300
311
|
OUTPUT_URI_FIELD_NUMBER: _ClassVar[int]
|
|
312
|
+
REPORT_URI_FIELD_NUMBER: _ClassVar[int]
|
|
301
313
|
output_uri: str
|
|
302
|
-
|
|
314
|
+
report_uri: str
|
|
315
|
+
def __init__(self, output_uri: _Optional[str] = ..., report_uri: _Optional[str] = ...) -> None: ...
|
|
303
316
|
|
|
304
317
|
class Inputs(_message.Message):
|
|
305
318
|
__slots__ = ["literals"]
|
|
@@ -11,12 +11,12 @@ from google.protobuf.internal import builder as _builder
|
|
|
11
11
|
_sym_db = _symbol_database.Default()
|
|
12
12
|
|
|
13
13
|
|
|
14
|
+
from flyte._protos.common import identifier_pb2 as common_dot_identifier__pb2
|
|
14
15
|
from flyte._protos.logs.dataplane import payload_pb2 as logs_dot_dataplane_dot_payload__pb2
|
|
15
16
|
from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
|
|
16
|
-
from flyte._protos.workflow import run_definition_pb2 as workflow_dot_run__definition__pb2
|
|
17
17
|
|
|
18
18
|
|
|
19
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1fworkflow/run_logs_service.proto\x12\x11\x63loudidl.workflow\x1a\x1clogs/dataplane/payload.proto\x1a\x17validate/validate.proto\
|
|
19
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1fworkflow/run_logs_service.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x1clogs/dataplane/payload.proto\x1a\x17validate/validate.proto\"~\n\x0fTailLogsRequest\x12H\n\taction_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\x12!\n\x07\x61ttempt\x18\x02 \x01(\rB\x07\xfa\x42\x04*\x02 \x00R\x07\x61ttempt\"\x90\x01\n\x10TailLogsResponse\x12<\n\x04logs\x18\x01 \x03(\x0b\x32(.cloudidl.workflow.TailLogsResponse.LogsR\x04logs\x1a>\n\x04Logs\x12\x36\n\x05lines\x18\x01 \x03(\x0b\x32 .cloudidl.logs.dataplane.LogLineR\x05lines2l\n\x0eRunLogsService\x12Z\n\x08TailLogs\x12\".cloudidl.workflow.TailLogsRequest\x1a#.cloudidl.workflow.TailLogsResponse\"\x03\x90\x02\x01\x30\x01\x42\xc0\x01\n\x15\x63om.cloudidl.workflowB\x13RunLogsServiceProtoH\x02P\x01Z+github.com/unionai/cloud/gen/pb-go/workflow\xa2\x02\x03\x43WX\xaa\x02\x11\x43loudidl.Workflow\xca\x02\x11\x43loudidl\\Workflow\xe2\x02\x1d\x43loudidl\\Workflow\\GPBMetadata\xea\x02\x12\x43loudidl::Workflowb\x06proto3')
|
|
20
20
|
|
|
21
21
|
_globals = globals()
|
|
22
22
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
@@ -30,12 +30,12 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
|
30
30
|
_TAILLOGSREQUEST.fields_by_name['attempt']._serialized_options = b'\372B\004*\002 \000'
|
|
31
31
|
_RUNLOGSSERVICE.methods_by_name['TailLogs']._options = None
|
|
32
32
|
_RUNLOGSSERVICE.methods_by_name['TailLogs']._serialized_options = b'\220\002\001'
|
|
33
|
-
_globals['_TAILLOGSREQUEST']._serialized_start=
|
|
34
|
-
_globals['_TAILLOGSREQUEST']._serialized_end=
|
|
35
|
-
_globals['_TAILLOGSRESPONSE']._serialized_start=
|
|
36
|
-
_globals['_TAILLOGSRESPONSE']._serialized_end=
|
|
37
|
-
_globals['_TAILLOGSRESPONSE_LOGS']._serialized_start=
|
|
38
|
-
_globals['_TAILLOGSRESPONSE_LOGS']._serialized_end=
|
|
39
|
-
_globals['_RUNLOGSSERVICE']._serialized_start=
|
|
40
|
-
_globals['_RUNLOGSSERVICE']._serialized_end=
|
|
33
|
+
_globals['_TAILLOGSREQUEST']._serialized_start=134
|
|
34
|
+
_globals['_TAILLOGSREQUEST']._serialized_end=260
|
|
35
|
+
_globals['_TAILLOGSRESPONSE']._serialized_start=263
|
|
36
|
+
_globals['_TAILLOGSRESPONSE']._serialized_end=407
|
|
37
|
+
_globals['_TAILLOGSRESPONSE_LOGS']._serialized_start=345
|
|
38
|
+
_globals['_TAILLOGSRESPONSE_LOGS']._serialized_end=407
|
|
39
|
+
_globals['_RUNLOGSSERVICE']._serialized_start=409
|
|
40
|
+
_globals['_RUNLOGSSERVICE']._serialized_end=517
|
|
41
41
|
# @@protoc_insertion_point(module_scope)
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
+
from flyte._protos.common import identifier_pb2 as _identifier_pb2
|
|
1
2
|
from flyte._protos.logs.dataplane import payload_pb2 as _payload_pb2
|
|
2
3
|
from flyte._protos.validate.validate import validate_pb2 as _validate_pb2
|
|
3
|
-
from flyte._protos.workflow import run_definition_pb2 as _run_definition_pb2
|
|
4
4
|
from google.protobuf.internal import containers as _containers
|
|
5
5
|
from google.protobuf import descriptor as _descriptor
|
|
6
6
|
from google.protobuf import message as _message
|
|
@@ -12,9 +12,9 @@ class TailLogsRequest(_message.Message):
|
|
|
12
12
|
__slots__ = ["action_id", "attempt"]
|
|
13
13
|
ACTION_ID_FIELD_NUMBER: _ClassVar[int]
|
|
14
14
|
ATTEMPT_FIELD_NUMBER: _ClassVar[int]
|
|
15
|
-
action_id:
|
|
15
|
+
action_id: _identifier_pb2.ActionIdentifier
|
|
16
16
|
attempt: int
|
|
17
|
-
def __init__(self, action_id: _Optional[_Union[
|
|
17
|
+
def __init__(self, action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ..., attempt: _Optional[int] = ...) -> None: ...
|
|
18
18
|
|
|
19
19
|
class TailLogsResponse(_message.Message):
|
|
20
20
|
__slots__ = ["logs"]
|
|
@@ -18,7 +18,7 @@ from flyte._protos.workflow import run_definition_pb2 as workflow_dot_run__defin
|
|
|
18
18
|
from flyte._protos.workflow import task_definition_pb2 as workflow_dot_task__definition__pb2
|
|
19
19
|
|
|
20
20
|
|
|
21
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1aworkflow/run_service.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x11\x63ommon/list.proto\x1a\x17validate/validate.proto\x1a\x1dworkflow/run_definition.proto\x1a\x1eworkflow/task_definition.proto\"\
|
|
21
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1aworkflow/run_service.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x11\x63ommon/list.proto\x1a\x17validate/validate.proto\x1a\x1dworkflow/run_definition.proto\x1a\x1eworkflow/task_definition.proto\"\xb4\x03\n\x10\x43reateRunRequest\x12\x41\n\x06run_id\x18\x01 \x01(\x0b\x32\x1e.cloudidl.common.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x05runId\x12M\n\nproject_id\x18\x06 \x01(\x0b\x32\".cloudidl.common.ProjectIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tprojectId\x12\x46\n\x07task_id\x18\x02 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x01R\x06taskId\x12\x44\n\ttask_spec\x18\x03 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x01R\x08taskSpec\x12\x31\n\x06inputs\x18\x04 \x01(\x0b\x32\x19.cloudidl.workflow.InputsR\x06inputs\x12\x35\n\x08run_spec\x18\x05 \x01(\x0b\x32\x1a.cloudidl.workflow.RunSpecR\x07runSpecB\t\n\x02id\x12\x03\xf8\x42\x01\x42\x0b\n\x04task\x12\x03\xf8\x42\x01\"=\n\x11\x43reateRunResponse\x12(\n\x03run\x18\x01 \x01(\x0b\x32\x16.cloudidl.workflow.RunR\x03run\"R\n\x0f\x41\x62ortRunRequest\x12?\n\x06run_id\x18\x01 \x01(\x0b\x32\x1e.cloudidl.common.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\"\x12\n\x10\x41\x62ortRunResponse\"W\n\x14GetRunDetailsRequest\x12?\n\x06run_id\x18\x01 \x01(\x0b\x32\x1e.cloudidl.common.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\"P\n\x15GetRunDetailsResponse\x12\x37\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32\x1d.cloudidl.workflow.RunDetailsR\x07\x64\x65tails\"Y\n\x16WatchRunDetailsRequest\x12?\n\x06run_id\x18\x01 \x01(\x0b\x32\x1e.cloudidl.common.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\"R\n\x17WatchRunDetailsResponse\x12\x37\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32\x1d.cloudidl.workflow.RunDetailsR\x07\x64\x65tails\"c\n\x17GetActionDetailsRequest\x12H\n\taction_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\"V\n\x18GetActionDetailsResponse\x12:\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.ActionDetailsR\x07\x64\x65tails\"e\n\x19WatchActionDetailsRequest\x12H\n\taction_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\"X\n\x1aWatchActionDetailsResponse\x12:\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.ActionDetailsR\x07\x64\x65tails\"`\n\x14GetActionDataRequest\x12H\n\taction_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\"\x80\x01\n\x15GetActionDataResponse\x12\x31\n\x06inputs\x18\x01 \x01(\x0b\x32\x19.cloudidl.workflow.InputsR\x06inputs\x12\x34\n\x07outputs\x18\x02 \x01(\x0b\x32\x1a.cloudidl.workflow.OutputsR\x07outputs\"\xd2\x01\n\x0fListRunsRequest\x12\x36\n\x07request\x18\x01 \x01(\x0b\x32\x1c.cloudidl.common.ListRequestR\x07request\x12\x1b\n\x03org\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x03org\x12M\n\nproject_id\x18\x04 \x01(\x0b\x32\".cloudidl.common.ProjectIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tprojectIdB\x0f\n\x08scope_by\x12\x03\xf8\x42\x01J\x04\x08\x03\x10\x04J\x04\x08\x05\x10\x06\"T\n\x10ListRunsResponse\x12*\n\x04runs\x18\x01 \x03(\x0b\x32\x16.cloudidl.workflow.RunR\x04runs\x12\x14\n\x05token\x18\x02 \x01(\tR\x05token\"\xa4\x02\n\x10WatchRunsRequest\x12\x1b\n\x03org\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x03org\x12M\n\ncluster_id\x18\x03 \x01(\x0b\x32\".cloudidl.common.ClusterIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tclusterId\x12M\n\nproject_id\x18\x04 \x01(\x0b\x32\".cloudidl.common.ProjectIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tprojectId\x12\x46\n\x07task_id\x18\x05 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x06taskIdB\r\n\x06target\x12\x03\xf8\x42\x01\"?\n\x11WatchRunsResponse\x12*\n\x04runs\x18\x01 \x03(\x0b\x32\x16.cloudidl.workflow.RunR\x04runs\"\x8d\x01\n\x12ListActionsRequest\x12\x36\n\x07request\x18\x01 \x01(\x0b\x32\x1c.cloudidl.common.ListRequestR\x07request\x12?\n\x06run_id\x18\x02 \x01(\x0b\x32\x1e.cloudidl.common.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\"`\n\x13ListActionsResponse\x12\x33\n\x07\x61\x63tions\x18\x01 \x03(\x0b\x32\x19.cloudidl.workflow.ActionR\x07\x61\x63tions\x12\x14\n\x05token\x18\x02 \x01(\tR\x05token\"\x87\x01\n\x13WatchActionsRequest\x12?\n\x06run_id\x18\x01 \x01(\x0b\x32\x1e.cloudidl.common.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\x12/\n\x06\x66ilter\x18\x02 \x03(\x0b\x32\x17.cloudidl.common.FilterR\x06\x66ilter\"d\n\x14WatchActionsResponse\x12L\n\x10\x65nriched_actions\x18\x01 \x03(\x0b\x32!.cloudidl.workflow.EnrichedActionR\x0f\x65nrichedActions\"{\n\x19WatchClusterEventsRequest\x12;\n\x02id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x02id\x12!\n\x07\x61ttempt\x18\x02 \x01(\rB\x07\xfa\x42\x04*\x02 \x00R\x07\x61ttempt\"d\n\x1aWatchClusterEventsResponse\x12\x46\n\x0e\x63luster_events\x18\x01 \x03(\x0b\x32\x1f.cloudidl.workflow.ClusterEventR\rclusterEvents2\xdb\t\n\nRunService\x12X\n\tCreateRun\x12#.cloudidl.workflow.CreateRunRequest\x1a$.cloudidl.workflow.CreateRunResponse\"\x00\x12U\n\x08\x41\x62ortRun\x12\".cloudidl.workflow.AbortRunRequest\x1a#.cloudidl.workflow.AbortRunResponse\"\x00\x12g\n\rGetRunDetails\x12\'.cloudidl.workflow.GetRunDetailsRequest\x1a(.cloudidl.workflow.GetRunDetailsResponse\"\x03\x90\x02\x01\x12l\n\x0fWatchRunDetails\x12).cloudidl.workflow.WatchRunDetailsRequest\x1a*.cloudidl.workflow.WatchRunDetailsResponse\"\x00\x30\x01\x12p\n\x10GetActionDetails\x12*.cloudidl.workflow.GetActionDetailsRequest\x1a+.cloudidl.workflow.GetActionDetailsResponse\"\x03\x90\x02\x01\x12u\n\x12WatchActionDetails\x12,.cloudidl.workflow.WatchActionDetailsRequest\x1a-.cloudidl.workflow.WatchActionDetailsResponse\"\x00\x30\x01\x12g\n\rGetActionData\x12\'.cloudidl.workflow.GetActionDataRequest\x1a(.cloudidl.workflow.GetActionDataResponse\"\x03\x90\x02\x01\x12X\n\x08ListRuns\x12\".cloudidl.workflow.ListRunsRequest\x1a#.cloudidl.workflow.ListRunsResponse\"\x03\x90\x02\x01\x12Z\n\tWatchRuns\x12#.cloudidl.workflow.WatchRunsRequest\x1a$.cloudidl.workflow.WatchRunsResponse\"\x00\x30\x01\x12\x61\n\x0bListActions\x12%.cloudidl.workflow.ListActionsRequest\x1a&.cloudidl.workflow.ListActionsResponse\"\x03\x90\x02\x01\x12\x63\n\x0cWatchActions\x12&.cloudidl.workflow.WatchActionsRequest\x1a\'.cloudidl.workflow.WatchActionsResponse\"\x00\x30\x01\x12u\n\x12WatchClusterEvents\x12,.cloudidl.workflow.WatchClusterEventsRequest\x1a-.cloudidl.workflow.WatchClusterEventsResponse\"\x00\x30\x01\x42\xbc\x01\n\x15\x63om.cloudidl.workflowB\x0fRunServiceProtoH\x02P\x01Z+github.com/unionai/cloud/gen/pb-go/workflow\xa2\x02\x03\x43WX\xaa\x02\x11\x43loudidl.Workflow\xca\x02\x11\x43loudidl\\Workflow\xe2\x02\x1d\x43loudidl\\Workflow\\GPBMetadata\xea\x02\x12\x43loudidl::Workflowb\x06proto3')
|
|
22
22
|
|
|
23
23
|
_globals = globals()
|
|
24
24
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
@@ -70,6 +70,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
|
70
70
|
_LISTACTIONSREQUEST.fields_by_name['run_id']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
71
71
|
_WATCHACTIONSREQUEST.fields_by_name['run_id']._options = None
|
|
72
72
|
_WATCHACTIONSREQUEST.fields_by_name['run_id']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
73
|
+
_WATCHCLUSTEREVENTSREQUEST.fields_by_name['id']._options = None
|
|
74
|
+
_WATCHCLUSTEREVENTSREQUEST.fields_by_name['id']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
75
|
+
_WATCHCLUSTEREVENTSREQUEST.fields_by_name['attempt']._options = None
|
|
76
|
+
_WATCHCLUSTEREVENTSREQUEST.fields_by_name['attempt']._serialized_options = b'\372B\004*\002 \000'
|
|
73
77
|
_RUNSERVICE.methods_by_name['GetRunDetails']._options = None
|
|
74
78
|
_RUNSERVICE.methods_by_name['GetRunDetails']._serialized_options = b'\220\002\001'
|
|
75
79
|
_RUNSERVICE.methods_by_name['GetActionDetails']._options = None
|
|
@@ -81,49 +85,53 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
|
81
85
|
_RUNSERVICE.methods_by_name['ListActions']._options = None
|
|
82
86
|
_RUNSERVICE.methods_by_name['ListActions']._serialized_options = b'\220\002\001'
|
|
83
87
|
_globals['_CREATERUNREQUEST']._serialized_start=182
|
|
84
|
-
_globals['_CREATERUNREQUEST']._serialized_end=
|
|
85
|
-
_globals['_CREATERUNRESPONSE']._serialized_start=
|
|
86
|
-
_globals['_CREATERUNRESPONSE']._serialized_end=
|
|
87
|
-
_globals['_ABORTRUNREQUEST']._serialized_start=
|
|
88
|
-
_globals['_ABORTRUNREQUEST']._serialized_end=
|
|
89
|
-
_globals['_ABORTRUNRESPONSE']._serialized_start=
|
|
90
|
-
_globals['_ABORTRUNRESPONSE']._serialized_end=
|
|
91
|
-
_globals['_GETRUNDETAILSREQUEST']._serialized_start=
|
|
92
|
-
_globals['_GETRUNDETAILSREQUEST']._serialized_end=
|
|
93
|
-
_globals['_GETRUNDETAILSRESPONSE']._serialized_start=
|
|
94
|
-
_globals['_GETRUNDETAILSRESPONSE']._serialized_end=
|
|
95
|
-
_globals['_WATCHRUNDETAILSREQUEST']._serialized_start=
|
|
96
|
-
_globals['_WATCHRUNDETAILSREQUEST']._serialized_end=
|
|
97
|
-
_globals['_WATCHRUNDETAILSRESPONSE']._serialized_start=
|
|
98
|
-
_globals['_WATCHRUNDETAILSRESPONSE']._serialized_end=
|
|
99
|
-
_globals['_GETACTIONDETAILSREQUEST']._serialized_start=
|
|
100
|
-
_globals['_GETACTIONDETAILSREQUEST']._serialized_end=
|
|
101
|
-
_globals['_GETACTIONDETAILSRESPONSE']._serialized_start=
|
|
102
|
-
_globals['_GETACTIONDETAILSRESPONSE']._serialized_end=
|
|
103
|
-
_globals['_WATCHACTIONDETAILSREQUEST']._serialized_start=
|
|
104
|
-
_globals['_WATCHACTIONDETAILSREQUEST']._serialized_end=
|
|
105
|
-
_globals['_WATCHACTIONDETAILSRESPONSE']._serialized_start=
|
|
106
|
-
_globals['_WATCHACTIONDETAILSRESPONSE']._serialized_end=
|
|
107
|
-
_globals['_GETACTIONDATAREQUEST']._serialized_start=
|
|
108
|
-
_globals['_GETACTIONDATAREQUEST']._serialized_end=
|
|
109
|
-
_globals['_GETACTIONDATARESPONSE']._serialized_start=
|
|
110
|
-
_globals['_GETACTIONDATARESPONSE']._serialized_end=
|
|
111
|
-
_globals['_LISTRUNSREQUEST']._serialized_start=
|
|
112
|
-
_globals['_LISTRUNSREQUEST']._serialized_end=
|
|
113
|
-
_globals['_LISTRUNSRESPONSE']._serialized_start=
|
|
114
|
-
_globals['_LISTRUNSRESPONSE']._serialized_end=
|
|
115
|
-
_globals['_WATCHRUNSREQUEST']._serialized_start=
|
|
116
|
-
_globals['_WATCHRUNSREQUEST']._serialized_end=
|
|
117
|
-
_globals['_WATCHRUNSRESPONSE']._serialized_start=
|
|
118
|
-
_globals['_WATCHRUNSRESPONSE']._serialized_end=
|
|
119
|
-
_globals['_LISTACTIONSREQUEST']._serialized_start=
|
|
120
|
-
_globals['_LISTACTIONSREQUEST']._serialized_end=
|
|
121
|
-
_globals['_LISTACTIONSRESPONSE']._serialized_start=
|
|
122
|
-
_globals['_LISTACTIONSRESPONSE']._serialized_end=
|
|
123
|
-
_globals['_WATCHACTIONSREQUEST']._serialized_start=
|
|
124
|
-
_globals['_WATCHACTIONSREQUEST']._serialized_end=
|
|
125
|
-
_globals['_WATCHACTIONSRESPONSE']._serialized_start=
|
|
126
|
-
_globals['_WATCHACTIONSRESPONSE']._serialized_end=
|
|
127
|
-
_globals['
|
|
128
|
-
_globals['
|
|
88
|
+
_globals['_CREATERUNREQUEST']._serialized_end=618
|
|
89
|
+
_globals['_CREATERUNRESPONSE']._serialized_start=620
|
|
90
|
+
_globals['_CREATERUNRESPONSE']._serialized_end=681
|
|
91
|
+
_globals['_ABORTRUNREQUEST']._serialized_start=683
|
|
92
|
+
_globals['_ABORTRUNREQUEST']._serialized_end=765
|
|
93
|
+
_globals['_ABORTRUNRESPONSE']._serialized_start=767
|
|
94
|
+
_globals['_ABORTRUNRESPONSE']._serialized_end=785
|
|
95
|
+
_globals['_GETRUNDETAILSREQUEST']._serialized_start=787
|
|
96
|
+
_globals['_GETRUNDETAILSREQUEST']._serialized_end=874
|
|
97
|
+
_globals['_GETRUNDETAILSRESPONSE']._serialized_start=876
|
|
98
|
+
_globals['_GETRUNDETAILSRESPONSE']._serialized_end=956
|
|
99
|
+
_globals['_WATCHRUNDETAILSREQUEST']._serialized_start=958
|
|
100
|
+
_globals['_WATCHRUNDETAILSREQUEST']._serialized_end=1047
|
|
101
|
+
_globals['_WATCHRUNDETAILSRESPONSE']._serialized_start=1049
|
|
102
|
+
_globals['_WATCHRUNDETAILSRESPONSE']._serialized_end=1131
|
|
103
|
+
_globals['_GETACTIONDETAILSREQUEST']._serialized_start=1133
|
|
104
|
+
_globals['_GETACTIONDETAILSREQUEST']._serialized_end=1232
|
|
105
|
+
_globals['_GETACTIONDETAILSRESPONSE']._serialized_start=1234
|
|
106
|
+
_globals['_GETACTIONDETAILSRESPONSE']._serialized_end=1320
|
|
107
|
+
_globals['_WATCHACTIONDETAILSREQUEST']._serialized_start=1322
|
|
108
|
+
_globals['_WATCHACTIONDETAILSREQUEST']._serialized_end=1423
|
|
109
|
+
_globals['_WATCHACTIONDETAILSRESPONSE']._serialized_start=1425
|
|
110
|
+
_globals['_WATCHACTIONDETAILSRESPONSE']._serialized_end=1513
|
|
111
|
+
_globals['_GETACTIONDATAREQUEST']._serialized_start=1515
|
|
112
|
+
_globals['_GETACTIONDATAREQUEST']._serialized_end=1611
|
|
113
|
+
_globals['_GETACTIONDATARESPONSE']._serialized_start=1614
|
|
114
|
+
_globals['_GETACTIONDATARESPONSE']._serialized_end=1742
|
|
115
|
+
_globals['_LISTRUNSREQUEST']._serialized_start=1745
|
|
116
|
+
_globals['_LISTRUNSREQUEST']._serialized_end=1955
|
|
117
|
+
_globals['_LISTRUNSRESPONSE']._serialized_start=1957
|
|
118
|
+
_globals['_LISTRUNSRESPONSE']._serialized_end=2041
|
|
119
|
+
_globals['_WATCHRUNSREQUEST']._serialized_start=2044
|
|
120
|
+
_globals['_WATCHRUNSREQUEST']._serialized_end=2336
|
|
121
|
+
_globals['_WATCHRUNSRESPONSE']._serialized_start=2338
|
|
122
|
+
_globals['_WATCHRUNSRESPONSE']._serialized_end=2401
|
|
123
|
+
_globals['_LISTACTIONSREQUEST']._serialized_start=2404
|
|
124
|
+
_globals['_LISTACTIONSREQUEST']._serialized_end=2545
|
|
125
|
+
_globals['_LISTACTIONSRESPONSE']._serialized_start=2547
|
|
126
|
+
_globals['_LISTACTIONSRESPONSE']._serialized_end=2643
|
|
127
|
+
_globals['_WATCHACTIONSREQUEST']._serialized_start=2646
|
|
128
|
+
_globals['_WATCHACTIONSREQUEST']._serialized_end=2781
|
|
129
|
+
_globals['_WATCHACTIONSRESPONSE']._serialized_start=2783
|
|
130
|
+
_globals['_WATCHACTIONSRESPONSE']._serialized_end=2883
|
|
131
|
+
_globals['_WATCHCLUSTEREVENTSREQUEST']._serialized_start=2885
|
|
132
|
+
_globals['_WATCHCLUSTEREVENTSREQUEST']._serialized_end=3008
|
|
133
|
+
_globals['_WATCHCLUSTEREVENTSRESPONSE']._serialized_start=3010
|
|
134
|
+
_globals['_WATCHCLUSTEREVENTSRESPONSE']._serialized_end=3110
|
|
135
|
+
_globals['_RUNSERVICE']._serialized_start=3113
|
|
136
|
+
_globals['_RUNSERVICE']._serialized_end=4356
|
|
129
137
|
# @@protoc_insertion_point(module_scope)
|
|
@@ -18,13 +18,13 @@ class CreateRunRequest(_message.Message):
|
|
|
18
18
|
TASK_SPEC_FIELD_NUMBER: _ClassVar[int]
|
|
19
19
|
INPUTS_FIELD_NUMBER: _ClassVar[int]
|
|
20
20
|
RUN_SPEC_FIELD_NUMBER: _ClassVar[int]
|
|
21
|
-
run_id:
|
|
21
|
+
run_id: _identifier_pb2.RunIdentifier
|
|
22
22
|
project_id: _identifier_pb2.ProjectIdentifier
|
|
23
23
|
task_id: _task_definition_pb2.TaskIdentifier
|
|
24
24
|
task_spec: _task_definition_pb2.TaskSpec
|
|
25
25
|
inputs: _run_definition_pb2.Inputs
|
|
26
26
|
run_spec: _run_definition_pb2.RunSpec
|
|
27
|
-
def __init__(self, run_id: _Optional[_Union[
|
|
27
|
+
def __init__(self, run_id: _Optional[_Union[_identifier_pb2.RunIdentifier, _Mapping]] = ..., project_id: _Optional[_Union[_identifier_pb2.ProjectIdentifier, _Mapping]] = ..., task_id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ..., task_spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ..., inputs: _Optional[_Union[_run_definition_pb2.Inputs, _Mapping]] = ..., run_spec: _Optional[_Union[_run_definition_pb2.RunSpec, _Mapping]] = ...) -> None: ...
|
|
28
28
|
|
|
29
29
|
class CreateRunResponse(_message.Message):
|
|
30
30
|
__slots__ = ["run"]
|
|
@@ -35,8 +35,8 @@ class CreateRunResponse(_message.Message):
|
|
|
35
35
|
class AbortRunRequest(_message.Message):
|
|
36
36
|
__slots__ = ["run_id"]
|
|
37
37
|
RUN_ID_FIELD_NUMBER: _ClassVar[int]
|
|
38
|
-
run_id:
|
|
39
|
-
def __init__(self, run_id: _Optional[_Union[
|
|
38
|
+
run_id: _identifier_pb2.RunIdentifier
|
|
39
|
+
def __init__(self, run_id: _Optional[_Union[_identifier_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
|
|
40
40
|
|
|
41
41
|
class AbortRunResponse(_message.Message):
|
|
42
42
|
__slots__ = []
|
|
@@ -45,8 +45,8 @@ class AbortRunResponse(_message.Message):
|
|
|
45
45
|
class GetRunDetailsRequest(_message.Message):
|
|
46
46
|
__slots__ = ["run_id"]
|
|
47
47
|
RUN_ID_FIELD_NUMBER: _ClassVar[int]
|
|
48
|
-
run_id:
|
|
49
|
-
def __init__(self, run_id: _Optional[_Union[
|
|
48
|
+
run_id: _identifier_pb2.RunIdentifier
|
|
49
|
+
def __init__(self, run_id: _Optional[_Union[_identifier_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
|
|
50
50
|
|
|
51
51
|
class GetRunDetailsResponse(_message.Message):
|
|
52
52
|
__slots__ = ["details"]
|
|
@@ -57,8 +57,8 @@ class GetRunDetailsResponse(_message.Message):
|
|
|
57
57
|
class WatchRunDetailsRequest(_message.Message):
|
|
58
58
|
__slots__ = ["run_id"]
|
|
59
59
|
RUN_ID_FIELD_NUMBER: _ClassVar[int]
|
|
60
|
-
run_id:
|
|
61
|
-
def __init__(self, run_id: _Optional[_Union[
|
|
60
|
+
run_id: _identifier_pb2.RunIdentifier
|
|
61
|
+
def __init__(self, run_id: _Optional[_Union[_identifier_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
|
|
62
62
|
|
|
63
63
|
class WatchRunDetailsResponse(_message.Message):
|
|
64
64
|
__slots__ = ["details"]
|
|
@@ -69,8 +69,8 @@ class WatchRunDetailsResponse(_message.Message):
|
|
|
69
69
|
class GetActionDetailsRequest(_message.Message):
|
|
70
70
|
__slots__ = ["action_id"]
|
|
71
71
|
ACTION_ID_FIELD_NUMBER: _ClassVar[int]
|
|
72
|
-
action_id:
|
|
73
|
-
def __init__(self, action_id: _Optional[_Union[
|
|
72
|
+
action_id: _identifier_pb2.ActionIdentifier
|
|
73
|
+
def __init__(self, action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
|
|
74
74
|
|
|
75
75
|
class GetActionDetailsResponse(_message.Message):
|
|
76
76
|
__slots__ = ["details"]
|
|
@@ -81,8 +81,8 @@ class GetActionDetailsResponse(_message.Message):
|
|
|
81
81
|
class WatchActionDetailsRequest(_message.Message):
|
|
82
82
|
__slots__ = ["action_id"]
|
|
83
83
|
ACTION_ID_FIELD_NUMBER: _ClassVar[int]
|
|
84
|
-
action_id:
|
|
85
|
-
def __init__(self, action_id: _Optional[_Union[
|
|
84
|
+
action_id: _identifier_pb2.ActionIdentifier
|
|
85
|
+
def __init__(self, action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
|
|
86
86
|
|
|
87
87
|
class WatchActionDetailsResponse(_message.Message):
|
|
88
88
|
__slots__ = ["details"]
|
|
@@ -93,8 +93,8 @@ class WatchActionDetailsResponse(_message.Message):
|
|
|
93
93
|
class GetActionDataRequest(_message.Message):
|
|
94
94
|
__slots__ = ["action_id"]
|
|
95
95
|
ACTION_ID_FIELD_NUMBER: _ClassVar[int]
|
|
96
|
-
action_id:
|
|
97
|
-
def __init__(self, action_id: _Optional[_Union[
|
|
96
|
+
action_id: _identifier_pb2.ActionIdentifier
|
|
97
|
+
def __init__(self, action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
|
|
98
98
|
|
|
99
99
|
class GetActionDataResponse(_message.Message):
|
|
100
100
|
__slots__ = ["inputs", "outputs"]
|
|
@@ -145,8 +145,8 @@ class ListActionsRequest(_message.Message):
|
|
|
145
145
|
REQUEST_FIELD_NUMBER: _ClassVar[int]
|
|
146
146
|
RUN_ID_FIELD_NUMBER: _ClassVar[int]
|
|
147
147
|
request: _list_pb2.ListRequest
|
|
148
|
-
run_id:
|
|
149
|
-
def __init__(self, request: _Optional[_Union[_list_pb2.ListRequest, _Mapping]] = ..., run_id: _Optional[_Union[
|
|
148
|
+
run_id: _identifier_pb2.RunIdentifier
|
|
149
|
+
def __init__(self, request: _Optional[_Union[_list_pb2.ListRequest, _Mapping]] = ..., run_id: _Optional[_Union[_identifier_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
|
|
150
150
|
|
|
151
151
|
class ListActionsResponse(_message.Message):
|
|
152
152
|
__slots__ = ["actions", "token"]
|
|
@@ -160,12 +160,26 @@ class WatchActionsRequest(_message.Message):
|
|
|
160
160
|
__slots__ = ["run_id", "filter"]
|
|
161
161
|
RUN_ID_FIELD_NUMBER: _ClassVar[int]
|
|
162
162
|
FILTER_FIELD_NUMBER: _ClassVar[int]
|
|
163
|
-
run_id:
|
|
163
|
+
run_id: _identifier_pb2.RunIdentifier
|
|
164
164
|
filter: _containers.RepeatedCompositeFieldContainer[_list_pb2.Filter]
|
|
165
|
-
def __init__(self, run_id: _Optional[_Union[
|
|
165
|
+
def __init__(self, run_id: _Optional[_Union[_identifier_pb2.RunIdentifier, _Mapping]] = ..., filter: _Optional[_Iterable[_Union[_list_pb2.Filter, _Mapping]]] = ...) -> None: ...
|
|
166
166
|
|
|
167
167
|
class WatchActionsResponse(_message.Message):
|
|
168
168
|
__slots__ = ["enriched_actions"]
|
|
169
169
|
ENRICHED_ACTIONS_FIELD_NUMBER: _ClassVar[int]
|
|
170
170
|
enriched_actions: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.EnrichedAction]
|
|
171
171
|
def __init__(self, enriched_actions: _Optional[_Iterable[_Union[_run_definition_pb2.EnrichedAction, _Mapping]]] = ...) -> None: ...
|
|
172
|
+
|
|
173
|
+
class WatchClusterEventsRequest(_message.Message):
|
|
174
|
+
__slots__ = ["id", "attempt"]
|
|
175
|
+
ID_FIELD_NUMBER: _ClassVar[int]
|
|
176
|
+
ATTEMPT_FIELD_NUMBER: _ClassVar[int]
|
|
177
|
+
id: _identifier_pb2.ActionIdentifier
|
|
178
|
+
attempt: int
|
|
179
|
+
def __init__(self, id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ..., attempt: _Optional[int] = ...) -> None: ...
|
|
180
|
+
|
|
181
|
+
class WatchClusterEventsResponse(_message.Message):
|
|
182
|
+
__slots__ = ["cluster_events"]
|
|
183
|
+
CLUSTER_EVENTS_FIELD_NUMBER: _ClassVar[int]
|
|
184
|
+
cluster_events: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.ClusterEvent]
|
|
185
|
+
def __init__(self, cluster_events: _Optional[_Iterable[_Union[_run_definition_pb2.ClusterEvent, _Mapping]]] = ...) -> None: ...
|
|
@@ -70,6 +70,11 @@ class RunServiceStub(object):
|
|
|
70
70
|
request_serializer=workflow_dot_run__service__pb2.WatchActionsRequest.SerializeToString,
|
|
71
71
|
response_deserializer=workflow_dot_run__service__pb2.WatchActionsResponse.FromString,
|
|
72
72
|
)
|
|
73
|
+
self.WatchClusterEvents = channel.unary_stream(
|
|
74
|
+
'/cloudidl.workflow.RunService/WatchClusterEvents',
|
|
75
|
+
request_serializer=workflow_dot_run__service__pb2.WatchClusterEventsRequest.SerializeToString,
|
|
76
|
+
response_deserializer=workflow_dot_run__service__pb2.WatchClusterEventsResponse.FromString,
|
|
77
|
+
)
|
|
73
78
|
|
|
74
79
|
|
|
75
80
|
class RunServiceServicer(object):
|
|
@@ -155,6 +160,13 @@ class RunServiceServicer(object):
|
|
|
155
160
|
context.set_details('Method not implemented!')
|
|
156
161
|
raise NotImplementedError('Method not implemented!')
|
|
157
162
|
|
|
163
|
+
def WatchClusterEvents(self, request, context):
|
|
164
|
+
"""Stream of k8s cluster events in human readable form
|
|
165
|
+
"""
|
|
166
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
167
|
+
context.set_details('Method not implemented!')
|
|
168
|
+
raise NotImplementedError('Method not implemented!')
|
|
169
|
+
|
|
158
170
|
|
|
159
171
|
def add_RunServiceServicer_to_server(servicer, server):
|
|
160
172
|
rpc_method_handlers = {
|
|
@@ -213,6 +225,11 @@ def add_RunServiceServicer_to_server(servicer, server):
|
|
|
213
225
|
request_deserializer=workflow_dot_run__service__pb2.WatchActionsRequest.FromString,
|
|
214
226
|
response_serializer=workflow_dot_run__service__pb2.WatchActionsResponse.SerializeToString,
|
|
215
227
|
),
|
|
228
|
+
'WatchClusterEvents': grpc.unary_stream_rpc_method_handler(
|
|
229
|
+
servicer.WatchClusterEvents,
|
|
230
|
+
request_deserializer=workflow_dot_run__service__pb2.WatchClusterEventsRequest.FromString,
|
|
231
|
+
response_serializer=workflow_dot_run__service__pb2.WatchClusterEventsResponse.SerializeToString,
|
|
232
|
+
),
|
|
216
233
|
}
|
|
217
234
|
generic_handler = grpc.method_handlers_generic_handler(
|
|
218
235
|
'cloudidl.workflow.RunService', rpc_method_handlers)
|
|
@@ -410,3 +427,20 @@ class RunService(object):
|
|
|
410
427
|
workflow_dot_run__service__pb2.WatchActionsResponse.FromString,
|
|
411
428
|
options, channel_credentials,
|
|
412
429
|
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
|
430
|
+
|
|
431
|
+
@staticmethod
|
|
432
|
+
def WatchClusterEvents(request,
|
|
433
|
+
target,
|
|
434
|
+
options=(),
|
|
435
|
+
channel_credentials=None,
|
|
436
|
+
call_credentials=None,
|
|
437
|
+
insecure=False,
|
|
438
|
+
compression=None,
|
|
439
|
+
wait_for_ready=None,
|
|
440
|
+
timeout=None,
|
|
441
|
+
metadata=None):
|
|
442
|
+
return grpc.experimental.unary_stream(request, target, '/cloudidl.workflow.RunService/WatchClusterEvents',
|
|
443
|
+
workflow_dot_run__service__pb2.WatchClusterEventsRequest.SerializeToString,
|
|
444
|
+
workflow_dot_run__service__pb2.WatchClusterEventsResponse.FromString,
|
|
445
|
+
options, channel_credentials,
|
|
446
|
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
|
@@ -11,13 +11,14 @@ from google.protobuf.internal import builder as _builder
|
|
|
11
11
|
_sym_db = _symbol_database.Default()
|
|
12
12
|
|
|
13
13
|
|
|
14
|
+
from flyte._protos.common import identifier_pb2 as common_dot_identifier__pb2
|
|
14
15
|
from flyteidl.core import execution_pb2 as flyteidl_dot_core_dot_execution__pb2
|
|
15
16
|
from google.rpc import status_pb2 as google_dot_rpc_dot_status__pb2
|
|
16
17
|
from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
|
|
17
18
|
from flyte._protos.workflow import run_definition_pb2 as workflow_dot_run__definition__pb2
|
|
18
19
|
|
|
19
20
|
|
|
20
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cworkflow/state_service.proto\x12\x11\x63loudidl.workflow\x1a\x1d\x66lyteidl/core/execution.proto\x1a\x17google/rpc/status.proto\x1a\x17validate/validate.proto\x1a\x1dworkflow/run_definition.proto\"\
|
|
21
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cworkflow/state_service.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x1d\x66lyteidl/core/execution.proto\x1a\x17google/rpc/status.proto\x1a\x17validate/validate.proto\x1a\x1dworkflow/run_definition.proto\"\xbf\x01\n\nPutRequest\x12H\n\taction_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\x12\x31\n\x12parent_action_name\x18\x02 \x01(\tH\x00R\x10parentActionName\x88\x01\x01\x12\x1d\n\x05state\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x05stateB\x15\n\x13_parent_action_name\"\x8d\x01\n\x0bPutResponse\x12H\n\taction_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\x12\x34\n\x06status\x18\x02 \x01(\x0b\x32\x12.google.rpc.StatusB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06status\"V\n\nGetRequest\x12H\n\taction_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\"\xac\x01\n\x0bGetResponse\x12H\n\taction_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\x12\x34\n\x06status\x18\x02 \x01(\x0b\x32\x12.google.rpc.StatusB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06status\x12\x1d\n\x05state\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x05state\"l\n\x0cWatchRequest\x12M\n\x10parent_action_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierH\x00R\x0eparentActionIdB\r\n\x06\x66ilter\x12\x03\xf8\x42\x01\"\xb0\x01\n\rWatchResponse\x12\x46\n\raction_update\x18\x01 \x01(\x0b\x32\x1f.cloudidl.workflow.ActionUpdateH\x00R\x0c\x61\x63tionUpdate\x12L\n\x0f\x63ontrol_message\x18\x02 \x01(\x0b\x32!.cloudidl.workflow.ControlMessageH\x00R\x0e\x63ontrolMessageB\t\n\x07message\",\n\x0e\x43ontrolMessage\x12\x1a\n\x08sentinel\x18\x01 \x01(\x08R\x08sentinel\"\xeb\x01\n\x0c\x41\x63tionUpdate\x12H\n\taction_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\x12.\n\x05phase\x18\x02 \x01(\x0e\x32\x18.cloudidl.workflow.PhaseR\x05phase\x12\x38\n\x05\x65rror\x18\x03 \x01(\x0b\x32\x1d.flyteidl.core.ExecutionErrorH\x00R\x05\x65rror\x88\x01\x01\x12\x1d\n\noutput_uri\x18\x04 \x01(\tR\toutputUriB\x08\n\x06_error2\xf6\x01\n\x0cStateService\x12J\n\x03Put\x12\x1d.cloudidl.workflow.PutRequest\x1a\x1e.cloudidl.workflow.PutResponse\"\x00(\x01\x30\x01\x12J\n\x03Get\x12\x1d.cloudidl.workflow.GetRequest\x1a\x1e.cloudidl.workflow.GetResponse\"\x00(\x01\x30\x01\x12N\n\x05Watch\x12\x1f.cloudidl.workflow.WatchRequest\x1a .cloudidl.workflow.WatchResponse\"\x00\x30\x01\x42\xbe\x01\n\x15\x63om.cloudidl.workflowB\x11StateServiceProtoH\x02P\x01Z+github.com/unionai/cloud/gen/pb-go/workflow\xa2\x02\x03\x43WX\xaa\x02\x11\x43loudidl.Workflow\xca\x02\x11\x43loudidl\\Workflow\xe2\x02\x1d\x43loudidl\\Workflow\\GPBMetadata\xea\x02\x12\x43loudidl::Workflowb\x06proto3')
|
|
21
22
|
|
|
22
23
|
_globals = globals()
|
|
23
24
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
@@ -45,22 +46,22 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
|
45
46
|
_WATCHREQUEST.oneofs_by_name['filter']._serialized_options = b'\370B\001'
|
|
46
47
|
_ACTIONUPDATE.fields_by_name['action_id']._options = None
|
|
47
48
|
_ACTIONUPDATE.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
48
|
-
_globals['_PUTREQUEST']._serialized_start=
|
|
49
|
-
_globals['_PUTREQUEST']._serialized_end=
|
|
50
|
-
_globals['_PUTRESPONSE']._serialized_start=
|
|
51
|
-
_globals['_PUTRESPONSE']._serialized_end=
|
|
52
|
-
_globals['_GETREQUEST']._serialized_start=
|
|
53
|
-
_globals['_GETREQUEST']._serialized_end=
|
|
54
|
-
_globals['_GETRESPONSE']._serialized_start=
|
|
55
|
-
_globals['_GETRESPONSE']._serialized_end=
|
|
56
|
-
_globals['_WATCHREQUEST']._serialized_start=
|
|
57
|
-
_globals['_WATCHREQUEST']._serialized_end=
|
|
58
|
-
_globals['_WATCHRESPONSE']._serialized_start=
|
|
59
|
-
_globals['_WATCHRESPONSE']._serialized_end=
|
|
60
|
-
_globals['_CONTROLMESSAGE']._serialized_start=
|
|
61
|
-
_globals['_CONTROLMESSAGE']._serialized_end=
|
|
62
|
-
_globals['_ACTIONUPDATE']._serialized_start=
|
|
63
|
-
_globals['_ACTIONUPDATE']._serialized_end=
|
|
64
|
-
_globals['_STATESERVICE']._serialized_start=
|
|
65
|
-
_globals['_STATESERVICE']._serialized_end=
|
|
49
|
+
_globals['_PUTREQUEST']._serialized_start=189
|
|
50
|
+
_globals['_PUTREQUEST']._serialized_end=380
|
|
51
|
+
_globals['_PUTRESPONSE']._serialized_start=383
|
|
52
|
+
_globals['_PUTRESPONSE']._serialized_end=524
|
|
53
|
+
_globals['_GETREQUEST']._serialized_start=526
|
|
54
|
+
_globals['_GETREQUEST']._serialized_end=612
|
|
55
|
+
_globals['_GETRESPONSE']._serialized_start=615
|
|
56
|
+
_globals['_GETRESPONSE']._serialized_end=787
|
|
57
|
+
_globals['_WATCHREQUEST']._serialized_start=789
|
|
58
|
+
_globals['_WATCHREQUEST']._serialized_end=897
|
|
59
|
+
_globals['_WATCHRESPONSE']._serialized_start=900
|
|
60
|
+
_globals['_WATCHRESPONSE']._serialized_end=1076
|
|
61
|
+
_globals['_CONTROLMESSAGE']._serialized_start=1078
|
|
62
|
+
_globals['_CONTROLMESSAGE']._serialized_end=1122
|
|
63
|
+
_globals['_ACTIONUPDATE']._serialized_start=1125
|
|
64
|
+
_globals['_ACTIONUPDATE']._serialized_end=1360
|
|
65
|
+
_globals['_STATESERVICE']._serialized_start=1363
|
|
66
|
+
_globals['_STATESERVICE']._serialized_end=1609
|
|
66
67
|
# @@protoc_insertion_point(module_scope)
|