Update agent.py

agent.py

@@ -7,7 +7,8 @@ from smolagents import (
     WikipediaSearchTool
 )
 
-MODEL_ID = "gpt-4o"
+MODEL_ID_1 = "gpt-4o"
+MODEL_ID_2 = "o3-mini"
 
 def run_agent(level, question, file_name):
     if level == 1:
@@ -21,11 +22,11 @@ def run_agent(level, question, file_name):
             add_base_tools = True,
             additional_authorized_imports = ["pandas", "numpy"],
             max_steps = max_steps,
-            model = OpenAIServerModel(model_id = MODEL_ID),
-            planning_interval=3,
+            model = OpenAIServerModel(model_id = MODEL_ID_1),
+            #planning_interval=3,
             tools = [DuckDuckGoSearchTool(), 
                      WikipediaSearchTool()],
-            verbosity_level = 2
+            #verbosity_level = 2
         )
 
     answer = agent.run(question)
@@ -36,10 +37,9 @@ def get_final_answer(question, answer):
     client = OpenAI()
 
     prompt_template = """
-You are a GAIA benchmark assistant, who is given a question and an answer:
+You are a GAIA benchmark assistant, who is given a question and an answer. Read the question carefully. You must only give the final answer requested. Do not include explanations, steps, reasoning, or additional text. Be direct and specific.
 **Question:** """ + str(question) + """
 **Answer:** """ + str(answer) + """
-**Instructions:** Read the question carefully. You must only give the final answer requested. Do not include explanations, steps, reasoning, or additional text. Be direct and specific.
 **Example:** If asked "What is the capital of France?", respond with "Paris".
 **Final answer:**:
 
@@ -48,7 +48,7 @@ You are a GAIA benchmark assistant, who is given a question and an answer:
     completion = client.chat.completions.create(
         max_tokens = 100,
         messages = [{"role": "user", "content": [{"type": "text", "text": prompt_template}]}],
-        model = MODEL_ID,
+        model = MODEL_ID_2,
         temperature = 0
     )