flyte 0.1.0__py3-none-any.whl → 0.2.0a0__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 (219) hide show
  1. flyte/__init__.py +78 -2
  2. flyte/_bin/__init__.py +0 -0
  3. flyte/_bin/runtime.py +152 -0
  4. flyte/_build.py +26 -0
  5. flyte/_cache/__init__.py +12 -0
  6. flyte/_cache/cache.py +145 -0
  7. flyte/_cache/defaults.py +9 -0
  8. flyte/_cache/policy_function_body.py +42 -0
  9. flyte/_code_bundle/__init__.py +8 -0
  10. flyte/_code_bundle/_ignore.py +113 -0
  11. flyte/_code_bundle/_packaging.py +187 -0
  12. flyte/_code_bundle/_utils.py +323 -0
  13. flyte/_code_bundle/bundle.py +209 -0
  14. flyte/_context.py +152 -0
  15. flyte/_deploy.py +243 -0
  16. flyte/_doc.py +29 -0
  17. flyte/_docstring.py +32 -0
  18. flyte/_environment.py +84 -0
  19. flyte/_excepthook.py +37 -0
  20. flyte/_group.py +32 -0
  21. flyte/_hash.py +23 -0
  22. flyte/_image.py +762 -0
  23. flyte/_initialize.py +492 -0
  24. flyte/_interface.py +84 -0
  25. flyte/_internal/__init__.py +3 -0
  26. flyte/_internal/controllers/__init__.py +128 -0
  27. flyte/_internal/controllers/_local_controller.py +193 -0
  28. flyte/_internal/controllers/_trace.py +41 -0
  29. flyte/_internal/controllers/remote/__init__.py +60 -0
  30. flyte/_internal/controllers/remote/_action.py +146 -0
  31. flyte/_internal/controllers/remote/_client.py +47 -0
  32. flyte/_internal/controllers/remote/_controller.py +494 -0
  33. flyte/_internal/controllers/remote/_core.py +410 -0
  34. flyte/_internal/controllers/remote/_informer.py +361 -0
  35. flyte/_internal/controllers/remote/_service_protocol.py +50 -0
  36. flyte/_internal/imagebuild/__init__.py +11 -0
  37. flyte/_internal/imagebuild/docker_builder.py +427 -0
  38. flyte/_internal/imagebuild/image_builder.py +246 -0
  39. flyte/_internal/imagebuild/remote_builder.py +0 -0
  40. flyte/_internal/resolvers/__init__.py +0 -0
  41. flyte/_internal/resolvers/_task_module.py +54 -0
  42. flyte/_internal/resolvers/common.py +31 -0
  43. flyte/_internal/resolvers/default.py +28 -0
  44. flyte/_internal/runtime/__init__.py +0 -0
  45. flyte/_internal/runtime/convert.py +342 -0
  46. flyte/_internal/runtime/entrypoints.py +135 -0
  47. flyte/_internal/runtime/io.py +136 -0
  48. flyte/_internal/runtime/resources_serde.py +138 -0
  49. flyte/_internal/runtime/task_serde.py +330 -0
  50. flyte/_internal/runtime/taskrunner.py +191 -0
  51. flyte/_internal/runtime/types_serde.py +54 -0
  52. flyte/_logging.py +135 -0
  53. flyte/_map.py +215 -0
  54. flyte/_pod.py +19 -0
  55. flyte/_protos/__init__.py +0 -0
  56. flyte/_protos/common/authorization_pb2.py +66 -0
  57. flyte/_protos/common/authorization_pb2.pyi +108 -0
  58. flyte/_protos/common/authorization_pb2_grpc.py +4 -0
  59. flyte/_protos/common/identifier_pb2.py +71 -0
  60. flyte/_protos/common/identifier_pb2.pyi +82 -0
  61. flyte/_protos/common/identifier_pb2_grpc.py +4 -0
  62. flyte/_protos/common/identity_pb2.py +48 -0
  63. flyte/_protos/common/identity_pb2.pyi +72 -0
  64. flyte/_protos/common/identity_pb2_grpc.py +4 -0
  65. flyte/_protos/common/list_pb2.py +36 -0
  66. flyte/_protos/common/list_pb2.pyi +71 -0
  67. flyte/_protos/common/list_pb2_grpc.py +4 -0
  68. flyte/_protos/common/policy_pb2.py +37 -0
  69. flyte/_protos/common/policy_pb2.pyi +27 -0
  70. flyte/_protos/common/policy_pb2_grpc.py +4 -0
  71. flyte/_protos/common/role_pb2.py +37 -0
  72. flyte/_protos/common/role_pb2.pyi +53 -0
  73. flyte/_protos/common/role_pb2_grpc.py +4 -0
  74. flyte/_protos/common/runtime_version_pb2.py +28 -0
  75. flyte/_protos/common/runtime_version_pb2.pyi +24 -0
  76. flyte/_protos/common/runtime_version_pb2_grpc.py +4 -0
  77. flyte/_protos/logs/dataplane/payload_pb2.py +100 -0
  78. flyte/_protos/logs/dataplane/payload_pb2.pyi +177 -0
  79. flyte/_protos/logs/dataplane/payload_pb2_grpc.py +4 -0
  80. flyte/_protos/secret/definition_pb2.py +49 -0
  81. flyte/_protos/secret/definition_pb2.pyi +93 -0
  82. flyte/_protos/secret/definition_pb2_grpc.py +4 -0
  83. flyte/_protos/secret/payload_pb2.py +62 -0
  84. flyte/_protos/secret/payload_pb2.pyi +94 -0
  85. flyte/_protos/secret/payload_pb2_grpc.py +4 -0
  86. flyte/_protos/secret/secret_pb2.py +38 -0
  87. flyte/_protos/secret/secret_pb2.pyi +6 -0
  88. flyte/_protos/secret/secret_pb2_grpc.py +198 -0
  89. flyte/_protos/secret/secret_pb2_grpc_grpc.py +198 -0
  90. flyte/_protos/validate/validate/validate_pb2.py +76 -0
  91. flyte/_protos/workflow/common_pb2.py +27 -0
  92. flyte/_protos/workflow/common_pb2.pyi +14 -0
  93. flyte/_protos/workflow/common_pb2_grpc.py +4 -0
  94. flyte/_protos/workflow/environment_pb2.py +29 -0
  95. flyte/_protos/workflow/environment_pb2.pyi +12 -0
  96. flyte/_protos/workflow/environment_pb2_grpc.py +4 -0
  97. flyte/_protos/workflow/node_execution_service_pb2.py +26 -0
  98. flyte/_protos/workflow/node_execution_service_pb2.pyi +4 -0
  99. flyte/_protos/workflow/node_execution_service_pb2_grpc.py +32 -0
  100. flyte/_protos/workflow/queue_service_pb2.py +105 -0
  101. flyte/_protos/workflow/queue_service_pb2.pyi +146 -0
  102. flyte/_protos/workflow/queue_service_pb2_grpc.py +172 -0
  103. flyte/_protos/workflow/run_definition_pb2.py +128 -0
  104. flyte/_protos/workflow/run_definition_pb2.pyi +314 -0
  105. flyte/_protos/workflow/run_definition_pb2_grpc.py +4 -0
  106. flyte/_protos/workflow/run_logs_service_pb2.py +41 -0
  107. flyte/_protos/workflow/run_logs_service_pb2.pyi +28 -0
  108. flyte/_protos/workflow/run_logs_service_pb2_grpc.py +69 -0
  109. flyte/_protos/workflow/run_service_pb2.py +129 -0
  110. flyte/_protos/workflow/run_service_pb2.pyi +171 -0
  111. flyte/_protos/workflow/run_service_pb2_grpc.py +412 -0
  112. flyte/_protos/workflow/state_service_pb2.py +66 -0
  113. flyte/_protos/workflow/state_service_pb2.pyi +75 -0
  114. flyte/_protos/workflow/state_service_pb2_grpc.py +138 -0
  115. flyte/_protos/workflow/task_definition_pb2.py +79 -0
  116. flyte/_protos/workflow/task_definition_pb2.pyi +81 -0
  117. flyte/_protos/workflow/task_definition_pb2_grpc.py +4 -0
  118. flyte/_protos/workflow/task_service_pb2.py +60 -0
  119. flyte/_protos/workflow/task_service_pb2.pyi +59 -0
  120. flyte/_protos/workflow/task_service_pb2_grpc.py +138 -0
  121. flyte/_resources.py +226 -0
  122. flyte/_retry.py +32 -0
  123. flyte/_reusable_environment.py +25 -0
  124. flyte/_run.py +482 -0
  125. flyte/_secret.py +61 -0
  126. flyte/_task.py +449 -0
  127. flyte/_task_environment.py +183 -0
  128. flyte/_timeout.py +47 -0
  129. flyte/_tools.py +27 -0
  130. flyte/_trace.py +120 -0
  131. flyte/_utils/__init__.py +26 -0
  132. flyte/_utils/asyn.py +119 -0
  133. flyte/_utils/async_cache.py +139 -0
  134. flyte/_utils/coro_management.py +23 -0
  135. flyte/_utils/file_handling.py +72 -0
  136. flyte/_utils/helpers.py +134 -0
  137. flyte/_utils/lazy_module.py +54 -0
  138. flyte/_utils/org_discovery.py +57 -0
  139. flyte/_utils/uv_script_parser.py +49 -0
  140. flyte/_version.py +21 -0
  141. flyte/cli/__init__.py +3 -0
  142. flyte/cli/_abort.py +28 -0
  143. flyte/cli/_common.py +337 -0
  144. flyte/cli/_create.py +145 -0
  145. flyte/cli/_delete.py +23 -0
  146. flyte/cli/_deploy.py +152 -0
  147. flyte/cli/_gen.py +163 -0
  148. flyte/cli/_get.py +310 -0
  149. flyte/cli/_params.py +538 -0
  150. flyte/cli/_run.py +231 -0
  151. flyte/cli/main.py +166 -0
  152. flyte/config/__init__.py +3 -0
  153. flyte/config/_config.py +216 -0
  154. flyte/config/_internal.py +64 -0
  155. flyte/config/_reader.py +207 -0
  156. flyte/connectors/__init__.py +0 -0
  157. flyte/errors.py +172 -0
  158. flyte/extras/__init__.py +5 -0
  159. flyte/extras/_container.py +263 -0
  160. flyte/io/__init__.py +27 -0
  161. flyte/io/_dir.py +448 -0
  162. flyte/io/_file.py +467 -0
  163. flyte/io/_structured_dataset/__init__.py +129 -0
  164. flyte/io/_structured_dataset/basic_dfs.py +219 -0
  165. flyte/io/_structured_dataset/structured_dataset.py +1061 -0
  166. flyte/models.py +391 -0
  167. flyte/remote/__init__.py +26 -0
  168. flyte/remote/_client/__init__.py +0 -0
  169. flyte/remote/_client/_protocols.py +133 -0
  170. flyte/remote/_client/auth/__init__.py +12 -0
  171. flyte/remote/_client/auth/_auth_utils.py +14 -0
  172. flyte/remote/_client/auth/_authenticators/__init__.py +0 -0
  173. flyte/remote/_client/auth/_authenticators/base.py +397 -0
  174. flyte/remote/_client/auth/_authenticators/client_credentials.py +73 -0
  175. flyte/remote/_client/auth/_authenticators/device_code.py +118 -0
  176. flyte/remote/_client/auth/_authenticators/external_command.py +79 -0
  177. flyte/remote/_client/auth/_authenticators/factory.py +200 -0
  178. flyte/remote/_client/auth/_authenticators/pkce.py +516 -0
  179. flyte/remote/_client/auth/_channel.py +215 -0
  180. flyte/remote/_client/auth/_client_config.py +83 -0
  181. flyte/remote/_client/auth/_default_html.py +32 -0
  182. flyte/remote/_client/auth/_grpc_utils/__init__.py +0 -0
  183. flyte/remote/_client/auth/_grpc_utils/auth_interceptor.py +288 -0
  184. flyte/remote/_client/auth/_grpc_utils/default_metadata_interceptor.py +151 -0
  185. flyte/remote/_client/auth/_keyring.py +143 -0
  186. flyte/remote/_client/auth/_token_client.py +260 -0
  187. flyte/remote/_client/auth/errors.py +16 -0
  188. flyte/remote/_client/controlplane.py +95 -0
  189. flyte/remote/_console.py +18 -0
  190. flyte/remote/_data.py +159 -0
  191. flyte/remote/_logs.py +176 -0
  192. flyte/remote/_project.py +85 -0
  193. flyte/remote/_run.py +970 -0
  194. flyte/remote/_secret.py +132 -0
  195. flyte/remote/_task.py +391 -0
  196. flyte/report/__init__.py +3 -0
  197. flyte/report/_report.py +178 -0
  198. flyte/report/_template.html +124 -0
  199. flyte/storage/__init__.py +29 -0
  200. flyte/storage/_config.py +233 -0
  201. flyte/storage/_remote_fs.py +34 -0
  202. flyte/storage/_storage.py +271 -0
  203. flyte/storage/_utils.py +5 -0
  204. flyte/syncify/__init__.py +56 -0
  205. flyte/syncify/_api.py +371 -0
  206. flyte/types/__init__.py +36 -0
  207. flyte/types/_interface.py +40 -0
  208. flyte/types/_pickle.py +118 -0
  209. flyte/types/_renderer.py +162 -0
  210. flyte/types/_string_literals.py +120 -0
  211. flyte/types/_type_engine.py +2287 -0
  212. flyte/types/_utils.py +80 -0
  213. flyte-0.2.0a0.dist-info/METADATA +249 -0
  214. flyte-0.2.0a0.dist-info/RECORD +218 -0
  215. {flyte-0.1.0.dist-info → flyte-0.2.0a0.dist-info}/WHEEL +2 -1
  216. flyte-0.2.0a0.dist-info/entry_points.txt +3 -0
  217. flyte-0.2.0a0.dist-info/top_level.txt +1 -0
  218. flyte-0.1.0.dist-info/METADATA +0 -6
  219. flyte-0.1.0.dist-info/RECORD +0 -5
