flyte 0.0.1b0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of flyte might be problematic. Click here for more details.

Files changed (390) hide show
  1. flyte/__init__.py +62 -0
  2. flyte/_api_commons.py +3 -0
  3. flyte/_bin/__init__.py +0 -0
  4. flyte/_bin/runtime.py +126 -0
  5. flyte/_build.py +25 -0
  6. flyte/_cache/__init__.py +12 -0
  7. flyte/_cache/cache.py +146 -0
  8. flyte/_cache/defaults.py +9 -0
  9. flyte/_cache/policy_function_body.py +42 -0
  10. flyte/_cli/__init__.py +0 -0
  11. flyte/_cli/_common.py +287 -0
  12. flyte/_cli/_create.py +42 -0
  13. flyte/_cli/_delete.py +23 -0
  14. flyte/_cli/_deploy.py +140 -0
  15. flyte/_cli/_get.py +235 -0
  16. flyte/_cli/_run.py +152 -0
  17. flyte/_cli/main.py +72 -0
  18. flyte/_code_bundle/__init__.py +8 -0
  19. flyte/_code_bundle/_ignore.py +113 -0
  20. flyte/_code_bundle/_packaging.py +187 -0
  21. flyte/_code_bundle/_utils.py +339 -0
  22. flyte/_code_bundle/bundle.py +178 -0
  23. flyte/_context.py +146 -0
  24. flyte/_datastructures.py +342 -0
  25. flyte/_deploy.py +202 -0
  26. flyte/_doc.py +29 -0
  27. flyte/_docstring.py +32 -0
  28. flyte/_environment.py +43 -0
  29. flyte/_group.py +31 -0
  30. flyte/_hash.py +23 -0
  31. flyte/_image.py +760 -0
  32. flyte/_initialize.py +634 -0
  33. flyte/_interface.py +84 -0
  34. flyte/_internal/__init__.py +3 -0
  35. flyte/_internal/controllers/__init__.py +115 -0
  36. flyte/_internal/controllers/_local_controller.py +118 -0
  37. flyte/_internal/controllers/_trace.py +40 -0
  38. flyte/_internal/controllers/pbhash.py +39 -0
  39. flyte/_internal/controllers/remote/__init__.py +40 -0
  40. flyte/_internal/controllers/remote/_action.py +141 -0
  41. flyte/_internal/controllers/remote/_client.py +43 -0
  42. flyte/_internal/controllers/remote/_controller.py +361 -0
  43. flyte/_internal/controllers/remote/_core.py +402 -0
  44. flyte/_internal/controllers/remote/_informer.py +361 -0
  45. flyte/_internal/controllers/remote/_service_protocol.py +50 -0
  46. flyte/_internal/imagebuild/__init__.py +11 -0
  47. flyte/_internal/imagebuild/docker_builder.py +416 -0
  48. flyte/_internal/imagebuild/image_builder.py +241 -0
  49. flyte/_internal/imagebuild/remote_builder.py +0 -0
  50. flyte/_internal/resolvers/__init__.py +0 -0
  51. flyte/_internal/resolvers/_task_module.py +54 -0
  52. flyte/_internal/resolvers/common.py +31 -0
  53. flyte/_internal/resolvers/default.py +28 -0
  54. flyte/_internal/runtime/__init__.py +0 -0
  55. flyte/_internal/runtime/convert.py +199 -0
  56. flyte/_internal/runtime/entrypoints.py +135 -0
  57. flyte/_internal/runtime/io.py +136 -0
  58. flyte/_internal/runtime/resources_serde.py +138 -0
  59. flyte/_internal/runtime/task_serde.py +210 -0
  60. flyte/_internal/runtime/taskrunner.py +190 -0
  61. flyte/_internal/runtime/types_serde.py +54 -0
  62. flyte/_logging.py +124 -0
  63. flyte/_protos/__init__.py +0 -0
  64. flyte/_protos/common/authorization_pb2.py +66 -0
  65. flyte/_protos/common/authorization_pb2.pyi +108 -0
  66. flyte/_protos/common/authorization_pb2_grpc.py +4 -0
  67. flyte/_protos/common/identifier_pb2.py +71 -0
  68. flyte/_protos/common/identifier_pb2.pyi +82 -0
  69. flyte/_protos/common/identifier_pb2_grpc.py +4 -0
  70. flyte/_protos/common/identity_pb2.py +48 -0
  71. flyte/_protos/common/identity_pb2.pyi +72 -0
  72. flyte/_protos/common/identity_pb2_grpc.py +4 -0
  73. flyte/_protos/common/list_pb2.py +36 -0
  74. flyte/_protos/common/list_pb2.pyi +69 -0
  75. flyte/_protos/common/list_pb2_grpc.py +4 -0
  76. flyte/_protos/common/policy_pb2.py +37 -0
  77. flyte/_protos/common/policy_pb2.pyi +27 -0
  78. flyte/_protos/common/policy_pb2_grpc.py +4 -0
  79. flyte/_protos/common/role_pb2.py +37 -0
  80. flyte/_protos/common/role_pb2.pyi +53 -0
  81. flyte/_protos/common/role_pb2_grpc.py +4 -0
  82. flyte/_protos/common/runtime_version_pb2.py +28 -0
  83. flyte/_protos/common/runtime_version_pb2.pyi +24 -0
  84. flyte/_protos/common/runtime_version_pb2_grpc.py +4 -0
  85. flyte/_protos/logs/dataplane/payload_pb2.py +96 -0
  86. flyte/_protos/logs/dataplane/payload_pb2.pyi +168 -0
  87. flyte/_protos/logs/dataplane/payload_pb2_grpc.py +4 -0
  88. flyte/_protos/secret/definition_pb2.py +49 -0
  89. flyte/_protos/secret/definition_pb2.pyi +93 -0
  90. flyte/_protos/secret/definition_pb2_grpc.py +4 -0
  91. flyte/_protos/secret/payload_pb2.py +62 -0
  92. flyte/_protos/secret/payload_pb2.pyi +94 -0
  93. flyte/_protos/secret/payload_pb2_grpc.py +4 -0
  94. flyte/_protos/secret/secret_pb2.py +38 -0
  95. flyte/_protos/secret/secret_pb2.pyi +6 -0
  96. flyte/_protos/secret/secret_pb2_grpc.py +198 -0
  97. flyte/_protos/secret/secret_pb2_grpc_grpc.py +198 -0
  98. flyte/_protos/validate/validate/validate_pb2.py +76 -0
  99. flyte/_protos/workflow/node_execution_service_pb2.py +26 -0
  100. flyte/_protos/workflow/node_execution_service_pb2.pyi +4 -0
  101. flyte/_protos/workflow/node_execution_service_pb2_grpc.py +32 -0
  102. flyte/_protos/workflow/queue_service_pb2.py +106 -0
  103. flyte/_protos/workflow/queue_service_pb2.pyi +141 -0
  104. flyte/_protos/workflow/queue_service_pb2_grpc.py +172 -0
  105. flyte/_protos/workflow/run_definition_pb2.py +128 -0
  106. flyte/_protos/workflow/run_definition_pb2.pyi +310 -0
  107. flyte/_protos/workflow/run_definition_pb2_grpc.py +4 -0
  108. flyte/_protos/workflow/run_logs_service_pb2.py +41 -0
  109. flyte/_protos/workflow/run_logs_service_pb2.pyi +28 -0
  110. flyte/_protos/workflow/run_logs_service_pb2_grpc.py +69 -0
  111. flyte/_protos/workflow/run_service_pb2.py +133 -0
  112. flyte/_protos/workflow/run_service_pb2.pyi +175 -0
  113. flyte/_protos/workflow/run_service_pb2_grpc.py +412 -0
  114. flyte/_protos/workflow/state_service_pb2.py +58 -0
  115. flyte/_protos/workflow/state_service_pb2.pyi +71 -0
  116. flyte/_protos/workflow/state_service_pb2_grpc.py +138 -0
  117. flyte/_protos/workflow/task_definition_pb2.py +72 -0
  118. flyte/_protos/workflow/task_definition_pb2.pyi +65 -0
  119. flyte/_protos/workflow/task_definition_pb2_grpc.py +4 -0
  120. flyte/_protos/workflow/task_service_pb2.py +44 -0
  121. flyte/_protos/workflow/task_service_pb2.pyi +31 -0
  122. flyte/_protos/workflow/task_service_pb2_grpc.py +104 -0
  123. flyte/_resources.py +226 -0
  124. flyte/_retry.py +32 -0
  125. flyte/_reusable_environment.py +25 -0
  126. flyte/_run.py +411 -0
  127. flyte/_secret.py +61 -0
  128. flyte/_task.py +367 -0
  129. flyte/_task_environment.py +200 -0
  130. flyte/_timeout.py +47 -0
  131. flyte/_tools.py +27 -0
  132. flyte/_trace.py +128 -0
  133. flyte/_utils/__init__.py +20 -0
  134. flyte/_utils/asyn.py +119 -0
  135. flyte/_utils/coro_management.py +25 -0
  136. flyte/_utils/file_handling.py +72 -0
  137. flyte/_utils/helpers.py +108 -0
  138. flyte/_utils/lazy_module.py +54 -0
  139. flyte/_utils/uv_script_parser.py +49 -0
  140. flyte/_version.py +21 -0
  141. flyte/connectors/__init__.py +0 -0
  142. flyte/errors.py +143 -0
  143. flyte/extras/__init__.py +5 -0
  144. flyte/extras/_container.py +273 -0
  145. flyte/io/__init__.py +11 -0
  146. flyte/io/_dataframe.py +0 -0
  147. flyte/io/_dir.py +448 -0
  148. flyte/io/_file.py +468 -0
  149. flyte/io/pickle/__init__.py +0 -0
  150. flyte/io/pickle/transformer.py +117 -0
  151. flyte/io/structured_dataset/__init__.py +129 -0
  152. flyte/io/structured_dataset/basic_dfs.py +219 -0
  153. flyte/io/structured_dataset/structured_dataset.py +1061 -0
  154. flyte/py.typed +0 -0
  155. flyte/remote/__init__.py +25 -0
  156. flyte/remote/_client/__init__.py +0 -0
  157. flyte/remote/_client/_protocols.py +131 -0
  158. flyte/remote/_client/auth/__init__.py +12 -0
  159. flyte/remote/_client/auth/_authenticators/__init__.py +0 -0
  160. flyte/remote/_client/auth/_authenticators/base.py +397 -0
  161. flyte/remote/_client/auth/_authenticators/client_credentials.py +73 -0
  162. flyte/remote/_client/auth/_authenticators/device_code.py +118 -0
  163. flyte/remote/_client/auth/_authenticators/external_command.py +79 -0
  164. flyte/remote/_client/auth/_authenticators/factory.py +200 -0
  165. flyte/remote/_client/auth/_authenticators/pkce.py +516 -0
  166. flyte/remote/_client/auth/_channel.py +184 -0
  167. flyte/remote/_client/auth/_client_config.py +83 -0
  168. flyte/remote/_client/auth/_default_html.py +32 -0
  169. flyte/remote/_client/auth/_grpc_utils/__init__.py +0 -0
  170. flyte/remote/_client/auth/_grpc_utils/auth_interceptor.py +288 -0
  171. flyte/remote/_client/auth/_grpc_utils/default_metadata_interceptor.py +151 -0
  172. flyte/remote/_client/auth/_keyring.py +143 -0
  173. flyte/remote/_client/auth/_token_client.py +260 -0
  174. flyte/remote/_client/auth/errors.py +16 -0
  175. flyte/remote/_client/controlplane.py +95 -0
  176. flyte/remote/_console.py +18 -0
  177. flyte/remote/_data.py +155 -0
  178. flyte/remote/_logs.py +116 -0
  179. flyte/remote/_project.py +86 -0
  180. flyte/remote/_run.py +873 -0
  181. flyte/remote/_secret.py +132 -0
  182. flyte/remote/_task.py +227 -0
  183. flyte/report/__init__.py +3 -0
  184. flyte/report/_report.py +178 -0
  185. flyte/report/_template.html +124 -0
  186. flyte/storage/__init__.py +24 -0
  187. flyte/storage/_remote_fs.py +34 -0
  188. flyte/storage/_storage.py +251 -0
  189. flyte/storage/_utils.py +5 -0
  190. flyte/types/__init__.py +13 -0
  191. flyte/types/_interface.py +25 -0
  192. flyte/types/_renderer.py +162 -0
  193. flyte/types/_string_literals.py +120 -0
  194. flyte/types/_type_engine.py +2210 -0
  195. flyte/types/_utils.py +80 -0
  196. flyte-0.0.1b0.dist-info/METADATA +179 -0
  197. flyte-0.0.1b0.dist-info/RECORD +390 -0
  198. flyte-0.0.1b0.dist-info/WHEEL +5 -0
  199. flyte-0.0.1b0.dist-info/entry_points.txt +3 -0
  200. flyte-0.0.1b0.dist-info/top_level.txt +1 -0
  201. union/__init__.py +54 -0
  202. union/_api_commons.py +3 -0
  203. union/_bin/__init__.py +0 -0
  204. union/_bin/runtime.py +113 -0
  205. union/_build.py +25 -0
  206. union/_cache/__init__.py +12 -0
  207. union/_cache/cache.py +141 -0
  208. union/_cache/defaults.py +9 -0
  209. union/_cache/policy_function_body.py +42 -0
  210. union/_cli/__init__.py +0 -0
  211. union/_cli/_common.py +263 -0
  212. union/_cli/_create.py +40 -0
  213. union/_cli/_delete.py +23 -0
  214. union/_cli/_deploy.py +120 -0
  215. union/_cli/_get.py +162 -0
  216. union/_cli/_params.py +579 -0
  217. union/_cli/_run.py +150 -0
  218. union/_cli/main.py +72 -0
  219. union/_code_bundle/__init__.py +8 -0
  220. union/_code_bundle/_ignore.py +113 -0
  221. union/_code_bundle/_packaging.py +187 -0
  222. union/_code_bundle/_utils.py +342 -0
  223. union/_code_bundle/bundle.py +176 -0
  224. union/_context.py +146 -0
  225. union/_datastructures.py +295 -0
  226. union/_deploy.py +185 -0
  227. union/_doc.py +29 -0
  228. union/_docstring.py +26 -0
  229. union/_environment.py +43 -0
  230. union/_group.py +31 -0
  231. union/_hash.py +23 -0
  232. union/_image.py +760 -0
  233. union/_initialize.py +585 -0
  234. union/_interface.py +84 -0
  235. union/_internal/__init__.py +3 -0
  236. union/_internal/controllers/__init__.py +77 -0
  237. union/_internal/controllers/_local_controller.py +77 -0
  238. union/_internal/controllers/pbhash.py +39 -0
  239. union/_internal/controllers/remote/__init__.py +40 -0
  240. union/_internal/controllers/remote/_action.py +131 -0
  241. union/_internal/controllers/remote/_client.py +43 -0
  242. union/_internal/controllers/remote/_controller.py +169 -0
  243. union/_internal/controllers/remote/_core.py +341 -0
  244. union/_internal/controllers/remote/_informer.py +260 -0
  245. union/_internal/controllers/remote/_service_protocol.py +44 -0
  246. union/_internal/imagebuild/__init__.py +11 -0
  247. union/_internal/imagebuild/docker_builder.py +416 -0
  248. union/_internal/imagebuild/image_builder.py +243 -0
  249. union/_internal/imagebuild/remote_builder.py +0 -0
  250. union/_internal/resolvers/__init__.py +0 -0
  251. union/_internal/resolvers/_task_module.py +31 -0
  252. union/_internal/resolvers/common.py +24 -0
  253. union/_internal/resolvers/default.py +27 -0
  254. union/_internal/runtime/__init__.py +0 -0
  255. union/_internal/runtime/convert.py +163 -0
  256. union/_internal/runtime/entrypoints.py +121 -0
  257. union/_internal/runtime/io.py +136 -0
  258. union/_internal/runtime/resources_serde.py +134 -0
  259. union/_internal/runtime/task_serde.py +202 -0
  260. union/_internal/runtime/taskrunner.py +179 -0
  261. union/_internal/runtime/types_serde.py +53 -0
  262. union/_logging.py +124 -0
  263. union/_protos/__init__.py +0 -0
  264. union/_protos/common/authorization_pb2.py +66 -0
  265. union/_protos/common/authorization_pb2.pyi +106 -0
  266. union/_protos/common/authorization_pb2_grpc.py +4 -0
  267. union/_protos/common/identifier_pb2.py +71 -0
  268. union/_protos/common/identifier_pb2.pyi +82 -0
  269. union/_protos/common/identifier_pb2_grpc.py +4 -0
  270. union/_protos/common/identity_pb2.py +48 -0
  271. union/_protos/common/identity_pb2.pyi +72 -0
  272. union/_protos/common/identity_pb2_grpc.py +4 -0
  273. union/_protos/common/list_pb2.py +36 -0
  274. union/_protos/common/list_pb2.pyi +69 -0
  275. union/_protos/common/list_pb2_grpc.py +4 -0
  276. union/_protos/common/policy_pb2.py +37 -0
  277. union/_protos/common/policy_pb2.pyi +27 -0
  278. union/_protos/common/policy_pb2_grpc.py +4 -0
  279. union/_protos/common/role_pb2.py +37 -0
  280. union/_protos/common/role_pb2.pyi +51 -0
  281. union/_protos/common/role_pb2_grpc.py +4 -0
  282. union/_protos/common/runtime_version_pb2.py +28 -0
  283. union/_protos/common/runtime_version_pb2.pyi +24 -0
  284. union/_protos/common/runtime_version_pb2_grpc.py +4 -0
  285. union/_protos/logs/dataplane/payload_pb2.py +96 -0
  286. union/_protos/logs/dataplane/payload_pb2.pyi +168 -0
  287. union/_protos/logs/dataplane/payload_pb2_grpc.py +4 -0
  288. union/_protos/secret/definition_pb2.py +49 -0
  289. union/_protos/secret/definition_pb2.pyi +93 -0
  290. union/_protos/secret/definition_pb2_grpc.py +4 -0
  291. union/_protos/secret/payload_pb2.py +62 -0
  292. union/_protos/secret/payload_pb2.pyi +94 -0
  293. union/_protos/secret/payload_pb2_grpc.py +4 -0
  294. union/_protos/secret/secret_pb2.py +38 -0
  295. union/_protos/secret/secret_pb2.pyi +6 -0
  296. union/_protos/secret/secret_pb2_grpc.py +198 -0
  297. union/_protos/validate/validate/validate_pb2.py +76 -0
  298. union/_protos/workflow/node_execution_service_pb2.py +26 -0
  299. union/_protos/workflow/node_execution_service_pb2.pyi +4 -0
  300. union/_protos/workflow/node_execution_service_pb2_grpc.py +32 -0
  301. union/_protos/workflow/queue_service_pb2.py +75 -0
  302. union/_protos/workflow/queue_service_pb2.pyi +103 -0
  303. union/_protos/workflow/queue_service_pb2_grpc.py +172 -0
  304. union/_protos/workflow/run_definition_pb2.py +100 -0
  305. union/_protos/workflow/run_definition_pb2.pyi +256 -0
  306. union/_protos/workflow/run_definition_pb2_grpc.py +4 -0
  307. union/_protos/workflow/run_logs_service_pb2.py +41 -0
  308. union/_protos/workflow/run_logs_service_pb2.pyi +28 -0
  309. union/_protos/workflow/run_logs_service_pb2_grpc.py +69 -0
  310. union/_protos/workflow/run_service_pb2.py +133 -0
  311. union/_protos/workflow/run_service_pb2.pyi +173 -0
  312. union/_protos/workflow/run_service_pb2_grpc.py +412 -0
  313. union/_protos/workflow/state_service_pb2.py +58 -0
  314. union/_protos/workflow/state_service_pb2.pyi +69 -0
  315. union/_protos/workflow/state_service_pb2_grpc.py +138 -0
  316. union/_protos/workflow/task_definition_pb2.py +72 -0
  317. union/_protos/workflow/task_definition_pb2.pyi +65 -0
  318. union/_protos/workflow/task_definition_pb2_grpc.py +4 -0
  319. union/_protos/workflow/task_service_pb2.py +44 -0
  320. union/_protos/workflow/task_service_pb2.pyi +31 -0
  321. union/_protos/workflow/task_service_pb2_grpc.py +104 -0
  322. union/_resources.py +226 -0
  323. union/_retry.py +32 -0
  324. union/_reusable_environment.py +25 -0
  325. union/_run.py +374 -0
  326. union/_secret.py +61 -0
  327. union/_task.py +354 -0
  328. union/_task_environment.py +186 -0
  329. union/_timeout.py +47 -0
  330. union/_tools.py +27 -0
  331. union/_utils/__init__.py +11 -0
  332. union/_utils/asyn.py +119 -0
  333. union/_utils/file_handling.py +71 -0
  334. union/_utils/helpers.py +46 -0
  335. union/_utils/lazy_module.py +54 -0
  336. union/_utils/uv_script_parser.py +49 -0
  337. union/_version.py +21 -0
  338. union/connectors/__init__.py +0 -0
  339. union/errors.py +128 -0
  340. union/extras/__init__.py +5 -0
  341. union/extras/_container.py +263 -0
  342. union/io/__init__.py +11 -0
  343. union/io/_dataframe.py +0 -0
  344. union/io/_dir.py +425 -0
  345. union/io/_file.py +418 -0
  346. union/io/pickle/__init__.py +0 -0
  347. union/io/pickle/transformer.py +117 -0
  348. union/io/structured_dataset/__init__.py +122 -0
  349. union/io/structured_dataset/basic_dfs.py +219 -0
  350. union/io/structured_dataset/structured_dataset.py +1057 -0
  351. union/py.typed +0 -0
  352. union/remote/__init__.py +23 -0
  353. union/remote/_client/__init__.py +0 -0
  354. union/remote/_client/_protocols.py +129 -0
  355. union/remote/_client/auth/__init__.py +12 -0
  356. union/remote/_client/auth/_authenticators/__init__.py +0 -0
  357. union/remote/_client/auth/_authenticators/base.py +391 -0
  358. union/remote/_client/auth/_authenticators/client_credentials.py +73 -0
  359. union/remote/_client/auth/_authenticators/device_code.py +120 -0
  360. union/remote/_client/auth/_authenticators/external_command.py +77 -0
  361. union/remote/_client/auth/_authenticators/factory.py +200 -0
  362. union/remote/_client/auth/_authenticators/pkce.py +515 -0
  363. union/remote/_client/auth/_channel.py +184 -0
  364. union/remote/_client/auth/_client_config.py +83 -0
  365. union/remote/_client/auth/_default_html.py +32 -0
  366. union/remote/_client/auth/_grpc_utils/__init__.py +0 -0
  367. union/remote/_client/auth/_grpc_utils/auth_interceptor.py +204 -0
  368. union/remote/_client/auth/_grpc_utils/default_metadata_interceptor.py +144 -0
  369. union/remote/_client/auth/_keyring.py +154 -0
  370. union/remote/_client/auth/_token_client.py +258 -0
  371. union/remote/_client/auth/errors.py +16 -0
  372. union/remote/_client/controlplane.py +86 -0
  373. union/remote/_data.py +149 -0
  374. union/remote/_logs.py +74 -0
  375. union/remote/_project.py +86 -0
  376. union/remote/_run.py +820 -0
  377. union/remote/_secret.py +132 -0
  378. union/remote/_task.py +193 -0
  379. union/report/__init__.py +3 -0
  380. union/report/_report.py +178 -0
  381. union/report/_template.html +124 -0
  382. union/storage/__init__.py +24 -0
  383. union/storage/_remote_fs.py +34 -0
  384. union/storage/_storage.py +247 -0
  385. union/storage/_utils.py +5 -0
  386. union/types/__init__.py +11 -0
  387. union/types/_renderer.py +162 -0
  388. union/types/_string_literals.py +120 -0
  389. union/types/_type_engine.py +2131 -0
  390. union/types/_utils.py +80 -0
