flyte 0.0.1b0__py3-none-any.whl → 2.0.0b46__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.
Files changed (455) hide show
  1. flyte/__init__.py +83 -30
  2. flyte/_bin/connect.py +61 -0
  3. flyte/_bin/debug.py +38 -0
  4. flyte/_bin/runtime.py +87 -19
  5. flyte/_bin/serve.py +351 -0
  6. flyte/_build.py +3 -2
  7. flyte/_cache/cache.py +6 -5
  8. flyte/_cache/local_cache.py +216 -0
  9. flyte/_code_bundle/_ignore.py +31 -5
  10. flyte/_code_bundle/_packaging.py +42 -11
  11. flyte/_code_bundle/_utils.py +57 -34
  12. flyte/_code_bundle/bundle.py +130 -27
  13. flyte/_constants.py +1 -0
  14. flyte/_context.py +21 -5
  15. flyte/_custom_context.py +73 -0
  16. flyte/_debug/constants.py +37 -0
  17. flyte/_debug/utils.py +17 -0
  18. flyte/_debug/vscode.py +315 -0
  19. flyte/_deploy.py +396 -75
  20. flyte/_deployer.py +109 -0
  21. flyte/_environment.py +94 -11
  22. flyte/_excepthook.py +37 -0
  23. flyte/_group.py +2 -1
  24. flyte/_hash.py +1 -16
  25. flyte/_image.py +544 -234
  26. flyte/_initialize.py +443 -294
  27. flyte/_interface.py +40 -5
  28. flyte/_internal/controllers/__init__.py +22 -8
  29. flyte/_internal/controllers/_local_controller.py +159 -35
  30. flyte/_internal/controllers/_trace.py +18 -10
  31. flyte/_internal/controllers/remote/__init__.py +38 -9
  32. flyte/_internal/controllers/remote/_action.py +82 -12
  33. flyte/_internal/controllers/remote/_client.py +6 -2
  34. flyte/_internal/controllers/remote/_controller.py +290 -64
  35. flyte/_internal/controllers/remote/_core.py +155 -95
  36. flyte/_internal/controllers/remote/_informer.py +40 -20
  37. flyte/_internal/controllers/remote/_service_protocol.py +2 -2
  38. flyte/_internal/imagebuild/__init__.py +2 -10
  39. flyte/_internal/imagebuild/docker_builder.py +391 -84
  40. flyte/_internal/imagebuild/image_builder.py +111 -55
  41. flyte/_internal/imagebuild/remote_builder.py +409 -0
  42. flyte/_internal/imagebuild/utils.py +79 -0
  43. flyte/_internal/resolvers/_app_env_module.py +92 -0
  44. flyte/_internal/resolvers/_task_module.py +5 -38
  45. flyte/_internal/resolvers/app_env.py +26 -0
  46. flyte/_internal/resolvers/common.py +8 -1
  47. flyte/_internal/resolvers/default.py +2 -2
  48. flyte/_internal/runtime/convert.py +322 -33
  49. flyte/_internal/runtime/entrypoints.py +106 -18
  50. flyte/_internal/runtime/io.py +71 -23
  51. flyte/_internal/runtime/resources_serde.py +21 -7
  52. flyte/_internal/runtime/reuse.py +125 -0
  53. flyte/_internal/runtime/rusty.py +196 -0
  54. flyte/_internal/runtime/task_serde.py +239 -66
  55. flyte/_internal/runtime/taskrunner.py +48 -8
  56. flyte/_internal/runtime/trigger_serde.py +162 -0
  57. flyte/_internal/runtime/types_serde.py +7 -16
  58. flyte/_keyring/file.py +115 -0
  59. flyte/_link.py +30 -0
  60. flyte/_logging.py +241 -42
  61. flyte/_map.py +312 -0
  62. flyte/_metrics.py +59 -0
  63. flyte/_module.py +74 -0
  64. flyte/_pod.py +30 -0
  65. flyte/_resources.py +296 -33
  66. flyte/_retry.py +1 -7
  67. flyte/_reusable_environment.py +72 -7
  68. flyte/_run.py +461 -132
  69. flyte/_secret.py +47 -11
  70. flyte/_serve.py +333 -0
  71. flyte/_task.py +245 -56
  72. flyte/_task_environment.py +219 -97
  73. flyte/_task_plugins.py +47 -0
  74. flyte/_tools.py +8 -8
  75. flyte/_trace.py +15 -24
  76. flyte/_trigger.py +1027 -0
  77. flyte/_utils/__init__.py +12 -1
  78. flyte/_utils/asyn.py +3 -1
  79. flyte/_utils/async_cache.py +139 -0
  80. flyte/_utils/coro_management.py +5 -4
  81. flyte/_utils/description_parser.py +19 -0
  82. flyte/_utils/docker_credentials.py +173 -0
  83. flyte/_utils/helpers.py +45 -19
  84. flyte/_utils/module_loader.py +123 -0
  85. flyte/_utils/org_discovery.py +57 -0
  86. flyte/_utils/uv_script_parser.py +8 -1
  87. flyte/_version.py +16 -3
  88. flyte/app/__init__.py +27 -0
  89. flyte/app/_app_environment.py +362 -0
  90. flyte/app/_connector_environment.py +40 -0
  91. flyte/app/_deploy.py +130 -0
  92. flyte/app/_parameter.py +343 -0
  93. flyte/app/_runtime/__init__.py +3 -0
  94. flyte/app/_runtime/app_serde.py +383 -0
  95. flyte/app/_types.py +113 -0
  96. flyte/app/extras/__init__.py +9 -0
  97. flyte/app/extras/_auth_middleware.py +217 -0
  98. flyte/app/extras/_fastapi.py +93 -0
  99. flyte/app/extras/_model_loader/__init__.py +3 -0
  100. flyte/app/extras/_model_loader/config.py +7 -0
  101. flyte/app/extras/_model_loader/loader.py +288 -0
  102. flyte/cli/__init__.py +12 -0
  103. flyte/cli/_abort.py +28 -0
  104. flyte/cli/_build.py +114 -0
  105. flyte/cli/_common.py +493 -0
  106. flyte/cli/_create.py +371 -0
  107. flyte/cli/_delete.py +45 -0
  108. flyte/cli/_deploy.py +401 -0
  109. flyte/cli/_gen.py +316 -0
  110. flyte/cli/_get.py +446 -0
  111. flyte/cli/_option.py +33 -0
  112. {union/_cli → flyte/cli}/_params.py +152 -153
  113. flyte/cli/_plugins.py +209 -0
  114. flyte/cli/_prefetch.py +292 -0
  115. flyte/cli/_run.py +690 -0
  116. flyte/cli/_serve.py +338 -0
  117. flyte/cli/_update.py +86 -0
  118. flyte/cli/_user.py +20 -0
  119. flyte/cli/main.py +246 -0
  120. flyte/config/__init__.py +3 -0
  121. flyte/config/_config.py +248 -0
  122. flyte/config/_internal.py +73 -0
  123. flyte/config/_reader.py +225 -0
  124. flyte/connectors/__init__.py +11 -0
  125. flyte/connectors/_connector.py +330 -0
  126. flyte/connectors/_server.py +194 -0
  127. flyte/connectors/utils.py +159 -0
  128. flyte/errors.py +134 -2
  129. flyte/extend.py +24 -0
  130. flyte/extras/_container.py +69 -56
  131. flyte/git/__init__.py +3 -0
  132. flyte/git/_config.py +279 -0
  133. flyte/io/__init__.py +8 -1
  134. flyte/io/{structured_dataset → _dataframe}/__init__.py +32 -30
  135. flyte/io/{structured_dataset → _dataframe}/basic_dfs.py +75 -68
  136. flyte/io/{structured_dataset/structured_dataset.py → _dataframe/dataframe.py} +207 -242
  137. flyte/io/_dir.py +575 -113
  138. flyte/io/_file.py +587 -141
  139. flyte/io/_hashing_io.py +342 -0
  140. flyte/io/extend.py +7 -0
  141. flyte/models.py +635 -0
  142. flyte/prefetch/__init__.py +22 -0
  143. flyte/prefetch/_hf_model.py +563 -0
  144. flyte/remote/__init__.py +14 -3
  145. flyte/remote/_action.py +879 -0
  146. flyte/remote/_app.py +346 -0
  147. flyte/remote/_auth_metadata.py +42 -0
  148. flyte/remote/_client/_protocols.py +62 -4
  149. flyte/remote/_client/auth/_auth_utils.py +19 -0
  150. flyte/remote/_client/auth/_authenticators/base.py +8 -2
  151. flyte/remote/_client/auth/_authenticators/device_code.py +4 -5
  152. flyte/remote/_client/auth/_authenticators/factory.py +4 -0
  153. flyte/remote/_client/auth/_authenticators/passthrough.py +79 -0
  154. flyte/remote/_client/auth/_authenticators/pkce.py +17 -18
  155. flyte/remote/_client/auth/_channel.py +47 -18
  156. flyte/remote/_client/auth/_client_config.py +5 -3
  157. flyte/remote/_client/auth/_keyring.py +15 -2
  158. flyte/remote/_client/auth/_token_client.py +3 -3
  159. flyte/remote/_client/controlplane.py +206 -18
  160. flyte/remote/_common.py +66 -0
  161. flyte/remote/_data.py +107 -22
  162. flyte/remote/_logs.py +116 -33
  163. flyte/remote/_project.py +21 -19
  164. flyte/remote/_run.py +164 -631
  165. flyte/remote/_secret.py +72 -29
  166. flyte/remote/_task.py +387 -46
  167. flyte/remote/_trigger.py +368 -0
  168. flyte/remote/_user.py +43 -0
  169. flyte/report/_report.py +10 -6
  170. flyte/storage/__init__.py +13 -1
  171. flyte/storage/_config.py +237 -0
  172. flyte/storage/_parallel_reader.py +289 -0
  173. flyte/storage/_storage.py +268 -59
  174. flyte/syncify/__init__.py +56 -0
  175. flyte/syncify/_api.py +414 -0
  176. flyte/types/__init__.py +39 -0
  177. flyte/types/_interface.py +22 -7
  178. flyte/{io/pickle/transformer.py → types/_pickle.py} +37 -9
  179. flyte/types/_string_literals.py +8 -9
  180. flyte/types/_type_engine.py +230 -129
  181. flyte/types/_utils.py +1 -1
  182. flyte-2.0.0b46.data/scripts/debug.py +38 -0
  183. flyte-2.0.0b46.data/scripts/runtime.py +194 -0
  184. flyte-2.0.0b46.dist-info/METADATA +352 -0
  185. flyte-2.0.0b46.dist-info/RECORD +221 -0
  186. flyte-2.0.0b46.dist-info/entry_points.txt +8 -0
  187. flyte-2.0.0b46.dist-info/licenses/LICENSE +201 -0
  188. flyte/_api_commons.py +0 -3
  189. flyte/_cli/_common.py +0 -287
  190. flyte/_cli/_create.py +0 -42
  191. flyte/_cli/_delete.py +0 -23
  192. flyte/_cli/_deploy.py +0 -140
  193. flyte/_cli/_get.py +0 -235
  194. flyte/_cli/_run.py +0 -152
  195. flyte/_cli/main.py +0 -72
  196. flyte/_datastructures.py +0 -342
  197. flyte/_internal/controllers/pbhash.py +0 -39
  198. flyte/_protos/common/authorization_pb2.py +0 -66
  199. flyte/_protos/common/authorization_pb2.pyi +0 -108
  200. flyte/_protos/common/authorization_pb2_grpc.py +0 -4
  201. flyte/_protos/common/identifier_pb2.py +0 -71
  202. flyte/_protos/common/identifier_pb2.pyi +0 -82
  203. flyte/_protos/common/identifier_pb2_grpc.py +0 -4
  204. flyte/_protos/common/identity_pb2.py +0 -48
  205. flyte/_protos/common/identity_pb2.pyi +0 -72
  206. flyte/_protos/common/identity_pb2_grpc.py +0 -4
  207. flyte/_protos/common/list_pb2.py +0 -36
  208. flyte/_protos/common/list_pb2.pyi +0 -69
  209. flyte/_protos/common/list_pb2_grpc.py +0 -4
  210. flyte/_protos/common/policy_pb2.py +0 -37
  211. flyte/_protos/common/policy_pb2.pyi +0 -27
  212. flyte/_protos/common/policy_pb2_grpc.py +0 -4
  213. flyte/_protos/common/role_pb2.py +0 -37
  214. flyte/_protos/common/role_pb2.pyi +0 -53
  215. flyte/_protos/common/role_pb2_grpc.py +0 -4
  216. flyte/_protos/common/runtime_version_pb2.py +0 -28
  217. flyte/_protos/common/runtime_version_pb2.pyi +0 -24
  218. flyte/_protos/common/runtime_version_pb2_grpc.py +0 -4
  219. flyte/_protos/logs/dataplane/payload_pb2.py +0 -96
  220. flyte/_protos/logs/dataplane/payload_pb2.pyi +0 -168
  221. flyte/_protos/logs/dataplane/payload_pb2_grpc.py +0 -4
  222. flyte/_protos/secret/definition_pb2.py +0 -49
  223. flyte/_protos/secret/definition_pb2.pyi +0 -93
  224. flyte/_protos/secret/definition_pb2_grpc.py +0 -4
  225. flyte/_protos/secret/payload_pb2.py +0 -62
  226. flyte/_protos/secret/payload_pb2.pyi +0 -94
  227. flyte/_protos/secret/payload_pb2_grpc.py +0 -4
  228. flyte/_protos/secret/secret_pb2.py +0 -38
  229. flyte/_protos/secret/secret_pb2.pyi +0 -6
  230. flyte/_protos/secret/secret_pb2_grpc.py +0 -198
  231. flyte/_protos/secret/secret_pb2_grpc_grpc.py +0 -198
  232. flyte/_protos/validate/validate/validate_pb2.py +0 -76
  233. flyte/_protos/workflow/node_execution_service_pb2.py +0 -26
  234. flyte/_protos/workflow/node_execution_service_pb2.pyi +0 -4
  235. flyte/_protos/workflow/node_execution_service_pb2_grpc.py +0 -32
  236. flyte/_protos/workflow/queue_service_pb2.py +0 -106
  237. flyte/_protos/workflow/queue_service_pb2.pyi +0 -141
  238. flyte/_protos/workflow/queue_service_pb2_grpc.py +0 -172
  239. flyte/_protos/workflow/run_definition_pb2.py +0 -128
  240. flyte/_protos/workflow/run_definition_pb2.pyi +0 -310
  241. flyte/_protos/workflow/run_definition_pb2_grpc.py +0 -4
  242. flyte/_protos/workflow/run_logs_service_pb2.py +0 -41
  243. flyte/_protos/workflow/run_logs_service_pb2.pyi +0 -28
  244. flyte/_protos/workflow/run_logs_service_pb2_grpc.py +0 -69
  245. flyte/_protos/workflow/run_service_pb2.py +0 -133
  246. flyte/_protos/workflow/run_service_pb2.pyi +0 -175
  247. flyte/_protos/workflow/run_service_pb2_grpc.py +0 -412
  248. flyte/_protos/workflow/state_service_pb2.py +0 -58
  249. flyte/_protos/workflow/state_service_pb2.pyi +0 -71
  250. flyte/_protos/workflow/state_service_pb2_grpc.py +0 -138
  251. flyte/_protos/workflow/task_definition_pb2.py +0 -72
  252. flyte/_protos/workflow/task_definition_pb2.pyi +0 -65
  253. flyte/_protos/workflow/task_definition_pb2_grpc.py +0 -4
  254. flyte/_protos/workflow/task_service_pb2.py +0 -44
  255. flyte/_protos/workflow/task_service_pb2.pyi +0 -31
  256. flyte/_protos/workflow/task_service_pb2_grpc.py +0 -104
  257. flyte/io/_dataframe.py +0 -0
  258. flyte/io/pickle/__init__.py +0 -0
  259. flyte/remote/_console.py +0 -18
  260. flyte-0.0.1b0.dist-info/METADATA +0 -179
  261. flyte-0.0.1b0.dist-info/RECORD +0 -390
  262. flyte-0.0.1b0.dist-info/entry_points.txt +0 -3
  263. union/__init__.py +0 -54
  264. union/_api_commons.py +0 -3
  265. union/_bin/__init__.py +0 -0
  266. union/_bin/runtime.py +0 -113
  267. union/_build.py +0 -25
  268. union/_cache/__init__.py +0 -12
  269. union/_cache/cache.py +0 -141
  270. union/_cache/defaults.py +0 -9
  271. union/_cache/policy_function_body.py +0 -42
  272. union/_cli/__init__.py +0 -0
  273. union/_cli/_common.py +0 -263
  274. union/_cli/_create.py +0 -40
  275. union/_cli/_delete.py +0 -23
  276. union/_cli/_deploy.py +0 -120
  277. union/_cli/_get.py +0 -162
  278. union/_cli/_run.py +0 -150
  279. union/_cli/main.py +0 -72
  280. union/_code_bundle/__init__.py +0 -8
  281. union/_code_bundle/_ignore.py +0 -113
  282. union/_code_bundle/_packaging.py +0 -187
  283. union/_code_bundle/_utils.py +0 -342
  284. union/_code_bundle/bundle.py +0 -176
  285. union/_context.py +0 -146
  286. union/_datastructures.py +0 -295
  287. union/_deploy.py +0 -185
  288. union/_doc.py +0 -29
  289. union/_docstring.py +0 -26
  290. union/_environment.py +0 -43
  291. union/_group.py +0 -31
  292. union/_hash.py +0 -23
  293. union/_image.py +0 -760
  294. union/_initialize.py +0 -585
  295. union/_interface.py +0 -84
  296. union/_internal/__init__.py +0 -3
  297. union/_internal/controllers/__init__.py +0 -77
  298. union/_internal/controllers/_local_controller.py +0 -77
  299. union/_internal/controllers/pbhash.py +0 -39
  300. union/_internal/controllers/remote/__init__.py +0 -40
  301. union/_internal/controllers/remote/_action.py +0 -131
  302. union/_internal/controllers/remote/_client.py +0 -43
  303. union/_internal/controllers/remote/_controller.py +0 -169
  304. union/_internal/controllers/remote/_core.py +0 -341
  305. union/_internal/controllers/remote/_informer.py +0 -260
  306. union/_internal/controllers/remote/_service_protocol.py +0 -44
  307. union/_internal/imagebuild/__init__.py +0 -11
  308. union/_internal/imagebuild/docker_builder.py +0 -416
  309. union/_internal/imagebuild/image_builder.py +0 -243
  310. union/_internal/imagebuild/remote_builder.py +0 -0
  311. union/_internal/resolvers/__init__.py +0 -0
  312. union/_internal/resolvers/_task_module.py +0 -31
  313. union/_internal/resolvers/common.py +0 -24
  314. union/_internal/resolvers/default.py +0 -27
  315. union/_internal/runtime/__init__.py +0 -0
  316. union/_internal/runtime/convert.py +0 -163
  317. union/_internal/runtime/entrypoints.py +0 -121
  318. union/_internal/runtime/io.py +0 -136
  319. union/_internal/runtime/resources_serde.py +0 -134
  320. union/_internal/runtime/task_serde.py +0 -202
  321. union/_internal/runtime/taskrunner.py +0 -179
  322. union/_internal/runtime/types_serde.py +0 -53
  323. union/_logging.py +0 -124
  324. union/_protos/__init__.py +0 -0
  325. union/_protos/common/authorization_pb2.py +0 -66
  326. union/_protos/common/authorization_pb2.pyi +0 -106
  327. union/_protos/common/authorization_pb2_grpc.py +0 -4
  328. union/_protos/common/identifier_pb2.py +0 -71
  329. union/_protos/common/identifier_pb2.pyi +0 -82
  330. union/_protos/common/identifier_pb2_grpc.py +0 -4
  331. union/_protos/common/identity_pb2.py +0 -48
  332. union/_protos/common/identity_pb2.pyi +0 -72
  333. union/_protos/common/identity_pb2_grpc.py +0 -4
  334. union/_protos/common/list_pb2.py +0 -36
  335. union/_protos/common/list_pb2.pyi +0 -69
  336. union/_protos/common/list_pb2_grpc.py +0 -4
  337. union/_protos/common/policy_pb2.py +0 -37
  338. union/_protos/common/policy_pb2.pyi +0 -27
  339. union/_protos/common/policy_pb2_grpc.py +0 -4
  340. union/_protos/common/role_pb2.py +0 -37
  341. union/_protos/common/role_pb2.pyi +0 -51
  342. union/_protos/common/role_pb2_grpc.py +0 -4
  343. union/_protos/common/runtime_version_pb2.py +0 -28
  344. union/_protos/common/runtime_version_pb2.pyi +0 -24
  345. union/_protos/common/runtime_version_pb2_grpc.py +0 -4
  346. union/_protos/logs/dataplane/payload_pb2.py +0 -96
  347. union/_protos/logs/dataplane/payload_pb2.pyi +0 -168
  348. union/_protos/logs/dataplane/payload_pb2_grpc.py +0 -4
  349. union/_protos/secret/definition_pb2.py +0 -49
  350. union/_protos/secret/definition_pb2.pyi +0 -93
  351. union/_protos/secret/definition_pb2_grpc.py +0 -4
  352. union/_protos/secret/payload_pb2.py +0 -62
  353. union/_protos/secret/payload_pb2.pyi +0 -94
  354. union/_protos/secret/payload_pb2_grpc.py +0 -4
  355. union/_protos/secret/secret_pb2.py +0 -38
  356. union/_protos/secret/secret_pb2.pyi +0 -6
  357. union/_protos/secret/secret_pb2_grpc.py +0 -198
  358. union/_protos/validate/validate/validate_pb2.py +0 -76
  359. union/_protos/workflow/node_execution_service_pb2.py +0 -26
  360. union/_protos/workflow/node_execution_service_pb2.pyi +0 -4
  361. union/_protos/workflow/node_execution_service_pb2_grpc.py +0 -32
  362. union/_protos/workflow/queue_service_pb2.py +0 -75
  363. union/_protos/workflow/queue_service_pb2.pyi +0 -103
  364. union/_protos/workflow/queue_service_pb2_grpc.py +0 -172
  365. union/_protos/workflow/run_definition_pb2.py +0 -100
  366. union/_protos/workflow/run_definition_pb2.pyi +0 -256
  367. union/_protos/workflow/run_definition_pb2_grpc.py +0 -4
  368. union/_protos/workflow/run_logs_service_pb2.py +0 -41
  369. union/_protos/workflow/run_logs_service_pb2.pyi +0 -28
  370. union/_protos/workflow/run_logs_service_pb2_grpc.py +0 -69
  371. union/_protos/workflow/run_service_pb2.py +0 -133
  372. union/_protos/workflow/run_service_pb2.pyi +0 -173
  373. union/_protos/workflow/run_service_pb2_grpc.py +0 -412
  374. union/_protos/workflow/state_service_pb2.py +0 -58
  375. union/_protos/workflow/state_service_pb2.pyi +0 -69
  376. union/_protos/workflow/state_service_pb2_grpc.py +0 -138
  377. union/_protos/workflow/task_definition_pb2.py +0 -72
  378. union/_protos/workflow/task_definition_pb2.pyi +0 -65
  379. union/_protos/workflow/task_definition_pb2_grpc.py +0 -4
  380. union/_protos/workflow/task_service_pb2.py +0 -44
  381. union/_protos/workflow/task_service_pb2.pyi +0 -31
  382. union/_protos/workflow/task_service_pb2_grpc.py +0 -104
  383. union/_resources.py +0 -226
  384. union/_retry.py +0 -32
  385. union/_reusable_environment.py +0 -25
  386. union/_run.py +0 -374
  387. union/_secret.py +0 -61
  388. union/_task.py +0 -354
  389. union/_task_environment.py +0 -186
  390. union/_timeout.py +0 -47
  391. union/_tools.py +0 -27
  392. union/_utils/__init__.py +0 -11
  393. union/_utils/asyn.py +0 -119
  394. union/_utils/file_handling.py +0 -71
  395. union/_utils/helpers.py +0 -46
  396. union/_utils/lazy_module.py +0 -54
  397. union/_utils/uv_script_parser.py +0 -49
  398. union/_version.py +0 -21
  399. union/connectors/__init__.py +0 -0
  400. union/errors.py +0 -128
  401. union/extras/__init__.py +0 -5
  402. union/extras/_container.py +0 -263
  403. union/io/__init__.py +0 -11
  404. union/io/_dataframe.py +0 -0
  405. union/io/_dir.py +0 -425
  406. union/io/_file.py +0 -418
  407. union/io/pickle/__init__.py +0 -0
  408. union/io/pickle/transformer.py +0 -117
  409. union/io/structured_dataset/__init__.py +0 -122
  410. union/io/structured_dataset/basic_dfs.py +0 -219
  411. union/io/structured_dataset/structured_dataset.py +0 -1057
  412. union/py.typed +0 -0
  413. union/remote/__init__.py +0 -23
  414. union/remote/_client/__init__.py +0 -0
  415. union/remote/_client/_protocols.py +0 -129
  416. union/remote/_client/auth/__init__.py +0 -12
  417. union/remote/_client/auth/_authenticators/__init__.py +0 -0
  418. union/remote/_client/auth/_authenticators/base.py +0 -391
  419. union/remote/_client/auth/_authenticators/client_credentials.py +0 -73
  420. union/remote/_client/auth/_authenticators/device_code.py +0 -120
  421. union/remote/_client/auth/_authenticators/external_command.py +0 -77
  422. union/remote/_client/auth/_authenticators/factory.py +0 -200
  423. union/remote/_client/auth/_authenticators/pkce.py +0 -515
  424. union/remote/_client/auth/_channel.py +0 -184
  425. union/remote/_client/auth/_client_config.py +0 -83
  426. union/remote/_client/auth/_default_html.py +0 -32
  427. union/remote/_client/auth/_grpc_utils/__init__.py +0 -0
  428. union/remote/_client/auth/_grpc_utils/auth_interceptor.py +0 -204
  429. union/remote/_client/auth/_grpc_utils/default_metadata_interceptor.py +0 -144
  430. union/remote/_client/auth/_keyring.py +0 -154
  431. union/remote/_client/auth/_token_client.py +0 -258
  432. union/remote/_client/auth/errors.py +0 -16
  433. union/remote/_client/controlplane.py +0 -86
  434. union/remote/_data.py +0 -149
  435. union/remote/_logs.py +0 -74
  436. union/remote/_project.py +0 -86
  437. union/remote/_run.py +0 -820
  438. union/remote/_secret.py +0 -132
  439. union/remote/_task.py +0 -193
  440. union/report/__init__.py +0 -3
  441. union/report/_report.py +0 -178
  442. union/report/_template.html +0 -124
  443. union/storage/__init__.py +0 -24
  444. union/storage/_remote_fs.py +0 -34
  445. union/storage/_storage.py +0 -247
  446. union/storage/_utils.py +0 -5
  447. union/types/__init__.py +0 -11
  448. union/types/_renderer.py +0 -162
  449. union/types/_string_literals.py +0 -120
  450. union/types/_type_engine.py +0 -2131
  451. union/types/_utils.py +0 -80
  452. /flyte/{_cli → _debug}/__init__.py +0 -0
  453. /flyte/{_protos → _keyring}/__init__.py +0 -0
  454. {flyte-0.0.1b0.dist-info → flyte-2.0.0b46.dist-info}/WHEEL +0 -0
  455. {flyte-0.0.1b0.dist-info → flyte-2.0.0b46.dist-info}/top_level.txt +0 -0
