hatchet-sdk 1.13.0__tar.gz → 1.14.1__tar.gz

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hatchet-sdk
-Version: 1.13.0
+Version: 1.14.1
 Summary: 
 License: MIT
 Author: Alexander Belanger

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/__init__.py

@@ -7,6 +7,7 @@ from hatchet_sdk.clients.listeners.durable_event_listener import (
     RegisterDurableEventRequest,
 )
 from hatchet_sdk.clients.listeners.run_event_listener import (
+    RunEventListener,
     StepRunEventType,
     WorkflowRunEventType,
 )
@@ -129,6 +130,14 @@ from hatchet_sdk.clients.rest.models.workflow_version_definition import (
     WorkflowVersionDefinition,
 )
 from hatchet_sdk.clients.rest.models.workflow_version_meta import WorkflowVersionMeta
+from hatchet_sdk.conditions import (
+    Condition,
+    OrGroup,
+    ParentCondition,
+    SleepCondition,
+    UserEventCondition,
+    or_,
+)
 from hatchet_sdk.config import ClientConfig, ClientTLSConfig, OpenTelemetryConfig
 from hatchet_sdk.context.context import Context, DurableContext
 from hatchet_sdk.context.worker_context import WorkerContext
@@ -140,6 +149,7 @@ from hatchet_sdk.contracts.workflows_pb2 import (
 from hatchet_sdk.exceptions import (
     DedupeViolationError,
     FailedTaskRunExceptionGroup,
+    NonRetryableException,
     TaskRunError,
 )
 from hatchet_sdk.features.runs import BulkCancelReplayOpts, RunFilter
@@ -154,16 +164,10 @@ from hatchet_sdk.runnables.types import (
     TaskDefaults,
     WorkflowConfig,
 )
+from hatchet_sdk.runnables.workflow import TaskRunRef
 from hatchet_sdk.utils.opentelemetry import OTelAttribute
-from hatchet_sdk.waits import (
-    Condition,
-    OrGroup,
-    ParentCondition,
-    SleepCondition,
-    UserEventCondition,
-    or_,
-)
 from hatchet_sdk.worker.worker import Worker, WorkerStartOptions, WorkerStatus
+from hatchet_sdk.workflow_run import WorkflowRunRef
 
 __all__ = [
     "APIError",
@@ -216,6 +220,7 @@ __all__ = [
     "LogLineList",
     "LogLineOrderByDirection",
     "LogLineOrderByField",
+    "NonRetryableException",
     "OTelAttribute",
     "OpenTelemetryConfig",
     "OrGroup",
@@ -229,6 +234,7 @@ __all__ = [
     "RejectInviteRequest",
     "ReplayEventRequest",
     "RerunStepRunRequest",
+    "RunEventListener",
     "RunFilter",
     "ScheduleTriggerWorkflowOptions",
     "SleepCondition",
@@ -240,6 +246,7 @@ __all__ = [
     "Task",
     "TaskDefaults",
     "TaskRunError",
+    "TaskRunRef",
     "Tenant",
     "TenantInvite",
     "TenantInviteList",
@@ -272,6 +279,7 @@ __all__ = [
     "WorkflowRun",
     "WorkflowRunEventType",
     "WorkflowRunList",
+    "WorkflowRunRef",
     "WorkflowRunStatus",
     "WorkflowRunTriggeredBy",
     "WorkflowTag",

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/clients/listeners/durable_event_listener.py

@@ -8,6 +8,7 @@ from pydantic import BaseModel, ConfigDict
 
 from hatchet_sdk.clients.listeners.pooled_listener import PooledListener
 from hatchet_sdk.clients.rest.tenacity_utils import tenacity_retry
+from hatchet_sdk.conditions import Condition, SleepCondition, UserEventCondition
 from hatchet_sdk.config import ClientConfig
 from hatchet_sdk.connection import new_conn
 from hatchet_sdk.contracts.v1.dispatcher_pb2 import (
@@ -20,7 +21,6 @@ from hatchet_sdk.contracts.v1.dispatcher_pb2 import (
 from hatchet_sdk.contracts.v1.dispatcher_pb2_grpc import V1DispatcherStub
 from hatchet_sdk.contracts.v1.shared.condition_pb2 import DurableEventListenerConditions
 from hatchet_sdk.metadata import get_metadata
-from hatchet_sdk.waits import SleepCondition, UserEventCondition
 
 DEFAULT_DURABLE_EVENT_LISTENER_RETRY_INTERVAL = 3  # seconds
 DEFAULT_DURABLE_EVENT_LISTENER_RETRY_COUNT = 5
@@ -32,7 +32,7 @@ class RegisterDurableEventRequest(BaseModel):
 
     task_id: str
     signal_key: str
-    conditions: list[SleepCondition | UserEventCondition]
+    conditions: list[Condition]
     config: ClientConfig
 
     def to_proto(self) -> RegisterDurableEventRequestProto:

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/clients/rest/models/semaphore_slots.py

@@ -42,7 +42,7 @@ class SemaphoreSlots(BaseModel):
     workflow_run_id: StrictStr = Field(
         description="The workflow run id.", alias="workflowRunId"
     )
-    status: StepRunStatus
+    status: Optional[StepRunStatus] = None
     __properties: ClassVar[List[str]] = [
         "stepRunId",
         "actionId",

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/clients/rest/models/v1_task_summary.py

@@ -111,6 +111,9 @@ class V1TaskSummary(BaseModel):
         description="The version ID of the workflow",
         alias="workflowVersionId",
     )
+    workflow_config: Optional[Dict[str, Any]] = Field(
+        default=None, alias="workflowConfig"
+    )
     __properties: ClassVar[List[str]] = [
         "metadata",
         "actionId",
@@ -138,6 +141,7 @@ class V1TaskSummary(BaseModel):
         "workflowName",
         "workflowRunExternalId",
         "workflowVersionId",
+        "workflowConfig",
     ]
 
     model_config = ConfigDict(
@@ -234,6 +238,7 @@ class V1TaskSummary(BaseModel):
                 "workflowName": obj.get("workflowName"),
                 "workflowRunExternalId": obj.get("workflowRunExternalId"),
                 "workflowVersionId": obj.get("workflowVersionId"),
+                "workflowConfig": obj.get("workflowConfig"),
             }
         )
         return _obj

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/clients/rest/models/v1_workflow_run_details.py

@@ -41,7 +41,16 @@ class V1WorkflowRunDetails(BaseModel):
     )
     shape: List[WorkflowRunShapeItemForWorkflowRunDetails]
     tasks: List[V1TaskSummary]
-    __properties: ClassVar[List[str]] = ["run", "taskEvents", "shape", "tasks"]
+    workflow_config: Optional[Dict[str, Any]] = Field(
+        default=None, alias="workflowConfig"
+    )
+    __properties: ClassVar[List[str]] = [
+        "run",
+        "taskEvents",
+        "shape",
+        "tasks",
+        "workflowConfig",
+    ]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -140,6 +149,7 @@ class V1WorkflowRunDetails(BaseModel):
                     if obj.get("tasks") is not None
                     else None
                 ),
+                "workflowConfig": obj.get("workflowConfig"),
             }
         )
         return _obj

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/clients/rest/models/workflow_version.py

@@ -51,6 +51,9 @@ class WorkflowVersion(BaseModel):
     triggers: Optional[WorkflowTriggers] = None
     schedule_timeout: Optional[StrictStr] = Field(default=None, alias="scheduleTimeout")
     jobs: Optional[List[Job]] = None
+    workflow_config: Optional[Dict[str, Any]] = Field(
+        default=None, alias="workflowConfig"
+    )
     __properties: ClassVar[List[str]] = [
         "metadata",
         "version",
@@ -63,6 +66,7 @@ class WorkflowVersion(BaseModel):
         "triggers",
         "scheduleTimeout",
         "jobs",
+        "workflowConfig",
     ]
 
     model_config = ConfigDict(
@@ -165,6 +169,7 @@ class WorkflowVersion(BaseModel):
                     if obj.get("jobs") is not None
                     else None
                 ),
+                "workflowConfig": obj.get("workflowConfig"),
             }
         )
         return _obj

hatchet_sdk-1.13.0/hatchet_sdk/waits.py → hatchet_sdk-1.14.1/hatchet_sdk/conditions.py

@@ -140,3 +140,18 @@ class OrGroup:
 
 def or_(*conditions: Condition) -> OrGroup:
     return OrGroup(conditions=list(conditions))
+
+
+def flatten_conditions(conditions: list[Condition | OrGroup]) -> list[Condition]:
+    flattened: list[Condition] = []
+
+    for condition in conditions:
+        if isinstance(condition, OrGroup):
+            for or_condition in condition.conditions:
+                or_condition.base.or_group_id = condition.or_group_id
+
+            flattened.extend(condition.conditions)
+        else:
+            flattened.append(condition)
+
+    return flattened

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/context/context.py

@@ -14,12 +14,17 @@ from hatchet_sdk.clients.listeners.durable_event_listener import (
     DurableEventListener,
     RegisterDurableEventRequest,
 )
+from hatchet_sdk.conditions import (
+    OrGroup,
+    SleepCondition,
+    UserEventCondition,
+    flatten_conditions,
+)
 from hatchet_sdk.context.worker_context import WorkerContext
 from hatchet_sdk.features.runs import RunsClient
 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:
@@ -367,7 +372,9 @@ class Context:
 
 class DurableContext(Context):
     async def aio_wait_for(
-        self, signal_key: str, *conditions: SleepCondition | UserEventCondition
+        self,
+        signal_key: str,
+        *conditions: SleepCondition | UserEventCondition | OrGroup,
     ) -> dict[str, Any]:
         """
         Durably wait for either a sleep or an event.
@@ -386,7 +393,7 @@ class DurableContext(Context):
         request = RegisterDurableEventRequest(
             task_id=task_id,
             signal_key=signal_key,
-            conditions=list(conditions),
+            conditions=flatten_conditions(list(conditions)),
             config=self.runs_client.client_config,
         )
 

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/exceptions.py

@@ -98,3 +98,7 @@ class FailedTaskRunExceptionGroup(Exception):  # noqa: N818
 
 class LoopAlreadyRunningError(Exception):
     pass
+
+
+class IllegalTaskOutputError(Exception):
+    pass

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/features/filters.py

@@ -26,47 +26,38 @@ class FiltersClient(BaseRestClient):
         self,
         limit: int | None = None,
         offset: int | None = None,
-        workflow_id_scope_pairs: list[tuple[str, str]] | None = None,
+        workflow_ids: list[str] | None = None,
+        scopes: list[str] | None = None,
     ) -> V1FilterList:
         """
         List filters for a given tenant.
 
         :param limit: The maximum number of filters to return.
         :param offset: The number of filters to skip before starting to collect the result set.
-        :param workflow_id_scope_pairs: A list of tuples containing workflow IDs and scopes to filter by. The workflow id is first, then the scope is second.
+        :param workflow_ids: A list of workflow IDs to filter by.
+        :param scopes: A list of scopes to filter by.
 
         :return: A list of filters matching the specified criteria.
         """
-        return await asyncio.to_thread(
-            self.list, limit, offset, workflow_id_scope_pairs
-        )
+        return await asyncio.to_thread(self.list, limit, offset, workflow_ids, scopes)
 
     def list(
         self,
         limit: int | None = None,
         offset: int | None = None,
-        workflow_id_scope_pairs: list[tuple[str, str]] | None = None,
+        workflow_ids: list[str] | None = None,
+        scopes: list[str] | None = None,
     ) -> V1FilterList:
         """
         List filters for a given tenant.
 
         :param limit: The maximum number of filters to return.
         :param offset: The number of filters to skip before starting to collect the result set.
-        :param workflow_id_scope_pairs: A list of tuples containing workflow IDs and scopes to filter by. The workflow id is first, then the scope is second.
+        :param workflow_ids: A list of workflow IDs to filter by.
+        :param scopes: A list of scopes to filter by.
 
         :return: A list of filters matching the specified criteria.
         """
-        workflow_ids = (
-            [pair[0] for pair in workflow_id_scope_pairs]
-            if workflow_id_scope_pairs
-            else None
-        )
-        scopes = (
-            [pair[1] for pair in workflow_id_scope_pairs]
-            if workflow_id_scope_pairs
-            else None
-        )
-
         with self.client() as client:
             return self._fa(client).v1_filter_list(
                 tenant=self.tenant_id,

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/runnables/task.py

@@ -1,6 +1,15 @@
 from collections.abc import Callable
 from typing import TYPE_CHECKING, Any, Generic, cast, get_type_hints
 
+from hatchet_sdk.conditions import (
+    Action,
+    Condition,
+    OrGroup,
+    ParentCondition,
+    SleepCondition,
+    UserEventCondition,
+    flatten_conditions,
+)
 from hatchet_sdk.context.context import Context, DurableContext
 from hatchet_sdk.contracts.v1.shared.condition_pb2 import TaskConditions
 from hatchet_sdk.contracts.v1.workflows_pb2 import (
@@ -24,14 +33,6 @@ from hatchet_sdk.utils.typing import (
     TaskIOValidator,
     is_basemodel_subclass,
 )
-from hatchet_sdk.waits import (
-    Action,
-    Condition,
-    OrGroup,
-    ParentCondition,
-    SleepCondition,
-    UserEventCondition,
-)
 
 if TYPE_CHECKING:
     from hatchet_sdk.runnables.workflow import Workflow
@@ -84,9 +85,9 @@ class Task(Generic[TWorkflowInput, R]):
         self.backoff_max_seconds = backoff_max_seconds
         self.concurrency = concurrency or []
 
-        self.wait_for = self._flatten_conditions(wait_for or [])
-        self.skip_if = self._flatten_conditions(skip_if or [])
-        self.cancel_if = self._flatten_conditions(cancel_if or [])
+        self.wait_for = flatten_conditions(wait_for or [])
+        self.skip_if = flatten_conditions(skip_if or [])
+        self.cancel_if = flatten_conditions(cancel_if or [])
 
         return_type = get_type_hints(_fn).get("return")
 
@@ -95,22 +96,6 @@ class Task(Generic[TWorkflowInput, R]):
             step_output=return_type if is_basemodel_subclass(return_type) else None,
         )
 
-    def _flatten_conditions(
-        self, conditions: list[Condition | OrGroup]
-    ) -> list[Condition]:
-        flattened: list[Condition] = []
-
-        for condition in conditions:
-            if isinstance(condition, OrGroup):
-                for or_condition in condition.conditions:
-                    or_condition.base.or_group_id = condition.or_group_id
-
-                flattened.extend(condition.conditions)
-            else:
-                flattened.append(condition)
-
-        return flattened
-
     def call(self, ctx: Context | DurableContext) -> R:
         if self.is_async_function:
             raise TypeError(f"{self.name} is not a sync function. Use `acall` instead.")

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/runnables/workflow.py

@@ -17,6 +17,7 @@ from hatchet_sdk.clients.rest.models.cron_workflows import CronWorkflows
 from hatchet_sdk.clients.rest.models.v1_filter import V1Filter
 from hatchet_sdk.clients.rest.models.v1_task_status import V1TaskStatus
 from hatchet_sdk.clients.rest.models.v1_task_summary import V1TaskSummary
+from hatchet_sdk.conditions import Condition, OrGroup
 from hatchet_sdk.context.context import Context, DurableContext
 from hatchet_sdk.contracts.v1.workflows_pb2 import (
     CreateWorkflowVersionRequest,
@@ -43,7 +44,6 @@ from hatchet_sdk.utils.typing import (
     JSONSerializableMapping,
     is_basemodel_subclass,
 )
-from hatchet_sdk.waits import Condition, OrGroup
 from hatchet_sdk.workflow_run import WorkflowRunRef
 
 if TYPE_CHECKING:

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/v0/__init__.py

@@ -1,3 +1,10 @@
+from hatchet_sdk.contracts.workflows_pb2 import (
+    ConcurrencyLimitStrategy,
+    CreateWorkflowVersionOpts,
+    RateLimitDuration,
+    StickyStrategy,
+    WorkerLabelComparator,
+)
 from hatchet_sdk.v0.clients.rest.models.accept_invite_request import AcceptInviteRequest
 
 # import models into sdk package
@@ -121,13 +128,6 @@ from hatchet_sdk.v0.clients.rest.models.workflow_version_definition import (
     WorkflowVersionDefinition,
 )
 from hatchet_sdk.v0.clients.rest.models.workflow_version_meta import WorkflowVersionMeta
-from hatchet_sdk.v0.contracts.workflows_pb2 import (
-    ConcurrencyLimitStrategy,
-    CreateWorkflowVersionOpts,
-    RateLimitDuration,
-    StickyStrategy,
-    WorkerLabelComparator,
-)
 from hatchet_sdk.v0.utils.aio_utils import sync_to_async
 
 from .client import new_client

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/v0/clients/admin.py

@@ -5,12 +5,7 @@ from typing import Any, Callable, Dict, List, Optional, TypedDict, TypeVar, Unio
 import grpc
 from google.protobuf import timestamp_pb2
 
-from hatchet_sdk.v0.clients.rest.models.workflow_run import WorkflowRun
-from hatchet_sdk.v0.clients.rest.tenacity_utils import tenacity_retry
-from hatchet_sdk.v0.clients.run_event_listener import new_listener
-from hatchet_sdk.v0.clients.workflow_listener import PooledWorkflowRunListener
-from hatchet_sdk.v0.connection import new_conn
-from hatchet_sdk.v0.contracts.workflows_pb2 import (
+from hatchet_sdk.contracts.workflows_pb2 import (
     BulkTriggerWorkflowRequest,
     BulkTriggerWorkflowResponse,
     CreateWorkflowVersionOpts,
@@ -22,7 +17,12 @@ from hatchet_sdk.v0.contracts.workflows_pb2 import (
     TriggerWorkflowResponse,
     WorkflowVersion,
 )
-from hatchet_sdk.v0.contracts.workflows_pb2_grpc import WorkflowServiceStub
+from hatchet_sdk.contracts.workflows_pb2_grpc import WorkflowServiceStub
+from hatchet_sdk.v0.clients.rest.models.workflow_run import WorkflowRun
+from hatchet_sdk.v0.clients.rest.tenacity_utils import tenacity_retry
+from hatchet_sdk.v0.clients.run_event_listener import new_listener
+from hatchet_sdk.v0.clients.workflow_listener import PooledWorkflowRunListener
+from hatchet_sdk.v0.connection import new_conn
 from hatchet_sdk.v0.workflow_run import RunRef, WorkflowRunRef
 
 from ..loader import ClientConfig

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/v0/clients/dispatcher/action_listener.py

@@ -7,12 +7,7 @@ from typing import Any, AsyncGenerator, List, Optional
 import grpc
 from grpc._cython import cygrpc
 
-from hatchet_sdk.v0.clients.event_ts import ThreadSafeEvent, read_with_interrupt
-from hatchet_sdk.v0.clients.run_event_listener import (
-    DEFAULT_ACTION_LISTENER_RETRY_INTERVAL,
-)
-from hatchet_sdk.v0.connection import new_conn
-from hatchet_sdk.v0.contracts.dispatcher_pb2 import (
+from hatchet_sdk.contracts.dispatcher_pb2 import (
     ActionType,
     AssignedAction,
     HeartbeatRequest,
@@ -20,8 +15,13 @@ from hatchet_sdk.v0.contracts.dispatcher_pb2 import (
     WorkerListenRequest,
     WorkerUnsubscribeRequest,
 )
-from hatchet_sdk.v0.contracts.dispatcher_pb2_grpc import DispatcherStub
-from hatchet_sdk.v0.logger import logger
+from hatchet_sdk.contracts.dispatcher_pb2_grpc import DispatcherStub
+from hatchet_sdk.logger import logger
+from hatchet_sdk.v0.clients.event_ts import ThreadSafeEvent, read_with_interrupt
+from hatchet_sdk.v0.clients.run_event_listener import (
+    DEFAULT_ACTION_LISTENER_RETRY_INTERVAL,
+)
+from hatchet_sdk.v0.connection import new_conn
 from hatchet_sdk.v0.utils.backoff import exp_backoff_sleep
 
 from ...loader import ClientConfig

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/v0/clients/dispatcher/dispatcher.py

@@ -2,14 +2,7 @@ from typing import Any, cast
 
 from google.protobuf.timestamp_pb2 import Timestamp
 
-from hatchet_sdk.v0.clients.dispatcher.action_listener import (
-    Action,
-    ActionListener,
-    GetActionListenerRequest,
-)
-from hatchet_sdk.v0.clients.rest.tenacity_utils import tenacity_retry
-from hatchet_sdk.v0.connection import new_conn
-from hatchet_sdk.v0.contracts.dispatcher_pb2 import (
+from hatchet_sdk.contracts.dispatcher_pb2 import (
     STEP_EVENT_TYPE_COMPLETED,
     STEP_EVENT_TYPE_FAILED,
     ActionEventResponse,
@@ -25,7 +18,14 @@ from hatchet_sdk.v0.contracts.dispatcher_pb2 import (
     WorkerRegisterRequest,
     WorkerRegisterResponse,
 )
-from hatchet_sdk.v0.contracts.dispatcher_pb2_grpc import DispatcherStub
+from hatchet_sdk.contracts.dispatcher_pb2_grpc import DispatcherStub
+from hatchet_sdk.v0.clients.dispatcher.action_listener import (
+    Action,
+    ActionListener,
+    GetActionListenerRequest,
+)
+from hatchet_sdk.v0.clients.rest.tenacity_utils import tenacity_retry
+from hatchet_sdk.v0.connection import new_conn
 
 from ...loader import ClientConfig
 from ...metadata import get_metadata

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/v0/clients/events.py

@@ -7,15 +7,15 @@ from uuid import uuid4
 import grpc
 from google.protobuf import timestamp_pb2
 
-from hatchet_sdk.v0.clients.rest.tenacity_utils import tenacity_retry
-from hatchet_sdk.v0.contracts.events_pb2 import (
+from hatchet_sdk.contracts.events_pb2 import (
     BulkPushEventRequest,
     Event,
     PushEventRequest,
     PutLogRequest,
     PutStreamEventRequest,
 )
-from hatchet_sdk.v0.contracts.events_pb2_grpc import EventsServiceStub
+from hatchet_sdk.contracts.events_pb2_grpc import EventsServiceStub
+from hatchet_sdk.v0.clients.rest.tenacity_utils import tenacity_retry
 
 from ..loader import ClientConfig
 from ..metadata import get_metadata

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/v0/clients/rest/tenacity_utils.py

@@ -3,7 +3,7 @@ from typing import Callable, ParamSpec, TypeVar
 import grpc
 import tenacity
 
-from hatchet_sdk.v0.logger import logger
+from hatchet_sdk.logger import logger
 
 P = ParamSpec("P")
 R = TypeVar("R")

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/v0/clients/run_event_listener.py

@@ -4,15 +4,15 @@ from typing import AsyncGenerator
 
 import grpc
 
-from hatchet_sdk.v0.connection import new_conn
-from hatchet_sdk.v0.contracts.dispatcher_pb2 import (
+from hatchet_sdk.contracts.dispatcher_pb2 import (
     RESOURCE_TYPE_STEP_RUN,
     RESOURCE_TYPE_WORKFLOW_RUN,
     ResourceEventType,
     SubscribeToWorkflowEventsRequest,
     WorkflowEvent,
 )
-from hatchet_sdk.v0.contracts.dispatcher_pb2_grpc import DispatcherStub
+from hatchet_sdk.contracts.dispatcher_pb2_grpc import DispatcherStub
+from hatchet_sdk.v0.connection import new_conn
 
 from ..loader import ClientConfig
 from ..metadata import get_metadata

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/v0/clients/workflow_listener.py

@@ -6,16 +6,16 @@ from typing import AsyncGenerator
 import grpc
 from grpc._cython import cygrpc
 
-from hatchet_sdk.v0.clients.event_ts import ThreadSafeEvent, read_with_interrupt
-from hatchet_sdk.v0.connection import new_conn
-from hatchet_sdk.v0.contracts.dispatcher_pb2 import (
+from hatchet_sdk.contracts.dispatcher_pb2 import (
     SubscribeToWorkflowRunsRequest,
     WorkflowRunEvent,
 )
-from hatchet_sdk.v0.contracts.dispatcher_pb2_grpc import DispatcherStub
+from hatchet_sdk.contracts.dispatcher_pb2_grpc import DispatcherStub
+from hatchet_sdk.v0.clients.event_ts import ThreadSafeEvent, read_with_interrupt
+from hatchet_sdk.v0.connection import new_conn
 
+from ...logger import logger
 from ..loader import ClientConfig
-from ..logger import logger
 from ..metadata import get_metadata
 
 DEFAULT_WORKFLOW_LISTENER_RETRY_INTERVAL = 3  # seconds

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/v0/context/context.py

@@ -7,21 +7,22 @@ from warnings import warn
 
 from pydantic import BaseModel, StrictStr
 
+from hatchet_sdk.contracts.dispatcher_pb2 import OverridesData
+from hatchet_sdk.contracts.workflows_pb2 import (
+    BulkTriggerWorkflowRequest,
+    TriggerWorkflowRequest,
+)
 from hatchet_sdk.v0.clients.events import EventClient
 from hatchet_sdk.v0.clients.rest.tenacity_utils import tenacity_retry
 from hatchet_sdk.v0.clients.rest_client import RestApi
 from hatchet_sdk.v0.clients.run_event_listener import RunEventListenerClient
 from hatchet_sdk.v0.clients.workflow_listener import PooledWorkflowRunListener
 from hatchet_sdk.v0.context.worker_context import WorkerContext
-from hatchet_sdk.v0.contracts.dispatcher_pb2 import OverridesData
-from hatchet_sdk.v0.contracts.workflows_pb2 import (
-    BulkTriggerWorkflowRequest,
-    TriggerWorkflowRequest,
-)
 from hatchet_sdk.v0.utils.types import WorkflowValidator
 from hatchet_sdk.v0.utils.typing import is_basemodel_subclass
 from hatchet_sdk.v0.workflow_run import WorkflowRunRef
 
+from ...logger import logger
 from ..clients.admin import (
     AdminClient,
     ChildTriggerWorkflowOptions,
@@ -33,7 +34,6 @@ from ..clients.dispatcher.dispatcher import (  # type: ignore[attr-defined]
     Action,
     DispatcherClient,
 )
-from ..logger import logger
 
 DEFAULT_WORKFLOW_POLLING_INTERVAL = 5  # Seconds
 

{hatchet_sdk-1.13.0 → hatchet_sdk-1.14.1}/hatchet_sdk/v0/hatchet.py

@@ -5,14 +5,14 @@ from typing import Any, Callable, Optional, ParamSpec, Type, TypeVar, Union
 from pydantic import BaseModel
 from typing_extensions import deprecated
 
-from hatchet_sdk.v0.clients.rest_client import RestApi
-from hatchet_sdk.v0.context.context import Context
-from hatchet_sdk.v0.contracts.workflows_pb2 import (
+from hatchet_sdk.contracts.workflows_pb2 import (
     ConcurrencyLimitStrategy,
     CreateStepRateLimit,
     DesiredWorkerLabels,
     StickyStrategy,
 )
+from hatchet_sdk.v0.clients.rest_client import RestApi
+from hatchet_sdk.v0.context.context import Context
 from hatchet_sdk.v0.features.cron import CronClient
 from hatchet_sdk.v0.features.scheduled import ScheduledClient
 from hatchet_sdk.v0.labels import DesiredWorkerLabel
@@ -20,12 +20,12 @@ from hatchet_sdk.v0.loader import ClientConfig, ConfigLoader
 from hatchet_sdk.v0.rate_limit import RateLimit
 from hatchet_sdk.v0.v2.callable import HatchetCallable
 
+from ..logger import logger
 from .client import Client, new_client, new_client_raw
 from .clients.admin import AdminClient
 from .clients.dispatcher.dispatcher import DispatcherClient
 from .clients.events import EventClient
 from .clients.run_event_listener import RunEventListenerClient
-from .logger import logger
 from .worker.worker import Worker
 from .workflow import (
     ConcurrencyExpression,