async-lambda-unstable 0.5.4__tar.gz → 0.5.6__tar.gz

{async-lambda-unstable-0.5.4 → async-lambda-unstable-0.5.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: async-lambda-unstable
-Version: 0.5.4
+Version: 0.5.6
 Summary: A framework for creating AWS Lambda Async Workflows. - Unstable Branch
 Author-email: "Nuclei, Inc" <engineering@nuclei.ai>
 Description-Content-Type: text/markdown

{async-lambda-unstable-0.5.4 → async-lambda-unstable-0.5.6}/async_lambda/__init__.py

@@ -1,3 +1,4 @@
+from .config import config_set_default_task_memory as config_set_default_task_memory
 from .config import config_set_name as config_set_name
 from .config import config_set_runtime as config_set_runtime
 from .config import config_set_s3_payload_retention as config_set_s3_payload_retention
@@ -20,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.5.4"
+__version__ = "0.5.6"

{async-lambda-unstable-0.5.4 → async-lambda-unstable-0.5.6}/async_lambda/config.py

@@ -5,6 +5,7 @@ class AsyncLambdaConfig:
     name: str = "async-lambda"
     runtime: str = "python3.10"
     s3_payload_retention: Optional[int] = 30
+    default_task_memory: int = 128
 
 
 config = AsyncLambdaConfig()
@@ -29,3 +30,7 @@ def config_set_s3_payload_retention(days: Optional[int]):
     Sets the s3_payload_retention policy in days.
     """
     config.s3_payload_retention = days
+
+
+def config_set_default_task_memory(memory: int = 128):
+    config.default_task_memory = memory

{async-lambda-unstable-0.5.4 → async-lambda-unstable-0.5.6}/async_lambda/controller.py

@@ -713,6 +713,9 @@ class AsyncLambdaController:
         is_dlq_task: bool = False,
         lane_count: Optional[int] = None,
         propagate_lane_assignment: Optional[bool] = None,
+        timeout: int = 60,
+        memory: Optional[int] = None,
+        ephemeral_storage: int = 512,
         **kwargs,
     ):
         """
@@ -742,6 +745,9 @@ class AsyncLambdaController:
                         "propagate_lane_assignment": propagate_lane_assignment,
                         "batch_size": 1,
                     },
+                    timeout=timeout,
+                    memory=memory,
+                    ephemeral_storage=ephemeral_storage,
                     **kwargs,
                 )
             )
@@ -759,6 +765,9 @@ class AsyncLambdaController:
         propagate_lane_assignment: Optional[bool] = None,
         batch_size: int = 20,
         max_batching_window: Optional[int] = None,
+        timeout: int = 60,
+        memory: Optional[int] = None,
+        ephemeral_storage: int = 512,
         **kwargs,
     ):
         """
@@ -789,6 +798,9 @@ class AsyncLambdaController:
                         "batch_size": batch_size,
                         "max_batching_window": max_batching_window,
                     },
+                    timeout=timeout,
+                    memory=memory,
+                    ephemeral_storage=ephemeral_storage,
                     **kwargs,
                 )
             )
@@ -796,7 +808,15 @@ class AsyncLambdaController:
 
         return _task
 
-    def sqs_task(self, task_id: str, queue_arn: str, **kwargs):
+    def sqs_task(
+        self,
+        task_id: str,
+        queue_arn: str,
+        timeout: int = 60,
+        memory: Optional[int] = None,
+        ephemeral_storage: int = 512,
+        **kwargs,
+    ):
         """
         Decorate a function to register it as an SQS task.
         These tasks will be triggered by messages in the given queue.
@@ -818,6 +838,9 @@ class AsyncLambdaController:
                     task_id=task_id,
                     trigger_type=TaskTriggerType.UNMANAGED_SQS,
                     trigger_config={"queue_arn": queue_arn},
+                    timeout=timeout,
+                    memory=memory,
+                    ephemeral_storage=ephemeral_storage,
                     **kwargs,
                 )
             )
@@ -826,7 +849,15 @@ class AsyncLambdaController:
 
         return _task
 
