Update policy_engine.py

policy_engine.py

@@ -4,11 +4,11 @@ Policy Engine for Automated Healing Actions.
 
 import threading
 import logging
-import datetime  # <-- ADDED
+import datetime
 from collections import OrderedDict
 from typing import Dict, List, Optional, Any
 
-from agentic_reliability_framework.core.models.event import HealingPolicy, HealingAction, ReliabilityEvent
+from agentic_reliability_framework.core.models.event import HealingPolicy, HealingAction, ReliabilityEvent, PolicyCondition
 
 logger = logging.getLogger(__name__)
 
@@ -16,7 +16,7 @@ logger = logging.getLogger(__name__)
 DEFAULT_HEALING_POLICIES = [
     HealingPolicy(
         name="high_latency_restart",
-        conditions=[{"metric": "latency_p99", "operator": "gt", "threshold": 500.0}],
+        conditions=[PolicyCondition(metric="latency_p99", operator="gt", threshold=500.0)],
         actions=[HealingAction.RESTART_CONTAINER, HealingAction.ALERT_TEAM],
         priority=1,
         cool_down_seconds=300,
@@ -24,7 +24,7 @@ DEFAULT_HEALING_POLICIES = [
     ),
     HealingPolicy(
         name="critical_error_rate_rollback",
-        conditions=[{"metric": "error_rate", "operator": "gt", "threshold": 0.3}],
+        conditions=[PolicyCondition(metric="error_rate", operator="gt", threshold=0.3)],
         actions=[HealingAction.ROLLBACK, HealingAction.CIRCUIT_BREAKER, HealingAction.ALERT_TEAM],
         priority=1,
         cool_down_seconds=600,
@@ -32,7 +32,7 @@ DEFAULT_HEALING_POLICIES = [
     ),
     HealingPolicy(
         name="high_error_rate_traffic_shift",
-        conditions=[{"metric": "error_rate", "operator": "gt", "threshold": 0.15}],
+        conditions=[PolicyCondition(metric="error_rate", operator="gt", threshold=0.15)],
         actions=[HealingAction.TRAFFIC_SHIFT, HealingAction.ALERT_TEAM],
         priority=2,
         cool_down_seconds=300,
@@ -41,8 +41,8 @@ DEFAULT_HEALING_POLICIES = [
     HealingPolicy(
         name="resource_exhaustion_scale",
         conditions=[
-            {"metric": "cpu_util", "operator": "gt", "threshold": 0.9},
-            {"metric": "memory_util", "operator": "gt", "threshold": 0.9}
+            PolicyCondition(metric="cpu_util", operator="gt", threshold=0.9),
+            PolicyCondition(metric="memory_util", operator="gt", threshold=0.9)
         ],
         actions=[HealingAction.SCALE_OUT],
         priority=2,
@@ -51,7 +51,7 @@ DEFAULT_HEALING_POLICIES = [
     ),
     HealingPolicy(
         name="moderate_latency_circuit_breaker",
-        conditions=[{"metric": "latency_p99", "operator": "gt", "threshold": 300.0}],
+        conditions=[PolicyCondition(metric="latency_p99", operator="gt", threshold=300.0)],
         actions=[HealingAction.CIRCUIT_BREAKER],
         priority=3,
         cool_down_seconds=180,
@@ -105,11 +105,11 @@ class PolicyEngine:
                 unique.append(a)
         return unique if unique else [HealingAction.NO_ACTION]
 
-    def _evaluate_conditions(self, conditions: List[Dict[str, Any]], event: ReliabilityEvent) -> bool:
+    def _evaluate_conditions(self, conditions: List[PolicyCondition], event: ReliabilityEvent) -> bool:
         for cond in conditions:
-            metric = cond["metric"]
-            op = cond["operator"]
-            thresh = cond["threshold"]
+            metric = cond.metric
+            op = cond.operator
+            thresh = cond.threshold
             val = getattr(event, metric, None)
             if val is None:
                 return False