hatchet-sdk 1.0.3__py3-none-any.whl → 1.2.0__py3-none-any.whl

hatchet_sdk/clients/workflow_listener.py

@@ -54,14 +54,6 @@ class _Subscription:
 
 class PooledWorkflowRunListener:
     def __init__(self, config: ClientConfig):
-        try:
-            asyncio.get_running_loop()
-        except RuntimeError:
-            loop = asyncio.new_event_loop()
-            asyncio.set_event_loop(loop)
-
-        conn = new_conn(config, True)
-        self.client = DispatcherStub(conn)  # type: ignore[no-untyped-call]
         self.token = config.token
         self.config = config
 
@@ -91,6 +83,10 @@ class PooledWorkflowRunListener:
 
         self.interrupter: asyncio.Task[None] | None = None
 
+        ## IMPORTANT: This needs to be created lazily so we don't require
+        ## an event loop to instantiate the client.
+        self.client: DispatcherStub | None = None
+
     async def _interrupter(self) -> None:
         """
         _interrupter runs in a separate thread and interrupts the listener according to a configurable duration.
@@ -239,7 +235,7 @@ class PooledWorkflowRunListener:
             if subscription_id:
                 self.cleanup_subscription(subscription_id)
 
-    async def result(self, workflow_run_id: str) -> dict[str, Any]:
+    async def aio_result(self, workflow_run_id: str) -> dict[str, Any]:
         from hatchet_sdk.clients.admin import DedupeViolationErr
 
         event = await self.subscribe(workflow_run_id)
@@ -261,6 +257,9 @@ class PooledWorkflowRunListener:
         self,
     ) -> grpc.aio.UnaryStreamCall[SubscribeToWorkflowRunsRequest, WorkflowRunEvent]:
         retries = 0
+        if self.client is None:
+            conn = new_conn(self.config, True)
+            self.client = DispatcherStub(conn)
 
         while retries < DEFAULT_WORKFLOW_LISTENER_RETRY_COUNT:
             try:
@@ -276,7 +275,7 @@ class PooledWorkflowRunListener:
                         SubscribeToWorkflowRunsRequest, WorkflowRunEvent
                     ],
                     self.client.SubscribeToWorkflowRuns(
-                        self._request(),
+                        self._request(),  # type: ignore[arg-type]
                         metadata=get_metadata(self.token),
                     ),
                 )

hatchet_sdk/contracts/dispatcher_pb2.py

@@ -15,7 +15,7 @@ _sym_db = _symbol_database.Default()
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x10\x64ispatcher.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"V\n\x0cWorkerLabels\x12\x15\n\x08strValue\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x15\n\x08intValue\x18\x02 \x01(\x05H\x01\x88\x01\x01\x42\x0b\n\t_strValueB\x0b\n\t_intValue\"\xc8\x01\n\x0bRuntimeInfo\x12\x17\n\nsdkVersion\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x1c\n\x08language\x18\x02 \x01(\x0e\x32\x05.SDKSH\x01\x88\x01\x01\x12\x1c\n\x0flanguageVersion\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x0f\n\x02os\x18\x04 \x01(\tH\x03\x88\x01\x01\x12\x12\n\x05\x65xtra\x18\x05 \x01(\tH\x04\x88\x01\x01\x42\r\n\x0b_sdkVersionB\x0b\n\t_languageB\x12\n\x10_languageVersionB\x05\n\x03_osB\x08\n\x06_extra\"\xc0\x02\n\x15WorkerRegisterRequest\x12\x12\n\nworkerName\x18\x01 \x01(\t\x12\x0f\n\x07\x61\x63tions\x18\x02 \x03(\t\x12\x10\n\x08services\x18\x03 \x03(\t\x12\x14\n\x07maxRuns\x18\x04 \x01(\x05H\x00\x88\x01\x01\x12\x32\n\x06labels\x18\x05 \x03(\x0b\x32\".WorkerRegisterRequest.LabelsEntry\x12\x16\n\twebhookId\x18\x06 \x01(\tH\x01\x88\x01\x01\x12&\n\x0bruntimeInfo\x18\x07 \x01(\x0b\x32\x0c.RuntimeInfoH\x02\x88\x01\x01\x1a<\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1c\n\x05value\x18\x02 \x01(\x0b\x32\r.WorkerLabels:\x02\x38\x01\x42\n\n\x08_maxRunsB\x0c\n\n_webhookIdB\x0e\n\x0c_runtimeInfo\"P\n\x16WorkerRegisterResponse\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x10\n\x08workerId\x18\x02 \x01(\t\x12\x12\n\nworkerName\x18\x03 \x01(\t\"\xa3\x01\n\x19UpsertWorkerLabelsRequest\x12\x10\n\x08workerId\x18\x01 \x01(\t\x12\x36\n\x06labels\x18\x02 \x03(\x0b\x32&.UpsertWorkerLabelsRequest.LabelsEntry\x1a<\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1c\n\x05value\x18\x02 \x01(\x0b\x32\r.WorkerLabels:\x02\x38\x01\"@\n\x1aUpsertWorkerLabelsResponse\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x10\n\x08workerId\x18\x02 \x01(\t\"\x86\x04\n\x0e\x41ssignedAction\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x15\n\rworkflowRunId\x18\x02 \x01(\t\x12\x18\n\x10getGroupKeyRunId\x18\x03 \x01(\t\x12\r\n\x05jobId\x18\x04 \x01(\t\x12\x0f\n\x07jobName\x18\x05 \x01(\t\x12\x10\n\x08jobRunId\x18\x06 \x01(\t\x12\x0e\n\x06stepId\x18\x07 \x01(\t\x12\x11\n\tstepRunId\x18\x08 \x01(\t\x12\x10\n\x08\x61\x63tionId\x18\t \x01(\t\x12\x1f\n\nactionType\x18\n \x01(\x0e\x32\x0b.ActionType\x12\x15\n\ractionPayload\x18\x0b \x01(\t\x12\x10\n\x08stepName\x18\x0c \x01(\t\x12\x12\n\nretryCount\x18\r \x01(\x05\x12 \n\x13\x61\x64\x64itional_metadata\x18\x0e \x01(\tH\x00\x88\x01\x01\x12!\n\x14\x63hild_workflow_index\x18\x0f \x01(\x05H\x01\x88\x01\x01\x12\x1f\n\x12\x63hild_workflow_key\x18\x10 \x01(\tH\x02\x88\x01\x01\x12#\n\x16parent_workflow_run_id\x18\x11 \x01(\tH\x03\x88\x01\x01\x42\x16\n\x14_additional_metadataB\x17\n\x15_child_workflow_indexB\x15\n\x13_child_workflow_keyB\x19\n\x17_parent_workflow_run_id\"\'\n\x13WorkerListenRequest\x12\x10\n\x08workerId\x18\x01 \x01(\t\",\n\x18WorkerUnsubscribeRequest\x12\x10\n\x08workerId\x18\x01 \x01(\t\"?\n\x19WorkerUnsubscribeResponse\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x10\n\x08workerId\x18\x02 \x01(\t\"\xe1\x01\n\x13GroupKeyActionEvent\x12\x10\n\x08workerId\x18\x01 \x01(\t\x12\x15\n\rworkflowRunId\x18\x02 \x01(\t\x12\x18\n\x10getGroupKeyRunId\x18\x03 \x01(\t\x12\x10\n\x08\x61\x63tionId\x18\x04 \x01(\t\x12\x32\n\x0e\x65ventTimestamp\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12+\n\teventType\x18\x06 \x01(\x0e\x32\x18.GroupKeyActionEventType\x12\x14\n\x0c\x65ventPayload\x18\x07 \x01(\t\"\x94\x02\n\x0fStepActionEvent\x12\x10\n\x08workerId\x18\x01 \x01(\t\x12\r\n\x05jobId\x18\x02 \x01(\t\x12\x10\n\x08jobRunId\x18\x03 \x01(\t\x12\x0e\n\x06stepId\x18\x04 \x01(\t\x12\x11\n\tstepRunId\x18\x05 \x01(\t\x12\x10\n\x08\x61\x63tionId\x18\x06 \x01(\t\x12\x32\n\x0e\x65ventTimestamp\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\'\n\teventType\x18\x08 \x01(\x0e\x32\x14.StepActionEventType\x12\x14\n\x0c\x65ventPayload\x18\t \x01(\t\x12\x17\n\nretryCount\x18\n \x01(\x05H\x00\x88\x01\x01\x42\r\n\x0b_retryCount\"9\n\x13\x41\x63tionEventResponse\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x10\n\x08workerId\x18\x02 \x01(\t\"\xc0\x01\n SubscribeToWorkflowEventsRequest\x12\x1a\n\rworkflowRunId\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x1e\n\x11\x61\x64\x64itionalMetaKey\x18\x02 \x01(\tH\x01\x88\x01\x01\x12 \n\x13\x61\x64\x64itionalMetaValue\x18\x03 \x01(\tH\x02\x88\x01\x01\x42\x10\n\x0e_workflowRunIdB\x14\n\x12_additionalMetaKeyB\x16\n\x14_additionalMetaValue\"7\n\x1eSubscribeToWorkflowRunsRequest\x12\x15\n\rworkflowRunId\x18\x01 \x01(\t\"\xb2\x02\n\rWorkflowEvent\x12\x15\n\rworkflowRunId\x18\x01 \x01(\t\x12#\n\x0cresourceType\x18\x02 \x01(\x0e\x32\r.ResourceType\x12%\n\teventType\x18\x03 \x01(\x0e\x32\x12.ResourceEventType\x12\x12\n\nresourceId\x18\x04 \x01(\t\x12\x32\n\x0e\x65ventTimestamp\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x14\n\x0c\x65ventPayload\x18\x06 \x01(\t\x12\x0e\n\x06hangup\x18\x07 \x01(\x08\x12\x18\n\x0bstepRetries\x18\x08 \x01(\x05H\x00\x88\x01\x01\x12\x17\n\nretryCount\x18\t \x01(\x05H\x01\x88\x01\x01\x42\x0e\n\x0c_stepRetriesB\r\n\x0b_retryCount\"\xa8\x01\n\x10WorkflowRunEvent\x12\x15\n\rworkflowRunId\x18\x01 \x01(\t\x12(\n\teventType\x18\x02 \x01(\x0e\x32\x15.WorkflowRunEventType\x12\x32\n\x0e\x65ventTimestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x1f\n\x07results\x18\x04 \x03(\x0b\x32\x0e.StepRunResult\"\x8a\x01\n\rStepRunResult\x12\x11\n\tstepRunId\x18\x01 \x01(\t\x12\x16\n\x0estepReadableId\x18\x02 \x01(\t\x12\x10\n\x08jobRunId\x18\x03 \x01(\t\x12\x12\n\x05\x65rror\x18\x04 \x01(\tH\x00\x88\x01\x01\x12\x13\n\x06output\x18\x05 \x01(\tH\x01\x88\x01\x01\x42\x08\n\x06_errorB\t\n\x07_output\"W\n\rOverridesData\x12\x11\n\tstepRunId\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\t\x12\r\n\x05value\x18\x03 \x01(\t\x12\x16\n\x0e\x63\x61llerFilename\x18\x04 \x01(\t\"\x17\n\x15OverridesDataResponse\"U\n\x10HeartbeatRequest\x12\x10\n\x08workerId\x18\x01 \x01(\t\x12/\n\x0bheartbeatAt\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\x13\n\x11HeartbeatResponse\"F\n\x15RefreshTimeoutRequest\x12\x11\n\tstepRunId\x18\x01 \x01(\t\x12\x1a\n\x12incrementTimeoutBy\x18\x02 \x01(\t\"G\n\x16RefreshTimeoutResponse\x12-\n\ttimeoutAt\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\'\n\x12ReleaseSlotRequest\x12\x11\n\tstepRunId\x18\x01 \x01(\t\"\x15\n\x13ReleaseSlotResponse*7\n\x04SDKS\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x06\n\x02GO\x10\x01\x12\n\n\x06PYTHON\x10\x02\x12\x0e\n\nTYPESCRIPT\x10\x03*N\n\nActionType\x12\x12\n\x0eSTART_STEP_RUN\x10\x00\x12\x13\n\x0f\x43\x41NCEL_STEP_RUN\x10\x01\x12\x17\n\x13START_GET_GROUP_KEY\x10\x02*\xa2\x01\n\x17GroupKeyActionEventType\x12 \n\x1cGROUP_KEY_EVENT_TYPE_UNKNOWN\x10\x00\x12 \n\x1cGROUP_KEY_EVENT_TYPE_STARTED\x10\x01\x12\"\n\x1eGROUP_KEY_EVENT_TYPE_COMPLETED\x10\x02\x12\x1f\n\x1bGROUP_KEY_EVENT_TYPE_FAILED\x10\x03*\xac\x01\n\x13StepActionEventType\x12\x1b\n\x17STEP_EVENT_TYPE_UNKNOWN\x10\x00\x12\x1b\n\x17STEP_EVENT_TYPE_STARTED\x10\x01\x12\x1d\n\x19STEP_EVENT_TYPE_COMPLETED\x10\x02\x12\x1a\n\x16STEP_EVENT_TYPE_FAILED\x10\x03\x12 \n\x1cSTEP_EVENT_TYPE_ACKNOWLEDGED\x10\x04*e\n\x0cResourceType\x12\x19\n\x15RESOURCE_TYPE_UNKNOWN\x10\x00\x12\x1a\n\x16RESOURCE_TYPE_STEP_RUN\x10\x01\x12\x1e\n\x1aRESOURCE_TYPE_WORKFLOW_RUN\x10\x02*\xfe\x01\n\x11ResourceEventType\x12\x1f\n\x1bRESOURCE_EVENT_TYPE_UNKNOWN\x10\x00\x12\x1f\n\x1bRESOURCE_EVENT_TYPE_STARTED\x10\x01\x12!\n\x1dRESOURCE_EVENT_TYPE_COMPLETED\x10\x02\x12\x1e\n\x1aRESOURCE_EVENT_TYPE_FAILED\x10\x03\x12!\n\x1dRESOURCE_EVENT_TYPE_CANCELLED\x10\x04\x12!\n\x1dRESOURCE_EVENT_TYPE_TIMED_OUT\x10\x05\x12\x1e\n\x1aRESOURCE_EVENT_TYPE_STREAM\x10\x06*<\n\x14WorkflowRunEventType\x12$\n WORKFLOW_RUN_EVENT_TYPE_FINISHED\x10\x00\x32\xf8\x06\n\nDispatcher\x12=\n\x08Register\x12\x16.WorkerRegisterRequest\x1a\x17.WorkerRegisterResponse\"\x00\x12\x33\n\x06Listen\x12\x14.WorkerListenRequest\x1a\x0f.AssignedAction\"\x00\x30\x01\x12\x35\n\x08ListenV2\x12\x14.WorkerListenRequest\x1a\x0f.AssignedAction\"\x00\x30\x01\x12\x34\n\tHeartbeat\x12\x11.HeartbeatRequest\x1a\x12.HeartbeatResponse\"\x00\x12R\n\x19SubscribeToWorkflowEvents\x12!.SubscribeToWorkflowEventsRequest\x1a\x0e.WorkflowEvent\"\x00\x30\x01\x12S\n\x17SubscribeToWorkflowRuns\x12\x1f.SubscribeToWorkflowRunsRequest\x1a\x11.WorkflowRunEvent\"\x00(\x01\x30\x01\x12?\n\x13SendStepActionEvent\x12\x10.StepActionEvent\x1a\x14.ActionEventResponse\"\x00\x12G\n\x17SendGroupKeyActionEvent\x12\x14.GroupKeyActionEvent\x1a\x14.ActionEventResponse\"\x00\x12<\n\x10PutOverridesData\x12\x0e.OverridesData\x1a\x16.OverridesDataResponse\"\x00\x12\x46\n\x0bUnsubscribe\x12\x19.WorkerUnsubscribeRequest\x1a\x1a.WorkerUnsubscribeResponse\"\x00\x12\x43\n\x0eRefreshTimeout\x12\x16.RefreshTimeoutRequest\x1a\x17.RefreshTimeoutResponse\"\x00\x12:\n\x0bReleaseSlot\x12\x13.ReleaseSlotRequest\x1a\x14.ReleaseSlotResponse\"\x00\x12O\n\x12UpsertWorkerLabels\x12\x1a.UpsertWorkerLabelsRequest\x1a\x1b.UpsertWorkerLabelsResponse\"\x00\x42GZEgithub.com/hatchet-dev/hatchet/internal/services/dispatcher/contractsb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x10\x64ispatcher.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"V\n\x0cWorkerLabels\x12\x15\n\x08strValue\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x15\n\x08intValue\x18\x02 \x01(\x05H\x01\x88\x01\x01\x42\x0b\n\t_strValueB\x0b\n\t_intValue\"\xc8\x01\n\x0bRuntimeInfo\x12\x17\n\nsdkVersion\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x1c\n\x08language\x18\x02 \x01(\x0e\x32\x05.SDKSH\x01\x88\x01\x01\x12\x1c\n\x0flanguageVersion\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x0f\n\x02os\x18\x04 \x01(\tH\x03\x88\x01\x01\x12\x12\n\x05\x65xtra\x18\x05 \x01(\tH\x04\x88\x01\x01\x42\r\n\x0b_sdkVersionB\x0b\n\t_languageB\x12\n\x10_languageVersionB\x05\n\x03_osB\x08\n\x06_extra\"\xc0\x02\n\x15WorkerRegisterRequest\x12\x12\n\nworkerName\x18\x01 \x01(\t\x12\x0f\n\x07\x61\x63tions\x18\x02 \x03(\t\x12\x10\n\x08services\x18\x03 \x03(\t\x12\x14\n\x07maxRuns\x18\x04 \x01(\x05H\x00\x88\x01\x01\x12\x32\n\x06labels\x18\x05 \x03(\x0b\x32\".WorkerRegisterRequest.LabelsEntry\x12\x16\n\twebhookId\x18\x06 \x01(\tH\x01\x88\x01\x01\x12&\n\x0bruntimeInfo\x18\x07 \x01(\x0b\x32\x0c.RuntimeInfoH\x02\x88\x01\x01\x1a<\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1c\n\x05value\x18\x02 \x01(\x0b\x32\r.WorkerLabels:\x02\x38\x01\x42\n\n\x08_maxRunsB\x0c\n\n_webhookIdB\x0e\n\x0c_runtimeInfo\"P\n\x16WorkerRegisterResponse\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x10\n\x08workerId\x18\x02 \x01(\t\x12\x12\n\nworkerName\x18\x03 \x01(\t\"\xa3\x01\n\x19UpsertWorkerLabelsRequest\x12\x10\n\x08workerId\x18\x01 \x01(\t\x12\x36\n\x06labels\x18\x02 \x03(\x0b\x32&.UpsertWorkerLabelsRequest.LabelsEntry\x1a<\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1c\n\x05value\x18\x02 \x01(\x0b\x32\r.WorkerLabels:\x02\x38\x01\"@\n\x1aUpsertWorkerLabelsResponse\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x10\n\x08workerId\x18\x02 \x01(\t\"\x86\x04\n\x0e\x41ssignedAction\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x15\n\rworkflowRunId\x18\x02 \x01(\t\x12\x18\n\x10getGroupKeyRunId\x18\x03 \x01(\t\x12\r\n\x05jobId\x18\x04 \x01(\t\x12\x0f\n\x07jobName\x18\x05 \x01(\t\x12\x10\n\x08jobRunId\x18\x06 \x01(\t\x12\x0e\n\x06stepId\x18\x07 \x01(\t\x12\x11\n\tstepRunId\x18\x08 \x01(\t\x12\x10\n\x08\x61\x63tionId\x18\t \x01(\t\x12\x1f\n\nactionType\x18\n \x01(\x0e\x32\x0b.ActionType\x12\x15\n\ractionPayload\x18\x0b \x01(\t\x12\x10\n\x08stepName\x18\x0c \x01(\t\x12\x12\n\nretryCount\x18\r \x01(\x05\x12 \n\x13\x61\x64\x64itional_metadata\x18\x0e \x01(\tH\x00\x88\x01\x01\x12!\n\x14\x63hild_workflow_index\x18\x0f \x01(\x05H\x01\x88\x01\x01\x12\x1f\n\x12\x63hild_workflow_key\x18\x10 \x01(\tH\x02\x88\x01\x01\x12#\n\x16parent_workflow_run_id\x18\x11 \x01(\tH\x03\x88\x01\x01\x42\x16\n\x14_additional_metadataB\x17\n\x15_child_workflow_indexB\x15\n\x13_child_workflow_keyB\x19\n\x17_parent_workflow_run_id\"\'\n\x13WorkerListenRequest\x12\x10\n\x08workerId\x18\x01 \x01(\t\",\n\x18WorkerUnsubscribeRequest\x12\x10\n\x08workerId\x18\x01 \x01(\t\"?\n\x19WorkerUnsubscribeResponse\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x10\n\x08workerId\x18\x02 \x01(\t\"\xe1\x01\n\x13GroupKeyActionEvent\x12\x10\n\x08workerId\x18\x01 \x01(\t\x12\x15\n\rworkflowRunId\x18\x02 \x01(\t\x12\x18\n\x10getGroupKeyRunId\x18\x03 \x01(\t\x12\x10\n\x08\x61\x63tionId\x18\x04 \x01(\t\x12\x32\n\x0e\x65ventTimestamp\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12+\n\teventType\x18\x06 \x01(\x0e\x32\x18.GroupKeyActionEventType\x12\x14\n\x0c\x65ventPayload\x18\x07 \x01(\t\"\xc4\x02\n\x0fStepActionEvent\x12\x10\n\x08workerId\x18\x01 \x01(\t\x12\r\n\x05jobId\x18\x02 \x01(\t\x12\x10\n\x08jobRunId\x18\x03 \x01(\t\x12\x0e\n\x06stepId\x18\x04 \x01(\t\x12\x11\n\tstepRunId\x18\x05 \x01(\t\x12\x10\n\x08\x61\x63tionId\x18\x06 \x01(\t\x12\x32\n\x0e\x65ventTimestamp\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\'\n\teventType\x18\x08 \x01(\x0e\x32\x14.StepActionEventType\x12\x14\n\x0c\x65ventPayload\x18\t \x01(\t\x12\x17\n\nretryCount\x18\n \x01(\x05H\x00\x88\x01\x01\x12\x1b\n\x0eshouldNotRetry\x18\x0b \x01(\x08H\x01\x88\x01\x01\x42\r\n\x0b_retryCountB\x11\n\x0f_shouldNotRetry\"9\n\x13\x41\x63tionEventResponse\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x10\n\x08workerId\x18\x02 \x01(\t\"\xc0\x01\n SubscribeToWorkflowEventsRequest\x12\x1a\n\rworkflowRunId\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x1e\n\x11\x61\x64\x64itionalMetaKey\x18\x02 \x01(\tH\x01\x88\x01\x01\x12 \n\x13\x61\x64\x64itionalMetaValue\x18\x03 \x01(\tH\x02\x88\x01\x01\x42\x10\n\x0e_workflowRunIdB\x14\n\x12_additionalMetaKeyB\x16\n\x14_additionalMetaValue\"7\n\x1eSubscribeToWorkflowRunsRequest\x12\x15\n\rworkflowRunId\x18\x01 \x01(\t\"\xb2\x02\n\rWorkflowEvent\x12\x15\n\rworkflowRunId\x18\x01 \x01(\t\x12#\n\x0cresourceType\x18\x02 \x01(\x0e\x32\r.ResourceType\x12%\n\teventType\x18\x03 \x01(\x0e\x32\x12.ResourceEventType\x12\x12\n\nresourceId\x18\x04 \x01(\t\x12\x32\n\x0e\x65ventTimestamp\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x14\n\x0c\x65ventPayload\x18\x06 \x01(\t\x12\x0e\n\x06hangup\x18\x07 \x01(\x08\x12\x18\n\x0bstepRetries\x18\x08 \x01(\x05H\x00\x88\x01\x01\x12\x17\n\nretryCount\x18\t \x01(\x05H\x01\x88\x01\x01\x42\x0e\n\x0c_stepRetriesB\r\n\x0b_retryCount\"\xa8\x01\n\x10WorkflowRunEvent\x12\x15\n\rworkflowRunId\x18\x01 \x01(\t\x12(\n\teventType\x18\x02 \x01(\x0e\x32\x15.WorkflowRunEventType\x12\x32\n\x0e\x65ventTimestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x1f\n\x07results\x18\x04 \x03(\x0b\x32\x0e.StepRunResult\"\x8a\x01\n\rStepRunResult\x12\x11\n\tstepRunId\x18\x01 \x01(\t\x12\x16\n\x0estepReadableId\x18\x02 \x01(\t\x12\x10\n\x08jobRunId\x18\x03 \x01(\t\x12\x12\n\x05\x65rror\x18\x04 \x01(\tH\x00\x88\x01\x01\x12\x13\n\x06output\x18\x05 \x01(\tH\x01\x88\x01\x01\x42\x08\n\x06_errorB\t\n\x07_output\"W\n\rOverridesData\x12\x11\n\tstepRunId\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\t\x12\r\n\x05value\x18\x03 \x01(\t\x12\x16\n\x0e\x63\x61llerFilename\x18\x04 \x01(\t\"\x17\n\x15OverridesDataResponse\"U\n\x10HeartbeatRequest\x12\x10\n\x08workerId\x18\x01 \x01(\t\x12/\n\x0bheartbeatAt\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\x13\n\x11HeartbeatResponse\"F\n\x15RefreshTimeoutRequest\x12\x11\n\tstepRunId\x18\x01 \x01(\t\x12\x1a\n\x12incrementTimeoutBy\x18\x02 \x01(\t\"G\n\x16RefreshTimeoutResponse\x12-\n\ttimeoutAt\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\'\n\x12ReleaseSlotRequest\x12\x11\n\tstepRunId\x18\x01 \x01(\t\"\x15\n\x13ReleaseSlotResponse*7\n\x04SDKS\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x06\n\x02GO\x10\x01\x12\n\n\x06PYTHON\x10\x02\x12\x0e\n\nTYPESCRIPT\x10\x03*N\n\nActionType\x12\x12\n\x0eSTART_STEP_RUN\x10\x00\x12\x13\n\x0f\x43\x41NCEL_STEP_RUN\x10\x01\x12\x17\n\x13START_GET_GROUP_KEY\x10\x02*\xa2\x01\n\x17GroupKeyActionEventType\x12 \n\x1cGROUP_KEY_EVENT_TYPE_UNKNOWN\x10\x00\x12 \n\x1cGROUP_KEY_EVENT_TYPE_STARTED\x10\x01\x12\"\n\x1eGROUP_KEY_EVENT_TYPE_COMPLETED\x10\x02\x12\x1f\n\x1bGROUP_KEY_EVENT_TYPE_FAILED\x10\x03*\xac\x01\n\x13StepActionEventType\x12\x1b\n\x17STEP_EVENT_TYPE_UNKNOWN\x10\x00\x12\x1b\n\x17STEP_EVENT_TYPE_STARTED\x10\x01\x12\x1d\n\x19STEP_EVENT_TYPE_COMPLETED\x10\x02\x12\x1a\n\x16STEP_EVENT_TYPE_FAILED\x10\x03\x12 \n\x1cSTEP_EVENT_TYPE_ACKNOWLEDGED\x10\x04*e\n\x0cResourceType\x12\x19\n\x15RESOURCE_TYPE_UNKNOWN\x10\x00\x12\x1a\n\x16RESOURCE_TYPE_STEP_RUN\x10\x01\x12\x1e\n\x1aRESOURCE_TYPE_WORKFLOW_RUN\x10\x02*\xfe\x01\n\x11ResourceEventType\x12\x1f\n\x1bRESOURCE_EVENT_TYPE_UNKNOWN\x10\x00\x12\x1f\n\x1bRESOURCE_EVENT_TYPE_STARTED\x10\x01\x12!\n\x1dRESOURCE_EVENT_TYPE_COMPLETED\x10\x02\x12\x1e\n\x1aRESOURCE_EVENT_TYPE_FAILED\x10\x03\x12!\n\x1dRESOURCE_EVENT_TYPE_CANCELLED\x10\x04\x12!\n\x1dRESOURCE_EVENT_TYPE_TIMED_OUT\x10\x05\x12\x1e\n\x1aRESOURCE_EVENT_TYPE_STREAM\x10\x06*<\n\x14WorkflowRunEventType\x12$\n WORKFLOW_RUN_EVENT_TYPE_FINISHED\x10\x00\x32\xf8\x06\n\nDispatcher\x12=\n\x08Register\x12\x16.WorkerRegisterRequest\x1a\x17.WorkerRegisterResponse\"\x00\x12\x33\n\x06Listen\x12\x14.WorkerListenRequest\x1a\x0f.AssignedAction\"\x00\x30\x01\x12\x35\n\x08ListenV2\x12\x14.WorkerListenRequest\x1a\x0f.AssignedAction\"\x00\x30\x01\x12\x34\n\tHeartbeat\x12\x11.HeartbeatRequest\x1a\x12.HeartbeatResponse\"\x00\x12R\n\x19SubscribeToWorkflowEvents\x12!.SubscribeToWorkflowEventsRequest\x1a\x0e.WorkflowEvent\"\x00\x30\x01\x12S\n\x17SubscribeToWorkflowRuns\x12\x1f.SubscribeToWorkflowRunsRequest\x1a\x11.WorkflowRunEvent\"\x00(\x01\x30\x01\x12?\n\x13SendStepActionEvent\x12\x10.StepActionEvent\x1a\x14.ActionEventResponse\"\x00\x12G\n\x17SendGroupKeyActionEvent\x12\x14.GroupKeyActionEvent\x1a\x14.ActionEventResponse\"\x00\x12<\n\x10PutOverridesData\x12\x0e.OverridesData\x1a\x16.OverridesDataResponse\"\x00\x12\x46\n\x0bUnsubscribe\x12\x19.WorkerUnsubscribeRequest\x1a\x1a.WorkerUnsubscribeResponse\"\x00\x12\x43\n\x0eRefreshTimeout\x12\x16.RefreshTimeoutRequest\x1a\x17.RefreshTimeoutResponse\"\x00\x12:\n\x0bReleaseSlot\x12\x13.ReleaseSlotRequest\x1a\x14.ReleaseSlotResponse\"\x00\x12O\n\x12UpsertWorkerLabels\x12\x1a.UpsertWorkerLabelsRequest\x1a\x1b.UpsertWorkerLabelsResponse\"\x00\x42GZEgithub.com/hatchet-dev/hatchet/internal/services/dispatcher/contractsb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -27,20 +27,20 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_WORKERREGISTERREQUEST_LABELSENTRY']._serialized_options = b'8\001'
   _globals['_UPSERTWORKERLABELSREQUEST_LABELSENTRY']._loaded_options = None
   _globals['_UPSERTWORKERLABELSREQUEST_LABELSENTRY']._serialized_options = b'8\001'
-  _globals['_SDKS']._serialized_start=3524
-  _globals['_SDKS']._serialized_end=3579
-  _globals['_ACTIONTYPE']._serialized_start=3581
-  _globals['_ACTIONTYPE']._serialized_end=3659
-  _globals['_GROUPKEYACTIONEVENTTYPE']._serialized_start=3662
-  _globals['_GROUPKEYACTIONEVENTTYPE']._serialized_end=3824
-  _globals['_STEPACTIONEVENTTYPE']._serialized_start=3827
-  _globals['_STEPACTIONEVENTTYPE']._serialized_end=3999
-  _globals['_RESOURCETYPE']._serialized_start=4001
-  _globals['_RESOURCETYPE']._serialized_end=4102
-  _globals['_RESOURCEEVENTTYPE']._serialized_start=4105
-  _globals['_RESOURCEEVENTTYPE']._serialized_end=4359
-  _globals['_WORKFLOWRUNEVENTTYPE']._serialized_start=4361
-  _globals['_WORKFLOWRUNEVENTTYPE']._serialized_end=4421
+  _globals['_SDKS']._serialized_start=3572
+  _globals['_SDKS']._serialized_end=3627
+  _globals['_ACTIONTYPE']._serialized_start=3629
+  _globals['_ACTIONTYPE']._serialized_end=3707
+  _globals['_GROUPKEYACTIONEVENTTYPE']._serialized_start=3710
+  _globals['_GROUPKEYACTIONEVENTTYPE']._serialized_end=3872
+  _globals['_STEPACTIONEVENTTYPE']._serialized_start=3875
+  _globals['_STEPACTIONEVENTTYPE']._serialized_end=4047
+  _globals['_RESOURCETYPE']._serialized_start=4049
+  _globals['_RESOURCETYPE']._serialized_end=4150
+  _globals['_RESOURCEEVENTTYPE']._serialized_start=4153
+  _globals['_RESOURCEEVENTTYPE']._serialized_end=4407
+  _globals['_WORKFLOWRUNEVENTTYPE']._serialized_start=4409
+  _globals['_WORKFLOWRUNEVENTTYPE']._serialized_end=4469
   _globals['_WORKERLABELS']._serialized_start=53
   _globals['_WORKERLABELS']._serialized_end=139
   _globals['_RUNTIMEINFO']._serialized_start=142
@@ -68,35 +68,35 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_GROUPKEYACTIONEVENT']._serialized_start=1655
   _globals['_GROUPKEYACTIONEVENT']._serialized_end=1880
   _globals['_STEPACTIONEVENT']._serialized_start=1883
-  _globals['_STEPACTIONEVENT']._serialized_end=2159
-  _globals['_ACTIONEVENTRESPONSE']._serialized_start=2161
-  _globals['_ACTIONEVENTRESPONSE']._serialized_end=2218
-  _globals['_SUBSCRIBETOWORKFLOWEVENTSREQUEST']._serialized_start=2221
-  _globals['_SUBSCRIBETOWORKFLOWEVENTSREQUEST']._serialized_end=2413
-  _globals['_SUBSCRIBETOWORKFLOWRUNSREQUEST']._serialized_start=2415
-  _globals['_SUBSCRIBETOWORKFLOWRUNSREQUEST']._serialized_end=2470
-  _globals['_WORKFLOWEVENT']._serialized_start=2473
-  _globals['_WORKFLOWEVENT']._serialized_end=2779
-  _globals['_WORKFLOWRUNEVENT']._serialized_start=2782
-  _globals['_WORKFLOWRUNEVENT']._serialized_end=2950
-  _globals['_STEPRUNRESULT']._serialized_start=2953
-  _globals['_STEPRUNRESULT']._serialized_end=3091
-  _globals['_OVERRIDESDATA']._serialized_start=3093
-  _globals['_OVERRIDESDATA']._serialized_end=3180
-  _globals['_OVERRIDESDATARESPONSE']._serialized_start=3182
-  _globals['_OVERRIDESDATARESPONSE']._serialized_end=3205
-  _globals['_HEARTBEATREQUEST']._serialized_start=3207
-  _globals['_HEARTBEATREQUEST']._serialized_end=3292
-  _globals['_HEARTBEATRESPONSE']._serialized_start=3294
-  _globals['_HEARTBEATRESPONSE']._serialized_end=3313
-  _globals['_REFRESHTIMEOUTREQUEST']._serialized_start=3315
-  _globals['_REFRESHTIMEOUTREQUEST']._serialized_end=3385
-  _globals['_REFRESHTIMEOUTRESPONSE']._serialized_start=3387
-  _globals['_REFRESHTIMEOUTRESPONSE']._serialized_end=3458
-  _globals['_RELEASESLOTREQUEST']._serialized_start=3460
-  _globals['_RELEASESLOTREQUEST']._serialized_end=3499
-  _globals['_RELEASESLOTRESPONSE']._serialized_start=3501
-  _globals['_RELEASESLOTRESPONSE']._serialized_end=3522
-  _globals['_DISPATCHER']._serialized_start=4424
-  _globals['_DISPATCHER']._serialized_end=5312
+  _globals['_STEPACTIONEVENT']._serialized_end=2207
+  _globals['_ACTIONEVENTRESPONSE']._serialized_start=2209
+  _globals['_ACTIONEVENTRESPONSE']._serialized_end=2266
+  _globals['_SUBSCRIBETOWORKFLOWEVENTSREQUEST']._serialized_start=2269
+  _globals['_SUBSCRIBETOWORKFLOWEVENTSREQUEST']._serialized_end=2461
+  _globals['_SUBSCRIBETOWORKFLOWRUNSREQUEST']._serialized_start=2463
+  _globals['_SUBSCRIBETOWORKFLOWRUNSREQUEST']._serialized_end=2518
+  _globals['_WORKFLOWEVENT']._serialized_start=2521
+  _globals['_WORKFLOWEVENT']._serialized_end=2827
+  _globals['_WORKFLOWRUNEVENT']._serialized_start=2830
+  _globals['_WORKFLOWRUNEVENT']._serialized_end=2998
+  _globals['_STEPRUNRESULT']._serialized_start=3001
+  _globals['_STEPRUNRESULT']._serialized_end=3139
+  _globals['_OVERRIDESDATA']._serialized_start=3141
+  _globals['_OVERRIDESDATA']._serialized_end=3228
+  _globals['_OVERRIDESDATARESPONSE']._serialized_start=3230
+  _globals['_OVERRIDESDATARESPONSE']._serialized_end=3253
+  _globals['_HEARTBEATREQUEST']._serialized_start=3255
+  _globals['_HEARTBEATREQUEST']._serialized_end=3340
+  _globals['_HEARTBEATRESPONSE']._serialized_start=3342
+  _globals['_HEARTBEATRESPONSE']._serialized_end=3361
+  _globals['_REFRESHTIMEOUTREQUEST']._serialized_start=3363
+  _globals['_REFRESHTIMEOUTREQUEST']._serialized_end=3433
+  _globals['_REFRESHTIMEOUTRESPONSE']._serialized_start=3435
+  _globals['_REFRESHTIMEOUTRESPONSE']._serialized_end=3506
+  _globals['_RELEASESLOTREQUEST']._serialized_start=3508
+  _globals['_RELEASESLOTREQUEST']._serialized_end=3547
+  _globals['_RELEASESLOTRESPONSE']._serialized_start=3549
+  _globals['_RELEASESLOTRESPONSE']._serialized_end=3570
+  _globals['_DISPATCHER']._serialized_start=4472
+  _globals['_DISPATCHER']._serialized_end=5360
 # @@protoc_insertion_point(module_scope)

hatchet_sdk/contracts/dispatcher_pb2.pyi

@@ -239,7 +239,7 @@ class GroupKeyActionEvent(_message.Message):
     def __init__(self, workerId: _Optional[str] = ..., workflowRunId: _Optional[str] = ..., getGroupKeyRunId: _Optional[str] = ..., actionId: _Optional[str] = ..., eventTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., eventType: _Optional[_Union[GroupKeyActionEventType, str]] = ..., eventPayload: _Optional[str] = ...) -> None: ...
 
 class StepActionEvent(_message.Message):
-    __slots__ = ("workerId", "jobId", "jobRunId", "stepId", "stepRunId", "actionId", "eventTimestamp", "eventType", "eventPayload", "retryCount")
+    __slots__ = ("workerId", "jobId", "jobRunId", "stepId", "stepRunId", "actionId", "eventTimestamp", "eventType", "eventPayload", "retryCount", "shouldNotRetry")
     WORKERID_FIELD_NUMBER: _ClassVar[int]
     JOBID_FIELD_NUMBER: _ClassVar[int]
     JOBRUNID_FIELD_NUMBER: _ClassVar[int]
@@ -250,6 +250,7 @@ class StepActionEvent(_message.Message):
     EVENTTYPE_FIELD_NUMBER: _ClassVar[int]
     EVENTPAYLOAD_FIELD_NUMBER: _ClassVar[int]
     RETRYCOUNT_FIELD_NUMBER: _ClassVar[int]
+    SHOULDNOTRETRY_FIELD_NUMBER: _ClassVar[int]
     workerId: str
     jobId: str
     jobRunId: str
@@ -260,7 +261,8 @@ class StepActionEvent(_message.Message):
     eventType: StepActionEventType
     eventPayload: str
     retryCount: int
-    def __init__(self, workerId: _Optional[str] = ..., jobId: _Optional[str] = ..., jobRunId: _Optional[str] = ..., stepId: _Optional[str] = ..., stepRunId: _Optional[str] = ..., actionId: _Optional[str] = ..., eventTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., eventType: _Optional[_Union[StepActionEventType, str]] = ..., eventPayload: _Optional[str] = ..., retryCount: _Optional[int] = ...) -> None: ...
+    shouldNotRetry: bool
+    def __init__(self, workerId: _Optional[str] = ..., jobId: _Optional[str] = ..., jobRunId: _Optional[str] = ..., stepId: _Optional[str] = ..., stepRunId: _Optional[str] = ..., actionId: _Optional[str] = ..., eventTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., eventType: _Optional[_Union[StepActionEventType, str]] = ..., eventPayload: _Optional[str] = ..., retryCount: _Optional[int] = ..., shouldNotRetry: bool = ...) -> None: ...
 
 class ActionEventResponse(_message.Message):
     __slots__ = ("tenantId", "workerId")

hatchet_sdk/contracts/dispatcher_pb2_grpc.py

@@ -33,7 +33,7 @@ if _version_not_supported:
 class DispatcherStub(object):
     """Missing associated documentation comment in .proto file."""
 
-    def __init__(self, channel):
+    def __init__(self, channel: grpc.Channel | grpc.aio.Channel) -> None:
         """Constructor.
 
         Args:

