async-lambda-unstable 0.6.4__tar.gz → 0.6.6__tar.gz

{async_lambda_unstable-0.6.4 → async_lambda_unstable-0.6.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: async-lambda-unstable
-Version: 0.6.4
+Version: 0.6.6
 Summary: A framework for creating AWS Lambda Async Workflows. - Unstable Branch
 Author-email: "Nuclei, Inc" <engineering@nuclei.ai>
 Requires-Dist: click>=8.0.0

{async_lambda_unstable-0.6.4 → async_lambda_unstable-0.6.6}/async_lambda/__init__.py

@@ -21,4 +21,4 @@ from .models.events.managed_sqs_event import ManagedSQSEvent as ManagedSQSEvent
 from .models.events.scheduled_event import ScheduledEvent as ScheduledEvent
 from .models.events.unmanaged_sqs_event import UnmanagedSQSEvent as UnmanagedSQSEvent
 
-__version__ = "0.6.4"
+__version__ = "0.6.6"

{async_lambda_unstable-0.6.4 → async_lambda_unstable-0.6.6}/async_lambda/controller.py

@@ -426,7 +426,6 @@ class AsyncLambdaController:
                             **build_config.environment_variables,
                         },
                     },
-                    "CodeUri": ".async_lambda/build/deployment.zip",
                     "MemorySize": config.default_task_memory,
                     "EphemeralStorage": {"Size": config.default_task_ephemeral_storage},
                     **build_config.function_properties,
@@ -1278,6 +1277,7 @@ class AsyncLambdaController:
         timeout: int = 60,
         memory: Optional[int] = None,
         ephemeral_storage: Optional[int] = None,
+        snap_start: bool = False,
         **kwargs,
     ):
         """
@@ -1335,6 +1335,7 @@ class AsyncLambdaController:
                     timeout=timeout,
                     memory=memory,
                     ephemeral_storage=ephemeral_storage,
+                    snap_start=snap_start,
                     **kwargs,
                 )
             )
@@ -1548,6 +1549,7 @@ class AsyncLambdaController:
         timeout: int = 60,
         memory: Optional[int] = None,
         ephemeral_storage: Optional[int] = None,
+        snap_start: bool = False,
         **kwargs,
     ):
         """
@@ -1595,6 +1597,7 @@ class AsyncLambdaController:
                     timeout=timeout,
                     memory=memory,
                     ephemeral_storage=ephemeral_storage,
+                    snap_start=snap_start,
                     **kwargs,
                 )
             )

{async_lambda_unstable-0.6.4 → async_lambda_unstable-0.6.6}/async_lambda/models/task.py

@@ -102,6 +102,7 @@ class AsyncLambdaTask(Generic[EventType, RT]):
     ephemeral_storage: Optional[int]
     maximum_concurrency: Optional[Union[int, List[int]]]
     init_tasks: List[Union[Callable[[str], Any], Callable[[], Any]]]
+    snap_start: bool
     _has_run_init_tasks: bool
 
     executable: Callable[[EventType], RT]
@@ -120,6 +121,7 @@ class AsyncLambdaTask(Generic[EventType, RT]):
         init_tasks: Optional[
             List[Union[Callable[[str], Any], Callable[[], Any]]]
         ] = None,
+        snap_start: bool = False,
     ):
         """
         Initialize an AsyncLambdaTask instance.
@@ -135,6 +137,7 @@ class AsyncLambdaTask(Generic[EventType, RT]):
             ephemeral_storage (Optional[int], optional): Ephemeral storage (MB). Defaults to None.
             maximum_concurrency (Optional[Union[int, List[int]]], optional): Maximum concurrency for the task. Defaults to None.
             init_tasks (Optional[List[Union[Callable[[str], Any], Callable[[], Any]]]], optional): Initialization tasks to run before execution. Defaults to None.
+            snap_start (bool): Enables SnapStart with static configuration. Defaults to False.
 
         Raises:
             Exception: If the DLQ (Dead Letter Queue) task ID is specified but does not exist or is not an async-task.
@@ -149,6 +152,7 @@ class AsyncLambdaTask(Generic[EventType, RT]):
         self.memory = memory
         self.ephemeral_storage = ephemeral_storage
         self.maximum_concurrency = maximum_concurrency
+        self.snap_start = snap_start
         if init_tasks is None:
             self.init_tasks = []
         else:
@@ -620,6 +624,7 @@ class AsyncLambdaTask(Generic[EventType, RT]):
                     "Timeout": self.timeout,
                     "Events": events,
                     "Policies": policies,
+                    "CodeUri": ".async_lambda/build/deployment.zip",
                     **override_config.function_properties,
                 },
             }
@@ -632,6 +637,10 @@ class AsyncLambdaTask(Generic[EventType, RT]):
             template[self.get_function_logical_id()]["Properties"][
                 "EphemeralStorage"
             ] = {"Size": self.ephemeral_storage}
+        if self.snap_start:
+            template[self.get_function_logical_id()]["Properties"]["SnapStart"] = {
+                "ApplyOn": "PublishedVersions"
+            }
 
         if self.trigger_type in MANAGED_SQS_TASK_TYPES:
             dlq_task = self.get_dlq_task()