Add Memory Usage Logging Functionality

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.

Author: beso

prometheus/functions/function_file.py

@@ -0,0 +1,45 @@
+import psutil
+import time
+
+def memory_usage_logger(func):
+    """
+    A decorator to log memory usage before and after a function call.
+
+    :param func: The function to be logged.
+    :return: The result of the function call.
+    """
+    def wrapper(*args, **kwargs):
+        # Log memory usage before function call
+        process = psutil.Process(os.getpid())
+        memory_before = process.memory_info().rss
+
+        # Call the function
+        result = func(*args, **kwargs)
+
+        # Log memory usage after function call
+        memory_after = process.memory_info().rss
+
+        # Calculate memory usage difference
+        memory_usage_diff = memory_after - memory_before
+
+        # Log memory usage difference
+        print(f"Memory usage difference for {func.__name__}: {memory_usage_diff} bytes")
+
+        return result
+
+    return wrapper
+
+@memory_usage_logger
+def function_to_log(arg1, arg2):
+    """
+    An example function to be logged for memory usage.
+
+    :param arg1: The first argument.
+    :param arg2: The second argument.
+    :return: None
+    """
+    time.sleep(1)  # Simulate function execution time
+    print(f"Function {function_to_log.__name__} executed with arguments: {arg1}, {arg2}")
+
+if __name__ == "__main__":
+    function_to_log("argument1", "argument2")

prometheus/utils/memory_usage_logger.py

@@ -0,0 +1,26 @@
+import psutil
+import time
+
+def memory_usage_logger(func):
+    """
+    Logs memory usage before and after calling the provided function.
+
+    :param func: The function to be logged.
+    :return: The result of the provided function.
+    """
+    def wrapper(*args, **kwargs):
+        # Log memory usage before calling the function
+        process = psutil.Process(os.getpid())
+        memory_before = process.memory_info().rss / (1024 * 1024)
+
+        # Call the function
+        result = func(*args, **kwargs)
+
+        # Log memory usage after calling the function
+        memory_after = process.memory_info().rss / (1024 * 1024)
+        print(f"Memory usage before: {memory_before:.2f} MB")
+        print(f"Memory usage after: {memory_after:.2f} MB")
+
+        return result
+
+    return wrapper