hatchet_sdk/contracts/events_pb2_grpc.py

@@ -33,7 +33,7 @@ if _version_not_supported:
 class EventsServiceStub(object):
     """Missing associated documentation comment in .proto file."""
 
-    def __init__(self, channel):
+    def __init__(self, channel: grpc.Channel | grpc.aio.Channel) -> None:
         """Constructor.
 
         Args:

hatchet_sdk/contracts/v1/dispatcher_pb2_grpc.py

@@ -33,7 +33,7 @@ if _version_not_supported:
 class V1DispatcherStub(object):
     """Missing associated documentation comment in .proto file."""
 
-    def __init__(self, channel):
+    def __init__(self, channel: grpc.Channel | grpc.aio.Channel) -> None:
         """Constructor.
 
         Args:

hatchet_sdk/contracts/v1/workflows_pb2_grpc.py

@@ -34,7 +34,7 @@ class AdminServiceStub(object):
     """AdminService represents a set of RPCs for admin management of tasks, workflows, etc.
     """
 
-    def __init__(self, channel):
+    def __init__(self, channel: grpc.Channel | grpc.aio.Channel) -> None:
         """Constructor.
 
         Args:

hatchet_sdk/contracts/workflows_pb2_grpc.py

@@ -34,7 +34,7 @@ class WorkflowServiceStub(object):
     """WorkflowService represents a set of RPCs for managing workflows.
     """
 
-    def __init__(self, channel):
+    def __init__(self, channel: grpc.Channel | grpc.aio.Channel) -> None:
         """Constructor.
 
         Args:

