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,129 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # source: workflow/run_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.common import list_pb2 as common_dot_list__pb2
16
+ from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
17
+ from flyte._protos.workflow import run_definition_pb2 as workflow_dot_run__definition__pb2
18
+ from flyte._protos.workflow import task_definition_pb2 as workflow_dot_task__definition__pb2
19
+
20
+
21
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1aworkflow/run_service.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x11\x63ommon/list.proto\x1a\x17validate/validate.proto\x1a\x1dworkflow/run_definition.proto\x1a\x1eworkflow/task_definition.proto\"\xb6\x03\n\x10\x43reateRunRequest\x12\x43\n\x06run_id\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x05runId\x12M\n\nproject_id\x18\x06 \x01(\x0b\x32\".cloudidl.common.ProjectIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tprojectId\x12\x46\n\x07task_id\x18\x02 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x01R\x06taskId\x12\x44\n\ttask_spec\x18\x03 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x01R\x08taskSpec\x12\x31\n\x06inputs\x18\x04 \x01(\x0b\x32\x19.cloudidl.workflow.InputsR\x06inputs\x12\x35\n\x08run_spec\x18\x05 \x01(\x0b\x32\x1a.cloudidl.workflow.RunSpecR\x07runSpecB\t\n\x02id\x12\x03\xf8\x42\x01\x42\x0b\n\x04task\x12\x03\xf8\x42\x01\"=\n\x11\x43reateRunResponse\x12(\n\x03run\x18\x01 \x01(\x0b\x32\x16.cloudidl.workflow.RunR\x03run\"T\n\x0f\x41\x62ortRunRequest\x12\x41\n\x06run_id\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\"\x12\n\x10\x41\x62ortRunResponse\"Y\n\x14GetRunDetailsRequest\x12\x41\n\x06run_id\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\"P\n\x15GetRunDetailsResponse\x12\x37\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32\x1d.cloudidl.workflow.RunDetailsR\x07\x64\x65tails\"[\n\x16WatchRunDetailsRequest\x12\x41\n\x06run_id\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\"R\n\x17WatchRunDetailsResponse\x12\x37\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32\x1d.cloudidl.workflow.RunDetailsR\x07\x64\x65tails\"e\n\x17GetActionDetailsRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\"V\n\x18GetActionDetailsResponse\x12:\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.ActionDetailsR\x07\x64\x65tails\"g\n\x19WatchActionDetailsRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\"X\n\x1aWatchActionDetailsResponse\x12:\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.ActionDetailsR\x07\x64\x65tails\"b\n\x14GetActionDataRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\"\x80\x01\n\x15GetActionDataResponse\x12\x31\n\x06inputs\x18\x01 \x01(\x0b\x32\x19.cloudidl.workflow.InputsR\x06inputs\x12\x34\n\x07outputs\x18\x02 \x01(\x0b\x32\x1a.cloudidl.workflow.OutputsR\x07outputs\"\xd2\x01\n\x0fListRunsRequest\x12\x36\n\x07request\x18\x01 \x01(\x0b\x32\x1c.cloudidl.common.ListRequestR\x07request\x12\x1b\n\x03org\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x03org\x12M\n\nproject_id\x18\x04 \x01(\x0b\x32\".cloudidl.common.ProjectIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tprojectIdB\x0f\n\x08scope_by\x12\x03\xf8\x42\x01J\x04\x08\x03\x10\x04J\x04\x08\x05\x10\x06\"T\n\x10ListRunsResponse\x12*\n\x04runs\x18\x01 \x03(\x0b\x32\x16.cloudidl.workflow.RunR\x04runs\x12\x14\n\x05token\x18\x02 \x01(\tR\x05token\"\xa4\x02\n\x10WatchRunsRequest\x12\x1b\n\x03org\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x03org\x12M\n\ncluster_id\x18\x03 \x01(\x0b\x32\".cloudidl.common.ClusterIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tclusterId\x12M\n\nproject_id\x18\x04 \x01(\x0b\x32\".cloudidl.common.ProjectIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tprojectId\x12\x46\n\x07task_id\x18\x05 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x06taskIdB\r\n\x06target\x12\x03\xf8\x42\x01\"?\n\x11WatchRunsResponse\x12*\n\x04runs\x18\x01 \x03(\x0b\x32\x16.cloudidl.workflow.RunR\x04runs\"\x8f\x01\n\x12ListActionsRequest\x12\x36\n\x07request\x18\x01 \x01(\x0b\x32\x1c.cloudidl.common.ListRequestR\x07request\x12\x41\n\x06run_id\x18\x02 \x01(\x0b\x32 .cloudidl.workflow.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\"`\n\x13ListActionsResponse\x12\x33\n\x07\x61\x63tions\x18\x01 \x03(\x0b\x32\x19.cloudidl.workflow.ActionR\x07\x61\x63tions\x12\x14\n\x05token\x18\x02 \x01(\tR\x05token\"\x89\x01\n\x13WatchActionsRequest\x12\x41\n\x06run_id\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\x12/\n\x06\x66ilter\x18\x02 \x03(\x0b\x32\x17.cloudidl.common.FilterR\x06\x66ilter\"d\n\x14WatchActionsResponse\x12L\n\x10\x65nriched_actions\x18\x01 \x03(\x0b\x32!.cloudidl.workflow.EnrichedActionR\x0f\x65nrichedActions2\xe4\x08\n\nRunService\x12X\n\tCreateRun\x12#.cloudidl.workflow.CreateRunRequest\x1a$.cloudidl.workflow.CreateRunResponse\"\x00\x12U\n\x08\x41\x62ortRun\x12\".cloudidl.workflow.AbortRunRequest\x1a#.cloudidl.workflow.AbortRunResponse\"\x00\x12g\n\rGetRunDetails\x12\'.cloudidl.workflow.GetRunDetailsRequest\x1a(.cloudidl.workflow.GetRunDetailsResponse\"\x03\x90\x02\x01\x12l\n\x0fWatchRunDetails\x12).cloudidl.workflow.WatchRunDetailsRequest\x1a*.cloudidl.workflow.WatchRunDetailsResponse\"\x00\x30\x01\x12p\n\x10GetActionDetails\x12*.cloudidl.workflow.GetActionDetailsRequest\x1a+.cloudidl.workflow.GetActionDetailsResponse\"\x03\x90\x02\x01\x12u\n\x12WatchActionDetails\x12,.cloudidl.workflow.WatchActionDetailsRequest\x1a-.cloudidl.workflow.WatchActionDetailsResponse\"\x00\x30\x01\x12g\n\rGetActionData\x12\'.cloudidl.workflow.GetActionDataRequest\x1a(.cloudidl.workflow.GetActionDataResponse\"\x03\x90\x02\x01\x12X\n\x08ListRuns\x12\".cloudidl.workflow.ListRunsRequest\x1a#.cloudidl.workflow.ListRunsResponse\"\x03\x90\x02\x01\x12Z\n\tWatchRuns\x12#.cloudidl.workflow.WatchRunsRequest\x1a$.cloudidl.workflow.WatchRunsResponse\"\x00\x30\x01\x12\x61\n\x0bListActions\x12%.cloudidl.workflow.ListActionsRequest\x1a&.cloudidl.workflow.ListActionsResponse\"\x03\x90\x02\x01\x12\x63\n\x0cWatchActions\x12&.cloudidl.workflow.WatchActionsRequest\x1a\'.cloudidl.workflow.WatchActionsResponse\"\x00\x30\x01\x42\xbc\x01\n\x15\x63om.cloudidl.workflowB\x0fRunServiceProtoH\x02P\x01Z+github.com/unionai/cloud/gen/pb-go/workflow\xa2\x02\x03\x43WX\xaa\x02\x11\x43loudidl.Workflow\xca\x02\x11\x43loudidl\\Workflow\xe2\x02\x1d\x43loudidl\\Workflow\\GPBMetadata\xea\x02\x12\x43loudidl::Workflowb\x06proto3')
22
+
23
+ _globals = globals()
24
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
25
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'workflow.run_service_pb2', _globals)
26
+ if _descriptor._USE_C_DESCRIPTORS == False:
27
+ DESCRIPTOR._options = None
28
+ DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\017RunServiceProtoH\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'
29
+ _CREATERUNREQUEST.oneofs_by_name['id']._options = None
30
+ _CREATERUNREQUEST.oneofs_by_name['id']._serialized_options = b'\370B\001'
31
+ _CREATERUNREQUEST.oneofs_by_name['task']._options = None
32
+ _CREATERUNREQUEST.oneofs_by_name['task']._serialized_options = b'\370B\001'
33
+ _CREATERUNREQUEST.fields_by_name['run_id']._options = None
34
+ _CREATERUNREQUEST.fields_by_name['run_id']._serialized_options = b'\372B\005\212\001\002\020\001'
35
+ _CREATERUNREQUEST.fields_by_name['project_id']._options = None
36
+ _CREATERUNREQUEST.fields_by_name['project_id']._serialized_options = b'\372B\005\212\001\002\020\001'
37
+ _CREATERUNREQUEST.fields_by_name['task_id']._options = None
38
+ _CREATERUNREQUEST.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
39
+ _CREATERUNREQUEST.fields_by_name['task_spec']._options = None
40
+ _CREATERUNREQUEST.fields_by_name['task_spec']._serialized_options = b'\372B\005\212\001\002\020\001'
41
+ _ABORTRUNREQUEST.fields_by_name['run_id']._options = None
42
+ _ABORTRUNREQUEST.fields_by_name['run_id']._serialized_options = b'\372B\005\212\001\002\020\001'
43
+ _GETRUNDETAILSREQUEST.fields_by_name['run_id']._options = None
44
+ _GETRUNDETAILSREQUEST.fields_by_name['run_id']._serialized_options = b'\372B\005\212\001\002\020\001'
45
+ _WATCHRUNDETAILSREQUEST.fields_by_name['run_id']._options = None
46
+ _WATCHRUNDETAILSREQUEST.fields_by_name['run_id']._serialized_options = b'\372B\005\212\001\002\020\001'
47
+ _GETACTIONDETAILSREQUEST.fields_by_name['action_id']._options = None
48
+ _GETACTIONDETAILSREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
49
+ _WATCHACTIONDETAILSREQUEST.fields_by_name['action_id']._options = None
50
+ _WATCHACTIONDETAILSREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
51
+ _GETACTIONDATAREQUEST.fields_by_name['action_id']._options = None
52
+ _GETACTIONDATAREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
53
+ _LISTRUNSREQUEST.oneofs_by_name['scope_by']._options = None
54
+ _LISTRUNSREQUEST.oneofs_by_name['scope_by']._serialized_options = b'\370B\001'
55
+ _LISTRUNSREQUEST.fields_by_name['org']._options = None
56
+ _LISTRUNSREQUEST.fields_by_name['org']._serialized_options = b'\372B\004r\002\020\001'
57
+ _LISTRUNSREQUEST.fields_by_name['project_id']._options = None
58
+ _LISTRUNSREQUEST.fields_by_name['project_id']._serialized_options = b'\372B\005\212\001\002\020\001'
59
+ _WATCHRUNSREQUEST.oneofs_by_name['target']._options = None
60
+ _WATCHRUNSREQUEST.oneofs_by_name['target']._serialized_options = b'\370B\001'
61
+ _WATCHRUNSREQUEST.fields_by_name['org']._options = None
62
+ _WATCHRUNSREQUEST.fields_by_name['org']._serialized_options = b'\372B\004r\002\020\001'
63
+ _WATCHRUNSREQUEST.fields_by_name['cluster_id']._options = None
64
+ _WATCHRUNSREQUEST.fields_by_name['cluster_id']._serialized_options = b'\372B\005\212\001\002\020\001'
65
+ _WATCHRUNSREQUEST.fields_by_name['project_id']._options = None
66
+ _WATCHRUNSREQUEST.fields_by_name['project_id']._serialized_options = b'\372B\005\212\001\002\020\001'
67
+ _WATCHRUNSREQUEST.fields_by_name['task_id']._options = None
68
+ _WATCHRUNSREQUEST.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
69
+ _LISTACTIONSREQUEST.fields_by_name['run_id']._options = None
70
+ _LISTACTIONSREQUEST.fields_by_name['run_id']._serialized_options = b'\372B\005\212\001\002\020\001'
71
+ _WATCHACTIONSREQUEST.fields_by_name['run_id']._options = None
72
+ _WATCHACTIONSREQUEST.fields_by_name['run_id']._serialized_options = b'\372B\005\212\001\002\020\001'
73
+ _RUNSERVICE.methods_by_name['GetRunDetails']._options = None
74
+ _RUNSERVICE.methods_by_name['GetRunDetails']._serialized_options = b'\220\002\001'
75
+ _RUNSERVICE.methods_by_name['GetActionDetails']._options = None
76
+ _RUNSERVICE.methods_by_name['GetActionDetails']._serialized_options = b'\220\002\001'
77
+ _RUNSERVICE.methods_by_name['GetActionData']._options = None
78
+ _RUNSERVICE.methods_by_name['GetActionData']._serialized_options = b'\220\002\001'
79
+ _RUNSERVICE.methods_by_name['ListRuns']._options = None
80
+ _RUNSERVICE.methods_by_name['ListRuns']._serialized_options = b'\220\002\001'
81
+ _RUNSERVICE.methods_by_name['ListActions']._options = None
82
+ _RUNSERVICE.methods_by_name['ListActions']._serialized_options = b'\220\002\001'
83
+ _globals['_CREATERUNREQUEST']._serialized_start=182
84
+ _globals['_CREATERUNREQUEST']._serialized_end=620
85
+ _globals['_CREATERUNRESPONSE']._serialized_start=622
86
+ _globals['_CREATERUNRESPONSE']._serialized_end=683
87
+ _globals['_ABORTRUNREQUEST']._serialized_start=685
88
+ _globals['_ABORTRUNREQUEST']._serialized_end=769
89
+ _globals['_ABORTRUNRESPONSE']._serialized_start=771
90
+ _globals['_ABORTRUNRESPONSE']._serialized_end=789
91
+ _globals['_GETRUNDETAILSREQUEST']._serialized_start=791
92
+ _globals['_GETRUNDETAILSREQUEST']._serialized_end=880
93
+ _globals['_GETRUNDETAILSRESPONSE']._serialized_start=882
94
+ _globals['_GETRUNDETAILSRESPONSE']._serialized_end=962
95
+ _globals['_WATCHRUNDETAILSREQUEST']._serialized_start=964
96
+ _globals['_WATCHRUNDETAILSREQUEST']._serialized_end=1055
97
+ _globals['_WATCHRUNDETAILSRESPONSE']._serialized_start=1057
98
+ _globals['_WATCHRUNDETAILSRESPONSE']._serialized_end=1139
99
+ _globals['_GETACTIONDETAILSREQUEST']._serialized_start=1141
100
+ _globals['_GETACTIONDETAILSREQUEST']._serialized_end=1242
101
+ _globals['_GETACTIONDETAILSRESPONSE']._serialized_start=1244
102
+ _globals['_GETACTIONDETAILSRESPONSE']._serialized_end=1330
103
+ _globals['_WATCHACTIONDETAILSREQUEST']._serialized_start=1332
104
+ _globals['_WATCHACTIONDETAILSREQUEST']._serialized_end=1435
105
+ _globals['_WATCHACTIONDETAILSRESPONSE']._serialized_start=1437
106
+ _globals['_WATCHACTIONDETAILSRESPONSE']._serialized_end=1525
107
+ _globals['_GETACTIONDATAREQUEST']._serialized_start=1527
108
+ _globals['_GETACTIONDATAREQUEST']._serialized_end=1625
109
+ _globals['_GETACTIONDATARESPONSE']._serialized_start=1628
110
+ _globals['_GETACTIONDATARESPONSE']._serialized_end=1756
111
+ _globals['_LISTRUNSREQUEST']._serialized_start=1759
112
+ _globals['_LISTRUNSREQUEST']._serialized_end=1969
113
+ _globals['_LISTRUNSRESPONSE']._serialized_start=1971
114
+ _globals['_LISTRUNSRESPONSE']._serialized_end=2055
115
+ _globals['_WATCHRUNSREQUEST']._serialized_start=2058
116
+ _globals['_WATCHRUNSREQUEST']._serialized_end=2350
117
+ _globals['_WATCHRUNSRESPONSE']._serialized_start=2352
118
+ _globals['_WATCHRUNSRESPONSE']._serialized_end=2415
119
+ _globals['_LISTACTIONSREQUEST']._serialized_start=2418
120
+ _globals['_LISTACTIONSREQUEST']._serialized_end=2561
121
+ _globals['_LISTACTIONSRESPONSE']._serialized_start=2563
122
+ _globals['_LISTACTIONSRESPONSE']._serialized_end=2659
123
+ _globals['_WATCHACTIONSREQUEST']._serialized_start=2662
124
+ _globals['_WATCHACTIONSREQUEST']._serialized_end=2799
125
+ _globals['_WATCHACTIONSRESPONSE']._serialized_start=2801
126
+ _globals['_WATCHACTIONSRESPONSE']._serialized_end=2901
127
+ _globals['_RUNSERVICE']._serialized_start=2904
128
+ _globals['_RUNSERVICE']._serialized_end=4028
129
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,171 @@
1
+ from flyte._protos.common import identifier_pb2 as _identifier_pb2
2
+ from flyte._protos.common import list_pb2 as _list_pb2
3
+ from flyte._protos.validate.validate import validate_pb2 as _validate_pb2
4
+ from flyte._protos.workflow import run_definition_pb2 as _run_definition_pb2
5
+ from flyte._protos.workflow import task_definition_pb2 as _task_definition_pb2
6
+ from google.protobuf.internal import containers as _containers
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import message as _message
9
+ from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
10
+
11
+ DESCRIPTOR: _descriptor.FileDescriptor
12
+
13
+ class CreateRunRequest(_message.Message):
14
+ __slots__ = ["run_id", "project_id", "task_id", "task_spec", "inputs", "run_spec"]
15
+ RUN_ID_FIELD_NUMBER: _ClassVar[int]
16
+ PROJECT_ID_FIELD_NUMBER: _ClassVar[int]
17
+ TASK_ID_FIELD_NUMBER: _ClassVar[int]
18
+ TASK_SPEC_FIELD_NUMBER: _ClassVar[int]
19
+ INPUTS_FIELD_NUMBER: _ClassVar[int]
20
+ RUN_SPEC_FIELD_NUMBER: _ClassVar[int]
21
+ run_id: _run_definition_pb2.RunIdentifier
22
+ project_id: _identifier_pb2.ProjectIdentifier
23
+ task_id: _task_definition_pb2.TaskIdentifier
24
+ task_spec: _task_definition_pb2.TaskSpec
25
+ inputs: _run_definition_pb2.Inputs
26
+ run_spec: _run_definition_pb2.RunSpec
27
+ def __init__(self, run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ..., project_id: _Optional[_Union[_identifier_pb2.ProjectIdentifier, _Mapping]] = ..., task_id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ..., task_spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ..., inputs: _Optional[_Union[_run_definition_pb2.Inputs, _Mapping]] = ..., run_spec: _Optional[_Union[_run_definition_pb2.RunSpec, _Mapping]] = ...) -> None: ...
28
+
29
+ class CreateRunResponse(_message.Message):
30
+ __slots__ = ["run"]
31
+ RUN_FIELD_NUMBER: _ClassVar[int]
32
+ run: _run_definition_pb2.Run
33
+ def __init__(self, run: _Optional[_Union[_run_definition_pb2.Run, _Mapping]] = ...) -> None: ...
34
+
35
+ class AbortRunRequest(_message.Message):
36
+ __slots__ = ["run_id"]
37
+ RUN_ID_FIELD_NUMBER: _ClassVar[int]
38
+ run_id: _run_definition_pb2.RunIdentifier
39
+ def __init__(self, run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
40
+
41
+ class AbortRunResponse(_message.Message):
42
+ __slots__ = []
43
+ def __init__(self) -> None: ...
44
+
45
+ class GetRunDetailsRequest(_message.Message):
46
+ __slots__ = ["run_id"]
47
+ RUN_ID_FIELD_NUMBER: _ClassVar[int]
48
+ run_id: _run_definition_pb2.RunIdentifier
49
+ def __init__(self, run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
50
+
51
+ class GetRunDetailsResponse(_message.Message):
52
+ __slots__ = ["details"]
53
+ DETAILS_FIELD_NUMBER: _ClassVar[int]
54
+ details: _run_definition_pb2.RunDetails
55
+ def __init__(self, details: _Optional[_Union[_run_definition_pb2.RunDetails, _Mapping]] = ...) -> None: ...
56
+
57
+ class WatchRunDetailsRequest(_message.Message):
58
+ __slots__ = ["run_id"]
59
+ RUN_ID_FIELD_NUMBER: _ClassVar[int]
60
+ run_id: _run_definition_pb2.RunIdentifier
61
+ def __init__(self, run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
62
+
63
+ class WatchRunDetailsResponse(_message.Message):
64
+ __slots__ = ["details"]
65
+ DETAILS_FIELD_NUMBER: _ClassVar[int]
66
+ details: _run_definition_pb2.RunDetails
67
+ def __init__(self, details: _Optional[_Union[_run_definition_pb2.RunDetails, _Mapping]] = ...) -> None: ...
68
+
69
+ class GetActionDetailsRequest(_message.Message):
70
+ __slots__ = ["action_id"]
71
+ ACTION_ID_FIELD_NUMBER: _ClassVar[int]
72
+ action_id: _run_definition_pb2.ActionIdentifier
73
+ def __init__(self, action_id: _Optional[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
74
+
75
+ class GetActionDetailsResponse(_message.Message):
76
+ __slots__ = ["details"]
77
+ DETAILS_FIELD_NUMBER: _ClassVar[int]
78
+ details: _run_definition_pb2.ActionDetails
79
+ def __init__(self, details: _Optional[_Union[_run_definition_pb2.ActionDetails, _Mapping]] = ...) -> None: ...
80
+
81
+ class WatchActionDetailsRequest(_message.Message):
82
+ __slots__ = ["action_id"]
83
+ ACTION_ID_FIELD_NUMBER: _ClassVar[int]
84
+ action_id: _run_definition_pb2.ActionIdentifier
85
+ def __init__(self, action_id: _Optional[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
86
+
87
+ class WatchActionDetailsResponse(_message.Message):
88
+ __slots__ = ["details"]
89
+ DETAILS_FIELD_NUMBER: _ClassVar[int]
90
+ details: _run_definition_pb2.ActionDetails
91
+ def __init__(self, details: _Optional[_Union[_run_definition_pb2.ActionDetails, _Mapping]] = ...) -> None: ...
92
+
93
+ class GetActionDataRequest(_message.Message):
94
+ __slots__ = ["action_id"]
95
+ ACTION_ID_FIELD_NUMBER: _ClassVar[int]
96
+ action_id: _run_definition_pb2.ActionIdentifier
97
+ def __init__(self, action_id: _Optional[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
98
+
99
+ class GetActionDataResponse(_message.Message):
100
+ __slots__ = ["inputs", "outputs"]
101
+ INPUTS_FIELD_NUMBER: _ClassVar[int]
102
+ OUTPUTS_FIELD_NUMBER: _ClassVar[int]
103
+ inputs: _run_definition_pb2.Inputs
104
+ outputs: _run_definition_pb2.Outputs
105
+ def __init__(self, inputs: _Optional[_Union[_run_definition_pb2.Inputs, _Mapping]] = ..., outputs: _Optional[_Union[_run_definition_pb2.Outputs, _Mapping]] = ...) -> None: ...
106
+
107
+ class ListRunsRequest(_message.Message):
108
+ __slots__ = ["request", "org", "project_id"]
109
+ REQUEST_FIELD_NUMBER: _ClassVar[int]
110
+ ORG_FIELD_NUMBER: _ClassVar[int]
111
+ PROJECT_ID_FIELD_NUMBER: _ClassVar[int]
112
+ request: _list_pb2.ListRequest
113
+ org: str
114
+ project_id: _identifier_pb2.ProjectIdentifier
115
+ def __init__(self, request: _Optional[_Union[_list_pb2.ListRequest, _Mapping]] = ..., org: _Optional[str] = ..., project_id: _Optional[_Union[_identifier_pb2.ProjectIdentifier, _Mapping]] = ...) -> None: ...
116
+
117
+ class ListRunsResponse(_message.Message):
118
+ __slots__ = ["runs", "token"]
119
+ RUNS_FIELD_NUMBER: _ClassVar[int]
120
+ TOKEN_FIELD_NUMBER: _ClassVar[int]
121
+ runs: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.Run]
122
+ token: str
123
+ def __init__(self, runs: _Optional[_Iterable[_Union[_run_definition_pb2.Run, _Mapping]]] = ..., token: _Optional[str] = ...) -> None: ...
124
+
125
+ class WatchRunsRequest(_message.Message):
126
+ __slots__ = ["org", "cluster_id", "project_id", "task_id"]
127
+ ORG_FIELD_NUMBER: _ClassVar[int]
128
+ CLUSTER_ID_FIELD_NUMBER: _ClassVar[int]
129
+ PROJECT_ID_FIELD_NUMBER: _ClassVar[int]
130
+ TASK_ID_FIELD_NUMBER: _ClassVar[int]
131
+ org: str
132
+ cluster_id: _identifier_pb2.ClusterIdentifier
133
+ project_id: _identifier_pb2.ProjectIdentifier
134
+ task_id: _task_definition_pb2.TaskIdentifier
135
+ def __init__(self, org: _Optional[str] = ..., cluster_id: _Optional[_Union[_identifier_pb2.ClusterIdentifier, _Mapping]] = ..., project_id: _Optional[_Union[_identifier_pb2.ProjectIdentifier, _Mapping]] = ..., task_id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ...) -> None: ...
136
+
137
+ class WatchRunsResponse(_message.Message):
138
+ __slots__ = ["runs"]
139
+ RUNS_FIELD_NUMBER: _ClassVar[int]
140
+ runs: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.Run]
141
+ def __init__(self, runs: _Optional[_Iterable[_Union[_run_definition_pb2.Run, _Mapping]]] = ...) -> None: ...
142
+
143
+ class ListActionsRequest(_message.Message):
144
+ __slots__ = ["request", "run_id"]
145
+ REQUEST_FIELD_NUMBER: _ClassVar[int]
146
+ RUN_ID_FIELD_NUMBER: _ClassVar[int]
147
+ request: _list_pb2.ListRequest
148
+ run_id: _run_definition_pb2.RunIdentifier
149
+ def __init__(self, request: _Optional[_Union[_list_pb2.ListRequest, _Mapping]] = ..., run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
150
+
151
+ class ListActionsResponse(_message.Message):
152
+ __slots__ = ["actions", "token"]
153
+ ACTIONS_FIELD_NUMBER: _ClassVar[int]
154
+ TOKEN_FIELD_NUMBER: _ClassVar[int]
155
+ actions: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.Action]
156
+ token: str
157
+ def __init__(self, actions: _Optional[_Iterable[_Union[_run_definition_pb2.Action, _Mapping]]] = ..., token: _Optional[str] = ...) -> None: ...
158
+
159
+ class WatchActionsRequest(_message.Message):
160
+ __slots__ = ["run_id", "filter"]
161
+ RUN_ID_FIELD_NUMBER: _ClassVar[int]
162
+ FILTER_FIELD_NUMBER: _ClassVar[int]
163
+ run_id: _run_definition_pb2.RunIdentifier
164
+ filter: _containers.RepeatedCompositeFieldContainer[_list_pb2.Filter]
165
+ def __init__(self, run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ..., filter: _Optional[_Iterable[_Union[_list_pb2.Filter, _Mapping]]] = ...) -> None: ...
166
+
167
+ class WatchActionsResponse(_message.Message):
168
+ __slots__ = ["enriched_actions"]
169
+ ENRICHED_ACTIONS_FIELD_NUMBER: _ClassVar[int]
170
+ enriched_actions: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.EnrichedAction]
171
+ def __init__(self, enriched_actions: _Optional[_Iterable[_Union[_run_definition_pb2.EnrichedAction, _Mapping]]] = ...) -> None: ...