hatchet-sdk 1.12.3__py3-none-any.whl → 1.13.0__py3-none-any.whl

hatchet_sdk/context/context.py

@@ -1,6 +1,5 @@
+import asyncio
 import json
-import traceback
-from concurrent.futures import Future, ThreadPoolExecutor
 from datetime import timedelta
 from typing import TYPE_CHECKING, Any, cast
 from warnings import warn
@@ -21,6 +20,7 @@ from hatchet_sdk.logger import logger
 from hatchet_sdk.utils.timedelta_to_expression import Duration, timedelta_to_expr
 from hatchet_sdk.utils.typing import JSONSerializableMapping
 from hatchet_sdk.waits import SleepCondition, UserEventCondition
+from hatchet_sdk.worker.runner.utils.capture_logs import AsyncLogSender, LogRecord
 
 if TYPE_CHECKING:
     from hatchet_sdk.runnables.task import Task
@@ -38,6 +38,7 @@ class Context:
         worker: WorkerContext,
         runs_client: RunsClient,
         lifespan_context: Any | None,
+        log_sender: AsyncLogSender,
     ):
         self.worker = worker
 
@@ -53,24 +54,41 @@ class Context:
         self.runs_client = runs_client
         self.durable_event_listener = durable_event_listener
 
-        # FIXME: this limits the number of concurrent log requests to 1, which means we can do about
-        # 100 log lines per second but this depends on network.
-        self.logger_thread_pool = ThreadPoolExecutor(max_workers=1)
-        self.stream_event_thread_pool = ThreadPoolExecutor(max_workers=1)
-
         self.input = self.data.input
         self.filter_payload = self.data.filter_payload
+        self.log_sender = log_sender
 
         self._lifespan_context = lifespan_context
 
+        self.stream_index = 0
+
+    def _increment_stream_index(self) -> int:
+        index = self.stream_index
+        self.stream_index += 1
+
+        return index
+
     def was_skipped(self, task: "Task[TWorkflowInput, R]") -> bool:
-        return self.data.parents.get(task.name, {}).get("skipped", False)
+        """
+        Check if a given task was skipped. You can read about skipping in [the docs](https://docs.hatchet.run/home/conditional-workflows#skip_if).
+
+        :param task: The task to check the status of (skipped or not).
+        :return: True if the task was skipped, False otherwise.
+        """
+        return self.data.parents.get(task.name, {}).get("skipped", False) is True
 
     @property
     def trigger_data(self) -> JSONSerializableMapping:
         return self.data.triggers
 
     def task_output(self, task: "Task[TWorkflowInput, R]") -> "R":
+        """
+        Get the output of a parent task in a DAG.
+
+        :param task: The task whose output you want to retrieve.
+        :return: The output of the parent task, validated against the task's validators.
+        :raises ValueError: If the task was skipped or if the step output for the task is not found.
+        """
         from hatchet_sdk.runnables.types import R
 
         if self.was_skipped(task):
@@ -78,8 +96,8 @@ class Context:
 
         try:
             parent_step_data = cast(R, self.data.parents[task.name])
-        except KeyError:
-            raise ValueError(f"Step output for '{task.name}' not found")
+        except KeyError as e:
+            raise ValueError(f"Step output for '{task.name}' not found") from e
 
         if parent_step_data and (v := task.validators.step_output):
             return cast(R, v.model_validate(parent_step_data))
@@ -90,6 +108,7 @@ class Context:
         warn(
             "`aio_task_output` is deprecated. Use `task_output` instead.",
             DeprecationWarning,
+            stacklevel=2,
         )
 
         if task.is_async_function:
@@ -101,48 +120,82 @@ class Context:
 
     @property
     def was_triggered_by_event(self) -> bool:
+        """
+        A property that indicates whether the workflow was triggered by an event.
+
+        :return: True if the workflow was triggered by an event, False otherwise.
+        """
         return self.data.triggered_by == "event"
 
     @property
     def workflow_input(self) -> JSONSerializableMapping:
+        """
+        The input to the workflow, as a dictionary. It's recommended to use the `input` parameter to the task (the first argument passed into the task at runtime) instead of this property.
+
+        :return: The input to the workflow.
+        """
         return self.input
 
     @property
     def lifespan(self) -> Any:
