hatchet-sdk 0.43.2__py3-none-any.whl → 0.44.0__py3-none-any.whl

hatchet_sdk/clients/admin.py

@@ -252,11 +252,11 @@ class AdminClientAioImpl(AdminClientBase):
                     workflow_listener=self.pooled_workflow_listener,
                     workflow_run_event_listener=self.listener_client,
                 )
-            except grpc.RpcError as e:
+            except (grpc.RpcError, grpc.aio.AioRpcError) as e:
                 if e.code() == grpc.StatusCode.ALREADY_EXISTS:
                     raise DedupeViolationErr(e.details())
 
-                raise ValueError(f"gRPC error: {e}")
+                raise e
 
     @tenacity_retry
     async def run_workflows(
@@ -266,56 +266,49 @@ class AdminClientAioImpl(AdminClientBase):
     ) -> List[WorkflowRunRef]:
         if len(workflows) == 0:
             raise ValueError("No workflows to run")
-        try:
-            if not self.pooled_workflow_listener:
-                self.pooled_workflow_listener = PooledWorkflowRunListener(self.config)
 
-            namespace = self.namespace
+        if not self.pooled_workflow_listener:
+            self.pooled_workflow_listener = PooledWorkflowRunListener(self.config)
 
-            if (
-                options is not None
-                and "namespace" in options
-                and options["namespace"] is not None
-            ):
-                namespace = options["namespace"]
-                del options["namespace"]
+        namespace = self.namespace
 
-            workflow_run_requests: TriggerWorkflowRequest = []
+        if (
+            options is not None
+            and "namespace" in options
+            and options["namespace"] is not None
+        ):
+            namespace = options["namespace"]
+            del options["namespace"]
 
-            for workflow in workflows:
-                workflow_name = workflow["workflow_name"]
-                input_data = workflow["input"]
-                options = workflow["options"]
+        workflow_run_requests: TriggerWorkflowRequest = []
 
-                if namespace != "" and not workflow_name.startswith(self.namespace):
-                    workflow_name = f"{namespace}{workflow_name}"
+        for workflow in workflows:
+            workflow_name = workflow["workflow_name"]
+            input_data = workflow["input"]
+            options = workflow["options"]
 
-                # Prepare and trigger workflow for each workflow name and input
-                request = self._prepare_workflow_request(
-                    workflow_name, input_data, options
-                )
-                workflow_run_requests.append(request)
+            if namespace != "" and not workflow_name.startswith(self.namespace):
+                workflow_name = f"{namespace}{workflow_name}"
 
-            request = BulkTriggerWorkflowRequest(workflows=workflow_run_requests)
+            # Prepare and trigger workflow for each workflow name and input
+            request = self._prepare_workflow_request(workflow_name, input_data, options)
+            workflow_run_requests.append(request)
 
-            resp: BulkTriggerWorkflowResponse = (
-                await self.aio_client.BulkTriggerWorkflow(
-                    request,
-                    metadata=get_metadata(self.token),
-                )
-            )
+        request = BulkTriggerWorkflowRequest(workflows=workflow_run_requests)
 
-            return [
-                WorkflowRunRef(
-                    workflow_run_id=workflow_run_id,
-                    workflow_listener=self.pooled_workflow_listener,
-                    workflow_run_event_listener=self.listener_client,
-                )
-                for workflow_run_id in resp.workflow_run_ids
-            ]
+        resp: BulkTriggerWorkflowResponse = await self.aio_client.BulkTriggerWorkflow(
+            request,
+            metadata=get_metadata(self.token),
+        )
 
-        except grpc.RpcError as e:
-            raise ValueError(f"gRPC error: {e}")
+        return [
+            WorkflowRunRef(
+                workflow_run_id=workflow_run_id,
+                workflow_listener=self.pooled_workflow_listener,
+                workflow_run_event_listener=self.listener_client,
+            )
+            for workflow_run_id in resp.workflow_run_ids
+        ]
 
     @tenacity_retry
     async def put_workflow(
@@ -324,15 +317,12 @@ class AdminClientAioImpl(AdminClientBase):
         workflow: CreateWorkflowVersionOpts | WorkflowMeta,
         overrides: CreateWorkflowVersionOpts | None = None,
     ) -> WorkflowVersion:
