flyte 2.0.0b9__py3-none-any.whl → 2.0.0b14__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 (60) hide show
  1. flyte/__init__.py +55 -31
  2. flyte/_bin/debug.py +38 -0
  3. flyte/_bin/runtime.py +13 -0
  4. flyte/_code_bundle/_utils.py +2 -0
  5. flyte/_code_bundle/bundle.py +4 -4
  6. flyte/_context.py +1 -1
  7. flyte/_debug/__init__.py +0 -0
  8. flyte/_debug/constants.py +39 -0
  9. flyte/_debug/utils.py +17 -0
  10. flyte/_debug/vscode.py +300 -0
  11. flyte/_environment.py +5 -5
  12. flyte/_image.py +34 -19
  13. flyte/_initialize.py +15 -29
  14. flyte/_internal/controllers/remote/_action.py +2 -2
  15. flyte/_internal/controllers/remote/_controller.py +1 -1
  16. flyte/_internal/imagebuild/docker_builder.py +11 -15
  17. flyte/_internal/imagebuild/remote_builder.py +71 -22
  18. flyte/_internal/runtime/entrypoints.py +3 -0
  19. flyte/_internal/runtime/reuse.py +7 -3
  20. flyte/_internal/runtime/task_serde.py +4 -3
  21. flyte/_internal/runtime/taskrunner.py +9 -3
  22. flyte/_logging.py +5 -2
  23. flyte/_protos/common/identifier_pb2.py +25 -19
  24. flyte/_protos/common/identifier_pb2.pyi +10 -0
  25. flyte/_protos/imagebuilder/definition_pb2.py +32 -31
  26. flyte/_protos/imagebuilder/definition_pb2.pyi +25 -12
  27. flyte/_protos/workflow/queue_service_pb2.py +24 -24
  28. flyte/_protos/workflow/queue_service_pb2.pyi +6 -6
  29. flyte/_protos/workflow/run_definition_pb2.py +48 -48
  30. flyte/_protos/workflow/run_definition_pb2.pyi +20 -10
  31. flyte/_reusable_environment.py +41 -19
  32. flyte/_run.py +9 -9
  33. flyte/_secret.py +9 -5
  34. flyte/_task.py +16 -11
  35. flyte/_task_environment.py +11 -13
  36. flyte/_tools.py +0 -13
  37. flyte/_version.py +16 -3
  38. flyte/cli/_build.py +2 -3
  39. flyte/cli/_common.py +16 -5
  40. flyte/cli/_gen.py +10 -1
  41. flyte/cli/_get.py +16 -14
  42. flyte/cli/_run.py +258 -25
  43. flyte/models.py +9 -0
  44. flyte/remote/_client/auth/_authenticators/base.py +8 -2
  45. flyte/remote/_client/auth/_authenticators/device_code.py +1 -1
  46. flyte/remote/_client/auth/_authenticators/pkce.py +1 -1
  47. flyte/remote/_client/auth/_channel.py +0 -6
  48. flyte/remote/_client/auth/_client_config.py +4 -2
  49. flyte/remote/_client/controlplane.py +14 -0
  50. flyte/remote/_task.py +18 -4
  51. flyte/storage/_storage.py +83 -7
  52. flyte/types/_type_engine.py +3 -33
  53. flyte-2.0.0b14.data/scripts/debug.py +38 -0
  54. {flyte-2.0.0b9.data → flyte-2.0.0b14.data}/scripts/runtime.py +13 -0
  55. {flyte-2.0.0b9.dist-info → flyte-2.0.0b14.dist-info}/METADATA +2 -2
  56. {flyte-2.0.0b9.dist-info → flyte-2.0.0b14.dist-info}/RECORD +60 -54
  57. {flyte-2.0.0b9.dist-info → flyte-2.0.0b14.dist-info}/WHEEL +0 -0
  58. {flyte-2.0.0b9.dist-info → flyte-2.0.0b14.dist-info}/entry_points.txt +0 -0
  59. {flyte-2.0.0b9.dist-info → flyte-2.0.0b14.dist-info}/licenses/LICENSE +0 -0
  60. {flyte-2.0.0b9.dist-info → flyte-2.0.0b14.dist-info}/top_level.txt +0 -0
@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
14
14
  from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
15
15
 
16
16
 
17
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x17\x63ommon/identifier.proto\x12\x0f\x63loudidl.common\x1a\x17validate/validate.proto\"~\n\x11ProjectIdentifier\x12+\n\x0corganization\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x0corganization\x12\x1f\n\x06\x64omain\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x06\x64omain\x12\x1b\n\x04name\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"T\n\x11\x43lusterIdentifier\x12\"\n\x0corganization\x18\x01 \x01(\tR\x0corganization\x12\x1b\n\x04name\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"O\n\x15\x43lusterPoolIdentifier\x12\"\n\x0corganization\x18\x01 \x01(\tR\x0corganization\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\"_\n\x17\x43lusterConfigIdentifier\x12+\n\x0corganization\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x0corganization\x12\x17\n\x02id\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x02id\"3\n\x0eUserIdentifier\x12!\n\x07subject\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07subject\":\n\x15\x41pplicationIdentifier\x12!\n\x07subject\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07subject\"Q\n\x0eRoleIdentifier\x12\"\n\x0corganization\x18\x01 \x01(\tR\x0corganization\x12\x1b\n\x04name\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"O\n\rOrgIdentifier\x12>\n\x04name\x18\x01 \x01(\tB*\xfa\x42\'r%\x10\x01\x18?2\x1f^[a-z0-9]([-a-z0-9]*[a-z0-9])?$R\x04name\"y\n\x18ManagedClusterIdentifier\x12\x1b\n\x04name\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\x12:\n\x03org\x18\x03 \x01(\x0b\x32\x1e.cloudidl.common.OrgIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x03orgJ\x04\x08\x01\x10\x02\"S\n\x10PolicyIdentifier\x12\"\n\x0corganization\x18\x01 \x01(\tR\x0corganization\x12\x1b\n\x04name\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"\x93\x01\n\rRunIdentifier\x12\x1b\n\x03org\x18\x01 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x03org\x12#\n\x07project\x18\x02 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x07project\x12!\n\x06\x64omain\x18\x03 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x06\x64omain\x12\x1d\n\x04name\x18\x04 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18\x1eR\x04name\"m\n\x10\x41\x63tionIdentifier\x12:\n\x03run\x18\x01 \x01(\x0b\x32\x1e.cloudidl.common.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x03run\x12\x1d\n\x04name\x18\x02 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18\x1eR\x04name\"\x86\x01\n\x17\x41\x63tionAttemptIdentifier\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\x61ttemptB\xb0\x01\n\x13\x63om.cloudidl.commonB\x0fIdentifierProtoH\x02P\x01Z)github.com/unionai/cloud/gen/pb-go/common\xa2\x02\x03\x43\x43X\xaa\x02\x0f\x43loudidl.Common\xca\x02\x0f\x43loudidl\\Common\xe2\x02\x1b\x43loudidl\\Common\\GPBMetadata\xea\x02\x10\x43loudidl::Commonb\x06proto3')
17
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x17\x63ommon/identifier.proto\x12\x0f\x63loudidl.common\x1a\x17validate/validate.proto\"~\n\x11ProjectIdentifier\x12+\n\x0corganization\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x0corganization\x12\x1f\n\x06\x64omain\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x06\x64omain\x12\x1b\n\x04name\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"T\n\x11\x43lusterIdentifier\x12\"\n\x0corganization\x18\x01 \x01(\tR\x0corganization\x12\x1b\n\x04name\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"O\n\x15\x43lusterPoolIdentifier\x12\"\n\x0corganization\x18\x01 \x01(\tR\x0corganization\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\"_\n\x17\x43lusterConfigIdentifier\x12+\n\x0corganization\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x0corganization\x12\x17\n\x02id\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x02id\"\x88\x01\n\x19\x43lusterNodepoolIdentifier\x12\"\n\x0corganization\x18\x01 \x01(\tR\x0corganization\x12*\n\x0c\x63luster_name\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x0b\x63lusterName\x12\x1b\n\x04name\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"3\n\x0eUserIdentifier\x12!\n\x07subject\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07subject\":\n\x15\x41pplicationIdentifier\x12!\n\x07subject\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07subject\"Q\n\x0eRoleIdentifier\x12\"\n\x0corganization\x18\x01 \x01(\tR\x0corganization\x12\x1b\n\x04name\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"O\n\rOrgIdentifier\x12>\n\x04name\x18\x01 \x01(\tB*\xfa\x42\'r%\x10\x01\x18?2\x1f^[a-z0-9]([-a-z0-9]*[a-z0-9])?$R\x04name\"y\n\x18ManagedClusterIdentifier\x12\x1b\n\x04name\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\x12:\n\x03org\x18\x03 \x01(\x0b\x32\x1e.cloudidl.common.OrgIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x03orgJ\x04\x08\x01\x10\x02\"S\n\x10PolicyIdentifier\x12\"\n\x0corganization\x18\x01 \x01(\tR\x0corganization\x12\x1b\n\x04name\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"\x93\x01\n\rRunIdentifier\x12\x1b\n\x03org\x18\x01 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x03org\x12#\n\x07project\x18\x02 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x07project\x12!\n\x06\x64omain\x18\x03 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x06\x64omain\x12\x1d\n\x04name\x18\x04 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18\x1eR\x04name\"m\n\x10\x41\x63tionIdentifier\x12:\n\x03run\x18\x01 \x01(\x0b\x32\x1e.cloudidl.common.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x03run\x12\x1d\n\x04name\x18\x02 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18\x1eR\x04name\"\x86\x01\n\x17\x41\x63tionAttemptIdentifier\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\x61ttemptB\xb0\x01\n\x13\x63om.cloudidl.commonB\x0fIdentifierProtoH\x02P\x01Z)github.com/unionai/cloud/gen/pb-go/common\xa2\x02\x03\x43\x43X\xaa\x02\x0f\x43loudidl.Common\xca\x02\x0f\x43loudidl\\Common\xe2\x02\x1b\x43loudidl\\Common\\GPBMetadata\xea\x02\x10\x43loudidl::Commonb\x06proto3')
18
18
 
19
19
  _globals = globals()
20
20
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -34,6 +34,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
34
34
  _CLUSTERCONFIGIDENTIFIER.fields_by_name['organization']._serialized_options = b'\372B\004r\002\020\001'
35
35
  _CLUSTERCONFIGIDENTIFIER.fields_by_name['id']._options = None
36
36
  _CLUSTERCONFIGIDENTIFIER.fields_by_name['id']._serialized_options = b'\372B\004r\002\020\001'
37
+ _CLUSTERNODEPOOLIDENTIFIER.fields_by_name['cluster_name']._options = None
38
+ _CLUSTERNODEPOOLIDENTIFIER.fields_by_name['cluster_name']._serialized_options = b'\372B\004r\002\020\001'
39
+ _CLUSTERNODEPOOLIDENTIFIER.fields_by_name['name']._options = None
40
+ _CLUSTERNODEPOOLIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\004r\002\020\001'
37
41
  _USERIDENTIFIER.fields_by_name['subject']._options = None
38
42
  _USERIDENTIFIER.fields_by_name['subject']._serialized_options = b'\372B\004r\002\020\001'
39
43
  _APPLICATIONIDENTIFIER.fields_by_name['subject']._options = None
@@ -72,22 +76,24 @@ if _descriptor._USE_C_DESCRIPTORS == False:
72
76
  _globals['_CLUSTERPOOLIDENTIFIER']._serialized_end=362
73
77
  _globals['_CLUSTERCONFIGIDENTIFIER']._serialized_start=364
