handle audio, remove extra sub-agent

basic_agent.py

@@ -26,12 +26,13 @@ Remember - The FINAL ANSWER must be a string, a number or a comma separated list
 
 @smolagents.tool
 def convert_from_url(url: str) -> str:
-  """Convert a file at a url into a markdown string.
+  """Convert data at a url into a markdown string.
 
-   This method takes a path to a file and converts it into
-   a markdown string. It can handle a variety of file
-   types as input including pdf, excel, xlsx, mp3, csv,
-   powerpoint, etc.
+   This method takes a url as input and converts the contents
+   into a markdown. It can convert file formats such as
+   pdf, excel, xlsx, csv, powerpoint, ppt, etc.
+
+   It can also transcribe audio files such as mp3.
 
    Args:
      url: the url to access the file. If the file is on the
@@ -42,7 +43,7 @@ def convert_from_url(url: str) -> str:
      string that is in the markdown format.
    """
   # Don't need third-party plugins.
-  md = markitdown.MarkItDown(enable_plugins=Disable)
+  md = markitdown.MarkItDown(enable_plugins=False)
   result = md.convert(url)
   return result.text_content
 
@@ -89,22 +90,6 @@ class BasicAgent:
 
     print(f"NEW5: BasicAgent {self.manager_model_id=} {self.code_model_id=}")
 
-    file_convert_tool = [
-        convert_from_url,
-        smolagents.FinalAnswerTool(),
-    ]
-    self.convert_files_to_markdown = smolagents.CodeAgent(
-        name="convert_files_to_markdown",
-        description=
-        "Given a url, it converts the file from various formats (such as pdf, excel, powerpoint, audio, csv, etc) into the markdown format",
-        model=self.code_model,
-        tools=file_convert_tool,
-        max_steps=6,
-        verbosity_level=2,
-        planning_interval=None,
-        additional_authorized_imports=["markitdown"],
-    )
-
     math_tools = [
         smolagents.FinalAnswerTool(),
     ]
@@ -153,23 +138,32 @@ class BasicAgent:
         additional_authorized_imports=["wikipedia-api"],
     )
 
+    manager_tools = [
+        convert_from_url,
+        smolagents.FinalAnswerTool(),
+    ]
     self.manager_agent = smolagents.CodeAgent(
         name="manager_agent",
         description="Manger of other agents",
-        tools=[smolagents.FinalAnswerTool()],
+        tools=manager_tools,
         model=self.manager_model,
         add_base_tools=True,
         max_steps=10,
         verbosity_level=2,
         planning_interval=None,
         additional_authorized_imports=[
-            "duckduckgo_search", "wikipedia-api", "markitdown"
+            "duckduckgo_search",
+            "wikipedia-api",
+            "markitdown",
+            "requests",
+            "pandas",
+            "io",
         ],
         managed_agents=[
             self.web_search_agent,
             self.wikipedia_search_agent,
             self.math_agent,
-            self.convert_files_to_markdown,
+            #            self.convert_files_to_markdown,
         ],
         final_answer_checks=[self.check_results],
     )

basic_agent_test.py

@@ -9,6 +9,8 @@ LOG = logging.getLogger(__name__)
 
 QUESTIONS_PATH = "../questions.json"
 
+DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
+
 # Local:
 #  https://huggingface.co/docs/smolagents/tutorials/inspect_runs
 #  https://cobusgreyling.medium.com/introduce-inspectability-to-huggingface-smolagents-571bd3f8da4c
@@ -53,7 +55,11 @@ print(f"{len(questions)=}")
 #  'task_id': '7bd855d8-463d-4ed5-93ca-5fe35145f733'},
 #
 
-want = 7
+# 18 has an xlsx file
+# 13 has an mp3 file
+# 11 has a py file
+# 9 has an mp3 file
+want = 9
 question = questions[want]
 
 q_task_id = question["task_id"]
@@ -61,12 +67,12 @@ q_text = question["question"]
 q_filename = question.get("file_name", "")
 
 if q_filename:
-  file_url = f"{api_url}/files/{task_id}"
+  file_url = f"{DEFAULT_API_URL}/files/{q_task_id}"
   print("Question {task_id=} has attachment: {file_url=}")
   quest_with_url = f"{q_text}\nFile url: {file_url}"
 else:
   quest_with_url = q_text
 
-answer = ba(question_text)
+answer = ba(quest_with_url)
 
 LOG.warning(f"{answer=}")