-        try:
-            opts = self._prepare_put_workflow_request(name, workflow, overrides)
+        opts = self._prepare_put_workflow_request(name, workflow, overrides)
 
-            return await self.aio_client.PutWorkflow(
-                opts,
-                metadata=get_metadata(self.token),
-            )
-        except grpc.RpcError as e:
-            raise ValueError(f"Could not put workflow: {e}")
+        return await self.aio_client.PutWorkflow(
+            opts,
+            metadata=get_metadata(self.token),
+        )
 
     @tenacity_retry
     async def put_rate_limit(
@@ -341,17 +331,14 @@ class AdminClientAioImpl(AdminClientBase):
         limit: int,
         duration: RateLimitDuration = RateLimitDuration.SECOND,
     ):
-        try:
-            await self.aio_client.PutRateLimit(
-                PutRateLimitRequest(
-                    key=key,
-                    limit=limit,
-                    duration=duration,
-                ),
-                metadata=get_metadata(self.token),
-            )
-        except grpc.RpcError as e:
-            raise ValueError(f"Could not put rate limit: {e}")
+        await self.aio_client.PutRateLimit(
+            PutRateLimitRequest(
+                key=key,
+                limit=limit,
+                duration=duration,
+            ),
+            metadata=get_metadata(self.token),
+        )
 
     @tenacity_retry
     async def schedule_workflow(
@@ -383,11 +370,11 @@ class AdminClientAioImpl(AdminClientBase):
                 request,
                 metadata=get_metadata(self.token),
             )
-        except grpc.RpcError as e:
+        except (grpc.aio.AioRpcError, grpc.RpcError) as e:
             if e.code() == grpc.StatusCode.ALREADY_EXISTS:
                 raise DedupeViolationErr(e.details())
 
-            raise ValueError(f"gRPC error: {e}")
+            raise e
 
 
 class AdminClient(AdminClientBase):
@@ -408,17 +395,14 @@ class AdminClient(AdminClientBase):
         workflow: CreateWorkflowVersionOpts | WorkflowMeta,
         overrides: CreateWorkflowVersionOpts | None = None,
     ) -> WorkflowVersion:
-        try:
-            opts = self._prepare_put_workflow_request(name, workflow, overrides)
+        opts = self._prepare_put_workflow_request(name, workflow, overrides)
 
-            resp: WorkflowVersion = self.client.PutWorkflow(
-                opts,
-                metadata=get_metadata(self.token),
-            )
+        resp: WorkflowVersion = self.client.PutWorkflow(
+            opts,
+            metadata=get_metadata(self.token),
+        )
 
-            return resp
-        except grpc.RpcError as e:
-            raise ValueError(f"Could not put workflow: {e}")
+        return resp
 
     @tenacity_retry
     def put_rate_limit(
@@ -427,17 +411,14 @@ class AdminClient(AdminClientBase):
         limit: int,
         duration: Union[RateLimitDuration.Value, str] = RateLimitDuration.SECOND,
     ):
-        try:
-            self.client.PutRateLimit(
-                PutRateLimitRequest(
-                    key=key,
-                    limit=limit,
-                    duration=duration,
-                ),
-                metadata=get_metadata(self.token),
-            )
-        except grpc.RpcError as e:
-            raise ValueError(f"Could not put rate limit: {e}")
+        self.client.PutRateLimit(
+            PutRateLimitRequest(
+                key=key,
+                limit=limit,
+                duration=duration,
+            ),
+            metadata=get_metadata(self.token),
+        )
 
     @tenacity_retry
     def schedule_workflow(
@@ -469,11 +450,11 @@ class AdminClient(AdminClientBase):
                 request,
                 metadata=get_metadata(self.token),
             )
-        except grpc.RpcError as e:
+        except (grpc.RpcError, grpc.aio.AioRpcError) as e:
             if e.code() == grpc.StatusCode.ALREADY_EXISTS:
                 raise DedupeViolationErr(e.details())
 
-            raise ValueError(f"gRPC error: {e}")
+            raise e
 
     ## TODO: `options` is treated as a dict (wrong type hint)
     ## TODO: `any` type hint should come from `typing`
