flyte 0.2.0b35__py3-none-any.whl → 0.2.0b36__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.

Files changed (37) hide show
  1. flyte/_image.py +1 -1
  2. flyte/_internal/controllers/_local_controller.py +3 -2
  3. flyte/_internal/controllers/_trace.py +14 -10
  4. flyte/_internal/controllers/remote/_action.py +37 -7
  5. flyte/_internal/controllers/remote/_controller.py +43 -21
  6. flyte/_internal/controllers/remote/_core.py +32 -16
  7. flyte/_internal/controllers/remote/_informer.py +18 -7
  8. flyte/_internal/runtime/task_serde.py +17 -6
  9. flyte/_protos/common/identifier_pb2.py +23 -1
  10. flyte/_protos/common/identifier_pb2.pyi +28 -0
  11. flyte/_protos/workflow/queue_service_pb2.py +33 -29
  12. flyte/_protos/workflow/queue_service_pb2.pyi +34 -16
  13. flyte/_protos/workflow/run_definition_pb2.py +64 -71
  14. flyte/_protos/workflow/run_definition_pb2.pyi +44 -31
  15. flyte/_protos/workflow/run_logs_service_pb2.py +10 -10
  16. flyte/_protos/workflow/run_logs_service_pb2.pyi +3 -3
  17. flyte/_protos/workflow/run_service_pb2.py +54 -46
  18. flyte/_protos/workflow/run_service_pb2.pyi +32 -18
  19. flyte/_protos/workflow/run_service_pb2_grpc.py +34 -0
  20. flyte/_protos/workflow/state_service_pb2.py +20 -19
  21. flyte/_protos/workflow/state_service_pb2.pyi +13 -12
  22. flyte/_run.py +11 -6
  23. flyte/_trace.py +4 -10
  24. flyte/_version.py +2 -2
  25. flyte/migrate/__init__.py +1 -0
  26. flyte/migrate/dynamic.py +13 -0
  27. flyte/migrate/task.py +99 -0
  28. flyte/migrate/workflow.py +13 -0
  29. flyte/remote/_action.py +56 -25
  30. flyte/remote/_logs.py +4 -3
  31. flyte/remote/_run.py +5 -4
  32. {flyte-0.2.0b35.dist-info → flyte-0.2.0b36.dist-info}/METADATA +2 -2
  33. {flyte-0.2.0b35.dist-info → flyte-0.2.0b36.dist-info}/RECORD +37 -33
  34. {flyte-0.2.0b35.data → flyte-0.2.0b36.data}/scripts/runtime.py +0 -0
  35. {flyte-0.2.0b35.dist-info → flyte-0.2.0b36.dist-info}/WHEEL +0 -0
  36. {flyte-0.2.0b35.dist-info → flyte-0.2.0b36.dist-info}/entry_points.txt +0 -0
  37. {flyte-0.2.0b35.dist-info → flyte-0.2.0b36.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
- 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 = ...) -> None: ...
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
- id: ActionIdentifier
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
- def __init__(self, id: _Optional[_Union[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]] = ...) -> None: ...
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
- def __init__(self, output_uri: _Optional[str] = ...) -> None: ...
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\x1a\x1dworkflow/run_definition.proto\"\x80\x01\n\x0fTailLogsRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.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')
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=141
34
- _globals['_TAILLOGSREQUEST']._serialized_end=269
35
- _globals['_TAILLOGSRESPONSE']._serialized_start=272
36
- _globals['_TAILLOGSRESPONSE']._serialized_end=416
37
- _globals['_TAILLOGSRESPONSE_LOGS']._serialized_start=354
38
- _globals['_TAILLOGSRESPONSE_LOGS']._serialized_end=416
39
- _globals['_RUNLOGSSERVICE']._serialized_start=418
40
- _globals['_RUNLOGSSERVICE']._serialized_end=526
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: _run_definition_pb2.ActionIdentifier
15
+ action_id: _identifier_pb2.ActionIdentifier
16
16
  attempt: int
