hatchet-sdk 1.20.0__tar.gz → 1.22.2__tar.gz

{hatchet_sdk-1.20.0 → hatchet_sdk-1.22.2}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: hatchet-sdk
-Version: 1.20.0
-Summary: 
+Version: 1.22.2
+Summary: This is the official Python SDK for Hatchet, a distributed, fault-tolerant task queue. The SDK allows you to easily integrate Hatchet's task scheduling and workflow orchestration capabilities into your Python applications.
 License: MIT
 Author: Alexander Belanger
 Author-email: alexander@hatchet.run
@@ -14,10 +14,8 @@ Classifier: Programming Language :: Python :: 3.12
 Provides-Extra: otel
 Provides-Extra: v0-sdk
 Requires-Dist: aiohttp (>=3.10.5,<4.0.0)
-Requires-Dist: grpcio (>=1.64.1,!=1.68.*) ; python_version < "3.13"
-Requires-Dist: grpcio (>=1.69.0) ; python_version >= "3.13"
-Requires-Dist: grpcio-tools (>=1.64.1,!=1.68.*) ; python_version < "3.13"
-Requires-Dist: grpcio-tools (>=1.69.0) ; python_version >= "3.13"
+Requires-Dist: grpcio (>=1.76.0,<2.0.0)
+Requires-Dist: grpcio-tools (>=1.76.0,<2.0.0)
 Requires-Dist: opentelemetry-api (>=1.28.0,<2.0.0) ; extra == "otel"
 Requires-Dist: opentelemetry-distro (>=0.49b0) ; extra == "otel"
 Requires-Dist: opentelemetry-exporter-otlp (>=1.28.0,<2.0.0) ; extra == "otel"
@@ -25,12 +23,12 @@ Requires-Dist: opentelemetry-exporter-otlp-proto-http (>=1.28.0,<2.0.0) ; extra
 Requires-Dist: opentelemetry-instrumentation (>=0.49b0) ; extra == "otel"
 Requires-Dist: opentelemetry-sdk (>=1.28.0,<2.0.0) ; extra == "otel"
 Requires-Dist: prometheus-client (>=0.21.1)
-Requires-Dist: protobuf (>=5.29.5,<6.0.0)
+Requires-Dist: protobuf (>=6.30.0,<7.0.0)
 Requires-Dist: pydantic (>=2.6.3,<3.0.0)
 Requires-Dist: pydantic-settings (>=2.7.1,<3.0.0)
 Requires-Dist: python-dateutil (>=2.9.0.post0,<3.0.0)
 Requires-Dist: tenacity (>=8.4.1)
-Requires-Dist: urllib3 (>=2.5.0,<3.0.0)
+Requires-Dist: urllib3 (>=2.6.0,<3.0.0)
 Description-Content-Type: text/markdown
 
 # Hatchet Python SDK

{hatchet_sdk-1.20.0 → hatchet_sdk-1.22.2}/hatchet_sdk/clients/admin.py

@@ -170,14 +170,12 @@ class AdminClient:
             priority=options.priority,
         )
 
-    @tenacity_retry
     async def aio_put_workflow(
         self,
         workflow: workflow_protos.CreateWorkflowVersionRequest,
     ) -> workflow_protos.CreateWorkflowVersionResponse:
         return await asyncio.to_thread(self.put_workflow, workflow)
 
-    @tenacity_retry
     async def aio_put_rate_limit(
         self,
         key: str,
@@ -186,7 +184,6 @@ class AdminClient:
     ) -> None:
         return await asyncio.to_thread(self.put_rate_limit, key, limit, duration)
 
-    @tenacity_retry
     async def aio_schedule_workflow(
         self,
         name: str,
@@ -198,7 +195,6 @@ class AdminClient:
             self.schedule_workflow, name, schedules, input, options
         )
 
