aws-durable-execution-sdk-python-testing 1.0.0.post1__tar.gz → 1.0.0.post2__tar.gz

{aws_durable_execution_sdk_python_testing-1.0.0.post1 → aws_durable_execution_sdk_python_testing-1.0.0.post2}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: aws-durable-execution-sdk-python-testing
-Version: 1.0.0.post1
-Summary: This is the Python SDK for AWS Lambda Durable Execution.
+Version: 1.0.0.post2
+Summary: AWS Durable Execution Testing SDK for Python
 Project-URL: Documentation, https://github.com/aws/aws-durable-execution-sdk-python-testing#readme
 Project-URL: Issues, https://github.com/aws/aws-durable-execution-sdk-python-testing/issues
 Project-URL: Source, https://github.com/aws/aws-durable-execution-sdk-python-testing

{aws_durable_execution_sdk_python_testing-1.0.0.post1 → aws_durable_execution_sdk_python_testing-1.0.0.post2}/aws_durable_execution_sdk_python_testing/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2025-present Amazon.com, Inc. or its affiliates.
 #
 # SPDX-License-Identifier: Apache-2.0
-__version__ = "1.0.0.post1"
+__version__ = "1.0.0.post2"

{aws_durable_execution_sdk_python_testing-1.0.0.post1 → aws_durable_execution_sdk_python_testing-1.0.0.post2}/aws_durable_execution_sdk_python_testing/executor.py

@@ -90,6 +90,7 @@ class Executor(ExecutionObserver):
         self._completion_events: dict[str, Event] = {}
         self._callback_timeouts: dict[str, Future] = {}
         self._callback_heartbeats: dict[str, Future] = {}
+        self._execution_timeout: Future | None = None
 
     def start_execution(
         self,
@@ -118,6 +119,21 @@ class Executor(ExecutionObserver):
         completion_event = self._scheduler.create_event()
         self._completion_events[execution.durable_execution_arn] = completion_event
 
+        # Schedule execution timeout
+        if input.execution_timeout_seconds > 0:
+
+            def timeout_handler():
+                error = ErrorObject.from_message(
+                    f"Execution timed out after {input.execution_timeout_seconds} seconds."
+                )
+                self.on_timed_out(execution.durable_execution_arn, error)
+
+            self._execution_timeout = self._scheduler.call_later(
+                timeout_handler,
+                delay=input.execution_timeout_seconds,
+                completion_event=completion_event,
+            )
+
         # Schedule initial invocation to run immediately
         self._invoke_execution(execution.durable_execution_arn)
 
@@ -897,6 +913,9 @@ class Executor(ExecutionObserver):
         # complete doesn't actually checkpoint explicitly
         if event := self._completion_events.get(execution_arn):
             event.set()
+        if self._execution_timeout:
+            self._execution_timeout.cancel()
+            self._execution_timeout = None
 
     def wait_until_complete(
         self, execution_arn: str, timeout: float | None = None

{aws_durable_execution_sdk_python_testing-1.0.0.post1 → aws_durable_execution_sdk_python_testing-1.0.0.post2}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "hatchling.build"
 [project]
 name = "aws-durable-execution-sdk-python-testing"
 dynamic = ["version"]
-description = 'This is the Python SDK for AWS Lambda Durable Execution.'
+description = 'AWS Durable Execution Testing SDK for Python'
 readme = "README.md"
 requires-python = ">=3.11"
 license = "Apache-2.0"