update tool description -- increase delay

agent.py

@@ -43,16 +43,16 @@ tavily_extract_tool = TavilyExtract()
 
 
 @tool
-def read_excel_file(excel_file_path: str) -> str:
+def read_excel_file(file_name: str) -> str:
     """Read the content of excel file.
 
     Args:
-      excel_file_path: path to excel file
+      file_name: name of excel file
     """
     # Load the Excel file using pandas
     try:
         # Read the Excel file
-        df = pd.read_excel(excel_file_path, sheet_name=None)  # sheet_name=None loads all sheets
+        df = pd.read_excel(file_name, sheet_name=None)  # sheet_name=None loads all sheets
         
         # Convert each sheet to a dictionary of rows
         json_output = {}
@@ -68,14 +68,14 @@ def read_excel_file(excel_file_path: str) -> str:
 
 
 @tool
-def transcribe_audio(file_path: str) -> str:
+def transcribe_audio(file_name: str) -> str:
   """Transcribe the audio file into text.
   
   Args:
-    file_path: path to audio file
+    file_name: name of audio file
   """
   config = aai.TranscriptionConfig(speech_model=aai.SpeechModel.best)
-  transcript = aai.Transcriber(config=config).transcribe(file_path)
+  transcript = aai.Transcriber(config=config).transcribe(file_name)
   if transcript.status == "error":
     raise RuntimeError(f"Transcription failed: {transcript.error}")
   return f"Here is the transcript: {transcript.text}"
@@ -92,6 +92,7 @@ def wiki_search(query: str) -> str:
             f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
             for doc in search_docs
         ])
+    print(formatted_search_docs)
     return {"wiki_results": formatted_search_docs}
 
 # @tool
@@ -122,8 +123,8 @@ sys_msg = SystemMessage(content=system_prompt)
 
 tools = [
     wiki_search,
-    # tavily_search_tool,
-    # tavily_extract_tool,
+    tavily_search_tool,
+    tavily_extract_tool,
     repl_tool,
     read_excel_file,
     transcribe_audio
@@ -168,9 +169,9 @@ def get_answer(query):
 
 if __name__ == "__main__":
     question = "In the video https://www.youtube.com/watch?v=L1vXCYZAYYM, what is the highest number of bird species to be on camera simultaneously?"
-    question = "Hi, I was out sick from my classes on Friday, so I'm trying to figure out what I need to study for my Calculus mid-term next week. My friend from class sent me an audio recording of Professor Willowbrook giving out the recommended reading for the test, but my headphones are broken :(\n\nCould you please listen to the recording for me and tell me the page numbers I'm supposed to go over? I've attached a file called Homework.mp3 that has the recording. Please provide just the page numbers as a comma-delimited list. And please provide the list in ascending order."
-    question = "What is the first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists?"
-    question = "Where were the Vietnamese specimens described by Kuznetzov in Nedoshivina's 2010 paper eventually deposited? Just give me the city name without abbreviations."
+    # question = "Hi, I was out sick from my classes on Friday, so I'm trying to figure out what I need to study for my Calculus mid-term next week. My friend from class sent me an audio recording of Professor Willowbrook giving out the recommended reading for the test, but my headphones are broken :(\n\nCould you please listen to the recording for me and tell me the page numbers I'm supposed to go over? I've attached a file called Homework.mp3 that has the recording. Please provide just the page numbers as a comma-delimited list. And please provide the list in ascending order."
+    # question = "What is the first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists?"
+    # question = "Where were the Vietnamese specimens described by Kuznetzov in Nedoshivina's 2010 paper eventually deposited? Just give me the city name without abbreviations."
     question = "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia."
     # getmessages = [HumanMessage(content=question)]
     # messages = graph.invoke({"messages": messages})

app.py

@@ -77,7 +77,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
     for item in questions_data:
-        time.sleep(5)
+        time.sleep(10)
         task_id = item.get("task_id")
         question_text = item.get("question")
         if not task_id or question_text is None: