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

hatchet_sdk/__init__.py

@@ -1,10 +1,9 @@
-from hatchet_sdk.client import new_client
 from hatchet_sdk.clients.admin import (
-    ChildTriggerWorkflowOptions,
     DedupeViolationErr,
     ScheduleTriggerWorkflowOptions,
     TriggerWorkflowOptions,
 )
+from hatchet_sdk.clients.durable_event_listener import RegisterDurableEventRequest
 from hatchet_sdk.clients.events import PushEventOptions
 from hatchet_sdk.clients.rest.models.accept_invite_request import AcceptInviteRequest
 
@@ -76,7 +75,6 @@ from hatchet_sdk.clients.rest.models.pull_request_state import PullRequestState
 from hatchet_sdk.clients.rest.models.reject_invite_request import RejectInviteRequest
 from hatchet_sdk.clients.rest.models.replay_event_request import ReplayEventRequest
 from hatchet_sdk.clients.rest.models.rerun_step_run_request import RerunStepRunRequest
-from hatchet_sdk.clients.rest.models.step import Step
 from hatchet_sdk.clients.rest.models.step_run import StepRun
 from hatchet_sdk.clients.rest.models.step_run_diff import StepRunDiff
 from hatchet_sdk.clients.rest.models.step_run_status import StepRunStatus
@@ -100,6 +98,7 @@ from hatchet_sdk.clients.rest.models.user_tenant_memberships_list import (
     UserTenantMembershipsList,
 )
 from hatchet_sdk.clients.rest.models.user_tenant_public import UserTenantPublic
