Update agent.py

agent.py

@@ -10,7 +10,7 @@ def run_agent(model_id, level, question, file_name):
         max_steps = 5
     elif level == 2:
         max_steps = 10
-    else:
+    elif level == 3:
         max_steps = 20
 
     print("###")
@@ -19,13 +19,13 @@ def run_agent(model_id, level, question, file_name):
     print("###")
     
     agent = CodeAgent(
-            add_base_tools=True,
-            additional_authorized_imports=["pandas", "numpy"],
-            max_steps=max_steps,
-            model=OpenAIServerModel(model_id=model_id),
-            tools=[DuckDuckGoSearchTool(), 
-                   WikipediaSearchTool()],
-            verbosity_level=2
+            add_base_tools = True,
+            additional_authorized_imports = ["pandas", "numpy"],
+            max_steps = max_steps,
+            model = OpenAIServerModel(model_id = model_id),
+            tools = [DuckDuckGoSearchTool(), 
+                     WikipediaSearchTool()],
+            verbosity_level = 2
         )
     
     return agent.run(question)