Update agent.py

agent.py

@@ -5,15 +5,22 @@ from smolagents import (
     WikipediaSearchTool
 )
 
-def run_agent(model_id, question):
+def run_agent(model_id, level, question, file_name):
+    if level == 1:
+        max_steps = 5
+    elif level == 2:
+        max_steps = 10
+    elif level == 3:
+        max_steps = 20
+        
     agent = CodeAgent(
+            add_base_tools=True,
+            additional_authorized_imports=["pandas", "numpy"]
+            max_steps=max_steps,
             model=OpenAIServerModel(model_id=model_id),
             tools=[DuckDuckGoSearchTool(), 
                    WikipediaSearchTool()],
-            add_base_tools=True,
-            max_steps=5,
             verbosity_level=2,
-            additional_authorized_imports=["pandas", "numpy"]
         )
     
     return agent.run(question)