hatchet-sdk 0.47.0__py3-none-any.whl → 1.0.0__py3-none-any.whl

hatchet_sdk/clients/rest/models/v1_workflow_run.py

@@ -0,0 +1,171 @@
+# 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 datetime import datetime
+from typing import Any, ClassVar, Dict, List, Optional, Set
+
+from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr
+from typing_extensions import Annotated, Self
+
+from hatchet_sdk.clients.rest.models.api_resource_meta import APIResourceMeta
+from hatchet_sdk.clients.rest.models.v1_task_status import V1TaskStatus
+
+
+class V1WorkflowRun(BaseModel):
+    """
+    V1WorkflowRun
+    """  # noqa: E501
+
+    metadata: APIResourceMeta
+    status: V1TaskStatus
+    started_at: Optional[datetime] = Field(
+        default=None,
+        description="The timestamp the task run started.",
+        alias="startedAt",
+    )
+    finished_at: Optional[datetime] = Field(
+        default=None,
+        description="The timestamp the task run finished.",
+        alias="finishedAt",
+    )
+    duration: Optional[StrictInt] = Field(
+        default=None, description="The duration of the task run, in milliseconds."
+    )
+    tenant_id: Annotated[str, Field(min_length=36, strict=True, max_length=36)] = Field(
+        description="The ID of the tenant.", alias="tenantId"
+    )
+    additional_metadata: Optional[Dict[str, Any]] = Field(
+        default=None,
+        description="Additional metadata for the task run.",
+        alias="additionalMetadata",
+    )
+    display_name: StrictStr = Field(
+        description="The display name of the task run.", alias="displayName"
+    )
+    workflow_id: StrictStr = Field(alias="workflowId")
+    output: Dict[str, Any] = Field(
+        description="The output of the task run (for the latest run)"
+    )
+    error_message: Optional[StrictStr] = Field(
+        default=None,
+        description="The error message of the task run (for the latest run)",
+        alias="errorMessage",
+    )
+    workflow_version_id: Optional[StrictStr] = Field(
+        default=None,
+        description="The ID of the workflow version.",
+        alias="workflowVersionId",
+    )
+    input: Dict[str, Any] = Field(description="The input of the task run.")
+    created_at: Optional[datetime] = Field(
+        default=None,
+        description="The timestamp the task run was created.",
+        alias="createdAt",
+    )
+    __properties: ClassVar[List[str]] = [
+        "metadata",
+        "status",
+        "startedAt",
+        "finishedAt",
+        "duration",
+        "tenantId",
+        "additionalMetadata",
+        "displayName",
+        "workflowId",
+        "output",
+        "errorMessage",
+        "workflowVersionId",
+        "input",
+        "createdAt",
+    ]
+
+    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 V1WorkflowRun 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 V1WorkflowRun 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
+                ),
+                "status": obj.get("status"),
+                "startedAt": obj.get("startedAt"),
+                "finishedAt": obj.get("finishedAt"),
+                "duration": obj.get("duration"),
+                "tenantId": obj.get("tenantId"),
+                "additionalMetadata": obj.get("additionalMetadata"),
+                "displayName": obj.get("displayName"),
+                "workflowId": obj.get("workflowId"),
+                "output": obj.get("output"),
+                "errorMessage": obj.get("errorMessage"),
+                "workflowVersionId": obj.get("workflowVersionId"),
+                "input": obj.get("input"),
+                "createdAt": obj.get("createdAt"),
+            }
+        )
+        return _obj

hatchet_sdk/clients/rest/models/v1_workflow_run_details.py

