Update agent.py
Browse files
agent.py
CHANGED
|
@@ -8,7 +8,7 @@ from smolagents import (
|
|
| 8 |
)
|
| 9 |
|
| 10 |
MODEL_ID_1 = "gpt-4o"
|
| 11 |
-
MODEL_ID_2 = "
|
| 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
|