flyte 2.0.0b23__py3-none-any.whl → 2.0.0b25__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 (162) hide show
  1. flyte/__init__.py +11 -2
  2. flyte/_cache/local_cache.py +4 -3
  3. flyte/_code_bundle/_utils.py +3 -3
  4. flyte/_code_bundle/bundle.py +12 -5
  5. flyte/_context.py +4 -1
  6. flyte/_custom_context.py +73 -0
  7. flyte/_deploy.py +31 -7
  8. flyte/_image.py +48 -16
  9. flyte/_initialize.py +69 -26
  10. flyte/_internal/controllers/_local_controller.py +1 -0
  11. flyte/_internal/controllers/_trace.py +1 -1
  12. flyte/_internal/controllers/remote/_action.py +9 -10
  13. flyte/_internal/controllers/remote/_client.py +1 -1
  14. flyte/_internal/controllers/remote/_controller.py +4 -2
  15. flyte/_internal/controllers/remote/_core.py +10 -13
  16. flyte/_internal/controllers/remote/_informer.py +3 -3
  17. flyte/_internal/controllers/remote/_service_protocol.py +7 -7
  18. flyte/_internal/imagebuild/docker_builder.py +45 -59
  19. flyte/_internal/imagebuild/remote_builder.py +51 -11
  20. flyte/_internal/imagebuild/utils.py +51 -3
  21. flyte/_internal/runtime/convert.py +39 -18
  22. flyte/_internal/runtime/io.py +8 -7
  23. flyte/_internal/runtime/resources_serde.py +20 -6
  24. flyte/_internal/runtime/reuse.py +1 -1
  25. flyte/_internal/runtime/task_serde.py +7 -10
  26. flyte/_internal/runtime/taskrunner.py +10 -1
  27. flyte/_internal/runtime/trigger_serde.py +13 -13
  28. flyte/_internal/runtime/types_serde.py +1 -1
  29. flyte/_keyring/file.py +2 -2
  30. flyte/_map.py +65 -13
  31. flyte/_pod.py +2 -2
  32. flyte/_resources.py +175 -31
  33. flyte/_run.py +37 -21
  34. flyte/_task.py +27 -6
  35. flyte/_task_environment.py +37 -10
  36. flyte/_utils/module_loader.py +2 -2
  37. flyte/_version.py +3 -3
  38. flyte/cli/_common.py +47 -5
  39. flyte/cli/_create.py +4 -0
  40. flyte/cli/_deploy.py +8 -0
  41. flyte/cli/_get.py +4 -0
  42. flyte/cli/_params.py +4 -4
  43. flyte/cli/_run.py +50 -7
  44. flyte/cli/_update.py +4 -3
  45. flyte/config/_config.py +2 -0
  46. flyte/config/_internal.py +1 -0
  47. flyte/config/_reader.py +3 -3
  48. flyte/errors.py +1 -1
  49. flyte/extend.py +4 -0
  50. flyte/extras/_container.py +6 -1
  51. flyte/git/_config.py +11 -9
  52. flyte/io/_dataframe/basic_dfs.py +1 -1
  53. flyte/io/_dataframe/dataframe.py +12 -8
  54. flyte/io/_dir.py +48 -15
  55. flyte/io/_file.py +48 -11
  56. flyte/models.py +12 -8
  57. flyte/remote/_action.py +18 -16
  58. flyte/remote/_client/_protocols.py +4 -3
  59. flyte/remote/_client/auth/_channel.py +1 -1
  60. flyte/remote/_client/controlplane.py +4 -8
  61. flyte/remote/_data.py +4 -3
  62. flyte/remote/_logs.py +3 -3
  63. flyte/remote/_run.py +5 -5
  64. flyte/remote/_secret.py +20 -13
  65. flyte/remote/_task.py +7 -8
  66. flyte/remote/_trigger.py +25 -27
  67. flyte/storage/_parallel_reader.py +274 -0
  68. flyte/storage/_storage.py +66 -2
  69. flyte/types/_interface.py +2 -2
  70. flyte/types/_pickle.py +1 -1
  71. flyte/types/_string_literals.py +8 -9
  72. flyte/types/_type_engine.py +25 -17
  73. flyte/types/_utils.py +1 -1
  74. {flyte-2.0.0b23.dist-info → flyte-2.0.0b25.dist-info}/METADATA +2 -1
  75. flyte-2.0.0b25.dist-info/RECORD +184 -0
  76. flyte/_protos/__init__.py +0 -0
  77. flyte/_protos/common/authorization_pb2.py +0 -66
  78. flyte/_protos/common/authorization_pb2.pyi +0 -108
  79. flyte/_protos/common/authorization_pb2_grpc.py +0 -4
  80. flyte/_protos/common/identifier_pb2.py +0 -117
  81. flyte/_protos/common/identifier_pb2.pyi +0 -142
  82. flyte/_protos/common/identifier_pb2_grpc.py +0 -4
  83. flyte/_protos/common/identity_pb2.py +0 -48
  84. flyte/_protos/common/identity_pb2.pyi +0 -72
  85. flyte/_protos/common/identity_pb2_grpc.py +0 -4
  86. flyte/_protos/common/list_pb2.py +0 -36
  87. flyte/_protos/common/list_pb2.pyi +0 -71
  88. flyte/_protos/common/list_pb2_grpc.py +0 -4
  89. flyte/_protos/common/policy_pb2.py +0 -37
  90. flyte/_protos/common/policy_pb2.pyi +0 -27
  91. flyte/_protos/common/policy_pb2_grpc.py +0 -4
  92. flyte/_protos/common/role_pb2.py +0 -37
  93. flyte/_protos/common/role_pb2.pyi +0 -53
  94. flyte/_protos/common/role_pb2_grpc.py +0 -4
  95. flyte/_protos/common/runtime_version_pb2.py +0 -28
  96. flyte/_protos/common/runtime_version_pb2.pyi +0 -24
  97. flyte/_protos/common/runtime_version_pb2_grpc.py +0 -4
  98. flyte/_protos/imagebuilder/definition_pb2.py +0 -60
  99. flyte/_protos/imagebuilder/definition_pb2.pyi +0 -153
  100. flyte/_protos/imagebuilder/definition_pb2_grpc.py +0 -4
  101. flyte/_protos/imagebuilder/payload_pb2.py +0 -32
  102. flyte/_protos/imagebuilder/payload_pb2.pyi +0 -21
  103. flyte/_protos/imagebuilder/payload_pb2_grpc.py +0 -4
  104. flyte/_protos/imagebuilder/service_pb2.py +0 -29
  105. flyte/_protos/imagebuilder/service_pb2.pyi +0 -5
  106. flyte/_protos/imagebuilder/service_pb2_grpc.py +0 -66
  107. flyte/_protos/logs/dataplane/payload_pb2.py +0 -100
  108. flyte/_protos/logs/dataplane/payload_pb2.pyi +0 -177
  109. flyte/_protos/logs/dataplane/payload_pb2_grpc.py +0 -4
  110. flyte/_protos/secret/definition_pb2.py +0 -49
  111. flyte/_protos/secret/definition_pb2.pyi +0 -93
  112. flyte/_protos/secret/definition_pb2_grpc.py +0 -4
  113. flyte/_protos/secret/payload_pb2.py +0 -62
  114. flyte/_protos/secret/payload_pb2.pyi +0 -94
  115. flyte/_protos/secret/payload_pb2_grpc.py +0 -4
  116. flyte/_protos/secret/secret_pb2.py +0 -38
  117. flyte/_protos/secret/secret_pb2.pyi +0 -6
  118. flyte/_protos/secret/secret_pb2_grpc.py +0 -198
  119. flyte/_protos/validate/validate/validate_pb2.py +0 -76
  120. flyte/_protos/workflow/common_pb2.py +0 -38
  121. flyte/_protos/workflow/common_pb2.pyi +0 -63
  122. flyte/_protos/workflow/common_pb2_grpc.py +0 -4
  123. flyte/_protos/workflow/environment_pb2.py +0 -29
  124. flyte/_protos/workflow/environment_pb2.pyi +0 -12
  125. flyte/_protos/workflow/environment_pb2_grpc.py +0 -4
  126. flyte/_protos/workflow/node_execution_service_pb2.py +0 -26
  127. flyte/_protos/workflow/node_execution_service_pb2.pyi +0 -4
  128. flyte/_protos/workflow/node_execution_service_pb2_grpc.py +0 -32
  129. flyte/_protos/workflow/queue_service_pb2.py +0 -117
  130. flyte/_protos/workflow/queue_service_pb2.pyi +0 -182
  131. flyte/_protos/workflow/queue_service_pb2_grpc.py +0 -206
  132. flyte/_protos/workflow/run_definition_pb2.py +0 -123
  133. flyte/_protos/workflow/run_definition_pb2.pyi +0 -354
  134. flyte/_protos/workflow/run_definition_pb2_grpc.py +0 -4
  135. flyte/_protos/workflow/run_logs_service_pb2.py +0 -41
  136. flyte/_protos/workflow/run_logs_service_pb2.pyi +0 -28
  137. flyte/_protos/workflow/run_logs_service_pb2_grpc.py +0 -69
  138. flyte/_protos/workflow/run_service_pb2.py +0 -147
  139. flyte/_protos/workflow/run_service_pb2.pyi +0 -203
  140. flyte/_protos/workflow/run_service_pb2_grpc.py +0 -480
  141. flyte/_protos/workflow/state_service_pb2.py +0 -67
  142. flyte/_protos/workflow/state_service_pb2.pyi +0 -76
  143. flyte/_protos/workflow/state_service_pb2_grpc.py +0 -138
  144. flyte/_protos/workflow/task_definition_pb2.py +0 -86
  145. flyte/_protos/workflow/task_definition_pb2.pyi +0 -105
  146. flyte/_protos/workflow/task_definition_pb2_grpc.py +0 -4
  147. flyte/_protos/workflow/task_service_pb2.py +0 -61
  148. flyte/_protos/workflow/task_service_pb2.pyi +0 -62
  149. flyte/_protos/workflow/task_service_pb2_grpc.py +0 -138
  150. flyte/_protos/workflow/trigger_definition_pb2.py +0 -66
  151. flyte/_protos/workflow/trigger_definition_pb2.pyi +0 -117
  152. flyte/_protos/workflow/trigger_definition_pb2_grpc.py +0 -4
  153. flyte/_protos/workflow/trigger_service_pb2.py +0 -96
  154. flyte/_protos/workflow/trigger_service_pb2.pyi +0 -110
  155. flyte/_protos/workflow/trigger_service_pb2_grpc.py +0 -281
  156. flyte-2.0.0b23.dist-info/RECORD +0 -262
  157. {flyte-2.0.0b23.data → flyte-2.0.0b25.data}/scripts/debug.py +0 -0
  158. {flyte-2.0.0b23.data → flyte-2.0.0b25.data}/scripts/runtime.py +0 -0
  159. {flyte-2.0.0b23.dist-info → flyte-2.0.0b25.dist-info}/WHEEL +0 -0
  160. {flyte-2.0.0b23.dist-info → flyte-2.0.0b25.dist-info}/entry_points.txt +0 -0
  161. {flyte-2.0.0b23.dist-info → flyte-2.0.0b25.dist-info}/licenses/LICENSE +0 -0
  162. {flyte-2.0.0b23.dist-info → flyte-2.0.0b25.dist-info}/top_level.txt +0 -0
