hatchet-sdk 1.9.0__py3-none-any.whl → 1.10.0__py3-none-any.whl

hatchet_sdk/clients/rest/api/task_api.py

@@ -954,6 +954,9 @@ class TaskApi:
             str,
             Field(min_length=36, strict=True, max_length=36, description="The task id"),
         ],
+        attempt: Annotated[
+            Optional[StrictInt], Field(description="The attempt number")
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -972,6 +975,8 @@ class TaskApi:
 
         :param task: The task id (required)
         :type task: str
+        :param attempt: The attempt number
+        :type attempt: int
         :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
@@ -996,6 +1001,7 @@ class TaskApi:
 
         _param = self._v1_task_get_serialize(
             task=task,
+            attempt=attempt,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -1025,6 +1031,9 @@ class TaskApi:
             str,
             Field(min_length=36, strict=True, max_length=36, description="The task id"),
         ],
+        attempt: Annotated[
+            Optional[StrictInt], Field(description="The attempt number")
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1043,6 +1052,8 @@ class TaskApi:
 
         :param task: The task id (required)
         :type task: str
+        :param attempt: The attempt number
+        :type attempt: int
         :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
@@ -1067,6 +1078,7 @@ class TaskApi:
 
         _param = self._v1_task_get_serialize(
             task=task,
+            attempt=attempt,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -1096,6 +1108,9 @@ class TaskApi:
             str,
             Field(min_length=36, strict=True, max_length=36, description="The task id"),
         ],
+        attempt: Annotated[
+            Optional[StrictInt], Field(description="The attempt number")
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1114,6 +1129,8 @@ class TaskApi:
 
         :param task: The task id (required)
         :type task: str
+        :param attempt: The attempt number
+        :type attempt: int
         :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
@@ -1138,6 +1155,7 @@ class TaskApi:
 
         _param = self._v1_task_get_serialize(
             task=task,
+            attempt=attempt,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -1159,6 +1177,7 @@ class TaskApi:
     def _v1_task_get_serialize(
         self,
         task,
+        attempt,
         _request_auth,
         _content_type,
         _headers,
@@ -1182,6 +1201,10 @@ class TaskApi:
         if task is not None:
             _path_params["task"] = task
         # process the query parameters
+        if attempt is not None:
+
+            _query_params.append(("attempt", attempt))
+
         # process the header parameters
         # process the form parameters
         # process the body parameter
@@ -1568,6 +1591,10 @@ class TaskApi:
             Optional[Annotated[str, Field(min_length=36, strict=True, max_length=36)]],
             Field(description="The parent task's external id"),
         ] = None,
+        triggering_event_external_id: Annotated[
+            Optional[Annotated[str, Field(min_length=36, strict=True, max_length=36)]],
+            Field(description="The id of the event that triggered the task"),
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1594,6 +1621,8 @@ class TaskApi:
         :type workflow_ids: List[str]
         :param parent_task_external_id: The parent task's external id
         :type parent_task_external_id: str
+        :param triggering_event_external_id: The id of the event that triggered the task
+        :type triggering_event_external_id: str
         :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
@@ -1622,6 +1651,7 @@ class TaskApi:
             until=until,
             workflow_ids=workflow_ids,
             parent_task_external_id=parent_task_external_id,
+            triggering_event_external_id=triggering_event_external_id,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -1668,6 +1698,10 @@ class TaskApi:
             Optional[Annotated[str, Field(min_length=36, strict=True, max_length=36)]],
             Field(description="The parent task's external id"),
         ] = None,
+        triggering_event_external_id: Annotated[
+            Optional[Annotated[str, Field(min_length=36, strict=True, max_length=36)]],
+            Field(description="The id of the event that triggered the task"),
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1694,6 +1728,8 @@ class TaskApi:
         :type workflow_ids: List[str]
         :param parent_task_external_id: The parent task's external id
         :type parent_task_external_id: str
+        :param triggering_event_external_id: The id of the event that triggered the task
+        :type triggering_event_external_id: str
         :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
@@ -1722,6 +1758,7 @@ class TaskApi:
             until=until,
             workflow_ids=workflow_ids,
             parent_task_external_id=parent_task_external_id,
+            triggering_event_external_id=triggering_event_external_id,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -1768,6 +1805,10 @@ class TaskApi:
             Optional[Annotated[str, Field(min_length=36, strict=True, max_length=36)]],
             Field(description="The parent task's external id"),
         ] = None,
+        triggering_event_external_id: Annotated[
+            Optional[Annotated[str, Field(min_length=36, strict=True, max_length=36)]],
+            Field(description="The id of the event that triggered the task"),
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1794,6 +1835,8 @@ class TaskApi:
         :type workflow_ids: List[str]
         :param parent_task_external_id: The parent task's external id
         :type parent_task_external_id: str
+        :param triggering_event_external_id: The id of the event that triggered the task
+        :type triggering_event_external_id: str
         :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
@@ -1822,6 +1865,7 @@ class TaskApi:
             until=until,
             workflow_ids=workflow_ids,
             parent_task_external_id=parent_task_external_id,
+            triggering_event_external_id=triggering_event_external_id,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -1846,6 +1890,7 @@ class TaskApi:
         until,
         workflow_ids,
         parent_task_external_id,
+        triggering_event_external_id,
         _request_auth,
         _content_type,
         _headers,
@@ -1901,6 +1946,12 @@ class TaskApi:
 
             _query_params.append(("parent_task_external_id", parent_task_external_id))
 
+        if triggering_event_external_id is not None:
+
+            _query_params.append(
+                ("triggering_event_external_id", triggering_event_external_id)
+            )
+
         # process the header parameters
         # process the form parameters
         # process the body parameter

hatchet_sdk/clients/rest/api/workflow_runs_api.py

@@ -1258,6 +1258,12 @@ class WorkflowRunsApi:
             Optional[Annotated[str, Field(min_length=36, strict=True, max_length=36)]],
             Field(description="The parent task external id to filter by"),
         ] = None,
+        triggering_event_external_id: Annotated[
+            Optional[Annotated[str, Field(min_length=36, strict=True, max_length=36)]],
+            Field(
+                description="The external id of the event that triggered the workflow run"
+            ),
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1296,6 +1302,8 @@ class WorkflowRunsApi:
         :type worker_id: str
         :param parent_task_external_id: The parent task external id to filter by
         :type parent_task_external_id: str
+        :param triggering_event_external_id: The external id of the event that triggered the workflow run
+        :type triggering_event_external_id: str
         :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
@@ -1330,6 +1338,7 @@ class WorkflowRunsApi:
             workflow_ids=workflow_ids,
             worker_id=worker_id,
             parent_task_external_id=parent_task_external_id,
+            triggering_event_external_id=triggering_event_external_id,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -1396,6 +1405,12 @@ class WorkflowRunsApi:
             Optional[Annotated[str, Field(min_length=36, strict=True, max_length=36)]],
             Field(description="The parent task external id to filter by"),
         ] = None,
+        triggering_event_external_id: Annotated[
+            Optional[Annotated[str, Field(min_length=36, strict=True, max_length=36)]],
+            Field(
+                description="The external id of the event that triggered the workflow run"
+            ),
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1434,6 +1449,8 @@ class WorkflowRunsApi:
         :type worker_id: str
         :param parent_task_external_id: The parent task external id to filter by
         :type parent_task_external_id: str
+        :param triggering_event_external_id: The external id of the event that triggered the workflow run
+        :type triggering_event_external_id: str
         :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
@@ -1468,6 +1485,7 @@ class WorkflowRunsApi:
             workflow_ids=workflow_ids,
             worker_id=worker_id,
             parent_task_external_id=parent_task_external_id,
+            triggering_event_external_id=triggering_event_external_id,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -1534,6 +1552,12 @@ class WorkflowRunsApi:
             Optional[Annotated[str, Field(min_length=36, strict=True, max_length=36)]],
             Field(description="The parent task external id to filter by"),
         ] = None,
+        triggering_event_external_id: Annotated[
+            Optional[Annotated[str, Field(min_length=36, strict=True, max_length=36)]],
+            Field(
+                description="The external id of the event that triggered the workflow run"
+            ),
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1572,6 +1596,8 @@ class WorkflowRunsApi:
         :type worker_id: str
         :param parent_task_external_id: The parent task external id to filter by
         :type parent_task_external_id: str
+        :param triggering_event_external_id: The external id of the event that triggered the workflow run
+        :type triggering_event_external_id: str
         :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
@@ -1606,6 +1632,7 @@ class WorkflowRunsApi:
             workflow_ids=workflow_ids,
             worker_id=worker_id,
             parent_task_external_id=parent_task_external_id,
+            triggering_event_external_id=triggering_event_external_id,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -1636,6 +1663,7 @@ class WorkflowRunsApi:
         workflow_ids,
         worker_id,
         parent_task_external_id,
+        triggering_event_external_id,
         _request_auth,
         _content_type,
         _headers,
@@ -1717,6 +1745,12 @@ class WorkflowRunsApi:
 
             _query_params.append(("parent_task_external_id", parent_task_external_id))
 
+        if triggering_event_external_id is not None:
+
+            _query_params.append(
+                ("triggering_event_external_id", triggering_event_external_id)
+            )
+
         # process the header parameters
         # process the form parameters
         # process the body parameter

hatchet_sdk/clients/rest/models/__init__.py

@@ -194,7 +194,17 @@ 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_create_filter_request import (
+    V1CreateFilterRequest,
+)
 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_workflow_run_summary import (
+    V1EventWorkflowRunSummary,
+)
+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_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

hatchet_sdk/clients/rest/models/create_event_request.py

@@ -19,7 +19,7 @@ import pprint
 import re  # noqa: F401
 from typing import Any, ClassVar, Dict, List, Optional, Set
 
-from pydantic import BaseModel, ConfigDict, Field, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr
 from typing_extensions import Self
 
 
@@ -35,7 +35,19 @@ class CreateEventRequest(BaseModel):
         description="Additional metadata for the event.",
         alias="additionalMetadata",
     )
-    __properties: ClassVar[List[str]] = ["key", "data", "additionalMetadata"]
+    priority: Optional[StrictInt] = Field(
+        default=None, description="The priority of the event."
+    )
+    scope: Optional[StrictStr] = Field(
+        default=None, description="The scope for event filtering."
+    )
+    __properties: ClassVar[List[str]] = [
+        "key",
+        "data",
+        "additionalMetadata",
+        "priority",
+        "scope",
+    ]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -90,6 +102,8 @@ class CreateEventRequest(BaseModel):
                 "key": obj.get("key"),
                 "data": obj.get("data"),
                 "additionalMetadata": obj.get("additionalMetadata"),
+                "priority": obj.get("priority"),
+                "scope": obj.get("scope"),
             }
         )
         return _obj

hatchet_sdk/clients/rest/models/v1_create_filter_request.py

@@ -0,0 +1,99 @@
+# 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
+
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+from typing import Any, ClassVar, Dict, List, Optional, Set
+
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
+from typing_extensions import Annotated, Self
+
+
+class V1CreateFilterRequest(BaseModel):
+    """
+    V1CreateFilterRequest
+    """  # noqa: E501
+
+    workflow_id: Annotated[str, Field(min_length=36, strict=True, max_length=36)] = (
+        Field(description="The workflow id", alias="workflowId")
+    )
+    expression: StrictStr = Field(description="The expression for the filter")
+    scope: StrictStr = Field(
+        description="The scope associated with this filter. Used for subsetting candidate filters at evaluation time"
+    )
+    payload: Optional[Dict[str, Any]] = Field(
+        default=None, description="The payload for the filter"
+    )
+    __properties: ClassVar[List[str]] = ["workflowId", "expression", "scope", "payload"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of V1CreateFilterRequest from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of V1CreateFilterRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate(
+            {
+                "workflowId": obj.get("workflowId"),
+                "expression": obj.get("expression"),
+                "scope": obj.get("scope"),
+                "payload": obj.get("payload"),
+            }
+        )
+        return _obj

hatchet_sdk/clients/rest/models/v1_event.py

@@ -0,0 +1,142 @@
+# 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
+
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+from typing import Any, ClassVar, Dict, List, Optional, Set
+
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
+from typing_extensions import Self
+
+from hatchet_sdk.clients.rest.models.api_resource_meta import APIResourceMeta
+from hatchet_sdk.clients.rest.models.tenant import Tenant
+from hatchet_sdk.clients.rest.models.v1_event_workflow_run_summary import (
+    V1EventWorkflowRunSummary,
+)
+
+
+class V1Event(BaseModel):
+    """
+    V1Event
+    """  # noqa: E501
+
+    metadata: APIResourceMeta
+    key: StrictStr = Field(description="The key for the event.")
+    tenant: Optional[Tenant] = Field(
+        default=None, description="The tenant associated with this event."
+    )
+    tenant_id: StrictStr = Field(
+        description="The ID of the tenant associated with this event.", alias="tenantId"
+    )
+    workflow_run_summary: V1EventWorkflowRunSummary = Field(
+        description="The workflow run summary for this event.",
+        alias="workflowRunSummary",
+    )
+    additional_metadata: Optional[Dict[str, Any]] = Field(
+        default=None,
+        description="Additional metadata for the event.",
+        alias="additionalMetadata",
+    )
+    __properties: ClassVar[List[str]] = [
+        "metadata",
+        "key",
+        "tenant",
+        "tenantId",
+        "workflowRunSummary",
+        "additionalMetadata",
+    ]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of V1Event from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of metadata
+        if self.metadata:
+            _dict["metadata"] = self.metadata.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of tenant
+        if self.tenant:
+            _dict["tenant"] = self.tenant.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of workflow_run_summary
+        if self.workflow_run_summary:
+            _dict["workflowRunSummary"] = self.workflow_run_summary.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of V1Event from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate(
+            {
+                "metadata": (
+                    APIResourceMeta.from_dict(obj["metadata"])
+                    if obj.get("metadata") is not None
+                    else None
+                ),
+                "key": obj.get("key"),
+                "tenant": (
+                    Tenant.from_dict(obj["tenant"])
+                    if obj.get("tenant") is not None
+                    else None
+                ),
+                "tenantId": obj.get("tenantId"),
+                "workflowRunSummary": (
+                    V1EventWorkflowRunSummary.from_dict(obj["workflowRunSummary"])
+                    if obj.get("workflowRunSummary") is not None
+                    else None
+                ),
+                "additionalMetadata": obj.get("additionalMetadata"),
+            }
+        )
+        return _obj