flyte 2.0.0b13__py3-none-any.whl → 2.0.0b30__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 (211) hide show
  1. flyte/__init__.py +18 -2
  2. flyte/_bin/debug.py +38 -0
  3. flyte/_bin/runtime.py +62 -8
  4. flyte/_cache/cache.py +4 -2
  5. flyte/_cache/local_cache.py +216 -0
  6. flyte/_code_bundle/_ignore.py +12 -4
  7. flyte/_code_bundle/_packaging.py +13 -9
  8. flyte/_code_bundle/_utils.py +18 -10
  9. flyte/_code_bundle/bundle.py +17 -9
  10. flyte/_constants.py +1 -0
  11. flyte/_context.py +4 -1
  12. flyte/_custom_context.py +73 -0
  13. flyte/_debug/constants.py +38 -0
  14. flyte/_debug/utils.py +17 -0
  15. flyte/_debug/vscode.py +307 -0
  16. flyte/_deploy.py +235 -61
  17. flyte/_environment.py +20 -6
  18. flyte/_excepthook.py +1 -1
  19. flyte/_hash.py +1 -16
  20. flyte/_image.py +178 -81
  21. flyte/_initialize.py +132 -51
  22. flyte/_interface.py +39 -2
  23. flyte/_internal/controllers/__init__.py +4 -5
  24. flyte/_internal/controllers/_local_controller.py +70 -29
  25. flyte/_internal/controllers/_trace.py +1 -1
  26. flyte/_internal/controllers/remote/__init__.py +0 -2
  27. flyte/_internal/controllers/remote/_action.py +14 -16
  28. flyte/_internal/controllers/remote/_client.py +1 -1
  29. flyte/_internal/controllers/remote/_controller.py +68 -70
  30. flyte/_internal/controllers/remote/_core.py +127 -99
  31. flyte/_internal/controllers/remote/_informer.py +19 -10
  32. flyte/_internal/controllers/remote/_service_protocol.py +7 -7
  33. flyte/_internal/imagebuild/docker_builder.py +181 -69
  34. flyte/_internal/imagebuild/image_builder.py +0 -5
  35. flyte/_internal/imagebuild/remote_builder.py +155 -64
  36. flyte/_internal/imagebuild/utils.py +51 -2
  37. flyte/_internal/resolvers/_task_module.py +5 -38
  38. flyte/_internal/resolvers/default.py +2 -2
  39. flyte/_internal/runtime/convert.py +110 -21
  40. flyte/_internal/runtime/entrypoints.py +27 -1
  41. flyte/_internal/runtime/io.py +21 -8
  42. flyte/_internal/runtime/resources_serde.py +20 -6
  43. flyte/_internal/runtime/reuse.py +1 -1
  44. flyte/_internal/runtime/rusty.py +20 -5
  45. flyte/_internal/runtime/task_serde.py +34 -19
  46. flyte/_internal/runtime/taskrunner.py +22 -4
  47. flyte/_internal/runtime/trigger_serde.py +160 -0
  48. flyte/_internal/runtime/types_serde.py +1 -1
  49. flyte/_keyring/__init__.py +0 -0
  50. flyte/_keyring/file.py +115 -0
  51. flyte/_logging.py +201 -39
  52. flyte/_map.py +111 -14
  53. flyte/_module.py +70 -0
  54. flyte/_pod.py +4 -3
  55. flyte/_resources.py +213 -31
  56. flyte/_run.py +110 -39
  57. flyte/_task.py +75 -16
  58. flyte/_task_environment.py +105 -29
  59. flyte/_task_plugins.py +4 -2
  60. flyte/_trace.py +5 -0
  61. flyte/_trigger.py +1000 -0
  62. flyte/_utils/__init__.py +2 -1
  63. flyte/_utils/asyn.py +3 -1
  64. flyte/_utils/coro_management.py +2 -1
  65. flyte/_utils/docker_credentials.py +173 -0
  66. flyte/_utils/module_loader.py +17 -2
  67. flyte/_version.py +3 -3
  68. flyte/cli/_abort.py +3 -3
  69. flyte/cli/_build.py +3 -6
  70. flyte/cli/_common.py +78 -7
  71. flyte/cli/_create.py +182 -4
  72. flyte/cli/_delete.py +23 -1
  73. flyte/cli/_deploy.py +63 -16
  74. flyte/cli/_get.py +79 -34
  75. flyte/cli/_params.py +26 -10
  76. flyte/cli/_plugins.py +209 -0
  77. flyte/cli/_run.py +151 -26
  78. flyte/cli/_serve.py +64 -0
  79. flyte/cli/_update.py +37 -0
  80. flyte/cli/_user.py +17 -0
  81. flyte/cli/main.py +30 -4
  82. flyte/config/_config.py +10 -6
  83. flyte/config/_internal.py +1 -0
  84. flyte/config/_reader.py +29 -8
  85. flyte/connectors/__init__.py +11 -0
  86. flyte/connectors/_connector.py +270 -0
  87. flyte/connectors/_server.py +197 -0
  88. flyte/connectors/utils.py +135 -0
  89. flyte/errors.py +22 -2
  90. flyte/extend.py +8 -1
  91. flyte/extras/_container.py +6 -1
  92. flyte/git/__init__.py +3 -0
  93. flyte/git/_config.py +21 -0
  94. flyte/io/__init__.py +2 -0
  95. flyte/io/_dataframe/__init__.py +2 -0
  96. flyte/io/_dataframe/basic_dfs.py +17 -8
  97. flyte/io/_dataframe/dataframe.py +98 -132
  98. flyte/io/_dir.py +575 -113
  99. flyte/io/_file.py +582 -139
  100. flyte/io/_hashing_io.py +342 -0
  101. flyte/models.py +74 -15
  102. flyte/remote/__init__.py +6 -1
  103. flyte/remote/_action.py +34 -26
  104. flyte/remote/_client/_protocols.py +39 -4
  105. flyte/remote/_client/auth/_authenticators/device_code.py +4 -5
  106. flyte/remote/_client/auth/_authenticators/pkce.py +1 -1
  107. flyte/remote/_client/auth/_channel.py +10 -6
  108. flyte/remote/_client/controlplane.py +17 -5
  109. flyte/remote/_console.py +3 -2
  110. flyte/remote/_data.py +6 -6
  111. flyte/remote/_logs.py +3 -3
  112. flyte/remote/_run.py +64 -8
  113. flyte/remote/_secret.py +26 -17
  114. flyte/remote/_task.py +75 -33
  115. flyte/remote/_trigger.py +306 -0
  116. flyte/remote/_user.py +33 -0
  117. flyte/report/_report.py +1 -1
  118. flyte/storage/__init__.py +6 -1
  119. flyte/storage/_config.py +5 -1
  120. flyte/storage/_parallel_reader.py +274 -0
  121. flyte/storage/_storage.py +200 -103
  122. flyte/types/__init__.py +16 -0
  123. flyte/types/_interface.py +2 -2
  124. flyte/types/_pickle.py +35 -8
  125. flyte/types/_string_literals.py +8 -9
  126. flyte/types/_type_engine.py +40 -70
  127. flyte/types/_utils.py +1 -1
  128. flyte-2.0.0b30.data/scripts/debug.py +38 -0
  129. {flyte-2.0.0b13.data → flyte-2.0.0b30.data}/scripts/runtime.py +62 -8
  130. {flyte-2.0.0b13.dist-info → flyte-2.0.0b30.dist-info}/METADATA +11 -3
  131. flyte-2.0.0b30.dist-info/RECORD +192 -0
  132. {flyte-2.0.0b13.dist-info → flyte-2.0.0b30.dist-info}/entry_points.txt +3 -0
  133. flyte/_protos/common/authorization_pb2.py +0 -66
  134. flyte/_protos/common/authorization_pb2.pyi +0 -108
  135. flyte/_protos/common/authorization_pb2_grpc.py +0 -4
  136. flyte/_protos/common/identifier_pb2.py +0 -93
  137. flyte/_protos/common/identifier_pb2.pyi +0 -110
  138. flyte/_protos/common/identifier_pb2_grpc.py +0 -4
  139. flyte/_protos/common/identity_pb2.py +0 -48
  140. flyte/_protos/common/identity_pb2.pyi +0 -72
  141. flyte/_protos/common/identity_pb2_grpc.py +0 -4
  142. flyte/_protos/common/list_pb2.py +0 -36
  143. flyte/_protos/common/list_pb2.pyi +0 -71
  144. flyte/_protos/common/list_pb2_grpc.py +0 -4
  145. flyte/_protos/common/policy_pb2.py +0 -37
  146. flyte/_protos/common/policy_pb2.pyi +0 -27
  147. flyte/_protos/common/policy_pb2_grpc.py +0 -4
  148. flyte/_protos/common/role_pb2.py +0 -37
  149. flyte/_protos/common/role_pb2.pyi +0 -53
  150. flyte/_protos/common/role_pb2_grpc.py +0 -4
  151. flyte/_protos/common/runtime_version_pb2.py +0 -28
  152. flyte/_protos/common/runtime_version_pb2.pyi +0 -24
  153. flyte/_protos/common/runtime_version_pb2_grpc.py +0 -4
  154. flyte/_protos/imagebuilder/definition_pb2.py +0 -59
  155. flyte/_protos/imagebuilder/definition_pb2.pyi +0 -140
  156. flyte/_protos/imagebuilder/definition_pb2_grpc.py +0 -4
  157. flyte/_protos/imagebuilder/payload_pb2.py +0 -32
  158. flyte/_protos/imagebuilder/payload_pb2.pyi +0 -21
  159. flyte/_protos/imagebuilder/payload_pb2_grpc.py +0 -4
  160. flyte/_protos/imagebuilder/service_pb2.py +0 -29
  161. flyte/_protos/imagebuilder/service_pb2.pyi +0 -5
  162. flyte/_protos/imagebuilder/service_pb2_grpc.py +0 -66
  163. flyte/_protos/logs/dataplane/payload_pb2.py +0 -100
  164. flyte/_protos/logs/dataplane/payload_pb2.pyi +0 -177
  165. flyte/_protos/logs/dataplane/payload_pb2_grpc.py +0 -4
  166. flyte/_protos/secret/definition_pb2.py +0 -49
  167. flyte/_protos/secret/definition_pb2.pyi +0 -93
  168. flyte/_protos/secret/definition_pb2_grpc.py +0 -4
  169. flyte/_protos/secret/payload_pb2.py +0 -62
  170. flyte/_protos/secret/payload_pb2.pyi +0 -94
  171. flyte/_protos/secret/payload_pb2_grpc.py +0 -4
  172. flyte/_protos/secret/secret_pb2.py +0 -38
  173. flyte/_protos/secret/secret_pb2.pyi +0 -6
  174. flyte/_protos/secret/secret_pb2_grpc.py +0 -198
  175. flyte/_protos/secret/secret_pb2_grpc_grpc.py +0 -198
  176. flyte/_protos/validate/validate/validate_pb2.py +0 -76
  177. flyte/_protos/workflow/common_pb2.py +0 -27
  178. flyte/_protos/workflow/common_pb2.pyi +0 -14
  179. flyte/_protos/workflow/common_pb2_grpc.py +0 -4
  180. flyte/_protos/workflow/environment_pb2.py +0 -29
  181. flyte/_protos/workflow/environment_pb2.pyi +0 -12
  182. flyte/_protos/workflow/environment_pb2_grpc.py +0 -4
  183. flyte/_protos/workflow/node_execution_service_pb2.py +0 -26
  184. flyte/_protos/workflow/node_execution_service_pb2.pyi +0 -4
  185. flyte/_protos/workflow/node_execution_service_pb2_grpc.py +0 -32
  186. flyte/_protos/workflow/queue_service_pb2.py +0 -109
  187. flyte/_protos/workflow/queue_service_pb2.pyi +0 -166
  188. flyte/_protos/workflow/queue_service_pb2_grpc.py +0 -172
  189. flyte/_protos/workflow/run_definition_pb2.py +0 -121
  190. flyte/_protos/workflow/run_definition_pb2.pyi +0 -327
  191. flyte/_protos/workflow/run_definition_pb2_grpc.py +0 -4
  192. flyte/_protos/workflow/run_logs_service_pb2.py +0 -41
  193. flyte/_protos/workflow/run_logs_service_pb2.pyi +0 -28
  194. flyte/_protos/workflow/run_logs_service_pb2_grpc.py +0 -69
  195. flyte/_protos/workflow/run_service_pb2.py +0 -137
  196. flyte/_protos/workflow/run_service_pb2.pyi +0 -185
  197. flyte/_protos/workflow/run_service_pb2_grpc.py +0 -446
  198. flyte/_protos/workflow/state_service_pb2.py +0 -67
  199. flyte/_protos/workflow/state_service_pb2.pyi +0 -76
  200. flyte/_protos/workflow/state_service_pb2_grpc.py +0 -138
  201. flyte/_protos/workflow/task_definition_pb2.py +0 -79
  202. flyte/_protos/workflow/task_definition_pb2.pyi +0 -81
  203. flyte/_protos/workflow/task_definition_pb2_grpc.py +0 -4
  204. flyte/_protos/workflow/task_service_pb2.py +0 -60
  205. flyte/_protos/workflow/task_service_pb2.pyi +0 -59
  206. flyte/_protos/workflow/task_service_pb2_grpc.py +0 -138
  207. flyte-2.0.0b13.dist-info/RECORD +0 -239
  208. /flyte/{_protos → _debug}/__init__.py +0 -0
  209. {flyte-2.0.0b13.dist-info → flyte-2.0.0b30.dist-info}/WHEEL +0 -0
  210. {flyte-2.0.0b13.dist-info → flyte-2.0.0b30.dist-info}/licenses/LICENSE +0 -0
  211. {flyte-2.0.0b13.dist-info → flyte-2.0.0b30.dist-info}/top_level.txt +0 -0
