Update agent.py

agent.py

@@ -1,6 +1,7 @@
 from smolagents import CodeAgent,  LiteLLMModel, tool, load_tool # DuckDuckGoSearchTool, WikipediaSearchTool, HfApiModel, OpenAIServerModel
 from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_community.document_loaders import WikipediaLoader
+from youtube_transcript_api import YouTubeTranscriptApi
 import asyncio
 import os
 import yaml
@@ -101,6 +102,18 @@ def ImageAnalysisTool(image_path: str) -> str:
     return f"The image description: '{response}'"
 
 
+@tool
+def youtube_transcript(url: str) -> str:
+    """
+    Get transcript of YouTube video.
+    Args:
+        url: YouTube video url in ""
+    """    
+    video_id = url.partition("https://www.youtube.com/watch?v=")[2]
+    transcript = YouTubeTranscriptApi.get_transcript(video_id)
+    transcript_text = " ".join([item["text"] for item in transcript])
+    return {"youtube_transcript": transcript_text}
+    
 #@tool
 #class LocalFileAudioTool:
 #    """Tool for transcribing audio files"""
@@ -134,6 +147,7 @@ class MagAgent:
             tools=[
                 wiki_search,
                 web_search,
+                youtube_transcript,
 #                GoogleSearchTool,
 #                DuckDuckGoSearchTool(),
 #                WikipediaSearchTool(),