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

hatchet_sdk/contracts/v1/workflows_pb2.pyi

@@ -0,0 +1,228 @@
+from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from hatchet_sdk.contracts.v1.shared import condition_pb2 as _condition_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+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 StickyStrategy(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    SOFT: _ClassVar[StickyStrategy]
+    HARD: _ClassVar[StickyStrategy]
+
+class RateLimitDuration(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    SECOND: _ClassVar[RateLimitDuration]
+    MINUTE: _ClassVar[RateLimitDuration]
+    HOUR: _ClassVar[RateLimitDuration]
+    DAY: _ClassVar[RateLimitDuration]
+    WEEK: _ClassVar[RateLimitDuration]
+    MONTH: _ClassVar[RateLimitDuration]
+    YEAR: _ClassVar[RateLimitDuration]
+
+class ConcurrencyLimitStrategy(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    CANCEL_IN_PROGRESS: _ClassVar[ConcurrencyLimitStrategy]
+    DROP_NEWEST: _ClassVar[ConcurrencyLimitStrategy]
+    QUEUE_NEWEST: _ClassVar[ConcurrencyLimitStrategy]
+    GROUP_ROUND_ROBIN: _ClassVar[ConcurrencyLimitStrategy]
+    CANCEL_NEWEST: _ClassVar[ConcurrencyLimitStrategy]
+
+class WorkerLabelComparator(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    EQUAL: _ClassVar[WorkerLabelComparator]
+    NOT_EQUAL: _ClassVar[WorkerLabelComparator]
+    GREATER_THAN: _ClassVar[WorkerLabelComparator]
+    GREATER_THAN_OR_EQUAL: _ClassVar[WorkerLabelComparator]
+    LESS_THAN: _ClassVar[WorkerLabelComparator]
+    LESS_THAN_OR_EQUAL: _ClassVar[WorkerLabelComparator]
+SOFT: StickyStrategy
+HARD: StickyStrategy
+SECOND: RateLimitDuration
+MINUTE: RateLimitDuration
+HOUR: RateLimitDuration
+DAY: RateLimitDuration
+WEEK: RateLimitDuration
+MONTH: RateLimitDuration
+YEAR: RateLimitDuration
+CANCEL_IN_PROGRESS: ConcurrencyLimitStrategy
+DROP_NEWEST: ConcurrencyLimitStrategy
+QUEUE_NEWEST: ConcurrencyLimitStrategy
+GROUP_ROUND_ROBIN: ConcurrencyLimitStrategy
+CANCEL_NEWEST: ConcurrencyLimitStrategy
+EQUAL: WorkerLabelComparator
+NOT_EQUAL: WorkerLabelComparator
+GREATER_THAN: WorkerLabelComparator
+GREATER_THAN_OR_EQUAL: WorkerLabelComparator
+LESS_THAN: WorkerLabelComparator
+LESS_THAN_OR_EQUAL: WorkerLabelComparator
+
+class CancelTasksRequest(_message.Message):
+    __slots__ = ("externalIds", "filter")
+    EXTERNALIDS_FIELD_NUMBER: _ClassVar[int]
+    FILTER_FIELD_NUMBER: _ClassVar[int]
+    externalIds: _containers.RepeatedScalarFieldContainer[str]
+    filter: TasksFilter
+    def __init__(self, externalIds: _Optional[_Iterable[str]] = ..., filter: _Optional[_Union[TasksFilter, _Mapping]] = ...) -> None: ...
+
+class ReplayTasksRequest(_message.Message):
+    __slots__ = ("externalIds", "filter")
+    EXTERNALIDS_FIELD_NUMBER: _ClassVar[int]
+    FILTER_FIELD_NUMBER: _ClassVar[int]
+    externalIds: _containers.RepeatedScalarFieldContainer[str]
+    filter: TasksFilter
+    def __init__(self, externalIds: _Optional[_Iterable[str]] = ..., filter: _Optional[_Union[TasksFilter, _Mapping]] = ...) -> None: ...
+
+class TasksFilter(_message.Message):
+    __slots__ = ("statuses", "since", "until", "workflow_ids", "additional_metadata")
+    STATUSES_FIELD_NUMBER: _ClassVar[int]
+    SINCE_FIELD_NUMBER: _ClassVar[int]
+    UNTIL_FIELD_NUMBER: _ClassVar[int]
+    WORKFLOW_IDS_FIELD_NUMBER: _ClassVar[int]
+    ADDITIONAL_METADATA_FIELD_NUMBER: _ClassVar[int]
+    statuses: _containers.RepeatedScalarFieldContainer[str]
+    since: _timestamp_pb2.Timestamp
+    until: _timestamp_pb2.Timestamp
+    workflow_ids: _containers.RepeatedScalarFieldContainer[str]
+    additional_metadata: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, statuses: _Optional[_Iterable[str]] = ..., since: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., until: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., workflow_ids: _Optional[_Iterable[str]] = ..., additional_metadata: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class CancelTasksResponse(_message.Message):
+    __slots__ = ("cancelled_tasks",)
+    CANCELLED_TASKS_FIELD_NUMBER: _ClassVar[int]
+    cancelled_tasks: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, cancelled_tasks: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class ReplayTasksResponse(_message.Message):
+    __slots__ = ("replayed_tasks",)
+    REPLAYED_TASKS_FIELD_NUMBER: _ClassVar[int]
+    replayed_tasks: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, replayed_tasks: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class TriggerWorkflowRunRequest(_message.Message):
+    __slots__ = ("workflow_name", "input", "additional_metadata")
+    WORKFLOW_NAME_FIELD_NUMBER: _ClassVar[int]
+    INPUT_FIELD_NUMBER: _ClassVar[int]
+    ADDITIONAL_METADATA_FIELD_NUMBER: _ClassVar[int]
+    workflow_name: str
+    input: bytes
+    additional_metadata: bytes
+    def __init__(self, workflow_name: _Optional[str] = ..., input: _Optional[bytes] = ..., additional_metadata: _Optional[bytes] = ...) -> None: ...
+
+class TriggerWorkflowRunResponse(_message.Message):
+    __slots__ = ("external_id",)
+    EXTERNAL_ID_FIELD_NUMBER: _ClassVar[int]
+    external_id: str
+    def __init__(self, external_id: _Optional[str] = ...) -> None: ...
+
+class CreateWorkflowVersionRequest(_message.Message):
+    __slots__ = ("name", "description", "version", "event_triggers", "cron_triggers", "tasks", "concurrency", "cron_input", "on_failure_task", "sticky")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
+    VERSION_FIELD_NUMBER: _ClassVar[int]
+    EVENT_TRIGGERS_FIELD_NUMBER: _ClassVar[int]
+    CRON_TRIGGERS_FIELD_NUMBER: _ClassVar[int]
+    TASKS_FIELD_NUMBER: _ClassVar[int]
+    CONCURRENCY_FIELD_NUMBER: _ClassVar[int]
+    CRON_INPUT_FIELD_NUMBER: _ClassVar[int]
+    ON_FAILURE_TASK_FIELD_NUMBER: _ClassVar[int]
+    STICKY_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    description: str
+    version: str
+    event_triggers: _containers.RepeatedScalarFieldContainer[str]
+    cron_triggers: _containers.RepeatedScalarFieldContainer[str]
+    tasks: _containers.RepeatedCompositeFieldContainer[CreateTaskOpts]
+    concurrency: Concurrency
+    cron_input: str
+    on_failure_task: CreateTaskOpts
+    sticky: StickyStrategy
+    def __init__(self, name: _Optional[str] = ..., description: _Optional[str] = ..., version: _Optional[str] = ..., event_triggers: _Optional[_Iterable[str]] = ..., cron_triggers: _Optional[_Iterable[str]] = ..., tasks: _Optional[_Iterable[_Union[CreateTaskOpts, _Mapping]]] = ..., concurrency: _Optional[_Union[Concurrency, _Mapping]] = ..., cron_input: _Optional[str] = ..., on_failure_task: _Optional[_Union[CreateTaskOpts, _Mapping]] = ..., sticky: _Optional[_Union[StickyStrategy, str]] = ...) -> None: ...
+
+class Concurrency(_message.Message):
+    __slots__ = ("expression", "max_runs", "limit_strategy")
+    EXPRESSION_FIELD_NUMBER: _ClassVar[int]
+    MAX_RUNS_FIELD_NUMBER: _ClassVar[int]
+    LIMIT_STRATEGY_FIELD_NUMBER: _ClassVar[int]
+    expression: str
+    max_runs: int
+    limit_strategy: ConcurrencyLimitStrategy
+    def __init__(self, expression: _Optional[str] = ..., max_runs: _Optional[int] = ..., limit_strategy: _Optional[_Union[ConcurrencyLimitStrategy, str]] = ...) -> None: ...
+
+class DesiredWorkerLabels(_message.Message):
+    __slots__ = ("strValue", "intValue", "required", "comparator", "weight")
+    STRVALUE_FIELD_NUMBER: _ClassVar[int]
+    INTVALUE_FIELD_NUMBER: _ClassVar[int]
+    REQUIRED_FIELD_NUMBER: _ClassVar[int]
+    COMPARATOR_FIELD_NUMBER: _ClassVar[int]
+    WEIGHT_FIELD_NUMBER: _ClassVar[int]
+    strValue: str
+    intValue: int
+    required: bool
+    comparator: WorkerLabelComparator
+    weight: int
+    def __init__(self, strValue: _Optional[str] = ..., intValue: _Optional[int] = ..., required: bool = ..., comparator: _Optional[_Union[WorkerLabelComparator, str]] = ..., weight: _Optional[int] = ...) -> None: ...
+
+class CreateTaskOpts(_message.Message):
+    __slots__ = ("readable_id", "action", "timeout", "inputs", "parents", "retries", "rate_limits", "worker_labels", "backoff_factor", "backoff_max_seconds", "concurrency", "conditions", "schedule_timeout")
+    class WorkerLabelsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: DesiredWorkerLabels
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[DesiredWorkerLabels, _Mapping]] = ...) -> None: ...
+    READABLE_ID_FIELD_NUMBER: _ClassVar[int]
+    ACTION_FIELD_NUMBER: _ClassVar[int]
+    TIMEOUT_FIELD_NUMBER: _ClassVar[int]
+    INPUTS_FIELD_NUMBER: _ClassVar[int]
+    PARENTS_FIELD_NUMBER: _ClassVar[int]
+    RETRIES_FIELD_NUMBER: _ClassVar[int]
+    RATE_LIMITS_FIELD_NUMBER: _ClassVar[int]
+    WORKER_LABELS_FIELD_NUMBER: _ClassVar[int]
+    BACKOFF_FACTOR_FIELD_NUMBER: _ClassVar[int]
+    BACKOFF_MAX_SECONDS_FIELD_NUMBER: _ClassVar[int]
+    CONCURRENCY_FIELD_NUMBER: _ClassVar[int]
+    CONDITIONS_FIELD_NUMBER: _ClassVar[int]
+    SCHEDULE_TIMEOUT_FIELD_NUMBER: _ClassVar[int]
+    readable_id: str
+    action: str
+    timeout: str
+    inputs: str
+    parents: _containers.RepeatedScalarFieldContainer[str]
+    retries: int
+    rate_limits: _containers.RepeatedCompositeFieldContainer[CreateTaskRateLimit]
+    worker_labels: _containers.MessageMap[str, DesiredWorkerLabels]
+    backoff_factor: float
+    backoff_max_seconds: int
+    concurrency: _containers.RepeatedCompositeFieldContainer[Concurrency]
+    conditions: _condition_pb2.TaskConditions
+    schedule_timeout: str
+    def __init__(self, readable_id: _Optional[str] = ..., action: _Optional[str] = ..., timeout: _Optional[str] = ..., inputs: _Optional[str] = ..., parents: _Optional[_Iterable[str]] = ..., retries: _Optional[int] = ..., rate_limits: _Optional[_Iterable[_Union[CreateTaskRateLimit, _Mapping]]] = ..., worker_labels: _Optional[_Mapping[str, DesiredWorkerLabels]] = ..., backoff_factor: _Optional[float] = ..., backoff_max_seconds: _Optional[int] = ..., concurrency: _Optional[_Iterable[_Union[Concurrency, _Mapping]]] = ..., conditions: _Optional[_Union[_condition_pb2.TaskConditions, _Mapping]] = ..., schedule_timeout: _Optional[str] = ...) -> None: ...
+
+class CreateTaskRateLimit(_message.Message):
+    __slots__ = ("key", "units", "key_expr", "units_expr", "limit_values_expr", "duration")
+    KEY_FIELD_NUMBER: _ClassVar[int]
+    UNITS_FIELD_NUMBER: _ClassVar[int]
+    KEY_EXPR_FIELD_NUMBER: _ClassVar[int]
+    UNITS_EXPR_FIELD_NUMBER: _ClassVar[int]
+    LIMIT_VALUES_EXPR_FIELD_NUMBER: _ClassVar[int]
+    DURATION_FIELD_NUMBER: _ClassVar[int]
+    key: str
+    units: int
+    key_expr: str
+    units_expr: str
+    limit_values_expr: str
+    duration: RateLimitDuration
+    def __init__(self, key: _Optional[str] = ..., units: _Optional[int] = ..., key_expr: _Optional[str] = ..., units_expr: _Optional[str] = ..., limit_values_expr: _Optional[str] = ..., duration: _Optional[_Union[RateLimitDuration, str]] = ...) -> None: ...
+
+class CreateWorkflowVersionResponse(_message.Message):
+    __slots__ = ("id", "workflow_id")
+    ID_FIELD_NUMBER: _ClassVar[int]
+    WORKFLOW_ID_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    workflow_id: str
+    def __init__(self, id: _Optional[str] = ..., workflow_id: _Optional[str] = ...) -> None: ...

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

@@ -1,6 +1,6 @@
-from typing import Union
+from typing import List, Union
 
-from pydantic import BaseModel, field_validator
+from pydantic import BaseModel, Field, field_validator
 
 from hatchet_sdk.client import Client
 from hatchet_sdk.clients.rest.models.cron_workflows import CronWorkflows
@@ -11,9 +11,10 @@ from hatchet_sdk.clients.rest.models.cron_workflows_order_by_field import (
 from hatchet_sdk.clients.rest.models.workflow_run_order_by_direction import (
     WorkflowRunOrderByDirection,
 )
+from hatchet_sdk.utils.typing import JSONSerializableMapping
 
 
-class CreateCronTriggerInput(BaseModel):
+class CreateCronTriggerConfig(BaseModel):
     """
     Schema for creating a workflow run triggered by a cron.
 
@@ -23,12 +24,13 @@ class CreateCronTriggerInput(BaseModel):
         additional_metadata (dict[str, str]): Additional metadata associated with the cron trigger (e.g. {"key1": "value1", "key2": "value2"}).
     """
 
-    expression: str = None
-    input: dict = {}
-    additional_metadata: dict[str, str] = {}
+    expression: str
+    input: JSONSerializableMapping = Field(default_factory=dict)
+    additional_metadata: JSONSerializableMapping = Field(default_factory=dict)
 
     @field_validator("expression")
-    def validate_cron_expression(cls, v):
+    @classmethod
+    def validate_cron_expression(cls, v: str) -> str:
         """
         Validates the cron expression to ensure it adheres to the expected format.
 
@@ -79,15 +81,14 @@ class CronClient:
             _client (Client): The client instance to be used for REST interactions.
         """
         self._client = _client
-        self.aio = CronClientAsync(_client)
 
     def create(
         self,
         workflow_name: str,
         cron_name: str,
         expression: str,
-        input: dict,
-        additional_metadata: dict[str, str],
+        input: JSONSerializableMapping,
+        additional_metadata: JSONSerializableMapping,
     ) -> CronWorkflows:
         """
         Creates a new workflow cron trigger.
@@ -102,7 +103,7 @@ class CronClient:
         Returns:
             CronWorkflows: The created cron workflow instance.
         """
-        validated_input = CreateCronTriggerInput(
+        validated_input = CreateCronTriggerConfig(
             expression=expression, input=input, additional_metadata=additional_metadata
         )
 
@@ -121,10 +122,11 @@ class CronClient:
         Args:
             cron_trigger (Union[str, CronWorkflows]): The cron trigger ID or CronWorkflows instance to delete.
         """
-        id_ = cron_trigger
-        if isinstance(cron_trigger, CronWorkflows):
-            id_ = cron_trigger.metadata.id
-        self._client.rest.cron_delete(id_)
+        self._client.rest.cron_delete(
+            cron_trigger.metadata.id
+            if isinstance(cron_trigger, CronWorkflows)
+            else cron_trigger
+        )
 
     def list(
         self,
@@ -168,38 +170,19 @@ class CronClient:
         Returns:
             CronWorkflows: The requested cron workflow instance.
         """
-        id_ = cron_trigger
-        if isinstance(cron_trigger, CronWorkflows):
-            id_ = cron_trigger.metadata.id
-        return self._client.rest.cron_get(id_)
-
-
-class CronClientAsync:
-    """
-    Asynchronous client for managing workflow cron triggers.
-
-    Attributes:
-        _client (Client): The underlying client used to interact with the REST API asynchronously.
-    """
-
-    _client: Client
-
-    def __init__(self, _client: Client):
-        """
-        Initializes the CronClientAsync with a given Client instance.
-
-        Args:
-            _client (Client): The client instance to be used for asynchronous REST interactions.
-        """
-        self._client = _client
+        return self._client.rest.cron_get(
+            cron_trigger.metadata.id
+            if isinstance(cron_trigger, CronWorkflows)
+            else cron_trigger
+        )
 
-    async def create(
+    async def aio_create(
         self,
         workflow_name: str,
         cron_name: str,
         expression: str,
-        input: dict,
-        additional_metadata: dict[str, str],
+        input: JSONSerializableMapping,
+        additional_metadata: JSONSerializableMapping,
     ) -> CronWorkflows:
         """
         Asynchronously creates a new workflow cron trigger.
@@ -214,11 +197,11 @@ class CronClientAsync:
         Returns:
             CronWorkflows: The created cron workflow instance.
         """
-        validated_input = CreateCronTriggerInput(
+        validated_input = CreateCronTriggerConfig(
             expression=expression, input=input, additional_metadata=additional_metadata
         )
 
-        return await self._client.rest.aio.cron_create(
+        return await self._client.rest.aio_create_cron(
             workflow_name=workflow_name,
             cron_name=cron_name,
             expression=validated_input.expression,
@@ -226,24 +209,25 @@ class CronClientAsync:
             additional_metadata=validated_input.additional_metadata,
         )
 
-    async def delete(self, cron_trigger: Union[str, CronWorkflows]) -> None:
+    async def aio_delete(self, cron_trigger: Union[str, CronWorkflows]) -> None:
         """
         Asynchronously deletes a workflow cron trigger.
 
         Args:
             cron_trigger (Union[str, CronWorkflows]): The cron trigger ID or CronWorkflows instance to delete.
         """
-        id_ = cron_trigger
-        if isinstance(cron_trigger, CronWorkflows):
-            id_ = cron_trigger.metadata.id
-        await self._client.rest.aio.cron_delete(id_)
+        await self._client.rest.aio_delete_cron(
+            cron_trigger.metadata.id
+            if isinstance(cron_trigger, CronWorkflows)
+            else cron_trigger
+        )
 
-    async def list(
+    async def aio_list(
         self,
         offset: int | None = None,
         limit: int | None = None,
         workflow_id: str | None = None,
-        additional_metadata: list[str] | None = None,
+        additional_metadata: List[str] | None = None,
         order_by_field: CronWorkflowsOrderByField | None = None,
         order_by_direction: WorkflowRunOrderByDirection | None = None,
     ) -> CronWorkflowsList:
@@ -261,7 +245,7 @@ class CronClientAsync:
         Returns:
             CronWorkflowsList: A list of cron workflows.
         """
-        return await self._client.rest.aio.cron_list(
+        return await self._client.rest.aio_list_crons(
             offset=offset,
             limit=limit,
             workflow_id=workflow_id,
@@ -270,7 +254,7 @@ class CronClientAsync:
             order_by_direction=order_by_direction,
         )
 
-    async def get(self, cron_trigger: Union[str, CronWorkflows]) -> CronWorkflows:
+    async def aio_get(self, cron_trigger: Union[str, CronWorkflows]) -> CronWorkflows:
         """
         Asynchronously retrieves a specific workflow cron trigger by ID.
 
@@ -280,7 +264,9 @@ class CronClientAsync:
         Returns:
             CronWorkflows: The requested cron workflow instance.
         """
-        id_ = cron_trigger
-        if isinstance(cron_trigger, CronWorkflows):
-            id_ = cron_trigger.metadata.id
-        return await self._client.rest.aio.cron_get(id_)
+
+        return await self._client.rest.aio_get_cron(
+            cron_trigger.metadata.id
+            if isinstance(cron_trigger, CronWorkflows)
+            else cron_trigger
+        )