Update optillm.py

Author: codelion

optillm.py

@@ -16,6 +16,7 @@
 from cot_reflection import cot_reflection
 from plansearch import plansearch
 from leap import leap
+from agent import agent_approach
 
 # Setup logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -98,6 +99,8 @@ def proxy():
             final_response = plansearch(system_prompt, initial_query, client, model, n=n)
         elif approach == 'leap':
             final_response = leap(system_prompt, initial_query, client, model)
+        elif approach == 'agent':
+            final_response = agent_approach(system_prompt, initial_query, client, model, max_attempts=3)
         else:
             raise ValueError(f"Unknown approach: {approach}")
     except Exception as e:
@@ -135,7 +138,7 @@ def proxy():
 def main():
     parser = argparse.ArgumentParser(description="Run LLM inference with various approaches.")
     parser.add_argument("--approach", type=str, choices=["auto", "mcts", "bon", "moa", "rto", "z3", "self_consistency", "pvg", "rstar",
-                                                          "cot_reflection", "plansearch", "leap"], default="auto", help="Inference approach to use")
+                                                          "cot_reflection", "plansearch", "leap", "agent"], default="auto", help="Inference approach to use")
     parser.add_argument("--simulations", type=int, default=2, help="Number of MCTS simulations")
     parser.add_argument("--exploration", type=float, default=0.2, help="Exploration weight for MCTS")
     parser.add_argument("--depth", type=int, default=1, help="Simulation depth for MCTS")