Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,8 @@ import os
|
|
3 |
|
4 |
from multi_agent import run_multi_agent
|
5 |
|
6 |
-
|
|
|
7 |
|
8 |
def invoke(openai_api_key, num_turns = 2):
|
9 |
if (openai_api_key == ""):
|
@@ -11,7 +12,7 @@ def invoke(openai_api_key, num_turns = 2):
|
|
11 |
|
12 |
os.environ["OPENAI_API_KEY"] = openai_api_key
|
13 |
|
14 |
-
return run_multi_agent(
|
15 |
|
16 |
gr.close_all()
|
17 |
|
|
|
3 |
|
4 |
from multi_agent import run_multi_agent
|
5 |
|
6 |
+
LLM_WHITE = "gpt-3.5-turbo"
|
7 |
+
LLM_BLACK = "gpt-4o"
|
8 |
|
9 |
def invoke(openai_api_key, num_turns = 2):
|
10 |
if (openai_api_key == ""):
|
|
|
12 |
|
13 |
os.environ["OPENAI_API_KEY"] = openai_api_key
|
14 |
|
15 |
+
return run_multi_agent(LLM_WHITE, LLM_BLACK, num_turns)
|
16 |
|
17 |
gr.close_all()
|
18 |
|