Define parameters used for required components
class YoutubeLoaderComponent(CustomComponent):
display_name: str = "Youtube Loader"
description: str = "Downloads the YouTube transcripts and video information."
documentation: str = (
"https://python.langchain.com/docs/integrations/providers/youtube"
)
beta = False
def build_config(self):
return {
"youtube_url": {
"display_name": "Video URL",
"is_list": False,
"required": True,
"value": "",
},
"language": {
"display_name": "Language",
"is_list": False,
"required": True,
"value": "en",
"info": "language code to extract transcript. please check subtitles/cc to know available transcripts.",
},
}
def build(self, youtube_url: str, language: str) -> List[Document]:
loader_instance = YoutubeLoader.from_youtube_url(
youtube_url=youtube_url, add_video_info=False, language=language
)
return loader_instance.load()
Last updated
Was this helpful?