flyte 2.0.0b23__py3-none-any.whl → 2.0.0b25__py3-none-any.whl

flyte/_protos/workflow/run_service_pb2_grpc.py

@@ -1,480 +0,0 @@
-# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
-"""Client and server classes corresponding to protobuf-defined services."""
-import grpc
-
-from flyte._protos.workflow import run_service_pb2 as workflow_dot_run__service__pb2
-
-
-class RunServiceStub(object):
-    """RunService provides an interface for managing runs.
-    """
-
-    def __init__(self, channel):
-        """Constructor.
-
-        Args:
-            channel: A grpc.Channel.
-        """
-        self.CreateRun = channel.unary_unary(
-                '/cloudidl.workflow.RunService/CreateRun',
-                request_serializer=workflow_dot_run__service__pb2.CreateRunRequest.SerializeToString,
-                response_deserializer=workflow_dot_run__service__pb2.CreateRunResponse.FromString,
-                )
-        self.AbortRun = channel.unary_unary(
-                '/cloudidl.workflow.RunService/AbortRun',
-                request_serializer=workflow_dot_run__service__pb2.AbortRunRequest.SerializeToString,
-                response_deserializer=workflow_dot_run__service__pb2.AbortRunResponse.FromString,
-                )
-        self.GetRunDetails = channel.unary_unary(
-                '/cloudidl.workflow.RunService/GetRunDetails',
-                request_serializer=workflow_dot_run__service__pb2.GetRunDetailsRequest.SerializeToString,
-                response_deserializer=workflow_dot_run__service__pb2.GetRunDetailsResponse.FromString,
-                )
-        self.WatchRunDetails = channel.unary_stream(
-                '/cloudidl.workflow.RunService/WatchRunDetails',
-                request_serializer=workflow_dot_run__service__pb2.WatchRunDetailsRequest.SerializeToString,
-                response_deserializer=workflow_dot_run__service__pb2.WatchRunDetailsResponse.FromString,
-                )
-        self.GetActionDetails = channel.unary_unary(
-                '/cloudidl.workflow.RunService/GetActionDetails',
-                request_serializer=workflow_dot_run__service__pb2.GetActionDetailsRequest.SerializeToString,
-                response_deserializer=workflow_dot_run__service__pb2.GetActionDetailsResponse.FromString,
-                )
-        self.WatchActionDetails = channel.unary_stream(
-                '/cloudidl.workflow.RunService/WatchActionDetails',
-                request_serializer=workflow_dot_run__service__pb2.WatchActionDetailsRequest.SerializeToString,
-                response_deserializer=workflow_dot_run__service__pb2.WatchActionDetailsResponse.FromString,
-                )
-        self.GetActionData = channel.unary_unary(
-                '/cloudidl.workflow.RunService/GetActionData',
-                request_serializer=workflow_dot_run__service__pb2.GetActionDataRequest.SerializeToString,
-                response_deserializer=workflow_dot_run__service__pb2.GetActionDataResponse.FromString,
-                )
-        self.ListRuns = channel.unary_unary(
-                '/cloudidl.workflow.RunService/ListRuns',
-                request_serializer=workflow_dot_run__service__pb2.ListRunsRequest.SerializeToString,
-                response_deserializer=workflow_dot_run__service__pb2.ListRunsResponse.FromString,
-                )
-        self.WatchRuns = channel.unary_stream(
-                '/cloudidl.workflow.RunService/WatchRuns',
-                request_serializer=workflow_dot_run__service__pb2.WatchRunsRequest.SerializeToString,
-                response_deserializer=workflow_dot_run__service__pb2.WatchRunsResponse.FromString,
-                )
-        self.ListActions = channel.unary_unary(
-                '/cloudidl.workflow.RunService/ListActions',
-                request_serializer=workflow_dot_run__service__pb2.ListActionsRequest.SerializeToString,
-                response_deserializer=workflow_dot_run__service__pb2.ListActionsResponse.FromString,
-                )
-        self.WatchActions = channel.unary_stream(
-                '/cloudidl.workflow.RunService/WatchActions',
-                request_serializer=workflow_dot_run__service__pb2.WatchActionsRequest.SerializeToString,
-                response_deserializer=workflow_dot_run__service__pb2.WatchActionsResponse.FromString,
-                )
-        self.WatchClusterEvents = channel.unary_stream(
-                '/cloudidl.workflow.RunService/WatchClusterEvents',
-                request_serializer=workflow_dot_run__service__pb2.WatchClusterEventsRequest.SerializeToString,
-                response_deserializer=workflow_dot_run__service__pb2.WatchClusterEventsResponse.FromString,
-                )
-        self.AbortAction = channel.unary_unary(
-                '/cloudidl.workflow.RunService/AbortAction',
-                request_serializer=workflow_dot_run__service__pb2.AbortActionRequest.SerializeToString,
-                response_deserializer=workflow_dot_run__service__pb2.AbortActionResponse.FromString,
-                )
-
-
-class RunServiceServicer(object):
-    """RunService provides an interface for managing runs.
-    """
-
-    def CreateRun(self, request, context):
-        """Create a new run of the given task.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def AbortRun(self, request, context):
-        """Abort a run.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def GetRunDetails(self, request, context):
-        """Get detailed information about a run.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def WatchRunDetails(self, request, context):
-        """Stream detailed information updates about a run. The call will terminate when the run reaches a terminal phase.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def GetActionDetails(self, request, context):
-        """Get detailed information about an action.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def WatchActionDetails(self, request, context):
-        """Stream detailed information updates about an action. The call will terminate when the action reaches a terminal phase.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def GetActionData(self, request, context):
-        """Get input and output for an action.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def ListRuns(self, request, context):
-        """List runs based on the provided filter criteria.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def WatchRuns(self, request, context):
-        """Stream updates for runs based on the provided filter criteria. Responses may include newly discovered
-        runs or updates to existing ones from the point of invocation.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def ListActions(self, request, context):
-        """List all actions for a given run.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def WatchActions(self, request, context):
-        """Stream updates for actions given a run. Responses may include newly discovered nested runs or updates
-        to  existing ones from the point of invocation.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def WatchClusterEvents(self, request, context):
-        """Stream of k8s cluster events in human readable form
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def AbortAction(self, request, context):
-        """AbortAction aborts a single action that was previously created or is currently being processed by a worker.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-
-def add_RunServiceServicer_to_server(servicer, server):
-    rpc_method_handlers = {
-            'CreateRun': grpc.unary_unary_rpc_method_handler(
-                    servicer.CreateRun,
-                    request_deserializer=workflow_dot_run__service__pb2.CreateRunRequest.FromString,
-                    response_serializer=workflow_dot_run__service__pb2.CreateRunResponse.SerializeToString,
-            ),
-            'AbortRun': grpc.unary_unary_rpc_method_handler(
-                    servicer.AbortRun,
-                    request_deserializer=workflow_dot_run__service__pb2.AbortRunRequest.FromString,
-                    response_serializer=workflow_dot_run__service__pb2.AbortRunResponse.SerializeToString,
-            ),
-            'GetRunDetails': grpc.unary_unary_rpc_method_handler(
-                    servicer.GetRunDetails,
-                    request_deserializer=workflow_dot_run__service__pb2.GetRunDetailsRequest.FromString,
-                    response_serializer=workflow_dot_run__service__pb2.GetRunDetailsResponse.SerializeToString,
-            ),
-            'WatchRunDetails': grpc.unary_stream_rpc_method_handler(
-                    servicer.WatchRunDetails,
-                    request_deserializer=workflow_dot_run__service__pb2.WatchRunDetailsRequest.FromString,
-                    response_serializer=workflow_dot_run__service__pb2.WatchRunDetailsResponse.SerializeToString,
-            ),
-            'GetActionDetails': grpc.unary_unary_rpc_method_handler(
-                    servicer.GetActionDetails,
-                    request_deserializer=workflow_dot_run__service__pb2.GetActionDetailsRequest.FromString,
-                    response_serializer=workflow_dot_run__service__pb2.GetActionDetailsResponse.SerializeToString,
-            ),
-            'WatchActionDetails': grpc.unary_stream_rpc_method_handler(
-                    servicer.WatchActionDetails,
-                    request_deserializer=workflow_dot_run__service__pb2.WatchActionDetailsRequest.FromString,
-                    response_serializer=workflow_dot_run__service__pb2.WatchActionDetailsResponse.SerializeToString,
-            ),
-            'GetActionData': grpc.unary_unary_rpc_method_handler(
-                    servicer.GetActionData,
-                    request_deserializer=workflow_dot_run__service__pb2.GetActionDataRequest.FromString,
-                    response_serializer=workflow_dot_run__service__pb2.GetActionDataResponse.SerializeToString,
-            ),
-            'ListRuns': grpc.unary_unary_rpc_method_handler(
-                    servicer.ListRuns,
-                    request_deserializer=workflow_dot_run__service__pb2.ListRunsRequest.FromString,
-                    response_serializer=workflow_dot_run__service__pb2.ListRunsResponse.SerializeToString,
-            ),
-            'WatchRuns': grpc.unary_stream_rpc_method_handler(
-                    servicer.WatchRuns,
-                    request_deserializer=workflow_dot_run__service__pb2.WatchRunsRequest.FromString,
-                    response_serializer=workflow_dot_run__service__pb2.WatchRunsResponse.SerializeToString,
-            ),
-            'ListActions': grpc.unary_unary_rpc_method_handler(
-                    servicer.ListActions,
-                    request_deserializer=workflow_dot_run__service__pb2.ListActionsRequest.FromString,
-                    response_serializer=workflow_dot_run__service__pb2.ListActionsResponse.SerializeToString,
-            ),
-            'WatchActions': grpc.unary_stream_rpc_method_handler(
-                    servicer.WatchActions,
-                    request_deserializer=workflow_dot_run__service__pb2.WatchActionsRequest.FromString,
-                    response_serializer=workflow_dot_run__service__pb2.WatchActionsResponse.SerializeToString,
-            ),
-            'WatchClusterEvents': grpc.unary_stream_rpc_method_handler(
-                    servicer.WatchClusterEvents,
-                    request_deserializer=workflow_dot_run__service__pb2.WatchClusterEventsRequest.FromString,
-                    response_serializer=workflow_dot_run__service__pb2.WatchClusterEventsResponse.SerializeToString,
-            ),
-            'AbortAction': grpc.unary_unary_rpc_method_handler(
-                    servicer.AbortAction,
-                    request_deserializer=workflow_dot_run__service__pb2.AbortActionRequest.FromString,
-                    response_serializer=workflow_dot_run__service__pb2.AbortActionResponse.SerializeToString,
-            ),
-    }
-    generic_handler = grpc.method_handlers_generic_handler(
-            'cloudidl.workflow.RunService', rpc_method_handlers)
-    server.add_generic_rpc_handlers((generic_handler,))
-
-
- # This class is part of an EXPERIMENTAL API.
-class RunService(object):
-    """RunService provides an interface for managing runs.
-    """
-
-    @staticmethod
-    def CreateRun(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.RunService/CreateRun',
-            workflow_dot_run__service__pb2.CreateRunRequest.SerializeToString,
-            workflow_dot_run__service__pb2.CreateRunResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def AbortRun(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.RunService/AbortRun',
-            workflow_dot_run__service__pb2.AbortRunRequest.SerializeToString,
-            workflow_dot_run__service__pb2.AbortRunResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def GetRunDetails(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.RunService/GetRunDetails',
-            workflow_dot_run__service__pb2.GetRunDetailsRequest.SerializeToString,
-            workflow_dot_run__service__pb2.GetRunDetailsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def WatchRunDetails(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_stream(request, target, '/cloudidl.workflow.RunService/WatchRunDetails',
-            workflow_dot_run__service__pb2.WatchRunDetailsRequest.SerializeToString,
-            workflow_dot_run__service__pb2.WatchRunDetailsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def GetActionDetails(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.RunService/GetActionDetails',
-            workflow_dot_run__service__pb2.GetActionDetailsRequest.SerializeToString,
-            workflow_dot_run__service__pb2.GetActionDetailsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def WatchActionDetails(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_stream(request, target, '/cloudidl.workflow.RunService/WatchActionDetails',
-            workflow_dot_run__service__pb2.WatchActionDetailsRequest.SerializeToString,
-            workflow_dot_run__service__pb2.WatchActionDetailsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def GetActionData(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.RunService/GetActionData',
-            workflow_dot_run__service__pb2.GetActionDataRequest.SerializeToString,
-            workflow_dot_run__service__pb2.GetActionDataResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def ListRuns(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.RunService/ListRuns',
-            workflow_dot_run__service__pb2.ListRunsRequest.SerializeToString,
-            workflow_dot_run__service__pb2.ListRunsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def WatchRuns(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_stream(request, target, '/cloudidl.workflow.RunService/WatchRuns',
-            workflow_dot_run__service__pb2.WatchRunsRequest.SerializeToString,
-            workflow_dot_run__service__pb2.WatchRunsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def ListActions(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.RunService/ListActions',
-            workflow_dot_run__service__pb2.ListActionsRequest.SerializeToString,
-            workflow_dot_run__service__pb2.ListActionsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def WatchActions(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_stream(request, target, '/cloudidl.workflow.RunService/WatchActions',
-            workflow_dot_run__service__pb2.WatchActionsRequest.SerializeToString,
-            workflow_dot_run__service__pb2.WatchActionsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def WatchClusterEvents(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_stream(request, target, '/cloudidl.workflow.RunService/WatchClusterEvents',
-            workflow_dot_run__service__pb2.WatchClusterEventsRequest.SerializeToString,
-            workflow_dot_run__service__pb2.WatchClusterEventsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def AbortAction(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.RunService/AbortAction',
-            workflow_dot_run__service__pb2.AbortActionRequest.SerializeToString,
-            workflow_dot_run__service__pb2.AbortActionResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

flyte/_protos/workflow/state_service_pb2.py

@@ -1,67 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: workflow/state_service.proto
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-from flyte._protos.common import identifier_pb2 as common_dot_identifier__pb2
-from flyteidl.core import execution_pb2 as flyteidl_dot_core_dot_execution__pb2
-from google.rpc import status_pb2 as google_dot_rpc_dot_status__pb2
-from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
-from flyte._protos.workflow import run_definition_pb2 as workflow_dot_run__definition__pb2
-
-
-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')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'workflow.state_service_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-  DESCRIPTOR._options = None
-  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'
-  _PUTREQUEST.fields_by_name['action_id']._options = None
-  _PUTREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
-  _PUTREQUEST.fields_by_name['state']._options = None
-  _PUTREQUEST.fields_by_name['state']._serialized_options = b'\372B\004r\002\020\001'
-  _PUTRESPONSE.fields_by_name['action_id']._options = None
-  _PUTRESPONSE.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
-  _PUTRESPONSE.fields_by_name['status']._options = None
-  _PUTRESPONSE.fields_by_name['status']._serialized_options = b'\372B\005\212\001\002\020\001'
-  _GETREQUEST.fields_by_name['action_id']._options = None
-  _GETREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
-  _GETRESPONSE.fields_by_name['action_id']._options = None
-  _GETRESPONSE.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
-  _GETRESPONSE.fields_by_name['status']._options = None
-  _GETRESPONSE.fields_by_name['status']._serialized_options = b'\372B\005\212\001\002\020\001'
-  _GETRESPONSE.fields_by_name['state']._options = None
-  _GETRESPONSE.fields_by_name['state']._serialized_options = b'\372B\004r\002\020\001'
-  _WATCHREQUEST.oneofs_by_name['filter']._options = None
-  _WATCHREQUEST.oneofs_by_name['filter']._serialized_options = b'\370B\001'
-  _ACTIONUPDATE.fields_by_name['action_id']._options = None
-  _ACTIONUPDATE.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
-  _globals['_PUTREQUEST']._serialized_start=189
-  _globals['_PUTREQUEST']._serialized_end=380
-  _globals['_PUTRESPONSE']._serialized_start=383
-  _globals['_PUTRESPONSE']._serialized_end=524
-  _globals['_GETREQUEST']._serialized_start=526
-  _globals['_GETREQUEST']._serialized_end=612
-  _globals['_GETRESPONSE']._serialized_start=615
-  _globals['_GETRESPONSE']._serialized_end=787
-  _globals['_WATCHREQUEST']._serialized_start=789
-  _globals['_WATCHREQUEST']._serialized_end=897
-  _globals['_WATCHRESPONSE']._serialized_start=900
-  _globals['_WATCHRESPONSE']._serialized_end=1076
-  _globals['_CONTROLMESSAGE']._serialized_start=1078
-  _globals['_CONTROLMESSAGE']._serialized_end=1122
-  _globals['_ACTIONUPDATE']._serialized_start=1125
-  _globals['_ACTIONUPDATE']._serialized_end=1360
-  _globals['_STATESERVICE']._serialized_start=1363
-  _globals['_STATESERVICE']._serialized_end=1609
-# @@protoc_insertion_point(module_scope)

flyte/_protos/workflow/state_service_pb2.pyi

@@ -1,76 +0,0 @@
-from flyte._protos.common import identifier_pb2 as _identifier_pb2
-from flyteidl.core import execution_pb2 as _execution_pb2
-from google.rpc import status_pb2 as _status_pb2
-from flyte._protos.validate.validate import validate_pb2 as _validate_pb2
-from flyte._protos.workflow import run_definition_pb2 as _run_definition_pb2
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import message as _message
-from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
-
-DESCRIPTOR: _descriptor.FileDescriptor
-
-class PutRequest(_message.Message):
-    __slots__ = ["action_id", "parent_action_name", "state"]
-    ACTION_ID_FIELD_NUMBER: _ClassVar[int]
-    PARENT_ACTION_NAME_FIELD_NUMBER: _ClassVar[int]
-    STATE_FIELD_NUMBER: _ClassVar[int]
-    action_id: _identifier_pb2.ActionIdentifier
-    parent_action_name: str
-    state: str
-    def __init__(self, action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ..., parent_action_name: _Optional[str] = ..., state: _Optional[str] = ...) -> None: ...
-
-class PutResponse(_message.Message):
-    __slots__ = ["action_id", "status"]
-    ACTION_ID_FIELD_NUMBER: _ClassVar[int]
-    STATUS_FIELD_NUMBER: _ClassVar[int]
-    action_id: _identifier_pb2.ActionIdentifier
-    status: _status_pb2.Status
-    def __init__(self, action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ..., status: _Optional[_Union[_status_pb2.Status, _Mapping]] = ...) -> None: ...
-
-class GetRequest(_message.Message):
-    __slots__ = ["action_id"]
-    ACTION_ID_FIELD_NUMBER: _ClassVar[int]
-    action_id: _identifier_pb2.ActionIdentifier
-    def __init__(self, action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
-
-class GetResponse(_message.Message):
-    __slots__ = ["action_id", "status", "state"]
-    ACTION_ID_FIELD_NUMBER: _ClassVar[int]
-    STATUS_FIELD_NUMBER: _ClassVar[int]
-    STATE_FIELD_NUMBER: _ClassVar[int]
-    action_id: _identifier_pb2.ActionIdentifier
-    status: _status_pb2.Status
-    state: str
-    def __init__(self, action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ..., status: _Optional[_Union[_status_pb2.Status, _Mapping]] = ..., state: _Optional[str] = ...) -> None: ...
-
-class WatchRequest(_message.Message):
-    __slots__ = ["parent_action_id"]
-    PARENT_ACTION_ID_FIELD_NUMBER: _ClassVar[int]
-    parent_action_id: _identifier_pb2.ActionIdentifier
-    def __init__(self, parent_action_id: _Optional[_Union[_identifier_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
-
-class WatchResponse(_message.Message):
-    __slots__ = ["action_update", "control_message"]
-    ACTION_UPDATE_FIELD_NUMBER: _ClassVar[int]
-    CONTROL_MESSAGE_FIELD_NUMBER: _ClassVar[int]
-    action_update: ActionUpdate
-    control_message: ControlMessage
-    def __init__(self, action_update: _Optional[_Union[ActionUpdate, _Mapping]] = ..., control_message: _Optional[_Union[ControlMessage, _Mapping]] = ...) -> None: ...
-
-class ControlMessage(_message.Message):
-    __slots__ = ["sentinel"]
-    SENTINEL_FIELD_NUMBER: _ClassVar[int]
-    sentinel: bool
-    def __init__(self, sentinel: bool = ...) -> None: ...
-
-class ActionUpdate(_message.Message):
-    __slots__ = ["action_id", "phase", "error", "output_uri"]
-    ACTION_ID_FIELD_NUMBER: _ClassVar[int]
-    PHASE_FIELD_NUMBER: _ClassVar[int]
-    ERROR_FIELD_NUMBER: _ClassVar[int]
-    OUTPUT_URI_FIELD_NUMBER: _ClassVar[int]
-    action_id: _identifier_pb2.ActionIdentifier
-    phase: _run_definition_pb2.Phase
-    error: _execution_pb2.ExecutionError
-    output_uri: str
-    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: ...