hatchet-sdk 1.11.1__py3-none-any.whl → 1.12.1__py3-none-any.whl

hatchet_sdk/__init__.py

@@ -144,6 +144,7 @@ from hatchet_sdk.runnables.task import Task
 from hatchet_sdk.runnables.types import (
     ConcurrencyExpression,
     ConcurrencyLimitStrategy,
+    DefaultFilter,
     EmptyModel,
     StickyStrategy,
     TaskDefaults,
@@ -275,4 +276,5 @@ __all__ = [
     "OTelAttribute",
     "OpenTelemetryConfig",
     "ClientTLSConfig",
+    "DefaultFilter",
 ]

hatchet_sdk/client.py

@@ -11,6 +11,7 @@ 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.tenant import TenantClient
 from hatchet_sdk.features.workers import WorkersClient
 from hatchet_sdk.features.workflows import WorkflowsClient
 
@@ -45,6 +46,7 @@ class Client:
             workflow_run_listener=self.workflow_listener,
         )
         self.scheduled = ScheduledClient(self.config)
+        self.tenant = TenantClient(self.config)
         self.workers = WorkersClient(self.config)
         self.workflows = WorkflowsClient(self.config)
 

hatchet_sdk/clients/events.py

@@ -10,8 +10,12 @@ from hatchet_sdk.clients.rest.api.event_api import EventApi
 from hatchet_sdk.clients.rest.api.workflow_runs_api import WorkflowRunsApi
 from hatchet_sdk.clients.rest.api_client import ApiClient
 from hatchet_sdk.clients.rest.models.v1_event_list import V1EventList
+from hatchet_sdk.clients.rest.models.v1_task_status import V1TaskStatus
 from hatchet_sdk.clients.rest.tenacity_utils import tenacity_retry
-from hatchet_sdk.clients.v1.api_client import BaseRestClient
+from hatchet_sdk.clients.v1.api_client import (
+    BaseRestClient,
+    maybe_additional_metadata_to_kv,
+)
 from hatchet_sdk.config import ClientConfig
 from hatchet_sdk.connection import new_conn
 from hatchet_sdk.contracts.events_pb2 import (
@@ -28,7 +32,7 @@ from hatchet_sdk.utils.typing import JSONSerializableMapping
 
 
 def proto_timestamp_now() -> timestamp_pb2.Timestamp:
-    t = datetime.datetime.now().timestamp()
+    t = datetime.datetime.now(tz=datetime.timezone.utc).timestamp()
     seconds = int(t)
     nanos = int(t % 1 * 1e9)
 
@@ -209,14 +213,40 @@ class EventClient(BaseRestClient):
         offset: int | None = None,
         limit: int | None = None,
         keys: list[str] | None = None,
+        since: datetime.datetime | None = None,
+        until: datetime.datetime | None = None,
+        workflow_ids: list[str] | None = None,
+        workflow_run_statuses: list[V1TaskStatus] | None = None,
+        event_ids: list[str] | None = None,
+        additional_metadata: JSONSerializableMapping | None = None,
+        scopes: list[str] | None = None,
     ) -> V1EventList:
-        return await asyncio.to_thread(self.list, offset=offset, limit=limit, keys=keys)
+        return await asyncio.to_thread(
+            self.list,
+            offset=offset,
+            limit=limit,
+            keys=keys,
+            since=since,
+            until=until,
+            workflow_ids=workflow_ids,
+            workflow_run_statuses=workflow_run_statuses,
+            event_ids=event_ids,
+            additional_metadata=additional_metadata,
+            scopes=scopes,
+        )
 
     def list(
         self,
         offset: int | None = None,
         limit: int | None = None,
         keys: list[str] | None = None,
+        since: datetime.datetime | None = None,
+        until: datetime.datetime | None = None,
+        workflow_ids: list[str] | None = None,
+        workflow_run_statuses: list[V1TaskStatus] | None = None,
+        event_ids: list[str] | None = None,
+        additional_metadata: JSONSerializableMapping | None = None,
+        scopes: list[str] | None = None,
     ) -> V1EventList:
         with self.client() as client:
             return self._ea(client).v1_event_list(
@@ -224,4 +254,13 @@ class EventClient(BaseRestClient):
                 offset=offset,
                 limit=limit,
                 keys=keys,
+                since=since,
+                until=until,
+                workflow_ids=workflow_ids,
+                workflow_run_statuses=workflow_run_statuses,
+                event_ids=event_ids,
+                additional_metadata=maybe_additional_metadata_to_kv(
+                    additional_metadata
+                ),
+                scopes=scopes,
             )

hatchet_sdk/clients/rest/__init__.py

@@ -207,6 +207,7 @@ from hatchet_sdk.clients.rest.models.tenant_resource_policy import TenantResourc
 from hatchet_sdk.clients.rest.models.tenant_step_run_queue_metrics import (
     TenantStepRunQueueMetrics,
 )
+from hatchet_sdk.clients.rest.models.tenant_ui_version import TenantUIVersion
 from hatchet_sdk.clients.rest.models.tenant_version import TenantVersion
 from hatchet_sdk.clients.rest.models.trigger_workflow_run_request import (
     TriggerWorkflowRunRequest,
@@ -236,6 +237,7 @@ from hatchet_sdk.clients.rest.models.v1_create_filter_request import (
 from hatchet_sdk.clients.rest.models.v1_dag_children import V1DagChildren
 from hatchet_sdk.clients.rest.models.v1_event import V1Event
 from hatchet_sdk.clients.rest.models.v1_event_list import V1EventList
+from hatchet_sdk.clients.rest.models.v1_event_triggered_run import V1EventTriggeredRun
 from hatchet_sdk.clients.rest.models.v1_event_workflow_run_summary import (
     V1EventWorkflowRunSummary,
 )
@@ -261,6 +263,9 @@ from hatchet_sdk.clients.rest.models.v1_task_timing_list import V1TaskTimingList
 from hatchet_sdk.clients.rest.models.v1_trigger_workflow_run_request import (
     V1TriggerWorkflowRunRequest,
 )
+from hatchet_sdk.clients.rest.models.v1_update_filter_request import (
+    V1UpdateFilterRequest,
+)
 from hatchet_sdk.clients.rest.models.v1_workflow_run import V1WorkflowRun
 from hatchet_sdk.clients.rest.models.v1_workflow_run_details import V1WorkflowRunDetails
 from hatchet_sdk.clients.rest.models.v1_workflow_run_display_name import (