Update optillm.py

Author: codelion

optillm.py

@@ -17,6 +17,7 @@
 from optillm.cot_reflection import cot_reflection
 from optillm.plansearch import plansearch
 from optillm.leap import leap
+from optillm.reread import re2_approach
 
 # Setup logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -69,7 +70,7 @@
 
 # List of known approaches
 known_approaches = ["mcts", "bon", "moa", "rto", "z3", "self_consistency", "pvg", "rstar",
-                    "cot_reflection", "plansearch", "leap"]
+                    "cot_reflection", "plansearch", "leap", "re2"]
 
 # Optional API key configuration to secure the proxy
 @app.before_request
@@ -149,6 +150,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 == 're2':
+            final_response = re2_approach(system_prompt, initial_query, client, model, n=n)
         else:
             raise ValueError(f"Unknown approach: {approach}")
     except Exception as e: