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

hatchet_sdk/clients/rest/models/v1_event_list.py

@@ -0,0 +1,110 @@
+# 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
+from typing_extensions import Self
+
+from hatchet_sdk.clients.rest.models.pagination_response import PaginationResponse
+from hatchet_sdk.clients.rest.models.v1_event import V1Event
+
+
+class V1EventList(BaseModel):
+    """
+    V1EventList
+    """  # noqa: E501
+
+    pagination: Optional[PaginationResponse] = None
+    rows: Optional[List[V1Event]] = None
+    __properties: ClassVar[List[str]] = ["pagination", "rows"]
+
+    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 V1EventList 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 pagination
+        if self.pagination:
+            _dict["pagination"] = self.pagination.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in rows (list)
+        _items = []
+        if self.rows:
+            for _item_rows in self.rows:
+                if _item_rows:
+                    _items.append(_item_rows.to_dict())
+            _dict["rows"] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of V1EventList from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate(
+            {
+                "pagination": (
+                    PaginationResponse.from_dict(obj["pagination"])
+                    if obj.get("pagination") is not None
+                    else None
+                ),
+                "rows": (
+                    [V1Event.from_dict(_item) for _item in obj["rows"]]
+                    if obj.get("rows") is not None
+                    else None
+                ),
+            }
+        )
+        return _obj

hatchet_sdk/clients/rest/models/v1_event_workflow_run_summary.py

@@ -0,0 +1,101 @@
+# 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, StrictInt
+from typing_extensions import Self
+
+
+class V1EventWorkflowRunSummary(BaseModel):
+    """
+    V1EventWorkflowRunSummary
+    """  # noqa: E501
+
+    running: StrictInt = Field(description="The number of running runs.")
+    queued: StrictInt = Field(description="The number of queued runs.")
+    succeeded: StrictInt = Field(description="The number of succeeded runs.")
+    failed: StrictInt = Field(description="The number of failed runs.")
+    cancelled: StrictInt = Field(description="The number of cancelled runs.")
+    __properties: ClassVar[List[str]] = [
+        "running",
+        "queued",
+        "succeeded",
+        "failed",
+        "cancelled",
+    ]
+
+    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 V1EventWorkflowRunSummary 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 V1EventWorkflowRunSummary from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate(
+            {
+                "running": obj.get("running"),
+                "queued": obj.get("queued"),
+                "succeeded": obj.get("succeeded"),
+                "failed": obj.get("failed"),
+                "cancelled": obj.get("cancelled"),
+            }
+        )
+        return _obj

hatchet_sdk/clients/rest/models/v1_filter.py

@@ -0,0 +1,127 @@
+# 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
+
+from hatchet_sdk.clients.rest.models.api_resource_meta import APIResourceMeta
+
+
+class V1Filter(BaseModel):
+    """
+    V1Filter
+    """  # noqa: E501
+
+    metadata: APIResourceMeta
+    tenant_id: StrictStr = Field(
+        description="The ID of the tenant associated with this filter.",
+        alias="tenantId",
+    )
+    workflow_id: Annotated[str, Field(min_length=36, strict=True, max_length=36)] = (
+        Field(
+            description="The workflow id associated with this filter.",
+            alias="workflowId",
+        )
+    )
+    scope: StrictStr = Field(
+        description="The scope associated with this filter. Used for subsetting candidate filters at evaluation time"
+    )
+    expression: StrictStr = Field(
+        description="The expression associated with this filter."
+    )
+    payload: Dict[str, Any] = Field(
+        description="Additional payload data associated with the filter"
+    )
+    __properties: ClassVar[List[str]] = [
+        "metadata",
+        "tenantId",
+        "workflowId",
+        "scope",
+        "expression",
+        "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 V1Filter 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()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of V1Filter 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
+                ),
+                "tenantId": obj.get("tenantId"),
+                "workflowId": obj.get("workflowId"),
+                "scope": obj.get("scope"),
+                "expression": obj.get("expression"),
+                "payload": obj.get("payload"),
+            }
+        )
+        return _obj

hatchet_sdk/clients/rest/models/v1_filter_list.py