@@ -541,55 +522,49 @@ class AdminClient(AdminClientBase):
                     workflow_listener=self.pooled_workflow_listener,
                     workflow_run_event_listener=self.listener_client,
                 )
-            except grpc.RpcError as e:
+            except (grpc.RpcError, grpc.aio.AioRpcError) as e:
                 if e.code() == grpc.StatusCode.ALREADY_EXISTS:
                     raise DedupeViolationErr(e.details())
 
-                raise ValueError(f"gRPC error: {e}")
+                raise e
 
     @tenacity_retry
     def run_workflows(
         self, workflows: List[WorkflowRunDict], options: TriggerWorkflowOptions = None
     ) -> list[WorkflowRunRef]:
         workflow_run_requests: TriggerWorkflowRequest = []
-        try:
-            if not self.pooled_workflow_listener:
-                self.pooled_workflow_listener = PooledWorkflowRunListener(self.config)
+        if not self.pooled_workflow_listener:
+            self.pooled_workflow_listener = PooledWorkflowRunListener(self.config)
 
-            for workflow in workflows:
-                workflow_name = workflow["workflow_name"]
-                input_data = workflow["input"]
-                options = workflow["options"]
+        for workflow in workflows:
+            workflow_name = workflow["workflow_name"]
+            input_data = workflow["input"]
+            options = workflow["options"]
 
-                namespace = self.namespace
-
-                if (
-                    options is not None
-                    and "namespace" in options
-                    and options["namespace"] is not None
-                ):
-                    namespace = options["namespace"]
-                    del options["namespace"]
+            namespace = self.namespace
 
-                if namespace != "" and not workflow_name.startswith(self.namespace):
-                    workflow_name = f"{namespace}{workflow_name}"
+            if (
+                options is not None
+                and "namespace" in options
+                and options["namespace"] is not None
+            ):
+                namespace = options["namespace"]
+                del options["namespace"]
 
-                # Prepare and trigger workflow for each workflow name and input
-                request = self._prepare_workflow_request(
-                    workflow_name, input_data, options
-                )
+            if namespace != "" and not workflow_name.startswith(self.namespace):
+                workflow_name = f"{namespace}{workflow_name}"
 
-                workflow_run_requests.append(request)
+            # Prepare and trigger workflow for each workflow name and input
+            request = self._prepare_workflow_request(workflow_name, input_data, options)
 
-                request = BulkTriggerWorkflowRequest(workflows=workflow_run_requests)
+            workflow_run_requests.append(request)
 
-            resp: BulkTriggerWorkflowResponse = self.client.BulkTriggerWorkflow(
-                request,
-                metadata=get_metadata(self.token),
-            )
+            request = BulkTriggerWorkflowRequest(workflows=workflow_run_requests)
 