-    def scheduled_task(self, task_id: str, schedule_expression: str, **kwargs):
+    def scheduled_task(
+        self,
+        task_id: str,
+        schedule_expression: str,
+        timeout: int = 60,
+        memory: Optional[int] = None,
+        ephemeral_storage: int = 512,
+        **kwargs,
+    ):
         """
         Decorate a function to register it as a scheduled task.
         These tasks will be triggered by the given schedule expression.
@@ -848,6 +879,9 @@ class AsyncLambdaController:
                     task_id=task_id,
                     trigger_type=TaskTriggerType.SCHEDULED_EVENT,
                     trigger_config={"schedule_expression": schedule_expression},
+                    timeout=timeout,
+                    memory=memory,
+                    ephemeral_storage=ephemeral_storage,
                     **kwargs,
                 )
             )
@@ -856,7 +890,16 @@ class AsyncLambdaController:
 
         return _task
 
-    def api_task(self, task_id: str, path: str, method: str, **kwargs):
+    def api_task(
+        self,
+        task_id: str,
+        path: str,
+        method: str,
+        timeout: int = 60,
+        memory: Optional[int] = None,
+        ephemeral_storage: int = 512,
+        **kwargs,
+    ):
         """
         Decorate a function to register it as an API task.
         These tasks will be triggered by an API call.
@@ -878,6 +921,9 @@ class AsyncLambdaController:
                     task_id=task_id,
                     trigger_type=TaskTriggerType.API_EVENT,
                     trigger_config={"path": path, "method": method},
+                    timeout=timeout,
+                    memory=memory,
+                    ephemeral_storage=ephemeral_storage,
                     **kwargs,
                 )
             )
@@ -893,6 +939,9 @@ class AsyncLambdaController:
         stream_arn: str,
         batch_size: int,
         max_batching_window: int = 0,
+        timeout: int = 60,
+        memory: Optional[int] = None,
+        ephemeral_storage: int = 512,
         **kwargs,
     ):
         """
@@ -920,6 +969,9 @@ class AsyncLambdaController:
                         "batch_size": batch_size,
                         "max_batching_window": max_batching_window,
                     },
+                    timeout=timeout,
+                    memory=memory,
+                    ephemeral_storage=ephemeral_storage,
                     **kwargs,
                 )
             )
@@ -928,7 +980,14 @@ class AsyncLambdaController:
 
         return _task
 
-    def pure_task(self, task_id: str, **kwargs):
+    def pure_task(
+        self,
+        task_id: str,
+        timeout: int = 60,
+        memory: Optional[int] = None,
+        ephemeral_storage: int = 512,
+        **kwargs,
+    ):
         """
         Decorate a function to register it as a pure lambda function.
         """
@@ -947,6 +1006,9 @@ class AsyncLambdaController:
                     task_id=task_id,
                     trigger_type=TaskTriggerType.BASE_EVENT,
                     trigger_config={},
+                    timeout=timeout,
+                    memory=memory,
+                    ephemeral_storage=ephemeral_storage,
                     **kwargs,
                 )
             )

{async-lambda-unstable-0.5.4 → async-lambda-unstable-0.5.6}/async_lambda/models/task.py

@@ -65,7 +65,7 @@ class AsyncLambdaTask(Generic[EventType, RT]):
     trigger_config: dict
 
     timeout: int
-    memory: int
+    memory: Optional[int]
     ephemeral_storage: int
     maximum_concurrency: Optional[Union[int, List[int]]]
     init_tasks: List[Union[Callable[[str], Any], Callable[[], Any]]]
@@ -81,7 +81,7 @@ class AsyncLambdaTask(Generic[EventType, RT]):
         trigger_type: TaskTriggerType,
         trigger_config: Optional[dict] = None,
         timeout: int = 60,
-        memory: int = 128,
+        memory: Optional[int] = None,
         ephemeral_storage: int = 512,
         maximum_concurrency: Optional[Union[int, List[int]]] = None,
         init_tasks: Optional[
@@ -447,7 +447,9 @@ class AsyncLambdaTask(Generic[EventType, RT]):
                     "FunctionName": self.get_function_name(),
                     "CodeUri": ".async_lambda/build/deployment.zip",
                     "EphemeralStorage": {"Size": self.ephemeral_storage},
-                    "MemorySize": self.memory,
+                    "MemorySize": (
+                        self.memory if self.memory else config.default_task_memory
+                    ),
                     "Timeout": self.timeout,
                     "Events": events,
                     "Policies": [

{async-lambda-unstable-0.5.4 → async-lambda-unstable-0.5.6}/async_lambda_unstable.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: async-lambda-unstable
-Version: 0.5.4
+Version: 0.5.6
 Summary: A framework for creating AWS Lambda Async Workflows. - Unstable Branch
 Author-email: "Nuclei, Inc" <engineering@nuclei.ai>
 Description-Content-Type: text/markdown