flyte 0.1.0__py3-none-any.whl → 0.2.0a0__py3-none-any.whl

flyte/_protos/workflow/state_service_pb2_grpc.py

@@ -0,0 +1,138 @@
+# 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 state_service_pb2 as workflow_dot_state__service__pb2
+
+
+class StateServiceStub(object):
+    """provides an interface for managing the state of actions.
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.Put = channel.stream_stream(
+                '/cloudidl.workflow.StateService/Put',
+                request_serializer=workflow_dot_state__service__pb2.PutRequest.SerializeToString,
+                response_deserializer=workflow_dot_state__service__pb2.PutResponse.FromString,
+                )
+        self.Get = channel.stream_stream(
+                '/cloudidl.workflow.StateService/Get',
+                request_serializer=workflow_dot_state__service__pb2.GetRequest.SerializeToString,
+                response_deserializer=workflow_dot_state__service__pb2.GetResponse.FromString,
+                )
+        self.Watch = channel.unary_stream(
+                '/cloudidl.workflow.StateService/Watch',
+                request_serializer=workflow_dot_state__service__pb2.WatchRequest.SerializeToString,
+                response_deserializer=workflow_dot_state__service__pb2.WatchResponse.FromString,
+                )
+
+
+class StateServiceServicer(object):
+    """provides an interface for managing the state of actions.
+    """
+
+    def Put(self, request_iterator, context):
+        """put the state of an action.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Get(self, request_iterator, context):
+        """get the state of an action.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Watch(self, request, context):
+        """watch for updates to the state of actions. this api guarantees at-least-once delivery semantics.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_StateServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'Put': grpc.stream_stream_rpc_method_handler(
+                    servicer.Put,
+                    request_deserializer=workflow_dot_state__service__pb2.PutRequest.FromString,
+                    response_serializer=workflow_dot_state__service__pb2.PutResponse.SerializeToString,
+            ),
+            'Get': grpc.stream_stream_rpc_method_handler(
+                    servicer.Get,
+                    request_deserializer=workflow_dot_state__service__pb2.GetRequest.FromString,
+                    response_serializer=workflow_dot_state__service__pb2.GetResponse.SerializeToString,
+            ),
+            'Watch': grpc.unary_stream_rpc_method_handler(
+                    servicer.Watch,
+                    request_deserializer=workflow_dot_state__service__pb2.WatchRequest.FromString,
+                    response_serializer=workflow_dot_state__service__pb2.WatchResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'cloudidl.workflow.StateService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class StateService(object):
+    """provides an interface for managing the state of actions.
+    """
+
+    @staticmethod
+    def Put(request_iterator,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.stream_stream(request_iterator, target, '/cloudidl.workflow.StateService/Put',
+            workflow_dot_state__service__pb2.PutRequest.SerializeToString,
+            workflow_dot_state__service__pb2.PutResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def Get(request_iterator,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.stream_stream(request_iterator, target, '/cloudidl.workflow.StateService/Get',
+            workflow_dot_state__service__pb2.GetRequest.SerializeToString,
+            workflow_dot_state__service__pb2.GetResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def Watch(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.StateService/Watch',
+            workflow_dot_state__service__pb2.WatchRequest.SerializeToString,
+            workflow_dot_state__service__pb2.WatchResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

flyte/_protos/workflow/task_definition_pb2.py

@@ -0,0 +1,79 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: workflow/task_definition.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 identity_pb2 as common_dot_identity__pb2
+from flyteidl.core import tasks_pb2 as flyteidl_dot_core_dot_tasks__pb2
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
+from flyte._protos.workflow import common_pb2 as workflow_dot_common__pb2
+from flyte._protos.workflow import environment_pb2 as workflow_dot_environment__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1eworkflow/task_definition.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x15\x63ommon/identity.proto\x1a\x19\x66lyteidl/core/tasks.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17validate/validate.proto\x1a\x15workflow/common.proto\x1a\x1aworkflow/environment.proto\"\x8f\x01\n\x08TaskName\x12\x1b\n\x03org\x18\x01 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x03org\x12#\n\x07project\x18\x02 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x07project\x12!\n\x06\x64omain\x18\x03 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x06\x64omain\x12\x1e\n\x04name\x18\x04 \x01(\tB\n\xfa\x42\x07r\x05\x10\x01\x18\xff\x01R\x04name\"\xba\x01\n\x0eTaskIdentifier\x12\x1b\n\x03org\x18\x01 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x03org\x12#\n\x07project\x18\x02 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x07project\x12!\n\x06\x64omain\x18\x03 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x06\x64omain\x12\x1e\n\x04name\x18\x04 \x01(\tB\n\xfa\x42\x07r\x05\x10\x01\x18\xff\x01R\x04name\x12#\n\x07version\x18\x05 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18?R\x07version\"\xed\x01\n\x0cTaskMetadata\x12L\n\x0b\x64\x65ployed_by\x18\x01 \x01(\x0b\x32!.cloudidl.common.EnrichedIdentityB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\ndeployedBy\x12\x1d\n\nshort_name\x18\x02 \x01(\tR\tshortName\x12\x45\n\x0b\x64\x65ployed_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x08\xfa\x42\x05\xb2\x01\x02\x08\x01R\ndeployedAt\x12)\n\x10\x65nvironment_name\x18\x04 \x01(\tR\x0f\x65nvironmentName\"\x93\x01\n\x04Task\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\x12\x45\n\x08metadata\x18\x02 \x01(\x0b\x32\x1f.cloudidl.workflow.TaskMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08metadata\"\x8a\x02\n\x08TaskSpec\x12J\n\rtask_template\x18\x01 \x01(\x0b\x32\x1b.flyteidl.core.TaskTemplateB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x0ctaskTemplate\x12H\n\x0e\x64\x65\x66\x61ult_inputs\x18\x02 \x03(\x0b\x32!.cloudidl.workflow.NamedParameterR\rdefaultInputs\x12&\n\nshort_name\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x18?R\tshortName\x12@\n\x0b\x65nvironment\x18\x04 \x01(\x0b\x32\x1e.cloudidl.workflow.EnvironmentR\x0b\x65nvironment\"\xd5\x01\n\x0bTaskDetails\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\x12\x45\n\x08metadata\x18\x02 \x01(\x0b\x32\x1f.cloudidl.workflow.TaskMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08metadata\x12\x39\n\x04spec\x18\x03 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x04specB\xc0\x01\n\x15\x63om.cloudidl.workflowB\x13TaskDefinitionProtoH\x02P\x01Z+github.com/unionai/cloud/gen/pb-go/workflow\xa2\x02\x03\x43WX\xaa\x02\x11\x43loudidl.Workflow\xca\x02\x11\x43loudidl\\Workflow\xe2\x02\x1d\x43loudidl\\Workflow\\GPBMetadata\xea\x02\x12\x43loudidl::Workflowb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'workflow.task_definition_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\023TaskDefinitionProtoH\002P\001Z+github.com/unionai/cloud/gen/pb-go/workflow\242\002\003CWX\252\002\021Cloudidl.Workflow\312\002\021Cloudidl\\Workflow\342\002\035Cloudidl\\Workflow\\GPBMetadata\352\002\022Cloudidl::Workflow'
+  _TASKNAME.fields_by_name['org']._options = None
+  _TASKNAME.fields_by_name['org']._serialized_options = b'\372B\006r\004\020\001\030?'
+  _TASKNAME.fields_by_name['project']._options = None
+  _TASKNAME.fields_by_name['project']._serialized_options = b'\372B\006r\004\020\001\030?'
+  _TASKNAME.fields_by_name['domain']._options = None
+  _TASKNAME.fields_by_name['domain']._serialized_options = b'\372B\006r\004\020\001\030?'
+  _TASKNAME.fields_by_name['name']._options = None
+  _TASKNAME.fields_by_name['name']._serialized_options = b'\372B\007r\005\020\001\030\377\001'
+  _TASKIDENTIFIER.fields_by_name['org']._options = None
+  _TASKIDENTIFIER.fields_by_name['org']._serialized_options = b'\372B\006r\004\020\001\030?'
+  _TASKIDENTIFIER.fields_by_name['project']._options = None
+  _TASKIDENTIFIER.fields_by_name['project']._serialized_options = b'\372B\006r\004\020\001\030?'
+  _TASKIDENTIFIER.fields_by_name['domain']._options = None
+  _TASKIDENTIFIER.fields_by_name['domain']._serialized_options = b'\372B\006r\004\020\001\030?'
+  _TASKIDENTIFIER.fields_by_name['name']._options = None
+  _TASKIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\007r\005\020\001\030\377\001'
+  _TASKIDENTIFIER.fields_by_name['version']._options = None
+  _TASKIDENTIFIER.fields_by_name['version']._serialized_options = b'\372B\006r\004\020\001\030?'
+  _TASKMETADATA.fields_by_name['deployed_by']._options = None
+  _TASKMETADATA.fields_by_name['deployed_by']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _TASKMETADATA.fields_by_name['deployed_at']._options = None
+  _TASKMETADATA.fields_by_name['deployed_at']._serialized_options = b'\372B\005\262\001\002\010\001'
+  _TASK.fields_by_name['task_id']._options = None
+  _TASK.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _TASK.fields_by_name['metadata']._options = None
+  _TASK.fields_by_name['metadata']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _TASKSPEC.fields_by_name['task_template']._options = None
+  _TASKSPEC.fields_by_name['task_template']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _TASKSPEC.fields_by_name['short_name']._options = None
+  _TASKSPEC.fields_by_name['short_name']._serialized_options = b'\372B\004r\002\030?'
+  _TASKDETAILS.fields_by_name['task_id']._options = None
+  _TASKDETAILS.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _TASKDETAILS.fields_by_name['metadata']._options = None
+  _TASKDETAILS.fields_by_name['metadata']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _TASKDETAILS.fields_by_name['spec']._options = None
+  _TASKDETAILS.fields_by_name['spec']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _globals['_TASKNAME']._serialized_start=238
+  _globals['_TASKNAME']._serialized_end=381
+  _globals['_TASKIDENTIFIER']._serialized_start=384
+  _globals['_TASKIDENTIFIER']._serialized_end=570
+  _globals['_TASKMETADATA']._serialized_start=573
+  _globals['_TASKMETADATA']._serialized_end=810
+  _globals['_TASK']._serialized_start=813
+  _globals['_TASK']._serialized_end=960
+  _globals['_TASKSPEC']._serialized_start=963
+  _globals['_TASKSPEC']._serialized_end=1229
+  _globals['_TASKDETAILS']._serialized_start=1232
+  _globals['_TASKDETAILS']._serialized_end=1445
+# @@protoc_insertion_point(module_scope)

flyte/_protos/workflow/task_definition_pb2.pyi

@@ -0,0 +1,81 @@
+from flyte._protos.common import identifier_pb2 as _identifier_pb2
+from flyte._protos.common import identity_pb2 as _identity_pb2
+from flyteidl.core import tasks_pb2 as _tasks_pb2
+from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from flyte._protos.validate.validate import validate_pb2 as _validate_pb2
+from flyte._protos.workflow import common_pb2 as _common_pb2
+from flyte._protos.workflow import environment_pb2 as _environment_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 TaskName(_message.Message):
+    __slots__ = ["org", "project", "domain", "name"]
+    ORG_FIELD_NUMBER: _ClassVar[int]
+    PROJECT_FIELD_NUMBER: _ClassVar[int]
+    DOMAIN_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    org: str
+    project: str
+    domain: str
+    name: str
+    def __init__(self, org: _Optional[str] = ..., project: _Optional[str] = ..., domain: _Optional[str] = ..., name: _Optional[str] = ...) -> None: ...
+
+class TaskIdentifier(_message.Message):
+    __slots__ = ["org", "project", "domain", "name", "version"]
+    ORG_FIELD_NUMBER: _ClassVar[int]
+    PROJECT_FIELD_NUMBER: _ClassVar[int]
+    DOMAIN_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    VERSION_FIELD_NUMBER: _ClassVar[int]
+    org: str
+    project: str
+    domain: str
+    name: str
+    version: str
+    def __init__(self, org: _Optional[str] = ..., project: _Optional[str] = ..., domain: _Optional[str] = ..., name: _Optional[str] = ..., version: _Optional[str] = ...) -> None: ...
+
+class TaskMetadata(_message.Message):
+    __slots__ = ["deployed_by", "short_name", "deployed_at", "environment_name"]
+    DEPLOYED_BY_FIELD_NUMBER: _ClassVar[int]
+    SHORT_NAME_FIELD_NUMBER: _ClassVar[int]
+    DEPLOYED_AT_FIELD_NUMBER: _ClassVar[int]
+    ENVIRONMENT_NAME_FIELD_NUMBER: _ClassVar[int]
+    deployed_by: _identity_pb2.EnrichedIdentity
+    short_name: str
+    deployed_at: _timestamp_pb2.Timestamp
+    environment_name: str
+    def __init__(self, deployed_by: _Optional[_Union[_identity_pb2.EnrichedIdentity, _Mapping]] = ..., short_name: _Optional[str] = ..., deployed_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., environment_name: _Optional[str] = ...) -> None: ...
+
+class Task(_message.Message):
+    __slots__ = ["task_id", "metadata"]
+    TASK_ID_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    task_id: TaskIdentifier
+    metadata: TaskMetadata
+    def __init__(self, task_id: _Optional[_Union[TaskIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[TaskMetadata, _Mapping]] = ...) -> None: ...
+
+class TaskSpec(_message.Message):
+    __slots__ = ["task_template", "default_inputs", "short_name", "environment"]
+    TASK_TEMPLATE_FIELD_NUMBER: _ClassVar[int]
+    DEFAULT_INPUTS_FIELD_NUMBER: _ClassVar[int]
+    SHORT_NAME_FIELD_NUMBER: _ClassVar[int]
+    ENVIRONMENT_FIELD_NUMBER: _ClassVar[int]
+    task_template: _tasks_pb2.TaskTemplate
+    default_inputs: _containers.RepeatedCompositeFieldContainer[_common_pb2.NamedParameter]
+    short_name: str
+    environment: _environment_pb2.Environment
+    def __init__(self, task_template: _Optional[_Union[_tasks_pb2.TaskTemplate, _Mapping]] = ..., default_inputs: _Optional[_Iterable[_Union[_common_pb2.NamedParameter, _Mapping]]] = ..., short_name: _Optional[str] = ..., environment: _Optional[_Union[_environment_pb2.Environment, _Mapping]] = ...) -> None: ...
+
+class TaskDetails(_message.Message):
+    __slots__ = ["task_id", "metadata", "spec"]
+    TASK_ID_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    SPEC_FIELD_NUMBER: _ClassVar[int]
+    task_id: TaskIdentifier
+    metadata: TaskMetadata
+    spec: TaskSpec
+    def __init__(self, task_id: _Optional[_Union[TaskIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[TaskMetadata, _Mapping]] = ..., spec: _Optional[_Union[TaskSpec, _Mapping]] = ...) -> None: ...

flyte/_protos/workflow/task_definition_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

flyte/_protos/workflow/task_service_pb2.py

@@ -0,0 +1,60 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: workflow/task_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 task_definition_pb2 as workflow_dot_task__definition__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1bworkflow/task_service.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x11\x63ommon/list.proto\x1a\x17validate/validate.proto\x1a\x1eworkflow/task_definition.proto\"\x94\x01\n\x11\x44\x65ployTaskRequest\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\x12\x39\n\x04spec\x18\x02 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x04spec\"\x14\n\x12\x44\x65ployTaskResponse\"]\n\x15GetTaskDetailsRequest\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\"R\n\x16GetTaskDetailsResponse\x12\x38\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32\x1e.cloudidl.workflow.TaskDetailsR\x07\x64\x65tails\"\xdc\x02\n\x10ListTasksRequest\x12\x36\n\x07request\x18\x01 \x01(\x0b\x32\x1c.cloudidl.common.ListRequestR\x07request\x12\x1b\n\x03org\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01H\x00R\x03org\x12M\n\nproject_id\x18\x03 \x01(\x0b\x32\".cloudidl.common.ProjectIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01H\x00R\tprojectId\x12T\n\rknown_filters\x18\x04 \x03(\x0b\x32/.cloudidl.workflow.ListTasksRequest.KnownFilterR\x0cknownFilters\x1a=\n\x0bKnownFilter\x12!\n\x0b\x64\x65ployed_by\x18\x01 \x01(\tH\x00R\ndeployedByB\x0b\n\tfilter_byB\x0f\n\x08scope_by\x12\x03\xf8\x42\x01\"X\n\x11ListTasksResponse\x12-\n\x05tasks\x18\x01 \x03(\x0b\x32\x17.cloudidl.workflow.TaskR\x05tasks\x12\x14\n\x05token\x18\x02 \x01(\tR\x05token2\xb3\x02\n\x0bTaskService\x12[\n\nDeployTask\x12$.cloudidl.workflow.DeployTaskRequest\x1a%.cloudidl.workflow.DeployTaskResponse\"\x00\x12j\n\x0eGetTaskDetails\x12(.cloudidl.workflow.GetTaskDetailsRequest\x1a).cloudidl.workflow.GetTaskDetailsResponse\"\x03\x90\x02\x01\x12[\n\tListTasks\x12#.cloudidl.workflow.ListTasksRequest\x1a$.cloudidl.workflow.ListTasksResponse\"\x03\x90\x02\x01\x42\xbd\x01\n\x15\x63om.cloudidl.workflowB\x10TaskServiceProtoH\x02P\x01Z+github.com/unionai/cloud/gen/pb-go/workflow\xa2\x02\x03\x43WX\xaa\x02\x11\x43loudidl.Workflow\xca\x02\x11\x43loudidl\\Workflow\xe2\x02\x1d\x43loudidl\\Workflow\\GPBMetadata\xea\x02\x12\x43loudidl::Workflowb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'workflow.task_service_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\020TaskServiceProtoH\002P\001Z+github.com/unionai/cloud/gen/pb-go/workflow\242\002\003CWX\252\002\021Cloudidl.Workflow\312\002\021Cloudidl\\Workflow\342\002\035Cloudidl\\Workflow\\GPBMetadata\352\002\022Cloudidl::Workflow'
+  _DEPLOYTASKREQUEST.fields_by_name['task_id']._options = None
+  _DEPLOYTASKREQUEST.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _DEPLOYTASKREQUEST.fields_by_name['spec']._options = None
+  _DEPLOYTASKREQUEST.fields_by_name['spec']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _GETTASKDETAILSREQUEST.fields_by_name['task_id']._options = None
+  _GETTASKDETAILSREQUEST.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _LISTTASKSREQUEST.oneofs_by_name['scope_by']._options = None
+  _LISTTASKSREQUEST.oneofs_by_name['scope_by']._serialized_options = b'\370B\001'
+  _LISTTASKSREQUEST.fields_by_name['org']._options = None
+  _LISTTASKSREQUEST.fields_by_name['org']._serialized_options = b'\372B\004r\002\020\001'
+  _LISTTASKSREQUEST.fields_by_name['project_id']._options = None
+  _LISTTASKSREQUEST.fields_by_name['project_id']._serialized_options = b'\372B\005\212\001\002\020\001'
+  _TASKSERVICE.methods_by_name['GetTaskDetails']._options = None
+  _TASKSERVICE.methods_by_name['GetTaskDetails']._serialized_options = b'\220\002\001'
+  _TASKSERVICE.methods_by_name['ListTasks']._options = None
+  _TASKSERVICE.methods_by_name['ListTasks']._serialized_options = b'\220\002\001'
+  _globals['_DEPLOYTASKREQUEST']._serialized_start=152
+  _globals['_DEPLOYTASKREQUEST']._serialized_end=300
+  _globals['_DEPLOYTASKRESPONSE']._serialized_start=302
+  _globals['_DEPLOYTASKRESPONSE']._serialized_end=322
+  _globals['_GETTASKDETAILSREQUEST']._serialized_start=324
+  _globals['_GETTASKDETAILSREQUEST']._serialized_end=417
+  _globals['_GETTASKDETAILSRESPONSE']._serialized_start=419
+  _globals['_GETTASKDETAILSRESPONSE']._serialized_end=501
+  _globals['_LISTTASKSREQUEST']._serialized_start=504
+  _globals['_LISTTASKSREQUEST']._serialized_end=852
+  _globals['_LISTTASKSREQUEST_KNOWNFILTER']._serialized_start=774
+  _globals['_LISTTASKSREQUEST_KNOWNFILTER']._serialized_end=835
+  _globals['_LISTTASKSRESPONSE']._serialized_start=854
+  _globals['_LISTTASKSRESPONSE']._serialized_end=942
+  _globals['_TASKSERVICE']._serialized_start=945
+  _globals['_TASKSERVICE']._serialized_end=1252
+# @@protoc_insertion_point(module_scope)

flyte/_protos/workflow/task_service_pb2.pyi

@@ -0,0 +1,59 @@
+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 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 DeployTaskRequest(_message.Message):
+    __slots__ = ["task_id", "spec"]
+    TASK_ID_FIELD_NUMBER: _ClassVar[int]
+    SPEC_FIELD_NUMBER: _ClassVar[int]
+    task_id: _task_definition_pb2.TaskIdentifier
+    spec: _task_definition_pb2.TaskSpec
+    def __init__(self, task_id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ..., spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ...) -> None: ...
+
+class DeployTaskResponse(_message.Message):
+    __slots__ = []
+    def __init__(self) -> None: ...
+
+class GetTaskDetailsRequest(_message.Message):
+    __slots__ = ["task_id"]
+    TASK_ID_FIELD_NUMBER: _ClassVar[int]
+    task_id: _task_definition_pb2.TaskIdentifier
+    def __init__(self, task_id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ...) -> None: ...
+
+class GetTaskDetailsResponse(_message.Message):
+    __slots__ = ["details"]
+    DETAILS_FIELD_NUMBER: _ClassVar[int]
+    details: _task_definition_pb2.TaskDetails
+    def __init__(self, details: _Optional[_Union[_task_definition_pb2.TaskDetails, _Mapping]] = ...) -> None: ...
+
+class ListTasksRequest(_message.Message):
+    __slots__ = ["request", "org", "project_id", "known_filters"]
+    class KnownFilter(_message.Message):
+        __slots__ = ["deployed_by"]
+        DEPLOYED_BY_FIELD_NUMBER: _ClassVar[int]
+        deployed_by: str
+        def __init__(self, deployed_by: _Optional[str] = ...) -> None: ...
+    REQUEST_FIELD_NUMBER: _ClassVar[int]
+    ORG_FIELD_NUMBER: _ClassVar[int]
+    PROJECT_ID_FIELD_NUMBER: _ClassVar[int]
+    KNOWN_FILTERS_FIELD_NUMBER: _ClassVar[int]
+    request: _list_pb2.ListRequest
+    org: str
+    project_id: _identifier_pb2.ProjectIdentifier
+    known_filters: _containers.RepeatedCompositeFieldContainer[ListTasksRequest.KnownFilter]
+    def __init__(self, request: _Optional[_Union[_list_pb2.ListRequest, _Mapping]] = ..., org: _Optional[str] = ..., project_id: _Optional[_Union[_identifier_pb2.ProjectIdentifier, _Mapping]] = ..., known_filters: _Optional[_Iterable[_Union[ListTasksRequest.KnownFilter, _Mapping]]] = ...) -> None: ...
+
+class ListTasksResponse(_message.Message):
+    __slots__ = ["tasks", "token"]
+    TASKS_FIELD_NUMBER: _ClassVar[int]
+    TOKEN_FIELD_NUMBER: _ClassVar[int]
+    tasks: _containers.RepeatedCompositeFieldContainer[_task_definition_pb2.Task]
+    token: str
+    def __init__(self, tasks: _Optional[_Iterable[_Union[_task_definition_pb2.Task, _Mapping]]] = ..., token: _Optional[str] = ...) -> None: ...

flyte/_protos/workflow/task_service_pb2_grpc.py

@@ -0,0 +1,138 @@
+# 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 task_service_pb2 as workflow_dot_task__service__pb2
+
+
+class TaskServiceStub(object):
+    """TaskService provides an interface for managing tasks.
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.DeployTask = channel.unary_unary(
+                '/cloudidl.workflow.TaskService/DeployTask',
+                request_serializer=workflow_dot_task__service__pb2.DeployTaskRequest.SerializeToString,
+                response_deserializer=workflow_dot_task__service__pb2.DeployTaskResponse.FromString,
+                )
+        self.GetTaskDetails = channel.unary_unary(
+                '/cloudidl.workflow.TaskService/GetTaskDetails',
+                request_serializer=workflow_dot_task__service__pb2.GetTaskDetailsRequest.SerializeToString,
+                response_deserializer=workflow_dot_task__service__pb2.GetTaskDetailsResponse.FromString,
+                )
+        self.ListTasks = channel.unary_unary(
+                '/cloudidl.workflow.TaskService/ListTasks',
+                request_serializer=workflow_dot_task__service__pb2.ListTasksRequest.SerializeToString,
+                response_deserializer=workflow_dot_task__service__pb2.ListTasksResponse.FromString,
+                )
+
+
+class TaskServiceServicer(object):
+    """TaskService provides an interface for managing tasks.
+    """
+
+    def DeployTask(self, request, context):
+        """Deploy a task.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetTaskDetails(self, request, context):
+        """Get detailed information about a task.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ListTasks(self, request, context):
+        """Lists tasks, one per task name, returning the latest version and who it was deployed by.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_TaskServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'DeployTask': grpc.unary_unary_rpc_method_handler(
+                    servicer.DeployTask,
+                    request_deserializer=workflow_dot_task__service__pb2.DeployTaskRequest.FromString,
+                    response_serializer=workflow_dot_task__service__pb2.DeployTaskResponse.SerializeToString,
+            ),
+            'GetTaskDetails': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetTaskDetails,
+                    request_deserializer=workflow_dot_task__service__pb2.GetTaskDetailsRequest.FromString,
+                    response_serializer=workflow_dot_task__service__pb2.GetTaskDetailsResponse.SerializeToString,
+            ),
+            'ListTasks': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListTasks,
+                    request_deserializer=workflow_dot_task__service__pb2.ListTasksRequest.FromString,
+                    response_serializer=workflow_dot_task__service__pb2.ListTasksResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'cloudidl.workflow.TaskService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class TaskService(object):
+    """TaskService provides an interface for managing tasks.
+    """
+
+    @staticmethod
+    def DeployTask(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.TaskService/DeployTask',
+            workflow_dot_task__service__pb2.DeployTaskRequest.SerializeToString,
+            workflow_dot_task__service__pb2.DeployTaskResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetTaskDetails(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.TaskService/GetTaskDetails',
+            workflow_dot_task__service__pb2.GetTaskDetailsRequest.SerializeToString,
+            workflow_dot_task__service__pb2.GetTaskDetailsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def ListTasks(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.TaskService/ListTasks',
+            workflow_dot_task__service__pb2.ListTasksRequest.SerializeToString,
+            workflow_dot_task__service__pb2.ListTasksResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)