chore: Update Suspension models

Updates:
 - WaitOptions
 - CallbackOptions
 - ChainedInvoke Options

Author: FullyTyped

src/aws_durable_execution_sdk_python/lambda_service.py

@@ -241,18 +241,41 @@ def to_dict(self) -> MutableMapping[str, Any]:
 
 @dataclass(frozen=True)
 class WaitOptions:
-    wait_seconds: int = 0
+    """
+    Wait Options provides details regarding suspension.
+
+    As of 2025/10/27:
+
+    - `wait_seconds` accepts values between 1, and 31622400
+    - When wait_second seconds does not exist,then we default to 1
+
+    """
+
+    wait_seconds: int = 1
 
     @classmethod
     def from_dict(cls, data: MutableMapping[str, Any]) -> WaitOptions:
-        return cls(wait_seconds=data.get("WaitSeconds", 0))
+        return cls(wait_seconds=data.get("WaitSeconds", 1))
 
     def to_dict(self) -> MutableMapping[str, Any]:
         return {"WaitSeconds": self.wait_seconds}
 
 
 @dataclass(frozen=True)
 class CallbackOptions:
+    """
+    Callback options provides details about the callback, wrt timeout
+    and heartbeat checks.
+
+    As of 2025/10/27:
+    - When timeout_seconds == 0, then the callback has no timeout
+    - When heartbeat_timeout_seconds == 0, then the callback has no timeout
+
+    - When timeout_seconds is not present, then default is 0
+    - When heartbeat_timeout_seconds, then default is 0
+
+    """
+
     timeout_seconds: TimeoutSeconds = 0
     heartbeat_timeout_seconds: int = 0
 
@@ -272,20 +295,22 @@ def to_dict(self) -> MutableMapping[str, Any]:
 
 @dataclass(frozen=True)
 class ChainedInvokeOptions:
+    """
+    As of 2025/10/27:
+     - Chained invoke options only contains a function name
+    """
+
     function_name: str
-    timeout_seconds: TimeoutSeconds = 0
 
     @classmethod
     def from_dict(cls, data: MutableMapping[str, Any]) -> ChainedInvokeOptions:
         return cls(
             function_name=data["FunctionName"],
-            timeout_seconds=data.get("TimeoutSeconds", 0),
         )
 
     def to_dict(self) -> MutableMapping[str, Any]:
         result: MutableMapping[str, Any] = {
             "FunctionName": self.function_name,
-            "TimeoutSeconds": self.timeout_seconds,
         }
         return result
 

src/aws_durable_execution_sdk_python/operation/invoke.py

@@ -87,9 +87,7 @@ def invoke_handler(
     start_operation: OperationUpdate = OperationUpdate.create_invoke_start(
         identifier=operation_identifier,
         payload=serialized_payload,
-        chained_invoke_options=ChainedInvokeOptions(
-            function_name=function_name, timeout_seconds=config.timeout_seconds
-        ),
+        chained_invoke_options=ChainedInvokeOptions(function_name=function_name),
     )
 
     state.create_checkpoint(operation_update=start_operation)

tests/lambda_service_test.py

@@ -402,15 +402,13 @@ def test_invoke_options_from_dict():
     data = {"FunctionName": "test-function", "TimeoutSeconds": 120}
     options = ChainedInvokeOptions.from_dict(data)
     assert options.function_name == "test-function"
-    assert options.timeout_seconds == 120
 
 
 def test_invoke_options_from_dict_required_only():
     """Test ChainedInvokeOptions.from_dict with only required field."""
     data = {"FunctionName": "test-function"}
     options = ChainedInvokeOptions.from_dict(data)
     assert options.function_name == "test-function"
-    assert options.timeout_seconds == 0
 
 
 def test_context_options_from_dict():
@@ -444,7 +442,7 @@ def test_callback_options_roundtrip():
 
 def test_invoke_options_roundtrip():
     """Test ChainedInvokeOptions to_dict -> from_dict roundtrip."""
-    original = ChainedInvokeOptions(function_name="test-func", timeout_seconds=120)
+    original = ChainedInvokeOptions(function_name="test-func")
     data = original.to_dict()
     restored = ChainedInvokeOptions.from_dict(data)
     assert restored == original
@@ -498,12 +496,10 @@ def test_invoke_options_to_dict():
     """Test ChainedInvokeOptions.to_dict method."""
     options = ChainedInvokeOptions(
         function_name="test_function",
-        timeout_seconds=30,
     )
     result = options.to_dict()
     expected = {
         "FunctionName": "test_function",
-        "TimeoutSeconds": 30,
     }
     assert result == expected
 
@@ -512,7 +508,7 @@ def test_invoke_options_to_dict_minimal():
     """Test ChainedInvokeOptions.to_dict with minimal fields."""
     options = ChainedInvokeOptions(function_name="test_function")
     result = options.to_dict()
-    assert result == {"FunctionName": "test_function", "TimeoutSeconds": 0}
+    assert result == {"FunctionName": "test_function"}
 
 
 def test_context_options_to_dict():
@@ -546,12 +542,11 @@ def test_invoke_options_from_dict_missing_function_name():
 
 def test_invoke_options_to_dict_complete():
     """Test ChainedInvokeOptions.to_dict with all fields."""
-    options = ChainedInvokeOptions(function_name="test_func", timeout_seconds=120)
+    options = ChainedInvokeOptions(function_name="test_func")
 
     result = options.to_dict()
 
     assert result["FunctionName"] == "test_func"
-    assert result["TimeoutSeconds"] == 120
 
 
 # =============================================================================
@@ -562,7 +557,7 @@ def test_invoke_options_to_dict_complete():
 def test_operation_update_create_invoke_start():
     """Test OperationUpdate.create_invoke_start method to cover line 545."""
     identifier = OperationIdentifier("test-id", "parent-id")
-    invoke_options = ChainedInvokeOptions("test-func", 120)
+    invoke_options = ChainedInvokeOptions("test-func")
     update = OperationUpdate.create_invoke_start(identifier, "payload", invoke_options)
     assert update.operation_id == "test-id"
 
@@ -609,9 +604,7 @@ def test_operation_update_to_dict_complete():
     callback_options = CallbackOptions(
         timeout_seconds=300, heartbeat_timeout_seconds=60
     )
-    chained_invoke_options = ChainedInvokeOptions(
-        function_name="test_func", timeout_seconds=60
-    )
+    chained_invoke_options = ChainedInvokeOptions(function_name="test_func")
 
     update = OperationUpdate(
         operation_id="op1",
@@ -639,7 +632,7 @@ def test_operation_update_to_dict_complete():
         "StepOptions": {"NextAttemptDelaySeconds": 30},
         "WaitOptions": {"WaitSeconds": 60},
         "CallbackOptions": {"TimeoutSeconds": 300, "HeartbeatTimeoutSeconds": 60},
-        "ChainedInvokeOptions": {"FunctionName": "test_func", "TimeoutSeconds": 60},
+        "ChainedInvokeOptions": {"FunctionName": "test_func"},
     }
     assert result == expected
 
@@ -884,9 +877,7 @@ def test_operation_update_complete_with_new_fields():
     callback_options = CallbackOptions(
         timeout_seconds=300, heartbeat_timeout_seconds=60
     )
-    chained_invoke_options = ChainedInvokeOptions(
-        function_name="test_func", timeout_seconds=60
-    )
+    chained_invoke_options = ChainedInvokeOptions(function_name="test_func")
 
     update = OperationUpdate(
         operation_id="op1",
@@ -918,7 +909,7 @@ def test_operation_update_complete_with_new_fields():
         "StepOptions": {"NextAttemptDelaySeconds": 30},
         "WaitOptions": {"WaitSeconds": 60},
         "CallbackOptions": {"TimeoutSeconds": 300, "HeartbeatTimeoutSeconds": 60},
-        "ChainedInvokeOptions": {"FunctionName": "test_func", "TimeoutSeconds": 60},
+        "ChainedInvokeOptions": {"FunctionName": "test_func"},
     }
     assert result == expected
 

tests/operation/invoke_test.py

@@ -247,7 +247,6 @@ def test_invoke_handler_new_operation():
     assert operation_update.name == "test_invoke"
     assert operation_update.payload == json.dumps("test_input")
     assert operation_update.chained_invoke_options.function_name == "test_function"
-    assert operation_update.chained_invoke_options.timeout_seconds == 60
 
 
 def test_invoke_handler_new_operation_with_timeout():
@@ -309,7 +308,10 @@ def test_invoke_handler_no_config():
 
     # Verify default config was used
     operation_update = mock_state.create_checkpoint.call_args[1]["operation_update"]
-    assert operation_update.chained_invoke_options.timeout_seconds == 0
+    assert (
+        operation_update.to_dict()["ChainedInvokeOptions"]["FunctionName"]
+        == "test_function"
+    )
 
 
 def test_invoke_handler_custom_serdes():