huytofu92 commited on
Commit
1db6e4e
·
1 Parent(s): d616b4c
Files changed (1) hide show
  1. community_tools.py +3 -3
community_tools.py CHANGED
@@ -15,14 +15,14 @@ arxiv_tools = [Tool.from_langchain(tool) for tool in load_tools(["arxiv"])]
15
  community_tools = [google_map_tool, wikipedia_tool, *arxiv_tools]
16
 
17
  @tool
18
- def get_youtube_transcript_from_url(video_url: str = None):
19
  """
20
  Get the transcript of a YouTube video
21
  Args:
22
  video_url: The URL of the YouTube video
23
  Returns:
24
- The transcript of the YouTube video
25
- """
26
  youtube_loader = YoutubeLoader(video_url=video_url)
27
  docs = youtube_loader.load()
28
  transcript = docs[0].page_content
 
15
  community_tools = [google_map_tool, wikipedia_tool, *arxiv_tools]
16
 
17
  @tool
18
+ def get_youtube_transcript_from_url(video_url: str = None)->str:
19
  """
20
  Get the transcript of a YouTube video
21
  Args:
22
  video_url: The URL of the YouTube video
23
  Returns:
24
+ The transcript of the YouTube video as a string
25
+ """
26
  youtube_loader = YoutubeLoader(video_url=video_url)
27
  docs = youtube_loader.load()
28
  transcript = docs[0].page_content