Update all examples to use Duration API

- Replace context.wait(seconds=N) with context.wait(Duration.from_seconds(N))
- Replace CallbackConfig(timeout_seconds=N) with CallbackConfig(timeout=Duration.from_seconds(N))
- Replace WaitForCallbackConfig timeout_seconds/heartbeat_timeout_seconds with Duration objects
- Replace RetryStrategyConfig initial_delay_seconds/max_delay_seconds with Duration objects
- Add Duration import to all affected files
- Aligns with Duration refactoring in main SDK

Author: FullyTyped

examples/src/block_example/block_example.py

@@ -2,6 +2,7 @@
 
 from typing import Any
 
+from aws_durable_execution_sdk_python.config import Duration
 from aws_durable_execution_sdk_python.context import (
     DurableContext,
     durable_with_child_context,
@@ -13,7 +14,7 @@
 def nested_block(ctx: DurableContext) -> str:
     """Nested block with its own child context."""
     # Wait in the nested block
-    ctx.wait(seconds=1)
+    ctx.wait(Duration.from_seconds(1))
     return "nested block result"
 
 

examples/src/callback/callback.py

@@ -1,6 +1,6 @@
 from typing import TYPE_CHECKING, Any
 
-from aws_durable_execution_sdk_python.config import CallbackConfig
+from aws_durable_execution_sdk_python.config import CallbackConfig, Duration
 from aws_durable_execution_sdk_python.context import DurableContext
 from aws_durable_execution_sdk_python.execution import durable_execution
 
@@ -11,7 +11,7 @@
 
 @durable_execution
 def handler(_event: Any, context: DurableContext) -> str:
-    callback_config = CallbackConfig(timeout_seconds=120, heartbeat_timeout_seconds=60)
+    callback_config = CallbackConfig(timeout=Duration.from_minutes(2), heartbeat_timeout=Duration.from_minutes(1))
 
     callback: Callback[str] = context.create_callback(
         name="example_callback", config=callback_config

examples/src/callback/callback_with_timeout.py

@@ -1,6 +1,6 @@
 from typing import TYPE_CHECKING, Any
 
-from aws_durable_execution_sdk_python.config import CallbackConfig
+from aws_durable_execution_sdk_python.config import CallbackConfig, Duration
 from aws_durable_execution_sdk_python.context import DurableContext
 from aws_durable_execution_sdk_python.execution import durable_execution
 
@@ -12,7 +12,7 @@
 @durable_execution
 def handler(_event: Any, context: DurableContext) -> str:
     # Callback with custom timeout configuration
-    config = CallbackConfig(timeout_seconds=60, heartbeat_timeout_seconds=30)
+    config = CallbackConfig(timeout=Duration.from_minutes(1), heartbeat_timeout=Duration.from_seconds(30))
 
     callback: Callback[str] = context.create_callback(
         name="timeout_callback", config=config

examples/src/parallel/parallel.py

@@ -2,7 +2,7 @@
 
 from typing import Any
 
-from aws_durable_execution_sdk_python.config import ParallelConfig
+from aws_durable_execution_sdk_python.config import Duration, ParallelConfig
 from aws_durable_execution_sdk_python.context import DurableContext
 from aws_durable_execution_sdk_python.execution import durable_execution
 
@@ -17,7 +17,7 @@ def handler(_event: Any, context: DurableContext) -> list[str]:
             lambda ctx: ctx.step(lambda _: "task 1 completed", name="task1"),
             lambda ctx: ctx.step(lambda _: "task 2 completed", name="task2"),
             lambda ctx: (
-                ctx.wait(1, name="wait_in_task3"),
+                ctx.wait(Duration.from_seconds(1), name="wait_in_task3"),
                 "task 3 completed after wait",
             )[1],
         ],

examples/src/parallel/parallel_with_wait.py

@@ -2,6 +2,7 @@
 
 from typing import Any
 
+from aws_durable_execution_sdk_python.config import Duration
 from aws_durable_execution_sdk_python.context import DurableContext
 from aws_durable_execution_sdk_python.execution import durable_execution
 
@@ -13,9 +14,9 @@ def handler(_event: Any, context: DurableContext) -> str:
     # Call get_results() to extract data and avoid BatchResult serialization
     context.parallel(
         functions=[
-            lambda ctx: ctx.wait(1, name="wait_1_second"),
-            lambda ctx: ctx.wait(2, name="wait_2_seconds"),
-            lambda ctx: ctx.wait(5, name="wait_5_seconds"),
+            lambda ctx: ctx.wait(Duration.from_seconds(1), name="wait_1_second"),
+            lambda ctx: ctx.wait(Duration.from_seconds(2), name="wait_2_seconds"),
+            lambda ctx: ctx.wait(Duration.from_seconds(5), name="wait_5_seconds"),
         ],
         name="parallel_waits",
     ).get_results()

examples/src/run_in_child_context/run_in_child_context_large_data.py

@@ -2,6 +2,7 @@
 
 from typing import Any
 
+from aws_durable_execution_sdk_python.config import Duration
 from aws_durable_execution_sdk_python.context import (
     DurableContext,
     durable_with_child_context,
@@ -55,7 +56,7 @@ def handler(_event: Any, context: DurableContext) -> dict[str, Any]:
     )
 
     # Add a wait after runInChildContext to test persistence across invocations
-    context.wait(seconds=1, name="post-processing-wait")
+    context.wait(Duration.from_seconds(1), name="post-processing-wait")
 
     # Verify the data is still intact after the wait
     data_integrity_check = (

examples/src/step/step_with_exponential_backoff.py

@@ -1,6 +1,6 @@
 from typing import Any
 
-from aws_durable_execution_sdk_python.config import StepConfig
+from aws_durable_execution_sdk_python.config import Duration, StepConfig
 from aws_durable_execution_sdk_python.context import DurableContext
 from aws_durable_execution_sdk_python.execution import durable_execution
 from aws_durable_execution_sdk_python.retries import (
@@ -13,7 +13,7 @@
 def handler(_event: Any, context: DurableContext) -> str:
     # Step with exponential backoff retry strategy
     retry_config = RetryStrategyConfig(
-        max_attempts=3, initial_delay_seconds=1, max_delay_seconds=10, backoff_rate=2.0
+        max_attempts=3, initial_delay=Duration.from_seconds(1), max_delay=Duration.from_seconds(10), backoff_rate=2.0
     )
 
     step_config = StepConfig(retry_strategy=create_retry_strategy(retry_config))

examples/src/step/steps_with_retry.py

@@ -3,7 +3,7 @@
 from random import random
 from typing import Any
 
-from aws_durable_execution_sdk_python.config import StepConfig
+from aws_durable_execution_sdk_python.config import Duration, StepConfig
 from aws_durable_execution_sdk_python.context import DurableContext
 from aws_durable_execution_sdk_python.execution import durable_execution
 from aws_durable_execution_sdk_python.retries import (
@@ -60,7 +60,7 @@ def handler(event: Any, context: DurableContext) -> dict[str, Any]:
                 break
 
             # Wait 1 second until next poll
-            context.wait(seconds=1)
+            context.wait(Duration.from_seconds(1))
 
     except RuntimeError as e:
         # Retries exhausted

examples/src/wait/multiple_wait.py

@@ -2,15 +2,16 @@
 
 from typing import Any
 
+from aws_durable_execution_sdk_python.config import Duration
 from aws_durable_execution_sdk_python.context import DurableContext
 from aws_durable_execution_sdk_python.execution import durable_execution
 
 
 @durable_execution
 def handler(_event: Any, context: DurableContext) -> dict[str, Any]:
     """Handler demonstrating multiple sequential wait operations."""
-    context.wait(seconds=5, name="wait-1")
-    context.wait(seconds=5, name="wait-2")
+    context.wait(Duration.from_seconds(5), name="wait-1")
+    context.wait(Duration.from_seconds(5), name="wait-2")
 
     return {
         "completedWaits": 2,

examples/src/wait/wait.py

@@ -1,10 +1,11 @@
 from typing import Any
 
+from aws_durable_execution_sdk_python.config import Duration
 from aws_durable_execution_sdk_python.context import DurableContext
 from aws_durable_execution_sdk_python.execution import durable_execution
 
 
 @durable_execution
 def handler(_event: Any, context: DurableContext) -> str:
-    context.wait(seconds=5)
+    context.wait(Duration.from_seconds(5))
     return "Wait completed"