Update agent.py

agent.py

@@ -13,11 +13,11 @@ MODEL_ID = "o4-mini"
 
 def run_agent(level, question, file_name):
     if level == 1:
-        max_steps = 5
-    elif level == 2:
         max_steps = 10
+    elif level == 2:
+        max_steps = 25
     elif level == 3:
-        max_steps = 20
+        max_steps = 50
    
     agent = CodeAgent(
             add_base_tools = True,