Upload aco/optimizer.py

aco/optimizer.py

@@ -90,16 +90,20 @@ class AgentCostOptimizer:
         prompt_layout = None
         if self.config.enable_cache_layout and context_budget:
             content_pieces = self._build_content_pieces(context_budget)
+            cache_discount = 0.5
+            if model_cfg:
+                cache_discount = model_cfg.cache_discount_rate
             prompt_layout = self.cache_layout.layout(
-                content_pieces, cost_1k, self.config.models.get(routing.model_id, {}).cache_discount_rate if model_cfg else 0.5
+                content_pieces, cost_1k, cache_discount
             )
         
         # 5. Check for meta-tool
         meta_tool_match = None
         if self.config.enable_meta_tool_miner:
             planned_tools = run_state.get("planned_tools", [])
+            planned_tool_names = [t[0] for t in planned_tools]
             meta_tool_match = self.meta_tool_miner.match_and_compress(
-                prediction.task_type, planned_tools
+                prediction.task_type, planned_tool_names
             )
         
         # 6. Gate tool calls