-    @tenacity_retry
     def put_workflow(
         self,
         workflow: workflow_protos.CreateWorkflowVersionRequest,
@@ -207,15 +203,15 @@ class AdminClient:
             conn = new_conn(self.config, False)
             self.client = AdminServiceStub(conn)
 
+        put_workflow = tenacity_retry(self.client.PutWorkflow, self.config.tenacity)
         return cast(
             workflow_protos.CreateWorkflowVersionResponse,
-            self.client.PutWorkflow(
+            put_workflow(
                 workflow,
                 metadata=get_metadata(self.token),
             ),
         )
 
-    @tenacity_retry
     def put_rate_limit(
         self,
         key: str,
@@ -227,8 +223,9 @@ class AdminClient:
         )
 
         client = self._get_or_create_v0_client()
+        put_rate_limit = tenacity_retry(client.PutRateLimit, self.config.tenacity)
 
-        client.PutRateLimit(
+        put_rate_limit(
             v0_workflow_protos.PutRateLimitRequest(
                 key=key,
                 limit=limit,
@@ -237,7 +234,6 @@ class AdminClient:
             metadata=get_metadata(self.token),
         )
 
-    @tenacity_retry
     def schedule_workflow(
         self,
         name: str,
@@ -255,10 +251,13 @@ class AdminClient:
             )
 
             client = self._get_or_create_v0_client()
+            schedule_workflow = tenacity_retry(
+                client.ScheduleWorkflow, self.config.tenacity
+            )
 
             return cast(
                 v0_workflow_protos.WorkflowVersion,
-                client.ScheduleWorkflow(
+                schedule_workflow(
                     request,
                     metadata=get_metadata(self.token),
                 ),
@@ -313,7 +312,6 @@ class AdminClient:
         return self._prepare_workflow_request(workflow_name, input, trigger_options)
 
     ## IMPORTANT: Keep this method's signature in sync with the wrapper in the OTel instrumentor
-    @tenacity_retry
     def run_workflow(
         self,
         workflow_name: str,
@@ -322,11 +320,12 @@ class AdminClient:
     ) -> WorkflowRunRef:
         request = self._create_workflow_run_request(workflow_name, input, options)
         client = self._get_or_create_v0_client()
+        trigger_workflow = tenacity_retry(client.TriggerWorkflow, self.config.tenacity)
 
         try:
             resp = cast(
                 v0_workflow_protos.TriggerWorkflowResponse,
-                client.TriggerWorkflow(
+                trigger_workflow(
                     request,
                     metadata=get_metadata(self.token),
                 ),
@@ -344,7 +343,6 @@ class AdminClient:
         )
 
     ## IMPORTANT: Keep this method's signature in sync with the wrapper in the OTel instrumentor
-    @tenacity_retry
     async def aio_run_workflow(
         self,
         workflow_name: str,
@@ -352,6 +350,7 @@ class AdminClient:
         options: TriggerWorkflowOptions = TriggerWorkflowOptions(),
     ) -> WorkflowRunRef:
         client = self._get_or_create_v0_client()
+        trigger_workflow = tenacity_retry(client.TriggerWorkflow, self.config.tenacity)
         async with spawn_index_lock:
             request = self._create_workflow_run_request(workflow_name, input, options)
 
@@ -359,7 +358,7 @@ class AdminClient:
             resp = cast(
                 v0_workflow_protos.TriggerWorkflowResponse,
                 await asyncio.to_thread(
-                    client.TriggerWorkflow,
+                    trigger_workflow,
                     request,
                     metadata=get_metadata(self.token),
                 ),
@@ -382,7 +381,6 @@ class AdminClient:
             yield xs[i : i + n]
 
     ## IMPORTANT: Keep this method's signature in sync with the wrapper in the OTel instrumentor
-    @tenacity_retry
     def run_workflows(
         self,
         workflows: list[WorkflowRunTriggerConfig],
@@ -394,6 +392,9 @@ class AdminClient:
             )
             for workflow in workflows
         ]
+        bulk_trigger_workflow = tenacity_retry(
+            client.BulkTriggerWorkflow, self.config.tenacity
+        )
 
         refs: list[WorkflowRunRef] = []
 
@@ -404,7 +405,7 @@ class AdminClient:
 
             resp = cast(
                 v0_workflow_protos.BulkTriggerWorkflowResponse,
-                client.BulkTriggerWorkflow(
+                bulk_trigger_workflow(
                     bulk_request,
                     metadata=get_metadata(self.token),
                 ),
@@ -424,7 +425,6 @@ class AdminClient:
 
         return refs
 
-    @tenacity_retry
     async def aio_run_workflows(
         self,
         workflows: list[WorkflowRunTriggerConfig],
@@ -432,6 +432,9 @@ class AdminClient:
         client = self._get_or_create_v0_client()
         chunks = self.chunk(workflows, MAX_BULK_WORKFLOW_RUN_BATCH_SIZE)
         refs: list[WorkflowRunRef] = []
+        bulk_trigger_workflow = tenacity_retry(
+            client.BulkTriggerWorkflow, self.config.tenacity
+        )
 
         for chunk in chunks:
             async with spawn_index_lock:
@@ -449,7 +452,7 @@ class AdminClient:
             resp = cast(
                 v0_workflow_protos.BulkTriggerWorkflowResponse,
                 await asyncio.to_thread(
-                    client.BulkTriggerWorkflow,
+                    bulk_trigger_workflow,
                     bulk_request,
                     metadata=get_metadata(self.token),
                 ),

{hatchet_sdk-1.20.0 → hatchet_sdk-1.22.2}/hatchet_sdk/clients/dispatcher/dispatcher.py

@@ -112,7 +112,6 @@ class DispatcherClient:
 
             return None
 
-    @tenacity_retry
     async def _try_send_step_action_event(
         self,
         action: Action,
@@ -141,9 +140,13 @@ class DispatcherClient:
             shouldNotRetry=should_not_retry,
         )
 
+        send_step_action_event = tenacity_retry(
+            self.aio_client.SendStepActionEvent, self.config.tenacity
+        )
+
         return cast(
             grpc.aio.UnaryUnaryCall[StepActionEvent, ActionEventResponse],
-            await self.aio_client.SendStepActionEvent(
+            await send_step_action_event(
                 event,
                 metadata=get_metadata(self.token),
             ),

{hatchet_sdk-1.20.0 → hatchet_sdk-1.22.2}/hatchet_sdk/clients/events.py

@@ -9,6 +9,7 @@ from pydantic import BaseModel, Field
 from hatchet_sdk.clients.rest.api.event_api import EventApi
 from hatchet_sdk.clients.rest.api.workflow_runs_api import WorkflowRunsApi
 from hatchet_sdk.clients.rest.api_client import ApiClient
+from hatchet_sdk.clients.rest.models.v1_event import V1Event
 from hatchet_sdk.clients.rest.models.v1_event_list import V1EventList
 from hatchet_sdk.clients.rest.models.v1_task_status import V1TaskStatus
 from hatchet_sdk.clients.rest.tenacity_utils import tenacity_retry
@@ -92,7 +93,6 @@ class EventClient(BaseRestClient):
         return await asyncio.to_thread(self.bulk_push, events=events, options=options)
 
     ## IMPORTANT: Keep this method's signature in sync with the wrapper in the OTel instrumentor
-    @tenacity_retry
     def push(
         self,
         event_key: str,
@@ -101,6 +101,9 @@ class EventClient(BaseRestClient):
     ) -> Event:
         namespace = options.namespace or self.namespace
         namespaced_event_key = self.client_config.apply_namespace(event_key, namespace)
+        push_event = tenacity_retry(
+            self.events_service_client.Push, self.client_config.tenacity
+        )
 
         try:
             meta_bytes = json.dumps(options.additional_metadata)
@@ -123,7 +126,7 @@ class EventClient(BaseRestClient):
 
         return cast(
             Event,
-            self.events_service_client.Push(request, metadata=get_metadata(self.token)),
+            push_event(request, metadata=get_metadata(self.token)),
         )
 
     def _create_push_event_request(
@@ -156,13 +159,15 @@ class EventClient(BaseRestClient):
         )
 
     ## IMPORTANT: Keep this method's signature in sync with the wrapper in the OTel instrumentor
-    @tenacity_retry
     def bulk_push(
         self,
         events: list[BulkPushEventWithMetadata],
         options: BulkPushEventOptions = BulkPushEventOptions(),
     ) -> list[Event]:
         namespace = options.namespace or self.namespace
+        bulk_push = tenacity_retry(
+            self.events_service_client.BulkPush, self.client_config.tenacity
+        )
 
         bulk_request = BulkPushEventRequest(
             events=[
@@ -173,16 +178,16 @@ class EventClient(BaseRestClient):
         return list(
             cast(
                 Events,
-                self.events_service_client.BulkPush(
-                    bulk_request, metadata=get_metadata(self.token)
-                ),
+                bulk_push(bulk_request, metadata=get_metadata(self.token)),
             ).events
         )
 
-    @tenacity_retry
     def log(
         self, message: str, step_run_id: str, level: LogLevel | None = None
     ) -> None:
+        put_log = tenacity_retry(
+            self.events_service_client.PutLog, self.client_config.tenacity
+        )
         request = PutLogRequest(
             stepRunId=step_run_id,
             createdAt=proto_timestamp_now(),
@@ -190,10 +195,12 @@ class EventClient(BaseRestClient):
             level=level.value if level else None,
         )
 
-        self.events_service_client.PutLog(request, metadata=get_metadata(self.token))
+        put_log(request, metadata=get_metadata(self.token))
 
-    @tenacity_retry
     def stream(self, data: str | bytes, step_run_id: str, index: int) -> None:
+        put_stream_event = tenacity_retry(
+            self.events_service_client.PutStreamEvent, self.client_config.tenacity
+        )
         if isinstance(data, str):
             data_bytes = data.encode("utf-8")
         elif isinstance(data, bytes):
@@ -209,9 +216,7 @@ class EventClient(BaseRestClient):
         )
 
         try:
-            self.events_service_client.PutStreamEvent(
-                request, metadata=get_metadata(self.token)
-            )
+            put_stream_event(request, metadata=get_metadata(self.token))
         except Exception:
             raise
 
@@ -271,3 +276,22 @@ class EventClient(BaseRestClient):
                 ),
                 scopes=scopes,
             )
+
+    def get(
+        self,
+        event_id: str,
+    ) -> V1Event:
+        with self.client() as client:
+            return self._ea(client).v1_event_get(
+                tenant=self.client_config.tenant_id,
+                v1_event=event_id,
+            )
+
+    async def aio_get(
+        self,
+        event_id: str,
+    ) -> V1Event:
+        return await asyncio.to_thread(
+            self.get,
+            event_id=event_id,
+        )

{hatchet_sdk-1.20.0 → hatchet_sdk-1.22.2}/hatchet_sdk/clients/listeners/durable_event_listener.py

@@ -103,14 +103,17 @@ class DurableEventListener(
             signal_key=key.signal_key,
         )
 
-    @tenacity_retry
     def register_durable_event(
         self, request: RegisterDurableEventRequest
     ) -> Literal[True]:
         conn = new_conn(self.config, True)
         client = V1DispatcherStub(conn)
 
-        client.RegisterDurableEvent(
+        register_durable_event = tenacity_retry(
+            client.RegisterDurableEvent, self.config.tenacity
+        )
+
+        register_durable_event(
             request.to_proto(),
             timeout=5,
             metadata=get_metadata(self.token),
@@ -118,7 +121,6 @@ class DurableEventListener(
 
         return True
 
-    @tenacity_retry
     async def result(self, task_id: str, signal_key: str) -> dict[str, Any]:
         key = self._generate_key(task_id, signal_key)
 

{hatchet_sdk-1.20.0 → hatchet_sdk-1.22.2}/hatchet_sdk/clients/rest/__init__.py

@@ -69,6 +69,7 @@ from hatchet_sdk.clients.rest.models.cancel_event_request import CancelEventRequ
 from hatchet_sdk.clients.rest.models.concurrency_limit_strategy import (
     ConcurrencyLimitStrategy,
 )
+from hatchet_sdk.clients.rest.models.concurrency_stat import ConcurrencyStat
 from hatchet_sdk.clients.rest.models.create_api_token_request import (
     CreateAPITokenRequest,
 )
@@ -150,6 +151,7 @@ from hatchet_sdk.clients.rest.models.rate_limit_order_by_field import (
     RateLimitOrderByField,
 )
 from hatchet_sdk.clients.rest.models.recent_step_runs import RecentStepRuns
+from hatchet_sdk.clients.rest.models.registered_workflow import RegisteredWorkflow
 from hatchet_sdk.clients.rest.models.reject_invite_request import RejectInviteRequest
 from hatchet_sdk.clients.rest.models.replay_event_request import ReplayEventRequest
 from hatchet_sdk.clients.rest.models.replay_workflow_runs_request import (
@@ -164,6 +166,27 @@ from hatchet_sdk.clients.rest.models.schedule_workflow_run_request import (
 )
 from hatchet_sdk.clients.rest.models.scheduled_run_status import ScheduledRunStatus
 from hatchet_sdk.clients.rest.models.scheduled_workflows import ScheduledWorkflows
+from hatchet_sdk.clients.rest.models.scheduled_workflows_bulk_delete_filter import (
+    ScheduledWorkflowsBulkDeleteFilter,
+)
+from hatchet_sdk.clients.rest.models.scheduled_workflows_bulk_delete_request import (
+    ScheduledWorkflowsBulkDeleteRequest,
+)
+from hatchet_sdk.clients.rest.models.scheduled_workflows_bulk_delete_response import (
+    ScheduledWorkflowsBulkDeleteResponse,
+)
+from hatchet_sdk.clients.rest.models.scheduled_workflows_bulk_error import (
+    ScheduledWorkflowsBulkError,
+)
+from hatchet_sdk.clients.rest.models.scheduled_workflows_bulk_update_item import (
+    ScheduledWorkflowsBulkUpdateItem,
+)
+from hatchet_sdk.clients.rest.models.scheduled_workflows_bulk_update_request import (
+    ScheduledWorkflowsBulkUpdateRequest,
+)
+from hatchet_sdk.clients.rest.models.scheduled_workflows_bulk_update_response import (
+    ScheduledWorkflowsBulkUpdateResponse,
+)
 from hatchet_sdk.clients.rest.models.scheduled_workflows_list import (
     ScheduledWorkflowsList,
 )
@@ -186,6 +209,8 @@ from hatchet_sdk.clients.rest.models.step_run_event_list import StepRunEventList
 from hatchet_sdk.clients.rest.models.step_run_event_reason import StepRunEventReason
 from hatchet_sdk.clients.rest.models.step_run_event_severity import StepRunEventSeverity
 from hatchet_sdk.clients.rest.models.step_run_status import StepRunStatus
+from hatchet_sdk.clients.rest.models.task_stat import TaskStat
+from hatchet_sdk.clients.rest.models.task_status_stat import TaskStatusStat
 from hatchet_sdk.clients.rest.models.tenant import Tenant
 from hatchet_sdk.clients.rest.models.tenant_alert_email_group import (
     TenantAlertEmailGroup,
@@ -196,6 +221,7 @@ from hatchet_sdk.clients.rest.models.tenant_alert_email_group_list import (
 from hatchet_sdk.clients.rest.models.tenant_alerting_settings import (
     TenantAlertingSettings,
 )
+from hatchet_sdk.clients.rest.models.tenant_environment import TenantEnvironment
 from hatchet_sdk.clients.rest.models.tenant_invite import TenantInvite
 from hatchet_sdk.clients.rest.models.tenant_invite_list import TenantInviteList
 from hatchet_sdk.clients.rest.models.tenant_list import TenantList
@@ -209,17 +235,25 @@ from hatchet_sdk.clients.rest.models.tenant_resource_policy import TenantResourc
 from hatchet_sdk.clients.rest.models.tenant_step_run_queue_metrics import (
     TenantStepRunQueueMetrics,
 )
-from hatchet_sdk.clients.rest.models.tenant_ui_version import TenantUIVersion
 from hatchet_sdk.clients.rest.models.tenant_version import TenantVersion
 from hatchet_sdk.clients.rest.models.trigger_workflow_run_request import (
     TriggerWorkflowRunRequest,
 )
+from hatchet_sdk.clients.rest.models.update_cron_workflow_trigger_request import (
+    UpdateCronWorkflowTriggerRequest,
+)
+from hatchet_sdk.clients.rest.models.update_scheduled_workflow_run_request import (
+    UpdateScheduledWorkflowRunRequest,
+)
 from hatchet_sdk.clients.rest.models.update_tenant_alert_email_group_request import (
     UpdateTenantAlertEmailGroupRequest,
 )
 from hatchet_sdk.clients.rest.models.update_tenant_invite_request import (
     UpdateTenantInviteRequest,
 )
+from hatchet_sdk.clients.rest.models.update_tenant_member_request import (
+    UpdateTenantMemberRequest,
+)
 from hatchet_sdk.clients.rest.models.update_tenant_request import UpdateTenantRequest
 from hatchet_sdk.clients.rest.models.update_worker_request import UpdateWorkerRequest
 from hatchet_sdk.clients.rest.models.user import User
@@ -290,6 +324,9 @@ from hatchet_sdk.clients.rest.models.v1_trigger_workflow_run_request import (
 from hatchet_sdk.clients.rest.models.v1_update_filter_request import (
     V1UpdateFilterRequest,
 )
+from hatchet_sdk.clients.rest.models.v1_update_webhook_request import (
+    V1UpdateWebhookRequest,
+)
 from hatchet_sdk.clients.rest.models.v1_webhook import V1Webhook
 from hatchet_sdk.clients.rest.models.v1_webhook_api_key_auth import V1WebhookAPIKeyAuth
 from hatchet_sdk.clients.rest.models.v1_webhook_auth_type import V1WebhookAuthType
@@ -302,9 +339,6 @@ from hatchet_sdk.clients.rest.models.v1_webhook_hmac_encoding import (
     V1WebhookHMACEncoding,
 )
 from hatchet_sdk.clients.rest.models.v1_webhook_list import V1WebhookList
-from hatchet_sdk.clients.rest.models.v1_webhook_receive200_response import (
-    V1WebhookReceive200Response,
-)
 from hatchet_sdk.clients.rest.models.v1_webhook_source_name import V1WebhookSourceName
 from hatchet_sdk.clients.rest.models.v1_workflow_run import V1WorkflowRun
 from hatchet_sdk.clients.rest.models.v1_workflow_run_details import V1WorkflowRunDetails