Spaces:
Running
Running
Commit
·
6f5caef
1
Parent(s):
1bb14f7
update api
Browse files
agents.py
CHANGED
@@ -6,12 +6,12 @@ from prompts import *
|
|
6 |
from utils import extract
|
7 |
|
8 |
|
9 |
-
def get_llm(model = "
|
10 |
return openai_llm(model)
|
11 |
|
12 |
def get_llms():
|
13 |
-
main_llm = get_llm("
|
14 |
-
cheap_llm = get_llm("gpt-4o-mini")
|
15 |
return main_llm,cheap_llm
|
16 |
|
17 |
def judge_idea(i,j,idea0,idea1,topic,llm):
|
|
|
6 |
from utils import extract
|
7 |
|
8 |
|
9 |
+
def get_llm(model = "gpt-4o-0806"):
|
10 |
return openai_llm(model)
|
11 |
|
12 |
def get_llms():
|
13 |
+
main_llm = get_llm("gpt-4o-0806")
|
14 |
+
cheap_llm = get_llm("gpt-4o-mini-0718")
|
15 |
return main_llm,cheap_llm
|
16 |
|
17 |
def judge_idea(i,j,idea0,idea1,topic,llm):
|