hatchet-sdk 0.42.1__tar.gz → 0.42.3__tar.gz

{hatchet_sdk-0.42.1 → hatchet_sdk-0.42.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hatchet-sdk
-Version: 0.42.1
+Version: 0.42.3
 Summary: 
 Author: Alexander Belanger
 Author-email: alexander@hatchet.run
@@ -12,6 +12,7 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: aiohttp (>=3.10.5,<4.0.0)
 Requires-Dist: aiohttp-retry (>=2.8.3,<3.0.0)
 Requires-Dist: aiostream (>=0.5.2,<0.6.0)
+Requires-Dist: cel-python (>=0.1.5,<0.2.0)
 Requires-Dist: grpcio (>=1.64.1,<2.0.0)
 Requires-Dist: grpcio-tools (>=1.60.0,<2.0.0)
 Requires-Dist: loguru (>=0.7.2,<0.8.0)

{hatchet_sdk-0.42.1 → hatchet_sdk-0.42.3}/hatchet_sdk/clients/rest/__init__.py

@@ -60,10 +60,10 @@ from hatchet_sdk.clients.rest.models.api_token import APIToken
 from hatchet_sdk.clients.rest.models.bulk_create_event_request import (
     BulkCreateEventRequest,
 )
-from hatchet_sdk.clients.rest.models.bulk_create_event_response import (
-    BulkCreateEventResponse,
-)
 from hatchet_sdk.clients.rest.models.cancel_event_request import CancelEventRequest
+from hatchet_sdk.clients.rest.models.concurrency_limit_strategy import (
+    ConcurrencyLimitStrategy,
+)
 from hatchet_sdk.clients.rest.models.create_api_token_request import (
     CreateAPITokenRequest,
 )
@@ -89,6 +89,7 @@ from hatchet_sdk.clients.rest.models.create_tenant_invite_request import (
 from hatchet_sdk.clients.rest.models.create_tenant_request import CreateTenantRequest
 from hatchet_sdk.clients.rest.models.cron_workflows import CronWorkflows
 from hatchet_sdk.clients.rest.models.cron_workflows_list import CronWorkflowsList
+from hatchet_sdk.clients.rest.models.cron_workflows_method import CronWorkflowsMethod
 from hatchet_sdk.clients.rest.models.cron_workflows_order_by_field import (
     CronWorkflowsOrderByField,
 )
@@ -106,6 +107,7 @@ from hatchet_sdk.clients.rest.models.event_update_cancel200_response import (
 from hatchet_sdk.clients.rest.models.event_workflow_run_summary import (
     EventWorkflowRunSummary,
 )
+from hatchet_sdk.clients.rest.models.events import Events
 from hatchet_sdk.clients.rest.models.get_step_run_diff_response import (
     GetStepRunDiffResponse,
 )
@@ -157,6 +159,9 @@ from hatchet_sdk.clients.rest.models.scheduled_workflows import ScheduledWorkflo
 from hatchet_sdk.clients.rest.models.scheduled_workflows_list import (
     ScheduledWorkflowsList,
 )
+from hatchet_sdk.clients.rest.models.scheduled_workflows_method import (
+    ScheduledWorkflowsMethod,
+)
 from hatchet_sdk.clients.rest.models.scheduled_workflows_order_by_field import (
     ScheduledWorkflowsOrderByField,
 )
@@ -240,6 +245,7 @@ from hatchet_sdk.clients.rest.models.worker_label import WorkerLabel
 from hatchet_sdk.clients.rest.models.worker_list import WorkerList
 from hatchet_sdk.clients.rest.models.worker_runtime_info import WorkerRuntimeInfo
 from hatchet_sdk.clients.rest.models.worker_runtime_sdks import WorkerRuntimeSDKs
+from hatchet_sdk.clients.rest.models.worker_type import WorkerType
 from hatchet_sdk.clients.rest.models.workflow import Workflow
 from hatchet_sdk.clients.rest.models.workflow_concurrency import WorkflowConcurrency
 from hatchet_sdk.clients.rest.models.workflow_kind import WorkflowKind

{hatchet_sdk-0.42.1 → hatchet_sdk-0.42.3}/hatchet_sdk/clients/rest/api/event_api.py

@@ -22,9 +22,6 @@ from hatchet_sdk.clients.rest.api_response import ApiResponse
 from hatchet_sdk.clients.rest.models.bulk_create_event_request import (
     BulkCreateEventRequest,
 )
-from hatchet_sdk.clients.rest.models.bulk_create_event_response import (
-    BulkCreateEventResponse,
-)
 from hatchet_sdk.clients.rest.models.cancel_event_request import CancelEventRequest
 from hatchet_sdk.clients.rest.models.create_event_request import CreateEventRequest
 from hatchet_sdk.clients.rest.models.event import Event
@@ -38,6 +35,7 @@ from hatchet_sdk.clients.rest.models.event_order_by_field import EventOrderByFie
 from hatchet_sdk.clients.rest.models.event_update_cancel200_response import (
     EventUpdateCancel200Response,
 )
+from hatchet_sdk.clients.rest.models.events import Events
 from hatchet_sdk.clients.rest.models.replay_event_request import ReplayEventRequest
 from hatchet_sdk.clients.rest.models.workflow_run_status import WorkflowRunStatus
 from hatchet_sdk.clients.rest.rest import RESTResponseType
@@ -375,7 +373,7 @@ class EventApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> BulkCreateEventResponse:
+    ) -> Events:
         """Bulk Create events
 
         Bulk creates new events.
@@ -416,7 +414,7 @@ class EventApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": "BulkCreateEventResponse",
+            "200": "Events",
             "400": "APIErrors",
             "403": "APIErrors",
             "429": "APIErrors",
@@ -453,7 +451,7 @@ class EventApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[BulkCreateEventResponse]:
+    ) -> ApiResponse[Events]:
         """Bulk Create events
 
         Bulk creates new events.
@@ -494,7 +492,7 @@ class EventApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": "BulkCreateEventResponse",
+            "200": "Events",
             "400": "APIErrors",
             "403": "APIErrors",
             "429": "APIErrors",
@@ -572,7 +570,7 @@ class EventApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": "BulkCreateEventResponse",
+            "200": "Events",
             "400": "APIErrors",
             "403": "APIErrors",
             "429": "APIErrors",

{hatchet_sdk-0.42.1 → hatchet_sdk-0.42.3}/hatchet_sdk/clients/rest/api/workflow_api.py

@@ -2540,6 +2540,15 @@ class WorkflowApi:
                 min_length=36, strict=True, max_length=36, description="The tenant id"
             ),
         ],