+        """
+        The worker lifespan, if it exists. You can read about lifespans in [the docs](https://docs.hatchet.run/home/lifespans).
+
+        **Note: You'll need to cast the return type of this property to the type returned by your lifespan generator.**
+        """
         return self._lifespan_context
 
     @property
     def workflow_run_id(self) -> str:
+        """
+        The id of the current workflow run.
+
+        :return: The id of the current workflow run.
+        """
         return self.action.workflow_run_id
 
     def _set_cancellation_flag(self) -> None:
         self.exit_flag = True
 
     def cancel(self) -> None:
+        """
+        Cancel the current task run. This will call the Hatchet API to cancel the step run and set the exit flag to True.
+
+        :return: None
+        """
         logger.debug("cancelling step...")
         self.runs_client.cancel(self.step_run_id)
         self._set_cancellation_flag()
 
     async def aio_cancel(self) -> None:
+        """
+        Cancel the current task run. This will call the Hatchet API to cancel the step run and set the exit flag to True.
+
+        :return: None
+        """
         logger.debug("cancelling step...")
         await self.runs_client.aio_cancel(self.step_run_id)
         self._set_cancellation_flag()
 
-    # done returns true if the context has been cancelled
     def done(self) -> bool:
-        return self.exit_flag
+        """
+        Check if the current task run has been cancelled.
 
-    def _log(self, line: str) -> tuple[bool, Exception | None]:
-        try:
-            self.event_client.log(message=line, step_run_id=self.step_run_id)
-            return True, None
-        except Exception as e:
-            # we don't want to raise an exception here, as it will kill the log thread
-            return False, e
+        :return: True if the task run has been cancelled, False otherwise.
+        """
+        return self.exit_flag
 
     def log(
         self, line: str | JSONSerializableMapping, raise_on_error: bool = False
     ) -> None:
+        """
+        Log a line to the Hatchet API. This will send the log line to the Hatchet API and return immediately.
+
+        :param line: The line to log. Can be a string or a JSON serializable mapping.
+        :param raise_on_error: If True, will raise an exception if the log fails. Defaults to False.
+        :return: None
+        """
+
         if self.step_run_id == "":
             return
 
@@ -152,43 +205,51 @@ class Context:
             except Exception:
                 line = str(line)
 
-        future = self.logger_thread_pool.submit(self._log, line)
-
-        def handle_result(future: Future[tuple[bool, Exception | None]]) -> None:
-            success, exception = future.result()
-
-            if not success and exception:
-                if raise_on_error:
-                    raise exception
-                else:
-                    thread_trace = "".join(
-                        traceback.format_exception(
-                            type(exception), exception, exception.__traceback__
-                        )
-                    )
-                    call_site_trace = "".join(traceback.format_stack())
-                    logger.error(
-                        f"Error in log thread: {exception}\n{thread_trace}\nCalled from:\n{call_site_trace}"
-                    )
-
-        future.add_done_callback(handle_result)
+        logger.info(line)
+        self.log_sender.publish(LogRecord(message=line, step_run_id=self.step_run_id))
 
     def release_slot(self) -> None:
+        """
+        Manually release the slot for the current step run to free up a slot on the worker. Note that this is an advanced feature and should be used with caution.
+
+        :return: None
+        """
         return self.dispatcher_client.release_slot(self.step_run_id)
 
-    def _put_stream(self, data: str | bytes) -> None:
+    def put_stream(self, data: str | bytes) -> None:
+        """
+        Put a stream event to the Hatchet API. This will send the data to the Hatchet API and return immediately. You can then subscribe to the stream from a separate consumer.
+
+        :param data: The data to send to the Hatchet API. Can be a string or bytes.
+        :return: None
+        """
         try:
-            self.event_client.stream(data=data, step_run_id=self.step_run_id)
+            ix = self._increment_stream_index()
+
+            self.event_client.stream(
+                data=data,
+                step_run_id=self.step_run_id,
+                index=ix,
+            )
         except Exception as e:
             logger.error(f"Error putting stream event: {e}")
 