@@ -0,0 +1,110 @@
+# 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
+from typing_extensions import Self
+
+from hatchet_sdk.clients.rest.models.pagination_response import PaginationResponse
+from hatchet_sdk.clients.rest.models.v1_filter import V1Filter
+
+
+class V1FilterList(BaseModel):
+    """
+    V1FilterList
+    """  # noqa: E501
+
+    pagination: Optional[PaginationResponse] = None
+    rows: Optional[List[V1Filter]] = None
+    __properties: ClassVar[List[str]] = ["pagination", "rows"]
+
+    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 V1FilterList 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 pagination
+        if self.pagination:
+            _dict["pagination"] = self.pagination.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in rows (list)
+        _items = []
+        if self.rows:
+            for _item_rows in self.rows:
+                if _item_rows:
+                    _items.append(_item_rows.to_dict())
+            _dict["rows"] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of V1FilterList from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate(
+            {
+                "pagination": (
+                    PaginationResponse.from_dict(obj["pagination"])
+                    if obj.get("pagination") is not None
+                    else None
+                ),
+                "rows": (
+                    [V1Filter.from_dict(_item) for _item in obj["rows"]]
+                    if obj.get("rows") is not None
+                    else None
+                ),
+            }
+        )
+        return _obj

hatchet_sdk/clients/rest/models/v1_log_line.py

@@ -20,9 +20,11 @@ import re  # noqa: F401
 from datetime import datetime
 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
 