@@ -1,106 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- # Generated by the protocol buffer compiler. DO NOT EDIT!
3
- # source: workflow/queue_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 flyteidl.core import types_pb2 as flyteidl_dot_core_dot_types__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
- from flyte._protos.workflow import task_definition_pb2 as workflow_dot_task__definition__pb2
18
-
19
-
20
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cworkflow/queue_service.proto\x12\x11\x63loudidl.workflow\x1a\x19\x66lyteidl/core/types.proto\x1a\x17validate/validate.proto\x1a\x1dworkflow/run_definition.proto\x1a\x1eworkflow/task_definition.proto\"\x8a\x04\n\x14\x45nqueueActionRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\x12\x31\n\x12parent_action_name\x18\x02 \x01(\tH\x01R\x10parentActionName\x88\x01\x01\x12$\n\tinput_uri\x18\x06 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x08inputUri\x12/\n\x0frun_output_base\x18\x07 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\rrunOutputBase\x12\x14\n\x05group\x18\x08 \x01(\tR\x05group\x12\x18\n\x07subject\x18\t \x01(\tR\x07subject\x12=\n\x04task\x18\n \x01(\x0b\x32\x1d.cloudidl.workflow.TaskActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x04task\x12@\n\x05trace\x18\x0b \x01(\x0b\x32\x1e.cloudidl.workflow.TraceActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x05trace\x12L\n\tcondition\x18\x0c \x01(\x0b\x32\".cloudidl.workflow.ConditionActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tconditionB\x06\n\x04specB\x15\n\x13_parent_action_name\"z\n\nTaskAction\x12\x31\n\x02id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierR\x02id\x12\x39\n\x04spec\x18\x02 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x04spec\"*\n\x0bTraceAction\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"\x8a\x02\n\x0f\x43onditionAction\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\x12 \n\x06run_id\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x05runId\x12&\n\taction_id\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x08\x61\x63tionId\x12\x18\n\x06global\x18\x04 \x01(\x08H\x00R\x06global\x12.\n\x04type\x18\x06 \x01(\x0b\x32\x1a.flyteidl.core.LiteralTypeR\x04type\x12\x16\n\x06prompt\x18\x07 \x01(\tR\x06prompt\x12 \n\x0b\x64\x65scription\x18\x08 \x01(\tR\x0b\x64\x65scriptionB\x0c\n\x05scope\x12\x03\xf8\x42\x01\"\x17\n\x15\x45nqueueActionResponse\"f\n\x18\x41\x62ortQueuedActionRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\"\x1b\n\x19\x41\x62ortQueuedActionResponse\"\x8e\x03\n\x10HeartbeatRequest\x12$\n\tworker_id\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x08workerId\x12&\n\ncluster_id\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\tclusterId\x12O\n\x11\x61\x63tive_action_ids\x18\x03 \x03(\x0b\x32#.cloudidl.workflow.ActionIdentifierR\x0f\x61\x63tiveActionIds\x12S\n\x13terminal_action_ids\x18\x04 \x03(\x0b\x32#.cloudidl.workflow.ActionIdentifierR\x11terminalActionIds\x12-\n\x0corganization\x18\x05 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x0corganization\x12-\n\x12\x61vailable_capacity\x18\x06 \x01(\x05R\x11\x61vailableCapacity\x12\x19\n\x03org\x18\x07 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x03orgB\r\n\x06\x66ilter\x12\x03\xf8\x42\x01\"\xf6\x01\n\x11HeartbeatResponse\x12\x37\n\nnew_leases\x18\x01 \x03(\x0b\x32\x18.cloudidl.workflow.LeaseR\tnewLeases\x12Q\n\x12\x61\x62orted_action_ids\x18\x02 \x03(\x0b\x32#.cloudidl.workflow.ActionIdentifierR\x10\x61\x62ortedActionIds\x12U\n\x14\x66inalized_action_ids\x18\x03 \x03(\x0b\x32#.cloudidl.workflow.ActionIdentifierR\x12\x66inalizedActionIds\"V\n\x13StreamLeasesRequest\x12$\n\tworker_id\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x08workerId\x12\x19\n\x03org\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x03org\"H\n\x14StreamLeasesResponse\x12\x30\n\x06leases\x18\x01 \x03(\x0b\x32\x18.cloudidl.workflow.LeaseR\x06leases\"\xe5\x03\n\x05Lease\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\x12\x31\n\x12parent_action_name\x18\x02 \x01(\tH\x01R\x10parentActionName\x88\x01\x01\x12$\n\tinput_uri\x18\x06 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x08inputUri\x12/\n\x0frun_output_base\x18\x07 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\rrunOutputBase\x12=\n\x04task\x18\x0b \x01(\x0b\x32\x1d.cloudidl.workflow.TaskActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x04task\x12L\n\tcondition\x18\x0c \x01(\x0b\x32\".cloudidl.workflow.ConditionActionB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tcondition\x12\x14\n\x05group\x18\x08 \x01(\tR\x05group\x12\x18\n\x07subject\x18\t \x01(\tR\x07subject\x12%\n\x0eprevious_state\x18\n \x01(\tR\rpreviousStateB\x0b\n\x04spec\x12\x03\xf8\x42\x01\x42\x15\n\x13_parent_action_name2\xa9\x03\n\x0cQueueService\x12\x64\n\rEnqueueAction\x12\'.cloudidl.workflow.EnqueueActionRequest\x1a(.cloudidl.workflow.EnqueueActionResponse\"\x00\x12p\n\x11\x41\x62ortQueuedAction\x12+.cloudidl.workflow.AbortQueuedActionRequest\x1a,.cloudidl.workflow.AbortQueuedActionResponse\"\x00\x12\\\n\tHeartbeat\x12#.cloudidl.workflow.HeartbeatRequest\x1a$.cloudidl.workflow.HeartbeatResponse\"\x00(\x01\x30\x01\x12\x63\n\x0cStreamLeases\x12&.cloudidl.workflow.StreamLeasesRequest\x1a\'.cloudidl.workflow.StreamLeasesResponse\"\x00\x30\x01\x42\xbe\x01\n\x15\x63om.cloudidl.workflowB\x11QueueServiceProtoH\x02P\x01Z+github.com/unionai/cloud/gen/pb-go/workflow\xa2\x02\x03\x43WX\xaa\x02\x11\x43loudidl.Workflow\xca\x02\x11\x43loudidl\\Workflow\xe2\x02\x1d\x43loudidl\\Workflow\\GPBMetadata\xea\x02\x12\x43loudidl::Workflowb\x06proto3')
21
-
22
- _globals = globals()
23
- _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
24
- _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'workflow.queue_service_pb2', _globals)
25
- if _descriptor._USE_C_DESCRIPTORS == False:
26
- DESCRIPTOR._options = None
27
- DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\021QueueServiceProtoH\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'
28
- _ENQUEUEACTIONREQUEST.fields_by_name['action_id']._options = None
29
- _ENQUEUEACTIONREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
30
- _ENQUEUEACTIONREQUEST.fields_by_name['input_uri']._options = None
31
- _ENQUEUEACTIONREQUEST.fields_by_name['input_uri']._serialized_options = b'\372B\004r\002\020\001'
32
- _ENQUEUEACTIONREQUEST.fields_by_name['run_output_base']._options = None
33
- _ENQUEUEACTIONREQUEST.fields_by_name['run_output_base']._serialized_options = b'\372B\004r\002\020\001'
34
- _ENQUEUEACTIONREQUEST.fields_by_name['task']._options = None
35
- _ENQUEUEACTIONREQUEST.fields_by_name['task']._serialized_options = b'\372B\005\212\001\002\020\001'
36
- _ENQUEUEACTIONREQUEST.fields_by_name['trace']._options = None
37
- _ENQUEUEACTIONREQUEST.fields_by_name['trace']._serialized_options = b'\372B\005\212\001\002\020\001'
38
- _ENQUEUEACTIONREQUEST.fields_by_name['condition']._options = None
39
- _ENQUEUEACTIONREQUEST.fields_by_name['condition']._serialized_options = b'\372B\005\212\001\002\020\001'
40
- _TASKACTION.fields_by_name['spec']._options = None
41
- _TASKACTION.fields_by_name['spec']._serialized_options = b'\372B\005\212\001\002\020\001'
42
- _TRACEACTION.fields_by_name['name']._options = None
43
- _TRACEACTION.fields_by_name['name']._serialized_options = b'\372B\004r\002\020\001'
44
- _CONDITIONACTION.oneofs_by_name['scope']._options = None
45
- _CONDITIONACTION.oneofs_by_name['scope']._serialized_options = b'\370B\001'
46
- _CONDITIONACTION.fields_by_name['name']._options = None
47
- _CONDITIONACTION.fields_by_name['name']._serialized_options = b'\372B\004r\002\020\001'
48
- _CONDITIONACTION.fields_by_name['run_id']._options = None
49
- _CONDITIONACTION.fields_by_name['run_id']._serialized_options = b'\372B\004r\002\020\001'
50
- _CONDITIONACTION.fields_by_name['action_id']._options = None
51
- _CONDITIONACTION.fields_by_name['action_id']._serialized_options = b'\372B\004r\002\020\001'
52
- _ABORTQUEUEDACTIONREQUEST.fields_by_name['action_id']._options = None
53
- _ABORTQUEUEDACTIONREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
54
- _HEARTBEATREQUEST.oneofs_by_name['filter']._options = None
55
- _HEARTBEATREQUEST.oneofs_by_name['filter']._serialized_options = b'\370B\001'
56
- _HEARTBEATREQUEST.fields_by_name['worker_id']._options = None
57
- _HEARTBEATREQUEST.fields_by_name['worker_id']._serialized_options = b'\372B\004r\002\020\001'
58
- _HEARTBEATREQUEST.fields_by_name['cluster_id']._options = None
59
- _HEARTBEATREQUEST.fields_by_name['cluster_id']._serialized_options = b'\372B\004r\002\020\001'
60
- _HEARTBEATREQUEST.fields_by_name['organization']._options = None
61
- _HEARTBEATREQUEST.fields_by_name['organization']._serialized_options = b'\372B\004r\002\020\001'
62
- _HEARTBEATREQUEST.fields_by_name['org']._options = None
63
- _HEARTBEATREQUEST.fields_by_name['org']._serialized_options = b'\372B\004r\002\020\001'
64
- _STREAMLEASESREQUEST.fields_by_name['worker_id']._options = None
65
- _STREAMLEASESREQUEST.fields_by_name['worker_id']._serialized_options = b'\372B\004r\002\020\001'
66
- _STREAMLEASESREQUEST.fields_by_name['org']._options = None
67
- _STREAMLEASESREQUEST.fields_by_name['org']._serialized_options = b'\372B\004r\002\020\001'
68
- _LEASE.oneofs_by_name['spec']._options = None
69
- _LEASE.oneofs_by_name['spec']._serialized_options = b'\370B\001'
70
- _LEASE.fields_by_name['action_id']._options = None
71
- _LEASE.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
72
- _LEASE.fields_by_name['input_uri']._options = None
73
- _LEASE.fields_by_name['input_uri']._serialized_options = b'\372B\004r\002\020\001'
74
- _LEASE.fields_by_name['run_output_base']._options = None
75
- _LEASE.fields_by_name['run_output_base']._serialized_options = b'\372B\004r\002\020\001'
76
- _LEASE.fields_by_name['task']._options = None
77
- _LEASE.fields_by_name['task']._serialized_options = b'\372B\005\212\001\002\020\001'
78
- _LEASE.fields_by_name['condition']._options = None
79
- _LEASE.fields_by_name['condition']._serialized_options = b'\372B\005\212\001\002\020\001'
80
- _globals['_ENQUEUEACTIONREQUEST']._serialized_start=167
81
- _globals['_ENQUEUEACTIONREQUEST']._serialized_end=689
82
- _globals['_TASKACTION']._serialized_start=691
83
- _globals['_TASKACTION']._serialized_end=813
84
- _globals['_TRACEACTION']._serialized_start=815
85
- _globals['_TRACEACTION']._serialized_end=857
86
- _globals['_CONDITIONACTION']._serialized_start=860
87
- _globals['_CONDITIONACTION']._serialized_end=1126
88
- _globals['_ENQUEUEACTIONRESPONSE']._serialized_start=1128
89
- _globals['_ENQUEUEACTIONRESPONSE']._serialized_end=1151
90
- _globals['_ABORTQUEUEDACTIONREQUEST']._serialized_start=1153
91
- _globals['_ABORTQUEUEDACTIONREQUEST']._serialized_end=1255
92
- _globals['_ABORTQUEUEDACTIONRESPONSE']._serialized_start=1257
93
- _globals['_ABORTQUEUEDACTIONRESPONSE']._serialized_end=1284
94
- _globals['_HEARTBEATREQUEST']._serialized_start=1287
95
- _globals['_HEARTBEATREQUEST']._serialized_end=1685
96
- _globals['_HEARTBEATRESPONSE']._serialized_start=1688
97
- _globals['_HEARTBEATRESPONSE']._serialized_end=1934
98
- _globals['_STREAMLEASESREQUEST']._serialized_start=1936
99
- _globals['_STREAMLEASESREQUEST']._serialized_end=2022
100
- _globals['_STREAMLEASESRESPONSE']._serialized_start=2024
101
- _globals['_STREAMLEASESRESPONSE']._serialized_end=2096
102
- _globals['_LEASE']._serialized_start=2099
103
- _globals['_LEASE']._serialized_end=2584
104
- _globals['_QUEUESERVICE']._serialized_start=2587
105
- _globals['_QUEUESERVICE']._serialized_end=3012
106
- # @@protoc_insertion_point(module_scope)
@@ -1,141 +0,0 @@
1
- from flyteidl.core import types_pb2 as _types_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 flyte._protos.workflow import task_definition_pb2 as _task_definition_pb2
5
- from google.protobuf.internal import containers as _containers
6
- from google.protobuf import descriptor as _descriptor
7
- from google.protobuf import message as _message
8
- from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
9
-
10
- DESCRIPTOR: _descriptor.FileDescriptor
11
-
12
- class EnqueueActionRequest(_message.Message):
13
- __slots__ = ["action_id", "parent_action_name", "input_uri", "run_output_base", "group", "subject", "task", "trace", "condition"]
14
- ACTION_ID_FIELD_NUMBER: _ClassVar[int]
15
- PARENT_ACTION_NAME_FIELD_NUMBER: _ClassVar[int]
16
- INPUT_URI_FIELD_NUMBER: _ClassVar[int]
17
- RUN_OUTPUT_BASE_FIELD_NUMBER: _ClassVar[int]
18
- GROUP_FIELD_NUMBER: _ClassVar[int]
19
- SUBJECT_FIELD_NUMBER: _ClassVar[int]
20
- TASK_FIELD_NUMBER: _ClassVar[int]
21
- TRACE_FIELD_NUMBER: _ClassVar[int]
22
- CONDITION_FIELD_NUMBER: _ClassVar[int]
23
- action_id: _run_definition_pb2.ActionIdentifier
24
- parent_action_name: str
25
- input_uri: str
26
- run_output_base: str
27
- group: str
28
- subject: str
29
- task: TaskAction
30
- trace: TraceAction
31
- condition: ConditionAction
32
- def __init__(self, action_id: _Optional[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]] = ..., parent_action_name: _Optional[str] = ..., input_uri: _Optional[str] = ..., run_output_base: _Optional[str] = ..., group: _Optional[str] = ..., subject: _Optional[str] = ..., task: _Optional[_Union[TaskAction, _Mapping]] = ..., trace: _Optional[_Union[TraceAction, _Mapping]] = ..., condition: _Optional[_Union[ConditionAction, _Mapping]] = ...) -> None: ...
33
-
34
- class TaskAction(_message.Message):
35
- __slots__ = ["id", "spec"]
36
- ID_FIELD_NUMBER: _ClassVar[int]
37
- SPEC_FIELD_NUMBER: _ClassVar[int]
38
- id: _task_definition_pb2.TaskIdentifier
39
- spec: _task_definition_pb2.TaskSpec
40
- def __init__(self, id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ..., spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ...) -> None: ...
41
-
42
- class TraceAction(_message.Message):
43
- __slots__ = ["name"]
44
- NAME_FIELD_NUMBER: _ClassVar[int]
45
- name: str
46
- def __init__(self, name: _Optional[str] = ...) -> None: ...
47
-
48
- class ConditionAction(_message.Message):
49
- __slots__ = ["name", "run_id", "action_id", "type", "prompt", "description"]
50
- NAME_FIELD_NUMBER: _ClassVar[int]
51
- RUN_ID_FIELD_NUMBER: _ClassVar[int]
52
- ACTION_ID_FIELD_NUMBER: _ClassVar[int]
53
- GLOBAL_FIELD_NUMBER: _ClassVar[int]
54
- TYPE_FIELD_NUMBER: _ClassVar[int]
55
- PROMPT_FIELD_NUMBER: _ClassVar[int]
56
- DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
57
- name: str
58
- run_id: str
59
- action_id: str
60
- type: _types_pb2.LiteralType
61
- prompt: str
62
- description: str
63
- def __init__(self, name: _Optional[str] = ..., run_id: _Optional[str] = ..., action_id: _Optional[str] = ..., type: _Optional[_Union[_types_pb2.LiteralType, _Mapping]] = ..., prompt: _Optional[str] = ..., description: _Optional[str] = ..., **kwargs) -> None: ...
64
-
65
- class EnqueueActionResponse(_message.Message):
66
- __slots__ = []
67
- def __init__(self) -> None: ...
68
-
69
- class AbortQueuedActionRequest(_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 AbortQueuedActionResponse(_message.Message):
76
- __slots__ = []
77
- def __init__(self) -> None: ...
78
-
79
- class HeartbeatRequest(_message.Message):
80
- __slots__ = ["worker_id", "cluster_id", "active_action_ids", "terminal_action_ids", "organization", "available_capacity", "org"]
81
- WORKER_ID_FIELD_NUMBER: _ClassVar[int]
82
- CLUSTER_ID_FIELD_NUMBER: _ClassVar[int]
83
- ACTIVE_ACTION_IDS_FIELD_NUMBER: _ClassVar[int]
84
- TERMINAL_ACTION_IDS_FIELD_NUMBER: _ClassVar[int]
85
- ORGANIZATION_FIELD_NUMBER: _ClassVar[int]
86
- AVAILABLE_CAPACITY_FIELD_NUMBER: _ClassVar[int]
87
- ORG_FIELD_NUMBER: _ClassVar[int]
88
- worker_id: str
89
- cluster_id: str
90
- active_action_ids: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.ActionIdentifier]
91
- terminal_action_ids: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.ActionIdentifier]
92
- organization: str
93
- available_capacity: int
94
- org: str
95
- def __init__(self, worker_id: _Optional[str] = ..., cluster_id: _Optional[str] = ..., active_action_ids: _Optional[_Iterable[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]]] = ..., terminal_action_ids: _Optional[_Iterable[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]]] = ..., organization: _Optional[str] = ..., available_capacity: _Optional[int] = ..., org: _Optional[str] = ...) -> None: ...
96
-
97
- class HeartbeatResponse(_message.Message):
98
- __slots__ = ["new_leases", "aborted_action_ids", "finalized_action_ids"]
99
- NEW_LEASES_FIELD_NUMBER: _ClassVar[int]
100
- ABORTED_ACTION_IDS_FIELD_NUMBER: _ClassVar[int]
101
- FINALIZED_ACTION_IDS_FIELD_NUMBER: _ClassVar[int]
102
- new_leases: _containers.RepeatedCompositeFieldContainer[Lease]
103
- aborted_action_ids: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.ActionIdentifier]
104
- finalized_action_ids: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.ActionIdentifier]
105
- def __init__(self, new_leases: _Optional[_Iterable[_Union[Lease, _Mapping]]] = ..., aborted_action_ids: _Optional[_Iterable[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]]] = ..., finalized_action_ids: _Optional[_Iterable[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]]] = ...) -> None: ...
106
-
107
- class StreamLeasesRequest(_message.Message):
108
- __slots__ = ["worker_id", "org"]
109
- WORKER_ID_FIELD_NUMBER: _ClassVar[int]
110
- ORG_FIELD_NUMBER: _ClassVar[int]
111
- worker_id: str
112
- org: str
113
- def __init__(self, worker_id: _Optional[str] = ..., org: _Optional[str] = ...) -> None: ...
114
-
115
- class StreamLeasesResponse(_message.Message):
116
- __slots__ = ["leases"]
117
- LEASES_FIELD_NUMBER: _ClassVar[int]
118
- leases: _containers.RepeatedCompositeFieldContainer[Lease]
119
- def __init__(self, leases: _Optional[_Iterable[_Union[Lease, _Mapping]]] = ...) -> None: ...
120
-
121
- class Lease(_message.Message):
122
- __slots__ = ["action_id", "parent_action_name", "input_uri", "run_output_base", "task", "condition", "group", "subject", "previous_state"]
123
- ACTION_ID_FIELD_NUMBER: _ClassVar[int]
124
- PARENT_ACTION_NAME_FIELD_NUMBER: _ClassVar[int]
125
- INPUT_URI_FIELD_NUMBER: _ClassVar[int]
126
- RUN_OUTPUT_BASE_FIELD_NUMBER: _ClassVar[int]
127
- TASK_FIELD_NUMBER: _ClassVar[int]
128
- CONDITION_FIELD_NUMBER: _ClassVar[int]
129
- GROUP_FIELD_NUMBER: _ClassVar[int]
130
- SUBJECT_FIELD_NUMBER: _ClassVar[int]
131
- PREVIOUS_STATE_FIELD_NUMBER: _ClassVar[int]
132
- action_id: _run_definition_pb2.ActionIdentifier
133
- parent_action_name: str
134
- input_uri: str
135
- run_output_base: str
136
- task: TaskAction
137
- condition: ConditionAction
138
- group: str
139
- subject: str
140
- previous_state: str
141
- def __init__(self, action_id: _Optional[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]] = ..., parent_action_name: _Optional[str] = ..., input_uri: _Optional[str] = ..., run_output_base: _Optional[str] = ..., task: _Optional[_Union[TaskAction, _Mapping]] = ..., condition: _Optional[_Union[ConditionAction, _Mapping]] = ..., group: _Optional[str] = ..., subject: _Optional[str] = ..., previous_state: _Optional[str] = ...) -> None: ...
@@ -1,172 +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 queue_service_pb2 as workflow_dot_queue__service__pb2
6
-
7
-
8
- class QueueServiceStub(object):
9
- """provides an interface for managing execution of runs over a collection of workers.
10
- """
11
-
12
- def __init__(self, channel):
13
- """Constructor.
14
-
15
- Args:
16
- channel: A grpc.Channel.
17
- """
18
- self.EnqueueAction = channel.unary_unary(
19
- '/cloudidl.workflow.QueueService/EnqueueAction',
20
- request_serializer=workflow_dot_queue__service__pb2.EnqueueActionRequest.SerializeToString,
21
- response_deserializer=workflow_dot_queue__service__pb2.EnqueueActionResponse.FromString,
22
- )
23
- self.AbortQueuedAction = channel.unary_unary(
24
- '/cloudidl.workflow.QueueService/AbortQueuedAction',
25
- request_serializer=workflow_dot_queue__service__pb2.AbortQueuedActionRequest.SerializeToString,
26
- response_deserializer=workflow_dot_queue__service__pb2.AbortQueuedActionResponse.FromString,
27
- )
28
- self.Heartbeat = channel.stream_stream(
29
- '/cloudidl.workflow.QueueService/Heartbeat',
30
- request_serializer=workflow_dot_queue__service__pb2.HeartbeatRequest.SerializeToString,
31
- response_deserializer=workflow_dot_queue__service__pb2.HeartbeatResponse.FromString,
32
- )
33
- self.StreamLeases = channel.unary_stream(
34
- '/cloudidl.workflow.QueueService/StreamLeases',
35
- request_serializer=workflow_dot_queue__service__pb2.StreamLeasesRequest.SerializeToString,
36
- response_deserializer=workflow_dot_queue__service__pb2.StreamLeasesResponse.FromString,
37
- )
38
-
39
-
40
- class QueueServiceServicer(object):
41
- """provides an interface for managing execution of runs over a collection of workers.
42
- """
43
-
44
- def EnqueueAction(self, request, context):
45
- """queue a new action for execution.
46
- """
47
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
48
- context.set_details('Method not implemented!')
49
- raise NotImplementedError('Method not implemented!')
50
-
51
- def AbortQueuedAction(self, request, context):
52
- """abort a queued action.
53
- """
54
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
55
- context.set_details('Method not implemented!')
56
- raise NotImplementedError('Method not implemented!')
57
-
58
- def Heartbeat(self, request_iterator, context):
59
- """heartbeat is used by workers to report their status and acquire new leases.
60
- """
61
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
62
- context.set_details('Method not implemented!')
63
- raise NotImplementedError('Method not implemented!')
64
-
65
- def StreamLeases(self, request, context):
66
- """Stream leases for a worker.
67
- """
68
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
69
- context.set_details('Method not implemented!')
70
- raise NotImplementedError('Method not implemented!')
71
-
72
-
73
- def add_QueueServiceServicer_to_server(servicer, server):
74
- rpc_method_handlers = {
75
- 'EnqueueAction': grpc.unary_unary_rpc_method_handler(
76
- servicer.EnqueueAction,
77
- request_deserializer=workflow_dot_queue__service__pb2.EnqueueActionRequest.FromString,
78
- response_serializer=workflow_dot_queue__service__pb2.EnqueueActionResponse.SerializeToString,
79
- ),
80
- 'AbortQueuedAction': grpc.unary_unary_rpc_method_handler(
81
- servicer.AbortQueuedAction,
82
- request_deserializer=workflow_dot_queue__service__pb2.AbortQueuedActionRequest.FromString,
83
- response_serializer=workflow_dot_queue__service__pb2.AbortQueuedActionResponse.SerializeToString,
84
- ),
85
- 'Heartbeat': grpc.stream_stream_rpc_method_handler(
86
- servicer.Heartbeat,
87
- request_deserializer=workflow_dot_queue__service__pb2.HeartbeatRequest.FromString,
88
- response_serializer=workflow_dot_queue__service__pb2.HeartbeatResponse.SerializeToString,
89
- ),
90
- 'StreamLeases': grpc.unary_stream_rpc_method_handler(
91
- servicer.StreamLeases,
92
- request_deserializer=workflow_dot_queue__service__pb2.StreamLeasesRequest.FromString,
93
- response_serializer=workflow_dot_queue__service__pb2.StreamLeasesResponse.SerializeToString,
94
- ),
95
- }
96
- generic_handler = grpc.method_handlers_generic_handler(
97
- 'cloudidl.workflow.QueueService', rpc_method_handlers)
98
- server.add_generic_rpc_handlers((generic_handler,))
99
-
100
-
101
- # This class is part of an EXPERIMENTAL API.
102
- class QueueService(object):
103
- """provides an interface for managing execution of runs over a collection of workers.
104
- """
105
-
106
- @staticmethod
107
- def EnqueueAction(request,
108
- target,
109
- options=(),
110
- channel_credentials=None,
111
- call_credentials=None,
112
- insecure=False,
113
- compression=None,
114
- wait_for_ready=None,
115
- timeout=None,
116
- metadata=None):
117
- return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.QueueService/EnqueueAction',
118
- workflow_dot_queue__service__pb2.EnqueueActionRequest.SerializeToString,
119
- workflow_dot_queue__service__pb2.EnqueueActionResponse.FromString,
120
- options, channel_credentials,
121
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
122
-
123
- @staticmethod
124
- def AbortQueuedAction(request,
125
- target,
126
- options=(),
127
- channel_credentials=None,
128
- call_credentials=None,
129
- insecure=False,
130
- compression=None,
131
- wait_for_ready=None,
132
- timeout=None,
133
- metadata=None):
134
- return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.QueueService/AbortQueuedAction',
135
- workflow_dot_queue__service__pb2.AbortQueuedActionRequest.SerializeToString,
136
- workflow_dot_queue__service__pb2.AbortQueuedActionResponse.FromString,
137
- options, channel_credentials,
138
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
139
-
140
- @staticmethod
141
- def Heartbeat(request_iterator,
142
- target,
143
- options=(),
144
- channel_credentials=None,
145
- call_credentials=None,
146
- insecure=False,
147
- compression=None,
148
- wait_for_ready=None,
149
- timeout=None,
150
- metadata=None):
151
- return grpc.experimental.stream_stream(request_iterator, target, '/cloudidl.workflow.QueueService/Heartbeat',
152
- workflow_dot_queue__service__pb2.HeartbeatRequest.SerializeToString,
153
- workflow_dot_queue__service__pb2.HeartbeatResponse.FromString,
154
- options, channel_credentials,
155
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
156
-
157
- @staticmethod
158
- def StreamLeases(request,
159
- target,
160
- options=(),
161
- channel_credentials=None,
162
- call_credentials=None,
163
- insecure=False,
164
- compression=None,
165
- wait_for_ready=None,
166
- timeout=None,
167
- metadata=None):
168
- return grpc.experimental.unary_stream(request, target, '/cloudidl.workflow.QueueService/StreamLeases',
169
- workflow_dot_queue__service__pb2.StreamLeasesRequest.SerializeToString,
170
- workflow_dot_queue__service__pb2.StreamLeasesResponse.FromString,
171
- options, channel_credentials,
172
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
@@ -1,128 +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 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\"\x8d\x01\n\rRunIdentifier\x12\x19\n\x03org\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x03org\x12!\n\x07project\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07project\x12\x1f\n\x06\x64omain\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\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\"d\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\")\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\"\xa9\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\x07outputsB\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\004r\002\020\001'
34
- _RUNIDENTIFIER.fields_by_name['project']._options = None
35
- _RUNIDENTIFIER.fields_by_name['project']._serialized_options = b'\372B\004r\002\020\001'
36
- _RUNIDENTIFIER.fields_by_name['domain']._options = None
37
- _RUNIDENTIFIER.fields_by_name['domain']._serialized_options = b'\372B\004r\002\020\001'
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=4822
71
- _globals['_PHASE']._serialized_end=5025
72
- _globals['_RUNIDENTIFIER']._serialized_start=284
73
- _globals['_RUNIDENTIFIER']._serialized_end=425
74
- _globals['_LABELS']._serialized_start=428
75
- _globals['_LABELS']._serialized_end=558
76
- _globals['_LABELS_VALUESENTRY']._serialized_start=501
77
- _globals['_LABELS_VALUESENTRY']._serialized_end=558
78
- _globals['_ANNOTATIONS']._serialized_start=561
79
- _globals['_ANNOTATIONS']._serialized_end=701
80
- _globals['_ANNOTATIONS_VALUESENTRY']._serialized_start=501
81
- _globals['_ANNOTATIONS_VALUESENTRY']._serialized_end=558
82
- _globals['_ENVS']._serialized_start=703
83
- _globals['_ENVS']._serialized_end=762
84
- _globals['_RUNSPEC']._serialized_start=765
85
- _globals['_RUNSPEC']._serialized_end=1043
86
- _globals['_RUN']._serialized_start=1045
87
- _globals['_RUN']._serialized_end=1101
88
- _globals['_RUNDETAILS']._serialized_start=1103
89
- _globals['_RUNDETAILS']._serialized_end=1228
90
- _globals['_ACTIONIDENTIFIER']._serialized_start=1230
91
- _globals['_ACTIONIDENTIFIER']._serialized_end=1341
92
- _globals['_TASKACTIONMETADATA']._serialized_start=1343
93
- _globals['_TASKACTIONMETADATA']._serialized_end=1443
94
- _globals['_TRACEACTIONMETADATA']._serialized_start=1445
95
- _globals['_TRACEACTIONMETADATA']._serialized_end=1486
96
- _globals['_CONDITIONACTIONMETADATA']._serialized_start=1489
97
- _globals['_CONDITIONACTIONMETADATA']._serialized_end=1648
98
- _globals['_ACTIONMETADATA']._serialized_start=1651
99
- _globals['_ACTIONMETADATA']._serialized_end=2020
100
- _globals['_ACTIONSTATUS']._serialized_start=2023
101
- _globals['_ACTIONSTATUS']._serialized_end=2254
102
- _globals['_ACTION']._serialized_start=2257
103
- _globals['_ACTION']._serialized_end=2438
104
- _globals['_ENRICHEDACTION']._serialized_start=2441
105
- _globals['_ENRICHEDACTION']._serialized_end=2727
106
- _globals['_ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY']._serialized_start=2657
107
- _globals['_ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY']._serialized_end=2727
108
- _globals['_ERRORINFO']._serialized_start=2730
109
- _globals['_ERRORINFO']._serialized_end=2884
110
- _globals['_ERRORINFO_KIND']._serialized_start=2824
111
- _globals['_ERRORINFO_KIND']._serialized_end=2884
112
- _globals['_ABORTINFO']._serialized_start=2886
113
- _globals['_ABORTINFO']._serialized_end=2987
114
- _globals['_ACTIONDETAILS']._serialized_start=2990
115
- _globals['_ACTIONDETAILS']._serialized_end=3451
116
- _globals['_ACTIONATTEMPT']._serialized_start=3454
117
- _globals['_ACTIONATTEMPT']._serialized_end=3879
118
- _globals['_ACTIONEVENT']._serialized_start=3882
119
- _globals['_ACTIONEVENT']._serialized_end=4543
120
- _globals['_NAMEDLITERAL']._serialized_start=4545
121
- _globals['_NAMEDLITERAL']._serialized_end=4625
122
- _globals['_OUTPUTREFERENCES']._serialized_start=4627
123
- _globals['_OUTPUTREFERENCES']._serialized_end=4676
124
- _globals['_INPUTS']._serialized_start=4678
125
- _globals['_INPUTS']._serialized_end=4747
126
- _globals['_OUTPUTS']._serialized_start=4749
127
- _globals['_OUTPUTS']._serialized_end=4819
128
- # @@protoc_insertion_point(module_scope)