Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -10,6 +10,7 @@ from optillm.self_consistency import advanced_self_consistency_approach
|
|
10 |
from optillm.rstar import RStar
|
11 |
from optillm.plansearch import plansearch
|
12 |
from optillm.leap import leap
|
|
|
13 |
|
14 |
|
15 |
API_KEY = os.environ.get("OPENROUTER_API_KEY")
|
@@ -59,6 +60,8 @@ def respond(message, history, model, approach, system_message, max_tokens, tempe
|
|
59 |
final_response = plansearch(system_prompt, initial_query, client, model)[0]
|
60 |
elif approach == 'leap':
|
61 |
final_response = leap(system_prompt, initial_query, client, model)
|
|
|
|
|
62 |
|
63 |
return final_response
|
64 |
|
@@ -85,7 +88,7 @@ def create_model_dropdown():
|
|
85 |
|
86 |
def create_approach_dropdown():
|
87 |
return gr.Dropdown(
|
88 |
-
["none", "leap", "plansearch", "rstar", "cot_reflection", "rto", "self_consistency", "z3"],
|
89 |
value="none", label="Approach"
|
90 |
)
|
91 |
|
|
|
10 |
from optillm.rstar import RStar
|
11 |
from optillm.plansearch import plansearch
|
12 |
from optillm.leap import leap
|
13 |
+
from optillm.reread import re2_approach
|
14 |
|
15 |
|
16 |
API_KEY = os.environ.get("OPENROUTER_API_KEY")
|
|
|
60 |
final_response = plansearch(system_prompt, initial_query, client, model)[0]
|
61 |
elif approach == 'leap':
|
62 |
final_response = leap(system_prompt, initial_query, client, model)
|
63 |
+
elif approach == 're2':
|
64 |
+
final_response = re2_approach(system_prompt, initial_query, client, model)
|
65 |
|
66 |
return final_response
|
67 |
|
|
|
88 |
|
89 |
def create_approach_dropdown():
|
90 |
return gr.Dropdown(
|
91 |
+
["none", "leap", "plansearch", "rstar", "cot_reflection", "rto", "self_consistency", "z3", "re2"],
|
92 |
value="none", label="Approach"
|
93 |
)
|
94 |
|