Update engine.py

engine.py

@@ -11,7 +11,7 @@ import numpy as np
 from typing import Optional, Dict, Any, List
 
 from agentic_reliability_framework.core.models.event import ReliabilityEvent, EventSeverity, HealingAction
-from policy_engine import PolicyEngine
+from policy_engine import PolicyEngine  # local patched version (see Dockerfile)
 from agentic_reliability_framework.runtime.analytics.anomaly import AdvancedAnomalyDetector
 from agentic_reliability_framework.runtime.analytics.predictive import BusinessImpactCalculator
 from agentic_reliability_framework.runtime.orchestration.manager import OrchestrationManager
@@ -41,6 +41,11 @@ class ThreadSafeEventStore:
 
 
 class EnhancedReliabilityEngine:
+    """
+    Main engine for processing infrastructure events.
+    Orchestrates agents, policy evaluation, risk scoring, and optional Claude enhancement.
+    """
+
     def __init__(self, orchestrator: Optional[OrchestrationManager] = None,
                  policy_engine: Optional[PolicyEngine] = None,
                  event_store: Optional[ThreadSafeEventStore] = None,
@@ -66,6 +71,20 @@ class EnhancedReliabilityEngine:
     async def process_event_enhanced(self, component: str, latency: float, error_rate: float,
                                       throughput: float = 1000, cpu_util: Optional[float] = None,
                                       memory_util: Optional[float] = None) -> Dict[str, Any]:
+        """
+        Process a single telemetry event and return analysis results.
+
+        Args:
+            component: Name of the component (e.g., "api-service").
+            latency: P99 latency in milliseconds.
+            error_rate: Error rate between 0 and 1.
+            throughput: Requests per second.
+            cpu_util: CPU utilization (0-1), optional.
+            memory_util: Memory utilization (0-1), optional.
+
+        Returns:
+            Dictionary containing analysis results.
+        """
         logger.info(f"Processing event for {component}: latency={latency}ms, error_rate={error_rate*100:.1f}%")
         from agentic_reliability_framework.core.models.event import validate_component_id
         is_valid, error_msg = validate_component_id(component)
@@ -157,6 +176,10 @@ class EnhancedReliabilityEngine:
         return result
 
     async def enhance_with_claude(self, event: ReliabilityEvent, agent_results: Dict[str, Any]) -> Dict[str, Any]:
+        """
+        Enhance agent results with a Claude‑generated executive summary.
+        Falls back gracefully if Claude is unavailable.
+        """
         context_parts = []
         context_parts.append("INCIDENT SUMMARY:")
         context_parts.append(f"Component: {event.component}")