flyte 0.0.1b0__py3-none-any.whl

flyte/_protos/workflow/run_logs_service_pb2.py

@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: workflow/run_logs_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.logs.dataplane import payload_pb2 as logs_dot_dataplane_dot_payload__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\x1fworkflow/run_logs_service.proto\x12\x11\x63loudidl.workflow\x1a\x1clogs/dataplane/payload.proto\x1a\x17validate/validate.proto\x1a\x1dworkflow/run_definition.proto\"\x80\x01\n\x0fTailLogsRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\x12!\n\x07\x61ttempt\x18\x02 \x01(\rB\x07\xfa\x42\x04*\x02 \x00R\x07\x61ttempt\"\x90\x01\n\x10TailLogsResponse\x12<\n\x04logs\x18\x01 \x03(\x0b\x32(.cloudidl.workflow.TailLogsResponse.LogsR\x04logs\x1a>\n\x04Logs\x12\x36\n\x05lines\x18\x01 \x03(\x0b\x32 .cloudidl.logs.dataplane.LogLineR\x05lines2l\n\x0eRunLogsService\x12Z\n\x08TailLogs\x12\".cloudidl.workflow.TailLogsRequest\x1a#.cloudidl.workflow.TailLogsResponse\"\x03\x90\x02\x01\x30\x01\x42\xc0\x01\n\x15\x63om.cloudidl.workflowB\x13RunLogsServiceProtoH\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.run_logs_service_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\023RunLogsServiceProtoH\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'
+  _TAILLOGSREQUEST.fields_by_name['action_id']._options = None
+  _TAILLOGSREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _TAILLOGSREQUEST.fields_by_name['attempt']._options = None
+  _TAILLOGSREQUEST.fields_by_name['attempt']._serialized_options = b'\372B\004*\002 \000'
+  _RUNLOGSSERVICE.methods_by_name['TailLogs']._options = None
+  _RUNLOGSSERVICE.methods_by_name['TailLogs']._serialized_options = b'\220\002\001'
+  _globals['_TAILLOGSREQUEST']._serialized_start=141
+  _globals['_TAILLOGSREQUEST']._serialized_end=269
+  _globals['_TAILLOGSRESPONSE']._serialized_start=272
+  _globals['_TAILLOGSRESPONSE']._serialized_end=416
+  _globals['_TAILLOGSRESPONSE_LOGS']._serialized_start=354
+  _globals['_TAILLOGSRESPONSE_LOGS']._serialized_end=416
+  _globals['_RUNLOGSSERVICE']._serialized_start=418
+  _globals['_RUNLOGSSERVICE']._serialized_end=526
+# @@protoc_insertion_point(module_scope)

flyte/_protos/workflow/run_logs_service_pb2.pyi

@@ -0,0 +1,28 @@
+from flyte._protos.logs.dataplane import payload_pb2 as _payload_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.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class TailLogsRequest(_message.Message):
+    __slots__ = ["action_id", "attempt"]
+    ACTION_ID_FIELD_NUMBER: _ClassVar[int]
+    ATTEMPT_FIELD_NUMBER: _ClassVar[int]
+    action_id: _run_definition_pb2.ActionIdentifier
+    attempt: int
+    def __init__(self, action_id: _Optional[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]] = ..., attempt: _Optional[int] = ...) -> None: ...
+
+class TailLogsResponse(_message.Message):
+    __slots__ = ["logs"]
+    class Logs(_message.Message):
+        __slots__ = ["lines"]
+        LINES_FIELD_NUMBER: _ClassVar[int]
+        lines: _containers.RepeatedCompositeFieldContainer[_payload_pb2.LogLine]
+        def __init__(self, lines: _Optional[_Iterable[_Union[_payload_pb2.LogLine, _Mapping]]] = ...) -> None: ...
+    LOGS_FIELD_NUMBER: _ClassVar[int]
+    logs: _containers.RepeatedCompositeFieldContainer[TailLogsResponse.Logs]
+    def __init__(self, logs: _Optional[_Iterable[_Union[TailLogsResponse.Logs, _Mapping]]] = ...) -> None: ...

