adapto 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

adapto/aws_lambda_monitor.py

@@ -0,0 +1,25 @@
+import time
+import logging
+from utils.aws_client import get_aws_client
+
+class AWSLambdaMonitor:
+    def __init__(self, function_name):
+        self.client = get_aws_client("cloudwatch")
+        self.function_name = function_name
+
+    def get_execution_time(self):
+        """Fetches the average execution time of the AWS Lambda function."""
+        try:
+            response = self.client.get_metric_statistics(
+                Namespace="AWS/Lambda",
+                MetricName="Duration",
+                Dimensions=[{"Name": "FunctionName", "Value": self.function_name}],
+                StartTime=time.time() - 3600,
+                EndTime=time.time(),
+                Period=300,
+                Statistics=["Average"]
+            )
+            return response["Datapoints"][-1]["Average"] if response["Datapoints"] else None
+        except Exception as e:
+            logging.error(f"Error fetching Lambda execution time: {e}")
+            return None

adapto/aws_lambda_scaler.py

@@ -0,0 +1,48 @@
+import logging
+from utils.aws_client import get_aws_client
+from adapto.config.settings import EXECUTION_TIME_THRESHOLD, LAMBDA_MEMORY_STEP, MIN_MEMORY, MAX_MEMORY
+from adapto.aws_lambda_monitor import AWSLambdaMonitor
+
+
+class AWSLambdaScaler:
+    def __init__(self, function_name):
+        self.client = get_aws_client("lambda")
+        self.monitor = AWSLambdaMonitor(function_name)
+        self.function_name = function_name
+
+    def get_current_memory(self):
+        """Retrieves the current memory allocation of the Lambda function."""
+        try:
+            response = self.client.get_function_configuration(FunctionName=self.function_name)
+            return response["MemorySize"]
+        except Exception as e:
+            logging.error(f"Error fetching Lambda memory configuration: {e}")
+            return None
+
+    def scale_lambda_memory(self):
+        """Adjusts Lambda memory allocation based on execution time."""
+        execution_time = self.monitor.get_execution_time()
+        if execution_time is None:
+            return
+
+        current_memory = self.get_current_memory()
+        if current_memory is None:
+            return
+
+        if execution_time > EXECUTION_TIME_THRESHOLD and current_memory < MAX_MEMORY:
+            new_memory = min(current_memory + LAMBDA_MEMORY_STEP, MAX_MEMORY)
+            self.update_lambda_memory(new_memory)
+        elif execution_time < EXECUTION_TIME_THRESHOLD * 0.6 and current_memory > MIN_MEMORY:
+            new_memory = max(current_memory - LAMBDA_MEMORY_STEP, MIN_MEMORY)
+            self.update_lambda_memory(new_memory)
+
+    def update_lambda_memory(self, new_memory):
+        """Updates the Lambda memory configuration."""
+        try:
+            self.client.update_function_configuration(
+                FunctionName=self.function_name,
+                MemorySize=new_memory
+            )
+            logging.info(f"Updated Lambda {self.function_name} memory to {new_memory}MB")
+        except Exception as e:
+            logging.error(f"Error updating Lambda memory: {e}")

{adapto-0.1.3.dist-info → adapto-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: adapto
-Version: 0.1.3
+Version: 0.1.4
 Summary: AI-driven auto-scaling library for dynamic resource allocation.
 Home-page: https://github.com/hrshlmeht/adapto
 Author: Harshal Mehta

adapto-0.1.4.dist-info/RECORD

@@ -0,0 +1,11 @@
+adapto/__init__.py,sha256=8SWysOKRVNsD3cOXLOfIdVZx6MxzIqGKEh9Z26QuCyM,30
+adapto/aws_lambda_monitor.py,sha256=XGv4QHe1DlZ4SmHvVZX_6KWOkoUoKGxC6W91-H1PAZs,959
+adapto/aws_lambda_scaler.py,sha256=r2oV2cvwJG9oqd1mFiGfv8Lmwpzp38BJ_v6Q7S5pPh4,2045
+adapto/predictor.py,sha256=ze-cMHdf290o3_9D-PQ7yQ8eMz1Utv3E3qTHRUFQiAI,842
+adapto/scaler.py,sha256=BqAs7rlIbrK4kkeBQzz3CgLHwpKWjhyQCciACqbFqTY,5654
+adapto/utils.py,sha256=6TgDvowlXL__v-OkLSTZF12ZOEqEKCLuNv1LmL70eAY,140
+adapto-0.1.4.dist-info/LICENSE,sha256=B73NNEiUy-CCs4zrOPlOWR39PfoG4zKtK97OnNctjvg,1070
+adapto-0.1.4.dist-info/METADATA,sha256=OfdbG0guMCztp96c1rksBnXhZlebQuPqcGAadHoIoWo,609
+adapto-0.1.4.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+adapto-0.1.4.dist-info/top_level.txt,sha256=IYsgAr6fnEC1R1ztYmK2ZpFNNGV0qRaf3Tvf-R8j0cM,7
+adapto-0.1.4.dist-info/RECORD,,

adapto-0.1.3.dist-info/RECORD

@@ -1,9 +0,0 @@
-adapto/__init__.py,sha256=8SWysOKRVNsD3cOXLOfIdVZx6MxzIqGKEh9Z26QuCyM,30
-adapto/predictor.py,sha256=ze-cMHdf290o3_9D-PQ7yQ8eMz1Utv3E3qTHRUFQiAI,842
-adapto/scaler.py,sha256=BqAs7rlIbrK4kkeBQzz3CgLHwpKWjhyQCciACqbFqTY,5654
-adapto/utils.py,sha256=6TgDvowlXL__v-OkLSTZF12ZOEqEKCLuNv1LmL70eAY,140
-adapto-0.1.3.dist-info/LICENSE,sha256=B73NNEiUy-CCs4zrOPlOWR39PfoG4zKtK97OnNctjvg,1070
-adapto-0.1.3.dist-info/METADATA,sha256=I09ftZmmqq2srMzQ2-ITC8hBIspuVym29oI9dZXYWjQ,609
-adapto-0.1.3.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-adapto-0.1.3.dist-info/top_level.txt,sha256=IYsgAr6fnEC1R1ztYmK2ZpFNNGV0qRaf3Tvf-R8j0cM,7
-adapto-0.1.3.dist-info/RECORD,,