-    def put_stream(self, data: str | bytes) -> None:
-        if self.step_run_id == "":
-            return
+    async def aio_put_stream(self, data: str | bytes) -> None:
+        """
+        Put a stream event to the Hatchet API. This will send the data to the Hatchet API and return immediately. You can then subscribe to the stream from a separate consumer.
 
-        self.stream_event_thread_pool.submit(self._put_stream, data)
+        :param data: The data to send to the Hatchet API. Can be a string or bytes.
+        :return: None
+        """
+        await asyncio.to_thread(self.put_stream, data)
 
     def refresh_timeout(self, increment_by: str | timedelta) -> None:
+        """
+        Refresh the timeout for the current task run. You can read about refreshing timeouts in [the docs](https://docs.hatchet.run/home/timeouts#refreshing-timeouts).
+
+        :param increment_by: The amount of time to increment the timeout by. Can be a string (e.g. "5m") or a timedelta object.
+        :return: None
+        """
         if isinstance(increment_by, timedelta):
             increment_by = timedelta_to_expr(increment_by)
 
@@ -201,10 +262,30 @@ class Context:
 
     @property
     def retry_count(self) -> int:
+        """
+        The retry count of the current task run, which corresponds to the number of times the task has been retried.
+
+        :return: The retry count of the current task run.
+        """
         return self.action.retry_count
 
+    @property
+    def attempt_number(self) -> int:
+        """
+        The attempt number of the current task run, which corresponds to the number of times the task has been attempted, including the initial attempt. This is one more than the retry count.
+
+        :return: The attempt number of the current task run.
+        """
+
+        return self.retry_count + 1
+
     @property
     def additional_metadata(self) -> JSONSerializableMapping | None:
+        """
+        The additional metadata sent with the current task run.
+
+        :return: The additional metadata sent with the current task run, or None if no additional metadata was sent.
+        """
         return self.action.additional_metadata
 
     @property
@@ -217,27 +298,54 @@ class Context:
 
     @property
     def parent_workflow_run_id(self) -> str | None:
+        """
+        The parent workflow run id of the current task run, if it exists. This is useful for knowing which workflow run spawned this run as a child.
+
+        :return: The parent workflow run id of the current task run, or None if it does not exist.
+        """
         return self.action.parent_workflow_run_id
 
     @property
     def priority(self) -> int | None:
+        """
+        The priority that the current task was run with.
+
+        :return: The priority of the current task run, or None if no priority was set.
+        """
         return self.action.priority
 
     @property
     def workflow_id(self) -> str | None:
+        """
+        The id of the workflow that this task belongs to.
+
+        :return: The id of the workflow that this task belongs to.
+        """
+
         return self.action.workflow_id
 
     @property
     def workflow_version_id(self) -> str | None:
+        """
+        The id of the workflow version that this task belongs to.
+
+        :return: The id of the workflow version that this task belongs to.
+        """
+
         return self.action.workflow_version_id
 
     @property
     def task_run_errors(self) -> dict[str, str]:
+        """
+        A helper intended to be used in an on-failure step to retrieve the errors that occurred in upstream task runs.
+
+        :return: A dictionary mapping task names to their error messages.
+        """
         errors = self.data.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"
+                "No step run errors found. `context.task_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
@@ -246,6 +354,12 @@ class Context:
         self,
         task: "Task[TWorkflowInput, R]",
     ) -> str | None:
+        """
+        A helper intended to be used in an on-failure step to retrieve the error that occurred in a specific upstream task run.
+
+        :param task: The task whose error you want to retrieve.
+        :return: The error message of the task run, or None if no error occurred.
+        """
         errors = self.data.step_run_errors
 
         return errors.get(task.name)
@@ -255,6 +369,15 @@ class DurableContext(Context):
     async def aio_wait_for(
         self, signal_key: str, *conditions: SleepCondition | UserEventCondition
     ) -> dict[str, Any]:
+        """
+        Durably wait for either a sleep or an event.
+
+        :param signal_key: The key to use for the durable event. This is used to identify the event in the Hatchet API.
+        :param *conditions: The conditions to wait for. Can be a SleepCondition or UserEventCondition.
+
+        :return: A dictionary containing the results of the wait.
+        :raises ValueError: If the durable event listener is not available.
+        """
         if self.durable_event_listener is None:
             raise ValueError("Durable event listener is not available")
 
@@ -264,6 +387,7 @@ class DurableContext(Context):
             task_id=task_id,
             signal_key=signal_key,
             conditions=list(conditions),