@@ -1,67 +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 flyte._protos.common import identifier_pb2 as common_dot_identifier__pb2
15
- from flyteidl.core import execution_pb2 as flyteidl_dot_core_dot_execution__pb2
16
- from google.rpc import status_pb2 as google_dot_rpc_dot_status__pb2
17
- from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
18
- from flyte._protos.workflow import run_definition_pb2 as workflow_dot_run__definition__pb2
19
-
20
-
21
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cworkflow/state_service.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x1d\x66lyteidl/core/execution.proto\x1a\x17google/rpc/status.proto\x1a\x17validate/validate.proto\x1a\x1dworkflow/run_definition.proto\"\xbf\x01\n\nPutRequest\x12H\n\taction_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.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\"\x8d\x01\n\x0bPutResponse\x12H\n\taction_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\x12\x34\n\x06status\x18\x02 \x01(\x0b\x32\x12.google.rpc.StatusB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06status\"V\n\nGetRequest\x12H\n\taction_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\"\xac\x01\n\x0bGetResponse\x12H\n\taction_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\x12\x34\n\x06status\x18\x02 \x01(\x0b\x32\x12.google.rpc.StatusB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06status\x12\x1d\n\x05state\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x05state\"l\n\x0cWatchRequest\x12M\n\x10parent_action_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.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\"\xeb\x01\n\x0c\x41\x63tionUpdate\x12H\n\taction_id\x18\x01 \x01(\x0b\x32!.cloudidl.common.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\xf6\x01\n\x0cStateService\x12J\n\x03Put\x12\x1d.cloudidl.workflow.PutRequest\x1a\x1e.cloudidl.workflow.PutResponse\"\x00(\x01\x30\x01\x12J\n\x03Get\x12\x1d.cloudidl.workflow.GetRequest\x1a\x1e.cloudidl.workflow.GetResponse\"\x00(\x01\x30\x01\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')
22
-
23
- _globals = globals()
24
- _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
25
- _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'workflow.state_service_pb2', _globals)
26
- if _descriptor._USE_C_DESCRIPTORS == False:
27
- DESCRIPTOR._options = None
28
- 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'
29
- _PUTREQUEST.fields_by_name['action_id']._options = None
30
- _PUTREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
31
- _PUTREQUEST.fields_by_name['state']._options = None
32
- _PUTREQUEST.fields_by_name['state']._serialized_options = b'\372B\004r\002\020\001'
33
- _PUTRESPONSE.fields_by_name['action_id']._options = None
34
- _PUTRESPONSE.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
35
- _PUTRESPONSE.fields_by_name['status']._options = None
36
- _PUTRESPONSE.fields_by_name['status']._serialized_options = b'\372B\005\212\001\002\020\001'
37
- _GETREQUEST.fields_by_name['action_id']._options = None
38
- _GETREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
39
- _GETRESPONSE.fields_by_name['action_id']._options = None
40
- _GETRESPONSE.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
41
- _GETRESPONSE.fields_by_name['status']._options = None
42
- _GETRESPONSE.fields_by_name['status']._serialized_options = b'\372B\005\212\001\002\020\001'
43
- _GETRESPONSE.fields_by_name['state']._options = None
44
- _GETRESPONSE.fields_by_name['state']._serialized_options = b'\372B\004r\002\020\001'
45
- _WATCHREQUEST.oneofs_by_name['filter']._options = None
46
- _WATCHREQUEST.oneofs_by_name['filter']._serialized_options = b'\370B\001'
47
- _ACTIONUPDATE.fields_by_name['action_id']._options = None
48
- _ACTIONUPDATE.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
49
- _globals['_PUTREQUEST']._serialized_start=189
50
- _globals['_PUTREQUEST']._serialized_end=380
51
- _globals['_PUTRESPONSE']._serialized_start=383
52
- _globals['_PUTRESPONSE']._serialized_end=524
53
- _globals['_GETREQUEST']._serialized_start=526
54
- _globals['_GETREQUEST']._serialized_end=612
55
- _globals['_GETRESPONSE']._serialized_start=615
56
- _globals['_GETRESPONSE']._serialized_end=787
57
- _globals['_WATCHREQUEST']._serialized_start=789
58
- _globals['_WATCHREQUEST']._serialized_end=897
59
- _globals['_WATCHRESPONSE']._serialized_start=900
60
- _globals['_WATCHRESPONSE']._serialized_end=1076
61
- _globals['_CONTROLMESSAGE']._serialized_start=1078
62
- _globals['_CONTROLMESSAGE']._serialized_end=1122
63
- _globals['_ACTIONUPDATE']._serialized_start=1125
64
- _globals['_ACTIONUPDATE']._serialized_end=1360
65
- _globals['_STATESERVICE']._serialized_start=1363
66
- _globals['_STATESERVICE']._serialized_end=1609
67
- # @@protoc_insertion_point(module_scope)
@@ -1,76 +0,0 @@
1
- from flyte._protos.common import identifier_pb2 as _identifier_pb2
2
- from flyteidl.core import execution_pb2 as _execution_pb2
3
- from google.rpc import status_pb2 as _status_pb2
4
- from flyte._protos.validate.validate import validate_pb2 as _validate_pb2
5
- from flyte._protos.workflow import run_definition_pb2 as _run_definition_pb2
6
- from google.protobuf import descriptor as _descriptor
7
- from google.protobuf import message as _message
8
- from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
9
-
10
- DESCRIPTOR: _descriptor.FileDescriptor
11
-
12
- class PutRequest(_message.Message):
13
- __slots__ = ["action_id", "parent_action_name", "state"]
14
- ACTION_ID_FIELD_NUMBER: _ClassVar[int]
15
- PARENT_ACTION_NAME_FIELD_NUMBER: _ClassVar[int]
16
- STATE_FIELD_NUMBER: _ClassVar[int]
17
- action_id: _identifier_pb2.ActionIdentifier
18
- parent_action_name: str
19
- state: str
20
- def __init__(self, action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ..., parent_action_name: _Optional[str] = ..., state: _Optional[str] = ...) -> None: ...
21
-
22
- class PutResponse(_message.Message):
23
- __slots__ = ["action_id", "status"]
24
- ACTION_ID_FIELD_NUMBER: _ClassVar[int]
25
- STATUS_FIELD_NUMBER: _ClassVar[int]
26
- action_id: _identifier_pb2.ActionIdentifier
27
- status: _status_pb2.Status
28
- def __init__(self, action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ..., status: _Optional[_Union[_status_pb2.Status, _Mapping]] = ...) -> None: ...
29
-
30
- class GetRequest(_message.Message):
31
- __slots__ = ["action_id"]
32
- ACTION_ID_FIELD_NUMBER: _ClassVar[int]
33
- action_id: _identifier_pb2.ActionIdentifier
34
- def __init__(self, action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
35
-
36
- class GetResponse(_message.Message):
37
- __slots__ = ["action_id", "status", "state"]
38
- ACTION_ID_FIELD_NUMBER: _ClassVar[int]
39
- STATUS_FIELD_NUMBER: _ClassVar[int]
40
- STATE_FIELD_NUMBER: _ClassVar[int]
41
- action_id: _identifier_pb2.ActionIdentifier
42
- status: _status_pb2.Status
43
- state: str
44
- def __init__(self, action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ..., status: _Optional[_Union[_status_pb2.Status, _Mapping]] = ..., state: _Optional[str] = ...) -> None: ...
45
-
46
- class WatchRequest(_message.Message):
47
- __slots__ = ["parent_action_id"]
48
- PARENT_ACTION_ID_FIELD_NUMBER: _ClassVar[int]
49
- parent_action_id: _identifier_pb2.ActionIdentifier
50
- def __init__(self, parent_action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
51
-
52
- class WatchResponse(_message.Message):
53
- __slots__ = ["action_update", "control_message"]
54
- ACTION_UPDATE_FIELD_NUMBER: _ClassVar[int]
55
- CONTROL_MESSAGE_FIELD_NUMBER: _ClassVar[int]
56
- action_update: ActionUpdate
57
- control_message: ControlMessage
58
- def __init__(self, action_update: _Optional[_Union[ActionUpdate, _Mapping]] = ..., control_message: _Optional[_Union[ControlMessage, _Mapping]] = ...) -> None: ...
59
-
60
- class ControlMessage(_message.Message):
61
- __slots__ = ["sentinel"]
62
- SENTINEL_FIELD_NUMBER: _ClassVar[int]
63
- sentinel: bool
64
- def __init__(self, sentinel: bool = ...) -> None: ...
65
-
66
- class ActionUpdate(_message.Message):
67
- __slots__ = ["action_id", "phase", "error", "output_uri"]
68
- ACTION_ID_FIELD_NUMBER: _ClassVar[int]
69
- PHASE_FIELD_NUMBER: _ClassVar[int]
70
- ERROR_FIELD_NUMBER: _ClassVar[int]
71
- OUTPUT_URI_FIELD_NUMBER: _ClassVar[int]
72
- action_id: _identifier_pb2.ActionIdentifier
73
- phase: _run_definition_pb2.Phase
74
- error: _execution_pb2.ExecutionError
75
- output_uri: str
76
- def __init__(self, action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ..., phase: _Optional[_Union[_run_definition_pb2.Phase, str]] = ..., error: _Optional[_Union[_execution_pb2.ExecutionError, _Mapping]] = ..., output_uri: _Optional[str] = ...) -> None: ...
@@ -1,138 +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 state_service_pb2 as workflow_dot_state__service__pb2
6
-
7
-
8
- class StateServiceStub(object):
9
- """provides an interface for managing the state of actions.
10
- """
11
-
12
- def __init__(self, channel):
13
- """Constructor.
14
-
15
- Args:
16
- channel: A grpc.Channel.
17
- """
18
- self.Put = channel.stream_stream(
19
- '/cloudidl.workflow.StateService/Put',
20
- request_serializer=workflow_dot_state__service__pb2.PutRequest.SerializeToString,
21
- response_deserializer=workflow_dot_state__service__pb2.PutResponse.FromString,
22
- )
23
- self.Get = channel.stream_stream(
24
- '/cloudidl.workflow.StateService/Get',
25
- request_serializer=workflow_dot_state__service__pb2.GetRequest.SerializeToString,
26
- response_deserializer=workflow_dot_state__service__pb2.GetResponse.FromString,
27
- )
28
- self.Watch = channel.unary_stream(
29
- '/cloudidl.workflow.StateService/Watch',
30
- request_serializer=workflow_dot_state__service__pb2.WatchRequest.SerializeToString,
31
- response_deserializer=workflow_dot_state__service__pb2.WatchResponse.FromString,
32
- )
33
-
34
-
35
- class StateServiceServicer(object):
36
- """provides an interface for managing the state of actions.
37
- """
38
-
39
- def Put(self, request_iterator, context):
40
- """put the state of an action.
41
- """
42
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
43
- context.set_details('Method not implemented!')
44
- raise NotImplementedError('Method not implemented!')
45
-
46
- def Get(self, request_iterator, context):
47
- """get the state of an action.
48
- """
49
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
50
- context.set_details('Method not implemented!')
51
- raise NotImplementedError('Method not implemented!')
52
-
53
- def Watch(self, request, context):
54
- """watch for updates to the state of actions. this api guarantees at-least-once delivery semantics.
55
- """
56
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
57
- context.set_details('Method not implemented!')
58
- raise NotImplementedError('Method not implemented!')
59
-
60
-
61
- def add_StateServiceServicer_to_server(servicer, server):
62
- rpc_method_handlers = {
63
- 'Put': grpc.stream_stream_rpc_method_handler(
64
- servicer.Put,
65
- request_deserializer=workflow_dot_state__service__pb2.PutRequest.FromString,
66
- response_serializer=workflow_dot_state__service__pb2.PutResponse.SerializeToString,
67
- ),
68
- 'Get': grpc.stream_stream_rpc_method_handler(
69
- servicer.Get,
70
- request_deserializer=workflow_dot_state__service__pb2.GetRequest.FromString,
71
- response_serializer=workflow_dot_state__service__pb2.GetResponse.SerializeToString,
72
- ),
73
- 'Watch': grpc.unary_stream_rpc_method_handler(
74
- servicer.Watch,
75
- request_deserializer=workflow_dot_state__service__pb2.WatchRequest.FromString,
76
- response_serializer=workflow_dot_state__service__pb2.WatchResponse.SerializeToString,
77
- ),
78
- }
79
- generic_handler = grpc.method_handlers_generic_handler(
80
- 'cloudidl.workflow.StateService', rpc_method_handlers)
81
- server.add_generic_rpc_handlers((generic_handler,))
82
-
83
-
84
- # This class is part of an EXPERIMENTAL API.
85
- class StateService(object):
86
- """provides an interface for managing the state of actions.
87
- """
88
-
89
- @staticmethod
90
- def Put(request_iterator,
91
- target,
92
- options=(),
93
- channel_credentials=None,
94
- call_credentials=None,
95
- insecure=False,
96
- compression=None,
97
- wait_for_ready=None,
98
- timeout=None,
99
- metadata=None):
100
- return grpc.experimental.stream_stream(request_iterator, target, '/cloudidl.workflow.StateService/Put',
101
- workflow_dot_state__service__pb2.PutRequest.SerializeToString,
102
- workflow_dot_state__service__pb2.PutResponse.FromString,
103
- options, channel_credentials,
104
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
105
-
106
- @staticmethod
107
- def Get(request_iterator,
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.stream_stream(request_iterator, target, '/cloudidl.workflow.StateService/Get',
118
- workflow_dot_state__service__pb2.GetRequest.SerializeToString,
119
- workflow_dot_state__service__pb2.GetResponse.FromString,
120
- options, channel_credentials,
121
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
122
-
123
- @staticmethod
124
- def Watch(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_stream(request, target, '/cloudidl.workflow.StateService/Watch',
135
- workflow_dot_state__service__pb2.WatchRequest.SerializeToString,
136
- workflow_dot_state__service__pb2.WatchResponse.FromString,
137
- options, channel_credentials,
138
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
@@ -1,79 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- # Generated by the protocol buffer compiler. DO NOT EDIT!
3
- # source: workflow/task_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 tasks_pb2 as flyteidl_dot_core_dot_tasks__pb2
17
- from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
18
- from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
19
- from flyte._protos.workflow import common_pb2 as workflow_dot_common__pb2
20
- from flyte._protos.workflow import environment_pb2 as workflow_dot_environment__pb2
21
-
22
-
23
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1eworkflow/task_definition.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x15\x63ommon/identity.proto\x1a\x19\x66lyteidl/core/tasks.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17validate/validate.proto\x1a\x15workflow/common.proto\x1a\x1aworkflow/environment.proto\"\x8f\x01\n\x08TaskName\x12\x1b\n\x03org\x18\x01 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x03org\x12#\n\x07project\x18\x02 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x07project\x12!\n\x06\x64omain\x18\x03 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x06\x64omain\x12\x1e\n\x04name\x18\x04 \x01(\tB\n\xfa\x42\x07r\x05\x10\x01\x18\xff\x01R\x04name\"\xba\x01\n\x0eTaskIdentifier\x12\x1b\n\x03org\x18\x01 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x03org\x12#\n\x07project\x18\x02 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x07project\x12!\n\x06\x64omain\x18\x03 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x06\x64omain\x12\x1e\n\x04name\x18\x04 \x01(\tB\n\xfa\x42\x07r\x05\x10\x01\x18\xff\x01R\x04name\x12#\n\x07version\x18\x05 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x07version\"\xed\x01\n\x0cTaskMetadata\x12L\n\x0b\x64\x65ployed_by\x18\x01 \x01(\x0b\x32!.cloudidl.common.EnrichedIdentityB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\ndeployedBy\x12\x1d\n\nshort_name\x18\x02 \x01(\tR\tshortName\x12\x45\n\x0b\x64\x65ployed_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x08\xfa\x42\x05\xb2\x01\x02\x08\x01R\ndeployedAt\x12)\n\x10\x65nvironment_name\x18\x04 \x01(\tR\x0f\x65nvironmentName\"\x93\x01\n\x04Task\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\x12\x45\n\x08metadata\x18\x02 \x01(\x0b\x32\x1f.cloudidl.workflow.TaskMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08metadata\"\x8a\x02\n\x08TaskSpec\x12J\n\rtask_template\x18\x01 \x01(\x0b\x32\x1b.flyteidl.core.TaskTemplateB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x0ctaskTemplate\x12H\n\x0e\x64\x65\x66\x61ult_inputs\x18\x02 \x03(\x0b\x32!.cloudidl.workflow.NamedParameterR\rdefaultInputs\x12&\n\nshort_name\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x18?R\tshortName\x12@\n\x0b\x65nvironment\x18\x04 \x01(\x0b\x32\x1e.cloudidl.workflow.EnvironmentR\x0b\x65nvironment\"\xd5\x01\n\x0bTaskDetails\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\x12\x45\n\x08metadata\x18\x02 \x01(\x0b\x32\x1f.cloudidl.workflow.TaskMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08metadata\x12\x39\n\x04spec\x18\x03 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x04specB\xc0\x01\n\x15\x63om.cloudidl.workflowB\x13TaskDefinitionProtoH\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')
24
-
25
- _globals = globals()
26
- _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
27
- _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'workflow.task_definition_pb2', _globals)
28
- if _descriptor._USE_C_DESCRIPTORS == False:
29
- DESCRIPTOR._options = None
30
- DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\023TaskDefinitionProtoH\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'
31
- _TASKNAME.fields_by_name['org']._options = None
32
- _TASKNAME.fields_by_name['org']._serialized_options = b'\372B\006r\004\020\001\030?'
33
- _TASKNAME.fields_by_name['project']._options = None
34
- _TASKNAME.fields_by_name['project']._serialized_options = b'\372B\006r\004\020\001\030?'
35
- _TASKNAME.fields_by_name['domain']._options = None
36
- _TASKNAME.fields_by_name['domain']._serialized_options = b'\372B\006r\004\020\001\030?'
37
- _TASKNAME.fields_by_name['name']._options = None
38
- _TASKNAME.fields_by_name['name']._serialized_options = b'\372B\007r\005\020\001\030\377\001'
39
- _TASKIDENTIFIER.fields_by_name['org']._options = None
40
- _TASKIDENTIFIER.fields_by_name['org']._serialized_options = b'\372B\006r\004\020\001\030?'
41
- _TASKIDENTIFIER.fields_by_name['project']._options = None
42
- _TASKIDENTIFIER.fields_by_name['project']._serialized_options = b'\372B\006r\004\020\001\030?'
43
- _TASKIDENTIFIER.fields_by_name['domain']._options = None
44
- _TASKIDENTIFIER.fields_by_name['domain']._serialized_options = b'\372B\006r\004\020\001\030?'
45
- _TASKIDENTIFIER.fields_by_name['name']._options = None
46
- _TASKIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\007r\005\020\001\030\377\001'
47
- _TASKIDENTIFIER.fields_by_name['version']._options = None
48
- _TASKIDENTIFIER.fields_by_name['version']._serialized_options = b'\372B\006r\004\020\001\030?'
49
- _TASKMETADATA.fields_by_name['deployed_by']._options = None
50
- _TASKMETADATA.fields_by_name['deployed_by']._serialized_options = b'\372B\005\212\001\002\020\001'
51
- _TASKMETADATA.fields_by_name['deployed_at']._options = None
52
- _TASKMETADATA.fields_by_name['deployed_at']._serialized_options = b'\372B\005\262\001\002\010\001'
53
- _TASK.fields_by_name['task_id']._options = None
54
- _TASK.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
55
- _TASK.fields_by_name['metadata']._options = None
56
- _TASK.fields_by_name['metadata']._serialized_options = b'\372B\005\212\001\002\020\001'
57
- _TASKSPEC.fields_by_name['task_template']._options = None
58
- _TASKSPEC.fields_by_name['task_template']._serialized_options = b'\372B\005\212\001\002\020\001'
59
- _TASKSPEC.fields_by_name['short_name']._options = None
60
- _TASKSPEC.fields_by_name['short_name']._serialized_options = b'\372B\004r\002\030?'
61
- _TASKDETAILS.fields_by_name['task_id']._options = None
62
- _TASKDETAILS.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
63
- _TASKDETAILS.fields_by_name['metadata']._options = None
64
- _TASKDETAILS.fields_by_name['metadata']._serialized_options = b'\372B\005\212\001\002\020\001'
65
- _TASKDETAILS.fields_by_name['spec']._options = None
66
- _TASKDETAILS.fields_by_name['spec']._serialized_options = b'\372B\005\212\001\002\020\001'
67
- _globals['_TASKNAME']._serialized_start=238
68
- _globals['_TASKNAME']._serialized_end=381
69
- _globals['_TASKIDENTIFIER']._serialized_start=384
70
- _globals['_TASKIDENTIFIER']._serialized_end=570
71
- _globals['_TASKMETADATA']._serialized_start=573
72
- _globals['_TASKMETADATA']._serialized_end=810
73
- _globals['_TASK']._serialized_start=813
74
- _globals['_TASK']._serialized_end=960
75
- _globals['_TASKSPEC']._serialized_start=963
76
- _globals['_TASKSPEC']._serialized_end=1229
77
- _globals['_TASKDETAILS']._serialized_start=1232
78
- _globals['_TASKDETAILS']._serialized_end=1445
79
- # @@protoc_insertion_point(module_scope)
@@ -1,81 +0,0 @@
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 tasks_pb2 as _tasks_pb2
4
- from google.protobuf import timestamp_pb2 as _timestamp_pb2
5
- from flyte._protos.validate.validate import validate_pb2 as _validate_pb2
6
- from flyte._protos.workflow import common_pb2 as _common_pb2
7
- from flyte._protos.workflow import environment_pb2 as _environment_pb2
8
- from google.protobuf.internal import containers as _containers
9
- from google.protobuf import descriptor as _descriptor
10
- from google.protobuf import message as _message
11
- from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
12
-
13
- DESCRIPTOR: _descriptor.FileDescriptor
14
-
15
- class TaskName(_message.Message):
16
- __slots__ = ["org", "project", "domain", "name"]
17
- ORG_FIELD_NUMBER: _ClassVar[int]
18
- PROJECT_FIELD_NUMBER: _ClassVar[int]
19
- DOMAIN_FIELD_NUMBER: _ClassVar[int]
20
- NAME_FIELD_NUMBER: _ClassVar[int]
21
- org: str
22
- project: str
23
- domain: str
24
- name: str
25
- def __init__(self, org: _Optional[str] = ..., project: _Optional[str] = ..., domain: _Optional[str] = ..., name: _Optional[str] = ...) -> None: ...
26
-
27
- class TaskIdentifier(_message.Message):
28
- __slots__ = ["org", "project", "domain", "name", "version"]
29
- ORG_FIELD_NUMBER: _ClassVar[int]
30
- PROJECT_FIELD_NUMBER: _ClassVar[int]
31
- DOMAIN_FIELD_NUMBER: _ClassVar[int]
32
- NAME_FIELD_NUMBER: _ClassVar[int]
33
- VERSION_FIELD_NUMBER: _ClassVar[int]
34
- org: str
35
- project: str
36
- domain: str
37
- name: str
38
- version: str
39
- def __init__(self, org: _Optional[str] = ..., project: _Optional[str] = ..., domain: _Optional[str] = ..., name: _Optional[str] = ..., version: _Optional[str] = ...) -> None: ...
40
-
41
- class TaskMetadata(_message.Message):
42
- __slots__ = ["deployed_by", "short_name", "deployed_at", "environment_name"]
43
- DEPLOYED_BY_FIELD_NUMBER: _ClassVar[int]
44
- SHORT_NAME_FIELD_NUMBER: _ClassVar[int]
45
- DEPLOYED_AT_FIELD_NUMBER: _ClassVar[int]
46
- ENVIRONMENT_NAME_FIELD_NUMBER: _ClassVar[int]
47
- deployed_by: _identity_pb2.EnrichedIdentity
48
- short_name: str
49
- deployed_at: _timestamp_pb2.Timestamp
50
- environment_name: str
51
- def __init__(self, deployed_by: _Optional[_Union[_identity_pb2.EnrichedIdentity, _Mapping]] = ..., short_name: _Optional[str] = ..., deployed_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., environment_name: _Optional[str] = ...) -> None: ...
52
-
53
- class Task(_message.Message):
54
- __slots__ = ["task_id", "metadata"]
55
- TASK_ID_FIELD_NUMBER: _ClassVar[int]
56
- METADATA_FIELD_NUMBER: _ClassVar[int]
57
- task_id: TaskIdentifier
58
- metadata: TaskMetadata
59
- def __init__(self, task_id: _Optional[_Union[TaskIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[TaskMetadata, _Mapping]] = ...) -> None: ...
60
-
61
- class TaskSpec(_message.Message):
62
- __slots__ = ["task_template", "default_inputs", "short_name", "environment"]
63
- TASK_TEMPLATE_FIELD_NUMBER: _ClassVar[int]
64
- DEFAULT_INPUTS_FIELD_NUMBER: _ClassVar[int]
65
- SHORT_NAME_FIELD_NUMBER: _ClassVar[int]
66
- ENVIRONMENT_FIELD_NUMBER: _ClassVar[int]
67
- task_template: _tasks_pb2.TaskTemplate
68
- default_inputs: _containers.RepeatedCompositeFieldContainer[_common_pb2.NamedParameter]
69
- short_name: str
70
- environment: _environment_pb2.Environment
71
- def __init__(self, task_template: _Optional[_Union[_tasks_pb2.TaskTemplate, _Mapping]] = ..., default_inputs: _Optional[_Iterable[_Union[_common_pb2.NamedParameter, _Mapping]]] = ..., short_name: _Optional[str] = ..., environment: _Optional[_Union[_environment_pb2.Environment, _Mapping]] = ...) -> None: ...
72
-
73
- class TaskDetails(_message.Message):
74
- __slots__ = ["task_id", "metadata", "spec"]
75
- TASK_ID_FIELD_NUMBER: _ClassVar[int]
76
- METADATA_FIELD_NUMBER: _ClassVar[int]
77
- SPEC_FIELD_NUMBER: _ClassVar[int]
78
- task_id: TaskIdentifier
79
- metadata: TaskMetadata
80
- spec: TaskSpec
81
- def __init__(self, task_id: _Optional[_Union[TaskIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[TaskMetadata, _Mapping]] = ..., spec: _Optional[_Union[TaskSpec, _Mapping]] = ...) -> None: ...
@@ -1,4 +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
-
@@ -1,60 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- # Generated by the protocol buffer compiler. DO NOT EDIT!
3
- # source: workflow/task_service.proto
4
- """Generated protocol buffer code."""
5
- from google.protobuf import descriptor as _descriptor
6
- from google.protobuf import descriptor_pool as _descriptor_pool
7
- from google.protobuf import symbol_database as _symbol_database
8
- from google.protobuf.internal import builder as _builder
9
- # @@protoc_insertion_point(imports)
10
-
11
- _sym_db = _symbol_database.Default()
12
-
13
-
14
- from flyte._protos.common import identifier_pb2 as common_dot_identifier__pb2
15
- from flyte._protos.common import list_pb2 as common_dot_list__pb2
16
- from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
17
- from flyte._protos.workflow import task_definition_pb2 as workflow_dot_task__definition__pb2
18
-
19
-
20
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1bworkflow/task_service.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x11\x63ommon/list.proto\x1a\x17validate/validate.proto\x1a\x1eworkflow/task_definition.proto\"\x94\x01\n\x11\x44\x65ployTaskRequest\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\x12\x39\n\x04spec\x18\x02 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x04spec\"\x14\n\x12\x44\x65ployTaskResponse\"]\n\x15GetTaskDetailsRequest\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\"R\n\x16GetTaskDetailsResponse\x12\x38\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32\x1e.cloudidl.workflow.TaskDetailsR\x07\x64\x65tails\"\xdc\x02\n\x10ListTasksRequest\x12\x36\n\x07request\x18\x01 \x01(\x0b\x32\x1c.cloudidl.common.ListRequestR\x07request\x12\x1b\n\x03org\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x03org\x12M\n\nproject_id\x18\x03 \x01(\x0b\x32\".cloudidl.common.ProjectIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tprojectId\x12T\n\rknown_filters\x18\x04 \x03(\x0b\x32/.cloudidl.workflow.ListTasksRequest.KnownFilterR\x0cknownFilters\x1a=\n\x0bKnownFilter\x12!\n\x0b\x64\x65ployed_by\x18\x01 \x01(\tH\x00R\ndeployedByB\x0b\n\tfilter_byB\x0f\n\x08scope_by\x12\x03\xf8\x42\x01\"X\n\x11ListTasksResponse\x12-\n\x05tasks\x18\x01 \x03(\x0b\x32\x17.cloudidl.workflow.TaskR\x05tasks\x12\x14\n\x05token\x18\x02 \x01(\tR\x05token2\xb3\x02\n\x0bTaskService\x12[\n\nDeployTask\x12$.cloudidl.workflow.DeployTaskRequest\x1a%.cloudidl.workflow.DeployTaskResponse\"\x00\x12j\n\x0eGetTaskDetails\x12(.cloudidl.workflow.GetTaskDetailsRequest\x1a).cloudidl.workflow.GetTaskDetailsResponse\"\x03\x90\x02\x01\x12[\n\tListTasks\x12#.cloudidl.workflow.ListTasksRequest\x1a$.cloudidl.workflow.ListTasksResponse\"\x03\x90\x02\x01\x42\xbd\x01\n\x15\x63om.cloudidl.workflowB\x10TaskServiceProtoH\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.task_service_pb2', _globals)
25
- if _descriptor._USE_C_DESCRIPTORS == False:
26
- DESCRIPTOR._options = None
27
- DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\020TaskServiceProtoH\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
- _DEPLOYTASKREQUEST.fields_by_name['task_id']._options = None
29
- _DEPLOYTASKREQUEST.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
30
- _DEPLOYTASKREQUEST.fields_by_name['spec']._options = None
31
- _DEPLOYTASKREQUEST.fields_by_name['spec']._serialized_options = b'\372B\005\212\001\002\020\001'
32
- _GETTASKDETAILSREQUEST.fields_by_name['task_id']._options = None
33
- _GETTASKDETAILSREQUEST.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
34
- _LISTTASKSREQUEST.oneofs_by_name['scope_by']._options = None
35
- _LISTTASKSREQUEST.oneofs_by_name['scope_by']._serialized_options = b'\370B\001'
36
- _LISTTASKSREQUEST.fields_by_name['org']._options = None
37
- _LISTTASKSREQUEST.fields_by_name['org']._serialized_options = b'\372B\004r\002\020\001'
38
- _LISTTASKSREQUEST.fields_by_name['project_id']._options = None
39
- _LISTTASKSREQUEST.fields_by_name['project_id']._serialized_options = b'\372B\005\212\001\002\020\001'
40
- _TASKSERVICE.methods_by_name['GetTaskDetails']._options = None
41
- _TASKSERVICE.methods_by_name['GetTaskDetails']._serialized_options = b'\220\002\001'
42
- _TASKSERVICE.methods_by_name['ListTasks']._options = None
43
- _TASKSERVICE.methods_by_name['ListTasks']._serialized_options = b'\220\002\001'
44
- _globals['_DEPLOYTASKREQUEST']._serialized_start=152
45
- _globals['_DEPLOYTASKREQUEST']._serialized_end=300
46
- _globals['_DEPLOYTASKRESPONSE']._serialized_start=302
47
- _globals['_DEPLOYTASKRESPONSE']._serialized_end=322
48
- _globals['_GETTASKDETAILSREQUEST']._serialized_start=324
49
- _globals['_GETTASKDETAILSREQUEST']._serialized_end=417
50
- _globals['_GETTASKDETAILSRESPONSE']._serialized_start=419
51
- _globals['_GETTASKDETAILSRESPONSE']._serialized_end=501
52
- _globals['_LISTTASKSREQUEST']._serialized_start=504
53
- _globals['_LISTTASKSREQUEST']._serialized_end=852
54
- _globals['_LISTTASKSREQUEST_KNOWNFILTER']._serialized_start=774
55
- _globals['_LISTTASKSREQUEST_KNOWNFILTER']._serialized_end=835
56
- _globals['_LISTTASKSRESPONSE']._serialized_start=854
57
- _globals['_LISTTASKSRESPONSE']._serialized_end=942
58
- _globals['_TASKSERVICE']._serialized_start=945
59
- _globals['_TASKSERVICE']._serialized_end=1252
60
- # @@protoc_insertion_point(module_scope)
@@ -1,59 +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 task_definition_pb2 as _task_definition_pb2
5
- from google.protobuf.internal import containers as _containers
6
- from google.protobuf import descriptor as _descriptor
7
- from google.protobuf import message as _message
8
- from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
9
-
10
- DESCRIPTOR: _descriptor.FileDescriptor
11
-
12
- class DeployTaskRequest(_message.Message):
13
- __slots__ = ["task_id", "spec"]
14
- TASK_ID_FIELD_NUMBER: _ClassVar[int]
15
- SPEC_FIELD_NUMBER: _ClassVar[int]
16
- task_id: _task_definition_pb2.TaskIdentifier
17
- spec: _task_definition_pb2.TaskSpec
18
- def __init__(self, task_id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ..., spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ...) -> None: ...
19
-
20
- class DeployTaskResponse(_message.Message):
21
- __slots__ = []
22
- def __init__(self) -> None: ...
23
-
24
- class GetTaskDetailsRequest(_message.Message):
25
- __slots__ = ["task_id"]
26
- TASK_ID_FIELD_NUMBER: _ClassVar[int]
27
- task_id: _task_definition_pb2.TaskIdentifier
28
- def __init__(self, task_id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ...) -> None: ...
29
-
30
- class GetTaskDetailsResponse(_message.Message):
31
- __slots__ = ["details"]
32
- DETAILS_FIELD_NUMBER: _ClassVar[int]
33
- details: _task_definition_pb2.TaskDetails
34
- def __init__(self, details: _Optional[_Union[_task_definition_pb2.TaskDetails, _Mapping]] = ...) -> None: ...
35
-
36
- class ListTasksRequest(_message.Message):
37
- __slots__ = ["request", "org", "project_id", "known_filters"]
38
- class KnownFilter(_message.Message):
39
- __slots__ = ["deployed_by"]
40
- DEPLOYED_BY_FIELD_NUMBER: _ClassVar[int]
41
- deployed_by: str
42
- def __init__(self, deployed_by: _Optional[str] = ...) -> None: ...
43
- REQUEST_FIELD_NUMBER: _ClassVar[int]
44
- ORG_FIELD_NUMBER: _ClassVar[int]
45
- PROJECT_ID_FIELD_NUMBER: _ClassVar[int]
46
- KNOWN_FILTERS_FIELD_NUMBER: _ClassVar[int]
47
- request: _list_pb2.ListRequest
48
- org: str
49
- project_id: _identifier_pb2.ProjectIdentifier
50
- known_filters: _containers.RepeatedCompositeFieldContainer[ListTasksRequest.KnownFilter]
51
- def __init__(self, request: _Optional[_Union[_list_pb2.ListRequest, _Mapping]] = ..., org: _Optional[str] = ..., project_id: _Optional[_Union[_identifier_pb2.ProjectIdentifier, _Mapping]] = ..., known_filters: _Optional[_Iterable[_Union[ListTasksRequest.KnownFilter, _Mapping]]] = ...) -> None: ...
52
-
53
- class ListTasksResponse(_message.Message):
54
- __slots__ = ["tasks", "token"]
55
- TASKS_FIELD_NUMBER: _ClassVar[int]
56
- TOKEN_FIELD_NUMBER: _ClassVar[int]
57
- tasks: _containers.RepeatedCompositeFieldContainer[_task_definition_pb2.Task]
58
- token: str
59
- def __init__(self, tasks: _Optional[_Iterable[_Union[_task_definition_pb2.Task, _Mapping]]] = ..., token: _Optional[str] = ...) -> None: ...