Update infrastructure/cost_estimator.py

infrastructure/cost_estimator.py

@@ -56,12 +56,23 @@ class CostEstimator:
         else:
             self._pricing = self.DEFAULT_PRICING.copy()
 
-    @lru_cache(maxsize=256)
+        # Cache for estimate_monthly_cost using a hashable key
+        self._cost_cache = {}
+
     def estimate_monthly_cost(self, intent: ProvisionResourceIntent) -> Optional[float]:
+        # Create a hashable key from the intent's immutable fields
+        # (resource_type, size, region, environment) – configuration is omitted as it doesn't affect cost in this demo.
+        key = (intent.resource_type, intent.size, intent.region, intent.environment)
+        if key in self._cost_cache:
+            return self._cost_cache[key]
+
         resource_pricing = self._pricing.get(intent.resource_type)
         if not resource_pricing:
+            self._cost_cache[key] = None
             return None
-        return resource_pricing.get(intent.size)
+        cost = resource_pricing.get(intent.size)
+        self._cost_cache[key] = cost
+        return cost
 
     def cost_delta_vs_baseline(
         self,