hatchet_sdk/features/cron.py

@@ -18,6 +18,7 @@ from hatchet_sdk.clients.v1.api_client import (
     BaseRestClient,
     maybe_additional_metadata_to_kv,
 )
+from hatchet_sdk.utils.aio import run_async_from_sync
 from hatchet_sdk.utils.typing import JSONSerializableMapping
 
 
@@ -121,7 +122,7 @@ class CronClient(BaseRestClient):
         input: JSONSerializableMapping,
         additional_metadata: JSONSerializableMapping,
     ) -> CronWorkflows:
-        return self._run_async_from_sync(
+        return run_async_from_sync(
             self.aio_create,
             workflow_name,
             cron_name,
@@ -143,7 +144,7 @@ class CronClient(BaseRestClient):
             )
 
     def delete(self, cron_id: str) -> None:
-        return self._run_async_from_sync(self.aio_delete, cron_id)
+        return run_async_from_sync(self.aio_delete, cron_id)
 
     async def aio_list(
         self,
@@ -204,7 +205,7 @@ class CronClient(BaseRestClient):
         Returns:
             CronWorkflowsList: A list of cron workflows.
         """
-        return self._run_async_from_sync(
+        return run_async_from_sync(
             self.aio_list,
             offset=offset,
             limit=limit,
@@ -239,4 +240,4 @@ class CronClient(BaseRestClient):
         Returns:
             CronWorkflows: The requested cron workflow instance.
         """
-        return self._run_async_from_sync(self.aio_get, cron_id)
+        return run_async_from_sync(self.aio_get, cron_id)

hatchet_sdk/features/logs.py

@@ -2,6 +2,7 @@ from hatchet_sdk.clients.rest.api.log_api import LogApi
 from hatchet_sdk.clients.rest.api_client import ApiClient
 from hatchet_sdk.clients.rest.models.v1_log_line_list import V1LogLineList
 from hatchet_sdk.clients.v1.api_client import BaseRestClient
+from hatchet_sdk.utils.aio import run_async_from_sync
 
 
 class LogsClient(BaseRestClient):
@@ -13,4 +14,4 @@ class LogsClient(BaseRestClient):
             return await self._la(client).v1_log_line_list(task=task_run_id)
 
     def list(self, task_run_id: str) -> V1LogLineList:
-        return self._run_async_from_sync(self.aio_list, task_run_id)
+        return run_async_from_sync(self.aio_list, task_run_id)

hatchet_sdk/features/metrics.py

@@ -11,6 +11,7 @@ from hatchet_sdk.clients.v1.api_client import (
     BaseRestClient,
     maybe_additional_metadata_to_kv,
 )
+from hatchet_sdk.utils.aio import run_async_from_sync
 from hatchet_sdk.utils.typing import JSONSerializableMapping
 
 
@@ -38,7 +39,7 @@ class MetricsClient(BaseRestClient):
         status: WorkflowRunStatus | None = None,
         group_key: str | None = None,
     ) -> WorkflowMetrics:
-        return self._run_async_from_sync(
+        return run_async_from_sync(
             self.aio_get_workflow_metrics, workflow_id, status, group_key
         )
 
@@ -61,7 +62,7 @@ class MetricsClient(BaseRestClient):
         workflow_ids: list[str] | None = None,
         additional_metadata: JSONSerializableMapping | None = None,
     ) -> TenantQueueMetrics:
-        return self._run_async_from_sync(
+        return run_async_from_sync(
             self.aio_get_queue_metrics, workflow_ids, additional_metadata
         )
 
@@ -72,4 +73,4 @@ class MetricsClient(BaseRestClient):
             )
 
     def get_task_metrics(self) -> TenantStepRunQueueMetrics:
-        return self._run_async_from_sync(self.aio_get_task_metrics)
+        return run_async_from_sync(self.aio_get_task_metrics)

hatchet_sdk/features/rate_limits.py

@@ -24,7 +24,7 @@ class RateLimitsClient(BaseRestClient):
         )
 
         conn = new_conn(self.client_config, False)
