human-eval-rust 2.1.0__py3-none-any.whl

human_eval/logging_config.py

@@ -0,0 +1,43 @@
+"""
+Logging configuration for HumanEval Rust evaluation.
+
+Provides structured logging setup for CLI and library usage.
+
+Copyright (c) 2025 Dave Tofflemire, SigilDERG Project
+Version: 2.1.0
+"""
+
+import logging
+import sys
+
+
+def setup_logging(
+    level: int = logging.INFO, json_format: bool = False
+) -> logging.Logger:
+    """Configure logging for human-eval-rust."""
+
+    logger = logging.getLogger("human_eval")
+    logger.setLevel(level)
+
+    handler = logging.StreamHandler(sys.stderr)
+    if json_format:
+        formatter = logging.Formatter(
+            '{"time": "%(asctime)s", "level": "%(levelname)s", '
+            '"module": "%(module)s", "message": "%(message)s"}'
+        )
+    else:
+        formatter = logging.Formatter(
+            "%(asctime)s [%(levelname)s] %(module)s: %(message)s"
+        )
+    handler.setFormatter(formatter)
+    if not logger.handlers:
+        logger.addHandler(handler)
+
+    return logger
+
+
+# Module-level logger for convenience imports
+logger = logging.getLogger("human_eval")
+
+
+__all__ = ["setup_logging", "logger"]

human_eval/resource_monitor.py

@@ -0,0 +1,58 @@
+"""
+Resource monitoring for HumanEval Rust evaluation.
+
+Provides worker slot management and memory usage monitoring during parallel evaluation.
+
+Copyright (c) 2025 Dave Tofflemire, SigilDERG Project
+Version: 2.1.0
+"""
+
+import threading
+
+import psutil
+
+
+class ResourceMonitor:
+    """Monitor and limit resource usage during evaluation."""
+
+    def __init__(
+        self,
+        max_memory_percent: float = 80.0,
+        max_workers: int = 24,
+        check_interval: float = 1.0,
+    ):
+        self.max_memory_percent = max_memory_percent
+        self.max_workers = max_workers
+        self.check_interval = check_interval
+        self._active_workers = 0
+        self._lock = threading.Lock()
+        self._stop_event = threading.Event()
+
+    def acquire_worker(self) -> bool:
+        """Try to acquire a worker slot. Returns False if limit reached."""
+
+        with self._lock:
+            if psutil.virtual_memory().percent > self.max_memory_percent:
+                return False
+            if self._active_workers >= self.max_workers:
+                return False
+            self._active_workers += 1
+            return True
+
+    def release_worker(self) -> None:
+        """Release a worker slot."""
+        with self._lock:
+            self._active_workers = max(0, self._active_workers - 1)
+
+    def stop(self) -> None:
+        """Signal the monitor to stop. Sets the internal stop event."""
+        self._stop_event.set()
+
+    @property
+    def active_workers(self) -> int:
+        """Return the current number of active workers."""
+        with self._lock:
+            return self._active_workers
+
+
+__all__ = ["ResourceMonitor"]