+from hatchet_sdk.clients.rest.models.v1_task_status import V1TaskStatus
 from hatchet_sdk.clients.rest.models.worker_list import WorkerList
 from hatchet_sdk.clients.rest.models.workflow import Workflow
 from hatchet_sdk.clients.rest.models.workflow_deployment_config import (
@@ -130,23 +129,33 @@ from hatchet_sdk.clients.run_event_listener import (
     WorkflowRunEventType,
 )
 from hatchet_sdk.config import ClientConfig
-from hatchet_sdk.context.context import Context
+from hatchet_sdk.context.context import Context, DurableContext
 from hatchet_sdk.context.worker_context import WorkerContext
 from hatchet_sdk.contracts.workflows_pb2 import (
     CreateWorkflowVersionOpts,
     RateLimitDuration,
     WorkerLabelComparator,
 )
+from hatchet_sdk.features.runs import BulkCancelReplayOpts, RunFilter
 from hatchet_sdk.hatchet import Hatchet
-from hatchet_sdk.utils.aio_utils import sync_to_async
-from hatchet_sdk.worker.worker import Worker, WorkerStartOptions, WorkerStatus
-from hatchet_sdk.workflow import (
-    BaseWorkflow,
+from hatchet_sdk.runnables.task import Task
+from hatchet_sdk.runnables.types import (
     ConcurrencyExpression,
     ConcurrencyLimitStrategy,
+    EmptyModel,
     StickyStrategy,
+    TaskDefaults,
     WorkflowConfig,
 )
+from hatchet_sdk.waits import (
+    Condition,
+    OrGroup,
+    ParentCondition,
+    SleepCondition,
+    UserEventCondition,
+    or_,
+)
+from hatchet_sdk.worker.worker import Worker, WorkerStartOptions, WorkerStatus
 
 __all__ = [
     "AcceptInviteRequest",
@@ -191,11 +200,9 @@ __all__ = [
     "RejectInviteRequest",
     "ReplayEventRequest",
     "RerunStepRunRequest",
-    "Step",
     "StepRun",
     "StepRunDiff",
     "StepRunStatus",
-    "sync_to_async",
     "Tenant",
     "TenantInvite",
     "TenantInviteList",
@@ -231,8 +238,6 @@ __all__ = [
     "CreateWorkflowVersionOpts",
     "RateLimitDuration",
     "StickyStrategy",
-    "new_client",
-    "ChildTriggerWorkflowOptions",
     "DedupeViolationErr",
     "ScheduleTriggerWorkflowOptions",
     "TriggerWorkflowOptions",
@@ -243,14 +248,25 @@ __all__ = [
     "WorkerContext",
     "ClientConfig",
     "Hatchet",
-    "concurrency",
-    "on_failure_step",
-    "step",
     "workflow",
     "Worker",
     "WorkerStartOptions",
     "WorkerStatus",
     "ConcurrencyExpression",
-    "BaseWorkflow",
+    "Workflow",
     "WorkflowConfig",
+    "Task",
+    "EmptyModel",
+    "Condition",
+    "OrGroup",
+    "or_",
+    "SleepCondition",
+    "UserEventCondition",
+    "ParentCondition",
+    "DurableContext",
+    "RegisterDurableEventRequest",
+    "TaskDefaults",
+    "BulkCancelReplayOpts",
+    "RunFilter",
+    "V1TaskStatus",
 ]

hatchet_sdk/client.py

@@ -1,76 +1,32 @@
 import asyncio
-from typing import Callable
 
 import grpc
 
 from hatchet_sdk.clients.admin import AdminClient
 from hatchet_sdk.clients.dispatcher.dispatcher import DispatcherClient
 from hatchet_sdk.clients.events import EventClient, new_event
-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.config import ClientConfig
 from hatchet_sdk.connection import new_conn
+from hatchet_sdk.features.cron import CronClient
+from hatchet_sdk.features.logs import LogsClient
+from hatchet_sdk.features.metrics import MetricsClient
+from hatchet_sdk.features.rate_limits import RateLimitsClient
+from hatchet_sdk.features.runs import RunsClient
+from hatchet_sdk.features.scheduled import ScheduledClient
+from hatchet_sdk.features.workers import WorkersClient
+from hatchet_sdk.features.workflows import WorkflowsClient
 
 
 class Client:
-    @classmethod
-    def from_environment(
-        cls,
-        defaults: ClientConfig = ClientConfig(),
-        debug: bool = False,
-        *opts_functions: Callable[[ClientConfig], None],
-    ) -> "Client":
-        try:
-            loop = asyncio.get_running_loop()
-        except RuntimeError:
-            loop = asyncio.new_event_loop()
-            asyncio.set_event_loop(loop)
-
-        for opt_function in opts_functions:
-            opt_function(defaults)
-
-        return cls.from_config(defaults, debug)
-
-    @classmethod
-    def from_config(
-        cls,
-        config: ClientConfig = ClientConfig(),
-        debug: bool = False,
-    ) -> "Client":
-        try:
-            loop = asyncio.get_running_loop()
-        except RuntimeError:
-            loop = asyncio.new_event_loop()
-            asyncio.set_event_loop(loop)
-
-        conn: grpc.Channel = new_conn(config, False)
-
-        # Instantiate clients
-        event_client = new_event(conn, config)
-        admin_client = AdminClient(config)
-        dispatcher_client = DispatcherClient(config)
-        rest_client = RestApi(config.server_url, config.token, config.tenant_id)
-        workflow_listener = None  # Initialize this if needed
-
-        return cls(
-            event_client,
-            admin_client,
-            dispatcher_client,
-            workflow_listener,
-            rest_client,
-            config,
-            debug,
-        )
-
     def __init__(
         self,
-        event_client: EventClient,
-        admin_client: AdminClient,
-        dispatcher_client: DispatcherClient,
-        workflow_listener: PooledWorkflowRunListener | None,
-        rest_client: RestApi,
         config: ClientConfig,
+        event_client: EventClient | None = None,
+        admin_client: AdminClient | None = None,
+        dispatcher_client: DispatcherClient | None = None,
+        workflow_listener: PooledWorkflowRunListener | None | None = None,
         debug: bool = False,
     ):
         try:
@@ -79,16 +35,22 @@ class Client:
             loop = asyncio.new_event_loop()
             asyncio.set_event_loop(loop)
 
-        self.admin = admin_client
-        self.dispatcher = dispatcher_client
-        self.event = event_client
-        self.rest = rest_client
+        conn: grpc.Channel = new_conn(config, False)
+
         self.config = config
+        self.admin = admin_client or AdminClient(config)
+        self.dispatcher = dispatcher_client or DispatcherClient(config)
+        self.event = event_client or new_event(conn, config)
         self.listener = RunEventListenerClient(config)
         self.workflow_listener = workflow_listener
         self.logInterceptor = config.logger
         self.debug = debug
 
-
-new_client = Client.from_environment
-new_client_raw = Client.from_config
+        self.cron = CronClient(self.config)
+        self.logs = LogsClient(self.config)
+        self.metrics = MetricsClient(self.config)
+        self.rate_limits = RateLimitsClient(self.config)
+        self.runs = RunsClient(self.config)
+        self.scheduled = ScheduledClient(self.config)
+        self.workers = WorkersClient(self.config)
+        self.workflows = WorkflowsClient(self.config)