hatchet-sdk 1.0.0__tar.gz → 1.0.1__tar.gz

{hatchet_sdk-1.0.0 → hatchet_sdk-1.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hatchet-sdk
-Version: 1.0.0
+Version: 1.0.1
 Summary: 
 Author: Alexander Belanger
 Author-email: alexander@hatchet.run
@@ -30,7 +30,6 @@ Requires-Dist: protobuf (>=5.29.1,<6.0.0)
 Requires-Dist: pydantic (>=2.6.3,<3.0.0)
 Requires-Dist: pydantic-settings (>=2.7.1,<3.0.0)
 Requires-Dist: python-dateutil (>=2.9.0.post0,<3.0.0)
-Requires-Dist: python-dotenv (>=1.0.0,<2.0.0)
 Requires-Dist: pyyaml (>=6.0.1,<7.0.0)
 Requires-Dist: tenacity (>=8.4.1)
 Requires-Dist: urllib3 (>=1.26.20)

{hatchet_sdk-1.0.0 → hatchet_sdk-1.0.1}/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-1.0.1/hatchet_sdk/client.py

@@ -0,0 +1,56 @@
+import asyncio
+
+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.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:
+    def __init__(
+        self,
+        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:
+            loop = asyncio.get_running_loop()
+        except RuntimeError:
+            loop = asyncio.new_event_loop()
+            asyncio.set_event_loop(loop)
+
+        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
+
+        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)