Update agent.py

agent.py

@@ -5,7 +5,7 @@ from smolagents import (
     WikipediaSearchTool
 )
 
-def run_agent(model_id, level, question, file_name):
+def run_agent(model_id, level, question, file_name, ground_truth):
     if level == 1:
         max_steps = 5
     elif level == 2:
@@ -27,5 +27,7 @@ def run_agent(model_id, level, question, file_name):
                      WikipediaSearchTool()],
             verbosity_level = 2
         )
+
+    answer = agent.run(question)
     
-    return agent.run(question)
+    return answer, answer == ground_truth