Add Memory Usage Logging Functionality

Implements koii-network#12905

Implements koii-network#12876

Implements koii-network#12871

Implements koii-network#12833

Implements koii-network#12816

Implements koii-network#12811

Implements koii-network#12791

Implements koii-network#12775

Implements koii-network#12711

Implements koii-network#12626

# Add Memory Usage Logging Functionality

## Task
Write a function to log memory usage before and after a function call.

## Acceptance Criteria
All tests must pass.

## Summary of Changes
Implemented a new utility function to log memory usage before and after function calls. This will help in tracking and analyzing memory consumption of critical functions in the application.

## Test Cases
 - Verify the memory logging function captures memory usage correctly
 - Ensure memory logging does not significantly impact function performance
 - Check that memory logging works for different types of functions
 - Validate memory usage logging for functions with varying execution times

This PR was created automatically by a Koii Network AI Agent powered by Together.ai.

This PR was created automatically by a Koii Network AI Agent powered by Together.ai.

This PR was created automatically by a Koii Network AI Agent powered by Together.ai.

This PR was created automatically by a Koii Network AI Agent powered by Together.ai.

This PR was created automatically by a Koii Network AI Agent powered by Together.ai.

This PR was created automatically by a Koii Network AI Agent powered by Together.ai.

This PR was created automatically by a Koii Network AI Agent powered by Together.ai.

This PR was created automatically by a Koii Network AI Agent powered by Together.ai.

This PR was created automatically by a Koii Network AI Agent powered by Together.ai.

Author: beso

prometheus/functions/function_file.py

@@ -0,0 +1,52 @@
+import psutil
+import time
+
+def memory_logger(func):
+    """
+    A decorator to log memory usage before and after a function call.
+
+    Args:
+        func: The function to decorate.
+
+    Returns:
+        The decorated function with memory logging functionality.
+
+    """
+    def wrapper(*args, **kwargs):
+        # Get the current memory usage
+        process = psutil.Process(os.getpid())
+        memory_before = process.memory_info().rss
+
+        # Call the function
+        start_time = time.time()
+        result = func(*args, **kwargs)
+        end_time = time.time()
+
+        # Get the memory usage after the function call
+        memory_after = process.memory_info().rss
+
+        # Calculate and log the memory difference
+        memory_difference = memory_after - memory_before
+        print(f"Function {func.__name__} memory usage: {memory_difference} bytes")
+
+        # Print the function execution time
+        print(f"Function {func.__name__} execution time: {end_time - start_time:.4f} seconds")
+
+        return result
+
+    return wrapper
+
+
+if __name__ == "__main__":
+    @memory_logger
+    def test_function():
+        time.sleep(1)
+
+
+    @memory_logger
+    def memory_intensive_function():
+        large_list = [i for i in range(1000000)]
+
+
+    test_function()
+    memory_intensive_function()

prometheus/utils/memory_logger.py

@@ -0,0 +1,37 @@
+import psutil
+import time
+
+def memory_logger(func):
+    """
+    A decorator to log memory usage before and after a function call.
+
+    Args:
+        func: The function to log memory usage for.
+
+    Returns:
+        The function with memory logging added.
+
+    """
+    def wrapper(*args, **kwargs):
+        # Get the current memory usage
+        process = psutil.Process(os.getpid())
+        memory_before = process.memory_info().rss
+
+        # Call the function
+        start_time = time.time()
+        result = func(*args, **kwargs)
+        end_time = time.time()
+
+        # Get the memory usage after the function call
+        memory_after = process.memory_info().rss
+
+        # Calculate the difference between the initial and final memory usage
+        memory_used = memory_after - memory_before
+
+        # Log the memory usage
+        print(f"Function {func.__name__} took {end_time - start_time:.6f} seconds "
+              f"and used {memory_used:,} bytes of memory")
+
+        return result
+
+    return wrapper