aws-durable-execution-sdk-python 1.1.2__py3-none-any.whl → 1.3.0__py3-none-any.whl

aws_durable_execution_sdk_python/__about__.py

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

aws_durable_execution_sdk_python/__init__.py

@@ -2,6 +2,8 @@
 
 # Main context - used in every durable function
 # Helper decorators - commonly used for step functions
+# Concurrency
+from aws_durable_execution_sdk_python.concurrency.models import BatchResult
 from aws_durable_execution_sdk_python.context import (
     DurableContext,
     durable_step,
@@ -20,7 +22,7 @@ from aws_durable_execution_sdk_python.exceptions import (
 from aws_durable_execution_sdk_python.execution import durable_execution
 
 # Essential context types - passed to user functions
-from aws_durable_execution_sdk_python.types import BatchResult, StepContext
+from aws_durable_execution_sdk_python.types import StepContext
 
 __all__ = [
     "BatchResult",

aws_durable_execution_sdk_python/context.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import hashlib
 import logging
+from dataclasses import dataclass
 from typing import TYPE_CHECKING, Any, Concatenate, Generic, ParamSpec, TypeVar
 
 from aws_durable_execution_sdk_python.config import (
@@ -74,6 +75,20 @@ logger = logging.getLogger(__name__)
 PASS_THROUGH_SERDES: SerDes[Any] = PassThroughSerDes()
 
 
+@dataclass(frozen=True)
+class ExecutionContext:
+    """Readonly metadata about the current durable execution context.
+
+    This class provides immutable access to execution-level metadata.
+
+    Attributes:
+        durable_execution_arn: The Amazon Resource Name (ARN) of the current
+            durable execution.
+    """
+
+    durable_execution_arn: str
+
+
 def durable_step(
     func: Callable[Concatenate[StepContext, Params], T],
 ) -> Callable[Params, Callable[[StepContext], T]]:
@@ -218,11 +233,13 @@ class DurableContext(DurableContextProtocol):
     def __init__(
         self,
         state: ExecutionState,
+        execution_context: ExecutionContext,
         lambda_context: LambdaContext | None = None,
         parent_id: str | None = None,
         logger: Logger | None = None,
     ) -> None:
         self.state: ExecutionState = state
+        self.execution_context: ExecutionContext = execution_context
         self.lambda_context = lambda_context
         self._parent_id: str | None = parent_id
         self._step_counter: OrderedCounter = OrderedCounter()
@@ -245,6 +262,9 @@ class DurableContext(DurableContextProtocol):
     ):
         return DurableContext(
             state=state,
+            execution_context=ExecutionContext(
+                durable_execution_arn=state.durable_execution_arn
+            ),
             lambda_context=lambda_context,
             parent_id=None,
         )
@@ -254,6 +274,7 @@ class DurableContext(DurableContextProtocol):
         logger.debug("Creating child context for parent %s", parent_id)
         return DurableContext(
             state=self.state,
+            execution_context=self.execution_context,
             lambda_context=self.lambda_context,
             parent_id=parent_id,
             logger=self.logger.with_log_info(

aws_durable_execution_sdk_python/logger.py

@@ -72,7 +72,7 @@ class Logger(LoggerInterface):
             # Use 'operation_name' instead of 'name' as key because the stdlib LogRecord internally reserved 'name' parameter
             extra["operationName"] = info.name
         if info.attempt is not None:
-            extra["attempt"] = info.attempt + 1
+            extra["attempt"] = info.attempt
         if info.operation_id:
             extra["operationId"] = info.operation_id
         return cls(

aws_durable_execution_sdk_python/operation/step.py

@@ -202,9 +202,10 @@ class StepOperationExecutor(OperationExecutor[T]):
             ExecutionError: For fatal errors that should not be retried
             May raise other exceptions that will be handled by retry_handler
         """
-        attempt: int = 0
+        # Get current attempt - checkpointed attempts + 1
+        attempt: int = 1
         if checkpointed_result.operation and checkpointed_result.operation.step_details:
-            attempt = checkpointed_result.operation.step_details.attempt
+            attempt = checkpointed_result.operation.step_details.attempt + 1
 
         step_context: StepContext = StepContext(
             logger=self.context_logger.with_log_info(

aws_durable_execution_sdk_python/operation/wait_for_condition.py

@@ -170,10 +170,11 @@ class WaitForConditionOperationExecutor(OperationExecutor[T]):
         else:
             current_state = self.config.initial_state
 
-        # Get attempt number
+        # Get attempt number - current attempt is checkpointed attempts + 1
+        # The checkpoint stores completed attempts, so the current attempt being executed is one more
         attempt: int = 1
         if checkpointed_result.operation and checkpointed_result.operation.step_details:
-            attempt = checkpointed_result.operation.step_details.attempt
+            attempt = checkpointed_result.operation.step_details.attempt + 1
 
         try:
             # Execute the check function with the injected logger

aws_durable_execution_sdk_python/state.py

@@ -45,12 +45,12 @@ class CheckpointBatcherConfig:
     Attributes:
         max_batch_size_bytes: Maximum batch size in bytes (default: 750KB)
         max_batch_time_seconds: Maximum time to wait before flushing batch (default: 1.0 second)
-        max_batch_operations: Maximum number of operations per batch (default: unlimited)
+        max_batch_operations: Maximum number of operations per batch (default: 250)
     """
 
-    max_batch_size_bytes: int = 750 * 1024  # 750KB - private readonly MAX_PAYLOAD_SIZE
-    max_batch_time_seconds: float = 1.0  # 1 second default
-    max_batch_operations: int | float = float("inf")  # No operation limit by default
+    max_batch_size_bytes: int = 750 * 1024  # 750KB
+    max_batch_time_seconds: float = 1.0
+    max_batch_operations: int = 250
 
 
 @dataclass(frozen=True)

{aws_durable_execution_sdk_python-1.1.2.dist-info → aws_durable_execution_sdk_python-1.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aws-durable-execution-sdk-python
-Version: 1.1.2
+Version: 1.3.0
 Summary: AWS Durable Execution SDK for Python
 Project-URL: Documentation, https://github.com/aws/aws-durable-execution-sdk-python#readme
 Project-URL: Issues, https://github.com/aws/aws-durable-execution-sdk-python/issues

{aws_durable_execution_sdk_python-1.1.2.dist-info → aws_durable_execution_sdk_python-1.3.0.dist-info}/RECORD

@@ -1,17 +1,17 @@
 aws_durable_execution_sdk_python/.gitignore,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aws_durable_execution_sdk_python/__about__.py,sha256=7qzc63t8OUhHP9YvXrUIRIK7LxcEvXL60VN7Z4b_y4M,137
-aws_durable_execution_sdk_python/__init__.py,sha256=reI6AQTpEyeK3ev6iZJuu7A6PSoa0V1VJGOgSB2RpGM,1023
+aws_durable_execution_sdk_python/__about__.py,sha256=PF_oYcM9RTUC3P846c3d27JU5o-gxm6cwcu_mFzcSDo,137
+aws_durable_execution_sdk_python/__init__.py,sha256=cPHNbPu8u7DLEotMt6x85AmM2tEGN4rUIK_R-FQhV0g,1100
 aws_durable_execution_sdk_python/config.py,sha256=vd_8IZO5KZU3qWENbWDVtyLglqw3sa6SU-R8CYUgm9g,18938
-aws_durable_execution_sdk_python/context.py,sha256=efJIc5KrsTzcI_azMjxqrHiPt71PuiQy4vbdsUiTtLI,22467
+aws_durable_execution_sdk_python/context.py,sha256=HwQn2ByStVxRErox23SqqwsRHjvKY439yNESPar9H2A,23154
 aws_durable_execution_sdk_python/exceptions.py,sha256=AvuXnBHAijXb3TepNTu5CVDb_cpR6Yw6x4W3jELVneQ,13631
 aws_durable_execution_sdk_python/execution.py,sha256=A20fOp6_wXdIcL1YiqJ_RGJbQ09D1lSivBQnUY0FT3Y,18571
 aws_durable_execution_sdk_python/identifier.py,sha256=0NyNTb8mTrUni7PlmFtzvJdW6lk0_XpuAaL89LNc378,324
 aws_durable_execution_sdk_python/lambda_service.py,sha256=FkkcmFxDZLA4JS7mMW2YT9BfqTNHHJtEk9kSADwu5bk,37858
-aws_durable_execution_sdk_python/logger.py,sha256=nde1fhe9uK-UmXRZI7T_sIoU-_T5PnZo16boaFQFtyw,4281
+aws_durable_execution_sdk_python/logger.py,sha256=DtPNfJS9PFUfeJqHTl34rSxYRizvVVWAtblV-5zpuY4,4277
 aws_durable_execution_sdk_python/py.typed,sha256=DbCDcAirv769HS1ibzZLXq8VPazYf7JCkKSzj3ctudM,58
 aws_durable_execution_sdk_python/retries.py,sha256=CNJ6msQHVhcdSjMuuQAvR0WvsaN7cFA7QgQTVI8pK2w,5829
 aws_durable_execution_sdk_python/serdes.py,sha256=-hzqPyAHi1rsOUwpi7RjiJBNiDc0Oe5_QVNMemYBfGY,16808
-aws_durable_execution_sdk_python/state.py,sha256=5kWSwtOCt0VkhhB2orhGyrXNg1nBt-HXfXm0T8VzAiE,33303
+aws_durable_execution_sdk_python/state.py,sha256=aJOB575jTZ1cCVNpLUMJTynj3twwlhAbkwMRduHloww,33191
 aws_durable_execution_sdk_python/suspend.py,sha256=PIgc6w5-diZIq58-tPv51dzBSHGW7e4oR0QFam8Vgi0,3323
 aws_durable_execution_sdk_python/threading.py,sha256=cU7vIEWHYco-qqkgoAMyPHSOQ8b_Q_r_rZGG7m1g7I4,7650
 aws_durable_execution_sdk_python/types.py,sha256=8vqwa5q2EikesX_NdhR_njin7l7uHakhWPH_ntHY_PI,5152
@@ -26,11 +26,11 @@ aws_durable_execution_sdk_python/operation/child.py,sha256=L2RMHhTt_xewAS20T1iZN
 aws_durable_execution_sdk_python/operation/invoke.py,sha256=R-31JLlazDWECe8Q242LMqiwuyg1_GzycDNKYJEoz_M,6801
 aws_durable_execution_sdk_python/operation/map.py,sha256=NMp-U3dvv2OK9kiUF1PQfkll1JKzpL7Sh_cxyda_Kyk,5024
 aws_durable_execution_sdk_python/operation/parallel.py,sha256=uzxa_XsTvimC63HpadwCjNhGoUcqxM8P4bvQLVHQTq0,4715
-aws_durable_execution_sdk_python/operation/step.py,sha256=5R-Rzmfh5uYHMQdq3WoDzwXUBwNg-ToeWW5DI7_l7zw,15567
+aws_durable_execution_sdk_python/operation/step.py,sha256=hnwiLtXNLAOl71FPduxyMYaUggQsrelLeP1W5kRXXlM,15629
 aws_durable_execution_sdk_python/operation/wait.py,sha256=P1R7-BrQ0eadFxQiGLHL3ySkuJUqPvVv7zewAzBeVT8,4193
-aws_durable_execution_sdk_python/operation/wait_for_condition.py,sha256=s33dEf5qosed_DxZiNN0A_t9DHfpHBpcfD7HqaEfbx0,11768
-aws_durable_execution_sdk_python-1.1.2.dist-info/METADATA,sha256=hYagEtD3Dl8MjSAH41gPAODHtYKb3Dyo3CNpm4mA9xs,6424
-aws_durable_execution_sdk_python-1.1.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-aws_durable_execution_sdk_python-1.1.2.dist-info/licenses/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-aws_durable_execution_sdk_python-1.1.2.dist-info/licenses/NOTICE,sha256=1CkO1kwu3Q_OHYTj-d-yiBJA_lNN73a4zSntavaD4oc,67
-aws_durable_execution_sdk_python-1.1.2.dist-info/RECORD,,
+aws_durable_execution_sdk_python/operation/wait_for_condition.py,sha256=9t-XsfKQfg66Ys57L61VWA5_JYPvVNnGD6ddFq-h4Xg,11921
+aws_durable_execution_sdk_python-1.3.0.dist-info/METADATA,sha256=jivJXjguOtgHqdOqSsVZhgu5zelPYEC85-tJKdY9fDA,6424
+aws_durable_execution_sdk_python-1.3.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+aws_durable_execution_sdk_python-1.3.0.dist-info/licenses/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+aws_durable_execution_sdk_python-1.3.0.dist-info/licenses/NOTICE,sha256=1CkO1kwu3Q_OHYTj-d-yiBJA_lNN73a4zSntavaD4oc,67
+aws_durable_execution_sdk_python-1.3.0.dist-info/RECORD,,