hatchet-sdk 1.11.0__py3-none-any.whl → 1.12.0__py3-none-any.whl

hatchet_sdk/clients/rest/models/create_tenant_request.py

@@ -22,6 +22,8 @@ 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.tenant_ui_version import TenantUIVersion
+
 
 class CreateTenantRequest(BaseModel):
     """
@@ -30,7 +32,12 @@ class CreateTenantRequest(BaseModel):
 
     name: StrictStr = Field(description="The name of the tenant.")
     slug: StrictStr = Field(description="The slug of the tenant.")
-    __properties: ClassVar[List[str]] = ["name", "slug"]
+    ui_version: Optional[TenantUIVersion] = Field(
+        default=None,
+        description="The UI version of the tenant. Defaults to V0.",
+        alias="uiVersion",
+    )
+    __properties: ClassVar[List[str]] = ["name", "slug", "uiVersion"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -80,5 +87,11 @@ class CreateTenantRequest(BaseModel):
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
-        _obj = cls.model_validate({"name": obj.get("name"), "slug": obj.get("slug")})
+        _obj = cls.model_validate(
+            {
+                "name": obj.get("name"),
+                "slug": obj.get("slug"),
+                "uiVersion": obj.get("uiVersion"),
+            }
+        )
         return _obj

hatchet_sdk/clients/rest/models/tenant.py

@@ -23,6 +23,7 @@ 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.tenant_ui_version import TenantUIVersion
 from hatchet_sdk.clients.rest.models.tenant_version import TenantVersion
 
 
@@ -45,6 +46,9 @@ class Tenant(BaseModel):
         alias="alertMemberEmails",
     )
     version: TenantVersion = Field(description="The version of the tenant.")
+    ui_version: Optional[TenantUIVersion] = Field(
+        default=None, description="The UI of the tenant.", alias="uiVersion"
+    )
     __properties: ClassVar[List[str]] = [
         "metadata",
         "name",
@@ -52,6 +56,7 @@ class Tenant(BaseModel):
         "analyticsOptOut",
         "alertMemberEmails",
         "version",
+        "uiVersion",
     ]
 
     model_config = ConfigDict(
@@ -117,6 +122,7 @@ class Tenant(BaseModel):
                 "analyticsOptOut": obj.get("analyticsOptOut"),
                 "alertMemberEmails": obj.get("alertMemberEmails"),
                 "version": obj.get("version"),
+                "uiVersion": obj.get("uiVersion"),
             }
         )
         return _obj

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

hatchet_sdk/clients/rest/models/update_tenant_request.py

@@ -22,6 +22,7 @@ from typing import Any, ClassVar, Dict, List, Optional, Set
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
 from typing_extensions import Self
 
+from hatchet_sdk.clients.rest.models.tenant_ui_version import TenantUIVersion
 from hatchet_sdk.clients.rest.models.tenant_version import TenantVersion
 
 
@@ -66,6 +67,9 @@ class UpdateTenantRequest(BaseModel):
     version: Optional[TenantVersion] = Field(
         default=None, description="The version of the tenant."
     )
+    ui_version: Optional[TenantUIVersion] = Field(
+        default=None, description="The UI of the tenant.", alias="uiVersion"
+    )
     __properties: ClassVar[List[str]] = [
         "name",
         "analyticsOptOut",
@@ -75,6 +79,7 @@ class UpdateTenantRequest(BaseModel):
         "enableTenantResourceLimitAlerts",
         "maxAlertingFrequency",
         "version",
+        "uiVersion",
     ]
 
     model_config = ConfigDict(
@@ -139,6 +144,7 @@ class UpdateTenantRequest(BaseModel):
                 ),
                 "maxAlertingFrequency": obj.get("maxAlertingFrequency"),
                 "version": obj.get("version"),
+                "uiVersion": obj.get("uiVersion"),
             }
         )
         return _obj

hatchet_sdk/clients/rest/models/v1_update_filter_request.py

@@ -0,0 +1,98 @@
+# 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
+
+
+class V1UpdateFilterRequest(BaseModel):
+    """
+    V1UpdateFilterRequest
+    """  # noqa: E501
+
+    expression: Optional[StrictStr] = Field(
+        default=None, description="The expression for the filter"
+    )
+    scope: Optional[StrictStr] = Field(
+        default=None,
+        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]] = ["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 V1UpdateFilterRequest 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 V1UpdateFilterRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate(
+            {
+                "expression": obj.get("expression"),
+                "scope": obj.get("scope"),
+                "payload": obj.get("payload"),
+            }
+        )
+        return _obj

hatchet_sdk/contracts/v1/workflows_pb2.py

@@ -16,7 +16,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 from hatchet_sdk.contracts.v1.shared import condition_pb2 as v1_dot_shared_dot_condition__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x12v1/workflows.proto\x12\x02v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x19v1/shared/condition.proto\"Z\n\x12\x43\x61ncelTasksRequest\x12\x13\n\x0b\x65xternalIds\x18\x01 \x03(\t\x12$\n\x06\x66ilter\x18\x02 \x01(\x0b\x32\x0f.v1.TasksFilterH\x00\x88\x01\x01\x42\t\n\x07_filter\"Z\n\x12ReplayTasksRequest\x12\x13\n\x0b\x65xternalIds\x18\x01 \x03(\t\x12$\n\x06\x66ilter\x18\x02 \x01(\x0b\x32\x0f.v1.TasksFilterH\x00\x88\x01\x01\x42\t\n\x07_filter\"\xb7\x01\n\x0bTasksFilter\x12\x10\n\x08statuses\x18\x01 \x03(\t\x12)\n\x05since\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\x05until\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00\x88\x01\x01\x12\x14\n\x0cworkflow_ids\x18\x04 \x03(\t\x12\x1b\n\x13\x61\x64\x64itional_metadata\x18\x05 \x03(\tB\x08\n\x06_until\".\n\x13\x43\x61ncelTasksResponse\x12\x17\n\x0f\x63\x61ncelled_tasks\x18\x01 \x03(\t\"-\n\x13ReplayTasksResponse\x12\x16\n\x0ereplayed_tasks\x18\x01 \x03(\t\"\x82\x01\n\x19TriggerWorkflowRunRequest\x12\x15\n\rworkflow_name\x18\x01 \x01(\t\x12\r\n\x05input\x18\x02 \x01(\x0c\x12\x1b\n\x13\x61\x64\x64itional_metadata\x18\x03 \x01(\x0c\x12\x15\n\x08priority\x18\x04 \x01(\x05H\x00\x88\x01\x01\x42\x0b\n\t_priority\"1\n\x1aTriggerWorkflowRunResponse\x12\x13\n\x0b\x65xternal_id\x18\x01 \x01(\t\"\xca\x03\n\x1c\x43reateWorkflowVersionRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x0f\n\x07version\x18\x03 \x01(\t\x12\x16\n\x0e\x65vent_triggers\x18\x04 \x03(\t\x12\x15\n\rcron_triggers\x18\x05 \x03(\t\x12!\n\x05tasks\x18\x06 \x03(\x0b\x32\x12.v1.CreateTaskOpts\x12$\n\x0b\x63oncurrency\x18\x07 \x01(\x0b\x32\x0f.v1.Concurrency\x12\x17\n\ncron_input\x18\x08 \x01(\tH\x00\x88\x01\x01\x12\x30\n\x0fon_failure_task\x18\t \x01(\x0b\x32\x12.v1.CreateTaskOptsH\x01\x88\x01\x01\x12\'\n\x06sticky\x18\n \x01(\x0e\x32\x12.v1.StickyStrategyH\x02\x88\x01\x01\x12\x1d\n\x10\x64\x65\x66\x61ult_priority\x18\x0b \x01(\x05H\x03\x88\x01\x01\x12(\n\x0f\x63oncurrency_arr\x18\x0c \x03(\x0b\x32\x0f.v1.ConcurrencyB\r\n\x0b_cron_inputB\x12\n\x10_on_failure_taskB\t\n\x07_stickyB\x13\n\x11_default_priority\"\x93\x01\n\x0b\x43oncurrency\x12\x12\n\nexpression\x18\x01 \x01(\t\x12\x15\n\x08max_runs\x18\x02 \x01(\x05H\x00\x88\x01\x01\x12\x39\n\x0elimit_strategy\x18\x03 \x01(\x0e\x32\x1c.v1.ConcurrencyLimitStrategyH\x01\x88\x01\x01\x42\x0b\n\t_max_runsB\x11\n\x0f_limit_strategy\"\xe4\x01\n\x13\x44\x65siredWorkerLabels\x12\x15\n\x08strValue\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x15\n\x08intValue\x18\x02 \x01(\x05H\x01\x88\x01\x01\x12\x15\n\x08required\x18\x03 \x01(\x08H\x02\x88\x01\x01\x12\x32\n\ncomparator\x18\x04 \x01(\x0e\x32\x19.v1.WorkerLabelComparatorH\x03\x88\x01\x01\x12\x13\n\x06weight\x18\x05 \x01(\x05H\x04\x88\x01\x01\x42\x0b\n\t_strValueB\x0b\n\t_intValueB\x0b\n\t_requiredB\r\n\x0b_comparatorB\t\n\x07_weight\"\xb1\x04\n\x0e\x43reateTaskOpts\x12\x13\n\x0breadable_id\x18\x01 \x01(\t\x12\x0e\n\x06\x61\x63tion\x18\x02 \x01(\t\x12\x0f\n\x07timeout\x18\x03 \x01(\t\x12\x0e\n\x06inputs\x18\x04 \x01(\t\x12\x0f\n\x07parents\x18\x05 \x03(\t\x12\x0f\n\x07retries\x18\x06 \x01(\x05\x12,\n\x0brate_limits\x18\x07 \x03(\x0b\x32\x17.v1.CreateTaskRateLimit\x12;\n\rworker_labels\x18\x08 \x03(\x0b\x32$.v1.CreateTaskOpts.WorkerLabelsEntry\x12\x1b\n\x0e\x62\x61\x63koff_factor\x18\t \x01(\x02H\x00\x88\x01\x01\x12 \n\x13\x62\x61\x63koff_max_seconds\x18\n \x01(\x05H\x01\x88\x01\x01\x12$\n\x0b\x63oncurrency\x18\x0b \x03(\x0b\x32\x0f.v1.Concurrency\x12+\n\nconditions\x18\x0c \x01(\x0b\x32\x12.v1.TaskConditionsH\x02\x88\x01\x01\x12\x1d\n\x10schedule_timeout\x18\r \x01(\tH\x03\x88\x01\x01\x1aL\n\x11WorkerLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12&\n\x05value\x18\x02 \x01(\x0b\x32\x17.v1.DesiredWorkerLabels:\x02\x38\x01\x42\x11\n\x0f_backoff_factorB\x16\n\x14_backoff_max_secondsB\r\n\x0b_conditionsB\x13\n\x11_schedule_timeout\"\xfd\x01\n\x13\x43reateTaskRateLimit\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x12\n\x05units\x18\x02 \x01(\x05H\x00\x88\x01\x01\x12\x15\n\x08key_expr\x18\x03 \x01(\tH\x01\x88\x01\x01\x12\x17\n\nunits_expr\x18\x04 \x01(\tH\x02\x88\x01\x01\x12\x1e\n\x11limit_values_expr\x18\x05 \x01(\tH\x03\x88\x01\x01\x12,\n\x08\x64uration\x18\x06 \x01(\x0e\x32\x15.v1.RateLimitDurationH\x04\x88\x01\x01\x42\x08\n\x06_unitsB\x0b\n\t_key_exprB\r\n\x0b_units_exprB\x14\n\x12_limit_values_exprB\x0b\n\t_duration\"@\n\x1d\x43reateWorkflowVersionResponse\x12\n\n\x02id\x18\x01 \x01(\t\x12\x13\n\x0bworkflow_id\x18\x02 \x01(\t*$\n\x0eStickyStrategy\x12\x08\n\x04SOFT\x10\x00\x12\x08\n\x04HARD\x10\x01*]\n\x11RateLimitDuration\x12\n\n\x06SECOND\x10\x00\x12\n\n\x06MINUTE\x10\x01\x12\x08\n\x04HOUR\x10\x02\x12\x07\n\x03\x44\x41Y\x10\x03\x12\x08\n\x04WEEK\x10\x04\x12\t\n\x05MONTH\x10\x05\x12\x08\n\x04YEAR\x10\x06*\x7f\n\x18\x43oncurrencyLimitStrategy\x12\x16\n\x12\x43\x41NCEL_IN_PROGRESS\x10\x00\x12\x0f\n\x0b\x44ROP_NEWEST\x10\x01\x12\x10\n\x0cQUEUE_NEWEST\x10\x02\x12\x15\n\x11GROUP_ROUND_ROBIN\x10\x03\x12\x11\n\rCANCEL_NEWEST\x10\x04*\x85\x01\n\x15WorkerLabelComparator\x12\t\n\x05\x45QUAL\x10\x00\x12\r\n\tNOT_EQUAL\x10\x01\x12\x10\n\x0cGREATER_THAN\x10\x02\x12\x19\n\x15GREATER_THAN_OR_EQUAL\x10\x03\x12\r\n\tLESS_THAN\x10\x04\x12\x16\n\x12LESS_THAN_OR_EQUAL\x10\x05\x32\xb7\x02\n\x0c\x41\x64minService\x12R\n\x0bPutWorkflow\x12 .v1.CreateWorkflowVersionRequest\x1a!.v1.CreateWorkflowVersionResponse\x12>\n\x0b\x43\x61ncelTasks\x12\x16.v1.CancelTasksRequest\x1a\x17.v1.CancelTasksResponse\x12>\n\x0bReplayTasks\x12\x16.v1.ReplayTasksRequest\x1a\x17.v1.ReplayTasksResponse\x12S\n\x12TriggerWorkflowRun\x12\x1d.v1.TriggerWorkflowRunRequest\x1a\x1e.v1.TriggerWorkflowRunResponseBBZ@github.com/hatchet-dev/hatchet/internal/services/shared/proto/v1b\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x12v1/workflows.proto\x12\x02v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x19v1/shared/condition.proto\"Z\n\x12\x43\x61ncelTasksRequest\x12\x13\n\x0b\x65xternalIds\x18\x01 \x03(\t\x12$\n\x06\x66ilter\x18\x02 \x01(\x0b\x32\x0f.v1.TasksFilterH\x00\x88\x01\x01\x42\t\n\x07_filter\"Z\n\x12ReplayTasksRequest\x12\x13\n\x0b\x65xternalIds\x18\x01 \x03(\t\x12$\n\x06\x66ilter\x18\x02 \x01(\x0b\x32\x0f.v1.TasksFilterH\x00\x88\x01\x01\x42\t\n\x07_filter\"\xb7\x01\n\x0bTasksFilter\x12\x10\n\x08statuses\x18\x01 \x03(\t\x12)\n\x05since\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\x05until\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00\x88\x01\x01\x12\x14\n\x0cworkflow_ids\x18\x04 \x03(\t\x12\x1b\n\x13\x61\x64\x64itional_metadata\x18\x05 \x03(\tB\x08\n\x06_until\".\n\x13\x43\x61ncelTasksResponse\x12\x17\n\x0f\x63\x61ncelled_tasks\x18\x01 \x03(\t\"-\n\x13ReplayTasksResponse\x12\x16\n\x0ereplayed_tasks\x18\x01 \x03(\t\"\x82\x01\n\x19TriggerWorkflowRunRequest\x12\x15\n\rworkflow_name\x18\x01 \x01(\t\x12\r\n\x05input\x18\x02 \x01(\x0c\x12\x1b\n\x13\x61\x64\x64itional_metadata\x18\x03 \x01(\x0c\x12\x15\n\x08priority\x18\x04 \x01(\x05H\x00\x88\x01\x01\x42\x0b\n\t_priority\"1\n\x1aTriggerWorkflowRunResponse\x12\x13\n\x0b\x65xternal_id\x18\x01 \x01(\t\"\xf6\x03\n\x1c\x43reateWorkflowVersionRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x0f\n\x07version\x18\x03 \x01(\t\x12\x16\n\x0e\x65vent_triggers\x18\x04 \x03(\t\x12\x15\n\rcron_triggers\x18\x05 \x03(\t\x12!\n\x05tasks\x18\x06 \x03(\x0b\x32\x12.v1.CreateTaskOpts\x12$\n\x0b\x63oncurrency\x18\x07 \x01(\x0b\x32\x0f.v1.Concurrency\x12\x17\n\ncron_input\x18\x08 \x01(\tH\x00\x88\x01\x01\x12\x30\n\x0fon_failure_task\x18\t \x01(\x0b\x32\x12.v1.CreateTaskOptsH\x01\x88\x01\x01\x12\'\n\x06sticky\x18\n \x01(\x0e\x32\x12.v1.StickyStrategyH\x02\x88\x01\x01\x12\x1d\n\x10\x64\x65\x66\x61ult_priority\x18\x0b \x01(\x05H\x03\x88\x01\x01\x12(\n\x0f\x63oncurrency_arr\x18\x0c \x03(\x0b\x32\x0f.v1.Concurrency\x12*\n\x0f\x64\x65\x66\x61ult_filters\x18\r \x03(\x0b\x32\x11.v1.DefaultFilterB\r\n\x0b_cron_inputB\x12\n\x10_on_failure_taskB\t\n\x07_stickyB\x13\n\x11_default_priority\"T\n\rDefaultFilter\x12\x12\n\nexpression\x18\x01 \x01(\t\x12\r\n\x05scope\x18\x02 \x01(\t\x12\x14\n\x07payload\x18\x03 \x01(\tH\x00\x88\x01\x01\x42\n\n\x08_payload\"\x93\x01\n\x0b\x43oncurrency\x12\x12\n\nexpression\x18\x01 \x01(\t\x12\x15\n\x08max_runs\x18\x02 \x01(\x05H\x00\x88\x01\x01\x12\x39\n\x0elimit_strategy\x18\x03 \x01(\x0e\x32\x1c.v1.ConcurrencyLimitStrategyH\x01\x88\x01\x01\x42\x0b\n\t_max_runsB\x11\n\x0f_limit_strategy\"\xe4\x01\n\x13\x44\x65siredWorkerLabels\x12\x15\n\x08strValue\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x15\n\x08intValue\x18\x02 \x01(\x05H\x01\x88\x01\x01\x12\x15\n\x08required\x18\x03 \x01(\x08H\x02\x88\x01\x01\x12\x32\n\ncomparator\x18\x04 \x01(\x0e\x32\x19.v1.WorkerLabelComparatorH\x03\x88\x01\x01\x12\x13\n\x06weight\x18\x05 \x01(\x05H\x04\x88\x01\x01\x42\x0b\n\t_strValueB\x0b\n\t_intValueB\x0b\n\t_requiredB\r\n\x0b_comparatorB\t\n\x07_weight\"\xb1\x04\n\x0e\x43reateTaskOpts\x12\x13\n\x0breadable_id\x18\x01 \x01(\t\x12\x0e\n\x06\x61\x63tion\x18\x02 \x01(\t\x12\x0f\n\x07timeout\x18\x03 \x01(\t\x12\x0e\n\x06inputs\x18\x04 \x01(\t\x12\x0f\n\x07parents\x18\x05 \x03(\t\x12\x0f\n\x07retries\x18\x06 \x01(\x05\x12,\n\x0brate_limits\x18\x07 \x03(\x0b\x32\x17.v1.CreateTaskRateLimit\x12;\n\rworker_labels\x18\x08 \x03(\x0b\x32$.v1.CreateTaskOpts.WorkerLabelsEntry\x12\x1b\n\x0e\x62\x61\x63koff_factor\x18\t \x01(\x02H\x00\x88\x01\x01\x12 \n\x13\x62\x61\x63koff_max_seconds\x18\n \x01(\x05H\x01\x88\x01\x01\x12$\n\x0b\x63oncurrency\x18\x0b \x03(\x0b\x32\x0f.v1.Concurrency\x12+\n\nconditions\x18\x0c \x01(\x0b\x32\x12.v1.TaskConditionsH\x02\x88\x01\x01\x12\x1d\n\x10schedule_timeout\x18\r \x01(\tH\x03\x88\x01\x01\x1aL\n\x11WorkerLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12&\n\x05value\x18\x02 \x01(\x0b\x32\x17.v1.DesiredWorkerLabels:\x02\x38\x01\x42\x11\n\x0f_backoff_factorB\x16\n\x14_backoff_max_secondsB\r\n\x0b_conditionsB\x13\n\x11_schedule_timeout\"\xfd\x01\n\x13\x43reateTaskRateLimit\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x12\n\x05units\x18\x02 \x01(\x05H\x00\x88\x01\x01\x12\x15\n\x08key_expr\x18\x03 \x01(\tH\x01\x88\x01\x01\x12\x17\n\nunits_expr\x18\x04 \x01(\tH\x02\x88\x01\x01\x12\x1e\n\x11limit_values_expr\x18\x05 \x01(\tH\x03\x88\x01\x01\x12,\n\x08\x64uration\x18\x06 \x01(\x0e\x32\x15.v1.RateLimitDurationH\x04\x88\x01\x01\x42\x08\n\x06_unitsB\x0b\n\t_key_exprB\r\n\x0b_units_exprB\x14\n\x12_limit_values_exprB\x0b\n\t_duration\"@\n\x1d\x43reateWorkflowVersionResponse\x12\n\n\x02id\x18\x01 \x01(\t\x12\x13\n\x0bworkflow_id\x18\x02 \x01(\t*$\n\x0eStickyStrategy\x12\x08\n\x04SOFT\x10\x00\x12\x08\n\x04HARD\x10\x01*]\n\x11RateLimitDuration\x12\n\n\x06SECOND\x10\x00\x12\n\n\x06MINUTE\x10\x01\x12\x08\n\x04HOUR\x10\x02\x12\x07\n\x03\x44\x41Y\x10\x03\x12\x08\n\x04WEEK\x10\x04\x12\t\n\x05MONTH\x10\x05\x12\x08\n\x04YEAR\x10\x06*\x7f\n\x18\x43oncurrencyLimitStrategy\x12\x16\n\x12\x43\x41NCEL_IN_PROGRESS\x10\x00\x12\x0f\n\x0b\x44ROP_NEWEST\x10\x01\x12\x10\n\x0cQUEUE_NEWEST\x10\x02\x12\x15\n\x11GROUP_ROUND_ROBIN\x10\x03\x12\x11\n\rCANCEL_NEWEST\x10\x04*\x85\x01\n\x15WorkerLabelComparator\x12\t\n\x05\x45QUAL\x10\x00\x12\r\n\tNOT_EQUAL\x10\x01\x12\x10\n\x0cGREATER_THAN\x10\x02\x12\x19\n\x15GREATER_THAN_OR_EQUAL\x10\x03\x12\r\n\tLESS_THAN\x10\x04\x12\x16\n\x12LESS_THAN_OR_EQUAL\x10\x05\x32\xb7\x02\n\x0c\x41\x64minService\x12R\n\x0bPutWorkflow\x12 .v1.CreateWorkflowVersionRequest\x1a!.v1.CreateWorkflowVersionResponse\x12>\n\x0b\x43\x61ncelTasks\x12\x16.v1.CancelTasksRequest\x1a\x17.v1.CancelTasksResponse\x12>\n\x0bReplayTasks\x12\x16.v1.ReplayTasksRequest\x1a\x17.v1.ReplayTasksResponse\x12S\n\x12TriggerWorkflowRun\x12\x1d.v1.TriggerWorkflowRunRequest\x1a\x1e.v1.TriggerWorkflowRunResponseBBZ@github.com/hatchet-dev/hatchet/internal/services/shared/proto/v1b\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -26,14 +26,14 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['DESCRIPTOR']._serialized_options = b'Z@github.com/hatchet-dev/hatchet/internal/services/shared/proto/v1'
   _globals['_CREATETASKOPTS_WORKERLABELSENTRY']._loaded_options = None
   _globals['_CREATETASKOPTS_WORKERLABELSENTRY']._serialized_options = b'8\001'
-  _globals['_STICKYSTRATEGY']._serialized_start=2463
-  _globals['_STICKYSTRATEGY']._serialized_end=2499
-  _globals['_RATELIMITDURATION']._serialized_start=2501
-  _globals['_RATELIMITDURATION']._serialized_end=2594
-  _globals['_CONCURRENCYLIMITSTRATEGY']._serialized_start=2596
-  _globals['_CONCURRENCYLIMITSTRATEGY']._serialized_end=2723
-  _globals['_WORKERLABELCOMPARATOR']._serialized_start=2726
-  _globals['_WORKERLABELCOMPARATOR']._serialized_end=2859
+  _globals['_STICKYSTRATEGY']._serialized_start=2593
+  _globals['_STICKYSTRATEGY']._serialized_end=2629
+  _globals['_RATELIMITDURATION']._serialized_start=2631
+  _globals['_RATELIMITDURATION']._serialized_end=2724
+  _globals['_CONCURRENCYLIMITSTRATEGY']._serialized_start=2726
+  _globals['_CONCURRENCYLIMITSTRATEGY']._serialized_end=2853
+  _globals['_WORKERLABELCOMPARATOR']._serialized_start=2856
+  _globals['_WORKERLABELCOMPARATOR']._serialized_end=2989
   _globals['_CANCELTASKSREQUEST']._serialized_start=86
   _globals['_CANCELTASKSREQUEST']._serialized_end=176
   _globals['_REPLAYTASKSREQUEST']._serialized_start=178
@@ -49,19 +49,21 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_TRIGGERWORKFLOWRUNRESPONSE']._serialized_start=684
   _globals['_TRIGGERWORKFLOWRUNRESPONSE']._serialized_end=733
   _globals['_CREATEWORKFLOWVERSIONREQUEST']._serialized_start=736
-  _globals['_CREATEWORKFLOWVERSIONREQUEST']._serialized_end=1194
-  _globals['_CONCURRENCY']._serialized_start=1197
-  _globals['_CONCURRENCY']._serialized_end=1344
-  _globals['_DESIREDWORKERLABELS']._serialized_start=1347
-  _globals['_DESIREDWORKERLABELS']._serialized_end=1575
-  _globals['_CREATETASKOPTS']._serialized_start=1578
-  _globals['_CREATETASKOPTS']._serialized_end=2139
-  _globals['_CREATETASKOPTS_WORKERLABELSENTRY']._serialized_start=1984
-  _globals['_CREATETASKOPTS_WORKERLABELSENTRY']._serialized_end=2060
-  _globals['_CREATETASKRATELIMIT']._serialized_start=2142
-  _globals['_CREATETASKRATELIMIT']._serialized_end=2395
-  _globals['_CREATEWORKFLOWVERSIONRESPONSE']._serialized_start=2397
-  _globals['_CREATEWORKFLOWVERSIONRESPONSE']._serialized_end=2461
-  _globals['_ADMINSERVICE']._serialized_start=2862
-  _globals['_ADMINSERVICE']._serialized_end=3173
+  _globals['_CREATEWORKFLOWVERSIONREQUEST']._serialized_end=1238
+  _globals['_DEFAULTFILTER']._serialized_start=1240
+  _globals['_DEFAULTFILTER']._serialized_end=1324
+  _globals['_CONCURRENCY']._serialized_start=1327
+  _globals['_CONCURRENCY']._serialized_end=1474
+  _globals['_DESIREDWORKERLABELS']._serialized_start=1477
+  _globals['_DESIREDWORKERLABELS']._serialized_end=1705
+  _globals['_CREATETASKOPTS']._serialized_start=1708
+  _globals['_CREATETASKOPTS']._serialized_end=2269
+  _globals['_CREATETASKOPTS_WORKERLABELSENTRY']._serialized_start=2114
+  _globals['_CREATETASKOPTS_WORKERLABELSENTRY']._serialized_end=2190
+  _globals['_CREATETASKRATELIMIT']._serialized_start=2272
+  _globals['_CREATETASKRATELIMIT']._serialized_end=2525
+  _globals['_CREATEWORKFLOWVERSIONRESPONSE']._serialized_start=2527
+  _globals['_CREATEWORKFLOWVERSIONRESPONSE']._serialized_end=2591
+  _globals['_ADMINSERVICE']._serialized_start=2992
+  _globals['_ADMINSERVICE']._serialized_end=3303
 # @@protoc_insertion_point(module_scope)

hatchet_sdk/contracts/v1/workflows_pb2.pyi

@@ -121,7 +121,7 @@ class TriggerWorkflowRunResponse(_message.Message):
     def __init__(self, external_id: _Optional[str] = ...) -> None: ...
 
 class CreateWorkflowVersionRequest(_message.Message):
-    __slots__ = ("name", "description", "version", "event_triggers", "cron_triggers", "tasks", "concurrency", "cron_input", "on_failure_task", "sticky", "default_priority", "concurrency_arr")
+    __slots__ = ("name", "description", "version", "event_triggers", "cron_triggers", "tasks", "concurrency", "cron_input", "on_failure_task", "sticky", "default_priority", "concurrency_arr", "default_filters")
     NAME_FIELD_NUMBER: _ClassVar[int]
     DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
     VERSION_FIELD_NUMBER: _ClassVar[int]
@@ -134,6 +134,7 @@ class CreateWorkflowVersionRequest(_message.Message):
     STICKY_FIELD_NUMBER: _ClassVar[int]
     DEFAULT_PRIORITY_FIELD_NUMBER: _ClassVar[int]
     CONCURRENCY_ARR_FIELD_NUMBER: _ClassVar[int]
+    DEFAULT_FILTERS_FIELD_NUMBER: _ClassVar[int]
     name: str
     description: str
     version: str
@@ -146,7 +147,18 @@ class CreateWorkflowVersionRequest(_message.Message):
     sticky: StickyStrategy
     default_priority: int
     concurrency_arr: _containers.RepeatedCompositeFieldContainer[Concurrency]
-    def __init__(self, name: _Optional[str] = ..., description: _Optional[str] = ..., version: _Optional[str] = ..., event_triggers: _Optional[_Iterable[str]] = ..., cron_triggers: _Optional[_Iterable[str]] = ..., tasks: _Optional[_Iterable[_Union[CreateTaskOpts, _Mapping]]] = ..., concurrency: _Optional[_Union[Concurrency, _Mapping]] = ..., cron_input: _Optional[str] = ..., on_failure_task: _Optional[_Union[CreateTaskOpts, _Mapping]] = ..., sticky: _Optional[_Union[StickyStrategy, str]] = ..., default_priority: _Optional[int] = ..., concurrency_arr: _Optional[_Iterable[_Union[Concurrency, _Mapping]]] = ...) -> None: ...
+    default_filters: _containers.RepeatedCompositeFieldContainer[DefaultFilter]
+    def __init__(self, name: _Optional[str] = ..., description: _Optional[str] = ..., version: _Optional[str] = ..., event_triggers: _Optional[_Iterable[str]] = ..., cron_triggers: _Optional[_Iterable[str]] = ..., tasks: _Optional[_Iterable[_Union[CreateTaskOpts, _Mapping]]] = ..., concurrency: _Optional[_Union[Concurrency, _Mapping]] = ..., cron_input: _Optional[str] = ..., on_failure_task: _Optional[_Union[CreateTaskOpts, _Mapping]] = ..., sticky: _Optional[_Union[StickyStrategy, str]] = ..., default_priority: _Optional[int] = ..., concurrency_arr: _Optional[_Iterable[_Union[Concurrency, _Mapping]]] = ..., default_filters: _Optional[_Iterable[_Union[DefaultFilter, _Mapping]]] = ...) -> None: ...
+
+class DefaultFilter(_message.Message):
+    __slots__ = ("expression", "scope", "payload")
+    EXPRESSION_FIELD_NUMBER: _ClassVar[int]
+    SCOPE_FIELD_NUMBER: _ClassVar[int]
+    PAYLOAD_FIELD_NUMBER: _ClassVar[int]
+    expression: str
+    scope: str
+    payload: str
+    def __init__(self, expression: _Optional[str] = ..., scope: _Optional[str] = ..., payload: _Optional[str] = ...) -> None: ...
 
 class Concurrency(_message.Message):
     __slots__ = ("expression", "max_runs", "limit_strategy")

hatchet_sdk/features/filters.py

@@ -7,6 +7,9 @@ from hatchet_sdk.clients.rest.models.v1_create_filter_request import (
 )
 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_update_filter_request import (
+    V1UpdateFilterRequest,
+)
 from hatchet_sdk.clients.v1.api_client import BaseRestClient
 from hatchet_sdk.utils.typing import JSONSerializableMapping
 
@@ -179,3 +182,36 @@ class FiltersClient(BaseRestClient):
         :return: The deleted filter.
         """
         return await asyncio.to_thread(self.delete, filter_id)