@@ -0,0 +1,172 @@
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)
@@ -0,0 +1,128 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # source: workflow/run_definition.proto
4
+ """Generated protocol buffer code."""
5
+ from google.protobuf import descriptor as _descriptor
6
+ from google.protobuf import descriptor_pool as _descriptor_pool
7
+ from google.protobuf import symbol_database as _symbol_database
8
+ from google.protobuf.internal import builder as _builder
9
+ # @@protoc_insertion_point(imports)
10
+
11
+ _sym_db = _symbol_database.Default()
12
+
13
+
14
+ from flyte._protos.common import identifier_pb2 as common_dot_identifier__pb2
15
+ from flyte._protos.common import identity_pb2 as common_dot_identity__pb2
16
+ from flyteidl.core import execution_pb2 as flyteidl_dot_core_dot_execution__pb2
17
+ from flyteidl.core import literals_pb2 as flyteidl_dot_core_dot_literals__pb2
18
+ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
19
+ from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
20
+ from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
21
+ from flyte._protos.workflow import task_definition_pb2 as workflow_dot_task__definition__pb2
22
+
23
+
24
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1dworkflow/run_definition.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x15\x63ommon/identity.proto\x1a\x1d\x66lyteidl/core/execution.proto\x1a\x1c\x66lyteidl/core/literals.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\x1a\x17validate/validate.proto\x1a\x1eworkflow/task_definition.proto\"\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)
@@ -0,0 +1,310 @@
1
+ from flyte._protos.common import identifier_pb2 as _identifier_pb2
2
+ from flyte._protos.common import identity_pb2 as _identity_pb2
3
+ from flyteidl.core import execution_pb2 as _execution_pb2
4
+ from flyteidl.core import literals_pb2 as _literals_pb2
5
+ from google.protobuf import timestamp_pb2 as _timestamp_pb2
6
+ from google.protobuf import wrappers_pb2 as _wrappers_pb2
7
+ from flyte._protos.validate.validate import validate_pb2 as _validate_pb2
8
+ from flyte._protos.workflow import task_definition_pb2 as _task_definition_pb2
9
+ from google.protobuf.internal import containers as _containers
10
+ from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
11
+ from google.protobuf import descriptor as _descriptor
12
+ from google.protobuf import message as _message
13
+ from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
14
+
15
+ DESCRIPTOR: _descriptor.FileDescriptor
16
+
17
+ class Phase(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
18
+ __slots__ = []
19
+ PHASE_UNSPECIFIED: _ClassVar[Phase]
20
+ PHASE_QUEUED: _ClassVar[Phase]
21
+ PHASE_WAITING_FOR_RESOURCES: _ClassVar[Phase]
22
+ PHASE_INITIALIZING: _ClassVar[Phase]
23
+ PHASE_RUNNING: _ClassVar[Phase]
24
+ PHASE_SUCCEEDED: _ClassVar[Phase]
25
+ PHASE_FAILED: _ClassVar[Phase]
26
+ PHASE_ABORTED: _ClassVar[Phase]
27
+ PHASE_TIMED_OUT: _ClassVar[Phase]
28
+ PHASE_UNSPECIFIED: Phase
29
+ PHASE_QUEUED: Phase
30
+ PHASE_WAITING_FOR_RESOURCES: Phase
31
+ PHASE_INITIALIZING: Phase
32
+ PHASE_RUNNING: Phase
33
+ PHASE_SUCCEEDED: Phase
34
+ PHASE_FAILED: Phase
35
+ PHASE_ABORTED: Phase
36
+ PHASE_TIMED_OUT: Phase
37
+
38
+ class RunIdentifier(_message.Message):
39
+ __slots__ = ["org", "project", "domain", "name"]
40
+ ORG_FIELD_NUMBER: _ClassVar[int]
41
+ PROJECT_FIELD_NUMBER: _ClassVar[int]
42
+ DOMAIN_FIELD_NUMBER: _ClassVar[int]
43
+ NAME_FIELD_NUMBER: _ClassVar[int]
44
+ org: str
45
+ project: str
46
+ domain: str
47
+ name: str
48
+ def __init__(self, org: _Optional[str] = ..., project: _Optional[str] = ..., domain: _Optional[str] = ..., name: _Optional[str] = ...) -> None: ...
49
+
50
+ class Labels(_message.Message):
51
+ __slots__ = ["values"]
52
+ class ValuesEntry(_message.Message):
53
+ __slots__ = ["key", "value"]
54
+ KEY_FIELD_NUMBER: _ClassVar[int]
55
+ VALUE_FIELD_NUMBER: _ClassVar[int]
56
+ key: str
57
+ value: str
58
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
59
+ VALUES_FIELD_NUMBER: _ClassVar[int]
60
+ values: _containers.ScalarMap[str, str]
61
+ def __init__(self, values: _Optional[_Mapping[str, str]] = ...) -> None: ...
62
+
63
+ class Annotations(_message.Message):
64
+ __slots__ = ["values"]
65
+ class ValuesEntry(_message.Message):
66
+ __slots__ = ["key", "value"]
67
+ KEY_FIELD_NUMBER: _ClassVar[int]
68
+ VALUE_FIELD_NUMBER: _ClassVar[int]
69
+ key: str
70
+ value: str
71
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
72
+ VALUES_FIELD_NUMBER: _ClassVar[int]
73
+ values: _containers.ScalarMap[str, str]
74
+ def __init__(self, values: _Optional[_Mapping[str, str]] = ...) -> None: ...
75
+
76
+ class Envs(_message.Message):
77
+ __slots__ = ["values"]
78
+ VALUES_FIELD_NUMBER: _ClassVar[int]
79
+ values: _containers.RepeatedCompositeFieldContainer[_literals_pb2.KeyValuePair]
80
+ def __init__(self, values: _Optional[_Iterable[_Union[_literals_pb2.KeyValuePair, _Mapping]]] = ...) -> None: ...
81
+
82
+ class RunSpec(_message.Message):
83
+ __slots__ = ["labels", "annotations", "envs", "interruptible", "overwrite_cache"]
84
+ LABELS_FIELD_NUMBER: _ClassVar[int]
85
+ ANNOTATIONS_FIELD_NUMBER: _ClassVar[int]
86
+ ENVS_FIELD_NUMBER: _ClassVar[int]
87
+ INTERRUPTIBLE_FIELD_NUMBER: _ClassVar[int]
88
+ OVERWRITE_CACHE_FIELD_NUMBER: _ClassVar[int]
89
+ labels: Labels
90
+ annotations: Annotations
91
+ envs: Envs
92
+ interruptible: _wrappers_pb2.BoolValue
93
+ overwrite_cache: bool
94
+ def __init__(self, labels: _Optional[_Union[Labels, _Mapping]] = ..., annotations: _Optional[_Union[Annotations, _Mapping]] = ..., envs: _Optional[_Union[Envs, _Mapping]] = ..., interruptible: _Optional[_Union[_wrappers_pb2.BoolValue, _Mapping]] = ..., overwrite_cache: bool = ...) -> None: ...
95
+
96
+ class Run(_message.Message):
97
+ __slots__ = ["action"]
98
+ ACTION_FIELD_NUMBER: _ClassVar[int]
99
+ action: Action
100
+ def __init__(self, action: _Optional[_Union[Action, _Mapping]] = ...) -> None: ...
101
+
102
+ class RunDetails(_message.Message):
103
+ __slots__ = ["run_spec", "action"]
104
+ RUN_SPEC_FIELD_NUMBER: _ClassVar[int]
105
+ ACTION_FIELD_NUMBER: _ClassVar[int]
106
+ run_spec: RunSpec
107
+ action: ActionDetails
108
+ def __init__(self, run_spec: _Optional[_Union[RunSpec, _Mapping]] = ..., action: _Optional[_Union[ActionDetails, _Mapping]] = ...) -> None: ...
109
+
110
+ class ActionIdentifier(_message.Message):
111
+ __slots__ = ["run", "name"]
112
+ RUN_FIELD_NUMBER: _ClassVar[int]
113
+ NAME_FIELD_NUMBER: _ClassVar[int]
114
+ run: RunIdentifier
115
+ name: str
116
+ def __init__(self, run: _Optional[_Union[RunIdentifier, _Mapping]] = ..., name: _Optional[str] = ...) -> None: ...
117
+
118
+ class TaskActionMetadata(_message.Message):
119
+ __slots__ = ["id", "task_type"]
120
+ ID_FIELD_NUMBER: _ClassVar[int]
121
+ TASK_TYPE_FIELD_NUMBER: _ClassVar[int]
122
+ id: _task_definition_pb2.TaskIdentifier
123
+ task_type: str
124
+ def __init__(self, id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ..., task_type: _Optional[str] = ...) -> None: ...
125
+
126
+ class TraceActionMetadata(_message.Message):
127
+ __slots__ = ["name"]
128
+ NAME_FIELD_NUMBER: _ClassVar[int]
129
+ name: str
130
+ def __init__(self, name: _Optional[str] = ...) -> None: ...
131
+
132
+ class ConditionActionMetadata(_message.Message):
133
+ __slots__ = ["name", "run_id", "action_id"]
134
+ NAME_FIELD_NUMBER: _ClassVar[int]
135
+ RUN_ID_FIELD_NUMBER: _ClassVar[int]
136
+ ACTION_ID_FIELD_NUMBER: _ClassVar[int]
137
+ GLOBAL_FIELD_NUMBER: _ClassVar[int]
138
+ name: str
139
+ run_id: str
140
+ action_id: str
141
+ def __init__(self, name: _Optional[str] = ..., run_id: _Optional[str] = ..., action_id: _Optional[str] = ..., **kwargs) -> None: ...
142
+
143
+ class ActionMetadata(_message.Message):
144
+ __slots__ = ["parent", "group", "executed_by", "task", "trace", "condition"]
145
+ PARENT_FIELD_NUMBER: _ClassVar[int]
146
+ GROUP_FIELD_NUMBER: _ClassVar[int]
147
+ EXECUTED_BY_FIELD_NUMBER: _ClassVar[int]
148
+ TASK_FIELD_NUMBER: _ClassVar[int]
149
+ TRACE_FIELD_NUMBER: _ClassVar[int]
150
+ CONDITION_FIELD_NUMBER: _ClassVar[int]
151
+ parent: str
152
+ group: str
153
+ executed_by: _identity_pb2.EnrichedIdentity
154
+ task: TaskActionMetadata
155
+ trace: TraceActionMetadata
156
+ condition: ConditionActionMetadata
157
+ def __init__(self, parent: _Optional[str] = ..., group: _Optional[str] = ..., executed_by: _Optional[_Union[_identity_pb2.EnrichedIdentity, _Mapping]] = ..., task: _Optional[_Union[TaskActionMetadata, _Mapping]] = ..., trace: _Optional[_Union[TraceActionMetadata, _Mapping]] = ..., condition: _Optional[_Union[ConditionActionMetadata, _Mapping]] = ...) -> None: ...
158
+
159
+ class ActionStatus(_message.Message):
160
+ __slots__ = ["phase", "start_time", "end_time", "attempts"]
161
+ PHASE_FIELD_NUMBER: _ClassVar[int]
162
+ START_TIME_FIELD_NUMBER: _ClassVar[int]
163
+ END_TIME_FIELD_NUMBER: _ClassVar[int]
164
+ ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
165
+ phase: Phase
166
+ start_time: _timestamp_pb2.Timestamp
167
+ end_time: _timestamp_pb2.Timestamp
168
+ attempts: int
169
+ def __init__(self, phase: _Optional[_Union[Phase, str]] = ..., start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., attempts: _Optional[int] = ...) -> None: ...
170
+
171
+ class Action(_message.Message):
172
+ __slots__ = ["id", "metadata", "status"]
173
+ ID_FIELD_NUMBER: _ClassVar[int]
174
+ METADATA_FIELD_NUMBER: _ClassVar[int]
175
+ STATUS_FIELD_NUMBER: _ClassVar[int]
176
+ id: ActionIdentifier
177
+ metadata: ActionMetadata
178
+ status: ActionStatus
179
+ def __init__(self, id: _Optional[_Union[ActionIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[ActionMetadata, _Mapping]] = ..., status: _Optional[_Union[ActionStatus, _Mapping]] = ...) -> None: ...
180
+
181
+ class EnrichedAction(_message.Message):
182
+ __slots__ = ["action", "meets_filter", "children_phase_counts"]
183
+ class ChildrenPhaseCountsEntry(_message.Message):
184
+ __slots__ = ["key", "value"]
185
+ KEY_FIELD_NUMBER: _ClassVar[int]
186
+ VALUE_FIELD_NUMBER: _ClassVar[int]
187
+ key: int
188
+ value: int
189
+ def __init__(self, key: _Optional[int] = ..., value: _Optional[int] = ...) -> None: ...
190
+ ACTION_FIELD_NUMBER: _ClassVar[int]
191
+ MEETS_FILTER_FIELD_NUMBER: _ClassVar[int]
192
+ CHILDREN_PHASE_COUNTS_FIELD_NUMBER: _ClassVar[int]
193
+ action: Action
194
+ meets_filter: bool
195
+ children_phase_counts: _containers.ScalarMap[int, int]
196
+ def __init__(self, action: _Optional[_Union[Action, _Mapping]] = ..., meets_filter: bool = ..., children_phase_counts: _Optional[_Mapping[int, int]] = ...) -> None: ...
197
+
198
+ class ErrorInfo(_message.Message):
199
+ __slots__ = ["message", "kind"]
200
+ class Kind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
201
+ __slots__ = []
202
+ KIND_UNSPECIFIED: _ClassVar[ErrorInfo.Kind]
203
+ KIND_USER: _ClassVar[ErrorInfo.Kind]
204
+ KIND_SYSTEM: _ClassVar[ErrorInfo.Kind]
205
+ KIND_UNSPECIFIED: ErrorInfo.Kind
206
+ KIND_USER: ErrorInfo.Kind
207
+ KIND_SYSTEM: ErrorInfo.Kind
208
+ MESSAGE_FIELD_NUMBER: _ClassVar[int]
209
+ KIND_FIELD_NUMBER: _ClassVar[int]
210
+ message: str
211
+ kind: ErrorInfo.Kind
212
+ def __init__(self, message: _Optional[str] = ..., kind: _Optional[_Union[ErrorInfo.Kind, str]] = ...) -> None: ...
213
+
214
+ class AbortInfo(_message.Message):
215
+ __slots__ = ["reason", "aborted_by"]
216
+ REASON_FIELD_NUMBER: _ClassVar[int]
217
+ ABORTED_BY_FIELD_NUMBER: _ClassVar[int]
218
+ reason: str
219
+ aborted_by: _identity_pb2.EnrichedIdentity
220
+ def __init__(self, reason: _Optional[str] = ..., aborted_by: _Optional[_Union[_identity_pb2.EnrichedIdentity, _Mapping]] = ...) -> None: ...
221
+
222
+ class ActionDetails(_message.Message):
223
+ __slots__ = ["id", "metadata", "status", "error_info", "abort_info", "resolved_task_spec", "attempts"]
224
+ ID_FIELD_NUMBER: _ClassVar[int]
225
+ METADATA_FIELD_NUMBER: _ClassVar[int]
226
+ STATUS_FIELD_NUMBER: _ClassVar[int]
227
+ ERROR_INFO_FIELD_NUMBER: _ClassVar[int]
228
+ ABORT_INFO_FIELD_NUMBER: _ClassVar[int]
229
+ RESOLVED_TASK_SPEC_FIELD_NUMBER: _ClassVar[int]
230
+ ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
231
+ id: ActionIdentifier
232
+ metadata: ActionMetadata
233
+ status: ActionStatus
234
+ error_info: ErrorInfo
235
+ abort_info: AbortInfo
236
+ resolved_task_spec: _task_definition_pb2.TaskSpec
237
+ attempts: _containers.RepeatedCompositeFieldContainer[ActionAttempt]
238
+ def __init__(self, id: _Optional[_Union[ActionIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[ActionMetadata, _Mapping]] = ..., status: _Optional[_Union[ActionStatus, _Mapping]] = ..., error_info: _Optional[_Union[ErrorInfo, _Mapping]] = ..., abort_info: _Optional[_Union[AbortInfo, _Mapping]] = ..., resolved_task_spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ..., attempts: _Optional[_Iterable[_Union[ActionAttempt, _Mapping]]] = ...) -> None: ...
239
+
240
+ class ActionAttempt(_message.Message):
241
+ __slots__ = ["phase", "start_time", "end_time", "error_info", "attempt", "log_info", "outputs"]
242
+ PHASE_FIELD_NUMBER: _ClassVar[int]
243
+ START_TIME_FIELD_NUMBER: _ClassVar[int]
244
+ END_TIME_FIELD_NUMBER: _ClassVar[int]
245
+ ERROR_INFO_FIELD_NUMBER: _ClassVar[int]
246
+ ATTEMPT_FIELD_NUMBER: _ClassVar[int]
247
+ LOG_INFO_FIELD_NUMBER: _ClassVar[int]
248
+ OUTPUTS_FIELD_NUMBER: _ClassVar[int]
249
+ phase: Phase
250
+ start_time: _timestamp_pb2.Timestamp
251
+ end_time: _timestamp_pb2.Timestamp
252
+ error_info: ErrorInfo
253
+ attempt: int
254
+ log_info: _containers.RepeatedCompositeFieldContainer[_execution_pb2.TaskLog]
255
+ outputs: OutputReferences
256
+ def __init__(self, phase: _Optional[_Union[Phase, str]] = ..., start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., error_info: _Optional[_Union[ErrorInfo, _Mapping]] = ..., attempt: _Optional[int] = ..., log_info: _Optional[_Iterable[_Union[_execution_pb2.TaskLog, _Mapping]]] = ..., outputs: _Optional[_Union[OutputReferences, _Mapping]] = ...) -> None: ...
257
+
258
+ class ActionEvent(_message.Message):
259
+ __slots__ = ["id", "attempt", "phase", "version", "start_time", "updated_time", "end_time", "error_info", "log_info", "log_context", "cluster", "outputs"]
260
+ ID_FIELD_NUMBER: _ClassVar[int]
261
+ ATTEMPT_FIELD_NUMBER: _ClassVar[int]
262
+ PHASE_FIELD_NUMBER: _ClassVar[int]
263
+ VERSION_FIELD_NUMBER: _ClassVar[int]
264
+ START_TIME_FIELD_NUMBER: _ClassVar[int]
265
+ UPDATED_TIME_FIELD_NUMBER: _ClassVar[int]
266
+ END_TIME_FIELD_NUMBER: _ClassVar[int]
267
+ ERROR_INFO_FIELD_NUMBER: _ClassVar[int]
268
+ LOG_INFO_FIELD_NUMBER: _ClassVar[int]
269
+ LOG_CONTEXT_FIELD_NUMBER: _ClassVar[int]
270
+ CLUSTER_FIELD_NUMBER: _ClassVar[int]
271
+ OUTPUTS_FIELD_NUMBER: _ClassVar[int]
272
+ id: ActionIdentifier
273
+ attempt: int
274
+ phase: Phase
275
+ version: int
276
+ start_time: _timestamp_pb2.Timestamp
277
+ updated_time: _timestamp_pb2.Timestamp
278
+ end_time: _timestamp_pb2.Timestamp
279
+ error_info: ErrorInfo
280
+ log_info: _containers.RepeatedCompositeFieldContainer[_execution_pb2.TaskLog]
281
+ log_context: _execution_pb2.LogContext
282
+ cluster: str
283
+ outputs: OutputReferences
284
+ def __init__(self, id: _Optional[_Union[ActionIdentifier, _Mapping]] = ..., attempt: _Optional[int] = ..., phase: _Optional[_Union[Phase, str]] = ..., version: _Optional[int] = ..., start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., updated_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., error_info: _Optional[_Union[ErrorInfo, _Mapping]] = ..., log_info: _Optional[_Iterable[_Union[_execution_pb2.TaskLog, _Mapping]]] = ..., log_context: _Optional[_Union[_execution_pb2.LogContext, _Mapping]] = ..., cluster: _Optional[str] = ..., outputs: _Optional[_Union[OutputReferences, _Mapping]] = ...) -> None: ...
285
+
286
+ class NamedLiteral(_message.Message):
287
+ __slots__ = ["name", "value"]
288
+ NAME_FIELD_NUMBER: _ClassVar[int]
289
+ VALUE_FIELD_NUMBER: _ClassVar[int]
290
+ name: str
291
+ value: _literals_pb2.Literal
292
+ def __init__(self, name: _Optional[str] = ..., value: _Optional[_Union[_literals_pb2.Literal, _Mapping]] = ...) -> None: ...
293
+
294
+ class OutputReferences(_message.Message):
295
+ __slots__ = ["output_uri"]
296
+ OUTPUT_URI_FIELD_NUMBER: _ClassVar[int]
297
+ output_uri: str
298
+ def __init__(self, output_uri: _Optional[str] = ...) -> None: ...
299
+
300
+ class Inputs(_message.Message):
301
+ __slots__ = ["literals"]
302
+ LITERALS_FIELD_NUMBER: _ClassVar[int]
303
+ literals: _containers.RepeatedCompositeFieldContainer[NamedLiteral]
304
+ def __init__(self, literals: _Optional[_Iterable[_Union[NamedLiteral, _Mapping]]] = ...) -> None: ...
305
+
306
+ class Outputs(_message.Message):
307
+ __slots__ = ["literals"]
308
+ LITERALS_FIELD_NUMBER: _ClassVar[int]
309
+ literals: _containers.RepeatedCompositeFieldContainer[NamedLiteral]
310
+ def __init__(self, literals: _Optional[_Iterable[_Union[NamedLiteral, _Mapping]]] = ...) -> None: ...
@@ -0,0 +1,4 @@
1
+ # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
+ """Client and server classes corresponding to protobuf-defined services."""
3
+ import grpc
4
+