+        offset: Annotated[
+            Optional[StrictInt], Field(description="The number to skip")
+        ] = None,
+        limit: Annotated[
+            Optional[StrictInt], Field(description="The number to limit by")
+        ] = None,
+        name: Annotated[
+            Optional[StrictStr], Field(description="Search by name")
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -2558,6 +2567,12 @@ class WorkflowApi:
 
         :param tenant: The tenant id (required)
         :type tenant: str
+        :param offset: The number to skip
+        :type offset: int
+        :param limit: The number to limit by
+        :type limit: int
+        :param name: Search by name
+        :type name: 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
@@ -2582,6 +2597,9 @@ class WorkflowApi:
 
         _param = self._workflow_list_serialize(
             tenant=tenant,
+            offset=offset,
+            limit=limit,
+            name=name,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -2611,6 +2629,15 @@ class WorkflowApi:
                 min_length=36, strict=True, max_length=36, description="The tenant id"
             ),
         ],
+        offset: Annotated[
+            Optional[StrictInt], Field(description="The number to skip")
+        ] = None,
+        limit: Annotated[
+            Optional[StrictInt], Field(description="The number to limit by")
+        ] = None,
+        name: Annotated[
+            Optional[StrictStr], Field(description="Search by name")
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -2629,6 +2656,12 @@ class WorkflowApi:
 
         :param tenant: The tenant id (required)
         :type tenant: str
+        :param offset: The number to skip
+        :type offset: int
+        :param limit: The number to limit by
+        :type limit: int
+        :param name: Search by name
+        :type name: 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
@@ -2653,6 +2686,9 @@ class WorkflowApi:
 
         _param = self._workflow_list_serialize(
             tenant=tenant,
+            offset=offset,
+            limit=limit,
+            name=name,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -2682,6 +2718,15 @@ class WorkflowApi:
                 min_length=36, strict=True, max_length=36, description="The tenant id"
             ),
         ],