+
+    def update(
+        self,
+        filter_id: str,
+        updates: V1UpdateFilterRequest,
+    ) -> V1Filter:
+        """
+        Update a filter by its ID.
+
+        :param filter_id: The ID of the filter to delete.
+        :param updates: The updates to apply to the filter.
+        :return: The updated filter.
+        """
+        with self.client() as client:
+            return self._fa(client).v1_filter_update(
+                tenant=self.tenant_id,
+                v1_filter=filter_id,
+                v1_update_filter_request=updates,
+            )
+
+    async def aio_update(
+        self,
+        filter_id: str,
+        updates: V1UpdateFilterRequest,
+    ) -> V1Filter:
+        """
+        Update a filter by its ID.
+
+        :param filter_id: The ID of the filter to delete.
+        :param updates: The updates to apply to the filter.
+        :return: The updated filter.
+        """
+        return await asyncio.to_thread(self.update, filter_id, updates)

hatchet_sdk/features/runs.py

@@ -1,5 +1,5 @@
 import asyncio
-from datetime import datetime, timedelta
+from datetime import datetime, timedelta, timezone
 from typing import TYPE_CHECKING, Literal, overload
 
 from pydantic import BaseModel, model_validator
@@ -129,6 +129,25 @@ class RunsClient(BaseRestClient):
         """
         return await asyncio.to_thread(self.get, workflow_run_id)
 
+    def get_status(self, workflow_run_id: str) -> V1TaskStatus:
+        """
+        Get workflow run status for a given workflow run ID.
+
+        :param workflow_run_id: The ID of the workflow run to retrieve details for.
+        :return: The task status
+        """
+        with self.client() as client:
+            return self._wra(client).v1_workflow_run_get_status(str(workflow_run_id))
+
+    async def aio_get_status(self, workflow_run_id: str) -> V1TaskStatus:
+        """
+        Get workflow run status for a given workflow run ID.
+
+        :param workflow_run_id: The ID of the workflow run to retrieve details for.
+        :return: The task status
+        """
+        return await asyncio.to_thread(self.get_status, workflow_run_id)
+
     async def aio_list(
         self,
         since: datetime | None = None,
@@ -162,7 +181,7 @@ class RunsClient(BaseRestClient):
         """
         return await asyncio.to_thread(
             self.list,
-            since=since or datetime.now() - timedelta(days=1),
+            since=since or datetime.now(tz=timezone.utc) - timedelta(days=1),
             only_tasks=only_tasks,
             offset=offset,
             limit=limit,
@@ -209,7 +228,7 @@ class RunsClient(BaseRestClient):
         with self.client() as client:
             return self._wra(client).v1_workflow_run_list(
                 tenant=self.client_config.tenant_id,
-                since=since or datetime.now() - timedelta(days=1),
+                since=since or datetime.now(tz=timezone.utc) - timedelta(days=1),
                 only_tasks=only_tasks,
                 offset=offset,
                 limit=limit,

hatchet_sdk/features/tenant.py

@@ -0,0 +1,32 @@
+import asyncio
+
+from hatchet_sdk.clients.rest.api.tenant_api import TenantApi
+from hatchet_sdk.clients.rest.api_client import ApiClient
+from hatchet_sdk.clients.rest.models.tenant import Tenant
+from hatchet_sdk.clients.v1.api_client import BaseRestClient
+
+
+class TenantClient(BaseRestClient):
+    """
+    The tenant client is a client for interacting with your Tenant.
+    """
+
+    def _ta(self, client: ApiClient) -> TenantApi:
+        return TenantApi(client)
+
+    def get(self) -> Tenant:
+        """
+        Get the current tenant.
+
+        :return: The tenant.
+        """
+        with self.client() as client:
+            return self._ta(client).tenant_get(self.client_config.tenant_id)
+
+    async def aio_get(self) -> Tenant:
+        """
+        Get the current tenant.
+
+        :return: The tenant.
+        """
+        return await asyncio.to_thread(self.get)