hatchet-sdk 1.6.5__py3-none-any.whl → 1.8.0__py3-none-any.whl

hatchet_sdk/clients/rest/models/v1_task_timing.py

@@ -0,0 +1,159 @@
+# 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 V1TaskTiming(BaseModel):
+    """
+    V1TaskTiming
+    """  # noqa: E501
+
+    metadata: APIResourceMeta
+    depth: StrictInt = Field(description="The depth of the task in the waterfall.")
+    status: V1TaskStatus
+    task_display_name: StrictStr = Field(
+        description="The display name of the task run.", alias="taskDisplayName"
+    )
+    task_external_id: Annotated[
+        str, Field(min_length=36, strict=True, max_length=36)
+    ] = Field(description="The external ID of the task.", alias="taskExternalId")
+    task_id: StrictInt = Field(description="The ID of the task.", alias="taskId")
+    task_inserted_at: datetime = Field(
+        description="The timestamp the task was inserted.", alias="taskInsertedAt"
+    )
+    tenant_id: Annotated[str, Field(min_length=36, strict=True, max_length=36)] = Field(
+        description="The ID of the tenant.", alias="tenantId"
+    )
+    parent_task_external_id: Optional[
+        Annotated[str, Field(min_length=36, strict=True, max_length=36)]
+    ] = Field(
+        default=None,
+        description="The external ID of the parent task.",
+        alias="parentTaskExternalId",
+    )
+    queued_at: Optional[datetime] = Field(
+        default=None,
+        description="The timestamp the task run was queued.",
+        alias="queuedAt",
+    )
+    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",
+    )
+    __properties: ClassVar[List[str]] = [
+        "metadata",
+        "depth",
+        "status",
+        "taskDisplayName",
+        "taskExternalId",
+        "taskId",
+        "taskInsertedAt",
+        "tenantId",
+        "parentTaskExternalId",
+        "queuedAt",
+        "startedAt",
+        "finishedAt",
+    ]
+
+    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 V1TaskTiming 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 V1TaskTiming 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
+                ),
+                "depth": obj.get("depth"),
+                "status": obj.get("status"),
+                "taskDisplayName": obj.get("taskDisplayName"),
+                "taskExternalId": obj.get("taskExternalId"),
+                "taskId": obj.get("taskId"),
+                "taskInsertedAt": obj.get("taskInsertedAt"),
+                "tenantId": obj.get("tenantId"),
+                "parentTaskExternalId": obj.get("parentTaskExternalId"),
+                "queuedAt": obj.get("queuedAt"),
+                "startedAt": obj.get("startedAt"),
+                "finishedAt": obj.get("finishedAt"),
+            }
+        )
+        return _obj

hatchet_sdk/clients/rest/models/v1_task_timing_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, Field
+from typing_extensions import Self
+
+from hatchet_sdk.clients.rest.models.pagination_response import PaginationResponse
+from hatchet_sdk.clients.rest.models.v1_task_timing import V1TaskTiming
+
+
+class V1TaskTimingList(BaseModel):
+    """
+    V1TaskTimingList
+    """  # noqa: E501
+
+    pagination: PaginationResponse
+    rows: List[V1TaskTiming] = Field(description="The list of task timings")
+    __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 V1TaskTimingList 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 V1TaskTimingList 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": (
+                    [V1TaskTiming.from_dict(_item) for _item in obj["rows"]]
+                    if obj.get("rows") is not None
+                    else None
+                ),
+            }
+        )
+        return _obj

hatchet_sdk/features/cron.py

@@ -172,6 +172,8 @@ class CronClient(BaseRestClient):
         additional_metadata: JSONSerializableMapping | None = None,
         order_by_field: CronWorkflowsOrderByField | None = None,
         order_by_direction: WorkflowRunOrderByDirection | None = None,
