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

hatchet_sdk/clients/rest/models/v1_workflow_run_display_name_list.py

@@ -0,0 +1,114 @@
+# coding: utf-8
+
+"""
+    Hatchet API
+
+    The Hatchet API
+
+    The version of the OpenAPI document: 1.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+from typing import Any, ClassVar, Dict, List, Optional, Set
+
+from pydantic import BaseModel, ConfigDict, Field
+from typing_extensions import Self
+
+from hatchet_sdk.clients.rest.models.pagination_response import PaginationResponse
+from hatchet_sdk.clients.rest.models.v1_workflow_run_display_name import (
+    V1WorkflowRunDisplayName,
+)
+
+
+class V1WorkflowRunDisplayNameList(BaseModel):
+    """
+    V1WorkflowRunDisplayNameList
+    """  # noqa: E501
+
+    pagination: PaginationResponse
+    rows: List[V1WorkflowRunDisplayName] = Field(
+        description="The list of display names"
+    )
+    __properties: ClassVar[List[str]] = ["pagination", "rows"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of V1WorkflowRunDisplayNameList from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of pagination
+        if self.pagination:
+            _dict["pagination"] = self.pagination.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in rows (list)
+        _items = []
+        if self.rows:
+            for _item_rows in self.rows:
+                if _item_rows:
+                    _items.append(_item_rows.to_dict())
+            _dict["rows"] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of V1WorkflowRunDisplayNameList from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate(
+            {
+                "pagination": (
+                    PaginationResponse.from_dict(obj["pagination"])
+                    if obj.get("pagination") is not None
+                    else None
+                ),
+                "rows": (
+                    [V1WorkflowRunDisplayName.from_dict(_item) for _item in obj["rows"]]
+                    if obj.get("rows") is not None
+                    else None
+                ),
+            }
+        )
+        return _obj

hatchet_sdk/clients/rest/models/workflow_run_shape_item_for_workflow_run_details.py

@@ -31,13 +31,17 @@ class WorkflowRunShapeItemForWorkflowRunDetails(BaseModel):
     task_external_id: Annotated[
         str, Field(min_length=36, strict=True, max_length=36)
     ] = Field(alias="taskExternalId")
-    children_external_ids: List[
+    step_id: Annotated[str, Field(min_length=36, strict=True, max_length=36)] = Field(
+        alias="stepId"
+    )
+    children_step_ids: List[
         Annotated[str, Field(min_length=36, strict=True, max_length=36)]
-    ] = Field(alias="childrenExternalIds")
+    ] = Field(alias="childrenStepIds")
     task_name: StrictStr = Field(alias="taskName")
     __properties: ClassVar[List[str]] = [
         "taskExternalId",
-        "childrenExternalIds",
+        "stepId",
+        "childrenStepIds",
         "taskName",
     ]
 
@@ -92,7 +96,8 @@ class WorkflowRunShapeItemForWorkflowRunDetails(BaseModel):
         _obj = cls.model_validate(
             {
                 "taskExternalId": obj.get("taskExternalId"),
-                "childrenExternalIds": obj.get("childrenExternalIds"),
+                "stepId": obj.get("stepId"),
+                "childrenStepIds": obj.get("childrenStepIds"),
                 "taskName": obj.get("taskName"),
             }
         )

hatchet_sdk/clients/rest_client.py

@@ -9,8 +9,10 @@ from pydantic import StrictInt
 from hatchet_sdk.clients.rest.api.event_api import EventApi
 from hatchet_sdk.clients.rest.api.log_api import LogApi
 from hatchet_sdk.clients.rest.api.step_run_api import StepRunApi
+from hatchet_sdk.clients.rest.api.worker_api import WorkerApi
 from hatchet_sdk.clients.rest.api.workflow_api import WorkflowApi
 from hatchet_sdk.clients.rest.api.workflow_run_api import WorkflowRunApi
+from hatchet_sdk.clients.rest.api.workflow_runs_api import WorkflowRunsApi
 from hatchet_sdk.clients.rest.api_client import ApiClient
 from hatchet_sdk.clients.rest.configuration import Configuration
 from hatchet_sdk.clients.rest.models.create_cron_workflow_trigger_request import (
@@ -106,9 +108,11 @@ class RestApi:
         self._api_client: ApiClient | None = None
         self._workflow_api: WorkflowApi | None = None
         self._workflow_run_api: WorkflowRunApi | None = None
+        self._workflow_runs_api: WorkflowRunsApi | None = None
         self._step_run_api: StepRunApi | None = None
         self._event_api: EventApi | None = None
         self._log_api: LogApi | None = None
+        self._worker_api: WorkerApi | None = None
 
     @property
     def api_client(self) -> ApiClient:
@@ -134,6 +138,23 @@ class RestApi:
             self._workflow_run_api = WorkflowRunApi(self.api_client)
         return self._workflow_run_api
 
+    @property
+    def workflow_runs_api(self) -> WorkflowRunsApi:
+        ## IMPORTANT: This client needs to be instantiated lazily because it relies on an event
+        ## loop to be running, which may not be the case when the `Hatchet` client is instantiated.
+        if self._workflow_runs_api is None:
+            self._workflow_runs_api = WorkflowRunsApi(self.api_client)
+        return self._workflow_runs_api
+
+    @property
+    def worker_api(self) -> WorkerApi:
+        ## IMPORTANT: This client needs to be instantiated lazily because it relies on an event
+        ## loop to be running, which may not be the case when the `Hatchet` client is instantiated.
+        if self._worker_api is None:
+            self._worker_api = WorkerApi(self.api_client)
+
+        return self._worker_api
+
     @property
     def step_run_api(self) -> StepRunApi:
         ## IMPORTANT: This client needs to be instantiated lazily because it relies on an event

hatchet_sdk/clients/run_event_listener.py

@@ -132,7 +132,6 @@ class RunEventListener:
 
                         try:
                             if workflow_event.eventPayload:
-                                ## TODO: Should this be `dumps` instead?
                                 payload = json.loads(workflow_event.eventPayload)
                         except Exception:
                             payload = workflow_event.eventPayload

hatchet_sdk/context/context.py

@@ -2,31 +2,34 @@ import inspect
 import json
 import traceback
 from concurrent.futures import Future, ThreadPoolExecutor
-from typing import Any, cast
+from datetime import timedelta
+from typing import TYPE_CHECKING, Any, cast
 
 from pydantic import BaseModel
 
-from hatchet_sdk.clients.admin import (
-    AdminClient,
-    ChildTriggerWorkflowOptions,
-    ChildWorkflowRunDict,
-    TriggerWorkflowOptions,
-    WorkflowRunDict,
-)
+from hatchet_sdk.clients.admin import AdminClient
 from hatchet_sdk.clients.dispatcher.dispatcher import (  # type: ignore[attr-defined]
     Action,
     DispatcherClient,
 )
+from hatchet_sdk.clients.durable_event_listener import (
+    DurableEventListener,
+    RegisterDurableEventRequest,
+)
 from hatchet_sdk.clients.events import EventClient
-from hatchet_sdk.clients.rest.tenacity_utils import tenacity_retry
 from hatchet_sdk.clients.rest_client import RestApi
 from hatchet_sdk.clients.run_event_listener import RunEventListenerClient
 from hatchet_sdk.clients.workflow_listener import PooledWorkflowRunListener
 from hatchet_sdk.context.worker_context import WorkerContext
-from hatchet_sdk.contracts.dispatcher_pb2 import OverridesData
 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, WorkflowValidator
-from hatchet_sdk.workflow_run import WorkflowRunRef
+from hatchet_sdk.waits import SleepCondition, UserEventCondition
+
+if TYPE_CHECKING:
+    from hatchet_sdk.runnables.task import Task
+    from hatchet_sdk.runnables.types import R, TWorkflowInput
+
 
 DEFAULT_WORKFLOW_POLLING_INTERVAL = 5  # Seconds
 
@@ -44,8 +47,6 @@ class StepRunError(BaseModel):
 
 
 class Context:
-    spawn_index = -1
-
     def __init__(
         self,
         action: Action,
@@ -54,6 +55,7 @@ class Context:
         event_client: EventClient,
         rest_client: RestApi,
         workflow_listener: PooledWorkflowRunListener | None,
+        durable_event_listener: DurableEventListener | None,
         workflow_run_event_listener: RunEventListenerClient,
         worker: WorkerContext,
         namespace: str = "",
@@ -66,9 +68,6 @@ class Context:
 
         self.action = action
 
-        # FIXME: stepRunId is a legacy field, we should remove it
-        self.stepRunId = action.step_run_id
-
         self.step_run_id: str = action.step_run_id
         self.exit_flag = False
         self.dispatcher_client = dispatcher_client
@@ -76,6 +75,7 @@ class Context:
         self.event_client = event_client
         self.rest_client = rest_client
         self.workflow_listener = workflow_listener
+        self.durable_event_listener = durable_event_listener
         self.workflow_run_event_listener = workflow_run_event_listener
         self.namespace = namespace
 
@@ -86,45 +86,46 @@ class Context:
 
         self.input = self.data.input
 
-    def _prepare_workflow_options(
-        self,
-        key: str | None = None,
-        options: ChildTriggerWorkflowOptions = ChildTriggerWorkflowOptions(),
-        worker_id: str | None = None,
-    ) -> TriggerWorkflowOptions:
-        workflow_run_id = self.action.workflow_run_id
-        step_run_id = self.action.step_run_id
-
-        trigger_options = TriggerWorkflowOptions(
-            parent_id=workflow_run_id,
-            parent_step_run_id=step_run_id,
-            child_key=key,
-            child_index=self.spawn_index,
-            additional_metadata=options.additional_metadata,
-            desired_worker_id=worker_id if options.sticky else None,
-        )
+    def was_skipped(self, task: "Task[TWorkflowInput, R]") -> bool:
+        return self.data.parents.get(task.name, {}).get("skipped", False)
+
+    @property
+    def trigger_data(self) -> JSONSerializableMapping:
+        return self.data.triggers
+
+    def task_output(self, task: "Task[TWorkflowInput, R]") -> "R":
+        from hatchet_sdk.runnables.types import R
 
-        self.spawn_index += 1
-        return trigger_options
+        if self.was_skipped(task):
+            raise ValueError("{task.name} was skipped")
+
+        action_prefix = self.action.action_id.split(":")[0]
 
-    def step_output(self, step: str) -> dict[str, Any] | BaseModel:
         workflow_validator = next(
-            (v for k, v in self.validator_registry.items() if k.split(":")[-1] == step),
+            (
+                v
+                for k, v in self.validator_registry.items()
+                if k == f"{action_prefix}:{task.name}"
+            ),
             None,
         )
 
         try:
-            parent_step_data = cast(dict[str, Any], self.data.parents[step])
+            parent_step_data = cast(R, self.data.parents[task.name])
         except KeyError:
-            raise ValueError(f"Step output for '{step}' not found")
+            raise ValueError(f"Step output for '{task.name}' not found")
 
-        if workflow_validator and (v := workflow_validator.step_output):
-            return v.model_validate(parent_step_data)
+        if (
+            parent_step_data
+            and workflow_validator
+            and (v := workflow_validator.step_output)
+        ):
+            return cast(R, v.model_validate(parent_step_data))
 
         return parent_step_data
 
     @property
-    def triggered_by_event(self) -> bool:
+    def was_triggered_by_event(self) -> bool:
         return self.data.triggered_by == "event"
 
     @property
@@ -143,23 +144,9 @@ class Context:
     def done(self) -> bool:
         return self.exit_flag
 
-    def playground(self, name: str, default: str | None = None) -> str | None:
-        caller_file = get_caller_file_path()
-
-        self.dispatcher_client.put_overrides_data(
-            OverridesData(
-                stepRunId=self.stepRunId,
-                path=name,
-                value=json.dumps(default),
-                callerFilename=caller_file,
-            )
-        )
-
-        return default
-
     def _log(self, line: str) -> tuple[bool, Exception | None]:
         try:
-            self.event_client.log(message=line, step_run_id=self.stepRunId)
+            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
@@ -168,7 +155,7 @@ class Context:
     def log(
         self, line: str | JSONSerializableMapping, raise_on_error: bool = False
     ) -> None:
-        if self.stepRunId == "":
+        if self.step_run_id == "":
             return
 
         if not isinstance(line, str):
@@ -198,24 +185,27 @@ class Context:
         future.add_done_callback(handle_result)
 
     def release_slot(self) -> None:
-        return self.dispatcher_client.release_slot(self.stepRunId)
+        return self.dispatcher_client.release_slot(self.step_run_id)
 
     def _put_stream(self, data: str | bytes) -> None:
         try:
-            self.event_client.stream(data=data, step_run_id=self.stepRunId)
+            self.event_client.stream(data=data, step_run_id=self.step_run_id)
         except Exception as e:
             logger.error(f"Error putting stream event: {e}")
 
     def put_stream(self, data: str | bytes) -> None:
-        if self.stepRunId == "":
+        if self.step_run_id == "":
             return
 
         self.stream_event_thread_pool.submit(self._put_stream, data)
 
-    def refresh_timeout(self, increment_by: str) -> None:
+    def refresh_timeout(self, increment_by: str | timedelta) -> None:
+        if isinstance(increment_by, timedelta):
+            increment_by = timedelta_to_expr(increment_by)
+
         try:
             return self.dispatcher_client.refresh_timeout(
-                step_run_id=self.stepRunId, increment_by=increment_by
+                step_run_id=self.step_run_id, increment_by=increment_by
             )
         except Exception as e:
             logger.error(f"Error refreshing timeout: {e}")
@@ -241,7 +231,7 @@ class Context:
         return self.action.parent_workflow_run_id
 
     @property
-    def step_run_errors(self) -> dict[str, str]:
+    def task_run_errors(self) -> dict[str, str]:
         errors = self.data.step_run_errors
 
         if not errors:
@@ -251,96 +241,44 @@ class Context:
 
         return errors
 
-    def fetch_run_failures(self) -> list[StepRunError]:
-        data = self.rest_client.workflow_run_get(self.action.workflow_run_id)
-        other_job_runs = [
-            run for run in (data.job_runs or []) if run.job_id != self.action.job_id
-        ]
-
-        return [
-            StepRunError(
-                step_id=step_run.step_id,
-                step_run_action_name=step_run.step.action,
-                error=step_run.error,
-            )
-            for job_run in other_job_runs
-            if job_run.step_runs
-            for step_run in job_run.step_runs
-            if step_run.error and step_run.step
-        ]
-
-    @tenacity_retry
-    async def aio_spawn_workflow(
+    def fetch_task_run_error(
         self,
-        workflow_name: str,
-        input: JSONSerializableMapping = {},
-        key: str | None = None,
-        options: ChildTriggerWorkflowOptions = ChildTriggerWorkflowOptions(),
-    ) -> WorkflowRunRef:
-        worker_id = self.worker.id()
+        task: "Task[TWorkflowInput, R]",
+    ) -> str | None:
+        errors = self.data.step_run_errors
 
-        trigger_options = self._prepare_workflow_options(key, options, worker_id)
+        return errors.get(task.name)
 
-        return await self.admin_client.aio_run_workflow(
-            workflow_name, input, trigger_options
-        )
 
-    @tenacity_retry
-    async def aio_spawn_workflows(
-        self, child_workflow_runs: list[ChildWorkflowRunDict]
-    ) -> list[WorkflowRunRef]:
+class DurableContext(Context):
+    async def aio_wait_for(
+        self, signal_key: str, *conditions: SleepCondition | UserEventCondition
+    ) -> dict[str, Any]:
+        if self.durable_event_listener is None:
+            raise ValueError("Durable event listener is not available")
 
-        if len(child_workflow_runs) == 0:
-            raise Exception("no child workflows to spawn")
+        task_id = self.step_run_id
 
-        worker_id = self.worker.id()
+        request = RegisterDurableEventRequest(
+            task_id=task_id,
+            signal_key=signal_key,
+            conditions=list(conditions),
+        )
 
-        bulk_trigger_workflow_runs = [
-            WorkflowRunDict(
-                workflow_name=child_workflow_run.workflow_name,
-                input=child_workflow_run.input,
-                options=self._prepare_workflow_options(
-                    child_workflow_run.key, child_workflow_run.options, worker_id
-                ),
-            )
-            for child_workflow_run in child_workflow_runs
-        ]
+        self.durable_event_listener.register_durable_event(request)
 
-        return await self.admin_client.aio_run_workflows(bulk_trigger_workflow_runs)
+        return await self.durable_event_listener.result(
+            task_id,
+            signal_key,
+        )
 
-    @tenacity_retry
-    def spawn_workflow(
-        self,
-        workflow_name: str,
-        input: JSONSerializableMapping = {},
-        key: str | None = None,
-        options: ChildTriggerWorkflowOptions = ChildTriggerWorkflowOptions(),
-    ) -> WorkflowRunRef:
-        worker_id = self.worker.id()
-
-        trigger_options = self._prepare_workflow_options(key, options, worker_id)
-
-        return self.admin_client.run_workflow(workflow_name, input, trigger_options)
-
-    @tenacity_retry
-    def spawn_workflows(
-        self, child_workflow_runs: list[ChildWorkflowRunDict]
-    ) -> list[WorkflowRunRef]:
-
-        if len(child_workflow_runs) == 0:
-            raise Exception("no child workflows to spawn")
-
-        worker_id = self.worker.id()
-
-        bulk_trigger_workflow_runs = [
-            WorkflowRunDict(
-                workflow_name=child_workflow_run.workflow_name,
-                input=child_workflow_run.input,
-                options=self._prepare_workflow_options(
-                    child_workflow_run.key, child_workflow_run.options, worker_id
-                ),
-            )
-            for child_workflow_run in child_workflow_runs
-        ]
+    async def aio_sleep_for(self, duration: Duration) -> dict[str, Any]:
+        """
+        Lightweight wrapper for durable sleep. Allows for shorthand usage of `ctx.aio_wait_for` when specifying a sleep condition.
+
+        For more complicated conditions, use `ctx.aio_wait_for` directly.
+        """
 
-        return self.admin_client.run_workflows(bulk_trigger_workflow_runs)
+        return await self.aio_wait_for(
+            f"sleep:{timedelta_to_expr(duration)}", SleepCondition(duration=duration)
+        )

hatchet_sdk/contracts/dispatcher_pb2_grpc.py

@@ -3,7 +3,7 @@
 import grpc
 import warnings
 
-from . import dispatcher_pb2 as dispatcher__pb2
+from hatchet_sdk.contracts import dispatcher_pb2 as dispatcher__pb2
 
 GRPC_GENERATED_VERSION = '1.64.1'
 GRPC_VERSION = grpc.__version__

hatchet_sdk/contracts/events_pb2.py

@@ -15,14 +15,14 @@ _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\"\xb4\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\x42\x15\n\x13_additionalMetadata\" \n\x06\x45vents\x12\x16\n\x06\x65vents\x18\x01 \x03(\x0b\x32\x06.Event\"\x92\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(\tB\x08\n\x06_level\"\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\"\x9c\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\x42\x15\n\x13_additionalMetadata\"%\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\x42GZEgithub.com/hatchet-dev/hatchet/internal/services/dispatcher/contractsb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0c\x65vents.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xb4\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\x42\x15\n\x13_additionalMetadata\" \n\x06\x45vents\x12\x16\n\x06\x65vents\x18\x01 \x03(\x0b\x32\x06.Event\"\x92\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(\tB\x08\n\x06_level\"\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\"\x9c\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\x42\x15\n\x13_additionalMetadata\"%\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)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'events_pb2', _globals)
 if not _descriptor._USE_C_DESCRIPTORS:
   _globals['DESCRIPTOR']._loaded_options = None
-  _globals['DESCRIPTOR']._serialized_options = b'ZEgithub.com/hatchet-dev/hatchet/internal/services/dispatcher/contracts'
+  _globals['DESCRIPTOR']._serialized_options = b'ZCgithub.com/hatchet-dev/hatchet/internal/services/ingestor/contracts'
   _globals['_EVENT']._serialized_start=50
   _globals['_EVENT']._serialized_end=230
   _globals['_EVENTS']._serialized_start=232

hatchet_sdk/contracts/events_pb2_grpc.py

@@ -3,7 +3,7 @@
 import grpc
 import warnings
 
-from . import events_pb2 as events__pb2
+from hatchet_sdk.contracts import events_pb2 as events__pb2
 
 GRPC_GENERATED_VERSION = '1.64.1'
 GRPC_VERSION = grpc.__version__

hatchet_sdk/contracts/v1/dispatcher_pb2.py

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: v1/dispatcher.proto
+# Protobuf Python Version: 5.26.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from hatchet_sdk.contracts.v1.shared import condition_pb2 as v1_dot_shared_dot_condition__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x13v1/dispatcher.proto\x12\x02v1\x1a\x19v1/shared/condition.proto\"z\n\x1bRegisterDurableEventRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x12\n\nsignal_key\x18\x02 \x01(\t\x12\x36\n\nconditions\x18\x03 \x01(\x0b\x32\".v1.DurableEventListenerConditions\"\x1e\n\x1cRegisterDurableEventResponse\"C\n\x1cListenForDurableEventRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x12\n\nsignal_key\x18\x02 \x01(\t\"A\n\x0c\x44urableEvent\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x12\n\nsignal_key\x18\x02 \x01(\t\x12\x0c\n\x04\x64\x61ta\x18\x03 \x01(\x0c\x32\xbe\x01\n\x0cV1Dispatcher\x12[\n\x14RegisterDurableEvent\x12\x1f.v1.RegisterDurableEventRequest\x1a .v1.RegisterDurableEventResponse\"\x00\x12Q\n\x15ListenForDurableEvent\x12 .v1.ListenForDurableEventRequest\x1a\x10.v1.DurableEvent\"\x00(\x01\x30\x01\x42\x42Z@github.com/hatchet-dev/hatchet/internal/services/shared/proto/v1b\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'v1.dispatcher_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z@github.com/hatchet-dev/hatchet/internal/services/shared/proto/v1'
+  _globals['_REGISTERDURABLEEVENTREQUEST']._serialized_start=54
+  _globals['_REGISTERDURABLEEVENTREQUEST']._serialized_end=176
+  _globals['_REGISTERDURABLEEVENTRESPONSE']._serialized_start=178
+  _globals['_REGISTERDURABLEEVENTRESPONSE']._serialized_end=208
+  _globals['_LISTENFORDURABLEEVENTREQUEST']._serialized_start=210
+  _globals['_LISTENFORDURABLEEVENTREQUEST']._serialized_end=277
+  _globals['_DURABLEEVENT']._serialized_start=279
+  _globals['_DURABLEEVENT']._serialized_end=344
+  _globals['_V1DISPATCHER']._serialized_start=347
+  _globals['_V1DISPATCHER']._serialized_end=537
+# @@protoc_insertion_point(module_scope)