@@ -0,0 +1,145 @@
+# 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
+from typing_extensions import Self
+
+from hatchet_sdk.clients.rest.models.v1_task_event import V1TaskEvent
+from hatchet_sdk.clients.rest.models.v1_task_summary import V1TaskSummary
+from hatchet_sdk.clients.rest.models.v1_workflow_run import V1WorkflowRun
+from hatchet_sdk.clients.rest.models.workflow_run_shape_item_for_workflow_run_details import (
+    WorkflowRunShapeItemForWorkflowRunDetails,
+)
+
+
+class V1WorkflowRunDetails(BaseModel):
+    """
+    V1WorkflowRunDetails
+    """  # noqa: E501
+
+    run: V1WorkflowRun
+    task_events: List[V1TaskEvent] = Field(
+        description="The list of task events for the workflow run", alias="taskEvents"
+    )
+    shape: List[WorkflowRunShapeItemForWorkflowRunDetails]
+    tasks: List[V1TaskSummary]
+    __properties: ClassVar[List[str]] = ["run", "taskEvents", "shape", "tasks"]
+
+    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 V1WorkflowRunDetails 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 run
+        if self.run:
+            _dict["run"] = self.run.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in task_events (list)
+        _items = []
+        if self.task_events:
+            for _item_task_events in self.task_events:
+                if _item_task_events:
+                    _items.append(_item_task_events.to_dict())
+            _dict["taskEvents"] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in shape (list)
+        _items = []
+        if self.shape:
+            for _item_shape in self.shape:
+                if _item_shape:
+                    _items.append(_item_shape.to_dict())
+            _dict["shape"] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in tasks (list)
+        _items = []
+        if self.tasks:
+            for _item_tasks in self.tasks:
+                if _item_tasks:
+                    _items.append(_item_tasks.to_dict())
+            _dict["tasks"] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of V1WorkflowRunDetails from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate(
+            {
+                "run": (
+                    V1WorkflowRun.from_dict(obj["run"])
+                    if obj.get("run") is not None
+                    else None
+                ),
+                "taskEvents": (
+                    [V1TaskEvent.from_dict(_item) for _item in obj["taskEvents"]]
+                    if obj.get("taskEvents") is not None
+                    else None
+                ),
+                "shape": (
+                    [
+                        WorkflowRunShapeItemForWorkflowRunDetails.from_dict(_item)
+                        for _item in obj["shape"]
+                    ]
+                    if obj.get("shape") is not None
+                    else None
+                ),
+                "tasks": (
+                    [V1TaskSummary.from_dict(_item) for _item in obj["tasks"]]
+                    if obj.get("tasks") is not None
+                    else None
+                ),
+            }
+        )
+        return _obj

hatchet_sdk/clients/rest/models/v1_workflow_type.py