17
- def __init__(self, action_id: _Optional[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]] = ..., attempt: _Optional[int] = ...) -> None: ...
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\"\xb6\x03\n\x10\x43reateRunRequest\x12\x43\n\x06run_id\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.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\"T\n\x0f\x41\x62ortRunRequest\x12\x41\n\x06run_id\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\"\x12\n\x10\x41\x62ortRunResponse\"Y\n\x14GetRunDetailsRequest\x12\x41\n\x06run_id\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.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\"[\n\x16WatchRunDetailsRequest\x12\x41\n\x06run_id\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.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\"e\n\x17GetActionDetailsRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.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\"g\n\x19WatchActionDetailsRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.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\"b\n\x14GetActionDataRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.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\"\x8f\x01\n\x12ListActionsRequest\x12\x36\n\x07request\x18\x01 \x01(\x0b\x32\x1c.cloudidl.common.ListRequestR\x07request\x12\x41\n\x06run_id\x18\x02 \x01(\x0b\x32 .cloudidl.workflow.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\"\x89\x01\n\x13WatchActionsRequest\x12\x41\n\x06run_id\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.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\x65nrichedActions2\xe4\x08\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\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')
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=620
85
- _globals['_CREATERUNRESPONSE']._serialized_start=622
86
- _globals['_CREATERUNRESPONSE']._serialized_end=683
87
- _globals['_ABORTRUNREQUEST']._serialized_start=685
88
- _globals['_ABORTRUNREQUEST']._serialized_end=769
89
- _globals['_ABORTRUNRESPONSE']._serialized_start=771
90
- _globals['_ABORTRUNRESPONSE']._serialized_end=789
91
- _globals['_GETRUNDETAILSREQUEST']._serialized_start=791
92
- _globals['_GETRUNDETAILSREQUEST']._serialized_end=880
93
- _globals['_GETRUNDETAILSRESPONSE']._serialized_start=882
94
- _globals['_GETRUNDETAILSRESPONSE']._serialized_end=962
95
- _globals['_WATCHRUNDETAILSREQUEST']._serialized_start=964
96
- _globals['_WATCHRUNDETAILSREQUEST']._serialized_end=1055
97
- _globals['_WATCHRUNDETAILSRESPONSE']._serialized_start=1057
98
- _globals['_WATCHRUNDETAILSRESPONSE']._serialized_end=1139
99
- _globals['_GETACTIONDETAILSREQUEST']._serialized_start=1141
100
- _globals['_GETACTIONDETAILSREQUEST']._serialized_end=1242
101
- _globals['_GETACTIONDETAILSRESPONSE']._serialized_start=1244
102
- _globals['_GETACTIONDETAILSRESPONSE']._serialized_end=1330
103
- _globals['_WATCHACTIONDETAILSREQUEST']._serialized_start=1332
104
- _globals['_WATCHACTIONDETAILSREQUEST']._serialized_end=1435
105
- _globals['_WATCHACTIONDETAILSRESPONSE']._serialized_start=1437
106
- _globals['_WATCHACTIONDETAILSRESPONSE']._serialized_end=1525
107
- _globals['_GETACTIONDATAREQUEST']._serialized_start=1527
108
- _globals['_GETACTIONDATAREQUEST']._serialized_end=1625
109
- _globals['_GETACTIONDATARESPONSE']._serialized_start=1628
110
- _globals['_GETACTIONDATARESPONSE']._serialized_end=1756
111
- _globals['_LISTRUNSREQUEST']._serialized_start=1759
112
- _globals['_LISTRUNSREQUEST']._serialized_end=1969
113
- _globals['_LISTRUNSRESPONSE']._serialized_start=1971
114
- _globals['_LISTRUNSRESPONSE']._serialized_end=2055
115
- _globals['_WATCHRUNSREQUEST']._serialized_start=2058
116
- _globals['_WATCHRUNSREQUEST']._serialized_end=2350
117
- _globals['_WATCHRUNSRESPONSE']._serialized_start=2352
118
- _globals['_WATCHRUNSRESPONSE']._serialized_end=2415
119
- _globals['_LISTACTIONSREQUEST']._serialized_start=2418
120
- _globals['_LISTACTIONSREQUEST']._serialized_end=2561
121
- _globals['_LISTACTIONSRESPONSE']._serialized_start=2563
122
- _globals['_LISTACTIONSRESPONSE']._serialized_end=2659
123
- _globals['_WATCHACTIONSREQUEST']._serialized_start=2662
124
- _globals['_WATCHACTIONSREQUEST']._serialized_end=2799
125
- _globals['_WATCHACTIONSRESPONSE']._serialized_start=2801
126
- _globals['_WATCHACTIONSRESPONSE']._serialized_end=2901
127
- _globals['_RUNSERVICE']._serialized_start=2904
128
- _globals['_RUNSERVICE']._serialized_end=4028
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: _run_definition_pb2.RunIdentifier
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[_run_definition_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: ...
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: _run_definition_pb2.RunIdentifier
39
- def __init__(self, run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
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: _run_definition_pb2.RunIdentifier
49
- def __init__(self, run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
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: _run_definition_pb2.RunIdentifier
61
- def __init__(self, run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
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: _run_definition_pb2.ActionIdentifier
73
- def __init__(self, action_id: _Optional[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
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: _run_definition_pb2.ActionIdentifier
85
- def __init__(self, action_id: _Optional[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
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: _run_definition_pb2.ActionIdentifier
97
- def __init__(self, action_id: _Optional[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
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: _run_definition_pb2.RunIdentifier
149
- def __init__(self, request: _Optional[_Union[_list_pb2.ListRequest, _Mapping]] = ..., run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
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: _run_definition_pb2.RunIdentifier
163
+ run_id: _identifier_pb2.RunIdentifier
164
164
  filter: _containers.RepeatedCompositeFieldContainer[_list_pb2.Filter]
165
- def __init__(self, run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ..., filter: _Optional[_Iterable[_Union[_list_pb2.Filter, _Mapping]]] = ...) -> None: ...
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\"\xc1\x01\n\nPutRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.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\"\x8f\x01\n\x0bPutResponse\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.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\"X\n\nGetRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\"\xae\x01\n\x0bGetResponse\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.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\"n\n\x0cWatchRequest\x12O\n\x10parent_action_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.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\"\xed\x01\n\x0c\x41\x63tionUpdate\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.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
+ 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=164
49
- _globals['_PUTREQUEST']._serialized_end=357
50
- _globals['_PUTRESPONSE']._serialized_start=360
51
- _globals['_PUTRESPONSE']._serialized_end=503
52
- _globals['_GETREQUEST']._serialized_start=505
53
- _globals['_GETREQUEST']._serialized_end=593
54
- _globals['_GETRESPONSE']._serialized_start=596
55
- _globals['_GETRESPONSE']._serialized_end=770
56
- _globals['_WATCHREQUEST']._serialized_start=772
57
- _globals['_WATCHREQUEST']._serialized_end=882
58
- _globals['_WATCHRESPONSE']._serialized_start=885
59
- _globals['_WATCHRESPONSE']._serialized_end=1061
60
- _globals['_CONTROLMESSAGE']._serialized_start=1063
61
- _globals['_CONTROLMESSAGE']._serialized_end=1107
62
- _globals['_ACTIONUPDATE']._serialized_start=1110
63
- _globals['_ACTIONUPDATE']._serialized_end=1347
64
- _globals['_STATESERVICE']._serialized_start=1350
65
- _globals['_STATESERVICE']._serialized_end=1596
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)