Add wait and retry parameters to TenacityConfig (#3651)

Author: juliusgeo

sdks/python/CHANGELOG.md

@@ -5,6 +5,12 @@ All notable changes to Hatchet's Python SDK will be documented in this changelog
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.33.0] - 2026-04-16
+
+### Changed
+
+- Adds `wait` and `before_sleep` parameters to `TenacityConfig` to allow custom retry strategies and retry callbacks.
+
 ## [1.32.3] - 2026-04-16
 
 ### Changed

sdks/python/hatchet_sdk/clients/dispatcher/dispatcher.py

@@ -9,11 +9,10 @@
 
 from hatchet_sdk.clients.dispatcher.action_listener import ActionListener
 from hatchet_sdk.clients.rest.tenacity_utils import (
-    tenacity_alert_retry,
     tenacity_retry,
     tenacity_should_retry,
 )
-from hatchet_sdk.config import ClientConfig, TenacityConfig
+from hatchet_sdk.config import ClientConfig, TenacityConfig, tenacity_before_sleep
 from hatchet_sdk.connection import new_conn
 from hatchet_sdk.contracts.dispatcher_pb2 import (
     SDKS,
@@ -105,7 +104,7 @@ async def get_action_listener(
         reraise=True,
         wait=tenacity.wait_exponential_jitter(initial=0.5, max=5),
         stop=tenacity.stop_after_attempt(3),
-        before_sleep=tenacity_alert_retry,
+        before_sleep=tenacity_before_sleep,
         retry=tenacity.retry_if_exception(tenacity_should_retry),
     )
     async def get_version(self) -> str | None:

sdks/python/hatchet_sdk/clients/rest/tenacity_utils.py

@@ -12,7 +12,6 @@
     ServiceException,
     TooManyRequestsException,
 )
-from hatchet_sdk.logger import logger
 
 if TYPE_CHECKING:
     from hatchet_sdk.config import TenacityConfig
@@ -30,21 +29,13 @@ def should_retry(ex: BaseException) -> bool:
 
     return tenacity.retry(
         reraise=True,
-        wait=tenacity.wait_exponential_jitter(),
+        wait=config.wait(),
         stop=tenacity.stop_after_attempt(config.max_attempts),
-        before_sleep=tenacity_alert_retry,
+        before_sleep=config.before_sleep,
         retry=tenacity.retry_if_exception(should_retry),
     )(func)
 
 
-def tenacity_alert_retry(retry_state: tenacity.RetryCallState) -> None:
-    """Called between tenacity retries."""
-    logger.debug(
-        f"retrying {retry_state.fn}: attempt "
-        f"{retry_state.attempt_number} ended with: {retry_state.outcome}",
-    )
-
-
 def tenacity_should_retry(
     ex: BaseException, config: TenacityConfig | None = None
 ) -> bool:

sdks/python/hatchet_sdk/config.py

@@ -1,12 +1,15 @@
 import json
+from collections.abc import Callable
 from datetime import timedelta
 from enum import Enum
 from logging import Logger, getLogger
 from typing import overload
 
+import tenacity
 from pydantic import Field, field_validator, model_validator
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
+from hatchet_sdk.logger import logger
 from hatchet_sdk.token import get_addresses_from_jwt, get_tenant_id_from_jwt
 from hatchet_sdk.utils.opentelemetry import OTelAttribute
 
@@ -96,6 +99,14 @@ class HTTPMethod(str, Enum):
     OPTIONS = "OPTIONS"
 
 
+def tenacity_before_sleep(retry_state: tenacity.RetryCallState) -> None:
+    """Called between tenacity retries."""
+    logger.debug(
+        f"retrying {retry_state.fn}: attempt "
+        f"{retry_state.attempt_number} ended with: {retry_state.outcome}",
+    )
+
+
 class TenacityConfig(BaseSettings):
     model_config = create_settings_config(
         env_prefix="HATCHET_CLIENT_TENACITY_",
@@ -115,6 +126,8 @@ class TenacityConfig(BaseSettings):
         default_factory=lambda: [HTTPMethod.GET, HTTPMethod.DELETE],
         description="HTTP methods to retry on transport errors when retry_transport_errors is enabled; excludes POST/PUT/PATCH by default due to idempotency concerns.",
     )
+    wait: type[tenacity.wait.wait_base] = tenacity.wait_exponential_jitter
+    before_sleep: Callable[[tenacity.RetryCallState], None] = tenacity_before_sleep
 
 
 DEFAULT_HOST_PORT = "localhost:7070"

sdks/python/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hatchet-sdk"
-version = "1.32.3"
+version = "1.33.0"
 description = "This is the official Python SDK for Hatchet, a distributed, fault-tolerant task queue. The SDK allows you to easily integrate Hatchet's task scheduling and workflow orchestration capabilities into your Python applications."
 readme = "README.md"
 license = { text = "MIT" }