fix: universal parsing + OpenRouter + state bug — purpose_agent/optimizer.py

purpose_agent/optimizer.py

@@ -248,12 +248,13 @@ class HeuristicOptimizer:
             )),
         ]
 
+        from purpose_agent.robust_parser import parse_optimizer_response
+
         try:
             result = self.llm.generate_structured(messages, schema=DISTILL_SCHEMA)
-        except Exception as e:
-            logger.error(f"Optimizer: Distillation failed ({e}), attempting text fallback")
-            raw = self.llm.generate(messages, temperature=0.5)
-            result = self._parse_distillation_text(raw)
+        except Exception:
+            raw = self.llm.generate(messages, temperature=0.5, max_tokens=2000)
+            result = parse_optimizer_response(raw)
 
         new_heuristics = []
         for h_data in result.get("heuristics", []):
@@ -356,7 +357,12 @@ class HeuristicOptimizer:
             )),
         ]
 
-        result = self.llm.generate_structured(messages, schema=DISTILL_SCHEMA)
+        from purpose_agent.robust_parser import parse_optimizer_response
+        try:
+            result = self.llm.generate_structured(messages, schema=DISTILL_SCHEMA)
+        except Exception:
+            raw = self.llm.generate(messages, temperature=0.5, max_tokens=2000)
+            result = parse_optimizer_response(raw)
 
         merged = []
         for h_data in result.get("heuristics", []):