test(examples): add wait and handler error test examples

- Add testing examples
  - wait for condition
  - handler error

- Update wait example with Duration change
- Update web runner, return error response so that we can test on funciton exception

Author: Alex Wang

.gitignore

@@ -29,6 +29,7 @@ dist/
 .kiro/
 .idea
 .env
+.env*
 
 .durable_executions
 

README.md

@@ -38,6 +38,8 @@ from durable_executions_python_language_sdk.context import (
     durable_with_child_context,
 )
 from durable_executions_python_language_sdk.execution import durable_execution
+from aws_durable_execution_sdk_python.config import Duration
+
 
 @durable_step
 def one(a: int, b: int) -> str:
@@ -68,7 +70,7 @@ def function_under_test(event: Any, context: DurableContext) -> list[str]:
     result_one: str = context.step(one(1, 2))
     results.append(result_one)
 
-    context.wait(seconds=1)
+    context.wait(Duration.from_seconds(1))
 
     result_two: str = context.run_in_child_context(two(3, 4))
     results.append(result_two)

examples/examples-catalog.json

@@ -297,6 +297,17 @@
         "ExecutionTimeout": 300
       },
       "path": "./src/parallel/parallel_with_batch_serdes.py"
+    },
+    {
+      "name": "Handler Error",
+      "description": "Simple function with handler error",
+      "handler": "handler_error.handler",
+      "integration": true,
+      "durableConfig": {
+        "RetentionPeriodInDays": 7,
+        "ExecutionTimeout": 300
+      },
+      "path": "./src/handler_error/handler_error.py"
     }
   ]
 }

examples/src/block_example/block_example.py

@@ -7,13 +7,14 @@
     durable_with_child_context,
 )
 from aws_durable_execution_sdk_python.execution import durable_execution
+from aws_durable_execution_sdk_python.config import Duration
 
 
 @durable_with_child_context
 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/handler_error/handler_error.py

@@ -0,0 +1,19 @@
+"""Demonstrates how handler-level errors are captured and structured in results."""
+
+from typing import Any
+
+from aws_durable_execution_sdk_python.context import DurableContext
+from aws_durable_execution_sdk_python.execution import durable_execution
+
+
+def generate_large_string(size_in_kb: int) -> str:
+    """Generate a string of approximately the specified size in KB."""
+    base_string = "B" * 1024  # 1KB string
+    return base_string * size_in_kb
+
+
+@durable_execution
+def handler(_event: Any, _context: DurableContext) -> None:
+    """Handler demonstrating handler-level error capture."""
+    # Simulate a handler-level error that might occur in real applications
+    raise Exception("Intentional handler failure")

examples/src/run_in_child_context/run_in_child_context_large_data.py

@@ -7,6 +7,7 @@
     durable_with_child_context,
 )
 from aws_durable_execution_sdk_python.execution import durable_execution
+from aws_durable_execution_sdk_python.config import Duration
 
 
 def generate_large_string(size_in_kb: int) -> str:
@@ -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/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 StepConfig, Duration
 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

@@ -4,13 +4,14 @@
 
 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.config import Duration
 
 
 @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

@@ -2,9 +2,10 @@
 
 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.config import Duration
 
 
 @durable_execution
 def handler(_event: Any, context: DurableContext) -> str:
-    context.wait(seconds=5)
+    context.wait(Duration.from_seconds(5))
     return "Wait completed"

examples/src/wait/wait_with_name.py

@@ -2,10 +2,11 @@
 
 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.config import Duration
 
 
 @durable_execution
 def handler(_event: Any, context: DurableContext) -> str:
     # Wait with explicit name
-    context.wait(seconds=2, name="custom_wait")
+    context.wait(Duration.from_seconds(2), name="custom_wait")
     return "Wait with name completed"