Update agent.py

agent.py

@@ -34,7 +34,7 @@ def run_agent(level, question, file_name):
     
     answer = agent.run(question)
 
-    return get_final_answer(question, answer)
+    return answer, get_final_answer(question, answer)
 
 def get_final_answer(question, answer):
     client = OpenAI()