74
78
  _globals['_CLUSTERCONFIGIDENTIFIER']._serialized_end=459
75
- _globals['_USERIDENTIFIER']._serialized_start=461
76
- _globals['_USERIDENTIFIER']._serialized_end=512
77
- _globals['_APPLICATIONIDENTIFIER']._serialized_start=514
78
- _globals['_APPLICATIONIDENTIFIER']._serialized_end=572
79
- _globals['_ROLEIDENTIFIER']._serialized_start=574
80
- _globals['_ROLEIDENTIFIER']._serialized_end=655
81
- _globals['_ORGIDENTIFIER']._serialized_start=657
82
- _globals['_ORGIDENTIFIER']._serialized_end=736
83
- _globals['_MANAGEDCLUSTERIDENTIFIER']._serialized_start=738
84
- _globals['_MANAGEDCLUSTERIDENTIFIER']._serialized_end=859
85
- _globals['_POLICYIDENTIFIER']._serialized_start=861
86
- _globals['_POLICYIDENTIFIER']._serialized_end=944
87
- _globals['_RUNIDENTIFIER']._serialized_start=947
88
- _globals['_RUNIDENTIFIER']._serialized_end=1094
89
- _globals['_ACTIONIDENTIFIER']._serialized_start=1096
90
- _globals['_ACTIONIDENTIFIER']._serialized_end=1205
91
- _globals['_ACTIONATTEMPTIDENTIFIER']._serialized_start=1208
92
- _globals['_ACTIONATTEMPTIDENTIFIER']._serialized_end=1342
79
+ _globals['_CLUSTERNODEPOOLIDENTIFIER']._serialized_start=462
80
+ _globals['_CLUSTERNODEPOOLIDENTIFIER']._serialized_end=598
81
+ _globals['_USERIDENTIFIER']._serialized_start=600
82
+ _globals['_USERIDENTIFIER']._serialized_end=651
83
+ _globals['_APPLICATIONIDENTIFIER']._serialized_start=653
84
+ _globals['_APPLICATIONIDENTIFIER']._serialized_end=711
85
+ _globals['_ROLEIDENTIFIER']._serialized_start=713
86
+ _globals['_ROLEIDENTIFIER']._serialized_end=794
87
+ _globals['_ORGIDENTIFIER']._serialized_start=796
88
+ _globals['_ORGIDENTIFIER']._serialized_end=875
89
+ _globals['_MANAGEDCLUSTERIDENTIFIER']._serialized_start=877
90
+ _globals['_MANAGEDCLUSTERIDENTIFIER']._serialized_end=998
91
+ _globals['_POLICYIDENTIFIER']._serialized_start=1000
92
+ _globals['_POLICYIDENTIFIER']._serialized_end=1083
93
+ _globals['_RUNIDENTIFIER']._serialized_start=1086
94
+ _globals['_RUNIDENTIFIER']._serialized_end=1233
95
+ _globals['_ACTIONIDENTIFIER']._serialized_start=1235
96
+ _globals['_ACTIONIDENTIFIER']._serialized_end=1344
97
+ _globals['_ACTIONATTEMPTIDENTIFIER']._serialized_start=1347
98
+ _globals['_ACTIONATTEMPTIDENTIFIER']._serialized_end=1481
93
99
  # @@protoc_insertion_point(module_scope)
@@ -39,6 +39,16 @@ class ClusterConfigIdentifier(_message.Message):
39
39
  id: str
40
40
  def __init__(self, organization: _Optional[str] = ..., id: _Optional[str] = ...) -> None: ...
41
41
 
42
+ class ClusterNodepoolIdentifier(_message.Message):
43
+ __slots__ = ["organization", "cluster_name", "name"]
44
+ ORGANIZATION_FIELD_NUMBER: _ClassVar[int]
45
+ CLUSTER_NAME_FIELD_NUMBER: _ClassVar[int]
46
+ NAME_FIELD_NUMBER: _ClassVar[int]
47
+ organization: str
48
+ cluster_name: str
49
+ name: str
50
+ def __init__(self, organization: _Optional[str] = ..., cluster_name: _Optional[str] = ..., name: _Optional[str] = ...) -> None: ...
51
+
42
52
  class UserIdentifier(_message.Message):
43
53
  __slots__ = ["subject"]
44
54
  SUBJECT_FIELD_NUMBER: _ClassVar[int]
@@ -11,10 +11,11 @@ from google.protobuf.internal import builder as _builder
11
11
  _sym_db = _symbol_database.Default()
12
12
 
13
13
 
14
+ from flyteidl.core import security_pb2 as flyteidl_dot_core_dot_security__pb2
14
15
  from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
15
16
 
16
17
 
