flyte 0.0.1b0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of flyte might be problematic. Click here for more details.
- flyte/__init__.py +62 -0
- flyte/_api_commons.py +3 -0
- flyte/_bin/__init__.py +0 -0
- flyte/_bin/runtime.py +126 -0
- flyte/_build.py +25 -0
- flyte/_cache/__init__.py +12 -0
- flyte/_cache/cache.py +146 -0
- flyte/_cache/defaults.py +9 -0
- flyte/_cache/policy_function_body.py +42 -0
- flyte/_cli/__init__.py +0 -0
- flyte/_cli/_common.py +287 -0
- flyte/_cli/_create.py +42 -0
- flyte/_cli/_delete.py +23 -0
- flyte/_cli/_deploy.py +140 -0
- flyte/_cli/_get.py +235 -0
- flyte/_cli/_run.py +152 -0
- flyte/_cli/main.py +72 -0
- flyte/_code_bundle/__init__.py +8 -0
- flyte/_code_bundle/_ignore.py +113 -0
- flyte/_code_bundle/_packaging.py +187 -0
- flyte/_code_bundle/_utils.py +339 -0
- flyte/_code_bundle/bundle.py +178 -0
- flyte/_context.py +146 -0
- flyte/_datastructures.py +342 -0
- flyte/_deploy.py +202 -0
- flyte/_doc.py +29 -0
- flyte/_docstring.py +32 -0
- flyte/_environment.py +43 -0
- flyte/_group.py +31 -0
- flyte/_hash.py +23 -0
- flyte/_image.py +760 -0
- flyte/_initialize.py +634 -0
- flyte/_interface.py +84 -0
- flyte/_internal/__init__.py +3 -0
- flyte/_internal/controllers/__init__.py +115 -0
- flyte/_internal/controllers/_local_controller.py +118 -0
- flyte/_internal/controllers/_trace.py +40 -0
- flyte/_internal/controllers/pbhash.py +39 -0
- flyte/_internal/controllers/remote/__init__.py +40 -0
- flyte/_internal/controllers/remote/_action.py +141 -0
- flyte/_internal/controllers/remote/_client.py +43 -0
- flyte/_internal/controllers/remote/_controller.py +361 -0
- flyte/_internal/controllers/remote/_core.py +402 -0
- flyte/_internal/controllers/remote/_informer.py +361 -0
- flyte/_internal/controllers/remote/_service_protocol.py +50 -0
- flyte/_internal/imagebuild/__init__.py +11 -0
- flyte/_internal/imagebuild/docker_builder.py +416 -0
- flyte/_internal/imagebuild/image_builder.py +241 -0
- flyte/_internal/imagebuild/remote_builder.py +0 -0
- flyte/_internal/resolvers/__init__.py +0 -0
- flyte/_internal/resolvers/_task_module.py +54 -0
- flyte/_internal/resolvers/common.py +31 -0
- flyte/_internal/resolvers/default.py +28 -0
- flyte/_internal/runtime/__init__.py +0 -0
- flyte/_internal/runtime/convert.py +199 -0
- flyte/_internal/runtime/entrypoints.py +135 -0
- flyte/_internal/runtime/io.py +136 -0
- flyte/_internal/runtime/resources_serde.py +138 -0
- flyte/_internal/runtime/task_serde.py +210 -0
- flyte/_internal/runtime/taskrunner.py +190 -0
- flyte/_internal/runtime/types_serde.py +54 -0
- flyte/_logging.py +124 -0
- flyte/_protos/__init__.py +0 -0
- flyte/_protos/common/authorization_pb2.py +66 -0
- flyte/_protos/common/authorization_pb2.pyi +108 -0
- flyte/_protos/common/authorization_pb2_grpc.py +4 -0
- flyte/_protos/common/identifier_pb2.py +71 -0
- flyte/_protos/common/identifier_pb2.pyi +82 -0
- flyte/_protos/common/identifier_pb2_grpc.py +4 -0
- flyte/_protos/common/identity_pb2.py +48 -0
- flyte/_protos/common/identity_pb2.pyi +72 -0
- flyte/_protos/common/identity_pb2_grpc.py +4 -0
- flyte/_protos/common/list_pb2.py +36 -0
- flyte/_protos/common/list_pb2.pyi +69 -0
- flyte/_protos/common/list_pb2_grpc.py +4 -0
- flyte/_protos/common/policy_pb2.py +37 -0
- flyte/_protos/common/policy_pb2.pyi +27 -0
- flyte/_protos/common/policy_pb2_grpc.py +4 -0
- flyte/_protos/common/role_pb2.py +37 -0
- flyte/_protos/common/role_pb2.pyi +53 -0
- flyte/_protos/common/role_pb2_grpc.py +4 -0
- flyte/_protos/common/runtime_version_pb2.py +28 -0
- flyte/_protos/common/runtime_version_pb2.pyi +24 -0
- flyte/_protos/common/runtime_version_pb2_grpc.py +4 -0
- flyte/_protos/logs/dataplane/payload_pb2.py +96 -0
- flyte/_protos/logs/dataplane/payload_pb2.pyi +168 -0
- flyte/_protos/logs/dataplane/payload_pb2_grpc.py +4 -0
- flyte/_protos/secret/definition_pb2.py +49 -0
- flyte/_protos/secret/definition_pb2.pyi +93 -0
- flyte/_protos/secret/definition_pb2_grpc.py +4 -0
- flyte/_protos/secret/payload_pb2.py +62 -0
- flyte/_protos/secret/payload_pb2.pyi +94 -0
- flyte/_protos/secret/payload_pb2_grpc.py +4 -0
- flyte/_protos/secret/secret_pb2.py +38 -0
- flyte/_protos/secret/secret_pb2.pyi +6 -0
- flyte/_protos/secret/secret_pb2_grpc.py +198 -0
- flyte/_protos/secret/secret_pb2_grpc_grpc.py +198 -0
- flyte/_protos/validate/validate/validate_pb2.py +76 -0
- flyte/_protos/workflow/node_execution_service_pb2.py +26 -0
- flyte/_protos/workflow/node_execution_service_pb2.pyi +4 -0
- flyte/_protos/workflow/node_execution_service_pb2_grpc.py +32 -0
- flyte/_protos/workflow/queue_service_pb2.py +106 -0
- flyte/_protos/workflow/queue_service_pb2.pyi +141 -0
- flyte/_protos/workflow/queue_service_pb2_grpc.py +172 -0
- flyte/_protos/workflow/run_definition_pb2.py +128 -0
- flyte/_protos/workflow/run_definition_pb2.pyi +310 -0
- flyte/_protos/workflow/run_definition_pb2_grpc.py +4 -0
- flyte/_protos/workflow/run_logs_service_pb2.py +41 -0
- flyte/_protos/workflow/run_logs_service_pb2.pyi +28 -0
- flyte/_protos/workflow/run_logs_service_pb2_grpc.py +69 -0
- flyte/_protos/workflow/run_service_pb2.py +133 -0
- flyte/_protos/workflow/run_service_pb2.pyi +175 -0
- flyte/_protos/workflow/run_service_pb2_grpc.py +412 -0
- flyte/_protos/workflow/state_service_pb2.py +58 -0
- flyte/_protos/workflow/state_service_pb2.pyi +71 -0
- flyte/_protos/workflow/state_service_pb2_grpc.py +138 -0
- flyte/_protos/workflow/task_definition_pb2.py +72 -0
- flyte/_protos/workflow/task_definition_pb2.pyi +65 -0
- flyte/_protos/workflow/task_definition_pb2_grpc.py +4 -0
- flyte/_protos/workflow/task_service_pb2.py +44 -0
- flyte/_protos/workflow/task_service_pb2.pyi +31 -0
- flyte/_protos/workflow/task_service_pb2_grpc.py +104 -0
- flyte/_resources.py +226 -0
- flyte/_retry.py +32 -0
- flyte/_reusable_environment.py +25 -0
- flyte/_run.py +411 -0
- flyte/_secret.py +61 -0
- flyte/_task.py +367 -0
- flyte/_task_environment.py +200 -0
- flyte/_timeout.py +47 -0
- flyte/_tools.py +27 -0
- flyte/_trace.py +128 -0
- flyte/_utils/__init__.py +20 -0
- flyte/_utils/asyn.py +119 -0
- flyte/_utils/coro_management.py +25 -0
- flyte/_utils/file_handling.py +72 -0
- flyte/_utils/helpers.py +108 -0
- flyte/_utils/lazy_module.py +54 -0
- flyte/_utils/uv_script_parser.py +49 -0
- flyte/_version.py +21 -0
- flyte/connectors/__init__.py +0 -0
- flyte/errors.py +143 -0
- flyte/extras/__init__.py +5 -0
- flyte/extras/_container.py +273 -0
- flyte/io/__init__.py +11 -0
- flyte/io/_dataframe.py +0 -0
- flyte/io/_dir.py +448 -0
- flyte/io/_file.py +468 -0
- flyte/io/pickle/__init__.py +0 -0
- flyte/io/pickle/transformer.py +117 -0
- flyte/io/structured_dataset/__init__.py +129 -0
- flyte/io/structured_dataset/basic_dfs.py +219 -0
- flyte/io/structured_dataset/structured_dataset.py +1061 -0
- flyte/py.typed +0 -0
- flyte/remote/__init__.py +25 -0
- flyte/remote/_client/__init__.py +0 -0
- flyte/remote/_client/_protocols.py +131 -0
- flyte/remote/_client/auth/__init__.py +12 -0
- flyte/remote/_client/auth/_authenticators/__init__.py +0 -0
- flyte/remote/_client/auth/_authenticators/base.py +397 -0
- flyte/remote/_client/auth/_authenticators/client_credentials.py +73 -0
- flyte/remote/_client/auth/_authenticators/device_code.py +118 -0
- flyte/remote/_client/auth/_authenticators/external_command.py +79 -0
- flyte/remote/_client/auth/_authenticators/factory.py +200 -0
- flyte/remote/_client/auth/_authenticators/pkce.py +516 -0
- flyte/remote/_client/auth/_channel.py +184 -0
- flyte/remote/_client/auth/_client_config.py +83 -0
- flyte/remote/_client/auth/_default_html.py +32 -0
- flyte/remote/_client/auth/_grpc_utils/__init__.py +0 -0
- flyte/remote/_client/auth/_grpc_utils/auth_interceptor.py +288 -0
- flyte/remote/_client/auth/_grpc_utils/default_metadata_interceptor.py +151 -0
- flyte/remote/_client/auth/_keyring.py +143 -0
- flyte/remote/_client/auth/_token_client.py +260 -0
- flyte/remote/_client/auth/errors.py +16 -0
- flyte/remote/_client/controlplane.py +95 -0
- flyte/remote/_console.py +18 -0
- flyte/remote/_data.py +155 -0
- flyte/remote/_logs.py +116 -0
- flyte/remote/_project.py +86 -0
- flyte/remote/_run.py +873 -0
- flyte/remote/_secret.py +132 -0
- flyte/remote/_task.py +227 -0
- flyte/report/__init__.py +3 -0
- flyte/report/_report.py +178 -0
- flyte/report/_template.html +124 -0
- flyte/storage/__init__.py +24 -0
- flyte/storage/_remote_fs.py +34 -0
- flyte/storage/_storage.py +251 -0
- flyte/storage/_utils.py +5 -0
- flyte/types/__init__.py +13 -0
- flyte/types/_interface.py +25 -0
- flyte/types/_renderer.py +162 -0
- flyte/types/_string_literals.py +120 -0
- flyte/types/_type_engine.py +2210 -0
- flyte/types/_utils.py +80 -0
- flyte-0.0.1b0.dist-info/METADATA +179 -0
- flyte-0.0.1b0.dist-info/RECORD +390 -0
- flyte-0.0.1b0.dist-info/WHEEL +5 -0
- flyte-0.0.1b0.dist-info/entry_points.txt +3 -0
- flyte-0.0.1b0.dist-info/top_level.txt +1 -0
- union/__init__.py +54 -0
- union/_api_commons.py +3 -0
- union/_bin/__init__.py +0 -0
- union/_bin/runtime.py +113 -0
- union/_build.py +25 -0
- union/_cache/__init__.py +12 -0
- union/_cache/cache.py +141 -0
- union/_cache/defaults.py +9 -0
- union/_cache/policy_function_body.py +42 -0
- union/_cli/__init__.py +0 -0
- union/_cli/_common.py +263 -0
- union/_cli/_create.py +40 -0
- union/_cli/_delete.py +23 -0
- union/_cli/_deploy.py +120 -0
- union/_cli/_get.py +162 -0
- union/_cli/_params.py +579 -0
- union/_cli/_run.py +150 -0
- union/_cli/main.py +72 -0
- union/_code_bundle/__init__.py +8 -0
- union/_code_bundle/_ignore.py +113 -0
- union/_code_bundle/_packaging.py +187 -0
- union/_code_bundle/_utils.py +342 -0
- union/_code_bundle/bundle.py +176 -0
- union/_context.py +146 -0
- union/_datastructures.py +295 -0
- union/_deploy.py +185 -0
- union/_doc.py +29 -0
- union/_docstring.py +26 -0
- union/_environment.py +43 -0
- union/_group.py +31 -0
- union/_hash.py +23 -0
- union/_image.py +760 -0
- union/_initialize.py +585 -0
- union/_interface.py +84 -0
- union/_internal/__init__.py +3 -0
- union/_internal/controllers/__init__.py +77 -0
- union/_internal/controllers/_local_controller.py +77 -0
- union/_internal/controllers/pbhash.py +39 -0
- union/_internal/controllers/remote/__init__.py +40 -0
- union/_internal/controllers/remote/_action.py +131 -0
- union/_internal/controllers/remote/_client.py +43 -0
- union/_internal/controllers/remote/_controller.py +169 -0
- union/_internal/controllers/remote/_core.py +341 -0
- union/_internal/controllers/remote/_informer.py +260 -0
- union/_internal/controllers/remote/_service_protocol.py +44 -0
- union/_internal/imagebuild/__init__.py +11 -0
- union/_internal/imagebuild/docker_builder.py +416 -0
- union/_internal/imagebuild/image_builder.py +243 -0
- union/_internal/imagebuild/remote_builder.py +0 -0
- union/_internal/resolvers/__init__.py +0 -0
- union/_internal/resolvers/_task_module.py +31 -0
- union/_internal/resolvers/common.py +24 -0
- union/_internal/resolvers/default.py +27 -0
- union/_internal/runtime/__init__.py +0 -0
- union/_internal/runtime/convert.py +163 -0
- union/_internal/runtime/entrypoints.py +121 -0
- union/_internal/runtime/io.py +136 -0
- union/_internal/runtime/resources_serde.py +134 -0
- union/_internal/runtime/task_serde.py +202 -0
- union/_internal/runtime/taskrunner.py +179 -0
- union/_internal/runtime/types_serde.py +53 -0
- union/_logging.py +124 -0
- union/_protos/__init__.py +0 -0
- union/_protos/common/authorization_pb2.py +66 -0
- union/_protos/common/authorization_pb2.pyi +106 -0
- union/_protos/common/authorization_pb2_grpc.py +4 -0
- union/_protos/common/identifier_pb2.py +71 -0
- union/_protos/common/identifier_pb2.pyi +82 -0
- union/_protos/common/identifier_pb2_grpc.py +4 -0
- union/_protos/common/identity_pb2.py +48 -0
- union/_protos/common/identity_pb2.pyi +72 -0
- union/_protos/common/identity_pb2_grpc.py +4 -0
- union/_protos/common/list_pb2.py +36 -0
- union/_protos/common/list_pb2.pyi +69 -0
- union/_protos/common/list_pb2_grpc.py +4 -0
- union/_protos/common/policy_pb2.py +37 -0
- union/_protos/common/policy_pb2.pyi +27 -0
- union/_protos/common/policy_pb2_grpc.py +4 -0
- union/_protos/common/role_pb2.py +37 -0
- union/_protos/common/role_pb2.pyi +51 -0
- union/_protos/common/role_pb2_grpc.py +4 -0
- union/_protos/common/runtime_version_pb2.py +28 -0
- union/_protos/common/runtime_version_pb2.pyi +24 -0
- union/_protos/common/runtime_version_pb2_grpc.py +4 -0
- union/_protos/logs/dataplane/payload_pb2.py +96 -0
- union/_protos/logs/dataplane/payload_pb2.pyi +168 -0
- union/_protos/logs/dataplane/payload_pb2_grpc.py +4 -0
- union/_protos/secret/definition_pb2.py +49 -0
- union/_protos/secret/definition_pb2.pyi +93 -0
- union/_protos/secret/definition_pb2_grpc.py +4 -0
- union/_protos/secret/payload_pb2.py +62 -0
- union/_protos/secret/payload_pb2.pyi +94 -0
- union/_protos/secret/payload_pb2_grpc.py +4 -0
- union/_protos/secret/secret_pb2.py +38 -0
- union/_protos/secret/secret_pb2.pyi +6 -0
- union/_protos/secret/secret_pb2_grpc.py +198 -0
- union/_protos/validate/validate/validate_pb2.py +76 -0
- union/_protos/workflow/node_execution_service_pb2.py +26 -0
- union/_protos/workflow/node_execution_service_pb2.pyi +4 -0
- union/_protos/workflow/node_execution_service_pb2_grpc.py +32 -0
- union/_protos/workflow/queue_service_pb2.py +75 -0
- union/_protos/workflow/queue_service_pb2.pyi +103 -0
- union/_protos/workflow/queue_service_pb2_grpc.py +172 -0
- union/_protos/workflow/run_definition_pb2.py +100 -0
- union/_protos/workflow/run_definition_pb2.pyi +256 -0
- union/_protos/workflow/run_definition_pb2_grpc.py +4 -0
- union/_protos/workflow/run_logs_service_pb2.py +41 -0
- union/_protos/workflow/run_logs_service_pb2.pyi +28 -0
- union/_protos/workflow/run_logs_service_pb2_grpc.py +69 -0
- union/_protos/workflow/run_service_pb2.py +133 -0
- union/_protos/workflow/run_service_pb2.pyi +173 -0
- union/_protos/workflow/run_service_pb2_grpc.py +412 -0
- union/_protos/workflow/state_service_pb2.py +58 -0
- union/_protos/workflow/state_service_pb2.pyi +69 -0
- union/_protos/workflow/state_service_pb2_grpc.py +138 -0
- union/_protos/workflow/task_definition_pb2.py +72 -0
- union/_protos/workflow/task_definition_pb2.pyi +65 -0
- union/_protos/workflow/task_definition_pb2_grpc.py +4 -0
- union/_protos/workflow/task_service_pb2.py +44 -0
- union/_protos/workflow/task_service_pb2.pyi +31 -0
- union/_protos/workflow/task_service_pb2_grpc.py +104 -0
- union/_resources.py +226 -0
- union/_retry.py +32 -0
- union/_reusable_environment.py +25 -0
- union/_run.py +374 -0
- union/_secret.py +61 -0
- union/_task.py +354 -0
- union/_task_environment.py +186 -0
- union/_timeout.py +47 -0
- union/_tools.py +27 -0
- union/_utils/__init__.py +11 -0
- union/_utils/asyn.py +119 -0
- union/_utils/file_handling.py +71 -0
- union/_utils/helpers.py +46 -0
- union/_utils/lazy_module.py +54 -0
- union/_utils/uv_script_parser.py +49 -0
- union/_version.py +21 -0
- union/connectors/__init__.py +0 -0
- union/errors.py +128 -0
- union/extras/__init__.py +5 -0
- union/extras/_container.py +263 -0
- union/io/__init__.py +11 -0
- union/io/_dataframe.py +0 -0
- union/io/_dir.py +425 -0
- union/io/_file.py +418 -0
- union/io/pickle/__init__.py +0 -0
- union/io/pickle/transformer.py +117 -0
- union/io/structured_dataset/__init__.py +122 -0
- union/io/structured_dataset/basic_dfs.py +219 -0
- union/io/structured_dataset/structured_dataset.py +1057 -0
- union/py.typed +0 -0
- union/remote/__init__.py +23 -0
- union/remote/_client/__init__.py +0 -0
- union/remote/_client/_protocols.py +129 -0
- union/remote/_client/auth/__init__.py +12 -0
- union/remote/_client/auth/_authenticators/__init__.py +0 -0
- union/remote/_client/auth/_authenticators/base.py +391 -0
- union/remote/_client/auth/_authenticators/client_credentials.py +73 -0
- union/remote/_client/auth/_authenticators/device_code.py +120 -0
- union/remote/_client/auth/_authenticators/external_command.py +77 -0
- union/remote/_client/auth/_authenticators/factory.py +200 -0
- union/remote/_client/auth/_authenticators/pkce.py +515 -0
- union/remote/_client/auth/_channel.py +184 -0
- union/remote/_client/auth/_client_config.py +83 -0
- union/remote/_client/auth/_default_html.py +32 -0
- union/remote/_client/auth/_grpc_utils/__init__.py +0 -0
- union/remote/_client/auth/_grpc_utils/auth_interceptor.py +204 -0
- union/remote/_client/auth/_grpc_utils/default_metadata_interceptor.py +144 -0
- union/remote/_client/auth/_keyring.py +154 -0
- union/remote/_client/auth/_token_client.py +258 -0
- union/remote/_client/auth/errors.py +16 -0
- union/remote/_client/controlplane.py +86 -0
- union/remote/_data.py +149 -0
- union/remote/_logs.py +74 -0
- union/remote/_project.py +86 -0
- union/remote/_run.py +820 -0
- union/remote/_secret.py +132 -0
- union/remote/_task.py +193 -0
- union/report/__init__.py +3 -0
- union/report/_report.py +178 -0
- union/report/_template.html +124 -0
- union/storage/__init__.py +24 -0
- union/storage/_remote_fs.py +34 -0
- union/storage/_storage.py +247 -0
- union/storage/_utils.py +5 -0
- union/types/__init__.py +11 -0
- union/types/_renderer.py +162 -0
- union/types/_string_literals.py +120 -0
- union/types/_type_engine.py +2131 -0
- union/types/_utils.py +80 -0
|
@@ -0,0 +1,172 @@
|
|
|
1
|
+
# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
|
|
2
|
+
"""Client and server classes corresponding to protobuf-defined services."""
|
|
3
|
+
import grpc
|
|
4
|
+
|
|
5
|
+
from flyte._protos.workflow import queue_service_pb2 as workflow_dot_queue__service__pb2
|
|
6
|
+
|
|
7
|
+
|
|
8
|
+
class QueueServiceStub(object):
|
|
9
|
+
"""provides an interface for managing execution of runs over a collection of workers.
|
|
10
|
+
"""
|
|
11
|
+
|
|
12
|
+
def __init__(self, channel):
|
|
13
|
+
"""Constructor.
|
|
14
|
+
|
|
15
|
+
Args:
|
|
16
|
+
channel: A grpc.Channel.
|
|
17
|
+
"""
|
|
18
|
+
self.EnqueueAction = channel.unary_unary(
|
|
19
|
+
'/cloudidl.workflow.QueueService/EnqueueAction',
|
|
20
|
+
request_serializer=workflow_dot_queue__service__pb2.EnqueueActionRequest.SerializeToString,
|
|
21
|
+
response_deserializer=workflow_dot_queue__service__pb2.EnqueueActionResponse.FromString,
|
|
22
|
+
)
|
|
23
|
+
self.AbortQueuedAction = channel.unary_unary(
|
|
24
|
+
'/cloudidl.workflow.QueueService/AbortQueuedAction',
|
|
25
|
+
request_serializer=workflow_dot_queue__service__pb2.AbortQueuedActionRequest.SerializeToString,
|
|
26
|
+
response_deserializer=workflow_dot_queue__service__pb2.AbortQueuedActionResponse.FromString,
|
|
27
|
+
)
|
|
28
|
+
self.Heartbeat = channel.stream_stream(
|
|
29
|
+
'/cloudidl.workflow.QueueService/Heartbeat',
|
|
30
|
+
request_serializer=workflow_dot_queue__service__pb2.HeartbeatRequest.SerializeToString,
|
|
31
|
+
response_deserializer=workflow_dot_queue__service__pb2.HeartbeatResponse.FromString,
|
|
32
|
+
)
|
|
33
|
+
self.StreamLeases = channel.unary_stream(
|
|
34
|
+
'/cloudidl.workflow.QueueService/StreamLeases',
|
|
35
|
+
request_serializer=workflow_dot_queue__service__pb2.StreamLeasesRequest.SerializeToString,
|
|
36
|
+
response_deserializer=workflow_dot_queue__service__pb2.StreamLeasesResponse.FromString,
|
|
37
|
+
)
|
|
38
|
+
|
|
39
|
+
|
|
40
|
+
class QueueServiceServicer(object):
|
|
41
|
+
"""provides an interface for managing execution of runs over a collection of workers.
|
|
42
|
+
"""
|
|
43
|
+
|
|
44
|
+
def EnqueueAction(self, request, context):
|
|
45
|
+
"""queue a new action for execution.
|
|
46
|
+
"""
|
|
47
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
48
|
+
context.set_details('Method not implemented!')
|
|
49
|
+
raise NotImplementedError('Method not implemented!')
|
|
50
|
+
|
|
51
|
+
def AbortQueuedAction(self, request, context):
|
|
52
|
+
"""abort a queued action.
|
|
53
|
+
"""
|
|
54
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
55
|
+
context.set_details('Method not implemented!')
|
|
56
|
+
raise NotImplementedError('Method not implemented!')
|
|
57
|
+
|
|
58
|
+
def Heartbeat(self, request_iterator, context):
|
|
59
|
+
"""heartbeat is used by workers to report their status and acquire new leases.
|
|
60
|
+
"""
|
|
61
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
62
|
+
context.set_details('Method not implemented!')
|
|
63
|
+
raise NotImplementedError('Method not implemented!')
|
|
64
|
+
|
|
65
|
+
def StreamLeases(self, request, context):
|
|
66
|
+
"""Stream leases for a worker.
|
|
67
|
+
"""
|
|
68
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
69
|
+
context.set_details('Method not implemented!')
|
|
70
|
+
raise NotImplementedError('Method not implemented!')
|
|
71
|
+
|
|
72
|
+
|
|
73
|
+
def add_QueueServiceServicer_to_server(servicer, server):
|
|
74
|
+
rpc_method_handlers = {
|
|
75
|
+
'EnqueueAction': grpc.unary_unary_rpc_method_handler(
|
|
76
|
+
servicer.EnqueueAction,
|
|
77
|
+
request_deserializer=workflow_dot_queue__service__pb2.EnqueueActionRequest.FromString,
|
|
78
|
+
response_serializer=workflow_dot_queue__service__pb2.EnqueueActionResponse.SerializeToString,
|
|
79
|
+
),
|
|
80
|
+
'AbortQueuedAction': grpc.unary_unary_rpc_method_handler(
|
|
81
|
+
servicer.AbortQueuedAction,
|
|
82
|
+
request_deserializer=workflow_dot_queue__service__pb2.AbortQueuedActionRequest.FromString,
|
|
83
|
+
response_serializer=workflow_dot_queue__service__pb2.AbortQueuedActionResponse.SerializeToString,
|
|
84
|
+
),
|
|
85
|
+
'Heartbeat': grpc.stream_stream_rpc_method_handler(
|
|
86
|
+
servicer.Heartbeat,
|
|
87
|
+
request_deserializer=workflow_dot_queue__service__pb2.HeartbeatRequest.FromString,
|
|
88
|
+
response_serializer=workflow_dot_queue__service__pb2.HeartbeatResponse.SerializeToString,
|
|
89
|
+
),
|
|
90
|
+
'StreamLeases': grpc.unary_stream_rpc_method_handler(
|
|
91
|
+
servicer.StreamLeases,
|
|
92
|
+
request_deserializer=workflow_dot_queue__service__pb2.StreamLeasesRequest.FromString,
|
|
93
|
+
response_serializer=workflow_dot_queue__service__pb2.StreamLeasesResponse.SerializeToString,
|
|
94
|
+
),
|
|
95
|
+
}
|
|
96
|
+
generic_handler = grpc.method_handlers_generic_handler(
|
|
97
|
+
'cloudidl.workflow.QueueService', rpc_method_handlers)
|
|
98
|
+
server.add_generic_rpc_handlers((generic_handler,))
|
|
99
|
+
|
|
100
|
+
|
|
101
|
+
# This class is part of an EXPERIMENTAL API.
|
|
102
|
+
class QueueService(object):
|
|
103
|
+
"""provides an interface for managing execution of runs over a collection of workers.
|
|
104
|
+
"""
|
|
105
|
+
|
|
106
|
+
@staticmethod
|
|
107
|
+
def EnqueueAction(request,
|
|
108
|
+
target,
|
|
109
|
+
options=(),
|
|
110
|
+
channel_credentials=None,
|
|
111
|
+
call_credentials=None,
|
|
112
|
+
insecure=False,
|
|
113
|
+
compression=None,
|
|
114
|
+
wait_for_ready=None,
|
|
115
|
+
timeout=None,
|
|
116
|
+
metadata=None):
|
|
117
|
+
return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.QueueService/EnqueueAction',
|
|
118
|
+
workflow_dot_queue__service__pb2.EnqueueActionRequest.SerializeToString,
|
|
119
|
+
workflow_dot_queue__service__pb2.EnqueueActionResponse.FromString,
|
|
120
|
+
options, channel_credentials,
|
|
121
|
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
|
122
|
+
|
|
123
|
+
@staticmethod
|
|
124
|
+
def AbortQueuedAction(request,
|
|
125
|
+
target,
|
|
126
|
+
options=(),
|
|
127
|
+
channel_credentials=None,
|
|
128
|
+
call_credentials=None,
|
|
129
|
+
insecure=False,
|
|
130
|
+
compression=None,
|
|
131
|
+
wait_for_ready=None,
|
|
132
|
+
timeout=None,
|
|
133
|
+
metadata=None):
|
|
134
|
+
return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.QueueService/AbortQueuedAction',
|
|
135
|
+
workflow_dot_queue__service__pb2.AbortQueuedActionRequest.SerializeToString,
|
|
136
|
+
workflow_dot_queue__service__pb2.AbortQueuedActionResponse.FromString,
|
|
137
|
+
options, channel_credentials,
|
|
138
|
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
|
139
|
+
|
|
140
|
+
@staticmethod
|
|
141
|
+
def Heartbeat(request_iterator,
|
|
142
|
+
target,
|
|
143
|
+
options=(),
|
|
144
|
+
channel_credentials=None,
|
|
145
|
+
call_credentials=None,
|
|
146
|
+
insecure=False,
|
|
147
|
+
compression=None,
|
|
148
|
+
wait_for_ready=None,
|
|
149
|
+
timeout=None,
|
|
150
|
+
metadata=None):
|
|
151
|
+
return grpc.experimental.stream_stream(request_iterator, target, '/cloudidl.workflow.QueueService/Heartbeat',
|
|
152
|
+
workflow_dot_queue__service__pb2.HeartbeatRequest.SerializeToString,
|
|
153
|
+
workflow_dot_queue__service__pb2.HeartbeatResponse.FromString,
|
|
154
|
+
options, channel_credentials,
|
|
155
|
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
|
156
|
+
|
|
157
|
+
@staticmethod
|
|
158
|
+
def StreamLeases(request,
|
|
159
|
+
target,
|
|
160
|
+
options=(),
|
|
161
|
+
channel_credentials=None,
|
|
162
|
+
call_credentials=None,
|
|
163
|
+
insecure=False,
|
|
164
|
+
compression=None,
|
|
165
|
+
wait_for_ready=None,
|
|
166
|
+
timeout=None,
|
|
167
|
+
metadata=None):
|
|
168
|
+
return grpc.experimental.unary_stream(request, target, '/cloudidl.workflow.QueueService/StreamLeases',
|
|
169
|
+
workflow_dot_queue__service__pb2.StreamLeasesRequest.SerializeToString,
|
|
170
|
+
workflow_dot_queue__service__pb2.StreamLeasesResponse.FromString,
|
|
171
|
+
options, channel_credentials,
|
|
172
|
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
|
@@ -0,0 +1,128 @@
|
|
|
1
|
+
# -*- coding: utf-8 -*-
|
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
3
|
+
# source: workflow/run_definition.proto
|
|
4
|
+
"""Generated protocol buffer code."""
|
|
5
|
+
from google.protobuf import descriptor as _descriptor
|
|
6
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
|
7
|
+
from google.protobuf import symbol_database as _symbol_database
|
|
8
|
+
from google.protobuf.internal import builder as _builder
|
|
9
|
+
# @@protoc_insertion_point(imports)
|
|
10
|
+
|
|
11
|
+
_sym_db = _symbol_database.Default()
|
|
12
|
+
|
|
13
|
+
|
|
14
|
+
from flyte._protos.common import identifier_pb2 as common_dot_identifier__pb2
|
|
15
|
+
from flyte._protos.common import identity_pb2 as common_dot_identity__pb2
|
|
16
|
+
from flyteidl.core import execution_pb2 as flyteidl_dot_core_dot_execution__pb2
|
|
17
|
+
from flyteidl.core import literals_pb2 as flyteidl_dot_core_dot_literals__pb2
|
|
18
|
+
from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
|
|
19
|
+
from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
|
|
20
|
+
from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
|
|
21
|
+
from flyte._protos.workflow import task_definition_pb2 as workflow_dot_task__definition__pb2
|
|
22
|
+
|
|
23
|
+
|
|
24
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1dworkflow/run_definition.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x15\x63ommon/identity.proto\x1a\x1d\x66lyteidl/core/execution.proto\x1a\x1c\x66lyteidl/core/literals.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\x1a\x17validate/validate.proto\x1a\x1eworkflow/task_definition.proto\"\x8d\x01\n\rRunIdentifier\x12\x19\n\x03org\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x03org\x12!\n\x07project\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07project\x12\x1f\n\x06\x64omain\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x06\x64omain\x12\x1d\n\x04name\x18\x04 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18\x1eR\x04name\"\x82\x01\n\x06Labels\x12=\n\x06values\x18\x01 \x03(\x0b\x32%.cloudidl.workflow.Labels.ValuesEntryR\x06values\x1a\x39\n\x0bValuesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\"\x8c\x01\n\x0b\x41nnotations\x12\x42\n\x06values\x18\x01 \x03(\x0b\x32*.cloudidl.workflow.Annotations.ValuesEntryR\x06values\x1a\x39\n\x0bValuesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\";\n\x04\x45nvs\x12\x33\n\x06values\x18\x01 \x03(\x0b\x32\x1b.flyteidl.core.KeyValuePairR\x06values\"\x96\x02\n\x07RunSpec\x12\x31\n\x06labels\x18\x01 \x01(\x0b\x32\x19.cloudidl.workflow.LabelsR\x06labels\x12@\n\x0b\x61nnotations\x18\x02 \x01(\x0b\x32\x1e.cloudidl.workflow.AnnotationsR\x0b\x61nnotations\x12+\n\x04\x65nvs\x18\x03 \x01(\x0b\x32\x17.cloudidl.workflow.EnvsR\x04\x65nvs\x12@\n\rinterruptible\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.BoolValueR\rinterruptible\x12\'\n\x0foverwrite_cache\x18\x05 \x01(\x08R\x0eoverwriteCache\"8\n\x03Run\x12\x31\n\x06\x61\x63tion\x18\x01 \x01(\x0b\x32\x19.cloudidl.workflow.ActionR\x06\x61\x63tion\"}\n\nRunDetails\x12\x35\n\x08run_spec\x18\x01 \x01(\x0b\x32\x1a.cloudidl.workflow.RunSpecR\x07runSpec\x12\x38\n\x06\x61\x63tion\x18\x02 \x01(\x0b\x32 .cloudidl.workflow.ActionDetailsR\x06\x61\x63tion\"o\n\x10\x41\x63tionIdentifier\x12<\n\x03run\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x03run\x12\x1d\n\x04name\x18\x02 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18\x1eR\x04name\"d\n\x12TaskActionMetadata\x12\x31\n\x02id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierR\x02id\x12\x1b\n\ttask_type\x18\x02 \x01(\tR\x08taskType\")\n\x13TraceActionMetadata\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\"\x9f\x01\n\x17\x43onditionActionMetadata\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12 \n\x06run_id\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x05runId\x12&\n\taction_id\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x08\x61\x63tionId\x12\x18\n\x06global\x18\x04 \x01(\x08H\x00R\x06globalB\x0c\n\x05scope\x12\x03\xf8\x42\x01\"\xf1\x02\n\x0e\x41\x63tionMetadata\x12\x16\n\x06parent\x18\x03 \x01(\tR\x06parent\x12\x14\n\x05group\x18\x05 \x01(\tR\x05group\x12\x42\n\x0b\x65xecuted_by\x18\x06 \x01(\x0b\x32!.cloudidl.common.EnrichedIdentityR\nexecutedBy\x12\x45\n\x04task\x18\x07 \x01(\x0b\x32%.cloudidl.workflow.TaskActionMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x04task\x12H\n\x05trace\x18\x08 \x01(\x0b\x32&.cloudidl.workflow.TraceActionMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x05trace\x12T\n\tcondition\x18\t \x01(\x0b\x32*.cloudidl.workflow.ConditionActionMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tconditionB\x06\n\x04spec\"\xe7\x01\n\x0c\x41\x63tionStatus\x12.\n\x05phase\x18\x01 \x01(\x0e\x32\x18.cloudidl.workflow.PhaseR\x05phase\x12\x39\n\nstart_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12:\n\x08\x65nd_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x07\x65ndTime\x88\x01\x01\x12#\n\x08\x61ttempts\x18\x04 \x01(\rB\x07\xfa\x42\x04*\x02 \x00R\x08\x61ttemptsB\x0b\n\t_end_time\"\xb5\x01\n\x06\x41\x63tion\x12\x33\n\x02id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierR\x02id\x12=\n\x08metadata\x18\x02 \x01(\x0b\x32!.cloudidl.workflow.ActionMetadataR\x08metadata\x12\x37\n\x06status\x18\x03 \x01(\x0b\x32\x1f.cloudidl.workflow.ActionStatusR\x06status\"\x9e\x02\n\x0e\x45nrichedAction\x12\x31\n\x06\x61\x63tion\x18\x01 \x01(\x0b\x32\x19.cloudidl.workflow.ActionR\x06\x61\x63tion\x12!\n\x0cmeets_filter\x18\x02 \x01(\x08R\x0bmeetsFilter\x12n\n\x15\x63hildren_phase_counts\x18\x03 \x03(\x0b\x32:.cloudidl.workflow.EnrichedAction.ChildrenPhaseCountsEntryR\x13\x63hildrenPhaseCounts\x1a\x46\n\x18\x43hildrenPhaseCountsEntry\x12\x10\n\x03key\x18\x01 \x01(\x05R\x03key\x12\x14\n\x05value\x18\x02 \x01(\x05R\x05value:\x02\x38\x01\"\x9a\x01\n\tErrorInfo\x12\x18\n\x07message\x18\x01 \x01(\tR\x07message\x12\x35\n\x04kind\x18\x02 \x01(\x0e\x32!.cloudidl.workflow.ErrorInfo.KindR\x04kind\"<\n\x04Kind\x12\x14\n\x10KIND_UNSPECIFIED\x10\x00\x12\r\n\tKIND_USER\x10\x01\x12\x0f\n\x0bKIND_SYSTEM\x10\x02\"e\n\tAbortInfo\x12\x16\n\x06reason\x18\x01 \x01(\tR\x06reason\x12@\n\naborted_by\x18\x02 \x01(\x0b\x32!.cloudidl.common.EnrichedIdentityR\tabortedBy\"\xcd\x03\n\rActionDetails\x12\x33\n\x02id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierR\x02id\x12=\n\x08metadata\x18\x02 \x01(\x0b\x32!.cloudidl.workflow.ActionMetadataR\x08metadata\x12\x37\n\x06status\x18\x03 \x01(\x0b\x32\x1f.cloudidl.workflow.ActionStatusR\x06status\x12=\n\nerror_info\x18\x04 \x01(\x0b\x32\x1c.cloudidl.workflow.ErrorInfoH\x00R\terrorInfo\x12=\n\nabort_info\x18\x05 \x01(\x0b\x32\x1c.cloudidl.workflow.AbortInfoH\x00R\tabortInfo\x12I\n\x12resolved_task_spec\x18\x06 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecR\x10resolvedTaskSpec\x12<\n\x08\x61ttempts\x18\x07 \x03(\x0b\x32 .cloudidl.workflow.ActionAttemptR\x08\x61ttemptsB\x08\n\x06result\"\xa9\x03\n\rActionAttempt\x12.\n\x05phase\x18\x01 \x01(\x0e\x32\x18.cloudidl.workflow.PhaseR\x05phase\x12\x39\n\nstart_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12:\n\x08\x65nd_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x07\x65ndTime\x88\x01\x01\x12@\n\nerror_info\x18\x04 \x01(\x0b\x32\x1c.cloudidl.workflow.ErrorInfoH\x01R\terrorInfo\x88\x01\x01\x12!\n\x07\x61ttempt\x18\x05 \x01(\rB\x07\xfa\x42\x04*\x02 \x00R\x07\x61ttempt\x12\x31\n\x08log_info\x18\x06 \x03(\x0b\x32\x16.flyteidl.core.TaskLogR\x07logInfo\x12=\n\x07outputs\x18\x07 \x01(\x0b\x32#.cloudidl.workflow.OutputReferencesR\x07outputsB\x0b\n\t_end_timeB\r\n\x0b_error_info\"\x95\x05\n\x0b\x41\x63tionEvent\x12=\n\x02id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x02id\x12!\n\x07\x61ttempt\x18\x02 \x01(\rB\x07\xfa\x42\x04*\x02 \x00R\x07\x61ttempt\x12.\n\x05phase\x18\x03 \x01(\x0e\x32\x18.cloudidl.workflow.PhaseR\x05phase\x12\x18\n\x07version\x18\x04 \x01(\rR\x07version\x12\x39\n\nstart_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12=\n\x0cupdated_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x0bupdatedTime\x12:\n\x08\x65nd_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x07\x65ndTime\x88\x01\x01\x12@\n\nerror_info\x18\x08 \x01(\x0b\x32\x1c.cloudidl.workflow.ErrorInfoH\x01R\terrorInfo\x88\x01\x01\x12\x31\n\x08log_info\x18\t \x03(\x0b\x32\x16.flyteidl.core.TaskLogR\x07logInfo\x12:\n\x0blog_context\x18\n \x01(\x0b\x32\x19.flyteidl.core.LogContextR\nlogContext\x12\x18\n\x07\x63luster\x18\x0b \x01(\tR\x07\x63luster\x12=\n\x07outputs\x18\x0c \x01(\x0b\x32#.cloudidl.workflow.OutputReferencesR\x07outputsB\x0b\n\t_end_timeB\r\n\x0b_error_info\"P\n\x0cNamedLiteral\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.flyteidl.core.LiteralR\x05value\"1\n\x10OutputReferences\x12\x1d\n\noutput_uri\x18\x01 \x01(\tR\toutputUri\"E\n\x06Inputs\x12;\n\x08literals\x18\x01 \x03(\x0b\x32\x1f.cloudidl.workflow.NamedLiteralR\x08literals\"F\n\x07Outputs\x12;\n\x08literals\x18\x01 \x03(\x0b\x32\x1f.cloudidl.workflow.NamedLiteralR\x08literals*\xcb\x01\n\x05Phase\x12\x15\n\x11PHASE_UNSPECIFIED\x10\x00\x12\x10\n\x0cPHASE_QUEUED\x10\x01\x12\x1f\n\x1bPHASE_WAITING_FOR_RESOURCES\x10\x02\x12\x16\n\x12PHASE_INITIALIZING\x10\x03\x12\x11\n\rPHASE_RUNNING\x10\x04\x12\x13\n\x0fPHASE_SUCCEEDED\x10\x05\x12\x10\n\x0cPHASE_FAILED\x10\x06\x12\x11\n\rPHASE_ABORTED\x10\x07\x12\x13\n\x0fPHASE_TIMED_OUT\x10\x08\x42\xbf\x01\n\x15\x63om.cloudidl.workflowB\x12RunDefinitionProtoH\x02P\x01Z+github.com/unionai/cloud/gen/pb-go/workflow\xa2\x02\x03\x43WX\xaa\x02\x11\x43loudidl.Workflow\xca\x02\x11\x43loudidl\\Workflow\xe2\x02\x1d\x43loudidl\\Workflow\\GPBMetadata\xea\x02\x12\x43loudidl::Workflowb\x06proto3')
|
|
25
|
+
|
|
26
|
+
_globals = globals()
|
|
27
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
28
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'workflow.run_definition_pb2', _globals)
|
|
29
|
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
|
30
|
+
DESCRIPTOR._options = None
|
|
31
|
+
DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\022RunDefinitionProtoH\002P\001Z+github.com/unionai/cloud/gen/pb-go/workflow\242\002\003CWX\252\002\021Cloudidl.Workflow\312\002\021Cloudidl\\Workflow\342\002\035Cloudidl\\Workflow\\GPBMetadata\352\002\022Cloudidl::Workflow'
|
|
32
|
+
_RUNIDENTIFIER.fields_by_name['org']._options = None
|
|
33
|
+
_RUNIDENTIFIER.fields_by_name['org']._serialized_options = b'\372B\004r\002\020\001'
|
|
34
|
+
_RUNIDENTIFIER.fields_by_name['project']._options = None
|
|
35
|
+
_RUNIDENTIFIER.fields_by_name['project']._serialized_options = b'\372B\004r\002\020\001'
|
|
36
|
+
_RUNIDENTIFIER.fields_by_name['domain']._options = None
|
|
37
|
+
_RUNIDENTIFIER.fields_by_name['domain']._serialized_options = b'\372B\004r\002\020\001'
|
|
38
|
+
_RUNIDENTIFIER.fields_by_name['name']._options = None
|
|
39
|
+
_RUNIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\006r\004\020\001\030\036'
|
|
40
|
+
_LABELS_VALUESENTRY._options = None
|
|
41
|
+
_LABELS_VALUESENTRY._serialized_options = b'8\001'
|
|
42
|
+
_ANNOTATIONS_VALUESENTRY._options = None
|
|
43
|
+
_ANNOTATIONS_VALUESENTRY._serialized_options = b'8\001'
|
|
44
|
+
_ACTIONIDENTIFIER.fields_by_name['run']._options = None
|
|
45
|
+
_ACTIONIDENTIFIER.fields_by_name['run']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
46
|
+
_ACTIONIDENTIFIER.fields_by_name['name']._options = None
|
|
47
|
+
_ACTIONIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\006r\004\020\001\030\036'
|
|
48
|
+
_CONDITIONACTIONMETADATA.oneofs_by_name['scope']._options = None
|
|
49
|
+
_CONDITIONACTIONMETADATA.oneofs_by_name['scope']._serialized_options = b'\370B\001'
|
|
50
|
+
_CONDITIONACTIONMETADATA.fields_by_name['run_id']._options = None
|
|
51
|
+
_CONDITIONACTIONMETADATA.fields_by_name['run_id']._serialized_options = b'\372B\004r\002\020\001'
|
|
52
|
+
_CONDITIONACTIONMETADATA.fields_by_name['action_id']._options = None
|
|
53
|
+
_CONDITIONACTIONMETADATA.fields_by_name['action_id']._serialized_options = b'\372B\004r\002\020\001'
|
|
54
|
+
_ACTIONMETADATA.fields_by_name['task']._options = None
|
|
55
|
+
_ACTIONMETADATA.fields_by_name['task']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
56
|
+
_ACTIONMETADATA.fields_by_name['trace']._options = None
|
|
57
|
+
_ACTIONMETADATA.fields_by_name['trace']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
58
|
+
_ACTIONMETADATA.fields_by_name['condition']._options = None
|
|
59
|
+
_ACTIONMETADATA.fields_by_name['condition']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
60
|
+
_ACTIONSTATUS.fields_by_name['attempts']._options = None
|
|
61
|
+
_ACTIONSTATUS.fields_by_name['attempts']._serialized_options = b'\372B\004*\002 \000'
|
|
62
|
+
_ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY._options = None
|
|
63
|
+
_ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY._serialized_options = b'8\001'
|
|
64
|
+
_ACTIONATTEMPT.fields_by_name['attempt']._options = None
|
|
65
|
+
_ACTIONATTEMPT.fields_by_name['attempt']._serialized_options = b'\372B\004*\002 \000'
|
|
66
|
+
_ACTIONEVENT.fields_by_name['id']._options = None
|
|
67
|
+
_ACTIONEVENT.fields_by_name['id']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
68
|
+
_ACTIONEVENT.fields_by_name['attempt']._options = None
|
|
69
|
+
_ACTIONEVENT.fields_by_name['attempt']._serialized_options = b'\372B\004*\002 \000'
|
|
70
|
+
_globals['_PHASE']._serialized_start=4822
|
|
71
|
+
_globals['_PHASE']._serialized_end=5025
|
|
72
|
+
_globals['_RUNIDENTIFIER']._serialized_start=284
|
|
73
|
+
_globals['_RUNIDENTIFIER']._serialized_end=425
|
|
74
|
+
_globals['_LABELS']._serialized_start=428
|
|
75
|
+
_globals['_LABELS']._serialized_end=558
|
|
76
|
+
_globals['_LABELS_VALUESENTRY']._serialized_start=501
|
|
77
|
+
_globals['_LABELS_VALUESENTRY']._serialized_end=558
|
|
78
|
+
_globals['_ANNOTATIONS']._serialized_start=561
|
|
79
|
+
_globals['_ANNOTATIONS']._serialized_end=701
|
|
80
|
+
_globals['_ANNOTATIONS_VALUESENTRY']._serialized_start=501
|
|
81
|
+
_globals['_ANNOTATIONS_VALUESENTRY']._serialized_end=558
|
|
82
|
+
_globals['_ENVS']._serialized_start=703
|
|
83
|
+
_globals['_ENVS']._serialized_end=762
|
|
84
|
+
_globals['_RUNSPEC']._serialized_start=765
|
|
85
|
+
_globals['_RUNSPEC']._serialized_end=1043
|
|
86
|
+
_globals['_RUN']._serialized_start=1045
|
|
87
|
+
_globals['_RUN']._serialized_end=1101
|
|
88
|
+
_globals['_RUNDETAILS']._serialized_start=1103
|
|
89
|
+
_globals['_RUNDETAILS']._serialized_end=1228
|
|
90
|
+
_globals['_ACTIONIDENTIFIER']._serialized_start=1230
|
|
91
|
+
_globals['_ACTIONIDENTIFIER']._serialized_end=1341
|
|
92
|
+
_globals['_TASKACTIONMETADATA']._serialized_start=1343
|
|
93
|
+
_globals['_TASKACTIONMETADATA']._serialized_end=1443
|
|
94
|
+
_globals['_TRACEACTIONMETADATA']._serialized_start=1445
|
|
95
|
+
_globals['_TRACEACTIONMETADATA']._serialized_end=1486
|
|
96
|
+
_globals['_CONDITIONACTIONMETADATA']._serialized_start=1489
|
|
97
|
+
_globals['_CONDITIONACTIONMETADATA']._serialized_end=1648
|
|
98
|
+
_globals['_ACTIONMETADATA']._serialized_start=1651
|
|
99
|
+
_globals['_ACTIONMETADATA']._serialized_end=2020
|
|
100
|
+
_globals['_ACTIONSTATUS']._serialized_start=2023
|
|
101
|
+
_globals['_ACTIONSTATUS']._serialized_end=2254
|
|
102
|
+
_globals['_ACTION']._serialized_start=2257
|
|
103
|
+
_globals['_ACTION']._serialized_end=2438
|
|
104
|
+
_globals['_ENRICHEDACTION']._serialized_start=2441
|
|
105
|
+
_globals['_ENRICHEDACTION']._serialized_end=2727
|
|
106
|
+
_globals['_ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY']._serialized_start=2657
|
|
107
|
+
_globals['_ENRICHEDACTION_CHILDRENPHASECOUNTSENTRY']._serialized_end=2727
|
|
108
|
+
_globals['_ERRORINFO']._serialized_start=2730
|
|
109
|
+
_globals['_ERRORINFO']._serialized_end=2884
|
|
110
|
+
_globals['_ERRORINFO_KIND']._serialized_start=2824
|
|
111
|
+
_globals['_ERRORINFO_KIND']._serialized_end=2884
|
|
112
|
+
_globals['_ABORTINFO']._serialized_start=2886
|
|
113
|
+
_globals['_ABORTINFO']._serialized_end=2987
|
|
114
|
+
_globals['_ACTIONDETAILS']._serialized_start=2990
|
|
115
|
+
_globals['_ACTIONDETAILS']._serialized_end=3451
|
|
116
|
+
_globals['_ACTIONATTEMPT']._serialized_start=3454
|
|
117
|
+
_globals['_ACTIONATTEMPT']._serialized_end=3879
|
|
118
|
+
_globals['_ACTIONEVENT']._serialized_start=3882
|
|
119
|
+
_globals['_ACTIONEVENT']._serialized_end=4543
|
|
120
|
+
_globals['_NAMEDLITERAL']._serialized_start=4545
|
|
121
|
+
_globals['_NAMEDLITERAL']._serialized_end=4625
|
|
122
|
+
_globals['_OUTPUTREFERENCES']._serialized_start=4627
|
|
123
|
+
_globals['_OUTPUTREFERENCES']._serialized_end=4676
|
|
124
|
+
_globals['_INPUTS']._serialized_start=4678
|
|
125
|
+
_globals['_INPUTS']._serialized_end=4747
|
|
126
|
+
_globals['_OUTPUTS']._serialized_start=4749
|
|
127
|
+
_globals['_OUTPUTS']._serialized_end=4819
|
|
128
|
+
# @@protoc_insertion_point(module_scope)
|
|
@@ -0,0 +1,310 @@
|
|
|
1
|
+
from flyte._protos.common import identifier_pb2 as _identifier_pb2
|
|
2
|
+
from flyte._protos.common import identity_pb2 as _identity_pb2
|
|
3
|
+
from flyteidl.core import execution_pb2 as _execution_pb2
|
|
4
|
+
from flyteidl.core import literals_pb2 as _literals_pb2
|
|
5
|
+
from google.protobuf import timestamp_pb2 as _timestamp_pb2
|
|
6
|
+
from google.protobuf import wrappers_pb2 as _wrappers_pb2
|
|
7
|
+
from flyte._protos.validate.validate import validate_pb2 as _validate_pb2
|
|
8
|
+
from flyte._protos.workflow import task_definition_pb2 as _task_definition_pb2
|
|
9
|
+
from google.protobuf.internal import containers as _containers
|
|
10
|
+
from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
|
|
11
|
+
from google.protobuf import descriptor as _descriptor
|
|
12
|
+
from google.protobuf import message as _message
|
|
13
|
+
from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
|
|
14
|
+
|
|
15
|
+
DESCRIPTOR: _descriptor.FileDescriptor
|
|
16
|
+
|
|
17
|
+
class Phase(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
|
|
18
|
+
__slots__ = []
|
|
19
|
+
PHASE_UNSPECIFIED: _ClassVar[Phase]
|
|
20
|
+
PHASE_QUEUED: _ClassVar[Phase]
|
|
21
|
+
PHASE_WAITING_FOR_RESOURCES: _ClassVar[Phase]
|
|
22
|
+
PHASE_INITIALIZING: _ClassVar[Phase]
|
|
23
|
+
PHASE_RUNNING: _ClassVar[Phase]
|
|
24
|
+
PHASE_SUCCEEDED: _ClassVar[Phase]
|
|
25
|
+
PHASE_FAILED: _ClassVar[Phase]
|
|
26
|
+
PHASE_ABORTED: _ClassVar[Phase]
|
|
27
|
+
PHASE_TIMED_OUT: _ClassVar[Phase]
|
|
28
|
+
PHASE_UNSPECIFIED: Phase
|
|
29
|
+
PHASE_QUEUED: Phase
|
|
30
|
+
PHASE_WAITING_FOR_RESOURCES: Phase
|
|
31
|
+
PHASE_INITIALIZING: Phase
|
|
32
|
+
PHASE_RUNNING: Phase
|
|
33
|
+
PHASE_SUCCEEDED: Phase
|
|
34
|
+
PHASE_FAILED: Phase
|
|
35
|
+
PHASE_ABORTED: Phase
|
|
36
|
+
PHASE_TIMED_OUT: Phase
|
|
37
|
+
|
|
38
|
+
class RunIdentifier(_message.Message):
|
|
39
|
+
__slots__ = ["org", "project", "domain", "name"]
|
|
40
|
+
ORG_FIELD_NUMBER: _ClassVar[int]
|
|
41
|
+
PROJECT_FIELD_NUMBER: _ClassVar[int]
|
|
42
|
+
DOMAIN_FIELD_NUMBER: _ClassVar[int]
|
|
43
|
+
NAME_FIELD_NUMBER: _ClassVar[int]
|
|
44
|
+
org: str
|
|
45
|
+
project: str
|
|
46
|
+
domain: str
|
|
47
|
+
name: str
|
|
48
|
+
def __init__(self, org: _Optional[str] = ..., project: _Optional[str] = ..., domain: _Optional[str] = ..., name: _Optional[str] = ...) -> None: ...
|
|
49
|
+
|
|
50
|
+
class Labels(_message.Message):
|
|
51
|
+
__slots__ = ["values"]
|
|
52
|
+
class ValuesEntry(_message.Message):
|
|
53
|
+
__slots__ = ["key", "value"]
|
|
54
|
+
KEY_FIELD_NUMBER: _ClassVar[int]
|
|
55
|
+
VALUE_FIELD_NUMBER: _ClassVar[int]
|
|
56
|
+
key: str
|
|
57
|
+
value: str
|
|
58
|
+
def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
|
|
59
|
+
VALUES_FIELD_NUMBER: _ClassVar[int]
|
|
60
|
+
values: _containers.ScalarMap[str, str]
|
|
61
|
+
def __init__(self, values: _Optional[_Mapping[str, str]] = ...) -> None: ...
|
|
62
|
+
|
|
63
|
+
class Annotations(_message.Message):
|
|
64
|
+
__slots__ = ["values"]
|
|
65
|
+
class ValuesEntry(_message.Message):
|
|
66
|
+
__slots__ = ["key", "value"]
|
|
67
|
+
KEY_FIELD_NUMBER: _ClassVar[int]
|
|
68
|
+
VALUE_FIELD_NUMBER: _ClassVar[int]
|
|
69
|
+
key: str
|
|
70
|
+
value: str
|
|
71
|
+
def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
|
|
72
|
+
VALUES_FIELD_NUMBER: _ClassVar[int]
|
|
73
|
+
values: _containers.ScalarMap[str, str]
|
|
74
|
+
def __init__(self, values: _Optional[_Mapping[str, str]] = ...) -> None: ...
|
|
75
|
+
|
|
76
|
+
class Envs(_message.Message):
|
|
77
|
+
__slots__ = ["values"]
|
|
78
|
+
VALUES_FIELD_NUMBER: _ClassVar[int]
|
|
79
|
+
values: _containers.RepeatedCompositeFieldContainer[_literals_pb2.KeyValuePair]
|
|
80
|
+
def __init__(self, values: _Optional[_Iterable[_Union[_literals_pb2.KeyValuePair, _Mapping]]] = ...) -> None: ...
|
|
81
|
+
|
|
82
|
+
class RunSpec(_message.Message):
|
|
83
|
+
__slots__ = ["labels", "annotations", "envs", "interruptible", "overwrite_cache"]
|
|
84
|
+
LABELS_FIELD_NUMBER: _ClassVar[int]
|
|
85
|
+
ANNOTATIONS_FIELD_NUMBER: _ClassVar[int]
|
|
86
|
+
ENVS_FIELD_NUMBER: _ClassVar[int]
|
|
87
|
+
INTERRUPTIBLE_FIELD_NUMBER: _ClassVar[int]
|
|
88
|
+
OVERWRITE_CACHE_FIELD_NUMBER: _ClassVar[int]
|
|
89
|
+
labels: Labels
|
|
90
|
+
annotations: Annotations
|
|
91
|
+
envs: Envs
|
|
92
|
+
interruptible: _wrappers_pb2.BoolValue
|
|
93
|
+
overwrite_cache: bool
|
|
94
|
+
def __init__(self, labels: _Optional[_Union[Labels, _Mapping]] = ..., annotations: _Optional[_Union[Annotations, _Mapping]] = ..., envs: _Optional[_Union[Envs, _Mapping]] = ..., interruptible: _Optional[_Union[_wrappers_pb2.BoolValue, _Mapping]] = ..., overwrite_cache: bool = ...) -> None: ...
|
|
95
|
+
|
|
96
|
+
class Run(_message.Message):
|
|
97
|
+
__slots__ = ["action"]
|
|
98
|
+
ACTION_FIELD_NUMBER: _ClassVar[int]
|
|
99
|
+
action: Action
|
|
100
|
+
def __init__(self, action: _Optional[_Union[Action, _Mapping]] = ...) -> None: ...
|
|
101
|
+
|
|
102
|
+
class RunDetails(_message.Message):
|
|
103
|
+
__slots__ = ["run_spec", "action"]
|
|
104
|
+
RUN_SPEC_FIELD_NUMBER: _ClassVar[int]
|
|
105
|
+
ACTION_FIELD_NUMBER: _ClassVar[int]
|
|
106
|
+
run_spec: RunSpec
|
|
107
|
+
action: ActionDetails
|
|
108
|
+
def __init__(self, run_spec: _Optional[_Union[RunSpec, _Mapping]] = ..., action: _Optional[_Union[ActionDetails, _Mapping]] = ...) -> None: ...
|
|
109
|
+
|
|
110
|
+
class ActionIdentifier(_message.Message):
|
|
111
|
+
__slots__ = ["run", "name"]
|
|
112
|
+
RUN_FIELD_NUMBER: _ClassVar[int]
|
|
113
|
+
NAME_FIELD_NUMBER: _ClassVar[int]
|
|
114
|
+
run: RunIdentifier
|
|
115
|
+
name: str
|
|
116
|
+
def __init__(self, run: _Optional[_Union[RunIdentifier, _Mapping]] = ..., name: _Optional[str] = ...) -> None: ...
|
|
117
|
+
|
|
118
|
+
class TaskActionMetadata(_message.Message):
|
|
119
|
+
__slots__ = ["id", "task_type"]
|
|
120
|
+
ID_FIELD_NUMBER: _ClassVar[int]
|
|
121
|
+
TASK_TYPE_FIELD_NUMBER: _ClassVar[int]
|
|
122
|
+
id: _task_definition_pb2.TaskIdentifier
|
|
123
|
+
task_type: str
|
|
124
|
+
def __init__(self, id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ..., task_type: _Optional[str] = ...) -> None: ...
|
|
125
|
+
|
|
126
|
+
class TraceActionMetadata(_message.Message):
|
|
127
|
+
__slots__ = ["name"]
|
|
128
|
+
NAME_FIELD_NUMBER: _ClassVar[int]
|
|
129
|
+
name: str
|
|
130
|
+
def __init__(self, name: _Optional[str] = ...) -> None: ...
|
|
131
|
+
|
|
132
|
+
class ConditionActionMetadata(_message.Message):
|
|
133
|
+
__slots__ = ["name", "run_id", "action_id"]
|
|
134
|
+
NAME_FIELD_NUMBER: _ClassVar[int]
|
|
135
|
+
RUN_ID_FIELD_NUMBER: _ClassVar[int]
|
|
136
|
+
ACTION_ID_FIELD_NUMBER: _ClassVar[int]
|
|
137
|
+
GLOBAL_FIELD_NUMBER: _ClassVar[int]
|
|
138
|
+
name: str
|
|
139
|
+
run_id: str
|
|
140
|
+
action_id: str
|
|
141
|
+
def __init__(self, name: _Optional[str] = ..., run_id: _Optional[str] = ..., action_id: _Optional[str] = ..., **kwargs) -> None: ...
|
|
142
|
+
|
|
143
|
+
class ActionMetadata(_message.Message):
|
|
144
|
+
__slots__ = ["parent", "group", "executed_by", "task", "trace", "condition"]
|
|
145
|
+
PARENT_FIELD_NUMBER: _ClassVar[int]
|
|
146
|
+
GROUP_FIELD_NUMBER: _ClassVar[int]
|
|
147
|
+
EXECUTED_BY_FIELD_NUMBER: _ClassVar[int]
|
|
148
|
+
TASK_FIELD_NUMBER: _ClassVar[int]
|
|
149
|
+
TRACE_FIELD_NUMBER: _ClassVar[int]
|
|
150
|
+
CONDITION_FIELD_NUMBER: _ClassVar[int]
|
|
151
|
+
parent: str
|
|
152
|
+
group: str
|
|
153
|
+
executed_by: _identity_pb2.EnrichedIdentity
|
|
154
|
+
task: TaskActionMetadata
|
|
155
|
+
trace: TraceActionMetadata
|
|
156
|
+
condition: ConditionActionMetadata
|
|
157
|
+
def __init__(self, parent: _Optional[str] = ..., group: _Optional[str] = ..., executed_by: _Optional[_Union[_identity_pb2.EnrichedIdentity, _Mapping]] = ..., task: _Optional[_Union[TaskActionMetadata, _Mapping]] = ..., trace: _Optional[_Union[TraceActionMetadata, _Mapping]] = ..., condition: _Optional[_Union[ConditionActionMetadata, _Mapping]] = ...) -> None: ...
|
|
158
|
+
|
|
159
|
+
class ActionStatus(_message.Message):
|
|
160
|
+
__slots__ = ["phase", "start_time", "end_time", "attempts"]
|
|
161
|
+
PHASE_FIELD_NUMBER: _ClassVar[int]
|
|
162
|
+
START_TIME_FIELD_NUMBER: _ClassVar[int]
|
|
163
|
+
END_TIME_FIELD_NUMBER: _ClassVar[int]
|
|
164
|
+
ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
|
|
165
|
+
phase: Phase
|
|
166
|
+
start_time: _timestamp_pb2.Timestamp
|
|
167
|
+
end_time: _timestamp_pb2.Timestamp
|
|
168
|
+
attempts: int
|
|
169
|
+
def __init__(self, phase: _Optional[_Union[Phase, str]] = ..., start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., attempts: _Optional[int] = ...) -> None: ...
|
|
170
|
+
|
|
171
|
+
class Action(_message.Message):
|
|
172
|
+
__slots__ = ["id", "metadata", "status"]
|
|
173
|
+
ID_FIELD_NUMBER: _ClassVar[int]
|
|
174
|
+
METADATA_FIELD_NUMBER: _ClassVar[int]
|
|
175
|
+
STATUS_FIELD_NUMBER: _ClassVar[int]
|
|
176
|
+
id: ActionIdentifier
|
|
177
|
+
metadata: ActionMetadata
|
|
178
|
+
status: ActionStatus
|
|
179
|
+
def __init__(self, id: _Optional[_Union[ActionIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[ActionMetadata, _Mapping]] = ..., status: _Optional[_Union[ActionStatus, _Mapping]] = ...) -> None: ...
|
|
180
|
+
|
|
181
|
+
class EnrichedAction(_message.Message):
|
|
182
|
+
__slots__ = ["action", "meets_filter", "children_phase_counts"]
|
|
183
|
+
class ChildrenPhaseCountsEntry(_message.Message):
|
|
184
|
+
__slots__ = ["key", "value"]
|
|
185
|
+
KEY_FIELD_NUMBER: _ClassVar[int]
|
|
186
|
+
VALUE_FIELD_NUMBER: _ClassVar[int]
|
|
187
|
+
key: int
|
|
188
|
+
value: int
|
|
189
|
+
def __init__(self, key: _Optional[int] = ..., value: _Optional[int] = ...) -> None: ...
|
|
190
|
+
ACTION_FIELD_NUMBER: _ClassVar[int]
|
|
191
|
+
MEETS_FILTER_FIELD_NUMBER: _ClassVar[int]
|
|
192
|
+
CHILDREN_PHASE_COUNTS_FIELD_NUMBER: _ClassVar[int]
|
|
193
|
+
action: Action
|
|
194
|
+
meets_filter: bool
|
|
195
|
+
children_phase_counts: _containers.ScalarMap[int, int]
|
|
196
|
+
def __init__(self, action: _Optional[_Union[Action, _Mapping]] = ..., meets_filter: bool = ..., children_phase_counts: _Optional[_Mapping[int, int]] = ...) -> None: ...
|
|
197
|
+
|
|
198
|
+
class ErrorInfo(_message.Message):
|
|
199
|
+
__slots__ = ["message", "kind"]
|
|
200
|
+
class Kind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
|
|
201
|
+
__slots__ = []
|
|
202
|
+
KIND_UNSPECIFIED: _ClassVar[ErrorInfo.Kind]
|
|
203
|
+
KIND_USER: _ClassVar[ErrorInfo.Kind]
|
|
204
|
+
KIND_SYSTEM: _ClassVar[ErrorInfo.Kind]
|
|
205
|
+
KIND_UNSPECIFIED: ErrorInfo.Kind
|
|
206
|
+
KIND_USER: ErrorInfo.Kind
|
|
207
|
+
KIND_SYSTEM: ErrorInfo.Kind
|
|
208
|
+
MESSAGE_FIELD_NUMBER: _ClassVar[int]
|
|
209
|
+
KIND_FIELD_NUMBER: _ClassVar[int]
|
|
210
|
+
message: str
|
|
211
|
+
kind: ErrorInfo.Kind
|
|
212
|
+
def __init__(self, message: _Optional[str] = ..., kind: _Optional[_Union[ErrorInfo.Kind, str]] = ...) -> None: ...
|
|
213
|
+
|
|
214
|
+
class AbortInfo(_message.Message):
|
|
215
|
+
__slots__ = ["reason", "aborted_by"]
|
|
216
|
+
REASON_FIELD_NUMBER: _ClassVar[int]
|
|
217
|
+
ABORTED_BY_FIELD_NUMBER: _ClassVar[int]
|
|
218
|
+
reason: str
|
|
219
|
+
aborted_by: _identity_pb2.EnrichedIdentity
|
|
220
|
+
def __init__(self, reason: _Optional[str] = ..., aborted_by: _Optional[_Union[_identity_pb2.EnrichedIdentity, _Mapping]] = ...) -> None: ...
|
|
221
|
+
|
|
222
|
+
class ActionDetails(_message.Message):
|
|
223
|
+
__slots__ = ["id", "metadata", "status", "error_info", "abort_info", "resolved_task_spec", "attempts"]
|
|
224
|
+
ID_FIELD_NUMBER: _ClassVar[int]
|
|
225
|
+
METADATA_FIELD_NUMBER: _ClassVar[int]
|
|
226
|
+
STATUS_FIELD_NUMBER: _ClassVar[int]
|
|
227
|
+
ERROR_INFO_FIELD_NUMBER: _ClassVar[int]
|
|
228
|
+
ABORT_INFO_FIELD_NUMBER: _ClassVar[int]
|
|
229
|
+
RESOLVED_TASK_SPEC_FIELD_NUMBER: _ClassVar[int]
|
|
230
|
+
ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
|
|
231
|
+
id: ActionIdentifier
|
|
232
|
+
metadata: ActionMetadata
|
|
233
|
+
status: ActionStatus
|
|
234
|
+
error_info: ErrorInfo
|
|
235
|
+
abort_info: AbortInfo
|
|
236
|
+
resolved_task_spec: _task_definition_pb2.TaskSpec
|
|
237
|
+
attempts: _containers.RepeatedCompositeFieldContainer[ActionAttempt]
|
|
238
|
+
def __init__(self, id: _Optional[_Union[ActionIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[ActionMetadata, _Mapping]] = ..., status: _Optional[_Union[ActionStatus, _Mapping]] = ..., error_info: _Optional[_Union[ErrorInfo, _Mapping]] = ..., abort_info: _Optional[_Union[AbortInfo, _Mapping]] = ..., resolved_task_spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ..., attempts: _Optional[_Iterable[_Union[ActionAttempt, _Mapping]]] = ...) -> None: ...
|
|
239
|
+
|
|
240
|
+
class ActionAttempt(_message.Message):
|
|
241
|
+
__slots__ = ["phase", "start_time", "end_time", "error_info", "attempt", "log_info", "outputs"]
|
|
242
|
+
PHASE_FIELD_NUMBER: _ClassVar[int]
|
|
243
|
+
START_TIME_FIELD_NUMBER: _ClassVar[int]
|
|
244
|
+
END_TIME_FIELD_NUMBER: _ClassVar[int]
|
|
245
|
+
ERROR_INFO_FIELD_NUMBER: _ClassVar[int]
|
|
246
|
+
ATTEMPT_FIELD_NUMBER: _ClassVar[int]
|
|
247
|
+
LOG_INFO_FIELD_NUMBER: _ClassVar[int]
|
|
248
|
+
OUTPUTS_FIELD_NUMBER: _ClassVar[int]
|
|
249
|
+
phase: Phase
|
|
250
|
+
start_time: _timestamp_pb2.Timestamp
|
|
251
|
+
end_time: _timestamp_pb2.Timestamp
|
|
252
|
+
error_info: ErrorInfo
|
|
253
|
+
attempt: int
|
|
254
|
+
log_info: _containers.RepeatedCompositeFieldContainer[_execution_pb2.TaskLog]
|
|
255
|
+
outputs: OutputReferences
|
|
256
|
+
def __init__(self, phase: _Optional[_Union[Phase, str]] = ..., start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., error_info: _Optional[_Union[ErrorInfo, _Mapping]] = ..., attempt: _Optional[int] = ..., log_info: _Optional[_Iterable[_Union[_execution_pb2.TaskLog, _Mapping]]] = ..., outputs: _Optional[_Union[OutputReferences, _Mapping]] = ...) -> None: ...
|
|
257
|
+
|
|
258
|
+
class ActionEvent(_message.Message):
|
|
259
|
+
__slots__ = ["id", "attempt", "phase", "version", "start_time", "updated_time", "end_time", "error_info", "log_info", "log_context", "cluster", "outputs"]
|
|
260
|
+
ID_FIELD_NUMBER: _ClassVar[int]
|
|
261
|
+
ATTEMPT_FIELD_NUMBER: _ClassVar[int]
|
|
262
|
+
PHASE_FIELD_NUMBER: _ClassVar[int]
|
|
263
|
+
VERSION_FIELD_NUMBER: _ClassVar[int]
|
|
264
|
+
START_TIME_FIELD_NUMBER: _ClassVar[int]
|
|
265
|
+
UPDATED_TIME_FIELD_NUMBER: _ClassVar[int]
|
|
266
|
+
END_TIME_FIELD_NUMBER: _ClassVar[int]
|
|
267
|
+
ERROR_INFO_FIELD_NUMBER: _ClassVar[int]
|
|
268
|
+
LOG_INFO_FIELD_NUMBER: _ClassVar[int]
|
|
269
|
+
LOG_CONTEXT_FIELD_NUMBER: _ClassVar[int]
|
|
270
|
+
CLUSTER_FIELD_NUMBER: _ClassVar[int]
|
|
271
|
+
OUTPUTS_FIELD_NUMBER: _ClassVar[int]
|
|
272
|
+
id: ActionIdentifier
|
|
273
|
+
attempt: int
|
|
274
|
+
phase: Phase
|
|
275
|
+
version: int
|
|
276
|
+
start_time: _timestamp_pb2.Timestamp
|
|
277
|
+
updated_time: _timestamp_pb2.Timestamp
|
|
278
|
+
end_time: _timestamp_pb2.Timestamp
|
|
279
|
+
error_info: ErrorInfo
|
|
280
|
+
log_info: _containers.RepeatedCompositeFieldContainer[_execution_pb2.TaskLog]
|
|
281
|
+
log_context: _execution_pb2.LogContext
|
|
282
|
+
cluster: str
|
|
283
|
+
outputs: OutputReferences
|
|
284
|
+
def __init__(self, id: _Optional[_Union[ActionIdentifier, _Mapping]] = ..., attempt: _Optional[int] = ..., phase: _Optional[_Union[Phase, str]] = ..., version: _Optional[int] = ..., start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., updated_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., error_info: _Optional[_Union[ErrorInfo, _Mapping]] = ..., log_info: _Optional[_Iterable[_Union[_execution_pb2.TaskLog, _Mapping]]] = ..., log_context: _Optional[_Union[_execution_pb2.LogContext, _Mapping]] = ..., cluster: _Optional[str] = ..., outputs: _Optional[_Union[OutputReferences, _Mapping]] = ...) -> None: ...
|
|
285
|
+
|
|
286
|
+
class NamedLiteral(_message.Message):
|
|
287
|
+
__slots__ = ["name", "value"]
|
|
288
|
+
NAME_FIELD_NUMBER: _ClassVar[int]
|
|
289
|
+
VALUE_FIELD_NUMBER: _ClassVar[int]
|
|
290
|
+
name: str
|
|
291
|
+
value: _literals_pb2.Literal
|
|
292
|
+
def __init__(self, name: _Optional[str] = ..., value: _Optional[_Union[_literals_pb2.Literal, _Mapping]] = ...) -> None: ...
|
|
293
|
+
|
|
294
|
+
class OutputReferences(_message.Message):
|
|
295
|
+
__slots__ = ["output_uri"]
|
|
296
|
+
OUTPUT_URI_FIELD_NUMBER: _ClassVar[int]
|
|
297
|
+
output_uri: str
|
|
298
|
+
def __init__(self, output_uri: _Optional[str] = ...) -> None: ...
|
|
299
|
+
|
|
300
|
+
class Inputs(_message.Message):
|
|
301
|
+
__slots__ = ["literals"]
|
|
302
|
+
LITERALS_FIELD_NUMBER: _ClassVar[int]
|
|
303
|
+
literals: _containers.RepeatedCompositeFieldContainer[NamedLiteral]
|
|
304
|
+
def __init__(self, literals: _Optional[_Iterable[_Union[NamedLiteral, _Mapping]]] = ...) -> None: ...
|
|
305
|
+
|
|
306
|
+
class Outputs(_message.Message):
|
|
307
|
+
__slots__ = ["literals"]
|
|
308
|
+
LITERALS_FIELD_NUMBER: _ClassVar[int]
|
|
309
|
+
literals: _containers.RepeatedCompositeFieldContainer[NamedLiteral]
|
|
310
|
+
def __init__(self, literals: _Optional[_Iterable[_Union[NamedLiteral, _Mapping]]] = ...) -> None: ...
|