+from hatchet_sdk.clients.rest.models.v1_log_line_level import V1LogLineLevel
+
 
 class V1LogLine(BaseModel):
     """
@@ -34,7 +36,21 @@ class V1LogLine(BaseModel):
     )
     message: StrictStr = Field(description="The log message.")
     metadata: Dict[str, Any] = Field(description="The log metadata.")
-    __properties: ClassVar[List[str]] = ["createdAt", "message", "metadata"]
+    retry_count: Optional[StrictInt] = Field(
+        default=None, description="The retry count of the log line.", alias="retryCount"
+    )
+    attempt: Optional[StrictInt] = Field(
+        default=None, description="The attempt number of the log line."
+    )
+    level: Optional[V1LogLineLevel] = Field(default=None, description="The log level.")
+    __properties: ClassVar[List[str]] = [
+        "createdAt",
+        "message",
+        "metadata",
+        "retryCount",
+        "attempt",
+        "level",
+    ]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -89,6 +105,9 @@ class V1LogLine(BaseModel):
                 "createdAt": obj.get("createdAt"),
                 "message": obj.get("message"),
                 "metadata": obj.get("metadata"),
+                "retryCount": obj.get("retryCount"),
+                "attempt": obj.get("attempt"),
+                "level": obj.get("level"),
             }
         )
         return _obj

hatchet_sdk/clients/rest/models/v1_task_event.py

@@ -42,6 +42,14 @@ class V1TaskEvent(BaseModel):
     task_display_name: Optional[StrictStr] = Field(
         default=None, alias="taskDisplayName"
     )
+    retry_count: Optional[StrictInt] = Field(
+        default=None,
+        description="The number of retries of the task.",
+        alias="retryCount",
+    )
+    attempt: Optional[StrictInt] = Field(
+        default=None, description="The attempt number of the task."
+    )
     __properties: ClassVar[List[str]] = [
         "id",
         "taskId",
@@ -52,6 +60,8 @@ class V1TaskEvent(BaseModel):
         "output",
         "workerId",
         "taskDisplayName",
+        "retryCount",
+        "attempt",
     ]
 
     model_config = ConfigDict(
@@ -113,6 +123,8 @@ class V1TaskEvent(BaseModel):
                 "output": obj.get("output"),
                 "workerId": obj.get("workerId"),
                 "taskDisplayName": obj.get("taskDisplayName"),
+                "retryCount": obj.get("retryCount"),
+                "attempt": obj.get("attempt"),
             }
         )
         return _obj

hatchet_sdk/clients/rest/models/v1_task_summary.py

@@ -37,6 +37,14 @@ class V1TaskSummary(BaseModel):
     action_id: Optional[StrictStr] = Field(
         default=None, description="The action ID of the task.", alias="actionId"
     )
+    retry_count: Optional[StrictInt] = Field(
+        default=None,
+        description="The number of retries of the task.",
+        alias="retryCount",
+    )
+    attempt: Optional[StrictInt] = Field(
+        default=None, description="The attempt number of the task."
+    )
     additional_metadata: Optional[Dict[str, Any]] = Field(
         default=None,
         description="Additional metadata for the task run.",
@@ -106,6 +114,8 @@ class V1TaskSummary(BaseModel):
     __properties: ClassVar[List[str]] = [
         "metadata",
         "actionId",
+        "retryCount",
+        "attempt",
         "additionalMetadata",
         "children",
         "createdAt",
@@ -196,6 +206,8 @@ class V1TaskSummary(BaseModel):
                     else None
                 ),
                 "actionId": obj.get("actionId"),
+                "retryCount": obj.get("retryCount"),
+                "attempt": obj.get("attempt"),
                 "additionalMetadata": obj.get("additionalMetadata"),
                 "children": (
                     [V1TaskSummary.from_dict(_item) for _item in obj["children"]]

hatchet_sdk/clients/rest/models/v1_task_timing.py

@@ -70,6 +70,19 @@ class V1TaskTiming(BaseModel):
         description="The timestamp the task run finished.",
         alias="finishedAt",
     )
+    workflow_run_id: Optional[StrictStr] = Field(
+        default=None,
+        description="The external ID of the workflow run.",
+        alias="workflowRunId",
+    )
+    retry_count: Optional[StrictInt] = Field(
+        default=None,
+        description="The number of retries of the task.",
+        alias="retryCount",
+    )
+    attempt: Optional[StrictInt] = Field(
+        default=None, description="The attempt number of the task."
+    )
     __properties: ClassVar[List[str]] = [
         "metadata",
         "depth",
@@ -83,6 +96,9 @@ class V1TaskTiming(BaseModel):
         "queuedAt",
         "startedAt",
         "finishedAt",
+        "workflowRunId",
+        "retryCount",
+        "attempt",
     ]
 
     model_config = ConfigDict(
@@ -154,6 +170,9 @@ class V1TaskTiming(BaseModel):
                 "queuedAt": obj.get("queuedAt"),
                 "startedAt": obj.get("startedAt"),
                 "finishedAt": obj.get("finishedAt"),
+                "workflowRunId": obj.get("workflowRunId"),
+                "retryCount": obj.get("retryCount"),
+                "attempt": obj.get("attempt"),
             }
         )
         return _obj

hatchet_sdk/clients/rest/models/workflow.py

@@ -47,6 +47,9 @@ class Workflow(BaseModel):
     jobs: Optional[List[Job]] = Field(
         default=None, description="The jobs of the workflow."
     )
+    tenant_id: StrictStr = Field(
+        description="The tenant id of the workflow.", alias="tenantId"
+    )
     __properties: ClassVar[List[str]] = [
         "metadata",
         "name",
@@ -55,6 +58,7 @@ class Workflow(BaseModel):
         "versions",
         "tags",
         "jobs",
+        "tenantId",
     ]
 
     model_config = ConfigDict(
@@ -154,6 +158,7 @@ class Workflow(BaseModel):
                     if obj.get("jobs") is not None
                     else None
                 ),
+                "tenantId": obj.get("tenantId"),
             }
         )
         return _obj

hatchet_sdk/config.py

@@ -1,5 +1,6 @@
 import json
 from logging import Logger, getLogger
+from typing import overload
 
 from pydantic import Field, field_validator, model_validator
 from pydantic_settings import BaseSettings, SettingsConfigDict
@@ -121,9 +122,37 @@ class ClientConfig(BaseSettings):
     def validate_namespace(cls, namespace: str) -> str:
         if not namespace:
             return ""
+
         if not namespace.endswith("_"):
             namespace = f"{namespace}_"
+
         return namespace.lower()
 
     def __hash__(self) -> int:
         return hash(json.dumps(self.model_dump(), default=str))
+
+    @overload
+    def apply_namespace(
+        self, resource_name: str, namespace_override: str | None = None
+    ) -> str: ...
+
+    @overload
+    def apply_namespace(
+        self, resource_name: None, namespace_override: str | None = None
+    ) -> None: ...
+
+    def apply_namespace(
+        self, resource_name: str | None, namespace_override: str | None = None
+    ) -> str | None:
+        if resource_name is None:
+            return None
+
+        namespace = namespace_override or self.namespace
+
+        if not namespace:
+            return resource_name
+
+        if resource_name.startswith(namespace):
+            return resource_name
+
+        return namespace + resource_name

hatchet_sdk/context/context.py

@@ -59,6 +59,7 @@ class Context:
         self.stream_event_thread_pool = ThreadPoolExecutor(max_workers=1)
 
         self.input = self.data.input
+        self.filter_payload = self.data.filter_payload
 
         self._lifespan_context = lifespan_context
 
@@ -222,6 +223,14 @@ class Context:
     def priority(self) -> int | None:
         return self.action.priority
 
+    @property
+    def workflow_id(self) -> str | None:
+        return self.action.workflow_id
+
+    @property
+    def workflow_version_id(self) -> str | None:
+        return self.action.workflow_version_id
+
     @property
     def task_run_errors(self) -> dict[str, str]:
         errors = self.data.step_run_errors