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,175 +0,0 @@
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", "cluster_id", "project_id", "task_id"]
109
- REQUEST_FIELD_NUMBER: _ClassVar[int]
110
- ORG_FIELD_NUMBER: _ClassVar[int]
111
- CLUSTER_ID_FIELD_NUMBER: _ClassVar[int]
112
- PROJECT_ID_FIELD_NUMBER: _ClassVar[int]
113
- TASK_ID_FIELD_NUMBER: _ClassVar[int]
114
- request: _list_pb2.ListRequest
115
- org: str
116
- cluster_id: _identifier_pb2.ClusterIdentifier
117
- project_id: _identifier_pb2.ProjectIdentifier
118
- task_id: _task_definition_pb2.TaskIdentifier
119
- def __init__(self, request: _Optional[_Union[_list_pb2.ListRequest, _Mapping]] = ..., 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: ...
120
-
121
- class ListRunsResponse(_message.Message):
122
- __slots__ = ["runs", "token"]
123
- RUNS_FIELD_NUMBER: _ClassVar[int]
124
- TOKEN_FIELD_NUMBER: _ClassVar[int]
125
- runs: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.Run]
126
- token: str
127
- def __init__(self, runs: _Optional[_Iterable[_Union[_run_definition_pb2.Run, _Mapping]]] = ..., token: _Optional[str] = ...) -> None: ...
128
-
129
- class WatchRunsRequest(_message.Message):
130
- __slots__ = ["org", "cluster_id", "project_id", "task_id"]
131
- ORG_FIELD_NUMBER: _ClassVar[int]
132
- CLUSTER_ID_FIELD_NUMBER: _ClassVar[int]
133
- PROJECT_ID_FIELD_NUMBER: _ClassVar[int]
134
- TASK_ID_FIELD_NUMBER: _ClassVar[int]
135
- org: str
136
- cluster_id: _identifier_pb2.ClusterIdentifier
137
- project_id: _identifier_pb2.ProjectIdentifier
138
- task_id: _task_definition_pb2.TaskIdentifier
139
- 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: ...
140
-
141
- class WatchRunsResponse(_message.Message):
142
- __slots__ = ["runs"]
143
- RUNS_FIELD_NUMBER: _ClassVar[int]
144
- runs: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.Run]
145
- def __init__(self, runs: _Optional[_Iterable[_Union[_run_definition_pb2.Run, _Mapping]]] = ...) -> None: ...
146
-
147
- class ListActionsRequest(_message.Message):
148
- __slots__ = ["request", "run_id"]
149
- REQUEST_FIELD_NUMBER: _ClassVar[int]
150
- RUN_ID_FIELD_NUMBER: _ClassVar[int]
151
- request: _list_pb2.ListRequest
152
- run_id: _run_definition_pb2.RunIdentifier
153
- def __init__(self, request: _Optional[_Union[_list_pb2.ListRequest, _Mapping]] = ..., run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
154
-
155
- class ListActionsResponse(_message.Message):
156
- __slots__ = ["actions", "token"]
157
- ACTIONS_FIELD_NUMBER: _ClassVar[int]
158
- TOKEN_FIELD_NUMBER: _ClassVar[int]
159
- actions: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.Action]
160
- token: str
161
- def __init__(self, actions: _Optional[_Iterable[_Union[_run_definition_pb2.Action, _Mapping]]] = ..., token: _Optional[str] = ...) -> None: ...
162
-
163
- class WatchActionsRequest(_message.Message):
164
- __slots__ = ["run_id", "filter"]
165
- RUN_ID_FIELD_NUMBER: _ClassVar[int]
166
- FILTER_FIELD_NUMBER: _ClassVar[int]
167
- run_id: _run_definition_pb2.RunIdentifier
168
- filter: _containers.RepeatedCompositeFieldContainer[_list_pb2.Filter]
169
- def __init__(self, run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ..., filter: _Optional[_Iterable[_Union[_list_pb2.Filter, _Mapping]]] = ...) -> None: ...
170
-
171
- class WatchActionsResponse(_message.Message):
172
- __slots__ = ["enriched_actions"]
173
- ENRICHED_ACTIONS_FIELD_NUMBER: _ClassVar[int]
174
- enriched_actions: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.EnrichedAction]
175
- def __init__(self, enriched_actions: _Optional[_Iterable[_Union[_run_definition_pb2.EnrichedAction, _Mapping]]] = ...) -> None: ...
@@ -1,412 +0,0 @@
1
- # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
- """Client and server classes corresponding to protobuf-defined services."""
3
- import grpc
4
-
5
- from flyte._protos.workflow import run_service_pb2 as workflow_dot_run__service__pb2
6
-
7
-
8
- class RunServiceStub(object):
9
- """RunService provides an interface for managing runs.
10
- """
11
-
12
- def __init__(self, channel):
13
- """Constructor.
14
-
15
- Args:
16
- channel: A grpc.Channel.
17
- """
18
- self.CreateRun = channel.unary_unary(
19
- '/cloudidl.workflow.RunService/CreateRun',
20
- request_serializer=workflow_dot_run__service__pb2.CreateRunRequest.SerializeToString,
21
- response_deserializer=workflow_dot_run__service__pb2.CreateRunResponse.FromString,
22
- )
23
- self.AbortRun = channel.unary_unary(
24
- '/cloudidl.workflow.RunService/AbortRun',
25
- request_serializer=workflow_dot_run__service__pb2.AbortRunRequest.SerializeToString,
26
- response_deserializer=workflow_dot_run__service__pb2.AbortRunResponse.FromString,
27
- )
28
- self.GetRunDetails = channel.unary_unary(
29
- '/cloudidl.workflow.RunService/GetRunDetails',
30
- request_serializer=workflow_dot_run__service__pb2.GetRunDetailsRequest.SerializeToString,
31
- response_deserializer=workflow_dot_run__service__pb2.GetRunDetailsResponse.FromString,
32
- )
33
- self.WatchRunDetails = channel.unary_stream(
34
- '/cloudidl.workflow.RunService/WatchRunDetails',
35
- request_serializer=workflow_dot_run__service__pb2.WatchRunDetailsRequest.SerializeToString,
36
- response_deserializer=workflow_dot_run__service__pb2.WatchRunDetailsResponse.FromString,
37
- )
38
- self.GetActionDetails = channel.unary_unary(
39
- '/cloudidl.workflow.RunService/GetActionDetails',
40
- request_serializer=workflow_dot_run__service__pb2.GetActionDetailsRequest.SerializeToString,
41
- response_deserializer=workflow_dot_run__service__pb2.GetActionDetailsResponse.FromString,
42
- )
43
- self.WatchActionDetails = channel.unary_stream(
44
- '/cloudidl.workflow.RunService/WatchActionDetails',
45
- request_serializer=workflow_dot_run__service__pb2.WatchActionDetailsRequest.SerializeToString,
46
- response_deserializer=workflow_dot_run__service__pb2.WatchActionDetailsResponse.FromString,
47
- )
48
- self.GetActionData = channel.unary_unary(
49
- '/cloudidl.workflow.RunService/GetActionData',
50
- request_serializer=workflow_dot_run__service__pb2.GetActionDataRequest.SerializeToString,
51
- response_deserializer=workflow_dot_run__service__pb2.GetActionDataResponse.FromString,
52
- )
53
- self.ListRuns = channel.unary_unary(
54
- '/cloudidl.workflow.RunService/ListRuns',
55
- request_serializer=workflow_dot_run__service__pb2.ListRunsRequest.SerializeToString,
56
- response_deserializer=workflow_dot_run__service__pb2.ListRunsResponse.FromString,
57
- )
58
- self.WatchRuns = channel.unary_stream(
59
- '/cloudidl.workflow.RunService/WatchRuns',
60
- request_serializer=workflow_dot_run__service__pb2.WatchRunsRequest.SerializeToString,
61
- response_deserializer=workflow_dot_run__service__pb2.WatchRunsResponse.FromString,
62
- )
63
- self.ListActions = channel.unary_unary(
64
- '/cloudidl.workflow.RunService/ListActions',
65
- request_serializer=workflow_dot_run__service__pb2.ListActionsRequest.SerializeToString,
66
- response_deserializer=workflow_dot_run__service__pb2.ListActionsResponse.FromString,
67
- )
68
- self.WatchActions = channel.unary_stream(
69
- '/cloudidl.workflow.RunService/WatchActions',
70
- request_serializer=workflow_dot_run__service__pb2.WatchActionsRequest.SerializeToString,
71
- response_deserializer=workflow_dot_run__service__pb2.WatchActionsResponse.FromString,
72
- )
73
-
74
-
75
- class RunServiceServicer(object):
76
- """RunService provides an interface for managing runs.
77
- """
78
-
79
- def CreateRun(self, request, context):
80
- """Create a new run of the given task.
81
- """
82
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
83
- context.set_details('Method not implemented!')
84
- raise NotImplementedError('Method not implemented!')
85
-
86
- def AbortRun(self, request, context):
87
- """Abort a run.
88
- """
89
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
90
- context.set_details('Method not implemented!')
91
- raise NotImplementedError('Method not implemented!')
92
-
93
- def GetRunDetails(self, request, context):
94
- """Get detailed information about a run.
95
- """
96
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
97
- context.set_details('Method not implemented!')
98
- raise NotImplementedError('Method not implemented!')
99
-
100
- def WatchRunDetails(self, request, context):
101
- """Stream detailed information updates about a run. The call will terminate when the run reaches a terminal phase.
102
- """
103
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
104
- context.set_details('Method not implemented!')
105
- raise NotImplementedError('Method not implemented!')
106
-
107
- def GetActionDetails(self, request, context):
108
- """Get detailed information about an action.
109
- """
110
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
111
- context.set_details('Method not implemented!')
112
- raise NotImplementedError('Method not implemented!')
113
-
114
- def WatchActionDetails(self, request, context):
115
- """Stream detailed information updates about an action. The call will terminate when the action reaches a terminal phase.
116
- """
117
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
118
- context.set_details('Method not implemented!')
119
- raise NotImplementedError('Method not implemented!')
120
-
121
- def GetActionData(self, request, context):
122
- """Get input and output for an action.
123
- """
124
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
125
- context.set_details('Method not implemented!')
126
- raise NotImplementedError('Method not implemented!')
127
-
128
- def ListRuns(self, request, context):
129
- """List runs based on the provided filter criteria.
130
- """
131
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
132
- context.set_details('Method not implemented!')
133
- raise NotImplementedError('Method not implemented!')
134
-
135
- def WatchRuns(self, request, context):
136
- """Stream updates for runs based on the provided filter criteria. Responses may include newly discovered
137
- runs or updates to existing ones from the point of invocation.
138
- """
139
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
140
- context.set_details('Method not implemented!')
141
- raise NotImplementedError('Method not implemented!')
142
-
143
- def ListActions(self, request, context):
144
- """List all actions for a given run.
145
- """
146
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
147
- context.set_details('Method not implemented!')
148
- raise NotImplementedError('Method not implemented!')
149
-
150
- def WatchActions(self, request, context):
151
- """Stream updates for actions given a run. Responses may include newly discovered nested runs or updates
152
- to existing ones from the point of invocation.
153
- """
154
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
155
- context.set_details('Method not implemented!')
156
- raise NotImplementedError('Method not implemented!')
157
-
158
-
159
- def add_RunServiceServicer_to_server(servicer, server):
160
- rpc_method_handlers = {
161
- 'CreateRun': grpc.unary_unary_rpc_method_handler(
162
- servicer.CreateRun,
163
- request_deserializer=workflow_dot_run__service__pb2.CreateRunRequest.FromString,
164
- response_serializer=workflow_dot_run__service__pb2.CreateRunResponse.SerializeToString,
165
- ),
166
- 'AbortRun': grpc.unary_unary_rpc_method_handler(
167
- servicer.AbortRun,
168
- request_deserializer=workflow_dot_run__service__pb2.AbortRunRequest.FromString,
169
- response_serializer=workflow_dot_run__service__pb2.AbortRunResponse.SerializeToString,
170
- ),
171
- 'GetRunDetails': grpc.unary_unary_rpc_method_handler(
172
- servicer.GetRunDetails,
173
- request_deserializer=workflow_dot_run__service__pb2.GetRunDetailsRequest.FromString,
174
- response_serializer=workflow_dot_run__service__pb2.GetRunDetailsResponse.SerializeToString,
175
- ),
176
- 'WatchRunDetails': grpc.unary_stream_rpc_method_handler(
177
- servicer.WatchRunDetails,
178
- request_deserializer=workflow_dot_run__service__pb2.WatchRunDetailsRequest.FromString,
179
- response_serializer=workflow_dot_run__service__pb2.WatchRunDetailsResponse.SerializeToString,
180
- ),
181
- 'GetActionDetails': grpc.unary_unary_rpc_method_handler(
182
- servicer.GetActionDetails,
183
- request_deserializer=workflow_dot_run__service__pb2.GetActionDetailsRequest.FromString,
184
- response_serializer=workflow_dot_run__service__pb2.GetActionDetailsResponse.SerializeToString,
185
- ),
186
- 'WatchActionDetails': grpc.unary_stream_rpc_method_handler(
187
- servicer.WatchActionDetails,
188
- request_deserializer=workflow_dot_run__service__pb2.WatchActionDetailsRequest.FromString,
189
- response_serializer=workflow_dot_run__service__pb2.WatchActionDetailsResponse.SerializeToString,
190
- ),
191
- 'GetActionData': grpc.unary_unary_rpc_method_handler(
192
- servicer.GetActionData,
193
- request_deserializer=workflow_dot_run__service__pb2.GetActionDataRequest.FromString,
194
- response_serializer=workflow_dot_run__service__pb2.GetActionDataResponse.SerializeToString,
195
- ),
196
- 'ListRuns': grpc.unary_unary_rpc_method_handler(
197
- servicer.ListRuns,
198
- request_deserializer=workflow_dot_run__service__pb2.ListRunsRequest.FromString,
199
- response_serializer=workflow_dot_run__service__pb2.ListRunsResponse.SerializeToString,
200
- ),
201
- 'WatchRuns': grpc.unary_stream_rpc_method_handler(
202
- servicer.WatchRuns,
203
- request_deserializer=workflow_dot_run__service__pb2.WatchRunsRequest.FromString,
204
- response_serializer=workflow_dot_run__service__pb2.WatchRunsResponse.SerializeToString,
205
- ),
206
- 'ListActions': grpc.unary_unary_rpc_method_handler(
207
- servicer.ListActions,
208
- request_deserializer=workflow_dot_run__service__pb2.ListActionsRequest.FromString,
209
- response_serializer=workflow_dot_run__service__pb2.ListActionsResponse.SerializeToString,
210
- ),
211
- 'WatchActions': grpc.unary_stream_rpc_method_handler(
212
- servicer.WatchActions,
213
- request_deserializer=workflow_dot_run__service__pb2.WatchActionsRequest.FromString,
214
- response_serializer=workflow_dot_run__service__pb2.WatchActionsResponse.SerializeToString,
215
- ),
216
- }
217
- generic_handler = grpc.method_handlers_generic_handler(
218
- 'cloudidl.workflow.RunService', rpc_method_handlers)
219
- server.add_generic_rpc_handlers((generic_handler,))
220
-
221
-
222
- # This class is part of an EXPERIMENTAL API.
223
- class RunService(object):
224
- """RunService provides an interface for managing runs.
225
- """
226
-
227
- @staticmethod
228
- def CreateRun(request,
229
- target,
230
- options=(),
231
- channel_credentials=None,
232
- call_credentials=None,
233
- insecure=False,
234
- compression=None,
235
- wait_for_ready=None,
236
- timeout=None,
237
- metadata=None):
238
- return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.RunService/CreateRun',
239
- workflow_dot_run__service__pb2.CreateRunRequest.SerializeToString,
240
- workflow_dot_run__service__pb2.CreateRunResponse.FromString,
241
- options, channel_credentials,
242
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
243
-
244
- @staticmethod
245
- def AbortRun(request,
246
- target,
247
- options=(),
248
- channel_credentials=None,
249
- call_credentials=None,
250
- insecure=False,
251
- compression=None,
252
- wait_for_ready=None,
253
- timeout=None,
254
- metadata=None):
255
- return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.RunService/AbortRun',
256
- workflow_dot_run__service__pb2.AbortRunRequest.SerializeToString,
257
- workflow_dot_run__service__pb2.AbortRunResponse.FromString,
258
- options, channel_credentials,
259
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
260
-
261
- @staticmethod
262
- def GetRunDetails(request,
263
- target,
264
- options=(),
265
- channel_credentials=None,
266
- call_credentials=None,
267
- insecure=False,
268
- compression=None,
269
- wait_for_ready=None,
270
- timeout=None,
271
- metadata=None):
272
- return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.RunService/GetRunDetails',
273
- workflow_dot_run__service__pb2.GetRunDetailsRequest.SerializeToString,
274
- workflow_dot_run__service__pb2.GetRunDetailsResponse.FromString,
275
- options, channel_credentials,
276
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
277
-
278
- @staticmethod
279
- def WatchRunDetails(request,
280
- target,
281
- options=(),
282
- channel_credentials=None,
283
- call_credentials=None,
284
- insecure=False,
285
- compression=None,
286
- wait_for_ready=None,
287
- timeout=None,
288
- metadata=None):
289
- return grpc.experimental.unary_stream(request, target, '/cloudidl.workflow.RunService/WatchRunDetails',
290
- workflow_dot_run__service__pb2.WatchRunDetailsRequest.SerializeToString,
291
- workflow_dot_run__service__pb2.WatchRunDetailsResponse.FromString,
292
- options, channel_credentials,
293
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
294
-
295
- @staticmethod
296
- def GetActionDetails(request,
297
- target,
298
- options=(),
299
- channel_credentials=None,
300
- call_credentials=None,
301
- insecure=False,
302
- compression=None,
303
- wait_for_ready=None,
304
- timeout=None,
305
- metadata=None):
306
- return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.RunService/GetActionDetails',
307
- workflow_dot_run__service__pb2.GetActionDetailsRequest.SerializeToString,
308
- workflow_dot_run__service__pb2.GetActionDetailsResponse.FromString,
309
- options, channel_credentials,
310
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
311
-
312
- @staticmethod
313
- def WatchActionDetails(request,
314
- target,
315
- options=(),
316
- channel_credentials=None,
317
- call_credentials=None,
318
- insecure=False,
319
- compression=None,
320
- wait_for_ready=None,
321
- timeout=None,
322
- metadata=None):
323
- return grpc.experimental.unary_stream(request, target, '/cloudidl.workflow.RunService/WatchActionDetails',
324
- workflow_dot_run__service__pb2.WatchActionDetailsRequest.SerializeToString,
325
- workflow_dot_run__service__pb2.WatchActionDetailsResponse.FromString,
326
- options, channel_credentials,
327
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
328
-
329
- @staticmethod
330
- def GetActionData(request,
331
- target,
332
- options=(),
333
- channel_credentials=None,
334
- call_credentials=None,
335
- insecure=False,
336
- compression=None,
337
- wait_for_ready=None,
338
- timeout=None,
339
- metadata=None):
340
- return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.RunService/GetActionData',
341
- workflow_dot_run__service__pb2.GetActionDataRequest.SerializeToString,
342
- workflow_dot_run__service__pb2.GetActionDataResponse.FromString,
343
- options, channel_credentials,
344
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
345
-
346
- @staticmethod
347
- def ListRuns(request,
348
- target,
349
- options=(),
350
- channel_credentials=None,
351
- call_credentials=None,
352
- insecure=False,
353
- compression=None,
354
- wait_for_ready=None,
355
- timeout=None,
356
- metadata=None):
357
- return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.RunService/ListRuns',
358
- workflow_dot_run__service__pb2.ListRunsRequest.SerializeToString,
359
- workflow_dot_run__service__pb2.ListRunsResponse.FromString,
360
- options, channel_credentials,
361
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
362
-
363
- @staticmethod
364
- def WatchRuns(request,
365
- target,
366
- options=(),
367
- channel_credentials=None,
368
- call_credentials=None,
369
- insecure=False,
370
- compression=None,
371
- wait_for_ready=None,
372
- timeout=None,
373
- metadata=None):
374
- return grpc.experimental.unary_stream(request, target, '/cloudidl.workflow.RunService/WatchRuns',
375
- workflow_dot_run__service__pb2.WatchRunsRequest.SerializeToString,
376
- workflow_dot_run__service__pb2.WatchRunsResponse.FromString,
377
- options, channel_credentials,
378
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
379
-
380
- @staticmethod
381
- def ListActions(request,
382
- target,
383
- options=(),
384
- channel_credentials=None,
385
- call_credentials=None,
386
- insecure=False,
387
- compression=None,
388
- wait_for_ready=None,
389
- timeout=None,
390
- metadata=None):
391
- return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.RunService/ListActions',
392
- workflow_dot_run__service__pb2.ListActionsRequest.SerializeToString,
393
- workflow_dot_run__service__pb2.ListActionsResponse.FromString,
394
- options, channel_credentials,
395
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
396
-
397
- @staticmethod
398
- def WatchActions(request,
399
- target,
400
- options=(),
401
- channel_credentials=None,
402
- call_credentials=None,
403
- insecure=False,
404
- compression=None,
405
- wait_for_ready=None,
406
- timeout=None,
407
- metadata=None):
408
- return grpc.experimental.unary_stream(request, target, '/cloudidl.workflow.RunService/WatchActions',
409
- workflow_dot_run__service__pb2.WatchActionsRequest.SerializeToString,
410
- workflow_dot_run__service__pb2.WatchActionsResponse.FromString,
411
- options, channel_credentials,
412
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
@@ -1,58 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- # Generated by the protocol buffer compiler. DO NOT EDIT!
3
- # source: workflow/state_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 execution_pb2 as flyteidl_dot_core_dot_execution__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/state_service.proto\x12\x11\x63loudidl.workflow\x1a\x1d\x66lyteidl/core/execution.proto\x1a\x17validate/validate.proto\x1a\x1dworkflow/run_definition.proto\x1a\x1eworkflow/task_definition.proto\"\xc3\x01\n\x0cStoreRequest\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\x00R\x10parentActionName\x88\x01\x01\x12\x1d\n\x05state\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x05stateB\x15\n\x13_parent_action_name\"\x0f\n\rStoreResponse\"\xb6\x01\n\x0bLoadRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\x12:\n\x07task_id\x18\x02 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierR\x06taskId\x12\x1f\n\x0binputs_hash\x18\x03 \x01(\x03R\ninputsHash\"-\n\x0cLoadResponse\x12\x1d\n\x05state\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x05state\"n\n\x0cWatchRequest\x12O\n\x10parent_action_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierH\x00R\x0eparentActionIdB\r\n\x06\x66ilter\x12\x03\xf8\x42\x01\"\xb0\x01\n\rWatchResponse\x12\x46\n\raction_update\x18\x01 \x01(\x0b\x32\x1f.cloudidl.workflow.ActionUpdateH\x00R\x0c\x61\x63tionUpdate\x12L\n\x0f\x63ontrol_message\x18\x02 \x01(\x0b\x32!.cloudidl.workflow.ControlMessageH\x00R\x0e\x63ontrolMessageB\t\n\x07message\",\n\x0e\x43ontrolMessage\x12\x1a\n\x08sentinel\x18\x01 \x01(\x08R\x08sentinel\"\xed\x01\n\x0c\x41\x63tionUpdate\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\x12.\n\x05phase\x18\x02 \x01(\x0e\x32\x18.cloudidl.workflow.PhaseR\x05phase\x12\x38\n\x05\x65rror\x18\x03 \x01(\x0b\x32\x1d.flyteidl.core.ExecutionErrorH\x00R\x05\x65rror\x88\x01\x01\x12\x1d\n\noutput_uri\x18\x04 \x01(\tR\toutputUriB\x08\n\x06_error2\xf7\x01\n\x0cStateService\x12L\n\x05Store\x12\x1f.cloudidl.workflow.StoreRequest\x1a .cloudidl.workflow.StoreResponse\"\x00\x12I\n\x04Load\x12\x1e.cloudidl.workflow.LoadRequest\x1a\x1f.cloudidl.workflow.LoadResponse\"\x00\x12N\n\x05Watch\x12\x1f.cloudidl.workflow.WatchRequest\x1a .cloudidl.workflow.WatchResponse\"\x00\x30\x01\x42\xbe\x01\n\x15\x63om.cloudidl.workflowB\x11StateServiceProtoH\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.state_service_pb2', _globals)
25
- if _descriptor._USE_C_DESCRIPTORS == False:
26
- DESCRIPTOR._options = None
27
- DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\021StateServiceProtoH\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
- _STOREREQUEST.fields_by_name['action_id']._options = None
29
- _STOREREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
30
- _STOREREQUEST.fields_by_name['state']._options = None
31
- _STOREREQUEST.fields_by_name['state']._serialized_options = b'\372B\004r\002\020\001'
32
- _LOADREQUEST.fields_by_name['action_id']._options = None
33
- _LOADREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
34
- _LOADRESPONSE.fields_by_name['state']._options = None
35
- _LOADRESPONSE.fields_by_name['state']._serialized_options = b'\372B\004r\002\020\001'
36
- _WATCHREQUEST.oneofs_by_name['filter']._options = None
37
- _WATCHREQUEST.oneofs_by_name['filter']._serialized_options = b'\370B\001'
38
- _ACTIONUPDATE.fields_by_name['action_id']._options = None
39
- _ACTIONUPDATE.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
40
- _globals['_STOREREQUEST']._serialized_start=171
41
- _globals['_STOREREQUEST']._serialized_end=366
42
- _globals['_STORERESPONSE']._serialized_start=368
43
- _globals['_STORERESPONSE']._serialized_end=383
44
- _globals['_LOADREQUEST']._serialized_start=386
45
- _globals['_LOADREQUEST']._serialized_end=568
46
- _globals['_LOADRESPONSE']._serialized_start=570
47
- _globals['_LOADRESPONSE']._serialized_end=615
48
- _globals['_WATCHREQUEST']._serialized_start=617
49
- _globals['_WATCHREQUEST']._serialized_end=727
50
- _globals['_WATCHRESPONSE']._serialized_start=730
51
- _globals['_WATCHRESPONSE']._serialized_end=906
52
- _globals['_CONTROLMESSAGE']._serialized_start=908
53
- _globals['_CONTROLMESSAGE']._serialized_end=952
54
- _globals['_ACTIONUPDATE']._serialized_start=955
55
- _globals['_ACTIONUPDATE']._serialized_end=1192
56
- _globals['_STATESERVICE']._serialized_start=1195
57
- _globals['_STATESERVICE']._serialized_end=1442
58
- # @@protoc_insertion_point(module_scope)