+            config=self.runs_client.client_config,
         )
 
         self.durable_event_listener.register_durable_event(request)

hatchet_sdk/context/worker_context.py

@@ -2,11 +2,11 @@ from hatchet_sdk.clients.dispatcher.dispatcher import DispatcherClient
 
 
 class WorkerContext:
-    _worker_id: str | None = None
-    _registered_workflow_names: list[str] = []
-    _labels: dict[str, str | int] = {}
-
     def __init__(self, labels: dict[str, str | int], client: DispatcherClient):
+        self._worker_id: str | None = None
+        self._registered_workflow_names: list[str] = []
+        self._labels: dict[str, str | int] = {}
+
         self._labels = labels
         self.client = client
 
@@ -23,6 +23,3 @@ class WorkerContext:
 
     def id(self) -> str | None:
         return self._worker_id
-
-    # def has_workflow(self, workflow_name: str):
-    #     return workflow_name in self._registered_workflow_names

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\"\xf6\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\x12\x10\n\x08priority\x18\x12 \x01(\x05\x12\x17\n\nworkflowId\x18\x13 \x01(\tH\x04\x88\x01\x01\x12\x1e\n\x11workflowVersionId\x18\x14 \x01(\tH\x05\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_idB\r\n\x0b_workflowIdB\x14\n\x12_workflowVersionId\"\'\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')
+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\"\xf6\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\x12\x10\n\x08priority\x18\x12 \x01(\x05\x12\x17\n\nworkflowId\x18\x13 \x01(\tH\x04\x88\x01\x01\x12\x1e\n\x11workflowVersionId\x18\x14 \x01(\tH\x05\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_idB\r\n\x0b_workflowIdB\x14\n\x12_workflowVersionId\"\'\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\"\xda\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\x12\x17\n\neventIndex\x18\n \x01(\x03H\x02\x88\x01\x01\x42\x0e\n\x0c_stepRetriesB\r\n\x0b_retryCountB\r\n\x0b_eventIndex\"\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=3684
-  _globals['_SDKS']._serialized_end=3739
-  _globals['_ACTIONTYPE']._serialized_start=3741
-  _globals['_ACTIONTYPE']._serialized_end=3819
-  _globals['_GROUPKEYACTIONEVENTTYPE']._serialized_start=3822
-  _globals['_GROUPKEYACTIONEVENTTYPE']._serialized_end=3984
-  _globals['_STEPACTIONEVENTTYPE']._serialized_start=3987
-  _globals['_STEPACTIONEVENTTYPE']._serialized_end=4159
-  _globals['_RESOURCETYPE']._serialized_start=4161
-  _globals['_RESOURCETYPE']._serialized_end=4262
-  _globals['_RESOURCEEVENTTYPE']._serialized_start=4265
-  _globals['_RESOURCEEVENTTYPE']._serialized_end=4519
-  _globals['_WORKFLOWRUNEVENTTYPE']._serialized_start=4521
-  _globals['_WORKFLOWRUNEVENTTYPE']._serialized_end=4581
+  _globals['_SDKS']._serialized_start=3724
+  _globals['_SDKS']._serialized_end=3779
+  _globals['_ACTIONTYPE']._serialized_start=3781
+  _globals['_ACTIONTYPE']._serialized_end=3859
+  _globals['_GROUPKEYACTIONEVENTTYPE']._serialized_start=3862
+  _globals['_GROUPKEYACTIONEVENTTYPE']._serialized_end=4024
+  _globals['_STEPACTIONEVENTTYPE']._serialized_start=4027
+  _globals['_STEPACTIONEVENTTYPE']._serialized_end=4199
+  _globals['_RESOURCETYPE']._serialized_start=4201
+  _globals['_RESOURCETYPE']._serialized_end=4302
+  _globals['_RESOURCEEVENTTYPE']._serialized_start=4305
+  _globals['_RESOURCEEVENTTYPE']._serialized_end=4559
+  _globals['_WORKFLOWRUNEVENTTYPE']._serialized_start=4561
+  _globals['_WORKFLOWRUNEVENTTYPE']._serialized_end=4621
   _globals['_WORKERLABELS']._serialized_start=53
   _globals['_WORKERLABELS']._serialized_end=139
   _globals['_RUNTIMEINFO']._serialized_start=142