@@ -1,123 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- # Generated by the protocol buffer compiler. DO NOT EDIT!
3
- # source: workflow/run_definition.proto
4
- """Generated protocol buffer code."""
5
- from google.protobuf import descriptor as _descriptor
6
- from google.protobuf import descriptor_pool as _descriptor_pool
7
- from google.protobuf import symbol_database as _symbol_database
8
- from google.protobuf.internal import builder as _builder
9
- # @@protoc_insertion_point(imports)
10
-
11
- _sym_db = _symbol_database.Default()
12
-
13
-
14
- from flyte._protos.common import identifier_pb2 as common_dot_identifier__pb2
15
- from flyte._protos.common import identity_pb2 as common_dot_identity__pb2
16
- from flyteidl.core import catalog_pb2 as flyteidl_dot_core_dot_catalog__pb2
17
- from flyteidl.core import execution_pb2 as flyteidl_dot_core_dot_execution__pb2
18
- from flyteidl.core import literals_pb2 as flyteidl_dot_core_dot_literals__pb2
19
- from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
20
- from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
21
- from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
22
- from flyte._protos.workflow import task_definition_pb2 as workflow_dot_task__definition__pb2
23
-
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\"\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\"\xf4\x03\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\tcondition\x12>\n\x0b\x61\x63tion_type\x18\n \x01(\x0e\x32\x1d.cloudidl.workflow.ActionTypeR\nactionType\x12\x41\n\ntrigger_id\x18\x0b \x01(\x0b\x32\".cloudidl.common.TriggerIdentifierR\ttriggerIdB\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\"\xf1\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\x12\x31\n\x04task\x18\x06 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecH\x01R\x04task\x12\x34\n\x05trace\x18\x08 \x01(\x0b\x32\x1c.cloudidl.workflow.TraceSpecH\x01R\x05trace\x12<\n\x08\x61ttempts\x18\x07 \x03(\x0b\x32 .cloudidl.workflow.ActionAttemptR\x08\x61ttemptsB\x08\n\x06resultB\x06\n\x04spec\"\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*q\n\nActionType\x12\x1b\n\x17\x41\x43TION_TYPE_UNSPECIFIED\x10\x00\x12\x14\n\x10\x41\x43TION_TYPE_TASK\x10\x01\x12\x15\n\x11\x41\x43TION_TYPE_TRACE\x10\x02\x12\x19\n\x15\x41\x43TION_TYPE_CONDITION\x10\x03\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
-
27
- _globals = globals()
28
- _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
29
- _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'workflow.run_definition_pb2', _globals)
30
- if _descriptor._USE_C_DESCRIPTORS == False:
31
- DESCRIPTOR._options = None
32
- DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\022RunDefinitionProtoH\002P\001Z+github.com/unionai/cloud/gen/pb-go/workflow\242\002\003CWX\252\002\021Cloudidl.Workflow\312\002\021Cloudidl\\Workflow\342\002\035Cloudidl\\Workflow\\GPBMetadata\352\002\022Cloudidl::Workflow'
33
- _LABELS_VALUESENTRY._options = None
34
- _LABELS_VALUESENTRY._serialized_options = b'8\001'
35
- _ANNOTATIONS_VALUESENTRY._options = None
36
- _ANNOTATIONS_VALUESENTRY._serialized_options = b'8\001'
37
- _CONDITIONACTIONMETADATA.oneofs_by_name['scope']._options = None
38
- _CONDITIONACTIONMETADATA.oneofs_by_name['scope']._serialized_options = b'\370B\001'
39
- _CONDITIONACTIONMETADATA.fields_by_name['run_id']._options = None
40
- _CONDITIONACTIONMETADATA.fields_by_name['run_id']._serialized_options = b'\372B\004r\002\020\001'
41
- _CONDITIONACTIONMETADATA.fields_by_name['action_id']._options = None
42
- _CONDITIONACTIONMETADATA.fields_by_name['action_id']._serialized_options = b'\372B\004r\002\020\001'
43
- _ACTIONMETADATA.fields_by_name['task']._options = None
44
- _ACTIONMETADATA.fields_by_name['task']._serialized_options = b'\372B\005\212\001\002\020\001'
45
- _ACTIONMETADATA.fields_by_name['trace']._options = None
46
- _ACTIONMETADATA.fields_by_name['trace']._serialized_options = b'\372B\005\212\001\002\020\001'
47
- _ACTIONMETADATA.fields_by_name['condition']._options = None
48
- _ACTIONMETADATA.fields_by_name['condition']._serialized_options = b'\372B\005\212\001\002\020\001'
49
- _ACTIONSTATUS.fields_by_name['attempts']._options = None
50
- _ACTIONSTATUS.fields_by_name['attempts']._serialized_options = b'\372B\004*\002 \000'
51
- _ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY._options = None
52
- _ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY._serialized_options = b'8\001'
53
- _ACTIONATTEMPT.fields_by_name['attempt']._options = None
54
- _ACTIONATTEMPT.fields_by_name['attempt']._serialized_options = b'\372B\004*\002 \000'
55
- _ACTIONEVENT.fields_by_name['id']._options = None
56
- _ACTIONEVENT.fields_by_name['id']._serialized_options = b'\372B\005\212\001\002\020\001'
57
- _ACTIONEVENT.fields_by_name['attempt']._options = None
58
- _ACTIONEVENT.fields_by_name['attempt']._serialized_options = b'\372B\004*\002 \000'
59
- _ACTIONEVENT.fields_by_name['start_time']._options = None
60
- _ACTIONEVENT.fields_by_name['start_time']._serialized_options = b'\030\001'
61
- _ACTIONEVENT.fields_by_name['end_time']._options = None
62
- _ACTIONEVENT.fields_by_name['end_time']._serialized_options = b'\030\001'
63
- _globals['_PHASE']._serialized_start=5837
64
- _globals['_PHASE']._serialized_end=6040
65
- _globals['_ACTIONTYPE']._serialized_start=6042
66
- _globals['_ACTIONTYPE']._serialized_end=6155
67
- _globals['_LABELS']._serialized_start=313
68
- _globals['_LABELS']._serialized_end=443
69
- _globals['_LABELS_VALUESENTRY']._serialized_start=386
70
- _globals['_LABELS_VALUESENTRY']._serialized_end=443
71
- _globals['_ANNOTATIONS']._serialized_start=446
72
- _globals['_ANNOTATIONS']._serialized_end=586
73
- _globals['_ANNOTATIONS_VALUESENTRY']._serialized_start=386
74
- _globals['_ANNOTATIONS_VALUESENTRY']._serialized_end=443
75
- _globals['_ENVS']._serialized_start=588
76
- _globals['_ENVS']._serialized_end=647
77
- _globals['_RUNSPEC']._serialized_start=650
78
- _globals['_RUNSPEC']._serialized_end=954
79
- _globals['_RUN']._serialized_start=956
80
- _globals['_RUN']._serialized_end=1012
81
- _globals['_RUNDETAILS']._serialized_start=1014
82
- _globals['_RUNDETAILS']._serialized_end=1139
83
- _globals['_TASKACTIONMETADATA']._serialized_start=1142
84
- _globals['_TASKACTIONMETADATA']._serialized_end=1273
85
- _globals['_TRACEACTIONMETADATA']._serialized_start=1275
86
- _globals['_TRACEACTIONMETADATA']._serialized_end=1316
87
- _globals['_CONDITIONACTIONMETADATA']._serialized_start=1319
88
- _globals['_CONDITIONACTIONMETADATA']._serialized_end=1478
89
- _globals['_ACTIONMETADATA']._serialized_start=1481
90
- _globals['_ACTIONMETADATA']._serialized_end=1981
91
- _globals['_ACTIONSTATUS']._serialized_start=1984
92
- _globals['_ACTIONSTATUS']._serialized_end=2285
93
- _globals['_ACTION']._serialized_start=2288
94
- _globals['_ACTION']._serialized_end=2467
95
- _globals['_ENRICHEDACTION']._serialized_start=2470
96
- _globals['_ENRICHEDACTION']._serialized_end=2756
97
- _globals['_ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY']._serialized_start=2686
98
- _globals['_ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY']._serialized_end=2756
99
- _globals['_ERRORINFO']._serialized_start=2759
100
- _globals['_ERRORINFO']._serialized_end=2913
101
- _globals['_ERRORINFO_KIND']._serialized_start=2853
102
- _globals['_ERRORINFO_KIND']._serialized_end=2913
103
- _globals['_ABORTINFO']._serialized_start=2915
104
- _globals['_ABORTINFO']._serialized_end=3016
105
- _globals['_ACTIONDETAILS']._serialized_start=3019
106
- _globals['_ACTIONDETAILS']._serialized_end=3516
107
- _globals['_ACTIONATTEMPT']._serialized_start=3519
108
- _globals['_ACTIONATTEMPT']._serialized_end=4292
109
- _globals['_CLUSTEREVENT']._serialized_start=4294
110
- _globals['_CLUSTEREVENT']._serialized_end=4395
111
- _globals['_PHASETRANSITION']._serialized_start=4398
112
- _globals['_PHASETRANSITION']._serialized_end=4595
113
- _globals['_ACTIONEVENT']._serialized_start=4598
114
- _globals['_ACTIONEVENT']._serialized_end=5472
115
- _globals['_NAMEDLITERAL']._serialized_start=5474
116
- _globals['_NAMEDLITERAL']._serialized_end=5554
117
- _globals['_OUTPUTREFERENCES']._serialized_start=5556
118
- _globals['_OUTPUTREFERENCES']._serialized_end=5636
119
- _globals['_INPUTS']._serialized_start=5638
120
- _globals['_INPUTS']._serialized_end=5762
121
- _globals['_OUTPUTS']._serialized_start=5764
122
- _globals['_OUTPUTS']._serialized_end=5834
123
- # @@protoc_insertion_point(module_scope)
@@ -1,354 +0,0 @@
1
- from flyte._protos.common import identifier_pb2 as _identifier_pb2
2
- from flyte._protos.common import identity_pb2 as _identity_pb2
3
- from flyteidl.core import catalog_pb2 as _catalog_pb2
4
- from flyteidl.core import execution_pb2 as _execution_pb2
5
- from flyteidl.core import literals_pb2 as _literals_pb2
6
- from google.protobuf import timestamp_pb2 as _timestamp_pb2
7
- from google.protobuf import wrappers_pb2 as _wrappers_pb2
8
- from flyte._protos.validate.validate import validate_pb2 as _validate_pb2
9
- from flyte._protos.workflow import task_definition_pb2 as _task_definition_pb2
10
- from google.protobuf.internal import containers as _containers
11
- from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
12
- from google.protobuf import descriptor as _descriptor
13
- from google.protobuf import message as _message
14
- from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
15
-
16
- DESCRIPTOR: _descriptor.FileDescriptor
17
-
18
- class Phase(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
19
- __slots__ = []
20
- PHASE_UNSPECIFIED: _ClassVar[Phase]
21
- PHASE_QUEUED: _ClassVar[Phase]
22
- PHASE_WAITING_FOR_RESOURCES: _ClassVar[Phase]
23
- PHASE_INITIALIZING: _ClassVar[Phase]
24
- PHASE_RUNNING: _ClassVar[Phase]
25
- PHASE_SUCCEEDED: _ClassVar[Phase]
26
- PHASE_FAILED: _ClassVar[Phase]
27
- PHASE_ABORTED: _ClassVar[Phase]
28
- PHASE_TIMED_OUT: _ClassVar[Phase]
29
-
30
- class ActionType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
31
- __slots__ = []
32
- ACTION_TYPE_UNSPECIFIED: _ClassVar[ActionType]
33
- ACTION_TYPE_TASK: _ClassVar[ActionType]
34
- ACTION_TYPE_TRACE: _ClassVar[ActionType]
35
- ACTION_TYPE_CONDITION: _ClassVar[ActionType]
36
- PHASE_UNSPECIFIED: Phase
37
- PHASE_QUEUED: Phase
38
- PHASE_WAITING_FOR_RESOURCES: Phase
39
- PHASE_INITIALIZING: Phase
40
- PHASE_RUNNING: Phase
41
- PHASE_SUCCEEDED: Phase
42
- PHASE_FAILED: Phase
43
- PHASE_ABORTED: Phase
44
- PHASE_TIMED_OUT: Phase
45
- ACTION_TYPE_UNSPECIFIED: ActionType
46
- ACTION_TYPE_TASK: ActionType
47
- ACTION_TYPE_TRACE: ActionType
48
- ACTION_TYPE_CONDITION: ActionType
49
-
50
- class Labels(_message.Message):
51
- __slots__ = ["values"]
52
- class ValuesEntry(_message.Message):
53
- __slots__ = ["key", "value"]
54
- KEY_FIELD_NUMBER: _ClassVar[int]
55
- VALUE_FIELD_NUMBER: _ClassVar[int]
56
- key: str
57
- value: str
58
- def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
59
- VALUES_FIELD_NUMBER: _ClassVar[int]
60
- values: _containers.ScalarMap[str, str]
61
- def __init__(self, values: _Optional[_Mapping[str, str]] = ...) -> None: ...
62
-
63
- class Annotations(_message.Message):
64
- __slots__ = ["values"]
65
- class ValuesEntry(_message.Message):
66
- __slots__ = ["key", "value"]
67
- KEY_FIELD_NUMBER: _ClassVar[int]
68
- VALUE_FIELD_NUMBER: _ClassVar[int]
69
- key: str
70
- value: str
71
- def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
72
- VALUES_FIELD_NUMBER: _ClassVar[int]
73
- values: _containers.ScalarMap[str, str]
74
- def __init__(self, values: _Optional[_Mapping[str, str]] = ...) -> None: ...
75
-
76
- class Envs(_message.Message):
77
- __slots__ = ["values"]
78
- VALUES_FIELD_NUMBER: _ClassVar[int]
79
- values: _containers.RepeatedCompositeFieldContainer[_literals_pb2.KeyValuePair]
80
- def __init__(self, values: _Optional[_Iterable[_Union[_literals_pb2.KeyValuePair, _Mapping]]] = ...) -> None: ...
81
-
82
- class RunSpec(_message.Message):
83
- __slots__ = ["labels", "annotations", "envs", "interruptible", "overwrite_cache", "cluster"]
84
- LABELS_FIELD_NUMBER: _ClassVar[int]
85
- ANNOTATIONS_FIELD_NUMBER: _ClassVar[int]
86
- ENVS_FIELD_NUMBER: _ClassVar[int]
87
- INTERRUPTIBLE_FIELD_NUMBER: _ClassVar[int]
88
- OVERWRITE_CACHE_FIELD_NUMBER: _ClassVar[int]
89
- CLUSTER_FIELD_NUMBER: _ClassVar[int]
90
- labels: Labels
91
- annotations: Annotations
92
- envs: Envs
93
- interruptible: _wrappers_pb2.BoolValue
94
- overwrite_cache: bool
95
- cluster: str
96
- def __init__(self, labels: _Optional[_Union[Labels, _Mapping]] = ..., annotations: _Optional[_Union[Annotations, _Mapping]] = ..., envs: _Optional[_Union[Envs, _Mapping]] = ..., interruptible: _Optional[_Union[_wrappers_pb2.BoolValue, _Mapping]] = ..., overwrite_cache: bool = ..., cluster: _Optional[str] = ...) -> None: ...
97
-
98
- class Run(_message.Message):
99
- __slots__ = ["action"]
100
- ACTION_FIELD_NUMBER: _ClassVar[int]
101
- action: Action
102
- def __init__(self, action: _Optional[_Union[Action, _Mapping]] = ...) -> None: ...
103
-
104
- class RunDetails(_message.Message):
105
- __slots__ = ["run_spec", "action"]
106
- RUN_SPEC_FIELD_NUMBER: _ClassVar[int]
107
- ACTION_FIELD_NUMBER: _ClassVar[int]
108
- run_spec: RunSpec
109
- action: ActionDetails
110
- def __init__(self, run_spec: _Optional[_Union[RunSpec, _Mapping]] = ..., action: _Optional[_Union[ActionDetails, _Mapping]] = ...) -> None: ...
111
-
112
- class TaskActionMetadata(_message.Message):
113
- __slots__ = ["id", "task_type", "short_name"]
114
- ID_FIELD_NUMBER: _ClassVar[int]
115
- TASK_TYPE_FIELD_NUMBER: _ClassVar[int]
116
- SHORT_NAME_FIELD_NUMBER: _ClassVar[int]
117
- id: _task_definition_pb2.TaskIdentifier
118
- task_type: str
119
- short_name: str
120
- def __init__(self, id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ..., task_type: _Optional[str] = ..., short_name: _Optional[str] = ...) -> None: ...
121
-
122
- class TraceActionMetadata(_message.Message):
123
- __slots__ = ["name"]
124
- NAME_FIELD_NUMBER: _ClassVar[int]
125
- name: str
126
- def __init__(self, name: _Optional[str] = ...) -> None: ...
127
-
128
- class ConditionActionMetadata(_message.Message):
129
- __slots__ = ["name", "run_id", "action_id"]
130
- NAME_FIELD_NUMBER: _ClassVar[int]
131
- RUN_ID_FIELD_NUMBER: _ClassVar[int]
132
- ACTION_ID_FIELD_NUMBER: _ClassVar[int]
133
- GLOBAL_FIELD_NUMBER: _ClassVar[int]
134
- name: str
135
- run_id: str
136
- action_id: str
137
- def __init__(self, name: _Optional[str] = ..., run_id: _Optional[str] = ..., action_id: _Optional[str] = ..., **kwargs) -> None: ...
138
-
139
- class ActionMetadata(_message.Message):
140
- __slots__ = ["parent", "group", "executed_by", "task", "trace", "condition", "action_type", "trigger_id"]
141
- PARENT_FIELD_NUMBER: _ClassVar[int]
142
- GROUP_FIELD_NUMBER: _ClassVar[int]
143
- EXECUTED_BY_FIELD_NUMBER: _ClassVar[int]
144
- TASK_FIELD_NUMBER: _ClassVar[int]
145
- TRACE_FIELD_NUMBER: _ClassVar[int]
146
- CONDITION_FIELD_NUMBER: _ClassVar[int]
147
- ACTION_TYPE_FIELD_NUMBER: _ClassVar[int]
148
- TRIGGER_ID_FIELD_NUMBER: _ClassVar[int]
149
- parent: str
150
- group: str
151
- executed_by: _identity_pb2.EnrichedIdentity
152
- task: TaskActionMetadata
153
- trace: TraceActionMetadata
154
- condition: ConditionActionMetadata
155
- action_type: ActionType
156
- trigger_id: _identifier_pb2.TriggerIdentifier
157
- def __init__(self, parent: _Optional[str] = ..., group: _Optional[str] = ..., executed_by: _Optional[_Union[_identity_pb2.EnrichedIdentity, _Mapping]] = ..., task: _Optional[_Union[TaskActionMetadata, _Mapping]] = ..., trace: _Optional[_Union[TraceActionMetadata, _Mapping]] = ..., condition: _Optional[_Union[ConditionActionMetadata, _Mapping]] = ..., action_type: _Optional[_Union[ActionType, str]] = ..., trigger_id: _Optional[_Union[_identifier_pb2.TriggerIdentifier, _Mapping]] = ...) -> None: ...
158
-
159
- class ActionStatus(_message.Message):
160
- __slots__ = ["phase", "start_time", "end_time", "attempts", "cache_status"]
161
- PHASE_FIELD_NUMBER: _ClassVar[int]
162
- START_TIME_FIELD_NUMBER: _ClassVar[int]
163
- END_TIME_FIELD_NUMBER: _ClassVar[int]
164
- ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
165
- CACHE_STATUS_FIELD_NUMBER: _ClassVar[int]
166
- phase: Phase
167
- start_time: _timestamp_pb2.Timestamp
168
- end_time: _timestamp_pb2.Timestamp
169
- attempts: int
170
- cache_status: _catalog_pb2.CatalogCacheStatus
171
- def __init__(self, phase: _Optional[_Union[Phase, str]] = ..., start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., attempts: _Optional[int] = ..., cache_status: _Optional[_Union[_catalog_pb2.CatalogCacheStatus, str]] = ...) -> None: ...
172
-
173
- class Action(_message.Message):
174
- __slots__ = ["id", "metadata", "status"]
175
- ID_FIELD_NUMBER: _ClassVar[int]
176
- METADATA_FIELD_NUMBER: _ClassVar[int]
177
- STATUS_FIELD_NUMBER: _ClassVar[int]
178
- id: _identifier_pb2.ActionIdentifier
179
- metadata: ActionMetadata
180
- status: ActionStatus
181
- def __init__(self, id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[ActionMetadata, _Mapping]] = ..., status: _Optional[_Union[ActionStatus, _Mapping]] = ...) -> None: ...
182
-
183
- class EnrichedAction(_message.Message):
184
- __slots__ = ["action", "meets_filter", "children_phase_counts"]
185
- class ChildrenPhaseCountsEntry(_message.Message):
186
- __slots__ = ["key", "value"]
187
- KEY_FIELD_NUMBER: _ClassVar[int]
188
- VALUE_FIELD_NUMBER: _ClassVar[int]
189
- key: int
190
- value: int
191
- def __init__(self, key: _Optional[int] = ..., value: _Optional[int] = ...) -> None: ...
192
- ACTION_FIELD_NUMBER: _ClassVar[int]
193
- MEETS_FILTER_FIELD_NUMBER: _ClassVar[int]
194
- CHILDREN_PHASE_COUNTS_FIELD_NUMBER: _ClassVar[int]
195
- action: Action
196
- meets_filter: bool
197
- children_phase_counts: _containers.ScalarMap[int, int]
198
- def __init__(self, action: _Optional[_Union[Action, _Mapping]] = ..., meets_filter: bool = ..., children_phase_counts: _Optional[_Mapping[int, int]] = ...) -> None: ...
199
-
200
- class ErrorInfo(_message.Message):
201
- __slots__ = ["message", "kind"]
202
- class Kind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
203
- __slots__ = []
204
- KIND_UNSPECIFIED: _ClassVar[ErrorInfo.Kind]
205
- KIND_USER: _ClassVar[ErrorInfo.Kind]
206
- KIND_SYSTEM: _ClassVar[ErrorInfo.Kind]
207
- KIND_UNSPECIFIED: ErrorInfo.Kind
208
- KIND_USER: ErrorInfo.Kind
209
- KIND_SYSTEM: ErrorInfo.Kind
210
- MESSAGE_FIELD_NUMBER: _ClassVar[int]
211
- KIND_FIELD_NUMBER: _ClassVar[int]
212
- message: str
213
- kind: ErrorInfo.Kind
214
- def __init__(self, message: _Optional[str] = ..., kind: _Optional[_Union[ErrorInfo.Kind, str]] = ...) -> None: ...
215
-
216
- class AbortInfo(_message.Message):
217
- __slots__ = ["reason", "aborted_by"]
218
- REASON_FIELD_NUMBER: _ClassVar[int]
219
- ABORTED_BY_FIELD_NUMBER: _ClassVar[int]
220
- reason: str
221
- aborted_by: _identity_pb2.EnrichedIdentity
222
- def __init__(self, reason: _Optional[str] = ..., aborted_by: _Optional[_Union[_identity_pb2.EnrichedIdentity, _Mapping]] = ...) -> None: ...
223
-
224
- class ActionDetails(_message.Message):
225
- __slots__ = ["id", "metadata", "status", "error_info", "abort_info", "task", "trace", "attempts"]
226
- ID_FIELD_NUMBER: _ClassVar[int]
227
- METADATA_FIELD_NUMBER: _ClassVar[int]
228
- STATUS_FIELD_NUMBER: _ClassVar[int]
229
- ERROR_INFO_FIELD_NUMBER: _ClassVar[int]
230
- ABORT_INFO_FIELD_NUMBER: _ClassVar[int]
231
- TASK_FIELD_NUMBER: _ClassVar[int]
232
- TRACE_FIELD_NUMBER: _ClassVar[int]
233
- ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
234
- id: _identifier_pb2.ActionIdentifier
235
- metadata: ActionMetadata
236
- status: ActionStatus
237
- error_info: ErrorInfo
238
- abort_info: AbortInfo
239
- task: _task_definition_pb2.TaskSpec
240
- trace: _task_definition_pb2.TraceSpec
241
- attempts: _containers.RepeatedCompositeFieldContainer[ActionAttempt]
242
- 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]] = ..., task: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ..., trace: _Optional[_Union[_task_definition_pb2.TraceSpec, _Mapping]] = ..., attempts: _Optional[_Iterable[_Union[ActionAttempt, _Mapping]]] = ...) -> None: ...
243
-
244
- class ActionAttempt(_message.Message):
245
- __slots__ = ["phase", "start_time", "end_time", "error_info", "attempt", "log_info", "outputs", "logs_available", "cache_status", "cluster_events", "phase_transitions", "cluster", "log_context"]
246
- PHASE_FIELD_NUMBER: _ClassVar[int]
247
- START_TIME_FIELD_NUMBER: _ClassVar[int]
248
- END_TIME_FIELD_NUMBER: _ClassVar[int]
249
- ERROR_INFO_FIELD_NUMBER: _ClassVar[int]
250
- ATTEMPT_FIELD_NUMBER: _ClassVar[int]
251
- LOG_INFO_FIELD_NUMBER: _ClassVar[int]
252
- OUTPUTS_FIELD_NUMBER: _ClassVar[int]
253
- LOGS_AVAILABLE_FIELD_NUMBER: _ClassVar[int]
254
- CACHE_STATUS_FIELD_NUMBER: _ClassVar[int]
255
- CLUSTER_EVENTS_FIELD_NUMBER: _ClassVar[int]
256
- PHASE_TRANSITIONS_FIELD_NUMBER: _ClassVar[int]
257
- CLUSTER_FIELD_NUMBER: _ClassVar[int]
258
- LOG_CONTEXT_FIELD_NUMBER: _ClassVar[int]
259
- phase: Phase
260
- start_time: _timestamp_pb2.Timestamp
261
- end_time: _timestamp_pb2.Timestamp
262
- error_info: ErrorInfo
263
- attempt: int
264
- log_info: _containers.RepeatedCompositeFieldContainer[_execution_pb2.TaskLog]
265
- outputs: OutputReferences
266
- logs_available: bool
267
- cache_status: _catalog_pb2.CatalogCacheStatus
268
- cluster_events: _containers.RepeatedCompositeFieldContainer[ClusterEvent]
269
- phase_transitions: _containers.RepeatedCompositeFieldContainer[PhaseTransition]
270
- cluster: str
271
- log_context: _execution_pb2.LogContext
272
- 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] = ..., log_context: _Optional[_Union[_execution_pb2.LogContext, _Mapping]] = ...) -> None: ...
273
-
274
- class ClusterEvent(_message.Message):
275
- __slots__ = ["occurred_at", "message"]
276
- OCCURRED_AT_FIELD_NUMBER: _ClassVar[int]
277
- MESSAGE_FIELD_NUMBER: _ClassVar[int]
278
- occurred_at: _timestamp_pb2.Timestamp
279
- message: str
280
- def __init__(self, occurred_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., message: _Optional[str] = ...) -> None: ...
281
-
282
- class PhaseTransition(_message.Message):
283
- __slots__ = ["phase", "start_time", "end_time"]
284
- PHASE_FIELD_NUMBER: _ClassVar[int]
285
- START_TIME_FIELD_NUMBER: _ClassVar[int]
286
- END_TIME_FIELD_NUMBER: _ClassVar[int]
287
- phase: Phase
288
- start_time: _timestamp_pb2.Timestamp
289
- end_time: _timestamp_pb2.Timestamp
290
- 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: ...
291
-
292
- class ActionEvent(_message.Message):
293
- __slots__ = ["id", "attempt", "phase", "version", "start_time", "updated_time", "end_time", "error_info", "log_info", "log_context", "cluster", "outputs", "cache_status", "cluster_events", "reported_time"]
294
- ID_FIELD_NUMBER: _ClassVar[int]
295
- ATTEMPT_FIELD_NUMBER: _ClassVar[int]
296
- PHASE_FIELD_NUMBER: _ClassVar[int]
297
- VERSION_FIELD_NUMBER: _ClassVar[int]
298
- START_TIME_FIELD_NUMBER: _ClassVar[int]
299
- UPDATED_TIME_FIELD_NUMBER: _ClassVar[int]
300
- END_TIME_FIELD_NUMBER: _ClassVar[int]
301
- ERROR_INFO_FIELD_NUMBER: _ClassVar[int]
302
- LOG_INFO_FIELD_NUMBER: _ClassVar[int]
303
- LOG_CONTEXT_FIELD_NUMBER: _ClassVar[int]
304
- CLUSTER_FIELD_NUMBER: _ClassVar[int]
305
- OUTPUTS_FIELD_NUMBER: _ClassVar[int]
306
- CACHE_STATUS_FIELD_NUMBER: _ClassVar[int]
307
- CLUSTER_EVENTS_FIELD_NUMBER: _ClassVar[int]
308
- REPORTED_TIME_FIELD_NUMBER: _ClassVar[int]
309
- id: _identifier_pb2.ActionIdentifier
310
- attempt: int
311
- phase: Phase
312
- version: int
313
- start_time: _timestamp_pb2.Timestamp
314
- updated_time: _timestamp_pb2.Timestamp
315
- end_time: _timestamp_pb2.Timestamp
316
- error_info: ErrorInfo
317
- log_info: _containers.RepeatedCompositeFieldContainer[_execution_pb2.TaskLog]
318
- log_context: _execution_pb2.LogContext
319
- cluster: str
320
- outputs: OutputReferences
321
- cache_status: _catalog_pb2.CatalogCacheStatus
322
- cluster_events: _containers.RepeatedCompositeFieldContainer[ClusterEvent]
323
- reported_time: _timestamp_pb2.Timestamp
324
- 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]]] = ..., reported_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
325
-
326
- class NamedLiteral(_message.Message):
327
- __slots__ = ["name", "value"]
328
- NAME_FIELD_NUMBER: _ClassVar[int]
329
- VALUE_FIELD_NUMBER: _ClassVar[int]
330
- name: str
331
- value: _literals_pb2.Literal
332
- def __init__(self, name: _Optional[str] = ..., value: _Optional[_Union[_literals_pb2.Literal, _Mapping]] = ...) -> None: ...
333
-
334
- class OutputReferences(_message.Message):
335
- __slots__ = ["output_uri", "report_uri"]
336
- OUTPUT_URI_FIELD_NUMBER: _ClassVar[int]
337
- REPORT_URI_FIELD_NUMBER: _ClassVar[int]
338
- output_uri: str
339
- report_uri: str
340
- def __init__(self, output_uri: _Optional[str] = ..., report_uri: _Optional[str] = ...) -> None: ...
341
-
342
- class Inputs(_message.Message):
343
- __slots__ = ["literals", "context"]
344
- LITERALS_FIELD_NUMBER: _ClassVar[int]
345
- CONTEXT_FIELD_NUMBER: _ClassVar[int]
346
- literals: _containers.RepeatedCompositeFieldContainer[NamedLiteral]
347
- context: _containers.RepeatedCompositeFieldContainer[_literals_pb2.KeyValuePair]
348
- def __init__(self, literals: _Optional[_Iterable[_Union[NamedLiteral, _Mapping]]] = ..., context: _Optional[_Iterable[_Union[_literals_pb2.KeyValuePair, _Mapping]]] = ...) -> None: ...
349
-
350
- class Outputs(_message.Message):
351
- __slots__ = ["literals"]
352
- LITERALS_FIELD_NUMBER: _ClassVar[int]
353
- literals: _containers.RepeatedCompositeFieldContainer[NamedLiteral]
354
- def __init__(self, literals: _Optional[_Iterable[_Union[NamedLiteral, _Mapping]]] = ...) -> None: ...
@@ -1,4 +0,0 @@
1
- # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
- """Client and server classes corresponding to protobuf-defined services."""
3
- import grpc
4
-
@@ -1,41 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- # Generated by the protocol buffer compiler. DO NOT EDIT!
3
- # source: workflow/run_logs_service.proto
4
- """Generated protocol buffer code."""
5
- from google.protobuf import descriptor as _descriptor
6
- from google.protobuf import descriptor_pool as _descriptor_pool
7
- from google.protobuf import symbol_database as _symbol_database
8
- from google.protobuf.internal import builder as _builder
9
- # @@protoc_insertion_point(imports)
10
-
11
- _sym_db = _symbol_database.Default()
12
-
13
-
14
- from flyte._protos.common import identifier_pb2 as common_dot_identifier__pb2
15
- from flyte._protos.logs.dataplane import payload_pb2 as logs_dot_dataplane_dot_payload__pb2
16
- from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
17
-
18
-
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
-
21
- _globals = globals()
22
- _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
23
- _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'workflow.run_logs_service_pb2', _globals)
24
- if _descriptor._USE_C_DESCRIPTORS == False:
25
- DESCRIPTOR._options = None
26
- DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\023RunLogsServiceProtoH\002P\001Z+github.com/unionai/cloud/gen/pb-go/workflow\242\002\003CWX\252\002\021Cloudidl.Workflow\312\002\021Cloudidl\\Workflow\342\002\035Cloudidl\\Workflow\\GPBMetadata\352\002\022Cloudidl::Workflow'
27
- _TAILLOGSREQUEST.fields_by_name['action_id']._options = None
28
- _TAILLOGSREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
29
- _TAILLOGSREQUEST.fields_by_name['attempt']._options = None
30
- _TAILLOGSREQUEST.fields_by_name['attempt']._serialized_options = b'\372B\004*\002 \000'
31
- _RUNLOGSSERVICE.methods_by_name['TailLogs']._options = None
32
- _RUNLOGSSERVICE.methods_by_name['TailLogs']._serialized_options = b'\220\002\001'
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
- # @@protoc_insertion_point(module_scope)
@@ -1,28 +0,0 @@
1
- from flyte._protos.common import identifier_pb2 as _identifier_pb2
2
- from flyte._protos.logs.dataplane import payload_pb2 as _payload_pb2
3
- from flyte._protos.validate.validate import validate_pb2 as _validate_pb2
4
- from google.protobuf.internal import containers as _containers
5
- from google.protobuf import descriptor as _descriptor
6
- from google.protobuf import message as _message
7
- from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
8
-
9
- DESCRIPTOR: _descriptor.FileDescriptor
10
-
11
- class TailLogsRequest(_message.Message):
12
- __slots__ = ["action_id", "attempt"]
13
- ACTION_ID_FIELD_NUMBER: _ClassVar[int]
14
- ATTEMPT_FIELD_NUMBER: _ClassVar[int]
15
- action_id: _identifier_pb2.ActionIdentifier
16
- attempt: int
17
- def __init__(self, action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ..., attempt: _Optional[int] = ...) -> None: ...
18
-
19
- class TailLogsResponse(_message.Message):
20
- __slots__ = ["logs"]
21
- class Logs(_message.Message):
22
- __slots__ = ["lines"]
23
- LINES_FIELD_NUMBER: _ClassVar[int]
24
- lines: _containers.RepeatedCompositeFieldContainer[_payload_pb2.LogLine]
25
- def __init__(self, lines: _Optional[_Iterable[_Union[_payload_pb2.LogLine, _Mapping]]] = ...) -> None: ...
26
- LOGS_FIELD_NUMBER: _ClassVar[int]
27
- logs: _containers.RepeatedCompositeFieldContainer[TailLogsResponse.Logs]
28
- def __init__(self, logs: _Optional[_Iterable[_Union[TailLogsResponse.Logs, _Mapping]]] = ...) -> None: ...
@@ -1,69 +0,0 @@
1
- # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
- """Client and server classes corresponding to protobuf-defined services."""
3
- import grpc
4
-
5
- from flyte._protos.workflow import run_logs_service_pb2 as workflow_dot_run__logs__service__pb2
6
-
7
-
8
- class RunLogsServiceStub(object):
9
- """RunLogsService provides an interface for streaming logs.
10
- """
11
-
12
- def __init__(self, channel):
13
- """Constructor.
14
-
15
- Args:
16
- channel: A grpc.Channel.
17
- """
18
- self.TailLogs = channel.unary_stream(
19
- '/cloudidl.workflow.RunLogsService/TailLogs',
20
- request_serializer=workflow_dot_run__logs__service__pb2.TailLogsRequest.SerializeToString,
21
- response_deserializer=workflow_dot_run__logs__service__pb2.TailLogsResponse.FromString,
22
- )
23
-
24
-
25
- class RunLogsServiceServicer(object):
26
- """RunLogsService provides an interface for streaming logs.
27
- """
28
-
29
- def TailLogs(self, request, context):
30
- """Missing associated documentation comment in .proto file."""
31
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
32
- context.set_details('Method not implemented!')
33
- raise NotImplementedError('Method not implemented!')
34
-
35
-
36
- def add_RunLogsServiceServicer_to_server(servicer, server):
37
- rpc_method_handlers = {
38
- 'TailLogs': grpc.unary_stream_rpc_method_handler(
39
- servicer.TailLogs,
40
- request_deserializer=workflow_dot_run__logs__service__pb2.TailLogsRequest.FromString,
41
- response_serializer=workflow_dot_run__logs__service__pb2.TailLogsResponse.SerializeToString,
42
- ),
43
- }
44
- generic_handler = grpc.method_handlers_generic_handler(
45
- 'cloudidl.workflow.RunLogsService', rpc_method_handlers)
46
- server.add_generic_rpc_handlers((generic_handler,))
47
-
48
-
49
- # This class is part of an EXPERIMENTAL API.
50
- class RunLogsService(object):
51
- """RunLogsService provides an interface for streaming logs.
52
- """
53
-
54
- @staticmethod
55
- def TailLogs(request,
56
- target,
57
- options=(),
58
- channel_credentials=None,
59
- call_credentials=None,
60
- insecure=False,
61
- compression=None,
62
- wait_for_ready=None,
63
- timeout=None,
64
- metadata=None):
65
- return grpc.experimental.unary_stream(request, target, '/cloudidl.workflow.RunLogsService/TailLogs',
66
- workflow_dot_run__logs__service__pb2.TailLogsRequest.SerializeToString,
67
- workflow_dot_run__logs__service__pb2.TailLogsResponse.FromString,
68
- options, channel_credentials,
69
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)