add youtube transcript

Tools/tools.py

@@ -5,6 +5,8 @@ from langchain_community.vectorstores import SupabaseVectorStore
 from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_core.tools import tool
 from langchain.tools.retriever import create_retriever_tool
+from langchain_community.document_loaders import YoutubeLoader
+from langchain_community.document_loaders.youtube import TranscriptFormat
 
 @tool
 def add(a: int, b:int) -> int:
@@ -49,13 +51,13 @@ def divide(a: int, b: int) -> int:
     return a / b
 
 @tool
-def remainder(a: int, b: int) -> int:
-    """ left over of division
-    args: 
-        a: first integer
-        b: second integer
+def modulus(a: int, b: int) -> int:
+    """
+    Get the modulus of two numbers.
+    Args:
+        a (int): the first number
+        b (int): the second number
     """
-    print ("remainder")
     return a % b
 
 @tool
@@ -109,6 +111,23 @@ def webSearch(searchQuery:str) -> str:
     )
     return {"web_search": formatted_results}
 
+@tool
+def youtubeTranscript(youtubeURL:str) -> str:
+    """
+      obtain youtube transcript by passing in the youtube url
+
+      args: 
+        youtubeURL: youtube url to pull out the transcript
+    """
+    print("youtube_transcript")
+    loader = YoutubeLoader.from_youtube_url(
+        youtubeURL,
+        add_video_info=True,
+        transcript_format=TranscriptFormat.CHUNKS,
+        chunk_size_seconds=30,
+    )
+    formatted_results = "\n\n".join(map(repr, loader.load()))
+    return {"Youtube transcript":formatted_results}