@@ -76,27 +76,27 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_SUBSCRIBETOWORKFLOWRUNSREQUEST']._serialized_start=2575
   _globals['_SUBSCRIBETOWORKFLOWRUNSREQUEST']._serialized_end=2630
   _globals['_WORKFLOWEVENT']._serialized_start=2633
-  _globals['_WORKFLOWEVENT']._serialized_end=2939
-  _globals['_WORKFLOWRUNEVENT']._serialized_start=2942
-  _globals['_WORKFLOWRUNEVENT']._serialized_end=3110
-  _globals['_STEPRUNRESULT']._serialized_start=3113
-  _globals['_STEPRUNRESULT']._serialized_end=3251
-  _globals['_OVERRIDESDATA']._serialized_start=3253
-  _globals['_OVERRIDESDATA']._serialized_end=3340
-  _globals['_OVERRIDESDATARESPONSE']._serialized_start=3342
-  _globals['_OVERRIDESDATARESPONSE']._serialized_end=3365
-  _globals['_HEARTBEATREQUEST']._serialized_start=3367
-  _globals['_HEARTBEATREQUEST']._serialized_end=3452
-  _globals['_HEARTBEATRESPONSE']._serialized_start=3454
-  _globals['_HEARTBEATRESPONSE']._serialized_end=3473
-  _globals['_REFRESHTIMEOUTREQUEST']._serialized_start=3475
-  _globals['_REFRESHTIMEOUTREQUEST']._serialized_end=3545
-  _globals['_REFRESHTIMEOUTRESPONSE']._serialized_start=3547
-  _globals['_REFRESHTIMEOUTRESPONSE']._serialized_end=3618
-  _globals['_RELEASESLOTREQUEST']._serialized_start=3620
-  _globals['_RELEASESLOTREQUEST']._serialized_end=3659
-  _globals['_RELEASESLOTRESPONSE']._serialized_start=3661
-  _globals['_RELEASESLOTRESPONSE']._serialized_end=3682
-  _globals['_DISPATCHER']._serialized_start=4584
-  _globals['_DISPATCHER']._serialized_end=5472
+  _globals['_WORKFLOWEVENT']._serialized_end=2979
+  _globals['_WORKFLOWRUNEVENT']._serialized_start=2982
+  _globals['_WORKFLOWRUNEVENT']._serialized_end=3150
+  _globals['_STEPRUNRESULT']._serialized_start=3153
+  _globals['_STEPRUNRESULT']._serialized_end=3291
+  _globals['_OVERRIDESDATA']._serialized_start=3293
+  _globals['_OVERRIDESDATA']._serialized_end=3380
+  _globals['_OVERRIDESDATARESPONSE']._serialized_start=3382
+  _globals['_OVERRIDESDATARESPONSE']._serialized_end=3405
+  _globals['_HEARTBEATREQUEST']._serialized_start=3407
+  _globals['_HEARTBEATREQUEST']._serialized_end=3492
+  _globals['_HEARTBEATRESPONSE']._serialized_start=3494
+  _globals['_HEARTBEATRESPONSE']._serialized_end=3513
+  _globals['_REFRESHTIMEOUTREQUEST']._serialized_start=3515
+  _globals['_REFRESHTIMEOUTREQUEST']._serialized_end=3585
+  _globals['_REFRESHTIMEOUTRESPONSE']._serialized_start=3587
+  _globals['_REFRESHTIMEOUTRESPONSE']._serialized_end=3658
+  _globals['_RELEASESLOTREQUEST']._serialized_start=3660
+  _globals['_RELEASESLOTREQUEST']._serialized_end=3699
+  _globals['_RELEASESLOTRESPONSE']._serialized_start=3701
+  _globals['_RELEASESLOTRESPONSE']._serialized_end=3722
+  _globals['_DISPATCHER']._serialized_start=4624
+  _globals['_DISPATCHER']._serialized_end=5512
 # @@protoc_insertion_point(module_scope)

hatchet_sdk/contracts/dispatcher_pb2.pyi

@@ -295,7 +295,7 @@ class SubscribeToWorkflowRunsRequest(_message.Message):
     def __init__(self, workflowRunId: _Optional[str] = ...) -> None: ...
 
 class WorkflowEvent(_message.Message):