flyte/_protos/workflow/run_logs_service_pb2_grpc.py

@@ -0,0 +1,69 @@
+# 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_logs_service_pb2 as workflow_dot_run__logs__service__pb2
+
+
+class RunLogsServiceStub(object):
+    """RunLogsService provides an interface for streaming logs.
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.TailLogs = channel.unary_stream(
+                '/cloudidl.workflow.RunLogsService/TailLogs',
+                request_serializer=workflow_dot_run__logs__service__pb2.TailLogsRequest.SerializeToString,
+                response_deserializer=workflow_dot_run__logs__service__pb2.TailLogsResponse.FromString,
+                )
+
+
+class RunLogsServiceServicer(object):
+    """RunLogsService provides an interface for streaming logs.
+    """
+
+    def TailLogs(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_RunLogsServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'TailLogs': grpc.unary_stream_rpc_method_handler(
+                    servicer.TailLogs,
+                    request_deserializer=workflow_dot_run__logs__service__pb2.TailLogsRequest.FromString,
+                    response_serializer=workflow_dot_run__logs__service__pb2.TailLogsResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'cloudidl.workflow.RunLogsService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class RunLogsService(object):
+    """RunLogsService provides an interface for streaming logs.
+    """
+
+    @staticmethod
+    def TailLogs(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.RunLogsService/TailLogs',
+            workflow_dot_run__logs__service__pb2.TailLogsRequest.SerializeToString,
+            workflow_dot_run__logs__service__pb2.TailLogsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

flyte/_protos/workflow/run_service_pb2.py

@@ -0,0 +1,133 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: workflow/run_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 flyte._protos.common import list_pb2 as common_dot_list__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
+from flyte._protos.workflow import task_definition_pb2 as workflow_dot_task__definition__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1aworkflow/run_service.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x11\x63ommon/list.proto\x1a\x17validate/validate.proto\x1a\x1dworkflow/run_definition.proto\x1a\x1eworkflow/task_definition.proto\"\xb6\x03\n\x10\x43reateRunRequest\x12\x43\n\x06run_id\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x05runId\x12M\n\nproject_id\x18\x06 \x01(\x0b\x32\".cloudidl.common.ProjectIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tprojectId\x12\x46\n\x07task_id\x18\x02 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x01R\x06taskId\x12\x44\n\ttask_spec\x18\x03 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x01R\x08taskSpec\x12\x31\n\x06inputs\x18\x04 \x01(\x0b\x32\x19.cloudidl.workflow.InputsR\x06inputs\x12\x35\n\x08run_spec\x18\x05 \x01(\x0b\x32\x1a.cloudidl.workflow.RunSpecR\x07runSpecB\t\n\x02id\x12\x03\xf8\x42\x01\x42\x0b\n\x04task\x12\x03\xf8\x42\x01\"=\n\x11\x43reateRunResponse\x12(\n\x03run\x18\x01 \x01(\x0b\x32\x16.cloudidl.workflow.RunR\x03run\"T\n\x0f\x41\x62ortRunRequest\x12\x41\n\x06run_id\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\"\x12\n\x10\x41\x62ortRunResponse\"Y\n\x14GetRunDetailsRequest\x12\x41\n\x06run_id\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\"P\n\x15GetRunDetailsResponse\x12\x37\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32\x1d.cloudidl.workflow.RunDetailsR\x07\x64\x65tails\"[\n\x16WatchRunDetailsRequest\x12\x41\n\x06run_id\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\"R\n\x17WatchRunDetailsResponse\x12\x37\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32\x1d.cloudidl.workflow.RunDetailsR\x07\x64\x65tails\"e\n\x17GetActionDetailsRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\"V\n\x18GetActionDetailsResponse\x12:\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.ActionDetailsR\x07\x64\x65tails\"g\n\x19WatchActionDetailsRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\"X\n\x1aWatchActionDetailsResponse\x12:\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.ActionDetailsR\x07\x64\x65tails\"b\n\x14GetActionDataRequest\x12J\n\taction_id\x18\x01 \x01(\x0b\x32#.cloudidl.workflow.ActionIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08\x61\x63tionId\"\x80\x01\n\x15GetActionDataResponse\x12\x31\n\x06inputs\x18\x01 \x01(\x0b\x32\x19.cloudidl.workflow.InputsR\x06inputs\x12\x34\n\x07outputs\x18\x02 \x01(\x0b\x32\x1a.cloudidl.workflow.OutputsR\x07outputs\"\xde\x02\n\x0fListRunsRequest\x12\x36\n\x07request\x18\x01 \x01(\x0b\x32\x1c.cloudidl.common.ListRequestR\x07request\x12\x1b\n\x03org\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x03org\x12M\n\ncluster_id\x18\x03 \x01(\x0b\x32\".cloudidl.common.ClusterIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tclusterId\x12M\n\nproject_id\x18\x04 \x01(\x0b\x32\".cloudidl.common.ProjectIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tprojectId\x12\x46\n\x07task_id\x18\x05 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x06taskIdB\x10\n\tfilter_by\x12\x03\xf8\x42\x01\"T\n\x10ListRunsResponse\x12*\n\x04runs\x18\x01 \x03(\x0b\x32\x16.cloudidl.workflow.RunR\x04runs\x12\x14\n\x05token\x18\x02 \x01(\tR\x05token\"\xa4\x02\n\x10WatchRunsRequest\x12\x1b\n\x03org\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x03org\x12M\n\ncluster_id\x18\x03 \x01(\x0b\x32\".cloudidl.common.ClusterIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tclusterId\x12M\n\nproject_id\x18\x04 \x01(\x0b\x32\".cloudidl.common.ProjectIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tprojectId\x12\x46\n\x07task_id\x18\x05 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\x06taskIdB\r\n\x06target\x12\x03\xf8\x42\x01\"?\n\x11WatchRunsResponse\x12*\n\x04runs\x18\x01 \x03(\x0b\x32\x16.cloudidl.workflow.RunR\x04runs\"\x8f\x01\n\x12ListActionsRequest\x12\x36\n\x07request\x18\x01 \x01(\x0b\x32\x1c.cloudidl.common.ListRequestR\x07request\x12\x41\n\x06run_id\x18\x02 \x01(\x0b\x32 .cloudidl.workflow.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\"`\n\x13ListActionsResponse\x12\x33\n\x07\x61\x63tions\x18\x01 \x03(\x0b\x32\x19.cloudidl.workflow.ActionR\x07\x61\x63tions\x12\x14\n\x05token\x18\x02 \x01(\tR\x05token\"\x89\x01\n\x13WatchActionsRequest\x12\x41\n\x06run_id\x18\x01 \x01(\x0b\x32 .cloudidl.workflow.RunIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x05runId\x12/\n\x06\x66ilter\x18\x02 \x03(\x0b\x32\x17.cloudidl.common.FilterR\x06\x66ilter\"d\n\x14WatchActionsResponse\x12L\n\x10\x65nriched_actions\x18\x01 \x03(\x0b\x32!.cloudidl.workflow.EnrichedActionR\x0f\x65nrichedActions2\xe4\x08\n\nRunService\x12X\n\tCreateRun\x12#.cloudidl.workflow.CreateRunRequest\x1a$.cloudidl.workflow.CreateRunResponse\"\x00\x12U\n\x08\x41\x62ortRun\x12\".cloudidl.workflow.AbortRunRequest\x1a#.cloudidl.workflow.AbortRunResponse\"\x00\x12g\n\rGetRunDetails\x12\'.cloudidl.workflow.GetRunDetailsRequest\x1a(.cloudidl.workflow.GetRunDetailsResponse\"\x03\x90\x02\x01\x12l\n\x0fWatchRunDetails\x12).cloudidl.workflow.WatchRunDetailsRequest\x1a*.cloudidl.workflow.WatchRunDetailsResponse\"\x00\x30\x01\x12p\n\x10GetActionDetails\x12*.cloudidl.workflow.GetActionDetailsRequest\x1a+.cloudidl.workflow.GetActionDetailsResponse\"\x03\x90\x02\x01\x12u\n\x12WatchActionDetails\x12,.cloudidl.workflow.WatchActionDetailsRequest\x1a-.cloudidl.workflow.WatchActionDetailsResponse\"\x00\x30\x01\x12g\n\rGetActionData\x12\'.cloudidl.workflow.GetActionDataRequest\x1a(.cloudidl.workflow.GetActionDataResponse\"\x03\x90\x02\x01\x12X\n\x08ListRuns\x12\".cloudidl.workflow.ListRunsRequest\x1a#.cloudidl.workflow.ListRunsResponse\"\x03\x90\x02\x01\x12Z\n\tWatchRuns\x12#.cloudidl.workflow.WatchRunsRequest\x1a$.cloudidl.workflow.WatchRunsResponse\"\x00\x30\x01\x12\x61\n\x0bListActions\x12%.cloudidl.workflow.ListActionsRequest\x1a&.cloudidl.workflow.ListActionsResponse\"\x03\x90\x02\x01\x12\x63\n\x0cWatchActions\x12&.cloudidl.workflow.WatchActionsRequest\x1a\'.cloudidl.workflow.WatchActionsResponse\"\x00\x30\x01\x42\xbc\x01\n\x15\x63om.cloudidl.workflowB\x0fRunServiceProtoH\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.run_service_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\017RunServiceProtoH\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'
+  _CREATERUNREQUEST.oneofs_by_name['id']._options = None
+  _CREATERUNREQUEST.oneofs_by_name['id']._serialized_options = b'\370B\001'
+  _CREATERUNREQUEST.oneofs_by_name['task']._options = None
+  _CREATERUNREQUEST.oneofs_by_name['task']._serialized_options = b'\370B\001'
+  _CREATERUNREQUEST.fields_by_name['run_id']._options = None
+  _CREATERUNREQUEST.fields_by_name['run_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _CREATERUNREQUEST.fields_by_name['project_id']._options = None
+  _CREATERUNREQUEST.fields_by_name['project_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _CREATERUNREQUEST.fields_by_name['task_id']._options = None
+  _CREATERUNREQUEST.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _CREATERUNREQUEST.fields_by_name['task_spec']._options = None
+  _CREATERUNREQUEST.fields_by_name['task_spec']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _ABORTRUNREQUEST.fields_by_name['run_id']._options = None
+  _ABORTRUNREQUEST.fields_by_name['run_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _GETRUNDETAILSREQUEST.fields_by_name['run_id']._options = None
+  _GETRUNDETAILSREQUEST.fields_by_name['run_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _WATCHRUNDETAILSREQUEST.fields_by_name['run_id']._options = None
+  _WATCHRUNDETAILSREQUEST.fields_by_name['run_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _GETACTIONDETAILSREQUEST.fields_by_name['action_id']._options = None
+  _GETACTIONDETAILSREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _WATCHACTIONDETAILSREQUEST.fields_by_name['action_id']._options = None
+  _WATCHACTIONDETAILSREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _GETACTIONDATAREQUEST.fields_by_name['action_id']._options = None
+  _GETACTIONDATAREQUEST.fields_by_name['action_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _LISTRUNSREQUEST.oneofs_by_name['filter_by']._options = None
+  _LISTRUNSREQUEST.oneofs_by_name['filter_by']._serialized_options = b'\370B\001'
+  _LISTRUNSREQUEST.fields_by_name['org']._options = None
+  _LISTRUNSREQUEST.fields_by_name['org']._serialized_options = b'\372B\004r\002\020\001'
+  _LISTRUNSREQUEST.fields_by_name['cluster_id']._options = None
+  _LISTRUNSREQUEST.fields_by_name['cluster_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _LISTRUNSREQUEST.fields_by_name['project_id']._options = None
+  _LISTRUNSREQUEST.fields_by_name['project_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _LISTRUNSREQUEST.fields_by_name['task_id']._options = None
+  _LISTRUNSREQUEST.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _WATCHRUNSREQUEST.oneofs_by_name['target']._options = None
+  _WATCHRUNSREQUEST.oneofs_by_name['target']._serialized_options = b'\370B\001'
+  _WATCHRUNSREQUEST.fields_by_name['org']._options = None
+  _WATCHRUNSREQUEST.fields_by_name['org']._serialized_options = b'\372B\004r\002\020\001'
+  _WATCHRUNSREQUEST.fields_by_name['cluster_id']._options = None
+  _WATCHRUNSREQUEST.fields_by_name['cluster_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _WATCHRUNSREQUEST.fields_by_name['project_id']._options = None
+  _WATCHRUNSREQUEST.fields_by_name['project_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _WATCHRUNSREQUEST.fields_by_name['task_id']._options = None
+  _WATCHRUNSREQUEST.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _LISTACTIONSREQUEST.fields_by_name['run_id']._options = None
+  _LISTACTIONSREQUEST.fields_by_name['run_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _WATCHACTIONSREQUEST.fields_by_name['run_id']._options = None
+  _WATCHACTIONSREQUEST.fields_by_name['run_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _RUNSERVICE.methods_by_name['GetRunDetails']._options = None
+  _RUNSERVICE.methods_by_name['GetRunDetails']._serialized_options = b'\220\002\001'
+  _RUNSERVICE.methods_by_name['GetActionDetails']._options = None
+  _RUNSERVICE.methods_by_name['GetActionDetails']._serialized_options = b'\220\002\001'
+  _RUNSERVICE.methods_by_name['GetActionData']._options = None
+  _RUNSERVICE.methods_by_name['GetActionData']._serialized_options = b'\220\002\001'
+  _RUNSERVICE.methods_by_name['ListRuns']._options = None
+  _RUNSERVICE.methods_by_name['ListRuns']._serialized_options = b'\220\002\001'
+  _RUNSERVICE.methods_by_name['ListActions']._options = None
+  _RUNSERVICE.methods_by_name['ListActions']._serialized_options = b'\220\002\001'
+  _globals['_CREATERUNREQUEST']._serialized_start=182
+  _globals['_CREATERUNREQUEST']._serialized_end=620
+  _globals['_CREATERUNRESPONSE']._serialized_start=622
+  _globals['_CREATERUNRESPONSE']._serialized_end=683
+  _globals['_ABORTRUNREQUEST']._serialized_start=685
+  _globals['_ABORTRUNREQUEST']._serialized_end=769
+  _globals['_ABORTRUNRESPONSE']._serialized_start=771
+  _globals['_ABORTRUNRESPONSE']._serialized_end=789
+  _globals['_GETRUNDETAILSREQUEST']._serialized_start=791
+  _globals['_GETRUNDETAILSREQUEST']._serialized_end=880
+  _globals['_GETRUNDETAILSRESPONSE']._serialized_start=882
+  _globals['_GETRUNDETAILSRESPONSE']._serialized_end=962
+  _globals['_WATCHRUNDETAILSREQUEST']._serialized_start=964
+  _globals['_WATCHRUNDETAILSREQUEST']._serialized_end=1055
+  _globals['_WATCHRUNDETAILSRESPONSE']._serialized_start=1057
+  _globals['_WATCHRUNDETAILSRESPONSE']._serialized_end=1139
+  _globals['_GETACTIONDETAILSREQUEST']._serialized_start=1141
+  _globals['_GETACTIONDETAILSREQUEST']._serialized_end=1242
+  _globals['_GETACTIONDETAILSRESPONSE']._serialized_start=1244
+  _globals['_GETACTIONDETAILSRESPONSE']._serialized_end=1330
+  _globals['_WATCHACTIONDETAILSREQUEST']._serialized_start=1332
+  _globals['_WATCHACTIONDETAILSREQUEST']._serialized_end=1435
+  _globals['_WATCHACTIONDETAILSRESPONSE']._serialized_start=1437
+  _globals['_WATCHACTIONDETAILSRESPONSE']._serialized_end=1525
+  _globals['_GETACTIONDATAREQUEST']._serialized_start=1527
+  _globals['_GETACTIONDATAREQUEST']._serialized_end=1625
+  _globals['_GETACTIONDATARESPONSE']._serialized_start=1628
+  _globals['_GETACTIONDATARESPONSE']._serialized_end=1756
+  _globals['_LISTRUNSREQUEST']._serialized_start=1759
+  _globals['_LISTRUNSREQUEST']._serialized_end=2109
+  _globals['_LISTRUNSRESPONSE']._serialized_start=2111
+  _globals['_LISTRUNSRESPONSE']._serialized_end=2195
+  _globals['_WATCHRUNSREQUEST']._serialized_start=2198
+  _globals['_WATCHRUNSREQUEST']._serialized_end=2490
+  _globals['_WATCHRUNSRESPONSE']._serialized_start=2492
+  _globals['_WATCHRUNSRESPONSE']._serialized_end=2555
+  _globals['_LISTACTIONSREQUEST']._serialized_start=2558
+  _globals['_LISTACTIONSREQUEST']._serialized_end=2701
+  _globals['_LISTACTIONSRESPONSE']._serialized_start=2703
+  _globals['_LISTACTIONSRESPONSE']._serialized_end=2799
+  _globals['_WATCHACTIONSREQUEST']._serialized_start=2802
+  _globals['_WATCHACTIONSREQUEST']._serialized_end=2939
+  _globals['_WATCHACTIONSRESPONSE']._serialized_start=2941
+  _globals['_WATCHACTIONSRESPONSE']._serialized_end=3041
+  _globals['_RUNSERVICE']._serialized_start=3044
+  _globals['_RUNSERVICE']._serialized_end=4168
+# @@protoc_insertion_point(module_scope)

flyte/_protos/workflow/run_service_pb2.pyi

@@ -0,0 +1,175 @@
+from flyte._protos.common import identifier_pb2 as _identifier_pb2
+from flyte._protos.common import list_pb2 as _list_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 flyte._protos.workflow import task_definition_pb2 as _task_definition_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class CreateRunRequest(_message.Message):
+    __slots__ = ["run_id", "project_id", "task_id", "task_spec", "inputs", "run_spec"]
+    RUN_ID_FIELD_NUMBER: _ClassVar[int]
+    PROJECT_ID_FIELD_NUMBER: _ClassVar[int]
+    TASK_ID_FIELD_NUMBER: _ClassVar[int]
+    TASK_SPEC_FIELD_NUMBER: _ClassVar[int]
+    INPUTS_FIELD_NUMBER: _ClassVar[int]
+    RUN_SPEC_FIELD_NUMBER: _ClassVar[int]
+    run_id: _run_definition_pb2.RunIdentifier
+    project_id: _identifier_pb2.ProjectIdentifier
+    task_id: _task_definition_pb2.TaskIdentifier
+    task_spec: _task_definition_pb2.TaskSpec
+    inputs: _run_definition_pb2.Inputs
+    run_spec: _run_definition_pb2.RunSpec
+    def __init__(self, run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ..., project_id: _Optional[_Union[_identifier_pb2.ProjectIdentifier, _Mapping]] = ..., task_id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ..., task_spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ..., inputs: _Optional[_Union[_run_definition_pb2.Inputs, _Mapping]] = ..., run_spec: _Optional[_Union[_run_definition_pb2.RunSpec, _Mapping]] = ...) -> None: ...
+
+class CreateRunResponse(_message.Message):
+    __slots__ = ["run"]
+    RUN_FIELD_NUMBER: _ClassVar[int]
+    run: _run_definition_pb2.Run
+    def __init__(self, run: _Optional[_Union[_run_definition_pb2.Run, _Mapping]] = ...) -> None: ...
+
+class AbortRunRequest(_message.Message):
+    __slots__ = ["run_id"]
+    RUN_ID_FIELD_NUMBER: _ClassVar[int]
+    run_id: _run_definition_pb2.RunIdentifier
+    def __init__(self, run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
+
+class AbortRunResponse(_message.Message):
+    __slots__ = []
+    def __init__(self) -> None: ...
+
+class GetRunDetailsRequest(_message.Message):
+    __slots__ = ["run_id"]
+    RUN_ID_FIELD_NUMBER: _ClassVar[int]
+    run_id: _run_definition_pb2.RunIdentifier
+    def __init__(self, run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
+
+class GetRunDetailsResponse(_message.Message):
+    __slots__ = ["details"]
+    DETAILS_FIELD_NUMBER: _ClassVar[int]
+    details: _run_definition_pb2.RunDetails
+    def __init__(self, details: _Optional[_Union[_run_definition_pb2.RunDetails, _Mapping]] = ...) -> None: ...
+
+class WatchRunDetailsRequest(_message.Message):
+    __slots__ = ["run_id"]
+    RUN_ID_FIELD_NUMBER: _ClassVar[int]
+    run_id: _run_definition_pb2.RunIdentifier
+    def __init__(self, run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
+
+class WatchRunDetailsResponse(_message.Message):
+    __slots__ = ["details"]
+    DETAILS_FIELD_NUMBER: _ClassVar[int]
+    details: _run_definition_pb2.RunDetails
+    def __init__(self, details: _Optional[_Union[_run_definition_pb2.RunDetails, _Mapping]] = ...) -> None: ...
+
+class GetActionDetailsRequest(_message.Message):
+    __slots__ = ["action_id"]
+    ACTION_ID_FIELD_NUMBER: _ClassVar[int]
+    action_id: _run_definition_pb2.ActionIdentifier
+    def __init__(self, action_id: _Optional[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
+
+class GetActionDetailsResponse(_message.Message):
+    __slots__ = ["details"]
+    DETAILS_FIELD_NUMBER: _ClassVar[int]
+    details: _run_definition_pb2.ActionDetails
+    def __init__(self, details: _Optional[_Union[_run_definition_pb2.ActionDetails, _Mapping]] = ...) -> None: ...
+
+class WatchActionDetailsRequest(_message.Message):
+    __slots__ = ["action_id"]
+    ACTION_ID_FIELD_NUMBER: _ClassVar[int]
+    action_id: _run_definition_pb2.ActionIdentifier
+    def __init__(self, action_id: _Optional[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
+
+class WatchActionDetailsResponse(_message.Message):
+    __slots__ = ["details"]
+    DETAILS_FIELD_NUMBER: _ClassVar[int]
+    details: _run_definition_pb2.ActionDetails
+    def __init__(self, details: _Optional[_Union[_run_definition_pb2.ActionDetails, _Mapping]] = ...) -> None: ...
+
+class GetActionDataRequest(_message.Message):
+    __slots__ = ["action_id"]
+    ACTION_ID_FIELD_NUMBER: _ClassVar[int]
+    action_id: _run_definition_pb2.ActionIdentifier
+    def __init__(self, action_id: _Optional[_Union[_run_definition_pb2.ActionIdentifier, _Mapping]] = ...) -> None: ...
+
+class GetActionDataResponse(_message.Message):
+    __slots__ = ["inputs", "outputs"]
+    INPUTS_FIELD_NUMBER: _ClassVar[int]
+    OUTPUTS_FIELD_NUMBER: _ClassVar[int]
+    inputs: _run_definition_pb2.Inputs
+    outputs: _run_definition_pb2.Outputs
+    def __init__(self, inputs: _Optional[_Union[_run_definition_pb2.Inputs, _Mapping]] = ..., outputs: _Optional[_Union[_run_definition_pb2.Outputs, _Mapping]] = ...) -> None: ...
+
+class ListRunsRequest(_message.Message):
+    __slots__ = ["request", "org", "cluster_id", "project_id", "task_id"]
+    REQUEST_FIELD_NUMBER: _ClassVar[int]
+    ORG_FIELD_NUMBER: _ClassVar[int]
+    CLUSTER_ID_FIELD_NUMBER: _ClassVar[int]
+    PROJECT_ID_FIELD_NUMBER: _ClassVar[int]
+    TASK_ID_FIELD_NUMBER: _ClassVar[int]
+    request: _list_pb2.ListRequest
+    org: str
+    cluster_id: _identifier_pb2.ClusterIdentifier
+    project_id: _identifier_pb2.ProjectIdentifier
+    task_id: _task_definition_pb2.TaskIdentifier
+    def __init__(self, request: _Optional[_Union[_list_pb2.ListRequest, _Mapping]] = ..., org: _Optional[str] = ..., cluster_id: _Optional[_Union[_identifier_pb2.ClusterIdentifier, _Mapping]] = ..., project_id: _Optional[_Union[_identifier_pb2.ProjectIdentifier, _Mapping]] = ..., task_id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ...) -> None: ...
+
+class ListRunsResponse(_message.Message):
+    __slots__ = ["runs", "token"]
+    RUNS_FIELD_NUMBER: _ClassVar[int]
+    TOKEN_FIELD_NUMBER: _ClassVar[int]
+    runs: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.Run]
+    token: str
+    def __init__(self, runs: _Optional[_Iterable[_Union[_run_definition_pb2.Run, _Mapping]]] = ..., token: _Optional[str] = ...) -> None: ...
+
+class WatchRunsRequest(_message.Message):
+    __slots__ = ["org", "cluster_id", "project_id", "task_id"]
+    ORG_FIELD_NUMBER: _ClassVar[int]
+    CLUSTER_ID_FIELD_NUMBER: _ClassVar[int]
+    PROJECT_ID_FIELD_NUMBER: _ClassVar[int]
+    TASK_ID_FIELD_NUMBER: _ClassVar[int]
+    org: str
+    cluster_id: _identifier_pb2.ClusterIdentifier
+    project_id: _identifier_pb2.ProjectIdentifier
+    task_id: _task_definition_pb2.TaskIdentifier
+    def __init__(self, org: _Optional[str] = ..., cluster_id: _Optional[_Union[_identifier_pb2.ClusterIdentifier, _Mapping]] = ..., project_id: _Optional[_Union[_identifier_pb2.ProjectIdentifier, _Mapping]] = ..., task_id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ...) -> None: ...
+
+class WatchRunsResponse(_message.Message):
+    __slots__ = ["runs"]
+    RUNS_FIELD_NUMBER: _ClassVar[int]
+    runs: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.Run]
+    def __init__(self, runs: _Optional[_Iterable[_Union[_run_definition_pb2.Run, _Mapping]]] = ...) -> None: ...
+
+class ListActionsRequest(_message.Message):
+    __slots__ = ["request", "run_id"]
+    REQUEST_FIELD_NUMBER: _ClassVar[int]
+    RUN_ID_FIELD_NUMBER: _ClassVar[int]
+    request: _list_pb2.ListRequest
+    run_id: _run_definition_pb2.RunIdentifier
+    def __init__(self, request: _Optional[_Union[_list_pb2.ListRequest, _Mapping]] = ..., run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ...) -> None: ...
+
+class ListActionsResponse(_message.Message):
+    __slots__ = ["actions", "token"]
+    ACTIONS_FIELD_NUMBER: _ClassVar[int]
+    TOKEN_FIELD_NUMBER: _ClassVar[int]
+    actions: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.Action]
+    token: str
+    def __init__(self, actions: _Optional[_Iterable[_Union[_run_definition_pb2.Action, _Mapping]]] = ..., token: _Optional[str] = ...) -> None: ...
+
+class WatchActionsRequest(_message.Message):
+    __slots__ = ["run_id", "filter"]
+    RUN_ID_FIELD_NUMBER: _ClassVar[int]
+    FILTER_FIELD_NUMBER: _ClassVar[int]
+    run_id: _run_definition_pb2.RunIdentifier
+    filter: _containers.RepeatedCompositeFieldContainer[_list_pb2.Filter]
+    def __init__(self, run_id: _Optional[_Union[_run_definition_pb2.RunIdentifier, _Mapping]] = ..., filter: _Optional[_Iterable[_Union[_list_pb2.Filter, _Mapping]]] = ...) -> None: ...
+
+class WatchActionsResponse(_message.Message):
+    __slots__ = ["enriched_actions"]
+    ENRICHED_ACTIONS_FIELD_NUMBER: _ClassVar[int]
+    enriched_actions: _containers.RepeatedCompositeFieldContainer[_run_definition_pb2.EnrichedAction]
+    def __init__(self, enriched_actions: _Optional[_Iterable[_Union[_run_definition_pb2.EnrichedAction, _Mapping]]] = ...) -> None: ...