jianghuyihei commited on
Commit
6f5caef
·
1 Parent(s): 1bb14f7

update api

Browse files
Files changed (1) hide show
  1. agents.py +3 -3
agents.py CHANGED
@@ -6,12 +6,12 @@ from prompts import *
6
  from utils import extract
7
 
8
 
9
- def get_llm(model = "gpt4o-0513"):
10
  return openai_llm(model)
11
 
12
  def get_llms():
13
- main_llm = get_llm("gpt4o-0513")
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):