-        except grpc.RpcError as e:
-            raise ValueError(f"gRPC error: {e}")
+        resp: BulkTriggerWorkflowResponse = self.client.BulkTriggerWorkflow(
+            request,
+            metadata=get_metadata(self.token),
+        )
 
         return [
             WorkflowRunRef(

hatchet_sdk/clients/events.py

@@ -125,10 +125,7 @@ class EventClient:
 
             span.add_event("Pushing event", attributes={"key": namespaced_event_key})
 
-            try:
-                return self.client.Push(request, metadata=get_metadata(self.token))
-            except grpc.RpcError as e:
-                raise ValueError(f"gRPC error: {e}")
+            return self.client.Push(request, metadata=get_metadata(self.token))
 
     @tenacity_retry
     def bulk_push(
@@ -188,13 +185,9 @@ class EventClient:
         bulk_request = BulkPushEventRequest(events=bulk_events)
 
         span.add_event("Pushing bulk events")
-        try:
-            response = self.client.BulkPush(
-                bulk_request, metadata=get_metadata(self.token)
-            )
-            return response.events
-        except grpc.RpcError as e:
-            raise ValueError(f"gRPC error: {e}")
+        response = self.client.BulkPush(bulk_request, metadata=get_metadata(self.token))
+
+        return response.events
 
     def log(self, message: str, step_run_id: str):
         try:

hatchet_sdk/clients/rest/tenacity_utils.py

@@ -28,8 +28,8 @@ def tenacity_alert_retry(retry_state: tenacity.RetryCallState) -> None:
 
 
 def tenacity_should_retry(ex: Exception) -> bool:
-    if isinstance(ex, grpc.aio.AioRpcError):
-        if ex.code in [
+    if isinstance(ex, (grpc.aio.AioRpcError, grpc.RpcError)):
+        if ex.code() in [
             grpc.StatusCode.UNIMPLEMENTED,
             grpc.StatusCode.NOT_FOUND,
         ]:

hatchet_sdk/context/context.py

@@ -376,6 +376,16 @@ class Context(BaseContext):
     def parent_workflow_run_id(self) -> str | None:
         return self.action.parent_workflow_run_id
 
+    def step_run_errors(self) -> dict[str, str]:
+        errors = cast(dict[str, str], self.data.get("step_run_errors", {}))
+
+        if not errors:
+            logger.error(
+                "No step run errors found. `context.step_run_errors` is intended to be run in an on-failure step, and will only work on engine versions more recent than v0.53.10"
+            )
+
+        return errors
+
     def fetch_run_failures(self) -> list[dict[str, StrictStr]]:
         data = self.rest_client.workflow_run_get(self.action.workflow_run_id)
         other_job_runs = [

{hatchet_sdk-0.43.2.dist-info → hatchet_sdk-0.44.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hatchet-sdk
-Version: 0.43.2
+Version: 0.44.0
 Summary: 
 Author: Alexander Belanger
 Author-email: alexander@hatchet.run

{hatchet_sdk-0.43.2.dist-info → hatchet_sdk-0.44.0.dist-info}/RECORD

@@ -1,10 +1,10 @@
 hatchet_sdk/__init__.py,sha256=R5ogd_Dn_6gA_u9a5W2URNq6eDtN1i56cObBv1tOwyU,9408
 hatchet_sdk/client.py,sha256=ajjLd-gZptVuAx25gG_SdAW8xDA4V7HMIhgYuh9MkVY,3486
-hatchet_sdk/clients/admin.py,sha256=RGi-cXGTn54EmgiqbUsnb9a9gbA3WFr9Qo2VdBHdSzQ,21530
+hatchet_sdk/clients/admin.py,sha256=1YsLK5AzpgFBKPblcPY5ACA0xLysynA4tihNJkcNyl4,20419
 hatchet_sdk/clients/dispatcher/action_listener.py,sha256=eOq5z29MhC7ynbOOegDRQr-RqDhpS3gfeLswIlZbuGg,15378
 hatchet_sdk/clients/dispatcher/dispatcher.py,sha256=qTcfIXn9EpfUlafff8Wun_fWTvz-UxOa4i4m6ZVIAnM,6402
 hatchet_sdk/clients/event_ts.py,sha256=ACGvDdfhvK6ZLKdsPxy-PksLhjIU69P9cdH3AxX-X10,728
-hatchet_sdk/clients/events.py,sha256=8QOHv4ObSNXb6t5EnaXPwkwr1hN4D087iWn02TRtXb0,7505
+hatchet_sdk/clients/events.py,sha256=pz68MOFEdvKV9ywu0e4DIaWFgqjwzm8zUV747Fim9ac,7258
 hatchet_sdk/clients/rest/__init__.py,sha256=AQBp3fX79IKrcjUmzLRGFaFKMYBnEbQD9DnwYOHU0jQ,14157
 hatchet_sdk/clients/rest/api/__init__.py,sha256=LaTEK7cYklb8R0iYj727C-jVk-MGHenADN8TJpoIASs,1067
 hatchet_sdk/clients/rest/api/api_token_api.py,sha256=C10FEIHHGBpwq-bIKkrBhvPlg6az4aHlREWEUlJHWl0,33577
@@ -184,13 +184,13 @@ hatchet_sdk/clients/rest/models/workflow_version_definition.py,sha256=e18BUh1XO0
 hatchet_sdk/clients/rest/models/workflow_version_meta.py,sha256=TW4R7bAuYAg_LraN-8psdZqp2E8wH9hYyL5Sji86aLk,3791
 hatchet_sdk/clients/rest/models/workflow_workers_count.py,sha256=qhzqfvjjIDyARkiiLGluMIqEmqO-diHTsjlu0Doi0yg,2875
 hatchet_sdk/clients/rest/rest.py,sha256=G83F1k4g_ePzvXW95rApzvaRDQPcaxrj-JmZyq1LvGw,6606
-hatchet_sdk/clients/rest/tenacity_utils.py,sha256=AmVMjML3pd8qzsWyvaEnaI33zkidou5EK_EQXF69SAE,1032
+hatchet_sdk/clients/rest/tenacity_utils.py,sha256=gy500kHXQ-4ZrZH1biHGmavhfh0NarCJcaIfW-A9Qd0,1051
 hatchet_sdk/clients/rest_client.py,sha256=wYCRQjjZha9XcYg6pdVgrFV4pcli89Y_G45EDEDteCk,21874
 hatchet_sdk/clients/run_event_listener.py,sha256=51WTg52_aISgYPOFPHJ21rb4IO6aEd7Ugp7FCf4HnfM,10184
 hatchet_sdk/clients/workflow_listener.py,sha256=Q_WJcGlZNHJGSpxzDac9wELjgxhP0vLaNTXRy_xnxc8,9466
 hatchet_sdk/connection.py,sha256=593aUGAj7Ouf00lcVwx_pmhdQ9NOC5ANT1Jrf8nwkHs,2165
 hatchet_sdk/context/__init__.py,sha256=Pl_seJ_SJpW34BBZp4KixuZ8GiRK9sJFfegf9u3m7zk,29
-hatchet_sdk/context/context.py,sha256=R2RsPgzBNYi1FJvg4MOowSY-V1Yf69qUmXe3MsAWkyo,13518
+hatchet_sdk/context/context.py,sha256=sRAih-dBqxkmQUxgo_IpQP2YIJJinzJ2C09RP-EM94o,13910
 hatchet_sdk/context/worker_context.py,sha256=OVcEWvdT_Kpd0nlg61VAPUgIPSFzSLs0aSrXWj-1GX4,974
 hatchet_sdk/contracts/dispatcher_pb2.py,sha256=-9YGf7pfuMyRpkjjcmFa6jlr-95gha4CJw9hG3seNFI,14573
 hatchet_sdk/contracts/dispatcher_pb2.pyi,sha256=2dvxvN4OETOc-OhLNWDntN7GvI4QUxbMiGLdK7FM1wE,18224
@@ -230,7 +230,7 @@ hatchet_sdk/worker/runner/utils/error_with_traceback.py,sha256=Iih_s8JNqrinXETFJ
 hatchet_sdk/worker/worker.py,sha256=7UPm3qTzNYSSm9QTNX6zBBMJqVA6nKFeCbAdqLLjUBs,13007
 hatchet_sdk/workflow.py,sha256=XRj5jcCQSvPQMXxBipf-ZlARua2E8Z9igRzGcQ5alkI,9375
 hatchet_sdk/workflow_run.py,sha256=BwK5cefvXXvyQ1Ednj_7LeejMwQJqWnvUC_FTBmJNxk,1805
-hatchet_sdk-0.43.2.dist-info/METADATA,sha256=4X-r_xOmoJR1IfkCwPuRybgiMjt2Z1l0bkI-6EpMBBk,1736
-hatchet_sdk-0.43.2.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-hatchet_sdk-0.43.2.dist-info/entry_points.txt,sha256=LTtQRABmSGYOxRI68cUVEz5dp9Qb57eqXGic9lU8RMo,1023
-hatchet_sdk-0.43.2.dist-info/RECORD,,
+hatchet_sdk-0.44.0.dist-info/METADATA,sha256=pJrEPNQ7Lyx6R7FVf2gaUNlWGCa0y9fz5qVwMi1UcuA,1736
+hatchet_sdk-0.44.0.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+hatchet_sdk-0.44.0.dist-info/entry_points.txt,sha256=LTtQRABmSGYOxRI68cUVEz5dp9Qb57eqXGic9lU8RMo,1023
+hatchet_sdk-0.44.0.dist-info/RECORD,,