adapto 0.1.3__tar.gz → 0.1.4__tar.gz

{adapto-0.1.3 → adapto-0.1.4}/PKG-INFO

@@ -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/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-0.1.4/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 → adapto-0.1.4}/adapto.egg-info/PKG-INFO

@@ -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.3 → adapto-0.1.4}/adapto.egg-info/SOURCES.txt

@@ -2,6 +2,8 @@ LICENSE
 README.md
 setup.py
 adapto/__init__.py
+adapto/aws_lambda_monitor.py
+adapto/aws_lambda_scaler.py
 adapto/predictor.py
 adapto/scaler.py
 adapto/utils.py

{adapto-0.1.3 → adapto-0.1.4}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='adapto',
-    version='0.1.3',
+    version='0.1.4',
     packages=find_packages(),
     install_requires=['psutil', 'numpy'],
     description='AI-driven auto-scaling library for dynamic resource allocation.',