@@ -0,0 +1,128 @@
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 execution_pb2 as flyteidl_dot_core_dot_execution__pb2
17
+ from flyteidl.core import literals_pb2 as flyteidl_dot_core_dot_literals__pb2
18
+ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
19
+ from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
20
+ from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
21
+ from flyte._protos.workflow import task_definition_pb2 as workflow_dot_task__definition__pb2
22
+
23
+
24
+ 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\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\"\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\"\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\"o\n\x10\x41\x63tionIdentifier\x12<\n\x03run\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.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\"\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\"\xb5\x01\n\x06\x41\x63tion\x12\x33\n\x02id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.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\"\xcd\x03\n\rActionDetails\x12\x33\n\x02id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.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\"\xd0\x03\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\rlogsAvailableB\x0b\n\t_end_timeB\r\n\x0b_error_info\"\x95\x05\n\x0b\x41\x63tionEvent\x12=\n\x02id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.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\x39\n\nstart_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\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.TimestampH\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\x07outputsB\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\"1\n\x10OutputReferences\x12\x1d\n\noutput_uri\x18\x01 \x01(\tR\toutputUri\"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
+
26
+ _globals = globals()
27
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
28
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'workflow.run_definition_pb2', _globals)
29
+ if _descriptor._USE_C_DESCRIPTORS == False:
30
+ DESCRIPTOR._options = None
31
+ 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'
32
+ _RUNIDENTIFIER.fields_by_name['org']._options = None
33
+ _RUNIDENTIFIER.fields_by_name['org']._serialized_options = b'\372B\006r\004\020\001\030?'
34
+ _RUNIDENTIFIER.fields_by_name['project']._options = None
35
+ _RUNIDENTIFIER.fields_by_name['project']._serialized_options = b'\372B\006r\004\020\001\030?'
36
+ _RUNIDENTIFIER.fields_by_name['domain']._options = None
37
+ _RUNIDENTIFIER.fields_by_name['domain']._serialized_options = b'\372B\006r\004\020\001\030?'
38
+ _RUNIDENTIFIER.fields_by_name['name']._options = None
39
+ _RUNIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\006r\004\020\001\030\036'
40
+ _LABELS_VALUESENTRY._options = None
41
+ _LABELS_VALUESENTRY._serialized_options = b'8\001'
42
+ _ANNOTATIONS_VALUESENTRY._options = None
43
+ _ANNOTATIONS_VALUESENTRY._serialized_options = b'8\001'
44
+ _ACTIONIDENTIFIER.fields_by_name['run']._options = None
45
+ _ACTIONIDENTIFIER.fields_by_name['run']._serialized_options = b'\372B\005\212\001\002\020\001'
46
+ _ACTIONIDENTIFIER.fields_by_name['name']._options = None
47
+ _ACTIONIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\006r\004\020\001\030\036'
48
+ _CONDITIONACTIONMETADATA.oneofs_by_name['scope']._options = None
49
+ _CONDITIONACTIONMETADATA.oneofs_by_name['scope']._serialized_options = b'\370B\001'
50
+ _CONDITIONACTIONMETADATA.fields_by_name['run_id']._options = None
51
+ _CONDITIONACTIONMETADATA.fields_by_name['run_id']._serialized_options = b'\372B\004r\002\020\001'
52
+ _CONDITIONACTIONMETADATA.fields_by_name['action_id']._options = None
53
+ _CONDITIONACTIONMETADATA.fields_by_name['action_id']._serialized_options = b'\372B\004r\002\020\001'
54
+ _ACTIONMETADATA.fields_by_name['task']._options = None
55
+ _ACTIONMETADATA.fields_by_name['task']._serialized_options = b'\372B\005\212\001\002\020\001'
56
+ _ACTIONMETADATA.fields_by_name['trace']._options = None
57
+ _ACTIONMETADATA.fields_by_name['trace']._serialized_options = b'\372B\005\212\001\002\020\001'
58
+ _ACTIONMETADATA.fields_by_name['condition']._options = None
59
+ _ACTIONMETADATA.fields_by_name['condition']._serialized_options = b'\372B\005\212\001\002\020\001'
60
+ _ACTIONSTATUS.fields_by_name['attempts']._options = None
61
+ _ACTIONSTATUS.fields_by_name['attempts']._serialized_options = b'\372B\004*\002 \000'
62
+ _ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY._options = None
63
+ _ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY._serialized_options = b'8\001'
64
+ _ACTIONATTEMPT.fields_by_name['attempt']._options = None
65
+ _ACTIONATTEMPT.fields_by_name['attempt']._serialized_options = b'\372B\004*\002 \000'
66
+ _ACTIONEVENT.fields_by_name['id']._options = None
67
+ _ACTIONEVENT.fields_by_name['id']._serialized_options = b'\372B\005\212\001\002\020\001'
68
+ _ACTIONEVENT.fields_by_name['attempt']._options = None
69
+ _ACTIONEVENT.fields_by_name['attempt']._serialized_options = b'\372B\004*\002 \000'
70
+ _globals['_PHASE']._serialized_start=4899
71
+ _globals['_PHASE']._serialized_end=5102
72
+ _globals['_RUNIDENTIFIER']._serialized_start=284
73
+ _globals['_RUNIDENTIFIER']._serialized_end=431
74
+ _globals['_LABELS']._serialized_start=434
75
+ _globals['_LABELS']._serialized_end=564
76
+ _globals['_LABELS_VALUESENTRY']._serialized_start=507
77
+ _globals['_LABELS_VALUESENTRY']._serialized_end=564
78
+ _globals['_ANNOTATIONS']._serialized_start=567
79
+ _globals['_ANNOTATIONS']._serialized_end=707
80
+ _globals['_ANNOTATIONS_VALUESENTRY']._serialized_start=507
81
+ _globals['_ANNOTATIONS_VALUESENTRY']._serialized_end=564
82
+ _globals['_ENVS']._serialized_start=709
83
+ _globals['_ENVS']._serialized_end=768
84
+ _globals['_RUNSPEC']._serialized_start=771
85
+ _globals['_RUNSPEC']._serialized_end=1049
86
+ _globals['_RUN']._serialized_start=1051
87
+ _globals['_RUN']._serialized_end=1107
88
+ _globals['_RUNDETAILS']._serialized_start=1109
89
+ _globals['_RUNDETAILS']._serialized_end=1234
90
+ _globals['_ACTIONIDENTIFIER']._serialized_start=1236
91
+ _globals['_ACTIONIDENTIFIER']._serialized_end=1347
92
+ _globals['_TASKACTIONMETADATA']._serialized_start=1350
93
+ _globals['_TASKACTIONMETADATA']._serialized_end=1481
94
+ _globals['_TRACEACTIONMETADATA']._serialized_start=1483
95
+ _globals['_TRACEACTIONMETADATA']._serialized_end=1524
96
+ _globals['_CONDITIONACTIONMETADATA']._serialized_start=1527
97
+ _globals['_CONDITIONACTIONMETADATA']._serialized_end=1686
98
+ _globals['_ACTIONMETADATA']._serialized_start=1689
99
+ _globals['_ACTIONMETADATA']._serialized_end=2058
100
+ _globals['_ACTIONSTATUS']._serialized_start=2061
101
+ _globals['_ACTIONSTATUS']._serialized_end=2292
102
+ _globals['_ACTION']._serialized_start=2295
103
+ _globals['_ACTION']._serialized_end=2476
104
+ _globals['_ENRICHEDACTION']._serialized_start=2479
105
+ _globals['_ENRICHEDACTION']._serialized_end=2765
106
+ _globals['_ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY']._serialized_start=2695
107
+ _globals['_ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY']._serialized_end=2765
108
+ _globals['_ERRORINFO']._serialized_start=2768
109
+ _globals['_ERRORINFO']._serialized_end=2922
110
+ _globals['_ERRORINFO_KIND']._serialized_start=2862
111
+ _globals['_ERRORINFO_KIND']._serialized_end=2922
112
+ _globals['_ABORTINFO']._serialized_start=2924
113
+ _globals['_ABORTINFO']._serialized_end=3025
114
+ _globals['_ACTIONDETAILS']._serialized_start=3028
115
+ _globals['_ACTIONDETAILS']._serialized_end=3489
116
+ _globals['_ACTIONATTEMPT']._serialized_start=3492
117
+ _globals['_ACTIONATTEMPT']._serialized_end=3956
118
+ _globals['_ACTIONEVENT']._serialized_start=3959
119
+ _globals['_ACTIONEVENT']._serialized_end=4620
120
+ _globals['_NAMEDLITERAL']._serialized_start=4622
121
+ _globals['_NAMEDLITERAL']._serialized_end=4702
122
+ _globals['_OUTPUTREFERENCES']._serialized_start=4704
123
+ _globals['_OUTPUTREFERENCES']._serialized_end=4753
124
+ _globals['_INPUTS']._serialized_start=4755
125
+ _globals['_INPUTS']._serialized_end=4824
126
+ _globals['_OUTPUTS']._serialized_start=4826
127
+ _globals['_OUTPUTS']._serialized_end=4896
128
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,314 @@
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 execution_pb2 as _execution_pb2
4
+ from flyteidl.core import literals_pb2 as _literals_pb2
5
+ from google.protobuf import timestamp_pb2 as _timestamp_pb2
6
+ from google.protobuf import wrappers_pb2 as _wrappers_pb2
7
+ from flyte._protos.validate.validate import validate_pb2 as _validate_pb2
8
+ from flyte._protos.workflow import task_definition_pb2 as _task_definition_pb2
9
+ from google.protobuf.internal import containers as _containers
10
+ from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
11
+ from google.protobuf import descriptor as _descriptor
12
+ from google.protobuf import message as _message
13
+ from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
14
+
15
+ DESCRIPTOR: _descriptor.FileDescriptor
16
+
17
+ class Phase(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
18
+ __slots__ = []
19
+ PHASE_UNSPECIFIED: _ClassVar[Phase]
20
+ PHASE_QUEUED: _ClassVar[Phase]
21
+ PHASE_WAITING_FOR_RESOURCES: _ClassVar[Phase]
22
+ PHASE_INITIALIZING: _ClassVar[Phase]
23
+ PHASE_RUNNING: _ClassVar[Phase]
24
+ PHASE_SUCCEEDED: _ClassVar[Phase]
25
+ PHASE_FAILED: _ClassVar[Phase]
26
+ PHASE_ABORTED: _ClassVar[Phase]
27
+ PHASE_TIMED_OUT: _ClassVar[Phase]
28
+ PHASE_UNSPECIFIED: Phase
29
+ PHASE_QUEUED: Phase
30
+ PHASE_WAITING_FOR_RESOURCES: Phase
31
+ PHASE_INITIALIZING: Phase
32
+ PHASE_RUNNING: Phase
33
+ PHASE_SUCCEEDED: Phase
34
+ PHASE_FAILED: Phase
35
+ PHASE_ABORTED: Phase
36
+ PHASE_TIMED_OUT: Phase
37
+
38
+ class RunIdentifier(_message.Message):
39
+ __slots__ = ["org", "project", "domain", "name"]
40
+ ORG_FIELD_NUMBER: _ClassVar[int]
41
+ PROJECT_FIELD_NUMBER: _ClassVar[int]
42
+ DOMAIN_FIELD_NUMBER: _ClassVar[int]
43
+ NAME_FIELD_NUMBER: _ClassVar[int]
44
+ org: str
45
+ project: str
46
+ domain: str
47
+ name: str
48
+ def __init__(self, org: _Optional[str] = ..., project: _Optional[str] = ..., domain: _Optional[str] = ..., name: _Optional[str] = ...) -> None: ...
49
+
50
+ 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"]
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
+ labels: Labels
90
+ annotations: Annotations
91
+ envs: Envs
92
+ interruptible: _wrappers_pb2.BoolValue
93
+ overwrite_cache: bool
94
+ 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 = ...) -> None: ...
95
+
96
+ class Run(_message.Message):
97
+ __slots__ = ["action"]
98
+ ACTION_FIELD_NUMBER: _ClassVar[int]
99
+ action: Action
100
+ def __init__(self, action: _Optional[_Union[Action, _Mapping]] = ...) -> None: ...
101
+
102
+ class RunDetails(_message.Message):
103
+ __slots__ = ["run_spec", "action"]
104
+ RUN_SPEC_FIELD_NUMBER: _ClassVar[int]
105
+ ACTION_FIELD_NUMBER: _ClassVar[int]
106
+ run_spec: RunSpec
107
+ action: ActionDetails
108
+ def __init__(self, run_spec: _Optional[_Union[RunSpec, _Mapping]] = ..., action: _Optional[_Union[ActionDetails, _Mapping]] = ...) -> None: ...
109
+
110
+ class ActionIdentifier(_message.Message):
111
+ __slots__ = ["run", "name"]
112
+ RUN_FIELD_NUMBER: _ClassVar[int]
113
+ NAME_FIELD_NUMBER: _ClassVar[int]
114
+ run: RunIdentifier
115
+ name: str
116
+ def __init__(self, run: _Optional[_Union[RunIdentifier, _Mapping]] = ..., name: _Optional[str] = ...) -> None: ...
117
+
118
+ class TaskActionMetadata(_message.Message):
119
+ __slots__ = ["id", "task_type", "short_name"]
120
+ ID_FIELD_NUMBER: _ClassVar[int]
121
+ TASK_TYPE_FIELD_NUMBER: _ClassVar[int]
122
+ SHORT_NAME_FIELD_NUMBER: _ClassVar[int]
123
+ id: _task_definition_pb2.TaskIdentifier
124
+ task_type: str
125
+ short_name: str
126
+ def __init__(self, id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ..., task_type: _Optional[str] = ..., short_name: _Optional[str] = ...) -> None: ...
127
+
128
+ class TraceActionMetadata(_message.Message):
129
+ __slots__ = ["name"]
130
+ NAME_FIELD_NUMBER: _ClassVar[int]
131
+ name: str
132
+ def __init__(self, name: _Optional[str] = ...) -> None: ...
133
+
134
+ class ConditionActionMetadata(_message.Message):
135
+ __slots__ = ["name", "run_id", "action_id"]
136
+ NAME_FIELD_NUMBER: _ClassVar[int]
137
+ RUN_ID_FIELD_NUMBER: _ClassVar[int]
138
+ ACTION_ID_FIELD_NUMBER: _ClassVar[int]
139
+ GLOBAL_FIELD_NUMBER: _ClassVar[int]
140
+ name: str
141
+ run_id: str
142
+ action_id: str
143
+ def __init__(self, name: _Optional[str] = ..., run_id: _Optional[str] = ..., action_id: _Optional[str] = ..., **kwargs) -> None: ...
144
+
145
+ class ActionMetadata(_message.Message):
146
+ __slots__ = ["parent", "group", "executed_by", "task", "trace", "condition"]
147
+ PARENT_FIELD_NUMBER: _ClassVar[int]
148
+ GROUP_FIELD_NUMBER: _ClassVar[int]
149
+ EXECUTED_BY_FIELD_NUMBER: _ClassVar[int]
150
+ TASK_FIELD_NUMBER: _ClassVar[int]
151
+ TRACE_FIELD_NUMBER: _ClassVar[int]
152
+ CONDITION_FIELD_NUMBER: _ClassVar[int]
153
+ parent: str
154
+ group: str
155
+ executed_by: _identity_pb2.EnrichedIdentity
156
+ task: TaskActionMetadata
157
+ trace: TraceActionMetadata
158
+ condition: ConditionActionMetadata
159
+ 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]] = ...) -> None: ...
160
+
161
+ class ActionStatus(_message.Message):
162
+ __slots__ = ["phase", "start_time", "end_time", "attempts"]
163
+ PHASE_FIELD_NUMBER: _ClassVar[int]
164
+ START_TIME_FIELD_NUMBER: _ClassVar[int]
165
+ END_TIME_FIELD_NUMBER: _ClassVar[int]
166
+ ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
167
+ phase: Phase
168
+ start_time: _timestamp_pb2.Timestamp
169
+ end_time: _timestamp_pb2.Timestamp
170
+ attempts: int
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] = ...) -> 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: ActionIdentifier
179
+ metadata: ActionMetadata
180
+ status: ActionStatus
181
+ def __init__(self, id: _Optional[_Union[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", "resolved_task_spec", "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
+ RESOLVED_TASK_SPEC_FIELD_NUMBER: _ClassVar[int]
232
+ ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
233
+ id: ActionIdentifier
234
+ metadata: ActionMetadata
235
+ status: ActionStatus
236
+ error_info: ErrorInfo
237
+ abort_info: AbortInfo
238
+ resolved_task_spec: _task_definition_pb2.TaskSpec
239
+ attempts: _containers.RepeatedCompositeFieldContainer[ActionAttempt]
240
+ def __init__(self, id: _Optional[_Union[ActionIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[ActionMetadata, _Mapping]] = ..., status: _Optional[_Union[ActionStatus, _Mapping]] = ..., error_info: _Optional[_Union[ErrorInfo, _Mapping]] = ..., abort_info: _Optional[_Union[AbortInfo, _Mapping]] = ..., resolved_task_spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ..., attempts: _Optional[_Iterable[_Union[ActionAttempt, _Mapping]]] = ...) -> None: ...
241
+
242
+ class ActionAttempt(_message.Message):
243
+ __slots__ = ["phase", "start_time", "end_time", "error_info", "attempt", "log_info", "outputs", "logs_available"]
244
+ PHASE_FIELD_NUMBER: _ClassVar[int]
245
+ START_TIME_FIELD_NUMBER: _ClassVar[int]
246
+ END_TIME_FIELD_NUMBER: _ClassVar[int]
247
+ ERROR_INFO_FIELD_NUMBER: _ClassVar[int]
248
+ ATTEMPT_FIELD_NUMBER: _ClassVar[int]
249
+ LOG_INFO_FIELD_NUMBER: _ClassVar[int]
250
+ OUTPUTS_FIELD_NUMBER: _ClassVar[int]
251
+ LOGS_AVAILABLE_FIELD_NUMBER: _ClassVar[int]
252
+ phase: Phase
253
+ start_time: _timestamp_pb2.Timestamp
254
+ end_time: _timestamp_pb2.Timestamp
255
+ error_info: ErrorInfo
256
+ attempt: int
257
+ log_info: _containers.RepeatedCompositeFieldContainer[_execution_pb2.TaskLog]
258
+ outputs: OutputReferences
259
+ logs_available: bool
260
+ def __init__(self, phase: _Optional[_Union[Phase, str]] = ..., start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., error_info: _Optional[_Union[ErrorInfo, _Mapping]] = ..., attempt: _Optional[int] = ..., log_info: _Optional[_Iterable[_Union[_execution_pb2.TaskLog, _Mapping]]] = ..., outputs: _Optional[_Union[OutputReferences, _Mapping]] = ..., logs_available: bool = ...) -> None: ...
261
+
262
+ class ActionEvent(_message.Message):
263
+ __slots__ = ["id", "attempt", "phase", "version", "start_time", "updated_time", "end_time", "error_info", "log_info", "log_context", "cluster", "outputs"]
264
+ ID_FIELD_NUMBER: _ClassVar[int]
265
+ ATTEMPT_FIELD_NUMBER: _ClassVar[int]
266
+ PHASE_FIELD_NUMBER: _ClassVar[int]
267
+ VERSION_FIELD_NUMBER: _ClassVar[int]
268
+ START_TIME_FIELD_NUMBER: _ClassVar[int]
269
+ UPDATED_TIME_FIELD_NUMBER: _ClassVar[int]
270
+ END_TIME_FIELD_NUMBER: _ClassVar[int]
271
+ ERROR_INFO_FIELD_NUMBER: _ClassVar[int]
272
+ LOG_INFO_FIELD_NUMBER: _ClassVar[int]
273
+ LOG_CONTEXT_FIELD_NUMBER: _ClassVar[int]
274
+ CLUSTER_FIELD_NUMBER: _ClassVar[int]
275
+ OUTPUTS_FIELD_NUMBER: _ClassVar[int]
276
+ id: ActionIdentifier
277
+ attempt: int
278
+ phase: Phase
279
+ version: int
280
+ start_time: _timestamp_pb2.Timestamp
281
+ updated_time: _timestamp_pb2.Timestamp
282
+ end_time: _timestamp_pb2.Timestamp
283
+ error_info: ErrorInfo
284
+ log_info: _containers.RepeatedCompositeFieldContainer[_execution_pb2.TaskLog]
285
+ log_context: _execution_pb2.LogContext
286
+ cluster: str
287
+ outputs: OutputReferences
288
+ def __init__(self, id: _Optional[_Union[ActionIdentifier, _Mapping]] = ..., attempt: _Optional[int] = ..., phase: _Optional[_Union[Phase, str]] = ..., version: _Optional[int] = ..., start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., updated_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., error_info: _Optional[_Union[ErrorInfo, _Mapping]] = ..., log_info: _Optional[_Iterable[_Union[_execution_pb2.TaskLog, _Mapping]]] = ..., log_context: _Optional[_Union[_execution_pb2.LogContext, _Mapping]] = ..., cluster: _Optional[str] = ..., outputs: _Optional[_Union[OutputReferences, _Mapping]] = ...) -> None: ...
289
+
290
+ class NamedLiteral(_message.Message):
291
+ __slots__ = ["name", "value"]
292
+ NAME_FIELD_NUMBER: _ClassVar[int]
293
+ VALUE_FIELD_NUMBER: _ClassVar[int]
294
+ name: str
295
+ value: _literals_pb2.Literal
296
+ def __init__(self, name: _Optional[str] = ..., value: _Optional[_Union[_literals_pb2.Literal, _Mapping]] = ...) -> None: ...
297
+
298
+ class OutputReferences(_message.Message):
299
+ __slots__ = ["output_uri"]
300
+ OUTPUT_URI_FIELD_NUMBER: _ClassVar[int]
301
+ output_uri: str
302
+ def __init__(self, output_uri: _Optional[str] = ...) -> None: ...
303
+
304
+ class Inputs(_message.Message):
305
+ __slots__ = ["literals"]
306
+ LITERALS_FIELD_NUMBER: _ClassVar[int]
307
+ literals: _containers.RepeatedCompositeFieldContainer[NamedLiteral]
308
+ def __init__(self, literals: _Optional[_Iterable[_Union[NamedLiteral, _Mapping]]] = ...) -> None: ...
309
+
310
+ class Outputs(_message.Message):
311
+ __slots__ = ["literals"]
312
+ LITERALS_FIELD_NUMBER: _ClassVar[int]
313
+ literals: _containers.RepeatedCompositeFieldContainer[NamedLiteral]
314
+ def __init__(self, literals: _Optional[_Iterable[_Union[NamedLiteral, _Mapping]]] = ...) -> None: ...
@@ -0,0 +1,4 @@
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
+
@@ -0,0 +1,41 @@
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.logs.dataplane import payload_pb2 as logs_dot_dataplane_dot_payload__pb2
15
+ from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
16
+ from flyte._protos.workflow import run_definition_pb2 as workflow_dot_run__definition__pb2
17
+
18
+
19
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1fworkflow/run_logs_service.proto\x12\x11\x63loudidl.workflow\x1a\x1clogs/dataplane/payload.proto\x1a\x17validate/validate.proto\x1a\x1dworkflow/run_definition.proto\"\x80\x01\n\x0fTailLogsRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\x12!\n\x07\x61ttempt\x18\x02 \x01(\rB\x07\xfa\x42\x04*\x02 \x00R\x07\x61ttempt\"\x90\x01\n\x10TailLogsResponse\x12<\n\x04logs\x18\x01 \x03(\x0b\x32(.cloudidl.workflow.TailLogsResponse.LogsR\x04logs\x1a>\n\x04Logs\x12\x36\n\x05lines\x18\x01 \x03(\x0b\x32 .cloudidl.logs.dataplane.LogLineR\x05lines2l\n\x0eRunLogsService\x12Z\n\x08TailLogs\x12\".cloudidl.workflow.TailLogsRequest\x1a#.cloudidl.workflow.TailLogsResponse\"\x03\x90\x02\x01\x30\x01\x42\xc0\x01\n\x15\x63om.cloudidl.workflowB\x13RunLogsServiceProtoH\x02P\x01Z+github.com/unionai/cloud/gen/pb-go/workflow\xa2\x02\x03\x43WX\xaa\x02\x11\x43loudidl.Workflow\xca\x02\x11\x43loudidl\\Workflow\xe2\x02\x1d\x43loudidl\\Workflow\\GPBMetadata\xea\x02\x12\x43loudidl::Workflowb\x06proto3')
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=141
34
+ _globals['_TAILLOGSREQUEST']._serialized_end=269
35
+ _globals['_TAILLOGSRESPONSE']._serialized_start=272
36
+ _globals['_TAILLOGSRESPONSE']._serialized_end=416
37
+ _globals['_TAILLOGSRESPONSE_LOGS']._serialized_start=354
38
+ _globals['_TAILLOGSRESPONSE_LOGS']._serialized_end=416
39
+ _globals['_RUNLOGSSERVICE']._serialized_start=418
40
+ _globals['_RUNLOGSSERVICE']._serialized_end=526
41
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,28 @@
1
+ from flyte._protos.logs.dataplane import payload_pb2 as _payload_pb2
2
+ from flyte._protos.validate.validate import validate_pb2 as _validate_pb2
3
+ from flyte._protos.workflow import run_definition_pb2 as _run_definition_pb2
4
+ 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: _run_definition_pb2.ActionIdentifier
16
+ attempt: int
17
+ def __init__(self, action_id: _Optional[_Union[_run_definition_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: ...
@@ -0,0 +1,69 @@
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)