Retrying failed tasks until they get solved, adding arxiv tool

app.py

@@ -20,6 +20,7 @@ from tools import (
     multiply,
     subtract,
     wikipedia_search,
+    arxiv_search,
     search_tool,
 )
 
@@ -52,6 +53,7 @@ class BasicAgent:
             subtract,
             wikipedia_search,
             search_tool,
+            arxiv_search
         ]
         chat_with_tools = chat.bind_tools(tools)
 
@@ -131,16 +133,22 @@ async def answer_all_questions(agent: BasicAgent, questions: list[dict]) -> list
     tasks = [agent.run(*question_info) for question_info in question_infos]
 
     results = await asyncio.gather(*tasks, return_exceptions=True)
-    # Optional: handle exceptions gracefully
-    final_results = []
-    for result in results:
-        if isinstance(result, Exception):
-            final_results.append(f"ERROR: {str(result)}")
-        else:
-            final_results.append(result)
 
+    failed_tasks = any(isinstance(result, Exception) for result in results)
+
+    while failed_tasks:
+        for result_index, result in enumerate(results):
+            if isinstance(result, Exception):
+                print(f"Retrying failed task {result_index + 1}/{len(results)}...")
+                try:
+                    new_result = await agent.run(*question_infos[result_index])
+                    results[result_index] = new_result
+                    print(f"Task {result_index + 1} succeeded on retry.")
+                except Exception as e:
+                    print(f"Task {result_index + 1} failed again: {e}")
+        failed_tasks = any(isinstance(result, Exception) for result in results)
     # Run all coroutines concurrently
-    return final_results
+    return results
 
 async def run_and_submit_all( profile: gr.OAuthProfile | None):
     """

utilities.py

@@ -58,11 +58,11 @@ def prepare_questions(questions_data: list) -> list:
                 df = pd.read_excel(io.BytesIO(file.content))
                 # Step 2: Convert the CSV data to a string
                 csv_data = df.to_string(index=False)
-                question_text += f"\n\nThe following is the content of the CSV file:\n{csv_data}"
+                question_text += f"\n\nThe following content is from the CSV file:\n{csv_data}"
             elif file_name.split(".")[-1] == "mp3":
                 audio_base64 = base64.b64encode(file.content).decode("utf-8")
             else:
-                question_text += f"{file.content}"
+                question_text += f"\n\nThe following content is from the python script:\n{file.content}"
 
         question_info.append((question_text, image_base64, audio_base64))