17
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1dimagebuilder/definition.proto\x12\x15\x63loudidl.imagebuilder\x1a\x17validate/validate.proto\".\n\x0fImageIdentifier\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"S\n\x05Image\x12\x36\n\x02id\x18\x01 \x01(\x0b\x32&.cloudidl.imagebuilder.ImageIdentifierR\x02id\x12\x12\n\x04\x66qin\x18\x02 \x01(\tR\x04\x66qin\")\n\x0b\x41ptPackages\x12\x1a\n\x08packages\x18\x01 \x03(\tR\x08packages\"\x84\x01\n\nPipOptions\x12\x1b\n\tindex_url\x18\x02 \x01(\tR\x08indexUrl\x12(\n\x10\x65xtra_index_urls\x18\x03 \x03(\tR\x0e\x65xtraIndexUrls\x12\x10\n\x03pre\x18\x04 \x01(\x08R\x03pre\x12\x1d\n\nextra_args\x18\x05 \x01(\tR\textraArgs\"f\n\x0bPipPackages\x12\x1a\n\x08packages\x18\x01 \x03(\tR\x08packages\x12;\n\x07options\x18\x02 \x01(\x0b\x32!.cloudidl.imagebuilder.PipOptionsR\x07options\"_\n\x0cRequirements\x12\x12\n\x04\x66ile\x18\x01 \x01(\tR\x04\x66ile\x12;\n\x07options\x18\x02 \x01(\x0b\x32!.cloudidl.imagebuilder.PipOptionsR\x07options\"]\n\x0cPythonWheels\x12\x10\n\x03\x64ir\x18\x01 \x01(\tR\x03\x64ir\x12;\n\x07options\x18\x02 \x01(\x0b\x32!.cloudidl.imagebuilder.PipOptionsR\x07options\"~\n\tUVProject\x12\x1c\n\tpyproject\x18\x01 \x01(\tR\tpyproject\x12\x16\n\x06uvlock\x18\x02 \x01(\tR\x06uvlock\x12;\n\x07options\x18\x03 \x01(\x0b\x32!.cloudidl.imagebuilder.PipOptionsR\x07options\"\x1c\n\x08\x43ommands\x12\x10\n\x03\x63md\x18\x02 \x03(\tR\x03\x63md\"#\n\x07WorkDir\x12\x18\n\x07workdir\x18\x01 \x01(\tR\x07workdir\"0\n\nCopyConfig\x12\x10\n\x03src\x18\x01 \x01(\tR\x03src\x12\x10\n\x03\x64st\x18\x02 \x01(\tR\x03\x64st\"\x99\x01\n\x03\x45nv\x12Q\n\renv_variables\x18\x01 \x03(\x0b\x32,.cloudidl.imagebuilder.Env.EnvVariablesEntryR\x0c\x65nvVariables\x1a?\n\x11\x45nvVariablesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\"\xed\x04\n\x05Layer\x12G\n\x0c\x61pt_packages\x18\x01 \x01(\x0b\x32\".cloudidl.imagebuilder.AptPackagesH\x00R\x0b\x61ptPackages\x12G\n\x0cpip_packages\x18\x02 \x01(\x0b\x32\".cloudidl.imagebuilder.PipPackagesH\x00R\x0bpipPackages\x12=\n\x08\x63ommands\x18\x03 \x01(\x0b\x32\x1f.cloudidl.imagebuilder.CommandsH\x00R\x08\x63ommands\x12I\n\x0crequirements\x18\x04 \x01(\x0b\x32#.cloudidl.imagebuilder.RequirementsH\x00R\x0crequirements\x12J\n\rpython_wheels\x18\x05 \x01(\x0b\x32#.cloudidl.imagebuilder.PythonWheelsH\x00R\x0cpythonWheels\x12:\n\x07workdir\x18\x06 \x01(\x0b\x32\x1e.cloudidl.imagebuilder.WorkDirH\x00R\x07workdir\x12\x44\n\x0b\x63opy_config\x18\x07 \x01(\x0b\x32!.cloudidl.imagebuilder.CopyConfigH\x00R\ncopyConfig\x12\x41\n\nuv_project\x18\x08 \x01(\x0b\x32 .cloudidl.imagebuilder.UVProjectH\x00R\tuvProject\x12.\n\x03\x65nv\x18\t \x01(\x0b\x32\x1a.cloudidl.imagebuilder.EnvH\x00R\x03\x65nvB\x07\n\x05layer\"\xa3\x01\n\tImageSpec\x12\x1d\n\nbase_image\x18\x01 \x01(\tR\tbaseImage\x12%\n\x0epython_version\x18\x02 \x01(\tR\rpythonVersion\x12\x34\n\x06layers\x18\x03 \x03(\x0b\x32\x1c.cloudidl.imagebuilder.LayerR\x06layers\x12\x1a\n\x08platform\x18\x04 \x03(\tR\x08platformB\xd4\x01\n\x19\x63om.cloudidl.imagebuilderB\x0f\x44\x65\x66initionProtoH\x02P\x01Z/github.com/unionai/cloud/gen/pb-go/imagebuilder\xa2\x02\x03\x43IX\xaa\x02\x15\x43loudidl.Imagebuilder\xca\x02\x15\x43loudidl\\Imagebuilder\xe2\x02!Cloudidl\\Imagebuilder\\GPBMetadata\xea\x02\x16\x43loudidl::Imagebuilderb\x06proto3')
18
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1dimagebuilder/definition.proto\x12\x15\x63loudidl.imagebuilder\x1a\x1c\x66lyteidl/core/security.proto\x1a\x17validate/validate.proto\".\n\x0fImageIdentifier\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"S\n\x05Image\x12\x36\n\x02id\x18\x01 \x01(\x0b\x32&.cloudidl.imagebuilder.ImageIdentifierR\x02id\x12\x12\n\x04\x66qin\x18\x02 \x01(\tR\x04\x66qin\"e\n\x0b\x41ptPackages\x12\x1a\n\x08packages\x18\x01 \x03(\tR\x08packages\x12:\n\rsecret_mounts\x18\x02 \x03(\x0b\x32\x15.flyteidl.core.SecretR\x0csecretMounts\"\x84\x01\n\nPipOptions\x12\x1b\n\tindex_url\x18\x02 \x01(\tR\x08indexUrl\x12(\n\x10\x65xtra_index_urls\x18\x03 \x03(\tR\x0e\x65xtraIndexUrls\x12\x10\n\x03pre\x18\x04 \x01(\x08R\x03pre\x12\x1d\n\nextra_args\x18\x05 \x01(\tR\textraArgs\"\xa2\x01\n\x0bPipPackages\x12\x1a\n\x08packages\x18\x01 \x03(\tR\x08packages\x12;\n\x07options\x18\x02 \x01(\x0b\x32!.cloudidl.imagebuilder.PipOptionsR\x07options\x12:\n\rsecret_mounts\x18\x03 \x03(\x0b\x32\x15.flyteidl.core.SecretR\x0csecretMounts\"\x9b\x01\n\x0cRequirements\x12\x12\n\x04\x66ile\x18\x01 \x01(\tR\x04\x66ile\x12;\n\x07options\x18\x02 \x01(\x0b\x32!.cloudidl.imagebuilder.PipOptionsR\x07options\x12:\n\rsecret_mounts\x18\x03 \x03(\x0b\x32\x15.flyteidl.core.SecretR\x0csecretMounts\"\x99\x01\n\x0cPythonWheels\x12\x10\n\x03\x64ir\x18\x01 \x01(\tR\x03\x64ir\x12;\n\x07options\x18\x02 \x01(\x0b\x32!.cloudidl.imagebuilder.PipOptionsR\x07options\x12:\n\rsecret_mounts\x18\x03 \x03(\x0b\x32\x15.flyteidl.core.SecretR\x0csecretMounts\"\xba\x01\n\tUVProject\x12\x1c\n\tpyproject\x18\x01 \x01(\tR\tpyproject\x12\x16\n\x06uvlock\x18\x02 \x01(\tR\x06uvlock\x12;\n\x07options\x18\x03 \x01(\x0b\x32!.cloudidl.imagebuilder.PipOptionsR\x07options\x12:\n\rsecret_mounts\x18\x04 \x03(\x0b\x32\x15.flyteidl.core.SecretR\x0csecretMounts\"X\n\x08\x43ommands\x12\x10\n\x03\x63md\x18\x02 \x03(\tR\x03\x63md\x12:\n\rsecret_mounts\x18\x03 \x03(\x0b\x32\x15.flyteidl.core.SecretR\x0csecretMounts\"#\n\x07WorkDir\x12\x18\n\x07workdir\x18\x01 \x01(\tR\x07workdir\"0\n\nCopyConfig\x12\x10\n\x03src\x18\x01 \x01(\tR\x03src\x12\x10\n\x03\x64st\x18\x02 \x01(\tR\x03\x64st\"\x99\x01\n\x03\x45nv\x12Q\n\renv_variables\x18\x01 \x03(\x0b\x32,.cloudidl.imagebuilder.Env.EnvVariablesEntryR\x0c\x65nvVariables\x1a?\n\x11\x45nvVariablesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\"\xed\x04\n\x05Layer\x12G\n\x0c\x61pt_packages\x18\x01 \x01(\x0b\x32\".cloudidl.imagebuilder.AptPackagesH\x00R\x0b\x61ptPackages\x12G\n\x0cpip_packages\x18\x02 \x01(\x0b\x32\".cloudidl.imagebuilder.PipPackagesH\x00R\x0bpipPackages\x12=\n\x08\x63ommands\x18\x03 \x01(\x0b\x32\x1f.cloudidl.imagebuilder.CommandsH\x00R\x08\x63ommands\x12I\n\x0crequirements\x18\x04 \x01(\x0b\x32#.cloudidl.imagebuilder.RequirementsH\x00R\x0crequirements\x12J\n\rpython_wheels\x18\x05 \x01(\x0b\x32#.cloudidl.imagebuilder.PythonWheelsH\x00R\x0cpythonWheels\x12:\n\x07workdir\x18\x06 \x01(\x0b\x32\x1e.cloudidl.imagebuilder.WorkDirH\x00R\x07workdir\x12\x44\n\x0b\x63opy_config\x18\x07 \x01(\x0b\x32!.cloudidl.imagebuilder.CopyConfigH\x00R\ncopyConfig\x12\x41\n\nuv_project\x18\x08 \x01(\x0b\x32 .cloudidl.imagebuilder.UVProjectH\x00R\tuvProject\x12.\n\x03\x65nv\x18\t \x01(\x0b\x32\x1a.cloudidl.imagebuilder.EnvH\x00R\x03\x65nvB\x07\n\x05layer\"\xa3\x01\n\tImageSpec\x12\x1d\n\nbase_image\x18\x01 \x01(\tR\tbaseImage\x12%\n\x0epython_version\x18\x02 \x01(\tR\rpythonVersion\x12\x34\n\x06layers\x18\x03 \x03(\x0b\x32\x1c.cloudidl.imagebuilder.LayerR\x06layers\x12\x1a\n\x08platform\x18\x04 \x03(\tR\x08platformB\xd4\x01\n\x19\x63om.cloudidl.imagebuilderB\x0f\x44\x65\x66initionProtoH\x02P\x01Z/github.com/unionai/cloud/gen/pb-go/imagebuilder\xa2\x02\x03\x43IX\xaa\x02\x15\x43loudidl.Imagebuilder\xca\x02\x15\x43loudidl\\Imagebuilder\xe2\x02!Cloudidl\\Imagebuilder\\GPBMetadata\xea\x02\x16\x43loudidl::Imagebuilderb\x06proto3')
18
19
 
19
20
  _globals = globals()
20
21
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -26,34 +27,34 @@ if _descriptor._USE_C_DESCRIPTORS == False:
26
27
  _IMAGEIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\004r\002\020\001'
27
28
  _ENV_ENVVARIABLESENTRY._options = None
28
29
  _ENV_ENVVARIABLESENTRY._serialized_options = b'8\001'
29
- _globals['_IMAGEIDENTIFIER']._serialized_start=81
30
- _globals['_IMAGEIDENTIFIER']._serialized_end=127
31
- _globals['_IMAGE']._serialized_start=129
32
- _globals['_IMAGE']._serialized_end=212
33
- _globals['_APTPACKAGES']._serialized_start=214
34
- _globals['_APTPACKAGES']._serialized_end=255
35
- _globals['_PIPOPTIONS']._serialized_start=258
36
- _globals['_PIPOPTIONS']._serialized_end=390
37
- _globals['_PIPPACKAGES']._serialized_start=392
38
- _globals['_PIPPACKAGES']._serialized_end=494
39
- _globals['_REQUIREMENTS']._serialized_start=496
40
- _globals['_REQUIREMENTS']._serialized_end=591
41
- _globals['_PYTHONWHEELS']._serialized_start=593
42
- _globals['_PYTHONWHEELS']._serialized_end=686
43
- _globals['_UVPROJECT']._serialized_start=688
44
- _globals['_UVPROJECT']._serialized_end=814
45
- _globals['_COMMANDS']._serialized_start=816
46
- _globals['_COMMANDS']._serialized_end=844
47
- _globals['_WORKDIR']._serialized_start=846
48
- _globals['_WORKDIR']._serialized_end=881
49
- _globals['_COPYCONFIG']._serialized_start=883
50
- _globals['_COPYCONFIG']._serialized_end=931
51
- _globals['_ENV']._serialized_start=934
52
- _globals['_ENV']._serialized_end=1087
53
- _globals['_ENV_ENVVARIABLESENTRY']._serialized_start=1024
54
- _globals['_ENV_ENVVARIABLESENTRY']._serialized_end=1087
55
- _globals['_LAYER']._serialized_start=1090
56
- _globals['_LAYER']._serialized_end=1711
57
- _globals['_IMAGESPEC']._serialized_start=1714
58
- _globals['_IMAGESPEC']._serialized_end=1877
30
+ _globals['_IMAGEIDENTIFIER']._serialized_start=111
31
+ _globals['_IMAGEIDENTIFIER']._serialized_end=157
32
+ _globals['_IMAGE']._serialized_start=159
33
+ _globals['_IMAGE']._serialized_end=242
34
+ _globals['_APTPACKAGES']._serialized_start=244
35
+ _globals['_APTPACKAGES']._serialized_end=345
36
+ _globals['_PIPOPTIONS']._serialized_start=348
37
+ _globals['_PIPOPTIONS']._serialized_end=480
38
+ _globals['_PIPPACKAGES']._serialized_start=483
39
+ _globals['_PIPPACKAGES']._serialized_end=645
40
+ _globals['_REQUIREMENTS']._serialized_start=648
41
+ _globals['_REQUIREMENTS']._serialized_end=803
42
+ _globals['_PYTHONWHEELS']._serialized_start=806
43
+ _globals['_PYTHONWHEELS']._serialized_end=959
44
+ _globals['_UVPROJECT']._serialized_start=962
45
+ _globals['_UVPROJECT']._serialized_end=1148
46
+ _globals['_COMMANDS']._serialized_start=1150
47
+ _globals['_COMMANDS']._serialized_end=1238
48
+ _globals['_WORKDIR']._serialized_start=1240
49
+ _globals['_WORKDIR']._serialized_end=1275
50
+ _globals['_COPYCONFIG']._serialized_start=1277
51
+ _globals['_COPYCONFIG']._serialized_end=1325
52
+ _globals['_ENV']._serialized_start=1328
53
+ _globals['_ENV']._serialized_end=1481
54
+ _globals['_ENV_ENVVARIABLESENTRY']._serialized_start=1418
55
+ _globals['_ENV_ENVVARIABLESENTRY']._serialized_end=1481
56
+ _globals['_LAYER']._serialized_start=1484
57
+ _globals['_LAYER']._serialized_end=2105
58
+ _globals['_IMAGESPEC']._serialized_start=2108
59
+ _globals['_IMAGESPEC']._serialized_end=2271
59
60
  # @@protoc_insertion_point(module_scope)
@@ -1,3 +1,4 @@
1
+ from flyteidl.core import security_pb2 as _security_pb2
1
2
  from flyte._protos.validate.validate import validate_pb2 as _validate_pb2
2
3
  from google.protobuf.internal import containers as _containers
3
4
  from google.protobuf import descriptor as _descriptor
@@ -21,10 +22,12 @@ class Image(_message.Message):
21
22
  def __init__(self, id: _Optional[_Union[ImageIdentifier, _Mapping]] = ..., fqin: _Optional[str] = ...) -> None: ...
22
23
 
23
24
  class AptPackages(_message.Message):
24
- __slots__ = ["packages"]
25
+ __slots__ = ["packages", "secret_mounts"]
25
26
  PACKAGES_FIELD_NUMBER: _ClassVar[int]
27
+ SECRET_MOUNTS_FIELD_NUMBER: _ClassVar[int]
26
28
  packages: _containers.RepeatedScalarFieldContainer[str]
27
- def __init__(self, packages: _Optional[_Iterable[str]] = ...) -> None: ...
29
+ secret_mounts: _containers.RepeatedCompositeFieldContainer[_security_pb2.Secret]
30
+ def __init__(self, packages: _Optional[_Iterable[str]] = ..., secret_mounts: _Optional[_Iterable[_Union[_security_pb2.Secret, _Mapping]]] = ...) -> None: ...
28
31
 
29
32
  class PipOptions(_message.Message):
30
33
  __slots__ = ["index_url", "extra_index_urls", "pre", "extra_args"]
@@ -39,44 +42,54 @@ class PipOptions(_message.Message):
39
42
  def __init__(self, index_url: _Optional[str] = ..., extra_index_urls: _Optional[_Iterable[str]] = ..., pre: bool = ..., extra_args: _Optional[str] = ...) -> None: ...
40
43
 
41
44
  class PipPackages(_message.Message):
42
- __slots__ = ["packages", "options"]
45
+ __slots__ = ["packages", "options", "secret_mounts"]
43
46
  PACKAGES_FIELD_NUMBER: _ClassVar[int]
44
47
  OPTIONS_FIELD_NUMBER: _ClassVar[int]
48
+ SECRET_MOUNTS_FIELD_NUMBER: _ClassVar[int]
45
49
  packages: _containers.RepeatedScalarFieldContainer[str]
46
50
  options: PipOptions
47
- def __init__(self, packages: _Optional[_Iterable[str]] = ..., options: _Optional[_Union[PipOptions, _Mapping]] = ...) -> None: ...
51
+ secret_mounts: _containers.RepeatedCompositeFieldContainer[_security_pb2.Secret]
52
+ def __init__(self, packages: _Optional[_Iterable[str]] = ..., options: _Optional[_Union[PipOptions, _Mapping]] = ..., secret_mounts: _Optional[_Iterable[_Union[_security_pb2.Secret, _Mapping]]] = ...) -> None: ...
48
53
 
49
54
  class Requirements(_message.Message):
50
- __slots__ = ["file", "options"]
55
+ __slots__ = ["file", "options", "secret_mounts"]
51
56
  FILE_FIELD_NUMBER: _ClassVar[int]
52
57
  OPTIONS_FIELD_NUMBER: _ClassVar[int]
58
+ SECRET_MOUNTS_FIELD_NUMBER: _ClassVar[int]
53
59
  file: str
54
60
  options: PipOptions
55
- def __init__(self, file: _Optional[str] = ..., options: _Optional[_Union[PipOptions, _Mapping]] = ...) -> None: ...
61
+ secret_mounts: _containers.RepeatedCompositeFieldContainer[_security_pb2.Secret]
62
+ def __init__(self, file: _Optional[str] = ..., options: _Optional[_Union[PipOptions, _Mapping]] = ..., secret_mounts: _Optional[_Iterable[_Union[_security_pb2.Secret, _Mapping]]] = ...) -> None: ...
56
63
 
57
64
  class PythonWheels(_message.Message):
58
- __slots__ = ["dir", "options"]
65
+ __slots__ = ["dir", "options", "secret_mounts"]
59
66
  DIR_FIELD_NUMBER: _ClassVar[int]
60
67
  OPTIONS_FIELD_NUMBER: _ClassVar[int]
68
+ SECRET_MOUNTS_FIELD_NUMBER: _ClassVar[int]
61
69
  dir: str
62
70
  options: PipOptions
63
- def __init__(self, dir: _Optional[str] = ..., options: _Optional[_Union[PipOptions, _Mapping]] = ...) -> None: ...
71
+ secret_mounts: _containers.RepeatedCompositeFieldContainer[_security_pb2.Secret]
72
+ def __init__(self, dir: _Optional[str] = ..., options: _Optional[_Union[PipOptions, _Mapping]] = ..., secret_mounts: _Optional[_Iterable[_Union[_security_pb2.Secret, _Mapping]]] = ...) -> None: ...
64
73
 
65
74
  class UVProject(_message.Message):
66
- __slots__ = ["pyproject", "uvlock", "options"]
75
+ __slots__ = ["pyproject", "uvlock", "options", "secret_mounts"]
67
76
  PYPROJECT_FIELD_NUMBER: _ClassVar[int]
68
77
  UVLOCK_FIELD_NUMBER: _ClassVar[int]
69
78
  OPTIONS_FIELD_NUMBER: _ClassVar[int]
79
+ SECRET_MOUNTS_FIELD_NUMBER: _ClassVar[int]
70
80
  pyproject: str
71
81
  uvlock: str
72
82
  options: PipOptions
73
- def __init__(self, pyproject: _Optional[str] = ..., uvlock: _Optional[str] = ..., options: _Optional[_Union[PipOptions, _Mapping]] = ...) -> None: ...
83
+ secret_mounts: _containers.RepeatedCompositeFieldContainer[_security_pb2.Secret]
84
+ def __init__(self, pyproject: _Optional[str] = ..., uvlock: _Optional[str] = ..., options: _Optional[_Union[PipOptions, _Mapping]] = ..., secret_mounts: _Optional[_Iterable[_Union[_security_pb2.Secret, _Mapping]]] = ...) -> None: ...
74
85
 
75
86
  class Commands(_message.Message):
76
- __slots__ = ["cmd"]
87
+ __slots__ = ["cmd", "secret_mounts"]
77
88
  CMD_FIELD_NUMBER: _ClassVar[int]
89
+ SECRET_MOUNTS_FIELD_NUMBER: _ClassVar[int]
78
90
  cmd: _containers.RepeatedScalarFieldContainer[str]
79
- def __init__(self, cmd: _Optional[_Iterable[str]] = ...) -> None: ...
91
+ secret_mounts: _containers.RepeatedCompositeFieldContainer[_security_pb2.Secret]
92
+ def __init__(self, cmd: _Optional[_Iterable[str]] = ..., secret_mounts: _Optional[_Iterable[_Union[_security_pb2.Secret, _Mapping]]] = ...) -> None: ...
80
93
 
81
94
  class WorkDir(_message.Message):
82
95
  __slots__ = ["workdir"]
@@ -20,7 +20,7 @@ from flyte._protos.workflow import run_definition_pb2 as workflow_dot_run__defin
20
20
  from flyte._protos.workflow import task_definition_pb2 as workflow_dot_task__definition__pb2
21
21
 
22
22
 
23
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cworkflow/queue_service.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x19\x66lyteidl/core/types.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\x1a\x17validate/validate.proto\x1a\x1dworkflow/run_definition.proto\x1a\x1eworkflow/task_definition.proto\"\x7f\n\x10WorkerIdentifier\x12+\n\x0corganization\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x0corganization\x12!\n\x07\x63luster\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07\x63luster\x12\x1b\n\x04name\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"\xbf\x04\n\x14\x45nqueueActionRequest\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\x01R\x10parentActionName\x88\x01\x01\x12\x35\n\x08run_spec\x18\x03 \x01(\x0b\x32\x1a.cloudidl.workflow.RunSpecR\x07runSpec\x12$\n\tinput_uri\x18\x06 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x08inputUri\x12/\n\x0frun_output_base\x18\x07 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\rrunOutputBase\x12\x14\n\x05group\x18\x08 \x01(\tR\x05group\x12\x18\n\x07subject\x18\t \x01(\tR\x07subject\x12=\n\x04task\x18\n \x01(\x0b\x32\x1d.cloudidl.workflow.TaskActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x04task\x12@\n\x05trace\x18\x0b \x01(\x0b\x32\x1e.cloudidl.workflow.TraceActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x05trace\x12L\n\tcondition\x18\x0c \x01(\x0b\x32\".cloudidl.workflow.ConditionActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tconditionB\x06\n\x04specB\x15\n\x13_parent_action_name\"\xb5\x01\n\nTaskAction\x12\x31\n\x02id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierR\x02id\x12\x39\n\x04spec\x18\x02 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x04spec\x12\x39\n\tcache_key\x18\x03 \x01(\x0b\x32\x1c.google.protobuf.StringValueR\x08\x63\x61\x63heKey\"\xce\x02\n\x0bTraceAction\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\x12.\n\x05phase\x18\x02 \x01(\x0e\x32\x18.cloudidl.workflow.PhaseR\x05phase\x12\x39\n\nstart_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12:\n\x08\x65nd_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x07\x65ndTime\x88\x01\x01\x12=\n\x07outputs\x18\x05 \x01(\x0b\x32#.cloudidl.workflow.OutputReferencesR\x07outputs\x12/\n\x04spec\x18\x06 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecR\x04specB\x0b\n\t_end_time\"\x8a\x02\n\x0f\x43onditionAction\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\x12 \n\x06run_id\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x05runId\x12&\n\taction_id\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x08\x61\x63tionId\x12\x18\n\x06global\x18\x04 \x01(\x08H\x00R\x06global\x12.\n\x04type\x18\x06 \x01(\x0b\x32\x1a.flyteidl.core.LiteralTypeR\x04type\x12\x16\n\x06prompt\x18\x07 \x01(\tR\x06prompt\x12 \n\x0b\x64\x65scription\x18\x08 \x01(\tR\x0b\x64\x65scriptionB\x0c\n\x05scope\x12\x03\xf8\x42\x01\"\x17\n\x15\x45nqueueActionResponse\"X\n\x15\x41\x62ortQueuedRunRequest\x12?\n\x06run_id\x18\x01 \x01(\x0b\x32\x1e.cloudidl.common.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\"\x18\n\x16\x41\x62ortQueuedRunResponse\"\x80\x03\n\x10HeartbeatRequest\x12J\n\tworker_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.WorkerIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08workerId\x12M\n\x11\x61\x63tive_action_ids\x18\x02 \x03(\x0b\x32!.cloudidl.common.ActionIdentifierR\x0f\x61\x63tiveActionIds\x12Q\n\x13terminal_action_ids\x18\x03 \x03(\x0b\x32!.cloudidl.common.ActionIdentifierR\x11terminalActionIds\x12O\n\x12\x61\x62orted_action_ids\x18\x04 \x03(\x0b\x32!.cloudidl.common.ActionIdentifierR\x10\x61\x62ortedActionIds\x12-\n\x12\x61vailable_capacity\x18\x05 \x01(\x05R\x11\x61vailableCapacity\"\xe2\x01\n\x11HeartbeatResponse\x12\x37\n\nnew_leases\x18\x01 \x03(\x0b\x32\x18.cloudidl.workflow.LeaseR\tnewLeases\x12?\n\x0e\x61\x62orted_leases\x18\x02 \x03(\x0b\x32\x18.cloudidl.workflow.LeaseR\rabortedLeases\x12S\n\x14\x66inalized_action_ids\x18\x03 \x03(\x0b\x32!.cloudidl.common.ActionIdentifierR\x12\x66inalizedActionIds\"a\n\x13StreamLeasesRequest\x12J\n\tworker_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.WorkerIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08workerId\"H\n\x14StreamLeasesResponse\x12\x30\n\x06leases\x18\x01 \x03(\x0b\x32\x18.cloudidl.workflow.LeaseR\x06leases\"\xc9\x04\n\x05Lease\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\x01R\x10parentActionName\x88\x01\x01\x12\x35\n\x08run_spec\x18\x03 \x01(\x0b\x32\x1a.cloudidl.workflow.RunSpecR\x07runSpec\x12$\n\tinput_uri\x18\x04 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x08inputUri\x12/\n\x0frun_output_base\x18\x05 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\rrunOutputBase\x12=\n\x04task\x18\x06 \x01(\x0b\x32\x1d.cloudidl.workflow.TaskActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x04task\x12L\n\tcondition\x18\x07 \x01(\x0b\x32\".cloudidl.workflow.ConditionActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tcondition\x12@\n\x05trace\x18\n \x01(\x0b\x32\x1e.cloudidl.workflow.TraceActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x05trace\x12\x14\n\x05group\x18\x08 \x01(\tR\x05group\x12\x18\n\x07subject\x18\t \x01(\tR\x07subject\x12\x12\n\x04host\x18\x0b \x01(\tR\x04hostB\x0b\n\x04spec\x12\x03\xf8\x42\x01\x42\x15\n\x13_parent_action_name2\xa0\x03\n\x0cQueueService\x12\x64\n\rEnqueueAction\x12\'.cloudidl.workflow.EnqueueActionRequest\x1a(.cloudidl.workflow.EnqueueActionResponse\"\x00\x12g\n\x0e\x41\x62ortQueuedRun\x12(.cloudidl.workflow.AbortQueuedRunRequest\x1a).cloudidl.workflow.AbortQueuedRunResponse\"\x00\x12\\\n\tHeartbeat\x12#.cloudidl.workflow.HeartbeatRequest\x1a$.cloudidl.workflow.HeartbeatResponse\"\x00(\x01\x30\x01\x12\x63\n\x0cStreamLeases\x12&.cloudidl.workflow.StreamLeasesRequest\x1a\'.cloudidl.workflow.StreamLeasesResponse\"\x00\x30\x01\x42\xbe\x01\n\x15\x63om.cloudidl.workflowB\x11QueueServiceProtoH\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')
23
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cworkflow/queue_service.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x19\x66lyteidl/core/types.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\x1a\x17validate/validate.proto\x1a\x1dworkflow/run_definition.proto\x1a\x1eworkflow/task_definition.proto\"\x7f\n\x10WorkerIdentifier\x12+\n\x0corganization\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x0corganization\x12!\n\x07\x63luster\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07\x63luster\x12\x1b\n\x04name\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"\xbf\x04\n\x14\x45nqueueActionRequest\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\x01R\x10parentActionName\x88\x01\x01\x12\x35\n\x08run_spec\x18\x03 \x01(\x0b\x32\x1a.cloudidl.workflow.RunSpecR\x07runSpec\x12$\n\tinput_uri\x18\x06 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x08inputUri\x12/\n\x0frun_output_base\x18\x07 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\rrunOutputBase\x12\x14\n\x05group\x18\x08 \x01(\tR\x05group\x12\x18\n\x07subject\x18\t \x01(\tR\x07subject\x12=\n\x04task\x18\n \x01(\x0b\x32\x1d.cloudidl.workflow.TaskActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x04task\x12@\n\x05trace\x18\x0b \x01(\x0b\x32\x1e.cloudidl.workflow.TraceActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x05trace\x12L\n\tcondition\x18\x0c \x01(\x0b\x32\".cloudidl.workflow.ConditionActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tconditionB\x06\n\x04specB\x15\n\x13_parent_action_name\"\xcf\x01\n\nTaskAction\x12\x31\n\x02id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierR\x02id\x12\x39\n\x04spec\x18\x02 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x04spec\x12\x39\n\tcache_key\x18\x03 \x01(\x0b\x32\x1c.google.protobuf.StringValueR\x08\x63\x61\x63heKey\x12\x18\n\x07\x63luster\x18\x04 \x01(\tR\x07\x63luster\"\x9d\x02\n\x0bTraceAction\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\x12.\n\x05phase\x18\x02 \x01(\x0e\x32\x18.cloudidl.workflow.PhaseR\x05phase\x12\x39\n\nstart_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12:\n\x08\x65nd_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x07\x65ndTime\x88\x01\x01\x12=\n\x07outputs\x18\x05 \x01(\x0b\x32#.cloudidl.workflow.OutputReferencesR\x07outputsB\x0b\n\t_end_time\"\x8a\x02\n\x0f\x43onditionAction\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\x12 \n\x06run_id\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x05runId\x12&\n\taction_id\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x08\x61\x63tionId\x12\x18\n\x06global\x18\x04 \x01(\x08H\x00R\x06global\x12.\n\x04type\x18\x06 \x01(\x0b\x32\x1a.flyteidl.core.LiteralTypeR\x04type\x12\x16\n\x06prompt\x18\x07 \x01(\tR\x06prompt\x12 \n\x0b\x64\x65scription\x18\x08 \x01(\tR\x0b\x64\x65scriptionB\x0c\n\x05scope\x12\x03\xf8\x42\x01\"\x17\n\x15\x45nqueueActionResponse\"X\n\x15\x41\x62ortQueuedRunRequest\x12?\n\x06run_id\x18\x01 \x01(\x0b\x32\x1e.cloudidl.common.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\"\x18\n\x16\x41\x62ortQueuedRunResponse\"\x80\x03\n\x10HeartbeatRequest\x12J\n\tworker_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.WorkerIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08workerId\x12M\n\x11\x61\x63tive_action_ids\x18\x02 \x03(\x0b\x32!.cloudidl.common.ActionIdentifierR\x0f\x61\x63tiveActionIds\x12Q\n\x13terminal_action_ids\x18\x03 \x03(\x0b\x32!.cloudidl.common.ActionIdentifierR\x11terminalActionIds\x12O\n\x12\x61\x62orted_action_ids\x18\x04 \x03(\x0b\x32!.cloudidl.common.ActionIdentifierR\x10\x61\x62ortedActionIds\x12-\n\x12\x61vailable_capacity\x18\x05 \x01(\x05R\x11\x61vailableCapacity\"\xe2\x01\n\x11HeartbeatResponse\x12\x37\n\nnew_leases\x18\x01 \x03(\x0b\x32\x18.cloudidl.workflow.LeaseR\tnewLeases\x12?\n\x0e\x61\x62orted_leases\x18\x02 \x03(\x0b\x32\x18.cloudidl.workflow.LeaseR\rabortedLeases\x12S\n\x14\x66inalized_action_ids\x18\x03 \x03(\x0b\x32!.cloudidl.common.ActionIdentifierR\x12\x66inalizedActionIds\"a\n\x13StreamLeasesRequest\x12J\n\tworker_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.WorkerIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08workerId\"H\n\x14StreamLeasesResponse\x12\x30\n\x06leases\x18\x01 \x03(\x0b\x32\x18.cloudidl.workflow.LeaseR\x06leases\"\xc9\x04\n\x05Lease\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\x01R\x10parentActionName\x88\x01\x01\x12\x35\n\x08run_spec\x18\x03 \x01(\x0b\x32\x1a.cloudidl.workflow.RunSpecR\x07runSpec\x12$\n\tinput_uri\x18\x04 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x08inputUri\x12/\n\x0frun_output_base\x18\x05 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\rrunOutputBase\x12=\n\x04task\x18\x06 \x01(\x0b\x32\x1d.cloudidl.workflow.TaskActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x04task\x12L\n\tcondition\x18\x07 \x01(\x0b\x32\".cloudidl.workflow.ConditionActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tcondition\x12@\n\x05trace\x18\n \x01(\x0b\x32\x1e.cloudidl.workflow.TraceActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x05trace\x12\x14\n\x05group\x18\x08 \x01(\tR\x05group\x12\x18\n\x07subject\x18\t \x01(\tR\x07subject\x12\x12\n\x04host\x18\x0b \x01(\tR\x04hostB\x0b\n\x04spec\x12\x03\xf8\x42\x01\x42\x15\n\x13_parent_action_name2\xa0\x03\n\x0cQueueService\x12\x64\n\rEnqueueAction\x12\'.cloudidl.workflow.EnqueueActionRequest\x1a(.cloudidl.workflow.EnqueueActionResponse\"\x00\x12g\n\x0e\x41\x62ortQueuedRun\x12(.cloudidl.workflow.AbortQueuedRunRequest\x1a).cloudidl.workflow.AbortQueuedRunResponse\"\x00\x12\\\n\tHeartbeat\x12#.cloudidl.workflow.HeartbeatRequest\x1a$.cloudidl.workflow.HeartbeatResponse\"\x00(\x01\x30\x01\x12\x63\n\x0cStreamLeases\x12&.cloudidl.workflow.StreamLeasesRequest\x1a\'.cloudidl.workflow.StreamLeasesResponse\"\x00\x30\x01\x42\xbe\x01\n\x15\x63om.cloudidl.workflowB\x11QueueServiceProtoH\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')
24
24
 
25
25
  _globals = globals()
26
26
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -83,27 +83,27 @@ if _descriptor._USE_C_DESCRIPTORS == False:
83
83
  _globals['_ENQUEUEACTIONREQUEST']._serialized_start=386
84
84
  _globals['_ENQUEUEACTIONREQUEST']._serialized_end=961
85
85
  _globals['_TASKACTION']._serialized_start=964
86
- _globals['_TASKACTION']._serialized_end=1145
87
- _globals['_TRACEACTION']._serialized_start=1148
88
- _globals['_TRACEACTION']._serialized_end=1482
89
- _globals['_CONDITIONACTION']._serialized_start=1485
90
- _globals['_CONDITIONACTION']._serialized_end=1751
91
- _globals['_ENQUEUEACTIONRESPONSE']._serialized_start=1753
92
- _globals['_ENQUEUEACTIONRESPONSE']._serialized_end=1776
93
- _globals['_ABORTQUEUEDRUNREQUEST']._serialized_start=1778
94
- _globals['_ABORTQUEUEDRUNREQUEST']._serialized_end=1866
95
- _globals['_ABORTQUEUEDRUNRESPONSE']._serialized_start=1868
96
- _globals['_ABORTQUEUEDRUNRESPONSE']._serialized_end=1892
97
- _globals['_HEARTBEATREQUEST']._serialized_start=1895
98
- _globals['_HEARTBEATREQUEST']._serialized_end=2279
99
- _globals['_HEARTBEATRESPONSE']._serialized_start=2282
100
- _globals['_HEARTBEATRESPONSE']._serialized_end=2508
101
- _globals['_STREAMLEASESREQUEST']._serialized_start=2510
102
- _globals['_STREAMLEASESREQUEST']._serialized_end=2607
103
- _globals['_STREAMLEASESRESPONSE']._serialized_start=2609
104
- _globals['_STREAMLEASESRESPONSE']._serialized_end=2681
105
- _globals['_LEASE']._serialized_start=2684
106
- _globals['_LEASE']._serialized_end=3269
107
- _globals['_QUEUESERVICE']._serialized_start=3272
108
- _globals['_QUEUESERVICE']._serialized_end=3688
86
+ _globals['_TASKACTION']._serialized_end=1171
87
+ _globals['_TRACEACTION']._serialized_start=1174
88
+ _globals['_TRACEACTION']._serialized_end=1459
89
+ _globals['_CONDITIONACTION']._serialized_start=1462
90
+ _globals['_CONDITIONACTION']._serialized_end=1728
91
+ _globals['_ENQUEUEACTIONRESPONSE']._serialized_start=1730
92
+ _globals['_ENQUEUEACTIONRESPONSE']._serialized_end=1753
93
+ _globals['_ABORTQUEUEDRUNREQUEST']._serialized_start=1755
94
+ _globals['_ABORTQUEUEDRUNREQUEST']._serialized_end=1843
95
+ _globals['_ABORTQUEUEDRUNRESPONSE']._serialized_start=1845
96
+ _globals['_ABORTQUEUEDRUNRESPONSE']._serialized_end=1869
97
+ _globals['_HEARTBEATREQUEST']._serialized_start=1872
98
+ _globals['_HEARTBEATREQUEST']._serialized_end=2256
99
+ _globals['_HEARTBEATRESPONSE']._serialized_start=2259
100
+ _globals['_HEARTBEATRESPONSE']._serialized_end=2485
101
+ _globals['_STREAMLEASESREQUEST']._serialized_start=2487
102
+ _globals['_STREAMLEASESREQUEST']._serialized_end=2584
103
+ _globals['_STREAMLEASESRESPONSE']._serialized_start=2586
104
+ _globals['_STREAMLEASESRESPONSE']._serialized_end=2658
105
+ _globals['_LEASE']._serialized_start=2661
106
+ _globals['_LEASE']._serialized_end=3246
107
+ _globals['_QUEUESERVICE']._serialized_start=3249
108
+ _globals['_QUEUESERVICE']._serialized_end=3665
109
109
  # @@protoc_insertion_point(module_scope)
@@ -47,30 +47,30 @@ class EnqueueActionRequest(_message.Message):
47
47
  def __init__(self, action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ..., parent_action_name: _Optional[str] = ..., run_spec: _Optional[_Union[_run_definition_pb2.RunSpec, _Mapping]] = ..., input_uri: _Optional[str] = ..., run_output_base: _Optional[str] = ..., group: _Optional[str] = ..., subject: _Optional[str] = ..., task: _Optional[_Union[TaskAction, _Mapping]] = ..., trace: _Optional[_Union[TraceAction, _Mapping]] = ..., condition: _Optional[_Union[ConditionAction, _Mapping]] = ...) -> None: ...
48
48
 
49
49
  class TaskAction(_message.Message):
50
- __slots__ = ["id", "spec", "cache_key"]
50
+ __slots__ = ["id", "spec", "cache_key", "cluster"]
51
51
  ID_FIELD_NUMBER: _ClassVar[int]
52
52
  SPEC_FIELD_NUMBER: _ClassVar[int]
53
53
  CACHE_KEY_FIELD_NUMBER: _ClassVar[int]
54
+ CLUSTER_FIELD_NUMBER: _ClassVar[int]
54
55
  id: _task_definition_pb2.TaskIdentifier
55
56
  spec: _task_definition_pb2.TaskSpec
56
57
  cache_key: _wrappers_pb2.StringValue
57
- def __init__(self, id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ..., spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ..., cache_key: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
58
+ cluster: str
59
+ def __init__(self, id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ..., spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ..., cache_key: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., cluster: _Optional[str] = ...) -> None: ...
58
60
 
59
61
  class TraceAction(_message.Message):
60
- __slots__ = ["name", "phase", "start_time", "end_time", "outputs", "spec"]
62
+ __slots__ = ["name", "phase", "start_time", "end_time", "outputs"]
61
63
  NAME_FIELD_NUMBER: _ClassVar[int]
62
64
  PHASE_FIELD_NUMBER: _ClassVar[int]
63
65
  START_TIME_FIELD_NUMBER: _ClassVar[int]
64
66
  END_TIME_FIELD_NUMBER: _ClassVar[int]
65
67
  OUTPUTS_FIELD_NUMBER: _ClassVar[int]
66
- SPEC_FIELD_NUMBER: _ClassVar[int]
67
68
  name: str
68
69
  phase: _run_definition_pb2.Phase
69
70
  start_time: _timestamp_pb2.Timestamp
70
71
  end_time: _timestamp_pb2.Timestamp
71
72
  outputs: _run_definition_pb2.OutputReferences
72
- spec: _task_definition_pb2.TaskSpec
73
- def __init__(self, name: _Optional[str] = ..., phase: _Optional[_Union[_run_definition_pb2.Phase, str]] = ..., start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., outputs: _Optional[_Union[_run_definition_pb2.OutputReferences, _Mapping]] = ..., spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ...) -> None: ...
73
+ def __init__(self, name: _Optional[str] = ..., phase: _Optional[_Union[_run_definition_pb2.Phase, str]] = ..., start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., outputs: _Optional[_Union[_run_definition_pb2.OutputReferences, _Mapping]] = ...) -> None: ...
74
74
 
75
75
  class ConditionAction(_message.Message):
76
76
  __slots__ = ["name", "run_id", "action_id", "type", "prompt", "description"]
@@ -22,7 +22,7 @@ from flyte._protos.validate.validate import validate_pb2 as validate_dot_validat
22
22
  from flyte._protos.workflow import task_definition_pb2 as workflow_dot_task__definition__pb2
23
23
 
24
24
 
25
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1dworkflow/run_definition.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x15\x63ommon/identity.proto\x1a\x1b\x66lyteidl/core/catalog.proto\x1a\x1d\x66lyteidl/core/execution.proto\x1a\x1c\x66lyteidl/core/literals.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\x1a\x17validate/validate.proto\x1a\x1eworkflow/task_definition.proto\"\x82\x01\n\x06Labels\x12=\n\x06values\x18\x01 \x03(\x0b\x32%.cloudidl.workflow.Labels.ValuesEntryR\x06values\x1a\x39\n\x0bValuesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\"\x8c\x01\n\x0b\x41nnotations\x12\x42\n\x06values\x18\x01 \x03(\x0b\x32*.cloudidl.workflow.Annotations.ValuesEntryR\x06values\x1a\x39\n\x0bValuesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\";\n\x04\x45nvs\x12\x33\n\x06values\x18\x01 \x03(\x0b\x32\x1b.flyteidl.core.KeyValuePairR\x06values\"\x96\x02\n\x07RunSpec\x12\x31\n\x06labels\x18\x01 \x01(\x0b\x32\x19.cloudidl.workflow.LabelsR\x06labels\x12@\n\x0b\x61nnotations\x18\x02 \x01(\x0b\x32\x1e.cloudidl.workflow.AnnotationsR\x0b\x61nnotations\x12+\n\x04\x65nvs\x18\x03 \x01(\x0b\x32\x17.cloudidl.workflow.EnvsR\x04\x65nvs\x12@\n\rinterruptible\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.BoolValueR\rinterruptible\x12\'\n\x0foverwrite_cache\x18\x05 \x01(\x08R\x0eoverwriteCache\"8\n\x03Run\x12\x31\n\x06\x61\x63tion\x18\x01 \x01(\x0b\x32\x19.cloudidl.workflow.ActionR\x06\x61\x63tion\"}\n\nRunDetails\x12\x35\n\x08run_spec\x18\x01 \x01(\x0b\x32\x1a.cloudidl.workflow.RunSpecR\x07runSpec\x12\x38\n\x06\x61\x63tion\x18\x02 \x01(\x0b\x32 .cloudidl.workflow.ActionDetailsR\x06\x61\x63tion\"\x83\x01\n\x12TaskActionMetadata\x12\x31\n\x02id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierR\x02id\x12\x1b\n\ttask_type\x18\x02 \x01(\tR\x08taskType\x12\x1d\n\nshort_name\x18\x03 \x01(\tR\tshortName\")\n\x13TraceActionMetadata\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\"\x9f\x01\n\x17\x43onditionActionMetadata\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12 \n\x06run_id\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x05runId\x12&\n\taction_id\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x08\x61\x63tionId\x12\x18\n\x06global\x18\x04 \x01(\x08H\x00R\x06globalB\x0c\n\x05scope\x12\x03\xf8\x42\x01\"\xf1\x02\n\x0e\x41\x63tionMetadata\x12\x16\n\x06parent\x18\x03 \x01(\tR\x06parent\x12\x14\n\x05group\x18\x05 \x01(\tR\x05group\x12\x42\n\x0b\x65xecuted_by\x18\x06 \x01(\x0b\x32!.cloudidl.common.EnrichedIdentityR\nexecutedBy\x12\x45\n\x04task\x18\x07 \x01(\x0b\x32%.cloudidl.workflow.TaskActionMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x04task\x12H\n\x05trace\x18\x08 \x01(\x0b\x32&.cloudidl.workflow.TraceActionMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x05trace\x12T\n\tcondition\x18\t \x01(\x0b\x32*.cloudidl.workflow.ConditionActionMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tconditionB\x06\n\x04spec\"\xe7\x01\n\x0c\x41\x63tionStatus\x12.\n\x05phase\x18\x01 \x01(\x0e\x32\x18.cloudidl.workflow.PhaseR\x05phase\x12\x39\n\nstart_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12:\n\x08\x65nd_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x07\x65ndTime\x88\x01\x01\x12#\n\x08\x61ttempts\x18\x04 \x01(\rB\x07\xfa\x42\x04*\x02 \x00R\x08\x61ttemptsB\x0b\n\t_end_time\"\xb3\x01\n\x06\x41\x63tion\x12\x31\n\x02id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierR\x02id\x12=\n\x08metadata\x18\x02 \x01(\x0b\x32!.cloudidl.workflow.ActionMetadataR\x08metadata\x12\x37\n\x06status\x18\x03 \x01(\x0b\x32\x1f.cloudidl.workflow.ActionStatusR\x06status\"\x9e\x02\n\x0e\x45nrichedAction\x12\x31\n\x06\x61\x63tion\x18\x01 \x01(\x0b\x32\x19.cloudidl.workflow.ActionR\x06\x61\x63tion\x12!\n\x0cmeets_filter\x18\x02 \x01(\x08R\x0bmeetsFilter\x12n\n\x15\x63hildren_phase_counts\x18\x03 \x03(\x0b\x32:.cloudidl.workflow.EnrichedAction.ChildrenPhaseCountsEntryR\x13\x63hildrenPhaseCounts\x1a\x46\n\x18\x43hildrenPhaseCountsEntry\x12\x10\n\x03key\x18\x01 \x01(\x05R\x03key\x12\x14\n\x05value\x18\x02 \x01(\x05R\x05value:\x02\x38\x01\"\x9a\x01\n\tErrorInfo\x12\x18\n\x07message\x18\x01 \x01(\tR\x07message\x12\x35\n\x04kind\x18\x02 \x01(\x0e\x32!.cloudidl.workflow.ErrorInfo.KindR\x04kind\"<\n\x04Kind\x12\x14\n\x10KIND_UNSPECIFIED\x10\x00\x12\r\n\tKIND_USER\x10\x01\x12\x0f\n\x0bKIND_SYSTEM\x10\x02\"e\n\tAbortInfo\x12\x16\n\x06reason\x18\x01 \x01(\tR\x06reason\x12@\n\naborted_by\x18\x02 \x01(\x0b\x32!.cloudidl.common.EnrichedIdentityR\tabortedBy\"\xcb\x03\n\rActionDetails\x12\x31\n\x02id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierR\x02id\x12=\n\x08metadata\x18\x02 \x01(\x0b\x32!.cloudidl.workflow.ActionMetadataR\x08metadata\x12\x37\n\x06status\x18\x03 \x01(\x0b\x32\x1f.cloudidl.workflow.ActionStatusR\x06status\x12=\n\nerror_info\x18\x04 \x01(\x0b\x32\x1c.cloudidl.workflow.ErrorInfoH\x00R\terrorInfo\x12=\n\nabort_info\x18\x05 \x01(\x0b\x32\x1c.cloudidl.workflow.AbortInfoH\x00R\tabortInfo\x12I\n\x12resolved_task_spec\x18\x06 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecR\x10resolvedTaskSpec\x12<\n\x08\x61ttempts\x18\x07 \x03(\x0b\x32 .cloudidl.workflow.ActionAttemptR\x08\x61ttemptsB\x08\n\x06result\"\xc9\x05\n\rActionAttempt\x12.\n\x05phase\x18\x01 \x01(\x0e\x32\x18.cloudidl.workflow.PhaseR\x05phase\x12\x39\n\nstart_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12:\n\x08\x65nd_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x07\x65ndTime\x88\x01\x01\x12@\n\nerror_info\x18\x04 \x01(\x0b\x32\x1c.cloudidl.workflow.ErrorInfoH\x01R\terrorInfo\x88\x01\x01\x12!\n\x07\x61ttempt\x18\x05 \x01(\rB\x07\xfa\x42\x04*\x02 \x00R\x07\x61ttempt\x12\x31\n\x08log_info\x18\x06 \x03(\x0b\x32\x16.flyteidl.core.TaskLogR\x07logInfo\x12=\n\x07outputs\x18\x07 \x01(\x0b\x32#.cloudidl.workflow.OutputReferencesR\x07outputs\x12%\n\x0elogs_available\x18\x08 \x01(\x08R\rlogsAvailable\x12\x44\n\x0c\x63\x61\x63he_status\x18\t \x01(\x0e\x32!.flyteidl.core.CatalogCacheStatusR\x0b\x63\x61\x63heStatus\x12\x46\n\x0e\x63luster_events\x18\n \x03(\x0b\x32\x1f.cloudidl.workflow.ClusterEventR\rclusterEvents\x12O\n\x11phase_transitions\x18\x0b \x03(\x0b\x32\".cloudidl.workflow.PhaseTransitionR\x10phaseTransitions\x12\x18\n\x07\x63luster\x18\x0c \x01(\tR\x07\x63lusterB\x0b\n\t_end_timeB\r\n\x0b_error_info\"e\n\x0c\x43lusterEvent\x12;\n\x0boccurred_at\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\noccurredAt\x12\x18\n\x07message\x18\x02 \x01(\tR\x07message\"\xc5\x01\n\x0fPhaseTransition\x12.\n\x05phase\x18\x01 \x01(\x0e\x32\x18.cloudidl.workflow.PhaseR\x05phase\x12\x39\n\nstart_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12:\n\x08\x65nd_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x07\x65ndTime\x88\x01\x01\x42\x0b\n\t_end_time\"\xa9\x06\n\x0b\x41\x63tionEvent\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\x12.\n\x05phase\x18\x03 \x01(\x0e\x32\x18.cloudidl.workflow.PhaseR\x05phase\x12\x18\n\x07version\x18\x04 \x01(\rR\x07version\x12=\n\nstart_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x02\x18\x01R\tstartTime\x12=\n\x0cupdated_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x0bupdatedTime\x12>\n\x08\x65nd_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x02\x18\x01H\x00R\x07\x65ndTime\x88\x01\x01\x12@\n\nerror_info\x18\x08 \x01(\x0b\x32\x1c.cloudidl.workflow.ErrorInfoH\x01R\terrorInfo\x88\x01\x01\x12\x31\n\x08log_info\x18\t \x03(\x0b\x32\x16.flyteidl.core.TaskLogR\x07logInfo\x12:\n\x0blog_context\x18\n \x01(\x0b\x32\x19.flyteidl.core.LogContextR\nlogContext\x12\x18\n\x07\x63luster\x18\x0b \x01(\tR\x07\x63luster\x12=\n\x07outputs\x18\x0c \x01(\x0b\x32#.cloudidl.workflow.OutputReferencesR\x07outputs\x12\x44\n\x0c\x63\x61\x63he_status\x18\r \x01(\x0e\x32!.flyteidl.core.CatalogCacheStatusR\x0b\x63\x61\x63heStatus\x12\x46\n\x0e\x63luster_events\x18\x0e \x03(\x0b\x32\x1f.cloudidl.workflow.ClusterEventR\rclusterEventsB\x0b\n\t_end_timeB\r\n\x0b_error_info\"P\n\x0cNamedLiteral\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.flyteidl.core.LiteralR\x05value\"P\n\x10OutputReferences\x12\x1d\n\noutput_uri\x18\x01 \x01(\tR\toutputUri\x12\x1d\n\nreport_uri\x18\x02 \x01(\tR\treportUri\"E\n\x06Inputs\x12;\n\x08literals\x18\x01 \x03(\x0b\x32\x1f.cloudidl.workflow.NamedLiteralR\x08literals\"F\n\x07Outputs\x12;\n\x08literals\x18\x01 \x03(\x0b\x32\x1f.cloudidl.workflow.NamedLiteralR\x08literals*\xcb\x01\n\x05Phase\x12\x15\n\x11PHASE_UNSPECIFIED\x10\x00\x12\x10\n\x0cPHASE_QUEUED\x10\x01\x12\x1f\n\x1bPHASE_WAITING_FOR_RESOURCES\x10\x02\x12\x16\n\x12PHASE_INITIALIZING\x10\x03\x12\x11\n\rPHASE_RUNNING\x10\x04\x12\x13\n\x0fPHASE_SUCCEEDED\x10\x05\x12\x10\n\x0cPHASE_FAILED\x10\x06\x12\x11\n\rPHASE_ABORTED\x10\x07\x12\x13\n\x0fPHASE_TIMED_OUT\x10\x08\x42\xbf\x01\n\x15\x63om.cloudidl.workflowB\x12RunDefinitionProtoH\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')
25
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1dworkflow/run_definition.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x15\x63ommon/identity.proto\x1a\x1b\x66lyteidl/core/catalog.proto\x1a\x1d\x66lyteidl/core/execution.proto\x1a\x1c\x66lyteidl/core/literals.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\x1a\x17validate/validate.proto\x1a\x1eworkflow/task_definition.proto\"\x82\x01\n\x06Labels\x12=\n\x06values\x18\x01 \x03(\x0b\x32%.cloudidl.workflow.Labels.ValuesEntryR\x06values\x1a\x39\n\x0bValuesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\"\x8c\x01\n\x0b\x41nnotations\x12\x42\n\x06values\x18\x01 \x03(\x0b\x32*.cloudidl.workflow.Annotations.ValuesEntryR\x06values\x1a\x39\n\x0bValuesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\";\n\x04\x45nvs\x12\x33\n\x06values\x18\x01 \x03(\x0b\x32\x1b.flyteidl.core.KeyValuePairR\x06values\"\xb0\x02\n\x07RunSpec\x12\x31\n\x06labels\x18\x01 \x01(\x0b\x32\x19.cloudidl.workflow.LabelsR\x06labels\x12@\n\x0b\x61nnotations\x18\x02 \x01(\x0b\x32\x1e.cloudidl.workflow.AnnotationsR\x0b\x61nnotations\x12+\n\x04\x65nvs\x18\x03 \x01(\x0b\x32\x17.cloudidl.workflow.EnvsR\x04\x65nvs\x12@\n\rinterruptible\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.BoolValueR\rinterruptible\x12\'\n\x0foverwrite_cache\x18\x05 \x01(\x08R\x0eoverwriteCache\x12\x18\n\x07\x63luster\x18\x06 \x01(\tR\x07\x63luster\"8\n\x03Run\x12\x31\n\x06\x61\x63tion\x18\x01 \x01(\x0b\x32\x19.cloudidl.workflow.ActionR\x06\x61\x63tion\"}\n\nRunDetails\x12\x35\n\x08run_spec\x18\x01 \x01(\x0b\x32\x1a.cloudidl.workflow.RunSpecR\x07runSpec\x12\x38\n\x06\x61\x63tion\x18\x02 \x01(\x0b\x32 .cloudidl.workflow.ActionDetailsR\x06\x61\x63tion\"\x83\x01\n\x12TaskActionMetadata\x12\x31\n\x02id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierR\x02id\x12\x1b\n\ttask_type\x18\x02 \x01(\tR\x08taskType\x12\x1d\n\nshort_name\x18\x03 \x01(\tR\tshortName\")\n\x13TraceActionMetadata\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\"\x9f\x01\n\x17\x43onditionActionMetadata\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12 \n\x06run_id\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x05runId\x12&\n\taction_id\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x08\x61\x63tionId\x12\x18\n\x06global\x18\x04 \x01(\x08H\x00R\x06globalB\x0c\n\x05scope\x12\x03\xf8\x42\x01\"\xf1\x02\n\x0e\x41\x63tionMetadata\x12\x16\n\x06parent\x18\x03 \x01(\tR\x06parent\x12\x14\n\x05group\x18\x05 \x01(\tR\x05group\x12\x42\n\x0b\x65xecuted_by\x18\x06 \x01(\x0b\x32!.cloudidl.common.EnrichedIdentityR\nexecutedBy\x12\x45\n\x04task\x18\x07 \x01(\x0b\x32%.cloudidl.workflow.TaskActionMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x04task\x12H\n\x05trace\x18\x08 \x01(\x0b\x32&.cloudidl.workflow.TraceActionMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x05trace\x12T\n\tcondition\x18\t \x01(\x0b\x32*.cloudidl.workflow.ConditionActionMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tconditionB\x06\n\x04spec\"\xad\x02\n\x0c\x41\x63tionStatus\x12.\n\x05phase\x18\x01 \x01(\x0e\x32\x18.cloudidl.workflow.PhaseR\x05phase\x12\x39\n\nstart_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12:\n\x08\x65nd_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x07\x65ndTime\x88\x01\x01\x12#\n\x08\x61ttempts\x18\x04 \x01(\rB\x07\xfa\x42\x04*\x02 \x00R\x08\x61ttempts\x12\x44\n\x0c\x63\x61\x63he_status\x18\x05 \x01(\x0e\x32!.flyteidl.core.CatalogCacheStatusR\x0b\x63\x61\x63heStatusB\x0b\n\t_end_time\"\xb3\x01\n\x06\x41\x63tion\x12\x31\n\x02id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierR\x02id\x12=\n\x08metadata\x18\x02 \x01(\x0b\x32!.cloudidl.workflow.ActionMetadataR\x08metadata\x12\x37\n\x06status\x18\x03 \x01(\x0b\x32\x1f.cloudidl.workflow.ActionStatusR\x06status\"\x9e\x02\n\x0e\x45nrichedAction\x12\x31\n\x06\x61\x63tion\x18\x01 \x01(\x0b\x32\x19.cloudidl.workflow.ActionR\x06\x61\x63tion\x12!\n\x0cmeets_filter\x18\x02 \x01(\x08R\x0bmeetsFilter\x12n\n\x15\x63hildren_phase_counts\x18\x03 \x03(\x0b\x32:.cloudidl.workflow.EnrichedAction.ChildrenPhaseCountsEntryR\x13\x63hildrenPhaseCounts\x1a\x46\n\x18\x43hildrenPhaseCountsEntry\x12\x10\n\x03key\x18\x01 \x01(\x05R\x03key\x12\x14\n\x05value\x18\x02 \x01(\x05R\x05value:\x02\x38\x01\"\x9a\x01\n\tErrorInfo\x12\x18\n\x07message\x18\x01 \x01(\tR\x07message\x12\x35\n\x04kind\x18\x02 \x01(\x0e\x32!.cloudidl.workflow.ErrorInfo.KindR\x04kind\"<\n\x04Kind\x12\x14\n\x10KIND_UNSPECIFIED\x10\x00\x12\r\n\tKIND_USER\x10\x01\x12\x0f\n\x0bKIND_SYSTEM\x10\x02\"e\n\tAbortInfo\x12\x16\n\x06reason\x18\x01 \x01(\tR\x06reason\x12@\n\naborted_by\x18\x02 \x01(\x0b\x32!.cloudidl.common.EnrichedIdentityR\tabortedBy\"\xcb\x03\n\rActionDetails\x12\x31\n\x02id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierR\x02id\x12=\n\x08metadata\x18\x02 \x01(\x0b\x32!.cloudidl.workflow.ActionMetadataR\x08metadata\x12\x37\n\x06status\x18\x03 \x01(\x0b\x32\x1f.cloudidl.workflow.ActionStatusR\x06status\x12=\n\nerror_info\x18\x04 \x01(\x0b\x32\x1c.cloudidl.workflow.ErrorInfoH\x00R\terrorInfo\x12=\n\nabort_info\x18\x05 \x01(\x0b\x32\x1c.cloudidl.workflow.AbortInfoH\x00R\tabortInfo\x12I\n\x12resolved_task_spec\x18\x06 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecR\x10resolvedTaskSpec\x12<\n\x08\x61ttempts\x18\x07 \x03(\x0b\x32 .cloudidl.workflow.ActionAttemptR\x08\x61ttemptsB\x08\n\x06result\"\x85\x06\n\rActionAttempt\x12.\n\x05phase\x18\x01 \x01(\x0e\x32\x18.cloudidl.workflow.PhaseR\x05phase\x12\x39\n\nstart_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12:\n\x08\x65nd_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x07\x65ndTime\x88\x01\x01\x12@\n\nerror_info\x18\x04 \x01(\x0b\x32\x1c.cloudidl.workflow.ErrorInfoH\x01R\terrorInfo\x88\x01\x01\x12!\n\x07\x61ttempt\x18\x05 \x01(\rB\x07\xfa\x42\x04*\x02 \x00R\x07\x61ttempt\x12\x31\n\x08log_info\x18\x06 \x03(\x0b\x32\x16.flyteidl.core.TaskLogR\x07logInfo\x12=\n\x07outputs\x18\x07 \x01(\x0b\x32#.cloudidl.workflow.OutputReferencesR\x07outputs\x12%\n\x0elogs_available\x18\x08 \x01(\x08R\rlogsAvailable\x12\x44\n\x0c\x63\x61\x63he_status\x18\t \x01(\x0e\x32!.flyteidl.core.CatalogCacheStatusR\x0b\x63\x61\x63heStatus\x12\x46\n\x0e\x63luster_events\x18\n \x03(\x0b\x32\x1f.cloudidl.workflow.ClusterEventR\rclusterEvents\x12O\n\x11phase_transitions\x18\x0b \x03(\x0b\x32\".cloudidl.workflow.PhaseTransitionR\x10phaseTransitions\x12\x18\n\x07\x63luster\x18\x0c \x01(\tR\x07\x63luster\x12:\n\x0blog_context\x18\r \x01(\x0b\x32\x19.flyteidl.core.LogContextR\nlogContextB\x0b\n\t_end_timeB\r\n\x0b_error_info\"e\n\x0c\x43lusterEvent\x12;\n\x0boccurred_at\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\noccurredAt\x12\x18\n\x07message\x18\x02 \x01(\tR\x07message\"\xc5\x01\n\x0fPhaseTransition\x12.\n\x05phase\x18\x01 \x01(\x0e\x32\x18.cloudidl.workflow.PhaseR\x05phase\x12\x39\n\nstart_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12:\n\x08\x65nd_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x07\x65ndTime\x88\x01\x01\x42\x0b\n\t_end_time\"\xea\x06\n\x0b\x41\x63tionEvent\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\x12.\n\x05phase\x18\x03 \x01(\x0e\x32\x18.cloudidl.workflow.PhaseR\x05phase\x12\x18\n\x07version\x18\x04 \x01(\rR\x07version\x12=\n\nstart_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x02\x18\x01R\tstartTime\x12=\n\x0cupdated_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x0bupdatedTime\x12>\n\x08\x65nd_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x02\x18\x01H\x00R\x07\x65ndTime\x88\x01\x01\x12@\n\nerror_info\x18\x08 \x01(\x0b\x32\x1c.cloudidl.workflow.ErrorInfoH\x01R\terrorInfo\x88\x01\x01\x12\x31\n\x08log_info\x18\t \x03(\x0b\x32\x16.flyteidl.core.TaskLogR\x07logInfo\x12:\n\x0blog_context\x18\n \x01(\x0b\x32\x19.flyteidl.core.LogContextR\nlogContext\x12\x18\n\x07\x63luster\x18\x0b \x01(\tR\x07\x63luster\x12=\n\x07outputs\x18\x0c \x01(\x0b\x32#.cloudidl.workflow.OutputReferencesR\x07outputs\x12\x44\n\x0c\x63\x61\x63he_status\x18\r \x01(\x0e\x32!.flyteidl.core.CatalogCacheStatusR\x0b\x63\x61\x63heStatus\x12\x46\n\x0e\x63luster_events\x18\x0e \x03(\x0b\x32\x1f.cloudidl.workflow.ClusterEventR\rclusterEvents\x12?\n\rreported_time\x18\x0f \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x0creportedTimeB\x0b\n\t_end_timeB\r\n\x0b_error_info\"P\n\x0cNamedLiteral\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.flyteidl.core.LiteralR\x05value\"P\n\x10OutputReferences\x12\x1d\n\noutput_uri\x18\x01 \x01(\tR\toutputUri\x12\x1d\n\nreport_uri\x18\x02 \x01(\tR\treportUri\"|\n\x06Inputs\x12;\n\x08literals\x18\x01 \x03(\x0b\x32\x1f.cloudidl.workflow.NamedLiteralR\x08literals\x12\x35\n\x07\x63ontext\x18\x02 \x03(\x0b\x32\x1b.flyteidl.core.KeyValuePairR\x07\x63ontext\"F\n\x07Outputs\x12;\n\x08literals\x18\x01 \x03(\x0b\x32\x1f.cloudidl.workflow.NamedLiteralR\x08literals*\xcb\x01\n\x05Phase\x12\x15\n\x11PHASE_UNSPECIFIED\x10\x00\x12\x10\n\x0cPHASE_QUEUED\x10\x01\x12\x1f\n\x1bPHASE_WAITING_FOR_RESOURCES\x10\x02\x12\x16\n\x12PHASE_INITIALIZING\x10\x03\x12\x11\n\rPHASE_RUNNING\x10\x04\x12\x13\n\x0fPHASE_SUCCEEDED\x10\x05\x12\x10\n\x0cPHASE_FAILED\x10\x06\x12\x11\n\rPHASE_ABORTED\x10\x07\x12\x13\n\x0fPHASE_TIMED_OUT\x10\x08\x42\xbf\x01\n\x15\x63om.cloudidl.workflowB\x12RunDefinitionProtoH\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')
26
26
 
27
27
  _globals = globals()
28
28
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -60,8 +60,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
60
60
  _ACTIONEVENT.fields_by_name['start_time']._serialized_options = b'\030\001'
61
61
  _ACTIONEVENT.fields_by_name['end_time']._options = None
62
62
  _ACTIONEVENT.fields_by_name['end_time']._serialized_options = b'\030\001'
63
- _globals['_PHASE']._serialized_start=5392
64
- _globals['_PHASE']._serialized_end=5595
63
+ _globals['_PHASE']._serialized_start=5668
64
+ _globals['_PHASE']._serialized_end=5871
65
65
  _globals['_LABELS']._serialized_start=313
66
66
  _globals['_LABELS']._serialized_end=443
67
67
  _globals['_LABELS_VALUESENTRY']._serialized_start=386
@@ -73,49 +73,49 @@ if _descriptor._USE_C_DESCRIPTORS == False:
73
73
  _globals['_ENVS']._serialized_start=588
74
74
  _globals['_ENVS']._serialized_end=647
75
75
  _globals['_RUNSPEC']._serialized_start=650
76
- _globals['_RUNSPEC']._serialized_end=928
77
- _globals['_RUN']._serialized_start=930
78
- _globals['_RUN']._serialized_end=986
79
- _globals['_RUNDETAILS']._serialized_start=988
80
- _globals['_RUNDETAILS']._serialized_end=1113
81
- _globals['_TASKACTIONMETADATA']._serialized_start=1116
82
- _globals['_TASKACTIONMETADATA']._serialized_end=1247
83
- _globals['_TRACEACTIONMETADATA']._serialized_start=1249
84
- _globals['_TRACEACTIONMETADATA']._serialized_end=1290
85
- _globals['_CONDITIONACTIONMETADATA']._serialized_start=1293
86
- _globals['_CONDITIONACTIONMETADATA']._serialized_end=1452
87
- _globals['_ACTIONMETADATA']._serialized_start=1455
88
- _globals['_ACTIONMETADATA']._serialized_end=1824
89
- _globals['_ACTIONSTATUS']._serialized_start=1827
90
- _globals['_ACTIONSTATUS']._serialized_end=2058
91
- _globals['_ACTION']._serialized_start=2061
92
- _globals['_ACTION']._serialized_end=2240
93
- _globals['_ENRICHEDACTION']._serialized_start=2243
94
- _globals['_ENRICHEDACTION']._serialized_end=2529
95
- _globals['_ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY']._serialized_start=2459
96
- _globals['_ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY']._serialized_end=2529
97
- _globals['_ERRORINFO']._serialized_start=2532
98
- _globals['_ERRORINFO']._serialized_end=2686
99
- _globals['_ERRORINFO_KIND']._serialized_start=2626
100
- _globals['_ERRORINFO_KIND']._serialized_end=2686
101
- _globals['_ABORTINFO']._serialized_start=2688
102
- _globals['_ABORTINFO']._serialized_end=2789
103
- _globals['_ACTIONDETAILS']._serialized_start=2792
104
- _globals['_ACTIONDETAILS']._serialized_end=3251
105
- _globals['_ACTIONATTEMPT']._serialized_start=3254
106
- _globals['_ACTIONATTEMPT']._serialized_end=3967
107
- _globals['_CLUSTEREVENT']._serialized_start=3969
108
- _globals['_CLUSTEREVENT']._serialized_end=4070
109
- _globals['_PHASETRANSITION']._serialized_start=4073
110
- _globals['_PHASETRANSITION']._serialized_end=4270
111
- _globals['_ACTIONEVENT']._serialized_start=4273
112
- _globals['_ACTIONEVENT']._serialized_end=5082
113
- _globals['_NAMEDLITERAL']._serialized_start=5084
114
- _globals['_NAMEDLITERAL']._serialized_end=5164
115
- _globals['_OUTPUTREFERENCES']._serialized_start=5166
116
- _globals['_OUTPUTREFERENCES']._serialized_end=5246
117
- _globals['_INPUTS']._serialized_start=5248
118
- _globals['_INPUTS']._serialized_end=5317
119
- _globals['_OUTPUTS']._serialized_start=5319
120
- _globals['_OUTPUTS']._serialized_end=5389
76
+ _globals['_RUNSPEC']._serialized_end=954
77
+ _globals['_RUN']._serialized_start=956
78
+ _globals['_RUN']._serialized_end=1012
79
+ _globals['_RUNDETAILS']._serialized_start=1014
80
+ _globals['_RUNDETAILS']._serialized_end=1139
81
+ _globals['_TASKACTIONMETADATA']._serialized_start=1142
82
+ _globals['_TASKACTIONMETADATA']._serialized_end=1273
83
+ _globals['_TRACEACTIONMETADATA']._serialized_start=1275
84
+ _globals['_TRACEACTIONMETADATA']._serialized_end=1316
85
+ _globals['_CONDITIONACTIONMETADATA']._serialized_start=1319
86
+ _globals['_CONDITIONACTIONMETADATA']._serialized_end=1478
87
+ _globals['_ACTIONMETADATA']._serialized_start=1481
88
+ _globals['_ACTIONMETADATA']._serialized_end=1850
89
+ _globals['_ACTIONSTATUS']._serialized_start=1853
90
+ _globals['_ACTIONSTATUS']._serialized_end=2154
91
+ _globals['_ACTION']._serialized_start=2157
92
+ _globals['_ACTION']._serialized_end=2336
93
+ _globals['_ENRICHEDACTION']._serialized_start=2339
94
+ _globals['_ENRICHEDACTION']._serialized_end=2625
95
+ _globals['_ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY']._serialized_start=2555
96
+ _globals['_ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY']._serialized_end=2625
97
+ _globals['_ERRORINFO']._serialized_start=2628
98
+ _globals['_ERRORINFO']._serialized_end=2782
99
+ _globals['_ERRORINFO_KIND']._serialized_start=2722
100
+ _globals['_ERRORINFO_KIND']._serialized_end=2782
101
+ _globals['_ABORTINFO']._serialized_start=2784
102
+ _globals['_ABORTINFO']._serialized_end=2885
103
+ _globals['_ACTIONDETAILS']._serialized_start=2888
104
+ _globals['_ACTIONDETAILS']._serialized_end=3347
105
+ _globals['_ACTIONATTEMPT']._serialized_start=3350
106
+ _globals['_ACTIONATTEMPT']._serialized_end=4123
107
+ _globals['_CLUSTEREVENT']._serialized_start=4125
108
+ _globals['_CLUSTEREVENT']._serialized_end=4226
109
+ _globals['_PHASETRANSITION']._serialized_start=4229
110
+ _globals['_PHASETRANSITION']._serialized_end=4426
111
+ _globals['_ACTIONEVENT']._serialized_start=4429
112
+ _globals['_ACTIONEVENT']._serialized_end=5303
113
+ _globals['_NAMEDLITERAL']._serialized_start=5305
114
+ _globals['_NAMEDLITERAL']._serialized_end=5385
115
+ _globals['_OUTPUTREFERENCES']._serialized_start=5387
116
+ _globals['_OUTPUTREFERENCES']._serialized_end=5467
117
+ _globals['_INPUTS']._serialized_start=5469
118
+ _globals['_INPUTS']._serialized_end=5593
119
+ _globals['_OUTPUTS']._serialized_start=5595
120
+ _globals['_OUTPUTS']._serialized_end=5665
121
121
  # @@protoc_insertion_point(module_scope)