+        offset: Annotated[
+            Optional[StrictInt], Field(description="The number to skip")
+        ] = None,
+        limit: Annotated[
+            Optional[StrictInt], Field(description="The number to limit by")
+        ] = None,
+        name: Annotated[
+            Optional[StrictStr], Field(description="Search by name")
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -2700,6 +2745,12 @@ class WorkflowApi:
 
         :param tenant: The tenant id (required)
         :type tenant: str
+        :param offset: The number to skip
+        :type offset: int
+        :param limit: The number to limit by
+        :type limit: int
+        :param name: Search by name
+        :type name: 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
@@ -2724,6 +2775,9 @@ class WorkflowApi:
 
         _param = self._workflow_list_serialize(
             tenant=tenant,
+            offset=offset,
+            limit=limit,
+            name=name,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -2743,6 +2797,9 @@ class WorkflowApi:
     def _workflow_list_serialize(
         self,
         tenant,
+        offset,
+        limit,
+        name,
         _request_auth,
         _content_type,
         _headers,
@@ -2766,6 +2823,18 @@ class WorkflowApi:
         if tenant is not None:
             _path_params["tenant"] = tenant
         # process the query parameters
+        if offset is not None:
+
+            _query_params.append(("offset", offset))
+
+        if limit is not None:
+
+            _query_params.append(("limit", limit))
+
+        if name is not None:
+
+            _query_params.append(("name", name))
+
         # process the header parameters
         # process the form parameters
         # process the body parameter

{hatchet_sdk-0.42.1 → hatchet_sdk-0.42.3}/hatchet_sdk/clients/rest/models/__init__.py

@@ -27,10 +27,10 @@ from hatchet_sdk.clients.rest.models.api_token import APIToken
 from hatchet_sdk.clients.rest.models.bulk_create_event_request import (
     BulkCreateEventRequest,
 )
-from hatchet_sdk.clients.rest.models.bulk_create_event_response import (
-    BulkCreateEventResponse,
-)
 from hatchet_sdk.clients.rest.models.cancel_event_request import CancelEventRequest
+from hatchet_sdk.clients.rest.models.concurrency_limit_strategy import (
+    ConcurrencyLimitStrategy,
+)
 from hatchet_sdk.clients.rest.models.create_api_token_request import (
     CreateAPITokenRequest,
 )
@@ -56,6 +56,7 @@ from hatchet_sdk.clients.rest.models.create_tenant_invite_request import (
 from hatchet_sdk.clients.rest.models.create_tenant_request import CreateTenantRequest
 from hatchet_sdk.clients.rest.models.cron_workflows import CronWorkflows
 from hatchet_sdk.clients.rest.models.cron_workflows_list import CronWorkflowsList
+from hatchet_sdk.clients.rest.models.cron_workflows_method import CronWorkflowsMethod
 from hatchet_sdk.clients.rest.models.cron_workflows_order_by_field import (
     CronWorkflowsOrderByField,
 )
@@ -73,6 +74,7 @@ from hatchet_sdk.clients.rest.models.event_update_cancel200_response import (
 from hatchet_sdk.clients.rest.models.event_workflow_run_summary import (
     EventWorkflowRunSummary,
 )
+from hatchet_sdk.clients.rest.models.events import Events
 from hatchet_sdk.clients.rest.models.get_step_run_diff_response import (
     GetStepRunDiffResponse,
 )
@@ -124,6 +126,9 @@ from hatchet_sdk.clients.rest.models.scheduled_workflows import ScheduledWorkflo
 from hatchet_sdk.clients.rest.models.scheduled_workflows_list import (
     ScheduledWorkflowsList,
 )
+from hatchet_sdk.clients.rest.models.scheduled_workflows_method import (
+    ScheduledWorkflowsMethod,
+)
 from hatchet_sdk.clients.rest.models.scheduled_workflows_order_by_field import (
     ScheduledWorkflowsOrderByField,
 )
@@ -207,6 +212,7 @@ from hatchet_sdk.clients.rest.models.worker_label import WorkerLabel
 from hatchet_sdk.clients.rest.models.worker_list import WorkerList
 from hatchet_sdk.clients.rest.models.worker_runtime_info import WorkerRuntimeInfo
 from hatchet_sdk.clients.rest.models.worker_runtime_sdks import WorkerRuntimeSDKs
+from hatchet_sdk.clients.rest.models.worker_type import WorkerType
 from hatchet_sdk.clients.rest.models.workflow import Workflow
 from hatchet_sdk.clients.rest.models.workflow_concurrency import WorkflowConcurrency
 from hatchet_sdk.clients.rest.models.workflow_kind import WorkflowKind

hatchet_sdk-0.42.3/hatchet_sdk/clients/rest/models/concurrency_limit_strategy.py

@@ -0,0 +1,39 @@
+# 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 ConcurrencyLimitStrategy(str, Enum):
+    """
+    ConcurrencyLimitStrategy
+    """
+
+    """
+    allowed enum values
+    """
+    CANCEL_IN_PROGRESS = "CANCEL_IN_PROGRESS"
+    DROP_NEWEST = "DROP_NEWEST"
+    QUEUE_NEWEST = "QUEUE_NEWEST"
+    GROUP_ROUND_ROBIN = "GROUP_ROUND_ROBIN"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of ConcurrencyLimitStrategy from a JSON string"""
+        return cls(json.loads(json_str))

{hatchet_sdk-0.42.1 → hatchet_sdk-0.42.3}/hatchet_sdk/clients/rest/models/cron_workflows.py

@@ -19,17 +19,11 @@ import pprint
 import re  # noqa: F401
 from typing import Any, ClassVar, Dict, List, Optional, Set
 
-from pydantic import (
-    BaseModel,
-    ConfigDict,
-    Field,
-    StrictBool,
-    StrictStr,
-    field_validator,
-)
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
 from typing_extensions import Self
 
 from hatchet_sdk.clients.rest.models.api_resource_meta import APIResourceMeta
+from hatchet_sdk.clients.rest.models.cron_workflows_method import CronWorkflowsMethod
 
 
 class CronWorkflows(BaseModel):
@@ -49,7 +43,7 @@ class CronWorkflows(BaseModel):
         default=None, alias="additionalMetadata"
     )
     enabled: StrictBool
-    method: StrictStr
+    method: CronWorkflowsMethod
     __properties: ClassVar[List[str]] = [
         "metadata",
         "tenantId",
@@ -64,13 +58,6 @@ class CronWorkflows(BaseModel):
         "method",
     ]
 
-    @field_validator("method")
-    def method_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in set(["DEFAULT", "API"]):
-            raise ValueError("must be one of enum values ('DEFAULT', 'API')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

hatchet_sdk-0.42.3/hatchet_sdk/clients/rest/models/cron_workflows_method.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 CronWorkflowsMethod(str, Enum):
+    """
+    CronWorkflowsMethod
+    """
+
+    """
+    allowed enum values
+    """
+    DEFAULT = "DEFAULT"
+    API = "API"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of CronWorkflowsMethod from a JSON string"""
+        return cls(json.loads(json_str))

hatchet_sdk-0.42.3/hatchet_sdk/clients/rest/models/events.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, Field
+from typing_extensions import Self
+
+from hatchet_sdk.clients.rest.models.api_resource_meta import APIResourceMeta
+from hatchet_sdk.clients.rest.models.event import Event
+
+
+class Events(BaseModel):
+    """
+    Events
+    """  # noqa: E501
+
+    metadata: APIResourceMeta
+    events: List[Event] = Field(description="The events.")
+    __properties: ClassVar[List[str]] = ["metadata", "events"]
+
+    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 Events 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 each item in events (list)
+        _items = []
+        if self.events:
+            for _item_events in self.events:
+                if _item_events:
+                    _items.append(_item_events.to_dict())
+            _dict["events"] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of Events 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
+                ),
+                "events": (
+                    [Event.from_dict(_item) for _item in obj["events"]]
+                    if obj.get("events") is not None
+                    else None
+                ),
+            }
+        )
+        return _obj

{hatchet_sdk-0.42.1 → hatchet_sdk-0.42.3}/hatchet_sdk/clients/rest/models/scheduled_workflows.py

@@ -20,10 +20,13 @@ import re  # noqa: F401
 from datetime import datetime
 from typing import Any, ClassVar, Dict, List, Optional, Set
 
-from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing_extensions import Annotated, Self
 
 from hatchet_sdk.clients.rest.models.api_resource_meta import APIResourceMeta
+from hatchet_sdk.clients.rest.models.scheduled_workflows_method import (
+    ScheduledWorkflowsMethod,
+)
 from hatchet_sdk.clients.rest.models.workflow_run_status import WorkflowRunStatus
 
 
@@ -54,7 +57,7 @@ class ScheduledWorkflows(BaseModel):
     workflow_run_id: Optional[
         Annotated[str, Field(min_length=36, strict=True, max_length=36)]
     ] = Field(default=None, alias="workflowRunId")
-    method: StrictStr
+    method: ScheduledWorkflowsMethod
     __properties: ClassVar[List[str]] = [
         "metadata",
         "tenantId",
@@ -71,13 +74,6 @@ class ScheduledWorkflows(BaseModel):
         "method",
     ]
 
-    @field_validator("method")
-    def method_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in set(["DEFAULT", "API"]):
-            raise ValueError("must be one of enum values ('DEFAULT', 'API')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

hatchet_sdk-0.42.3/hatchet_sdk/clients/rest/models/scheduled_workflows_method.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 ScheduledWorkflowsMethod(str, Enum):
+    """
+    ScheduledWorkflowsMethod
+    """
+
+    """
+    allowed enum values
+    """
+    DEFAULT = "DEFAULT"
+    API = "API"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of ScheduledWorkflowsMethod from a JSON string"""
+        return cls(json.loads(json_str))

{hatchet_sdk-0.42.1 → hatchet_sdk-0.42.3}/hatchet_sdk/clients/rest/models/worker.py

@@ -28,6 +28,7 @@ from hatchet_sdk.clients.rest.models.recent_step_runs import RecentStepRuns
 from hatchet_sdk.clients.rest.models.semaphore_slots import SemaphoreSlots
 from hatchet_sdk.clients.rest.models.worker_label import WorkerLabel
 from hatchet_sdk.clients.rest.models.worker_runtime_info import WorkerRuntimeInfo
+from hatchet_sdk.clients.rest.models.worker_type import WorkerType
 
 
 class Worker(BaseModel):
@@ -37,7 +38,7 @@ class Worker(BaseModel):
 
     metadata: APIResourceMeta
     name: StrictStr = Field(description="The name of the worker.")
-    type: StrictStr
+    type: WorkerType
     last_heartbeat_at: Optional[datetime] = Field(
         default=None,
         description="The time this worker last sent a heartbeat.",
@@ -108,15 +109,6 @@ class Worker(BaseModel):
         "runtimeInfo",
     ]
 
-    @field_validator("type")
-    def type_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in set(["SELFHOSTED", "MANAGED", "WEBHOOK"]):
-            raise ValueError(
-                "must be one of enum values ('SELFHOSTED', 'MANAGED', 'WEBHOOK')"
-            )
-        return value
-
     @field_validator("status")
     def status_validate_enum(cls, value):
         """Validates the enum"""