hatchet-sdk 1.15.3__py3-none-any.whl → 1.16.1__py3-none-any.whl

hatchet_sdk/__init__.py

@@ -152,6 +152,7 @@ from hatchet_sdk.exceptions import (
     NonRetryableException,
     TaskRunError,
 )
+from hatchet_sdk.features.cel import CELEvaluationResult, CELFailure, CELSuccess
 from hatchet_sdk.features.runs import BulkCancelReplayOpts, RunFilter
 from hatchet_sdk.hatchet import Hatchet
 from hatchet_sdk.runnables.task import Task
@@ -180,6 +181,9 @@ __all__ = [
     "APIToken",
     "AcceptInviteRequest",
     "BulkCancelReplayOpts",
+    "CELEvaluationResult",
+    "CELFailure",
+    "CELSuccess",
     "ClientConfig",
     "ClientTLSConfig",
     "ConcurrencyExpression",

hatchet_sdk/client.py

@@ -4,6 +4,7 @@ from hatchet_sdk.clients.events import EventClient
 from hatchet_sdk.clients.listeners.run_event_listener import RunEventListenerClient
 from hatchet_sdk.clients.listeners.workflow_listener import PooledWorkflowRunListener
 from hatchet_sdk.config import ClientConfig
+from hatchet_sdk.features.cel import CELClient
 from hatchet_sdk.features.cron import CronClient
 from hatchet_sdk.features.filters import FiltersClient
 from hatchet_sdk.features.logs import LogsClient
@@ -35,6 +36,7 @@ class Client:
         self.log_interceptor = config.logger
         self.debug = debug
 
+        self.cel = CELClient(self.config)
         self.cron = CronClient(self.config)
         self.filters = FiltersClient(self.config)
         self.logs = LogsClient(self.config)

hatchet_sdk/clients/admin.py

@@ -23,6 +23,7 @@ from hatchet_sdk.metadata import get_metadata
 from hatchet_sdk.rate_limit import RateLimitDuration
 from hatchet_sdk.runnables.contextvars import (
     ctx_action_key,
+    ctx_additional_metadata,
     ctx_step_run_id,
     ctx_worker_id,
     ctx_workflow_run_id,
@@ -278,6 +279,7 @@ class AdminClient:
         step_run_id = ctx_step_run_id.get()
         worker_id = ctx_worker_id.get()
         action_key = ctx_action_key.get()
+        additional_metadata = ctx_additional_metadata.get() or {}
         spawn_index = workflow_spawn_indices[action_key] if action_key else 0
 
         ## Increment the spawn_index for the parent workflow
@@ -296,7 +298,7 @@ class AdminClient:
             parent_step_run_id=options.parent_step_run_id or step_run_id,
             child_key=options.child_key,
             child_index=child_index,
-            additional_metadata=options.additional_metadata,
+            additional_metadata={**additional_metadata, **options.additional_metadata},
             desired_worker_id=desired_worker_id,
             priority=options.priority,
             namespace=options.namespace,

hatchet_sdk/clients/dispatcher/action_listener.py

@@ -140,15 +140,15 @@ class ActionListener:
                     # todo case on "recvmsg:Connection reset by peer" for updates?
                     if self.missed_heartbeats >= 3:
                         # we don't reraise the error here, as we don't want to stop the heartbeat thread
-                        logger.error(
-                            f"⛔️ failed heartbeat ({self.missed_heartbeats}): {e.details()}"
+                        logger.exception(
+                            f"⛔️ failed heartbeat ({self.missed_heartbeats})"
                         )
                     elif self.missed_heartbeats > 1:
                         logger.warning(
                             f"failed to send heartbeat ({self.missed_heartbeats}): {e.details()}"
                         )
                 else:
-                    logger.error(f"failed to send heartbeat: {e}")
+                    logger.exception("failed to send heartbeat")
 
                 if self.interrupt is not None:
                     self.interrupt.set()
@@ -195,7 +195,7 @@ class ActionListener:
 
                     if not t.done():
                         logger.warning(
-                            "Interrupted read_with_interrupt task of action listener"
+                            "interrupted read_with_interrupt task of action listener"
                         )
 
                         t.cancel()
@@ -206,7 +206,7 @@ class ActionListener:
                     result = t.result()
 
                     if isinstance(result, UnexpectedEOF):
-                        logger.debug("Handling EOF in Action Listener")
+                        logger.debug("handling EOF in Action Listener")
                         self.retries = self.retries + 1
                         break
 
@@ -222,8 +222,8 @@ class ActionListener:
                                 assigned_action.actionPayload
                             )
                         )
-                    except (ValueError, json.JSONDecodeError) as e:
-                        logger.error(f"Error decoding payload: {e}")
+                    except (ValueError, json.JSONDecodeError):
+                        logger.exception("error decoding payload")
 
                         action_payload = ActionPayload()
 
@@ -263,9 +263,9 @@ class ActionListener:
                 # Handle different types of errors
                 if e.code() == grpc.StatusCode.CANCELLED:
                     # Context cancelled, unsubscribe and close
-                    logger.debug("Context cancelled, closing listener")
+                    logger.debug("context cancelled, closing listener")
                 elif e.code() == grpc.StatusCode.DEADLINE_EXCEEDED:
-                    logger.info("Deadline exceeded, retrying subscription")
+                    logger.info("deadline exceeded, retrying subscription")
                 elif (
                     self.listen_strategy == "v2"
                     and e.code() == grpc.StatusCode.UNIMPLEMENTED
@@ -277,10 +277,10 @@ class ActionListener:
                 else:
                     # TODO retry
                     if e.code() == grpc.StatusCode.UNAVAILABLE:
-                        logger.error(f"action listener error: {e.details()}")
+                        logger.exception("action listener error")
                     else:
                         # Unknown error, report and break
-                        logger.error(f"action listener error: {e}")
+                        logger.exception("action listener error")
 
                     self.retries = self.retries + 1
 
@@ -346,8 +346,8 @@ class ActionListener:
 
         try:
             self.unregister()
-        except Exception as e:
-            logger.error(f"failed to unregister: {e}")
+        except Exception:
+            logger.exception("failed to unregister")
 
         if self.interrupt:  # type: ignore[truthy-bool]
             self.interrupt.set()

hatchet_sdk/clients/event_ts.py

@@ -66,7 +66,7 @@ async def read_with_interrupt(
         result = cast(TResponse, await listener.read())
 
         if result is cygrpc.EOF:
-            logger.warning("Received EOF from engine")
+            logger.warning("received EOF from engine")
             return UnexpectedEOF()
 
         key = key_generator(result) if key_generator else ""

hatchet_sdk/clients/listeners/pooled_listener.py

@@ -103,7 +103,7 @@ class PooledListener(Generic[R, T, L], ABC):
                     try:
                         self.listener = await self._retry_subscribe()
 
-                        logger.debug("Listener connected.")
+                        logger.debug("listener connected.")
 
                         # spawn an interrupter task
                         if self.interrupter is not None and not self.interrupter.done():
@@ -125,7 +125,7 @@ class PooledListener(Generic[R, T, L], ABC):
 
                             if not t.done():
                                 logger.warning(
-                                    "Interrupted read_with_interrupt task of listener"
+                                    "interrupted read_with_interrupt task of listener"
                                 )
 
                                 t.cancel()
@@ -138,7 +138,7 @@ class PooledListener(Generic[R, T, L], ABC):
 
                             if isinstance(event, UnexpectedEOF):
                                 logger.debug(
-                                    f"Handling EOF in Pooled Listener {self.__class__.__name__}"
+                                    f"handling EOF in Pooled Listener {self.__class__.__name__}"
                                 )
                                 break
 
@@ -153,7 +153,7 @@ class PooledListener(Generic[R, T, L], ABC):
                         continue
 
         except Exception as e:
-            logger.error(f"Error in listener: {e}")
+            logger.exception("error in listener")
 
             self.listener = None
 

hatchet_sdk/clients/rest/__init__.py

@@ -18,6 +18,7 @@ __version__ = "1.0.0"
 
 # import apis into sdk package
 from hatchet_sdk.clients.rest.api.api_token_api import APITokenApi
+from hatchet_sdk.clients.rest.api.cel_api import CELApi
 from hatchet_sdk.clients.rest.api.default_api import DefaultApi
 from hatchet_sdk.clients.rest.api.event_api import EventApi
 from hatchet_sdk.clients.rest.api.filter_api import FilterApi
@@ -231,6 +232,12 @@ from hatchet_sdk.clients.rest.models.user_tenant_memberships_list import (
 )
 from hatchet_sdk.clients.rest.models.user_tenant_public import UserTenantPublic
 from hatchet_sdk.clients.rest.models.v1_cancel_task_request import V1CancelTaskRequest
+from hatchet_sdk.clients.rest.models.v1_cancelled_tasks import V1CancelledTasks
+from hatchet_sdk.clients.rest.models.v1_cel_debug_request import V1CELDebugRequest
+from hatchet_sdk.clients.rest.models.v1_cel_debug_response import V1CELDebugResponse
+from hatchet_sdk.clients.rest.models.v1_cel_debug_response_status import (
+    V1CELDebugResponseStatus,
+)
 from hatchet_sdk.clients.rest.models.v1_create_filter_request import (
     V1CreateFilterRequest,
 )
@@ -247,6 +254,7 @@ from hatchet_sdk.clients.rest.models.v1_log_line import V1LogLine
 from hatchet_sdk.clients.rest.models.v1_log_line_level import V1LogLineLevel
 from hatchet_sdk.clients.rest.models.v1_log_line_list import V1LogLineList
 from hatchet_sdk.clients.rest.models.v1_replay_task_request import V1ReplayTaskRequest
+from hatchet_sdk.clients.rest.models.v1_replayed_tasks import V1ReplayedTasks
 from hatchet_sdk.clients.rest.models.v1_task_event import V1TaskEvent
 from hatchet_sdk.clients.rest.models.v1_task_event_list import V1TaskEventList
 from hatchet_sdk.clients.rest.models.v1_task_event_type import V1TaskEventType

hatchet_sdk/clients/rest/api/__init__.py

@@ -2,6 +2,7 @@
 
 # import apis into api package
 from hatchet_sdk.clients.rest.api.api_token_api import APITokenApi
+from hatchet_sdk.clients.rest.api.cel_api import CELApi
 from hatchet_sdk.clients.rest.api.default_api import DefaultApi
 from hatchet_sdk.clients.rest.api.event_api import EventApi
 from hatchet_sdk.clients.rest.api.filter_api import FilterApi

hatchet_sdk/clients/rest/api/cel_api.py

@@ -0,0 +1,334 @@
+# coding: utf-8
+
+"""
+    Hatchet API
+
+    The Hatchet API
+
+    The version of the OpenAPI document: 1.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+import warnings
+from typing import Any, Dict, List, Optional, Tuple, Union
+
+from pydantic import Field, StrictFloat, StrictInt, StrictStr, validate_call
+from typing_extensions import Annotated
+
+from hatchet_sdk.clients.rest.api_client import ApiClient, RequestSerialized
+from hatchet_sdk.clients.rest.api_response import ApiResponse
+from hatchet_sdk.clients.rest.models.v1_cel_debug_request import V1CELDebugRequest
+from hatchet_sdk.clients.rest.models.v1_cel_debug_response import V1CELDebugResponse
+from hatchet_sdk.clients.rest.rest import RESTResponseType
+
+
+class CELApi:
+    """NOTE: This class is auto generated by OpenAPI Generator
+    Ref: https://openapi-generator.tech
+
+    Do not edit the class manually.
+    """
+
+    def __init__(self, api_client=None) -> None:
+        if api_client is None:
+            api_client = ApiClient.get_default()
+        self.api_client = api_client
+
+    @validate_call
+    def v1_cel_debug(
+        self,
+        tenant: Annotated[
+            str,
+            Field(
+                min_length=36, strict=True, max_length=36, description="The tenant id"
+            ),
+        ],
+        v1_cel_debug_request: Annotated[
+            V1CELDebugRequest,
+            Field(description="The inputs to test the CEL expression against"),
+        ],
+        _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,
+    ) -> V1CELDebugResponse:
+        """Debug a CEL expression
+
+        Evaluate a CEL expression against provided input data.
+
+        :param tenant: The tenant id (required)
+        :type tenant: str
+        :param v1_cel_debug_request: The inputs to test the CEL expression against (required)
+        :type v1_cel_debug_request: V1CELDebugRequest
+        :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_cel_debug_serialize(
+            tenant=tenant,
+            v1_cel_debug_request=v1_cel_debug_request,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "V1CELDebugResponse",
+            "400": "APIErrors",
+            "403": "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_cel_debug_with_http_info(
+        self,
+        tenant: Annotated[
+            str,
+            Field(
+                min_length=36, strict=True, max_length=36, description="The tenant id"
+            ),
+        ],
+        v1_cel_debug_request: Annotated[
+            V1CELDebugRequest,
+            Field(description="The inputs to test the CEL expression against"),
+        ],
+        _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[V1CELDebugResponse]:
+        """Debug a CEL expression
+
+        Evaluate a CEL expression against provided input data.
+
+        :param tenant: The tenant id (required)
+        :type tenant: str
+        :param v1_cel_debug_request: The inputs to test the CEL expression against (required)
+        :type v1_cel_debug_request: V1CELDebugRequest
+        :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_cel_debug_serialize(
+            tenant=tenant,
+            v1_cel_debug_request=v1_cel_debug_request,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "V1CELDebugResponse",
+            "400": "APIErrors",
+            "403": "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_cel_debug_without_preload_content(
+        self,
+        tenant: Annotated[
+            str,
+            Field(
+                min_length=36, strict=True, max_length=36, description="The tenant id"
+            ),
+        ],
+        v1_cel_debug_request: Annotated[
+            V1CELDebugRequest,
+            Field(description="The inputs to test the CEL expression against"),
+        ],
+        _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:
+        """Debug a CEL expression
+
+        Evaluate a CEL expression against provided input data.
+
+        :param tenant: The tenant id (required)
+        :type tenant: str
+        :param v1_cel_debug_request: The inputs to test the CEL expression against (required)
+        :type v1_cel_debug_request: V1CELDebugRequest
+        :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_cel_debug_serialize(
+            tenant=tenant,
+            v1_cel_debug_request=v1_cel_debug_request,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "V1CELDebugResponse",
+            "400": "APIErrors",
+            "403": "APIErrors",
+        }
+        response_data = self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+    def _v1_cel_debug_serialize(
+        self,
+        tenant,
+        v1_cel_debug_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
+        # process the query parameters
+        # process the header parameters
+        # process the form parameters
+        # process the body parameter
+        if v1_cel_debug_request is not None:
+            _body_params = v1_cel_debug_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="POST",
+            resource_path="/api/v1/stable/tenants/{tenant}/cel/debug",
+            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,
+        )

hatchet_sdk/clients/rest/api/task_api.py

@@ -21,8 +21,10 @@ from typing_extensions import Annotated
 from hatchet_sdk.clients.rest.api_client import ApiClient, RequestSerialized
 from hatchet_sdk.clients.rest.api_response import ApiResponse
 from hatchet_sdk.clients.rest.models.v1_cancel_task_request import V1CancelTaskRequest
+from hatchet_sdk.clients.rest.models.v1_cancelled_tasks import V1CancelledTasks
 from hatchet_sdk.clients.rest.models.v1_dag_children import V1DagChildren
 from hatchet_sdk.clients.rest.models.v1_replay_task_request import V1ReplayTaskRequest
+from hatchet_sdk.clients.rest.models.v1_replayed_tasks import V1ReplayedTasks
 from hatchet_sdk.clients.rest.models.v1_task_event_list import V1TaskEventList
 from hatchet_sdk.clients.rest.models.v1_task_point_metrics import V1TaskPointMetrics
 from hatchet_sdk.clients.rest.models.v1_task_run_metric import V1TaskRunMetric
@@ -361,7 +363,7 @@ class TaskApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> None:
+    ) -> V1CancelledTasks:
         """Cancel tasks
 
         Cancel tasks
@@ -402,7 +404,7 @@ class TaskApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": None,
+            "200": "V1CancelledTasks",
             "400": "APIErrors",
             "403": "APIErrors",
             "404": "APIErrors",
@@ -440,7 +442,7 @@ class TaskApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[None]:
+    ) -> ApiResponse[V1CancelledTasks]:
         """Cancel tasks
 
         Cancel tasks
@@ -481,7 +483,7 @@ class TaskApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": None,
+            "200": "V1CancelledTasks",
             "400": "APIErrors",
             "403": "APIErrors",
             "404": "APIErrors",
@@ -560,7 +562,7 @@ class TaskApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": None,
+            "200": "V1CancelledTasks",
             "400": "APIErrors",
             "403": "APIErrors",
             "404": "APIErrors",
@@ -2003,7 +2005,7 @@ class TaskApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> None:
+    ) -> V1ReplayedTasks:
         """Replay tasks
 
         Replay tasks
@@ -2044,7 +2046,7 @@ class TaskApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": None,
+            "200": "V1ReplayedTasks",
             "400": "APIErrors",
             "403": "APIErrors",
             "404": "APIErrors",
@@ -2082,7 +2084,7 @@ class TaskApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[None]:
+    ) -> ApiResponse[V1ReplayedTasks]:
         """Replay tasks
 
         Replay tasks
@@ -2123,7 +2125,7 @@ class TaskApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": None,
+            "200": "V1ReplayedTasks",
             "400": "APIErrors",
             "403": "APIErrors",
             "404": "APIErrors",
@@ -2202,7 +2204,7 @@ class TaskApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": None,
+            "200": "V1ReplayedTasks",
             "400": "APIErrors",
             "403": "APIErrors",
             "404": "APIErrors",

hatchet_sdk/clients/rest/models/__init__.py

@@ -195,6 +195,12 @@ from hatchet_sdk.clients.rest.models.user_tenant_memberships_list import (
 )
 from hatchet_sdk.clients.rest.models.user_tenant_public import UserTenantPublic
 from hatchet_sdk.clients.rest.models.v1_cancel_task_request import V1CancelTaskRequest
+from hatchet_sdk.clients.rest.models.v1_cancelled_tasks import V1CancelledTasks
+from hatchet_sdk.clients.rest.models.v1_cel_debug_request import V1CELDebugRequest
+from hatchet_sdk.clients.rest.models.v1_cel_debug_response import V1CELDebugResponse
+from hatchet_sdk.clients.rest.models.v1_cel_debug_response_status import (
+    V1CELDebugResponseStatus,
+)
 from hatchet_sdk.clients.rest.models.v1_create_filter_request import (
     V1CreateFilterRequest,
 )
@@ -211,6 +217,7 @@ from hatchet_sdk.clients.rest.models.v1_log_line import V1LogLine
 from hatchet_sdk.clients.rest.models.v1_log_line_level import V1LogLineLevel
 from hatchet_sdk.clients.rest.models.v1_log_line_list import V1LogLineList
 from hatchet_sdk.clients.rest.models.v1_replay_task_request import V1ReplayTaskRequest
+from hatchet_sdk.clients.rest.models.v1_replayed_tasks import V1ReplayedTasks
 from hatchet_sdk.clients.rest.models.v1_task_event import V1TaskEvent
 from hatchet_sdk.clients.rest.models.v1_task_event_list import V1TaskEventList
 from hatchet_sdk.clients.rest.models.v1_task_event_type import V1TaskEventType