Update agent.py

agent.py

@@ -5,7 +5,9 @@ from smolagents import (
     WikipediaSearchTool
 )
 
-def run_agent(model_id, level, question, file_name, ground_truth):
+MODEL_ID = "gpt-4o"
+
+def run_agent(level, question, file_name, ground_truth):
     if level == 1:
         max_steps = 5
     elif level == 2:
@@ -17,7 +19,7 @@ def run_agent(model_id, level, question, file_name, ground_truth):
             add_base_tools = True,
             additional_authorized_imports = ["pandas", "numpy"],
             max_steps = max_steps,
-            model = OpenAIServerModel(model_id = model_id),
+            model = OpenAIServerModel(model_id = MODEL_ID),
             tools = [DuckDuckGoSearchTool(), 
                      WikipediaSearchTool()],
             verbosity_level = 2