Update policy_engine.py

policy_engine.py

@@ -8,7 +8,7 @@ import datetime
 from collections import OrderedDict
 from typing import Dict, List, Optional, Any
 
-from agentic_reliability_framework.core.models.event import HealingPolicy, HealingAction, ReliabilityEvent, PolicyCondition
+from agentic_reliability_framework.core.models.event import HealingPolicy, HealingAction, PolicyCondition, ReliabilityEvent
 
 logger = logging.getLogger(__name__)
 
@@ -63,12 +63,10 @@ DEFAULT_HEALING_POLICIES = [
 class PolicyEngine:
     """
     Thread‑safe policy engine with cooldown and rate limiting.
-
     Policies are evaluated in priority order. Each policy has:
       - conditions (AND logic)
       - cooldown per (policy, component)
       - rate limit per hour
-
     The engine maintains an LRU cache of last execution timestamps
     (using OrderedDict) to bound memory usage.
     """
@@ -174,4 +172,33 @@ class PolicyEngine:
             self.execution_timestamps[key] = []
         self.execution_timestamps[key].append(ts)
         if len(self.execution_timestamps[key]) > self.max_execution_history:
-            self.execution_timestamps[key] = self.execution_timestamps[key][-self.max_execution_history:]
+            self.execution_timestamps[key] = self.execution_timestamps[key][-self.max_execution_history:]
+
+    # ========== NEW: Simplified evaluation for demo ==========
+    def evaluate(self, event_type: str, severity: str, component: str) -> List[Dict[str, Any]]:
+        """
+        Simplified policy evaluation for the governance demo.
+        Returns a list of recommended actions based on severity and event type.
+        Each action is a dict with keys: policy, action, reason.
+        """
+        actions = []
+        if severity == "critical":
+            actions.append({
+                "policy": "POL-002",
+                "action": "isolate_affected",
+                "reason": "Critical failure detected"
+            })
+        elif severity == "high":
+            actions.append({
+                "policy": "POL-004",
+                "action": "require_approval",
+                "reason": "High risk"
+            })
+        elif severity == "medium" and event_type == "text_generation":
+            actions.append({
+                "policy": "POL-001",
+                "action": "regenerate",
+                "reason": "Low confidence"
+            })
+        # You can add more rules based on component or event_type as needed
+        return actions