hatchet-sdk 1.0.0__py3-none-any.whl → 1.0.1__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/models/workflow_runs_metrics.py

@@ -22,13 +22,17 @@ from typing import Any, ClassVar, Dict, List, Optional, Set
 from pydantic import BaseModel, ConfigDict
 from typing_extensions import Self
 
+from hatchet_sdk.clients.rest.models.workflow_runs_metrics_counts import (
+    WorkflowRunsMetricsCounts,
+)
+
 
 class WorkflowRunsMetrics(BaseModel):
     """
     WorkflowRunsMetrics
     """  # noqa: E501
 
-    counts: Optional[Dict[str, Any]] = None
+    counts: Optional[WorkflowRunsMetricsCounts] = None
     __properties: ClassVar[List[str]] = ["counts"]
 
     model_config = ConfigDict(

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/clients/v1/api_client.py

@@ -0,0 +1,81 @@
+import asyncio
+from concurrent.futures import ThreadPoolExecutor
+from typing import AsyncContextManager, Callable, Coroutine, ParamSpec, TypeVar
+
+from hatchet_sdk.clients.rest.api_client import ApiClient
+from hatchet_sdk.clients.rest.configuration import Configuration
+from hatchet_sdk.config import ClientConfig
+from hatchet_sdk.utils.typing import JSONSerializableMapping
+
+## Type variables to use with coroutines.
+## See https://stackoverflow.com/questions/73240620/the-right-way-to-type-hint-a-coroutine-function
+## Return type
+R = TypeVar("R")
+
+## Yield type
+Y = TypeVar("Y")
+
+## Send type
+S = TypeVar("S")
+
+P = ParamSpec("P")
+
+
+def maybe_additional_metadata_to_kv(
+    additional_metadata: dict[str, str] | JSONSerializableMapping | None
+) -> list[str] | None:
+    if not additional_metadata:
+        return None
+
+    return [f"{k}:{v}" for k, v in additional_metadata.items()]
+
+
+class BaseRestClient:
+    def __init__(self, config: ClientConfig) -> None:
+        self.tenant_id = config.tenant_id
+
+        self.client_config = config
+        self.api_config = Configuration(
+            host=config.server_url,
+            access_token=config.token,
+        )
+
+        self.api_config.datetime_format = "%Y-%m-%dT%H:%M:%S.%fZ"
+
+    def client(self) -> AsyncContextManager[ApiClient]:
+        return ApiClient(self.api_config)
+
+    def _run_async_function_do_not_use_directly(
+        self,
+        async_func: Callable[P, Coroutine[Y, S, R]],
+        *args: P.args,
+        **kwargs: P.kwargs,
+    ) -> R:
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+        try:
+            return loop.run_until_complete(async_func(*args, **kwargs))
+        finally:
+            loop.close()
+
+    def _run_async_from_sync(
+        self,
+        async_func: Callable[P, Coroutine[Y, S, R]],
+        *args: P.args,
+        **kwargs: P.kwargs,
+    ) -> R:
+        try:
+            loop = asyncio.get_event_loop()
+        except RuntimeError:
+            loop = None
+
+        if loop and loop.is_running():
+            return loop.run_until_complete(async_func(*args, **kwargs))
+        else:
+            with ThreadPoolExecutor() as executor:
+                future = executor.submit(
+                    lambda: self._run_async_function_do_not_use_directly(
+                        async_func, *args, **kwargs
+                    )
+                )
+                return future.result()

hatchet_sdk/context/context.py

@@ -2,31 +2,31 @@ 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,19 +44,14 @@ class StepRunError(BaseModel):
 
 
 class Context:
-    spawn_index = -1
-
     def __init__(
         self,
         action: Action,
         dispatcher_client: DispatcherClient,
         admin_client: AdminClient,
         event_client: EventClient,
-        rest_client: RestApi,
-        workflow_listener: PooledWorkflowRunListener | None,
-        workflow_run_event_listener: RunEventListenerClient,
+        durable_event_listener: DurableEventListener | None,
         worker: WorkerContext,
-        namespace: str = "",
         validator_registry: dict[str, WorkflowValidator] = {},
     ):
         self.worker = worker
@@ -66,18 +61,12 @@ 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.step_run_id = action.step_run_id
         self.exit_flag = False
         self.dispatcher_client = dispatcher_client
         self.admin_client = admin_client
         self.event_client = event_client
-        self.rest_client = rest_client
-        self.workflow_listener = workflow_listener
-        self.workflow_run_event_listener = workflow_run_event_listener
-        self.namespace = namespace
+        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.
@@ -86,45 +75,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 +133,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 +144,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 +174,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 +220,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 +230,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__