hatchet-sdk 1.11.0__py3-none-any.whl → 1.12.0__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/admin.py

@@ -361,7 +361,8 @@ class AdminClient:
         try:
             resp = cast(
                 v0_workflow_protos.TriggerWorkflowResponse,
-                client.TriggerWorkflow(
+                await asyncio.to_thread(
+                    client.TriggerWorkflow,
                     request,
                     metadata=get_metadata(self.token),
                 ),
@@ -450,7 +451,8 @@ class AdminClient:
 
             resp = cast(
                 v0_workflow_protos.BulkTriggerWorkflowResponse,
-                client.BulkTriggerWorkflow(
+                await asyncio.to_thread(
+                    client.BulkTriggerWorkflow,
                     bulk_request,
                     metadata=get_metadata(self.token),
                 ),

hatchet_sdk/clients/events.py

@@ -28,7 +28,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 +209,20 @@ 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,
     ) -> 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
+        )
 
     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,
     ) -> V1EventList:
         with self.client() as client:
             return self._ea(client).v1_event_list(
@@ -224,4 +230,6 @@ class EventClient(BaseRestClient):
                 offset=offset,
                 limit=limit,
                 keys=keys,
+                since=since,
+                until=until,
             )

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,
@@ -261,6 +262,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 (

hatchet_sdk/clients/rest/api/event_api.py

@@ -12,6 +12,7 @@
 """  # noqa: E501
 
 import warnings
+from datetime import datetime
 from typing import Any, Dict, List, Optional, Tuple, Union
 
 from pydantic import Field, StrictFloat, StrictInt, StrictStr, validate_call
@@ -2566,6 +2567,14 @@ class EventApi:
         keys: Annotated[
             Optional[List[StrictStr]], Field(description="A list of keys to filter by")
         ] = None,
+        since: Annotated[
+            Optional[datetime],
+            Field(description="Consider events that occurred after this time"),
+        ] = None,
+        until: Annotated[
+            Optional[datetime],
+            Field(description="Consider events that occurred before this time"),
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -2590,6 +2599,10 @@ class EventApi:
         :type limit: int
         :param keys: A list of keys to filter by
         :type keys: List[str]
+        :param since: Consider events that occurred after this time
+        :type since: datetime
+        :param until: Consider events that occurred before this time
+        :type until: datetime
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -2617,6 +2630,8 @@ class EventApi:
             offset=offset,
             limit=limit,
             keys=keys,
+            since=since,
+            until=until,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -2655,6 +2670,14 @@ class EventApi:
         keys: Annotated[
             Optional[List[StrictStr]], Field(description="A list of keys to filter by")
         ] = None,
+        since: Annotated[
+            Optional[datetime],
+            Field(description="Consider events that occurred after this time"),
+        ] = None,
+        until: Annotated[
+            Optional[datetime],
+            Field(description="Consider events that occurred before this time"),
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -2679,6 +2702,10 @@ class EventApi:
         :type limit: int
         :param keys: A list of keys to filter by
         :type keys: List[str]
+        :param since: Consider events that occurred after this time
+        :type since: datetime
+        :param until: Consider events that occurred before this time
+        :type until: datetime
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -2706,6 +2733,8 @@ class EventApi:
             offset=offset,
             limit=limit,
             keys=keys,
+            since=since,
+            until=until,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -2744,6 +2773,14 @@ class EventApi:
         keys: Annotated[
             Optional[List[StrictStr]], Field(description="A list of keys to filter by")
         ] = None,
+        since: Annotated[
+            Optional[datetime],
+            Field(description="Consider events that occurred after this time"),
+        ] = None,
+        until: Annotated[
+            Optional[datetime],
+            Field(description="Consider events that occurred before this time"),
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -2768,6 +2805,10 @@ class EventApi:
         :type limit: int
         :param keys: A list of keys to filter by
         :type keys: List[str]
+        :param since: Consider events that occurred after this time
+        :type since: datetime
+        :param until: Consider events that occurred before this time
+        :type until: datetime
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -2795,6 +2836,8 @@ class EventApi:
             offset=offset,
             limit=limit,
             keys=keys,
+            since=since,
+            until=until,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -2817,6 +2860,8 @@ class EventApi:
         offset,
         limit,
         keys,
+        since,
+        until,
         _request_auth,
         _content_type,
         _headers,
@@ -2854,6 +2899,28 @@ class EventApi:
 
             _query_params.append(("keys", keys))
 
+        if since is not None:
+            if isinstance(since, datetime):
+                _query_params.append(
+                    (
+                        "since",
+                        since.strftime(self.api_client.configuration.datetime_format),
+                    )
+                )
+            else:
+                _query_params.append(("since", since))
+
+        if until is not None:
+            if isinstance(until, datetime):
+                _query_params.append(
+                    (
+                        "until",
+                        until.strftime(self.api_client.configuration.datetime_format),
+                    )
+                )
+            else:
+                _query_params.append(("until", until))
+
         # process the header parameters
         # process the form parameters
         # process the body parameter

hatchet_sdk/clients/rest/api/filter_api.py

@@ -24,6 +24,9 @@ from hatchet_sdk.clients.rest.models.v1_create_filter_request import (
 )
 from hatchet_sdk.clients.rest.models.v1_filter import V1Filter
 from hatchet_sdk.clients.rest.models.v1_filter_list import V1FilterList
+from hatchet_sdk.clients.rest.models.v1_update_filter_request import (
+    V1UpdateFilterRequest,
+)
 from hatchet_sdk.clients.rest.rest import RESTResponseType
 
 
@@ -1303,3 +1306,339 @@ class FilterApi:
             _host=_host,
             _request_auth=_request_auth,
         )
+
+    @validate_call
+    def v1_filter_update(
+        self,
+        tenant: Annotated[
+            str,
+            Field(
+                min_length=36, strict=True, max_length=36, description="The tenant id"
+            ),
+        ],
+        v1_filter: Annotated[
+            str,
+            Field(
+                min_length=36,
+                strict=True,
+                max_length=36,
+                description="The filter id to update",
+            ),
+        ],
+        v1_update_filter_request: Annotated[
+            V1UpdateFilterRequest, Field(description="The input to the filter update")
+        ],
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)], Annotated[StrictFloat, Field(gt=0)]
+            ],
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> V1Filter:
+        """v1_filter_update
+
+        Update a filter
+
+        :param tenant: The tenant id (required)
+        :type tenant: str
+        :param v1_filter: The filter id to update (required)
+        :type v1_filter: str
+        :param v1_update_filter_request: The input to the filter update (required)
+        :type v1_update_filter_request: V1UpdateFilterRequest
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """  # noqa: E501
+
+        _param = self._v1_filter_update_serialize(
+            tenant=tenant,
+            v1_filter=v1_filter,
+            v1_update_filter_request=v1_update_filter_request,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "V1Filter",
+            "400": "APIErrors",
+            "403": "APIErrors",
+            "404": "APIErrors",
+        }
+        response_data = self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+    @validate_call
+    def v1_filter_update_with_http_info(
+        self,
+        tenant: Annotated[
+            str,
+            Field(
+                min_length=36, strict=True, max_length=36, description="The tenant id"
+            ),
+        ],
+        v1_filter: Annotated[
+            str,
+            Field(
+                min_length=36,
+                strict=True,
+                max_length=36,
+                description="The filter id to update",
+            ),
+        ],
+        v1_update_filter_request: Annotated[
+            V1UpdateFilterRequest, Field(description="The input to the filter update")
+        ],
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)], Annotated[StrictFloat, Field(gt=0)]
+            ],
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ApiResponse[V1Filter]:
+        """v1_filter_update
+
+        Update a filter
+
+        :param tenant: The tenant id (required)
+        :type tenant: str
+        :param v1_filter: The filter id to update (required)
+        :type v1_filter: str
+        :param v1_update_filter_request: The input to the filter update (required)
+        :type v1_update_filter_request: V1UpdateFilterRequest
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """  # noqa: E501
+
+        _param = self._v1_filter_update_serialize(
+            tenant=tenant,
+            v1_filter=v1_filter,
+            v1_update_filter_request=v1_update_filter_request,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "V1Filter",
+            "400": "APIErrors",
+            "403": "APIErrors",
+            "404": "APIErrors",
+        }
+        response_data = self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+    @validate_call
+    def v1_filter_update_without_preload_content(
+        self,
+        tenant: Annotated[
+            str,
+            Field(
+                min_length=36, strict=True, max_length=36, description="The tenant id"
+            ),
+        ],
+        v1_filter: Annotated[
+            str,
+            Field(
+                min_length=36,
+                strict=True,
+                max_length=36,
+                description="The filter id to update",
+            ),
+        ],
+        v1_update_filter_request: Annotated[
+            V1UpdateFilterRequest, Field(description="The input to the filter update")
+        ],
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)], Annotated[StrictFloat, Field(gt=0)]
+            ],
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> RESTResponseType:
+        """v1_filter_update
+
+        Update a filter
+
+        :param tenant: The tenant id (required)
+        :type tenant: str
+        :param v1_filter: The filter id to update (required)
+        :type v1_filter: str
+        :param v1_update_filter_request: The input to the filter update (required)
+        :type v1_update_filter_request: V1UpdateFilterRequest
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """  # noqa: E501
+
+        _param = self._v1_filter_update_serialize(
+            tenant=tenant,
+            v1_filter=v1_filter,
+            v1_update_filter_request=v1_update_filter_request,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "V1Filter",
+            "400": "APIErrors",
+            "403": "APIErrors",
+            "404": "APIErrors",
+        }
+        response_data = self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+    def _v1_filter_update_serialize(
+        self,
+        tenant,
+        v1_filter,
+        v1_update_filter_request,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: Dict[str, str] = {}
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        if tenant is not None:
+            _path_params["tenant"] = tenant
+        if v1_filter is not None:
+            _path_params["v1-filter"] = v1_filter
+        # process the query parameters
+        # process the header parameters
+        # process the form parameters
+        # process the body parameter
+        if v1_update_filter_request is not None:
+            _body_params = v1_update_filter_request
+
+        # set the HTTP header `Accept`
+        if "Accept" not in _header_params:
+            _header_params["Accept"] = self.api_client.select_header_accept(
+                ["application/json"]
+            )
+
+        # set the HTTP header `Content-Type`
+        if _content_type:
+            _header_params["Content-Type"] = _content_type
+        else:
+            _default_content_type = self.api_client.select_header_content_type(
+                ["application/json"]
+            )
+            if _default_content_type is not None:
+                _header_params["Content-Type"] = _default_content_type
+
+        # authentication setting
+        _auth_settings: List[str] = ["cookieAuth", "bearerAuth"]
+
+        return self.api_client.param_serialize(
+            method="PATCH",
+            resource_path="/api/v1/stable/tenants/{tenant}/filters/{v1-filter}",
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth,
+        )