@@ -0,0 +1,37 @@
+# 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
+from enum import Enum
+
+from typing_extensions import Self
+
+
+class V1WorkflowType(str, Enum):
+    """
+    V1WorkflowType
+    """
+
+    """
+    allowed enum values
+    """
+    DAG = "DAG"
+    TASK = "TASK"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of V1WorkflowType from a JSON string"""
+        return cls(json.loads(json_str))

hatchet_sdk/clients/rest/models/workflow_run_shape_item_for_workflow_run_details.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 WorkflowRunShapeItemForWorkflowRunDetails(BaseModel):
+    """
+    WorkflowRunShapeItemForWorkflowRunDetails
+    """  # noqa: E501
+
+    task_external_id: Annotated[
+        str, Field(min_length=36, strict=True, max_length=36)
+    ] = Field(alias="taskExternalId")
+    children_external_ids: List[
+        Annotated[str, Field(min_length=36, strict=True, max_length=36)]
+    ] = Field(alias="childrenExternalIds")
+    task_name: StrictStr = Field(alias="taskName")
+    __properties: ClassVar[List[str]] = [
+        "taskExternalId",
+        "childrenExternalIds",
+        "taskName",
+    ]
+
+    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 WorkflowRunShapeItemForWorkflowRunDetails 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 WorkflowRunShapeItemForWorkflowRunDetails from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate(
+            {
+                "taskExternalId": obj.get("taskExternalId"),
+                "childrenExternalIds": obj.get("childrenExternalIds"),
+                "taskName": obj.get("taskName"),
+            }
+        )
+        return _obj

hatchet_sdk/clients/rest/rest.py

@@ -55,40 +55,32 @@ class RESTClientObject:
     def __init__(self, configuration) -> None:
 
         # maxsize is number of requests to host that are allowed in parallel
-        maxsize = configuration.connection_pool_maxsize
+        self.maxsize = configuration.connection_pool_maxsize
 
-        ssl_context = ssl.create_default_context(cafile=configuration.ssl_ca_cert)
+        self.ssl_context = ssl.create_default_context(
+            cafile=configuration.ssl_ca_cert,
+            cadata=configuration.ca_cert_data,
+        )
         if configuration.cert_file:
-            ssl_context.load_cert_chain(
+            self.ssl_context.load_cert_chain(
                 configuration.cert_file, keyfile=configuration.key_file
             )
 
         if not configuration.verify_ssl:
-            ssl_context.check_hostname = False
-            ssl_context.verify_mode = ssl.CERT_NONE
-
-        connector = aiohttp.TCPConnector(limit=maxsize, ssl=ssl_context)
+            self.ssl_context.check_hostname = False
+            self.ssl_context.verify_mode = ssl.CERT_NONE
 
         self.proxy = configuration.proxy
         self.proxy_headers = configuration.proxy_headers
 
-        # https pool manager
-        self.pool_manager = aiohttp.ClientSession(connector=connector, trust_env=True)
-
-        retries = configuration.retries
-        self.retry_client: Optional[aiohttp_retry.RetryClient]
-        if retries is not None:
-            self.retry_client = aiohttp_retry.RetryClient(
-                client_session=self.pool_manager,
-                retry_options=aiohttp_retry.ExponentialRetry(
-                    attempts=retries, factor=2.0, start_timeout=0.1, max_timeout=120.0
-                ),
-            )
-        else:
-            self.retry_client = None
+        self.retries = configuration.retries
+
+        self.pool_manager: Optional[aiohttp.ClientSession] = None
+        self.retry_client: Optional[aiohttp_retry.RetryClient] = None
 
-    async def close(self):
-        await self.pool_manager.close()
+    async def close(self) -> None:
+        if self.pool_manager:
+            await self.pool_manager.close()
         if self.retry_client is not None:
             await self.retry_client.close()
 
@@ -177,10 +169,29 @@ class RESTClientObject:
                 raise ApiException(status=0, reason=msg)
 
         pool_manager: Union[aiohttp.ClientSession, aiohttp_retry.RetryClient]
-        if self.retry_client is not None and method in ALLOW_RETRY_METHODS:
+
+        # https pool manager
+        if self.pool_manager is None:
+            self.pool_manager = aiohttp.ClientSession(
+                connector=aiohttp.TCPConnector(
+                    limit=self.maxsize, ssl=self.ssl_context
+                ),
+                trust_env=True,
+            )
+        pool_manager = self.pool_manager
+
+        if self.retries is not None and method in ALLOW_RETRY_METHODS:
+            if self.retry_client is None:
+                self.retry_client = aiohttp_retry.RetryClient(
+                    client_session=self.pool_manager,
+                    retry_options=aiohttp_retry.ExponentialRetry(
+                        attempts=self.retries,
+                        factor=2.0,
+                        start_timeout=0.1,
+                        max_timeout=120.0,
+                    ),
+                )
             pool_manager = self.retry_client
-        else:
-            pool_manager = self.pool_manager
 
         r = await pool_manager.request(**args)
 

hatchet_sdk/clients/rest/tenacity_utils.py

@@ -27,7 +27,7 @@ def tenacity_alert_retry(retry_state: tenacity.RetryCallState) -> None:
     )
 
 
-def tenacity_should_retry(ex: Exception) -> bool:
+def tenacity_should_retry(ex: BaseException) -> bool:
     if isinstance(ex, (grpc.aio.AioRpcError, grpc.RpcError)):
         if ex.code() in [
             grpc.StatusCode.UNIMPLEMENTED,