update imports

.gitignore

@@ -1,3 +1,3 @@
 *.pyc
 __pycache__/*
-.env
+.env

requirements.txt

@@ -1,6 +1,6 @@
 gradio
 requests
-smolagents[litellm, transformers, wikipedia, youtube, duckduckgo, langchain, langgraph]
+smolagents[litellm, transformers, audio, wikipedia, youtube, duckduckgo, langchain, langgraph]
 requests
 python-dotenv
 datasets
@@ -15,4 +15,5 @@ langchain-chroma
 wikipedia
 youtube-transcript-api
 wikipedia-api
-duckduckgo-search
+duckduckgo-search
+markdownify

task_force.py

@@ -107,6 +107,16 @@ SCOUT_TOOLS = [
     YouTubeTranscript(),
 ]
 
+AUTHORIZED_IMPORTS = [
+    "pytube",
+    "opencv-python",
+    "pandas",
+    "numpy",
+    "bs4",
+    "requests",
+    "markdownify"
+]
+
 # a genius hacker that can use the power of the web to answer questions
 HACKER = CodeAgent(
     name="hacker",
@@ -115,7 +125,7 @@ HACKER = CodeAgent(
     tools=HACKER_TOOLS,
     verbosity_level=VERBOSITY_LEVEL,
     add_base_tools=True,
-    additional_authorized_imports=["pandas", "numpy","bs4", "requests"],
+    additional_authorized_imports=AUTHORIZED_IMPORTS,
     max_steps=6
 )
 
@@ -126,7 +136,7 @@ SCOUT = CodeAgent(
     tools=SCOUT_TOOLS,
     verbosity_level=VERBOSITY_LEVEL,
     add_base_tools=True,
-    additional_authorized_imports=["pytube", "opencv-python", "pandas", "numpy","bs4", "requests"],
+    additional_authorized_imports=AUTHORIZED_IMPORTS,
     max_steps=6
 )
 
@@ -137,6 +147,7 @@ THINKER = CodeAgent(
     tools=[FinalAnswerTool()],
     verbosity_level=VERBOSITY_LEVEL,
     add_base_tools=True,
+    additional_authorized_imports=AUTHORIZED_IMPORTS,
     max_steps=6
 )
 
@@ -150,6 +161,7 @@ COMMANDER = CodeAgent(
     managed_agents=[HACKER, SCOUT, THINKER],
     planning_interval=2,
     verbosity_level=VERBOSITY_LEVEL,
+    additional_authorized_imports=AUTHORIZED_IMPORTS,
     prompt_templates=custom_prompts.commander_prompt_templates,
     max_steps=12,   
 )