Update agent.py

agent.py

@@ -11,30 +11,21 @@ from smolagents import (
 
 MODEL_ID = "o4-mini"
 
-def run_agent(level, question, file_name):
-    if level == 1:
-        max_steps = 10
-    elif level == 2:
-        max_steps = 25
-    elif level == 3:
-        max_steps = 50
-   
+def run_agent(question, file_name):
     agent = CodeAgent(
             add_base_tools = True,
             additional_authorized_imports = ["pandas", "numpy"],
-            max_steps = max_steps,
+            max_steps = 25,
             model = OpenAIServerModel(model_id = MODEL_ID),
             #planning_interval=3,
             tools = [#DuckDuckGoSearchTool(), 
                      WikipediaSearchTool()],
-            verbosity_level = 1
+            verbosity_level = 0
         )
 
-    agent.logger.console.width = 66
-    
     answer = agent.run(question)
 
-    return answer, get_final_answer(question, answer)
+    return get_final_answer(question, answer)
 
 def get_final_answer(question, answer):
     client = OpenAI()
@@ -52,6 +43,4 @@ You are a GAIA benchmark assistant, who is given a question and an answer. Read
         model = MODEL_ID
     )
 
-    final_answer = completion.choices[0].message.content
-
-    return final_answer
+    return completion.choices[0].message.content