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

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,
 )
@@ -140,6 +141,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,6 +156,7 @@ 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,
@@ -164,6 +167,7 @@ from hatchet_sdk.waits import (
     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/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/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/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/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/exceptions.py

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

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/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/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/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/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/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/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/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/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/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/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,

hatchet_sdk/v0/opentelemetry/instrumentor.py

@@ -25,6 +25,7 @@ except (RuntimeError, ImportError, ModuleNotFoundError):
     )
 
 import hatchet_sdk
+from hatchet_sdk.contracts.events_pb2 import Event
 from hatchet_sdk.v0.clients.admin import (
     AdminClient,
     TriggerWorkflowOptions,
@@ -36,7 +37,6 @@ from hatchet_sdk.v0.clients.events import (
     EventClient,
     PushEventOptions,
 )
-from hatchet_sdk.v0.contracts.events_pb2 import Event
 from hatchet_sdk.v0.worker.runner.runner import Runner
 from hatchet_sdk.v0.workflow_run import WorkflowRunRef
 

hatchet_sdk/v0/rate_limit.py

@@ -3,7 +3,7 @@ from typing import Union
 
 from celpy import CELEvalError, Environment
 
-from hatchet_sdk.v0.contracts.workflows_pb2 import CreateStepRateLimit
+from hatchet_sdk.contracts.workflows_pb2 import CreateStepRateLimit
 
 
 def validate_cel_expression(expr: str) -> bool:

hatchet_sdk/v0/v2/callable.py

@@ -11,9 +11,7 @@ from typing import (
     Union,
 )
 
-from hatchet_sdk.v0.clients.admin import ChildTriggerWorkflowOptions
-from hatchet_sdk.v0.context.context import Context
-from hatchet_sdk.v0.contracts.workflows_pb2 import (  # type: ignore[attr-defined]
+from hatchet_sdk.contracts.workflows_pb2 import (  # type: ignore[attr-defined]
     CreateStepRateLimit,
     CreateWorkflowJobOpts,
     CreateWorkflowStepOpts,
@@ -23,8 +21,10 @@ from hatchet_sdk.v0.contracts.workflows_pb2 import (  # type: ignore[attr-define
     WorkflowConcurrencyOpts,
     WorkflowKind,
 )
+from hatchet_sdk.logger import logger
+from hatchet_sdk.v0.clients.admin import ChildTriggerWorkflowOptions
+from hatchet_sdk.v0.context.context import Context
 from hatchet_sdk.v0.labels import DesiredWorkerLabel
-from hatchet_sdk.v0.logger import logger
 from hatchet_sdk.v0.rate_limit import RateLimit
 from hatchet_sdk.v0.v2.concurrency import ConcurrencyFunction
 from hatchet_sdk.v0.workflow_run import RunRef

hatchet_sdk/v0/v2/concurrency.py

@@ -1,9 +1,9 @@
 from typing import Any, Callable
 
-from hatchet_sdk.v0.context.context import Context
-from hatchet_sdk.v0.contracts.workflows_pb2 import (  # type: ignore[attr-defined]
+from hatchet_sdk.contracts.workflows_pb2 import (  # type: ignore[attr-defined]
     ConcurrencyLimitStrategy,
 )
+from hatchet_sdk.v0.context.context import Context
 
 
 class ConcurrencyFunction:

hatchet_sdk/v0/v2/hatchet.py

@@ -1,11 +1,11 @@
 from typing import Any, Callable, TypeVar, Union
 
-from hatchet_sdk.v0 import Worker
-from hatchet_sdk.v0.context.context import Context
-from hatchet_sdk.v0.contracts.workflows_pb2 import (  # type: ignore[attr-defined]
+from hatchet_sdk.contracts.workflows_pb2 import (  # type: ignore[attr-defined]
     ConcurrencyLimitStrategy,
     StickyStrategy,
 )
+from hatchet_sdk.v0 import Worker
+from hatchet_sdk.v0.context.context import Context
 from hatchet_sdk.v0.hatchet import Hatchet as HatchetV1
 from hatchet_sdk.v0.hatchet import workflow
 from hatchet_sdk.v0.labels import DesiredWorkerLabel

hatchet_sdk/v0/worker/action_listener_process.py

@@ -8,6 +8,12 @@ from typing import Any, List, Mapping, Optional
 
 import grpc
 
+from hatchet_sdk.contracts.dispatcher_pb2 import (
+    GROUP_KEY_EVENT_TYPE_STARTED,
+    STEP_EVENT_TYPE_STARTED,
+    ActionType,
+)
+from hatchet_sdk.logger import logger
 from hatchet_sdk.v0.client import Client, new_client_raw
 from hatchet_sdk.v0.clients.dispatcher.action_listener import Action
 from hatchet_sdk.v0.clients.dispatcher.dispatcher import (
@@ -16,13 +22,7 @@ from hatchet_sdk.v0.clients.dispatcher.dispatcher import (
     new_dispatcher,
 )
 from hatchet_sdk.v0.clients.rest.models.update_worker_request import UpdateWorkerRequest
-from hatchet_sdk.v0.contracts.dispatcher_pb2 import (
-    GROUP_KEY_EVENT_TYPE_STARTED,
-    STEP_EVENT_TYPE_STARTED,
-    ActionType,
-)
 from hatchet_sdk.v0.loader import ClientConfig
-from hatchet_sdk.v0.logger import logger
 from hatchet_sdk.v0.utils.backoff import exp_backoff_sleep
 
 ACTION_EVENT_RETRY_COUNT = 5

hatchet_sdk/v0/worker/runner/run_loop_manager.py

@@ -4,11 +4,11 @@ from dataclasses import dataclass, field
 from multiprocessing import Queue
 from typing import Callable, TypeVar
 
+from hatchet_sdk.logger import logger
 from hatchet_sdk.v0 import Context
 from hatchet_sdk.v0.client import Client, new_client_raw
 from hatchet_sdk.v0.clients.dispatcher.action_listener import Action
 from hatchet_sdk.v0.loader import ClientConfig
-from hatchet_sdk.v0.logger import logger
 from hatchet_sdk.v0.utils.types import WorkflowValidator
 from hatchet_sdk.v0.worker.runner.runner import Runner
 from hatchet_sdk.v0.worker.runner.utils.capture_logs import capture_logs

hatchet_sdk/v0/worker/runner/runner.py

@@ -13,15 +13,7 @@ from typing import Any, Callable, Dict, cast
 
 from pydantic import BaseModel
 
-from hatchet_sdk.v0.client import new_client_raw
-from hatchet_sdk.v0.clients.admin import new_admin
-from hatchet_sdk.v0.clients.dispatcher.action_listener import Action
-from hatchet_sdk.v0.clients.dispatcher.dispatcher import new_dispatcher
-from hatchet_sdk.v0.clients.run_event_listener import new_listener
-from hatchet_sdk.v0.clients.workflow_listener import PooledWorkflowRunListener
-from hatchet_sdk.v0.context import Context  # type: ignore[attr-defined]
-from hatchet_sdk.v0.context.worker_context import WorkerContext
-from hatchet_sdk.v0.contracts.dispatcher_pb2 import (
+from hatchet_sdk.contracts.dispatcher_pb2 import (
     GROUP_KEY_EVENT_TYPE_COMPLETED,
     GROUP_KEY_EVENT_TYPE_FAILED,
     GROUP_KEY_EVENT_TYPE_STARTED,
@@ -30,8 +22,16 @@ from hatchet_sdk.v0.contracts.dispatcher_pb2 import (
     STEP_EVENT_TYPE_STARTED,
     ActionType,
 )
+from hatchet_sdk.logger import logger
+from hatchet_sdk.v0.client import new_client_raw
+from hatchet_sdk.v0.clients.admin import new_admin
+from hatchet_sdk.v0.clients.dispatcher.action_listener import Action
+from hatchet_sdk.v0.clients.dispatcher.dispatcher import new_dispatcher
+from hatchet_sdk.v0.clients.run_event_listener import new_listener
+from hatchet_sdk.v0.clients.workflow_listener import PooledWorkflowRunListener
+from hatchet_sdk.v0.context import Context  # type: ignore[attr-defined]
+from hatchet_sdk.v0.context.worker_context import WorkerContext
 from hatchet_sdk.v0.loader import ClientConfig
-from hatchet_sdk.v0.logger import logger
 from hatchet_sdk.v0.utils.types import WorkflowValidator
 from hatchet_sdk.v0.v2.callable import DurableContext
 from hatchet_sdk.v0.worker.action_listener_process import ActionEvent

hatchet_sdk/v0/worker/runner/utils/capture_logs.py

@@ -5,7 +5,7 @@ from concurrent.futures import ThreadPoolExecutor
 from io import StringIO
 from typing import Any, Coroutine
 
-from hatchet_sdk.v0 import logger
+from hatchet_sdk import logger
 from hatchet_sdk.v0.clients.events import EventClient
 
 wr: contextvars.ContextVar[str | None] = contextvars.ContextVar(

hatchet_sdk/v0/worker/worker.py

@@ -17,11 +17,11 @@ from aiohttp.web_request import Request
 from aiohttp.web_response import Response
 from prometheus_client import CONTENT_TYPE_LATEST, Gauge, generate_latest
 
+from hatchet_sdk.contracts.workflows_pb2 import CreateWorkflowVersionOpts
+from hatchet_sdk.logger import logger
 from hatchet_sdk.v0 import Context
 from hatchet_sdk.v0.client import Client, new_client_raw
-from hatchet_sdk.v0.contracts.workflows_pb2 import CreateWorkflowVersionOpts
 from hatchet_sdk.v0.loader import ClientConfig
-from hatchet_sdk.v0.logger import logger
 from hatchet_sdk.v0.utils.types import WorkflowValidator
 from hatchet_sdk.v0.utils.typing import is_basemodel_subclass
 from hatchet_sdk.v0.v2.callable import HatchetCallable