-        client = WorkflowServiceStub(conn)  # type: ignore[no-untyped-call]
+        client = WorkflowServiceStub(conn)
 
         client.PutRateLimit(
             v0_workflow_protos.PutRateLimitRequest(

hatchet_sdk/features/runs.py

@@ -19,6 +19,7 @@ from hatchet_sdk.clients.v1.api_client import (
     BaseRestClient,
     maybe_additional_metadata_to_kv,
 )
+from hatchet_sdk.utils.aio import run_async_from_sync
 from hatchet_sdk.utils.typing import JSONSerializableMapping
 
 
@@ -93,7 +94,7 @@ class RunsClient(BaseRestClient):
             return await self._wra(client).v1_workflow_run_get(str(workflow_run_id))
 
     def get(self, workflow_run_id: str) -> V1WorkflowRunDetails:
-        return self._run_async_from_sync(self.aio_get, workflow_run_id)
+        return run_async_from_sync(self.aio_get, workflow_run_id)
 
     async def aio_list(
         self,
@@ -138,7 +139,7 @@ class RunsClient(BaseRestClient):
         worker_id: str | None = None,
         parent_task_external_id: str | None = None,
     ) -> V1TaskSummaryList:
-        return self._run_async_from_sync(
+        return run_async_from_sync(
             self.aio_list,
             since=since,
             only_tasks=only_tasks,
@@ -174,7 +175,7 @@ class RunsClient(BaseRestClient):
         input: JSONSerializableMapping,
         additional_metadata: JSONSerializableMapping = {},
     ) -> V1WorkflowRunDetails:
-        return self._run_async_from_sync(
+        return run_async_from_sync(
             self.aio_create, workflow_name, input, additional_metadata
         )
 
@@ -182,7 +183,7 @@ class RunsClient(BaseRestClient):
         await self.aio_bulk_replay(opts=BulkCancelReplayOpts(ids=[run_id]))
 
     def replay(self, run_id: str) -> None:
-        return self._run_async_from_sync(self.aio_replay, run_id)
+        return run_async_from_sync(self.aio_replay, run_id)
 
     async def aio_bulk_replay(self, opts: BulkCancelReplayOpts) -> None:
         async with self.client() as client:
@@ -192,13 +193,13 @@ class RunsClient(BaseRestClient):
             )
 
     def bulk_replay(self, opts: BulkCancelReplayOpts) -> None:
-        return self._run_async_from_sync(self.aio_bulk_replay, opts)
+        return run_async_from_sync(self.aio_bulk_replay, opts)
 
     async def aio_cancel(self, run_id: str) -> None:
         await self.aio_bulk_cancel(opts=BulkCancelReplayOpts(ids=[run_id]))
 
     def cancel(self, run_id: str) -> None:
-        return self._run_async_from_sync(self.aio_cancel, run_id)
+        return run_async_from_sync(self.aio_cancel, run_id)
 
     async def aio_bulk_cancel(self, opts: BulkCancelReplayOpts) -> None:
         async with self.client() as client:
@@ -208,7 +209,7 @@ class RunsClient(BaseRestClient):
             )
 
     def bulk_cancel(self, opts: BulkCancelReplayOpts) -> None:
-        return self._run_async_from_sync(self.aio_bulk_cancel, opts)
+        return run_async_from_sync(self.aio_bulk_cancel, opts)
 
     async def aio_get_result(self, run_id: str) -> JSONSerializableMapping:
         details = await self.aio_get(run_id)

hatchet_sdk/features/scheduled.py

@@ -22,6 +22,7 @@ from hatchet_sdk.clients.v1.api_client import (
     BaseRestClient,
     maybe_additional_metadata_to_kv,
 )
+from hatchet_sdk.utils.aio import run_async_from_sync
 from hatchet_sdk.utils.typing import JSONSerializableMapping
 
 
@@ -82,7 +83,7 @@ class ScheduledClient(BaseRestClient):
             ScheduledWorkflows: The created scheduled workflow instance.
         """
 
-        return self._run_async_from_sync(
+        return run_async_from_sync(
             self.aio_create,
             workflow_name,
             trigger_at,
@@ -104,7 +105,7 @@ class ScheduledClient(BaseRestClient):
             )
 
     def delete(self, scheduled_id: str) -> None:
-        self._run_async_from_sync(self.aio_delete, scheduled_id)
+        run_async_from_sync(self.aio_delete, scheduled_id)
 
     async def aio_list(
         self,
@@ -175,7 +176,7 @@ class ScheduledClient(BaseRestClient):
         Returns:
             List[ScheduledWorkflows]: A list of scheduled workflows matching the criteria.
         """
-        return self._run_async_from_sync(
+        return run_async_from_sync(
             self.aio_list,
             offset=offset,
             limit=limit,
@@ -214,4 +215,4 @@ class ScheduledClient(BaseRestClient):
         Returns:
             ScheduledWorkflows: The requested scheduled workflow instance.
         """
-        return self._run_async_from_sync(self.aio_get, scheduled_id)
+        return run_async_from_sync(self.aio_get, scheduled_id)

hatchet_sdk/features/workers.py

@@ -4,6 +4,7 @@ from hatchet_sdk.clients.rest.models.update_worker_request import UpdateWorkerRe
 from hatchet_sdk.clients.rest.models.worker import Worker
 from hatchet_sdk.clients.rest.models.worker_list import WorkerList
 from hatchet_sdk.clients.v1.api_client import BaseRestClient
+from hatchet_sdk.utils.aio import run_async_from_sync
 
 
 class WorkersClient(BaseRestClient):
@@ -15,7 +16,7 @@ class WorkersClient(BaseRestClient):
             return await self._wa(client).worker_get(worker_id)
 
     def get(self, worker_id: str) -> Worker:
-        return self._run_async_from_sync(self.aio_get, worker_id)
+        return run_async_from_sync(self.aio_get, worker_id)
 
     async def aio_list(
         self,
@@ -28,7 +29,7 @@ class WorkersClient(BaseRestClient):
     def list(
         self,
     ) -> WorkerList:
-        return self._run_async_from_sync(self.aio_list)
+        return run_async_from_sync(self.aio_list)
 
     async def aio_update(self, worker_id: str, opts: UpdateWorkerRequest) -> Worker:
         async with self.client() as client:
@@ -38,4 +39,4 @@ class WorkersClient(BaseRestClient):
             )
 
     def update(self, worker_id: str, opts: UpdateWorkerRequest) -> Worker:
-        return self._run_async_from_sync(self.aio_update, worker_id, opts)
+        return run_async_from_sync(self.aio_update, worker_id, opts)

hatchet_sdk/features/workflows.py

@@ -5,6 +5,7 @@ from hatchet_sdk.clients.rest.models.workflow import Workflow
 from hatchet_sdk.clients.rest.models.workflow_list import WorkflowList
 from hatchet_sdk.clients.rest.models.workflow_version import WorkflowVersion
 from hatchet_sdk.clients.v1.api_client import BaseRestClient
+from hatchet_sdk.utils.aio import run_async_from_sync
 
 
 class WorkflowsClient(BaseRestClient):
@@ -19,7 +20,7 @@ class WorkflowsClient(BaseRestClient):
             return await self._wa(client).workflow_get(workflow_id)
 
     def get(self, workflow_id: str) -> Workflow:
-        return self._run_async_from_sync(self.aio_get, workflow_id)
+        return run_async_from_sync(self.aio_get, workflow_id)
 
     async def aio_list(
         self,
@@ -41,7 +42,7 @@ class WorkflowsClient(BaseRestClient):
         limit: int | None = None,
         offset: int | None = None,
     ) -> WorkflowList:
-        return self._run_async_from_sync(self.aio_list, workflow_name, limit, offset)
+        return run_async_from_sync(self.aio_list, workflow_name, limit, offset)
 
     async def aio_get_version(
         self, workflow_id: str, version: str | None = None
@@ -52,4 +53,4 @@ class WorkflowsClient(BaseRestClient):
     def get_version(
         self, workflow_id: str, version: str | None = None
     ) -> WorkflowVersion:
-        return self._run_async_from_sync(self.aio_get_version, workflow_id, version)
+        return run_async_from_sync(self.aio_get_version, workflow_id, version)

hatchet_sdk/metadata.py

@@ -1,2 +1,2 @@
-def get_metadata(token: str) -> list[tuple[str, str]]:
-    return [("authorization", "bearer " + token)]
+def get_metadata(token: str) -> tuple[tuple[str, str]]:
+    return (("authorization", "bearer " + token),)

hatchet_sdk/runnables/standalone.py

@@ -1,4 +1,3 @@
-import asyncio
 from datetime import datetime
 from typing import Any, Generic, cast, get_type_hints
 
@@ -12,7 +11,7 @@ from hatchet_sdk.contracts.workflows_pb2 import WorkflowVersion
 from hatchet_sdk.runnables.task import Task
 from hatchet_sdk.runnables.types import EmptyModel, R, TWorkflowInput
 from hatchet_sdk.runnables.workflow import BaseWorkflow, Workflow
-from hatchet_sdk.utils.aio_utils import get_active_event_loop
+from hatchet_sdk.utils.aio import run_async_from_sync
 from hatchet_sdk.utils.typing import JSONSerializableMapping, is_basemodel_subclass
 from hatchet_sdk.workflow_run import WorkflowRunRef
 
@@ -27,25 +26,11 @@ class TaskRunRef(Generic[TWorkflowInput, R]):
         self._wrr = workflow_run_ref
 
     async def aio_result(self) -> R:
-        result = await self._wrr.workflow_listener.result(self._wrr.workflow_run_id)
+        result = await self._wrr.workflow_listener.aio_result(self._wrr.workflow_run_id)
         return self._s._extract_result(result)
 
     def result(self) -> R:
-        coro = self._wrr.workflow_listener.result(self._wrr.workflow_run_id)
-
-        loop = get_active_event_loop()
-
-        if loop is None:
-            loop = asyncio.new_event_loop()
-            asyncio.set_event_loop(loop)
-            try:
-                result = loop.run_until_complete(coro)
-            finally:
-                asyncio.set_event_loop(None)
-        else:
-            result = loop.run_until_complete(coro)
-
-        return self._s._extract_result(result)
+        return run_async_from_sync(self.aio_result)
 
 
 class Standalone(BaseWorkflow[TWorkflowInput], Generic[TWorkflowInput, R]):

hatchet_sdk/runnables/task.py

@@ -44,6 +44,10 @@ def fall_back_to_default(value: T, default: T, fallback_value: T) -> T:
     return fallback_value
 
 
+class NonRetryableException(Exception):
+    pass
+
+
 class Task(Generic[TWorkflowInput, R]):
     def __init__(
         self,