hatchet-sdk 1.0.0__py3-none-any.whl → 1.0.0a1__py3-none-any.whl

hatchet_sdk/contracts/v1/workflows_pb2_grpc.py

@@ -0,0 +1,234 @@
+# 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 hatchet_sdk.contracts.v1 import workflows_pb2 as v1_dot_workflows__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 v1/workflows_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 AdminServiceStub(object):
+    """AdminService represents a set of RPCs for admin management of tasks, workflows, etc.
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.PutWorkflow = channel.unary_unary(
+                '/v1.AdminService/PutWorkflow',
+                request_serializer=v1_dot_workflows__pb2.CreateWorkflowVersionRequest.SerializeToString,
+                response_deserializer=v1_dot_workflows__pb2.CreateWorkflowVersionResponse.FromString,
+                _registered_method=True)
+        self.CancelTasks = channel.unary_unary(
+                '/v1.AdminService/CancelTasks',
+                request_serializer=v1_dot_workflows__pb2.CancelTasksRequest.SerializeToString,
+                response_deserializer=v1_dot_workflows__pb2.CancelTasksResponse.FromString,
+                _registered_method=True)
+        self.ReplayTasks = channel.unary_unary(
+                '/v1.AdminService/ReplayTasks',
+                request_serializer=v1_dot_workflows__pb2.ReplayTasksRequest.SerializeToString,
+                response_deserializer=v1_dot_workflows__pb2.ReplayTasksResponse.FromString,
+                _registered_method=True)
+        self.TriggerWorkflowRun = channel.unary_unary(
+                '/v1.AdminService/TriggerWorkflowRun',
+                request_serializer=v1_dot_workflows__pb2.TriggerWorkflowRunRequest.SerializeToString,
+                response_deserializer=v1_dot_workflows__pb2.TriggerWorkflowRunResponse.FromString,
+                _registered_method=True)
+
+
+class AdminServiceServicer(object):
+    """AdminService represents a set of RPCs for admin management of tasks, workflows, etc.
+    """
+
+    def PutWorkflow(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 CancelTasks(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 ReplayTasks(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 TriggerWorkflowRun(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_AdminServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'PutWorkflow': grpc.unary_unary_rpc_method_handler(
+                    servicer.PutWorkflow,
+                    request_deserializer=v1_dot_workflows__pb2.CreateWorkflowVersionRequest.FromString,
+                    response_serializer=v1_dot_workflows__pb2.CreateWorkflowVersionResponse.SerializeToString,
+            ),
+            'CancelTasks': grpc.unary_unary_rpc_method_handler(
+                    servicer.CancelTasks,
+                    request_deserializer=v1_dot_workflows__pb2.CancelTasksRequest.FromString,
+                    response_serializer=v1_dot_workflows__pb2.CancelTasksResponse.SerializeToString,
+            ),
+            'ReplayTasks': grpc.unary_unary_rpc_method_handler(
+                    servicer.ReplayTasks,
+                    request_deserializer=v1_dot_workflows__pb2.ReplayTasksRequest.FromString,
+                    response_serializer=v1_dot_workflows__pb2.ReplayTasksResponse.SerializeToString,
+            ),
+            'TriggerWorkflowRun': grpc.unary_unary_rpc_method_handler(
+                    servicer.TriggerWorkflowRun,
+                    request_deserializer=v1_dot_workflows__pb2.TriggerWorkflowRunRequest.FromString,
+                    response_serializer=v1_dot_workflows__pb2.TriggerWorkflowRunResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'v1.AdminService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('v1.AdminService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class AdminService(object):
+    """AdminService represents a set of RPCs for admin management of tasks, workflows, etc.
+    """
+
+    @staticmethod
+    def PutWorkflow(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,
+            '/v1.AdminService/PutWorkflow',
+            v1_dot_workflows__pb2.CreateWorkflowVersionRequest.SerializeToString,
+            v1_dot_workflows__pb2.CreateWorkflowVersionResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def CancelTasks(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,
+            '/v1.AdminService/CancelTasks',
+            v1_dot_workflows__pb2.CancelTasksRequest.SerializeToString,
+            v1_dot_workflows__pb2.CancelTasksResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def ReplayTasks(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,
+            '/v1.AdminService/ReplayTasks',
+            v1_dot_workflows__pb2.ReplayTasksRequest.SerializeToString,
+            v1_dot_workflows__pb2.ReplayTasksResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def TriggerWorkflowRun(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,
+            '/v1.AdminService/TriggerWorkflowRun',
+            v1_dot_workflows__pb2.TriggerWorkflowRunRequest.SerializeToString,
+            v1_dot_workflows__pb2.TriggerWorkflowRunResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

hatchet_sdk/contracts/workflows_pb2_grpc.py

@@ -3,7 +3,7 @@
 import grpc
 import warnings
 
-from . import workflows_pb2 as workflows__pb2
+from hatchet_sdk.contracts import workflows_pb2 as workflows__pb2
 
 GRPC_GENERATED_VERSION = '1.64.1'
 GRPC_VERSION = grpc.__version__

hatchet_sdk/features/cron.py

@@ -14,7 +14,7 @@ from hatchet_sdk.clients.rest.models.workflow_run_order_by_direction import (
 from hatchet_sdk.utils.typing import JSONSerializableMapping
 
 
-class CreateCronTriggerJSONSerializableMapping(BaseModel):
+class CreateCronTriggerConfig(BaseModel):
     """
     Schema for creating a workflow run triggered by a cron.
 
@@ -103,7 +103,7 @@ class CronClient:
         Returns:
             CronWorkflows: The created cron workflow instance.
         """
-        validated_input = CreateCronTriggerJSONSerializableMapping(
+        validated_input = CreateCronTriggerConfig(
             expression=expression, input=input, additional_metadata=additional_metadata
         )
 
@@ -197,7 +197,7 @@ class CronClient:
         Returns:
             CronWorkflows: The created cron workflow instance.
         """
-        validated_input = CreateCronTriggerJSONSerializableMapping(
+        validated_input = CreateCronTriggerConfig(
             expression=expression, input=input, additional_metadata=additional_metadata
         )
 

hatchet_sdk/features/scheduled.py

@@ -20,7 +20,7 @@ from hatchet_sdk.clients.rest.models.workflow_run_order_by_direction import (
 from hatchet_sdk.utils.typing import JSONSerializableMapping
 
 
-class CreateScheduledTriggerJSONSerializableMapping(BaseModel):
+class CreateScheduledTriggerConfig(BaseModel):
     """
     Schema for creating a scheduled workflow run.
 
@@ -75,7 +75,7 @@ class ScheduledClient:
             ScheduledWorkflows: The created scheduled workflow instance.
         """
 
-        validated_input = CreateScheduledTriggerJSONSerializableMapping(
+        validated_input = CreateScheduledTriggerConfig(
             trigger_at=trigger_at, input=input, additional_metadata=additional_metadata
         )