-    __slots__ = ("workflowRunId", "resourceType", "eventType", "resourceId", "eventTimestamp", "eventPayload", "hangup", "stepRetries", "retryCount")
+    __slots__ = ("workflowRunId", "resourceType", "eventType", "resourceId", "eventTimestamp", "eventPayload", "hangup", "stepRetries", "retryCount", "eventIndex")
     WORKFLOWRUNID_FIELD_NUMBER: _ClassVar[int]
     RESOURCETYPE_FIELD_NUMBER: _ClassVar[int]
     EVENTTYPE_FIELD_NUMBER: _ClassVar[int]
@@ -305,6 +305,7 @@ class WorkflowEvent(_message.Message):
     HANGUP_FIELD_NUMBER: _ClassVar[int]
     STEPRETRIES_FIELD_NUMBER: _ClassVar[int]
     RETRYCOUNT_FIELD_NUMBER: _ClassVar[int]
+    EVENTINDEX_FIELD_NUMBER: _ClassVar[int]
     workflowRunId: str
     resourceType: ResourceType
     eventType: ResourceEventType
@@ -314,7 +315,8 @@ class WorkflowEvent(_message.Message):
     hangup: bool
     stepRetries: int
     retryCount: int
-    def __init__(self, workflowRunId: _Optional[str] = ..., resourceType: _Optional[_Union[ResourceType, str]] = ..., eventType: _Optional[_Union[ResourceEventType, str]] = ..., resourceId: _Optional[str] = ..., eventTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., eventPayload: _Optional[str] = ..., hangup: bool = ..., stepRetries: _Optional[int] = ..., retryCount: _Optional[int] = ...) -> None: ...
+    eventIndex: int
+    def __init__(self, workflowRunId: _Optional[str] = ..., resourceType: _Optional[_Union[ResourceType, str]] = ..., eventType: _Optional[_Union[ResourceEventType, str]] = ..., resourceId: _Optional[str] = ..., eventTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., eventPayload: _Optional[str] = ..., hangup: bool = ..., stepRetries: _Optional[int] = ..., retryCount: _Optional[int] = ..., eventIndex: _Optional[int] = ...) -> None: ...
 
 class WorkflowRunEvent(_message.Message):
     __slots__ = ("workflowRunId", "eventType", "eventTimestamp", "results")

