jianghuyihei commited on
Commit
2d7cd9d
1 Parent(s): 1b42af4

update api

Browse files
Files changed (1) hide show
  1. agents.py +2 -2
agents.py CHANGED
@@ -10,8 +10,8 @@ def get_llm(model,deployment ,endpoint,api_key):
10
  return openai_llm(model,deployment,endpoint,api_key)
11
 
12
  def get_llms():
13
- main_llm = get_llm("gpt4o (05-13)","gpt-4o-0513",os.environ.get("AZURE_OPENAI_API_ENDPOINT",None),os.environ.get("AZURE_OPENAI_API",None))
14
- cheap_llm = get_llm("gpt-4o-mini (0718)","gpt-4o-mini-0718",os.environ.get("CHEAP_AZURE_OPENAI_API_ENDPOINT",None),os.environ.get("CHEAP_AZURE_OPENAI_API",None))
15
  return main_llm,cheap_llm
16
 
17
  def judge_idea(i,j,idea0,idea1,topic,llm):
 
10
  return openai_llm(model,deployment,endpoint,api_key)
11
 
12
  def get_llms():
13
+ main_llm = get_llm("gpt4o (05-13)","gpt-4o-0513",os.environ.get("AZURE_OPENAI_API_ENDPOINT",None),os.environ.get("AZURE_OPENAI_KEY",None))
14
+ cheap_llm = get_llm("gpt-4o-mini (0718)","gpt-4o-mini-0718",os.environ.get("CHEAP_AZURE_OPENAI_API_ENDPOINT",None),os.environ.get("CHEAP_AZURE_OPENAI_KEY",None))
15
  return main_llm,cheap_llm
16
 
17
  def judge_idea(i,j,idea0,idea1,topic,llm):