Adds exception handling to agent answer retrieval

app.py

@@ -56,19 +56,22 @@ class BasicAgent:
         as you want. The file name is: {file_name}.\n
         """
 
-
+        try:
         # Retrieve the answer from the cache if it exists
-        cached = get_cached_answer(task_id)
-        if cached and cached.get("isCorrect") and cached.get("answer"):
-            print(f"Returning cached correct answer for task_id {task_id}: {cached['answer']}")
-            
-            return str(cached["answer"])
+            cached = get_cached_answer(task_id)
+            if cached and cached.get("isCorrect") and cached.get("answer"):
+                print(f"Returning cached correct answer for task_id {task_id}: {cached['answer']}")
+                
+                return str(cached["answer"])
 
-        answer = await self.agent.run(prompt)
+            answer = await self.agent.run(prompt)
 
-        print(f"Agent returning answer: {answer}")
-        
-        return str(answer)
+            print(f"Agent returning answer: {answer}")
+            
+            return str(answer)
+        except Exception as e:
+            print(f"Error running agent on question {question['task_id']}: {e}")
+            return "AGENT ERROR: " + str(e)
 
 def instantiate_agent():
     try: