Update tools.py

tools.py

@@ -36,11 +36,10 @@ datetime_tools = [current_date, day_of_week, days_until]
 @tool
 def transcribe_audio(audio_file: str, file_extension: str) -> str:
     """ Transcribes an audio file to text
-
     Args:
         audio_file (str): local file path to the audio file (.mp3, .m4a, etc.)
         file_extension (str): file extension of the audio, e.g. mp3
-    
+
     Returns:
         str: The transcribed text from the audio.
     """
@@ -71,21 +70,18 @@ def transcribe_audio(audio_file: str, file_extension: str) -> str:
 
 # TESTTTTT
 @tool
-def transcribe_youtube(page_url: str) -> str:
-    """Get the transcript of a YouTube video
+def transcribe_youtube(youtube_url: str) -> str:
+    """ Transcribes a YouTube video
     Args:
-        page_url (str): YouTube URL of the video
+        youtube_url (str): youtube video's url
     """
     try:
-        # get video ID from URL
-        video_id = extract.video_id(page_url)
-
-        # get transcript
-        ytt_api = YouTubeTranscriptApi()
-        transcript = ytt_api.fetch(video_id)
+        video_id = extract.video_id(youtube_url)
+        youtube_api = YouTubeTranscriptApi()
+        transcript = youtube_api.fetch(video_id)
 
         # keep only text
         txt = '\n'.join([s.text for s in transcript.snippets])
         return txt
     except Exception as e:
-        return f"get_youtube_transcript failed: {e}"
+        return f"transcribe_youtube failed: {e}"