flyte 0.2.0b1__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 (266) 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 -231
  26. flyte/_initialize.py +456 -316
  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 +319 -36
  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 +462 -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. flyte/{_cli → cli}/_params.py +57 -17
  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 +2 -167
  121. flyte/config/_config.py +215 -163
  122. flyte/config/_internal.py +10 -1
  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 +226 -126
  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 -299
  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 -174
  195. flyte/_cli/main.py +0 -98
  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.2.0b1.dist-info/METADATA +0 -179
  261. flyte-0.2.0b1.dist-info/RECORD +0 -204
  262. flyte-0.2.0b1.dist-info/entry_points.txt +0 -3
  263. /flyte/{_cli → _debug}/__init__.py +0 -0
  264. /flyte/{_protos → _keyring}/__init__.py +0 -0
  265. {flyte-0.2.0b1.dist-info → flyte-2.0.0b46.dist-info}/WHEEL +0 -0
  266. {flyte-0.2.0b1.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)