diff options
Diffstat (limited to '.venv/lib/python3.12/site-packages/hatchet_sdk/contracts/dispatcher_pb2_grpc.py')
| -rw-r--r-- | .venv/lib/python3.12/site-packages/hatchet_sdk/contracts/dispatcher_pb2_grpc.py | 621 |
1 files changed, 621 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/hatchet_sdk/contracts/dispatcher_pb2_grpc.py b/.venv/lib/python3.12/site-packages/hatchet_sdk/contracts/dispatcher_pb2_grpc.py new file mode 100644 index 00000000..6b0d8cf3 --- /dev/null +++ b/.venv/lib/python3.12/site-packages/hatchet_sdk/contracts/dispatcher_pb2_grpc.py @@ -0,0 +1,621 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc +import warnings + +from . import dispatcher_pb2 as dispatcher__pb2 + +GRPC_GENERATED_VERSION = '1.64.1' +GRPC_VERSION = grpc.__version__ +EXPECTED_ERROR_RELEASE = '1.65.0' +SCHEDULED_RELEASE_DATE = 'June 25, 2024' +_version_not_supported = False + +try: + from grpc._utilities import first_version_is_lower + _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION) +except ImportError: + _version_not_supported = True + +if _version_not_supported: + warnings.warn( + f'The grpc package installed is at version {GRPC_VERSION},' + + f' but the generated code in dispatcher_pb2_grpc.py depends on' + + f' grpcio>={GRPC_GENERATED_VERSION}.' + + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}' + + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.' + + f' This warning will become an error in {EXPECTED_ERROR_RELEASE},' + + f' scheduled for release on {SCHEDULED_RELEASE_DATE}.', + RuntimeWarning + ) + + +class DispatcherStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.Register = channel.unary_unary( + '/Dispatcher/Register', + request_serializer=dispatcher__pb2.WorkerRegisterRequest.SerializeToString, + response_deserializer=dispatcher__pb2.WorkerRegisterResponse.FromString, + _registered_method=True) + self.Listen = channel.unary_stream( + '/Dispatcher/Listen', + request_serializer=dispatcher__pb2.WorkerListenRequest.SerializeToString, + response_deserializer=dispatcher__pb2.AssignedAction.FromString, + _registered_method=True) + self.ListenV2 = channel.unary_stream( + '/Dispatcher/ListenV2', + request_serializer=dispatcher__pb2.WorkerListenRequest.SerializeToString, + response_deserializer=dispatcher__pb2.AssignedAction.FromString, + _registered_method=True) + self.Heartbeat = channel.unary_unary( + '/Dispatcher/Heartbeat', + request_serializer=dispatcher__pb2.HeartbeatRequest.SerializeToString, + response_deserializer=dispatcher__pb2.HeartbeatResponse.FromString, + _registered_method=True) + self.SubscribeToWorkflowEvents = channel.unary_stream( + '/Dispatcher/SubscribeToWorkflowEvents', + request_serializer=dispatcher__pb2.SubscribeToWorkflowEventsRequest.SerializeToString, + response_deserializer=dispatcher__pb2.WorkflowEvent.FromString, + _registered_method=True) + self.SubscribeToWorkflowRuns = channel.stream_stream( + '/Dispatcher/SubscribeToWorkflowRuns', + request_serializer=dispatcher__pb2.SubscribeToWorkflowRunsRequest.SerializeToString, + response_deserializer=dispatcher__pb2.WorkflowRunEvent.FromString, + _registered_method=True) + self.SendStepActionEvent = channel.unary_unary( + '/Dispatcher/SendStepActionEvent', + request_serializer=dispatcher__pb2.StepActionEvent.SerializeToString, + response_deserializer=dispatcher__pb2.ActionEventResponse.FromString, + _registered_method=True) + self.SendGroupKeyActionEvent = channel.unary_unary( + '/Dispatcher/SendGroupKeyActionEvent', + request_serializer=dispatcher__pb2.GroupKeyActionEvent.SerializeToString, + response_deserializer=dispatcher__pb2.ActionEventResponse.FromString, + _registered_method=True) + self.PutOverridesData = channel.unary_unary( + '/Dispatcher/PutOverridesData', + request_serializer=dispatcher__pb2.OverridesData.SerializeToString, + response_deserializer=dispatcher__pb2.OverridesDataResponse.FromString, + _registered_method=True) + self.Unsubscribe = channel.unary_unary( + '/Dispatcher/Unsubscribe', + request_serializer=dispatcher__pb2.WorkerUnsubscribeRequest.SerializeToString, + response_deserializer=dispatcher__pb2.WorkerUnsubscribeResponse.FromString, + _registered_method=True) + self.RefreshTimeout = channel.unary_unary( + '/Dispatcher/RefreshTimeout', + request_serializer=dispatcher__pb2.RefreshTimeoutRequest.SerializeToString, + response_deserializer=dispatcher__pb2.RefreshTimeoutResponse.FromString, + _registered_method=True) + self.ReleaseSlot = channel.unary_unary( + '/Dispatcher/ReleaseSlot', + request_serializer=dispatcher__pb2.ReleaseSlotRequest.SerializeToString, + response_deserializer=dispatcher__pb2.ReleaseSlotResponse.FromString, + _registered_method=True) + self.UpsertWorkerLabels = channel.unary_unary( + '/Dispatcher/UpsertWorkerLabels', + request_serializer=dispatcher__pb2.UpsertWorkerLabelsRequest.SerializeToString, + response_deserializer=dispatcher__pb2.UpsertWorkerLabelsResponse.FromString, + _registered_method=True) + + +class DispatcherServicer(object): + """Missing associated documentation comment in .proto file.""" + + def Register(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 Listen(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 ListenV2(self, request, context): + """ListenV2 is like listen, but implementation does not include heartbeats. This should only used by SDKs + against engine version v0.18.1+ + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def Heartbeat(self, request, context): + """Heartbeat is a method for workers to send heartbeats to the dispatcher + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def SubscribeToWorkflowEvents(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 SubscribeToWorkflowRuns(self, request_iterator, 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 SendStepActionEvent(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 SendGroupKeyActionEvent(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 PutOverridesData(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 Unsubscribe(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 RefreshTimeout(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 ReleaseSlot(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 UpsertWorkerLabels(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_DispatcherServicer_to_server(servicer, server): + rpc_method_handlers = { + 'Register': grpc.unary_unary_rpc_method_handler( + servicer.Register, + request_deserializer=dispatcher__pb2.WorkerRegisterRequest.FromString, + response_serializer=dispatcher__pb2.WorkerRegisterResponse.SerializeToString, + ), + 'Listen': grpc.unary_stream_rpc_method_handler( + servicer.Listen, + request_deserializer=dispatcher__pb2.WorkerListenRequest.FromString, + response_serializer=dispatcher__pb2.AssignedAction.SerializeToString, + ), + 'ListenV2': grpc.unary_stream_rpc_method_handler( + servicer.ListenV2, + request_deserializer=dispatcher__pb2.WorkerListenRequest.FromString, + response_serializer=dispatcher__pb2.AssignedAction.SerializeToString, + ), + 'Heartbeat': grpc.unary_unary_rpc_method_handler( + servicer.Heartbeat, + request_deserializer=dispatcher__pb2.HeartbeatRequest.FromString, + response_serializer=dispatcher__pb2.HeartbeatResponse.SerializeToString, + ), + 'SubscribeToWorkflowEvents': grpc.unary_stream_rpc_method_handler( + servicer.SubscribeToWorkflowEvents, + request_deserializer=dispatcher__pb2.SubscribeToWorkflowEventsRequest.FromString, + response_serializer=dispatcher__pb2.WorkflowEvent.SerializeToString, + ), + 'SubscribeToWorkflowRuns': grpc.stream_stream_rpc_method_handler( + servicer.SubscribeToWorkflowRuns, + request_deserializer=dispatcher__pb2.SubscribeToWorkflowRunsRequest.FromString, + response_serializer=dispatcher__pb2.WorkflowRunEvent.SerializeToString, + ), + 'SendStepActionEvent': grpc.unary_unary_rpc_method_handler( + servicer.SendStepActionEvent, + request_deserializer=dispatcher__pb2.StepActionEvent.FromString, + response_serializer=dispatcher__pb2.ActionEventResponse.SerializeToString, + ), + 'SendGroupKeyActionEvent': grpc.unary_unary_rpc_method_handler( + servicer.SendGroupKeyActionEvent, + request_deserializer=dispatcher__pb2.GroupKeyActionEvent.FromString, + response_serializer=dispatcher__pb2.ActionEventResponse.SerializeToString, + ), + 'PutOverridesData': grpc.unary_unary_rpc_method_handler( + servicer.PutOverridesData, + request_deserializer=dispatcher__pb2.OverridesData.FromString, + response_serializer=dispatcher__pb2.OverridesDataResponse.SerializeToString, + ), + 'Unsubscribe': grpc.unary_unary_rpc_method_handler( + servicer.Unsubscribe, + request_deserializer=dispatcher__pb2.WorkerUnsubscribeRequest.FromString, + response_serializer=dispatcher__pb2.WorkerUnsubscribeResponse.SerializeToString, + ), + 'RefreshTimeout': grpc.unary_unary_rpc_method_handler( + servicer.RefreshTimeout, + request_deserializer=dispatcher__pb2.RefreshTimeoutRequest.FromString, + response_serializer=dispatcher__pb2.RefreshTimeoutResponse.SerializeToString, + ), + 'ReleaseSlot': grpc.unary_unary_rpc_method_handler( + servicer.ReleaseSlot, + request_deserializer=dispatcher__pb2.ReleaseSlotRequest.FromString, + response_serializer=dispatcher__pb2.ReleaseSlotResponse.SerializeToString, + ), + 'UpsertWorkerLabels': grpc.unary_unary_rpc_method_handler( + servicer.UpsertWorkerLabels, + request_deserializer=dispatcher__pb2.UpsertWorkerLabelsRequest.FromString, + response_serializer=dispatcher__pb2.UpsertWorkerLabelsResponse.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'Dispatcher', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + server.add_registered_method_handlers('Dispatcher', rpc_method_handlers) + + + # This class is part of an EXPERIMENTAL API. +class Dispatcher(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def Register(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, + '/Dispatcher/Register', + dispatcher__pb2.WorkerRegisterRequest.SerializeToString, + dispatcher__pb2.WorkerRegisterResponse.FromString, + options, + channel_credentials, + insecure, + call_credentials, + compression, + wait_for_ready, + timeout, + metadata, + _registered_method=True) + + @staticmethod + def Listen(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, + '/Dispatcher/Listen', + dispatcher__pb2.WorkerListenRequest.SerializeToString, + dispatcher__pb2.AssignedAction.FromString, + options, + channel_credentials, + insecure, + call_credentials, + compression, + wait_for_ready, + timeout, + metadata, + _registered_method=True) + + @staticmethod + def ListenV2(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, + '/Dispatcher/ListenV2', + dispatcher__pb2.WorkerListenRequest.SerializeToString, + dispatcher__pb2.AssignedAction.FromString, + options, + channel_credentials, + insecure, + call_credentials, + compression, + wait_for_ready, + timeout, + metadata, + _registered_method=True) + + @staticmethod + def Heartbeat(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, + '/Dispatcher/Heartbeat', + dispatcher__pb2.HeartbeatRequest.SerializeToString, + dispatcher__pb2.HeartbeatResponse.FromString, + options, + channel_credentials, + insecure, + call_credentials, + compression, + wait_for_ready, + timeout, + metadata, + _registered_method=True) + + @staticmethod + def SubscribeToWorkflowEvents(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, + '/Dispatcher/SubscribeToWorkflowEvents', + dispatcher__pb2.SubscribeToWorkflowEventsRequest.SerializeToString, + dispatcher__pb2.WorkflowEvent.FromString, + options, + channel_credentials, + insecure, + call_credentials, + compression, + wait_for_ready, + timeout, + metadata, + _registered_method=True) + + @staticmethod + def SubscribeToWorkflowRuns(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, + '/Dispatcher/SubscribeToWorkflowRuns', + dispatcher__pb2.SubscribeToWorkflowRunsRequest.SerializeToString, + dispatcher__pb2.WorkflowRunEvent.FromString, + options, + channel_credentials, + insecure, + call_credentials, + compression, + wait_for_ready, + timeout, + metadata, + _registered_method=True) + + @staticmethod + def SendStepActionEvent(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, + '/Dispatcher/SendStepActionEvent', + dispatcher__pb2.StepActionEvent.SerializeToString, + dispatcher__pb2.ActionEventResponse.FromString, + options, + channel_credentials, + insecure, + call_credentials, + compression, + wait_for_ready, + timeout, + metadata, + _registered_method=True) + + @staticmethod + def SendGroupKeyActionEvent(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, + '/Dispatcher/SendGroupKeyActionEvent', + dispatcher__pb2.GroupKeyActionEvent.SerializeToString, + dispatcher__pb2.ActionEventResponse.FromString, + options, + channel_credentials, + insecure, + call_credentials, + compression, + wait_for_ready, + timeout, + metadata, + _registered_method=True) + + @staticmethod + def PutOverridesData(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, + '/Dispatcher/PutOverridesData', + dispatcher__pb2.OverridesData.SerializeToString, + dispatcher__pb2.OverridesDataResponse.FromString, + options, + channel_credentials, + insecure, + call_credentials, + compression, + wait_for_ready, + timeout, + metadata, + _registered_method=True) + + @staticmethod + def Unsubscribe(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, + '/Dispatcher/Unsubscribe', + dispatcher__pb2.WorkerUnsubscribeRequest.SerializeToString, + dispatcher__pb2.WorkerUnsubscribeResponse.FromString, + options, + channel_credentials, + insecure, + call_credentials, + compression, + wait_for_ready, + timeout, + metadata, + _registered_method=True) + + @staticmethod + def RefreshTimeout(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, + '/Dispatcher/RefreshTimeout', + dispatcher__pb2.RefreshTimeoutRequest.SerializeToString, + dispatcher__pb2.RefreshTimeoutResponse.FromString, + options, + channel_credentials, + insecure, + call_credentials, + compression, + wait_for_ready, + timeout, + metadata, + _registered_method=True) + + @staticmethod + def ReleaseSlot(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, + '/Dispatcher/ReleaseSlot', + dispatcher__pb2.ReleaseSlotRequest.SerializeToString, + dispatcher__pb2.ReleaseSlotResponse.FromString, + options, + channel_credentials, + insecure, + call_credentials, + compression, + wait_for_ready, + timeout, + metadata, + _registered_method=True) + + @staticmethod + def UpsertWorkerLabels(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, + '/Dispatcher/UpsertWorkerLabels', + dispatcher__pb2.UpsertWorkerLabelsRequest.SerializeToString, + dispatcher__pb2.UpsertWorkerLabelsResponse.FromString, + options, + channel_credentials, + insecure, + call_credentials, + compression, + wait_for_ready, + timeout, + metadata, + _registered_method=True) |