Create uei_core/policy.py

uei_core/policy.py

@@ -0,0 +1,26 @@
+class UEIPolicy:
+
+    def __init__(self, uncertainty_threshold=0.20, energy_ratio_threshold=0.5):
+        self.uncertainty_threshold = uncertainty_threshold
+        self.energy_ratio_threshold = energy_ratio_threshold
+
+    def decide(self, U_small, U_large, E_small, E_large):
+        """
+        Decide if escalation is needed based on uncertainty-energy marginal utility.
+        """
+
+        if float(U_small) < self.uncertainty_threshold:
+            return "small"
+
+        marginal_gain = float(U_small - U_large)
+        marginal_energy = float(E_large - E_small)
+
+        if marginal_gain <= 0:
+            return "small"
+
+        ratio = marginal_gain / (marginal_energy + 1e-8)
+
+        if ratio < self.energy_ratio_threshold:
+            return "small"
+        else:
+            return "large"