bstraehle commited on
Commit
f600343
·
verified ·
1 Parent(s): 90e87ac

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -2
agent.py CHANGED
@@ -8,7 +8,7 @@ from smolagents import (
8
  )
9
 
10
  MODEL_ID_1 = "gpt-4o"
11
- MODEL_ID_2 = "o3-mini"
12
 
13
  def run_agent(level, question, file_name):
14
  if level == 1:
@@ -46,7 +46,6 @@ You are a GAIA benchmark assistant, who is given a question and an answer. Read
46
  """
47
 
48
  completion = client.chat.completions.create(
49
- max_tokens = 100,
50
  messages = [{"role": "user", "content": [{"type": "text", "text": prompt_template}]}],
51
  model = MODEL_ID_2,
52
  temperature = 0
 
8
  )
9
 
10
  MODEL_ID_1 = "gpt-4o"
11
+ MODEL_ID_2 = "o4-mini"
12
 
13
  def run_agent(level, question, file_name):
14
  if level == 1:
 
46
  """
47
 
48
  completion = client.chat.completions.create(
 
49
  messages = [{"role": "user", "content": [{"type": "text", "text": prompt_template}]}],
50
  model = MODEL_ID_2,
51
  temperature = 0