+        workflow_name: str | None = None,
+        cron_name: str | None = None,
     ) -> CronWorkflowsList:
         """
         Retrieve a list of all workflow cron triggers matching the criteria.
@@ -182,6 +184,8 @@ class CronClient(BaseRestClient):
         :param additional_metadata: Filter by additional metadata keys.
         :param order_by_field: The field to order the list by.
         :param order_by_direction: The direction to order the list by.
+        :param workflow_name: The name of the workflow to filter by.
+        :param cron_name: The name of the cron trigger to filter by.
 
         :return: A list of cron workflows.
         """
@@ -193,6 +197,8 @@ class CronClient(BaseRestClient):
             additional_metadata=additional_metadata,
             order_by_field=order_by_field,
             order_by_direction=order_by_direction,
+            workflow_name=workflow_name,
+            cron_name=cron_name,
         )
 
     def list(
@@ -203,6 +209,8 @@ class CronClient(BaseRestClient):
         additional_metadata: JSONSerializableMapping | None = None,
         order_by_field: CronWorkflowsOrderByField | None = None,
         order_by_direction: WorkflowRunOrderByDirection | None = None,
+        workflow_name: str | None = None,
+        cron_name: str | None = None,
     ) -> CronWorkflowsList:
         """
         Retrieve a list of all workflow cron triggers matching the criteria.
@@ -213,6 +221,8 @@ class CronClient(BaseRestClient):
         :param additional_metadata: Filter by additional metadata keys.
         :param order_by_field: The field to order the list by.
         :param order_by_direction: The direction to order the list by.
+        :param workflow_name: The name of the workflow to filter by.
+        :param cron_name: The name of the cron trigger to filter by.
 
         :return: A list of cron workflows.
         """
@@ -227,6 +237,8 @@ class CronClient(BaseRestClient):
                 ),
                 order_by_field=order_by_field,
                 order_by_direction=order_by_direction,
+                workflow_name=workflow_name,
+                cron_name=cron_name,
             )
 
     def get(self, cron_id: str) -> CronWorkflows:

hatchet_sdk/features/runs.py

