Change standard to default mode

Author: SamRemis

packages/smithy-core/src/smithy_core/retries.py

@@ -12,18 +12,18 @@
 from .interfaces import retries as retries_interface
 from .interfaces.retries import RetryStrategy
 
-RetryStrategyType = Literal["simple"]
+RetryStrategyType = Literal["simple", "standard"]
 
 
 @dataclass(kw_only=True, frozen=True)
 class RetryStrategyOptions:
     """Options for configuring retry behavior."""
 
-    retry_mode: RetryStrategyType = "simple"
+    retry_mode: RetryStrategyType = "standard"
     """The retry mode to use."""
 
-    max_attempts: int = 3
-    """Maximum number of attempts (initial attempt plus retries)."""
+    max_attempts: int | None = None
+    """Maximum number of attempts (initial attempt plus retries). If None, uses the strategy's default."""
 
 
 class RetryStrategyResolver:
@@ -44,11 +44,19 @@ async def resolve_retry_strategy(
 
     @lru_cache
     def _create_retry_strategy(
-        self, retry_mode: RetryStrategyType, max_attempts: int
+        self, retry_mode: RetryStrategyType, max_attempts: int | None
     ) -> RetryStrategy:
         match retry_mode:
             case "simple":
-                return SimpleRetryStrategy(max_attempts=max_attempts)
+                if max_attempts is None:
+                    return SimpleRetryStrategy()
+                else:
+                    return SimpleRetryStrategy(max_attempts=max_attempts)
+            case "standard":
+                if max_attempts is None:
+                    return StandardRetryStrategy()
+                else:
+                    return StandardRetryStrategy(max_attempts=max_attempts)
             case _:
                 raise ValueError(f"Unknown retry mode: {retry_mode}")
 

packages/smithy-core/tests/unit/test_retries.py

@@ -217,7 +217,7 @@ async def test_caching_retry_strategy_default_resolution() -> None:
 
     strategy = await resolver.resolve_retry_strategy(options=options)
 
-    assert isinstance(strategy, SimpleRetryStrategy)
+    assert isinstance(strategy, StandardRetryStrategy)
     assert strategy.max_attempts == 3