Upload tools.py

src/tools.py

@@ -2,7 +2,7 @@ import json
 import os
 
 from langchain_community.document_loaders import \
-    AssemblyAIAudioTranscriptLoader, WikipediaLoader
+    AssemblyAIAudioTranscriptLoader, WikipediaLoader, YoutubeLoader
 from langchain_core.tools import tool
 
 
@@ -106,3 +106,22 @@ def audio_transcript(
     )
     docs = loader.load()
     return docs[0].page_content if docs else "No transcription available."
+
+
+@tool
+def youtube_transcript(
+    video_url: str
+) -> str:
+    """
+    Get the transcript of a YouTube video.
+
+    Args:
+        video_url: The URL of the YouTube video.
+    """
+    loader = YoutubeLoader(
+        url=video_url,
+        language="en",
+        add_video_info=False
+    )
+    docs = loader.load()
+    return docs[0].page_content if docs else "No transcript available."