hatchet_sdk/contracts/events_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\x0c\x65vents.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xd2\x01\n\x05\x45vent\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x0f\n\x07\x65ventId\x18\x02 \x01(\t\x12\x0b\n\x03key\x18\x03 \x01(\t\x12\x0f\n\x07payload\x18\x04 \x01(\t\x12\x32\n\x0e\x65ventTimestamp\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x1f\n\x12\x61\x64\x64itionalMetadata\x18\x06 \x01(\tH\x00\x88\x01\x01\x12\x12\n\x05scope\x18\x07 \x01(\tH\x01\x88\x01\x01\x42\x15\n\x13_additionalMetadataB\x08\n\x06_scope\" \n\x06\x45vents\x12\x16\n\x06\x65vents\x18\x01 \x03(\x0b\x32\x06.Event\"\xc2\x01\n\rPutLogRequest\x12\x11\n\tstepRunId\x18\x01 \x01(\t\x12-\n\tcreatedAt\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x0f\n\x07message\x18\x03 \x01(\t\x12\x12\n\x05level\x18\x04 \x01(\tH\x00\x88\x01\x01\x12\x10\n\x08metadata\x18\x05 \x01(\t\x12\x1b\n\x0etaskRetryCount\x18\x06 \x01(\x05H\x01\x88\x01\x01\x42\x08\n\x06_levelB\x11\n\x0f_taskRetryCount\"\x10\n\x0ePutLogResponse\"|\n\x15PutStreamEventRequest\x12\x11\n\tstepRunId\x18\x01 \x01(\t\x12-\n\tcreatedAt\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x0f\n\x07message\x18\x03 \x01(\x0c\x12\x10\n\x08metadata\x18\x05 \x01(\t\"\x18\n\x16PutStreamEventResponse\"9\n\x14\x42ulkPushEventRequest\x12!\n\x06\x65vents\x18\x01 \x03(\x0b\x32\x11.PushEventRequest\"\xde\x01\n\x10PushEventRequest\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x0f\n\x07payload\x18\x02 \x01(\t\x12\x32\n\x0e\x65ventTimestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x1f\n\x12\x61\x64\x64itionalMetadata\x18\x04 \x01(\tH\x00\x88\x01\x01\x12\x15\n\x08priority\x18\x05 \x01(\x05H\x01\x88\x01\x01\x12\x12\n\x05scope\x18\x06 \x01(\tH\x02\x88\x01\x01\x42\x15\n\x13_additionalMetadataB\x0b\n\t_priorityB\x08\n\x06_scope\"%\n\x12ReplayEventRequest\x12\x0f\n\x07\x65ventId\x18\x01 \x01(\t2\x88\x02\n\rEventsService\x12#\n\x04Push\x12\x11.PushEventRequest\x1a\x06.Event\"\x00\x12,\n\x08\x42ulkPush\x12\x15.BulkPushEventRequest\x1a\x07.Events\"\x00\x12\x32\n\x11ReplaySingleEvent\x12\x13.ReplayEventRequest\x1a\x06.Event\"\x00\x12+\n\x06PutLog\x12\x0e.PutLogRequest\x1a\x0f.PutLogResponse\"\x00\x12\x43\n\x0ePutStreamEvent\x12\x16.PutStreamEventRequest\x1a\x17.PutStreamEventResponse\"\x00\x42\x45ZCgithub.com/hatchet-dev/hatchet/internal/services/ingestor/contractsb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0c\x65vents.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xd2\x01\n\x05\x45vent\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x0f\n\x07\x65ventId\x18\x02 \x01(\t\x12\x0b\n\x03key\x18\x03 \x01(\t\x12\x0f\n\x07payload\x18\x04 \x01(\t\x12\x32\n\x0e\x65ventTimestamp\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x1f\n\x12\x61\x64\x64itionalMetadata\x18\x06 \x01(\tH\x00\x88\x01\x01\x12\x12\n\x05scope\x18\x07 \x01(\tH\x01\x88\x01\x01\x42\x15\n\x13_additionalMetadataB\x08\n\x06_scope\" \n\x06\x45vents\x12\x16\n\x06\x65vents\x18\x01 \x03(\x0b\x32\x06.Event\"\xc2\x01\n\rPutLogRequest\x12\x11\n\tstepRunId\x18\x01 \x01(\t\x12-\n\tcreatedAt\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x0f\n\x07message\x18\x03 \x01(\t\x12\x12\n\x05level\x18\x04 \x01(\tH\x00\x88\x01\x01\x12\x10\n\x08metadata\x18\x05 \x01(\t\x12\x1b\n\x0etaskRetryCount\x18\x06 \x01(\x05H\x01\x88\x01\x01\x42\x08\n\x06_levelB\x11\n\x0f_taskRetryCount\"\x10\n\x0ePutLogResponse\"\xa4\x01\n\x15PutStreamEventRequest\x12\x11\n\tstepRunId\x18\x01 \x01(\t\x12-\n\tcreatedAt\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x0f\n\x07message\x18\x03 \x01(\x0c\x12\x10\n\x08metadata\x18\x05 \x01(\t\x12\x17\n\neventIndex\x18\x06 \x01(\x03H\x00\x88\x01\x01\x42\r\n\x0b_eventIndex\"\x18\n\x16PutStreamEventResponse\"9\n\x14\x42ulkPushEventRequest\x12!\n\x06\x65vents\x18\x01 \x03(\x0b\x32\x11.PushEventRequest\"\xde\x01\n\x10PushEventRequest\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x0f\n\x07payload\x18\x02 \x01(\t\x12\x32\n\x0e\x65ventTimestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x1f\n\x12\x61\x64\x64itionalMetadata\x18\x04 \x01(\tH\x00\x88\x01\x01\x12\x15\n\x08priority\x18\x05 \x01(\x05H\x01\x88\x01\x01\x12\x12\n\x05scope\x18\x06 \x01(\tH\x02\x88\x01\x01\x42\x15\n\x13_additionalMetadataB\x0b\n\t_priorityB\x08\n\x06_scope\"%\n\x12ReplayEventRequest\x12\x0f\n\x07\x65ventId\x18\x01 \x01(\t2\x88\x02\n\rEventsService\x12#\n\x04Push\x12\x11.PushEventRequest\x1a\x06.Event\"\x00\x12,\n\x08\x42ulkPush\x12\x15.BulkPushEventRequest\x1a\x07.Events\"\x00\x12\x32\n\x11ReplaySingleEvent\x12\x13.ReplayEventRequest\x1a\x06.Event\"\x00\x12+\n\x06PutLog\x12\x0e.PutLogRequest\x1a\x0f.PutLogResponse\"\x00\x12\x43\n\x0ePutStreamEvent\x12\x16.PutStreamEventRequest\x1a\x17.PutStreamEventResponse\"\x00\x42\x45ZCgithub.com/hatchet-dev/hatchet/internal/services/ingestor/contractsb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -31,16 +31,16 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_PUTLOGREQUEST']._serialized_end=491
   _globals['_PUTLOGRESPONSE']._serialized_start=493
   _globals['_PUTLOGRESPONSE']._serialized_end=509
-  _globals['_PUTSTREAMEVENTREQUEST']._serialized_start=511
-  _globals['_PUTSTREAMEVENTREQUEST']._serialized_end=635
-  _globals['_PUTSTREAMEVENTRESPONSE']._serialized_start=637
-  _globals['_PUTSTREAMEVENTRESPONSE']._serialized_end=661
-  _globals['_BULKPUSHEVENTREQUEST']._serialized_start=663
-  _globals['_BULKPUSHEVENTREQUEST']._serialized_end=720
-  _globals['_PUSHEVENTREQUEST']._serialized_start=723
-  _globals['_PUSHEVENTREQUEST']._serialized_end=945
-  _globals['_REPLAYEVENTREQUEST']._serialized_start=947
-  _globals['_REPLAYEVENTREQUEST']._serialized_end=984
-  _globals['_EVENTSSERVICE']._serialized_start=987
-  _globals['_EVENTSSERVICE']._serialized_end=1251
+  _globals['_PUTSTREAMEVENTREQUEST']._serialized_start=512
+  _globals['_PUTSTREAMEVENTREQUEST']._serialized_end=676
+  _globals['_PUTSTREAMEVENTRESPONSE']._serialized_start=678
+  _globals['_PUTSTREAMEVENTRESPONSE']._serialized_end=702
+  _globals['_BULKPUSHEVENTREQUEST']._serialized_start=704
+  _globals['_BULKPUSHEVENTREQUEST']._serialized_end=761
+  _globals['_PUSHEVENTREQUEST']._serialized_start=764
+  _globals['_PUSHEVENTREQUEST']._serialized_end=986
+  _globals['_REPLAYEVENTREQUEST']._serialized_start=988
+  _globals['_REPLAYEVENTREQUEST']._serialized_end=1025
+  _globals['_EVENTSSERVICE']._serialized_start=1028
+  _globals['_EVENTSSERVICE']._serialized_end=1292
 # @@protoc_insertion_point(module_scope)

hatchet_sdk/contracts/events_pb2.pyi

@@ -51,16 +51,18 @@ class PutLogResponse(_message.Message):
     def __init__(self) -> None: ...
 
 class PutStreamEventRequest(_message.Message):
-    __slots__ = ("stepRunId", "createdAt", "message", "metadata")
+    __slots__ = ("stepRunId", "createdAt", "message", "metadata", "eventIndex")
     STEPRUNID_FIELD_NUMBER: _ClassVar[int]
     CREATEDAT_FIELD_NUMBER: _ClassVar[int]
     MESSAGE_FIELD_NUMBER: _ClassVar[int]
     METADATA_FIELD_NUMBER: _ClassVar[int]
+    EVENTINDEX_FIELD_NUMBER: _ClassVar[int]
     stepRunId: str
     createdAt: _timestamp_pb2.Timestamp
     message: bytes
     metadata: str
-    def __init__(self, stepRunId: _Optional[str] = ..., createdAt: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., message: _Optional[bytes] = ..., metadata: _Optional[str] = ...) -> None: ...
+    eventIndex: int
+    def __init__(self, stepRunId: _Optional[str] = ..., createdAt: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., message: _Optional[bytes] = ..., metadata: _Optional[str] = ..., eventIndex: _Optional[int] = ...) -> None: ...
 
 class PutStreamEventResponse(_message.Message):
     __slots__ = ()