async-lambda-unstable 0.2.11__tar.gz → 0.2.12__tar.gz

{async-lambda-unstable-0.2.11 → async-lambda-unstable-0.2.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: async-lambda-unstable
-Version: 0.2.11
+Version: 0.2.12
 Summary: A framework for creating AWS Lambda Async Workflows. - Unstable Branch
 Author-email: "Nuclei, Inc" <engineering@nuclei.ai>
 Description-Content-Type: text/markdown
@@ -46,6 +46,7 @@ All task decorators share common arguments for configuring the underlying lambda
 - `memory: int = 128` Sets the memory allocation for the function.
 - `timeout: int = 60` Sets the timeout for the function (max 900 seconds).
 - `ephemeral_storage: int = 512` Sets the ephemeral storage allocation for the function.
+- `maximum_concurrency: Optional[int] = None` Sets the maximum concurrency value for the SQS trigger for the function. (only applies to `async_task` and `sqs_task` tasks.)
 
 ## Async Task
 

{async-lambda-unstable-0.2.11 → async-lambda-unstable-0.2.12}/README.md

@@ -39,6 +39,7 @@ All task decorators share common arguments for configuring the underlying lambda
 - `memory: int = 128` Sets the memory allocation for the function.
 - `timeout: int = 60` Sets the timeout for the function (max 900 seconds).
 - `ephemeral_storage: int = 512` Sets the ephemeral storage allocation for the function.
+- `maximum_concurrency: Optional[int] = None` Sets the maximum concurrency value for the SQS trigger for the function. (only applies to `async_task` and `sqs_task` tasks.)
 
 ## Async Task
 

{async-lambda-unstable-0.2.11 → async-lambda-unstable-0.2.12}/async_lambda/__init__.py

@@ -9,4 +9,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.2.11"
+__version__ = "0.2.12"

{async-lambda-unstable-0.2.11 → async-lambda-unstable-0.2.12}/async_lambda/models/task.py

@@ -31,6 +31,7 @@ class AsyncLambdaTask(Generic[EventType]):
     timeout: int
     memory: int
     ephemeral_storage: int
+    maximum_concurrency: Optional[int]
 
     executable: Callable[[EventType], Any]
 
@@ -43,6 +44,7 @@ class AsyncLambdaTask(Generic[EventType]):
         timeout: int = 60,
         memory: int = 128,
         ephemeral_storage: int = 512,
+        maximum_concurrency: Optional[int] = None,
     ):
         AsyncLambdaTask.validate_task_id(task_id)
         self.executable = executable
@@ -52,6 +54,7 @@ class AsyncLambdaTask(Generic[EventType]):
         self.timeout = timeout
         self.memory = memory
         self.ephemeral_storage = ephemeral_storage
+        self.maximum_concurrency = maximum_concurrency
 
     @staticmethod
     def validate_task_id(task_id: str):
@@ -93,6 +96,11 @@ class AsyncLambdaTask(Generic[EventType]):
         return f"{self.get_function_logical_id()}Extra{index}"
 
     def get_template_events(self):
+        sqs_properties = {}
+        if self.maximum_concurrency is not None:
+            sqs_properties["ScalingConfig"] = {
+                "MaximumConcurrency": self.maximum_concurrency
+            }
         if self.trigger_type == TaskTriggerType.MANAGED_SQS:
             return {
                 "ManagedSQS": {
@@ -106,6 +114,7 @@ class AsyncLambdaTask(Generic[EventType]):
                                 "Arn",
                             ]
                         },
+                        **sqs_properties,
                     },
                 }
             }
@@ -117,6 +126,7 @@ class AsyncLambdaTask(Generic[EventType]):
                         "BatchSize": 1,
                         "Enabled": True,
                         "Queue": self.trigger_config["queue_arn"],
+                        **sqs_properties,
                     },
                 }
             }

{async-lambda-unstable-0.2.11 → async-lambda-unstable-0.2.12}/async_lambda_unstable.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: async-lambda-unstable
-Version: 0.2.11
+Version: 0.2.12
 Summary: A framework for creating AWS Lambda Async Workflows. - Unstable Branch
 Author-email: "Nuclei, Inc" <engineering@nuclei.ai>
 Description-Content-Type: text/markdown
@@ -46,6 +46,7 @@ All task decorators share common arguments for configuring the underlying lambda
 - `memory: int = 128` Sets the memory allocation for the function.
 - `timeout: int = 60` Sets the timeout for the function (max 900 seconds).
 - `ephemeral_storage: int = 512` Sets the ephemeral storage allocation for the function.
+- `maximum_concurrency: Optional[int] = None` Sets the maximum concurrency value for the SQS trigger for the function. (only applies to `async_task` and `sqs_task` tasks.)
 
 ## Async Task