@@ -131,7 +131,7 @@ class RunsClient(BaseRestClient):
 
     async def aio_list(
         self,
-        since: datetime = datetime.now() - timedelta(hours=1),
+        since: datetime | None = None,
         only_tasks: bool = False,
         offset: int | None = None,
         limit: int | None = None,
@@ -160,7 +160,7 @@ class RunsClient(BaseRestClient):
         """
         return await asyncio.to_thread(
             self.list,
-            since=since,
+            since=since or datetime.now() - timedelta(days=1),
             only_tasks=only_tasks,
             offset=offset,
             limit=limit,
@@ -174,7 +174,7 @@ class RunsClient(BaseRestClient):
 
     def list(
         self,
-        since: datetime = datetime.now() - timedelta(hours=1),
+        since: datetime | None = None,
         only_tasks: bool = False,
         offset: int | None = None,
         limit: int | None = None,
@@ -204,7 +204,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,
+                since=since or datetime.now() - timedelta(days=1),
                 only_tasks=only_tasks,
                 offset=offset,
                 limit=limit,

hatchet_sdk/hatchet.py

@@ -1,5 +1,6 @@
 import asyncio
 import logging
+from datetime import timedelta
 from typing import Any, Callable, Type, Union, cast, overload
 
 from hatchet_sdk import Context, DurableContext
@@ -21,8 +22,6 @@ from hatchet_sdk.logger import logger
 from hatchet_sdk.rate_limit import RateLimit
 from hatchet_sdk.runnables.standalone import Standalone
 from hatchet_sdk.runnables.types import (
-    DEFAULT_EXECUTION_TIMEOUT,
-    DEFAULT_SCHEDULE_TIMEOUT,
     ConcurrencyExpression,
     EmptyModel,
     R,
@@ -294,8 +293,8 @@ class Hatchet:
         sticky: StickyStrategy | None = None,
         default_priority: int = 1,
         concurrency: ConcurrencyExpression | list[ConcurrencyExpression] | None = None,
-        schedule_timeout: Duration = DEFAULT_SCHEDULE_TIMEOUT,
-        execution_timeout: Duration = DEFAULT_EXECUTION_TIMEOUT,
+        schedule_timeout: Duration = timedelta(minutes=5),
+        execution_timeout: Duration = timedelta(seconds=60),
         retries: int = 0,
         rate_limits: list[RateLimit] = [],
         desired_worker_labels: dict[str, DesiredWorkerLabel] = {},
@@ -316,8 +315,8 @@ class Hatchet:
         sticky: StickyStrategy | None = None,
         default_priority: int = 1,
         concurrency: ConcurrencyExpression | list[ConcurrencyExpression] | None = None,
-        schedule_timeout: Duration = DEFAULT_SCHEDULE_TIMEOUT,
-        execution_timeout: Duration = DEFAULT_EXECUTION_TIMEOUT,
+        schedule_timeout: Duration = timedelta(minutes=5),
+        execution_timeout: Duration = timedelta(seconds=60),
         retries: int = 0,
         rate_limits: list[RateLimit] = [],
         desired_worker_labels: dict[str, DesiredWorkerLabel] = {},
@@ -339,8 +338,8 @@ class Hatchet:
         sticky: StickyStrategy | None = None,
         default_priority: int = 1,
         concurrency: ConcurrencyExpression | list[ConcurrencyExpression] | None = None,
-        schedule_timeout: Duration = DEFAULT_SCHEDULE_TIMEOUT,
-        execution_timeout: Duration = DEFAULT_EXECUTION_TIMEOUT,
+        schedule_timeout: Duration = timedelta(minutes=5),
+        execution_timeout: Duration = timedelta(seconds=60),
         retries: int = 0,
         rate_limits: list[RateLimit] = [],
         desired_worker_labels: dict[str, DesiredWorkerLabel] = {},
@@ -451,8 +450,8 @@ class Hatchet:
         sticky: StickyStrategy | None = None,
         default_priority: int = 1,
         concurrency: ConcurrencyExpression | None = None,
-        schedule_timeout: Duration = DEFAULT_SCHEDULE_TIMEOUT,
-        execution_timeout: Duration = DEFAULT_EXECUTION_TIMEOUT,
+        schedule_timeout: Duration = timedelta(minutes=5),
+        execution_timeout: Duration = timedelta(seconds=60),
         retries: int = 0,
         rate_limits: list[RateLimit] = [],
         desired_worker_labels: dict[str, DesiredWorkerLabel] = {},
@@ -475,8 +474,8 @@ class Hatchet:
         sticky: StickyStrategy | None = None,
         default_priority: int = 1,
         concurrency: ConcurrencyExpression | None = None,
-        schedule_timeout: Duration = DEFAULT_SCHEDULE_TIMEOUT,
-        execution_timeout: Duration = DEFAULT_EXECUTION_TIMEOUT,
+        schedule_timeout: Duration = timedelta(minutes=5),
+        execution_timeout: Duration = timedelta(seconds=60),
         retries: int = 0,
         rate_limits: list[RateLimit] = [],
         desired_worker_labels: dict[str, DesiredWorkerLabel] = {},
@@ -498,8 +497,8 @@ class Hatchet:
         sticky: StickyStrategy | None = None,
         default_priority: int = 1,
         concurrency: ConcurrencyExpression | None = None,
-        schedule_timeout: Duration = DEFAULT_SCHEDULE_TIMEOUT,
-        execution_timeout: Duration = DEFAULT_EXECUTION_TIMEOUT,
+        schedule_timeout: Duration = timedelta(minutes=5),
+        execution_timeout: Duration = timedelta(seconds=60),
         retries: int = 0,
         rate_limits: list[RateLimit] = [],
         desired_worker_labels: dict[str, DesiredWorkerLabel] = {},

hatchet_sdk/runnables/standalone.py

@@ -1,4 +1,5 @@
-from datetime import datetime
+import asyncio
+from datetime import datetime, timedelta
 from typing import Any, Generic, cast, get_type_hints
 
 from hatchet_sdk.clients.admin import (
@@ -7,7 +8,10 @@ from hatchet_sdk.clients.admin import (
     WorkflowRunTriggerConfig,
 )
 from hatchet_sdk.clients.rest.models.cron_workflows import CronWorkflows
+from hatchet_sdk.clients.rest.models.v1_task_status import V1TaskStatus
+from hatchet_sdk.clients.rest.models.v1_task_summary import V1TaskSummary
 from hatchet_sdk.contracts.workflows_pb2 import WorkflowVersion
+from hatchet_sdk.logger import logger
 from hatchet_sdk.runnables.task import Task
 from hatchet_sdk.runnables.types import EmptyModel, R, TWorkflowInput
 from hatchet_sdk.runnables.workflow import BaseWorkflow, Workflow
@@ -294,3 +298,88 @@ class Standalone(BaseWorkflow[TWorkflowInput], Generic[TWorkflowInput, R]):
 
     def to_task(self) -> Task[TWorkflowInput, R]:
         return self._task
+
+    def list_runs(
+        self,
+        since: datetime | None = None,
+        until: datetime | None = None,
+        limit: int = 100,
+        offset: int | None = None,
+        statuses: list[V1TaskStatus] | None = None,
+        additional_metadata: dict[str, str] | None = None,
+        worker_id: str | None = None,
+        parent_task_external_id: str | None = None,
+    ) -> list[V1TaskSummary]:
+        """
+        List runs of the workflow.
+
+        :param since: The start time for the runs to be listed.
+        :param until: The end time for the runs to be listed.
+        :param limit: The maximum number of runs to be listed.
+        :param offset: The offset for pagination.
+        :param statuses: The statuses of the runs to be listed.
+        :param additional_metadata: Additional metadata for filtering the runs.
+        :param worker_id: The ID of the worker that ran the tasks.
+        :param parent_task_external_id: The external ID of the parent task.
+
+        :returns: A list of `V1TaskSummary` objects representing the runs of the workflow.
+        """
+        workflows = self.client.workflows.list(workflow_name=self._workflow.name)
+
+        if not workflows.rows:
+            logger.warning(f"No runs found for {self.name}")
+            return []
+
+        workflow = workflows.rows[0]
+
+        response = self.client.runs.list(
+            workflow_ids=[workflow.metadata.id],
+            since=since or datetime.now() - timedelta(days=1),
+            only_tasks=True,
+            offset=offset,
+            limit=limit,
+            statuses=statuses,
+            until=until,
+            additional_metadata=additional_metadata,
+            worker_id=worker_id,
+            parent_task_external_id=parent_task_external_id,
+        )
+
+        return response.rows
+
+    async def aio_list_runs(
+        self,
+        since: datetime | None = None,
+        until: datetime | None = None,
+        limit: int = 100,
+        offset: int | None = None,
+        statuses: list[V1TaskStatus] | None = None,
+        additional_metadata: dict[str, str] | None = None,
+        worker_id: str | None = None,
+        parent_task_external_id: str | None = None,
+    ) -> list[V1TaskSummary]:
+        """
+        List runs of the workflow.
+
+        :param since: The start time for the runs to be listed.
+        :param until: The end time for the runs to be listed.
+        :param limit: The maximum number of runs to be listed.
+        :param offset: The offset for pagination.
+        :param statuses: The statuses of the runs to be listed.
+        :param additional_metadata: Additional metadata for filtering the runs.
+        :param worker_id: The ID of the worker that ran the tasks.
+        :param parent_task_external_id: The external ID of the parent task.
+
+        :returns: A list of `V1TaskSummary` objects representing the runs of the workflow.
+        """
+        return await asyncio.to_thread(
+            self.list_runs,
+            since=since or datetime.now() - timedelta(days=1),
+            offset=offset,
+            limit=limit,
+            statuses=statuses,
+            until=until,
+            additional_metadata=additional_metadata,
+            worker_id=worker_id,
+            parent_task_external_id=parent_task_external_id,
+        )

hatchet_sdk/runnables/task.py

@@ -1,10 +1,10 @@
+from datetime import timedelta
 from typing import (
     TYPE_CHECKING,
     Any,
     Awaitable,
     Callable,
     Generic,
-    TypeVar,
     Union,
     cast,
     get_type_hints,
@@ -18,8 +18,6 @@ from hatchet_sdk.contracts.v1.workflows_pb2 import (
     DesiredWorkerLabels,
 )
 from hatchet_sdk.runnables.types import (
-    DEFAULT_EXECUTION_TIMEOUT,
-    DEFAULT_SCHEDULE_TIMEOUT,
     ConcurrencyExpression,
     R,
     StepType,
@@ -43,18 +41,6 @@ if TYPE_CHECKING:
     from hatchet_sdk.runnables.workflow import Workflow
 
 
-T = TypeVar("T")
-
-
-def fall_back_to_default(value: T, default: T, fallback_value: T) -> T:
-    ## If the value is not the default, it's set
-    if value != default:
-        return value
-
-    ## Otherwise, it's unset, so return the fallback value
-    return fallback_value
-
-
 class Task(Generic[TWorkflowInput, R]):
     def __init__(
         self,
@@ -68,8 +54,8 @@ class Task(Generic[TWorkflowInput, R]):
         type: StepType,
         workflow: "Workflow[TWorkflowInput]",
         name: str,
-        execution_timeout: Duration = DEFAULT_EXECUTION_TIMEOUT,
-        schedule_timeout: Duration = DEFAULT_SCHEDULE_TIMEOUT,
+        execution_timeout: Duration = timedelta(seconds=60),
+        schedule_timeout: Duration = timedelta(minutes=5),
         parents: "list[Task[TWorkflowInput, Any]]" = [],
         retries: int = 0,
         rate_limits: list[CreateTaskRateLimit] = [],
@@ -89,12 +75,8 @@ class Task(Generic[TWorkflowInput, R]):
         self.workflow = workflow
 
         self.type = type
-        self.execution_timeout = fall_back_to_default(
-            execution_timeout, DEFAULT_EXECUTION_TIMEOUT, DEFAULT_EXECUTION_TIMEOUT
-        )
-        self.schedule_timeout = fall_back_to_default(
-            schedule_timeout, DEFAULT_SCHEDULE_TIMEOUT, DEFAULT_SCHEDULE_TIMEOUT
-        )
+        self.execution_timeout = execution_timeout
+        self.schedule_timeout = schedule_timeout
         self.name = name
         self.parents = parents
         self.retries = retries

hatchet_sdk/runnables/types.py

@@ -1,9 +1,8 @@
 import asyncio
-from datetime import timedelta
 from enum import Enum
 from typing import Any, Awaitable, Callable, ParamSpec, Type, TypeGuard, TypeVar, Union
 
-from pydantic import BaseModel, ConfigDict, Field, StrictInt, model_validator
+from pydantic import BaseModel, ConfigDict, Field, model_validator
 
 from hatchet_sdk.context.context import Context, DurableContext
 from hatchet_sdk.contracts.v1.workflows_pb2 import Concurrency
@@ -16,11 +15,6 @@ R = TypeVar("R", bound=Union[ValidTaskReturnType, Awaitable[ValidTaskReturnType]
 P = ParamSpec("P")
 
 
-DEFAULT_EXECUTION_TIMEOUT = timedelta(seconds=60)
-DEFAULT_SCHEDULE_TIMEOUT = timedelta(minutes=5)
-DEFAULT_PRIORITY = 1
-
-
 class EmptyModel(BaseModel):
     model_config = ConfigDict(extra="allow", frozen=True)
 
@@ -65,9 +59,12 @@ TWorkflowInput = TypeVar("TWorkflowInput", bound=BaseModel)
 
 
 class TaskDefaults(BaseModel):
-    schedule_timeout: Duration = DEFAULT_SCHEDULE_TIMEOUT
-    execution_timeout: Duration = DEFAULT_EXECUTION_TIMEOUT
-    priority: StrictInt = Field(gt=0, lt=4, default=DEFAULT_PRIORITY)
+    schedule_timeout: Duration | None = None
+    execution_timeout: Duration | None = None
+    priority: int | None = Field(gt=0, lt=4, default=None)
+    retries: int | None = None
+    backoff_factor: float | None = None
+    backoff_max_seconds: int | None = None
 
 
 class WorkflowConfig(BaseModel):