OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update youtube.py

Browse Source
This commit is contained in:
Classic298 2025-06-10 12:46:21 +02:00 committed by GitHub
parent 17e100661c
commit 3216e8f904
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
backend/open_webui/retrieval/loaders/youtube.py
View File

@ -40,13 +40,11 @@ def _parse_video_id(url: str) -> Optional[str]:
video_id = ids if isinstance(ids, str) else ids[0] video_id = ids if isinstance(ids, str) else ids[0]
else: else:
return None return None
elif parsed_url.netloc == "youtu.be":
video_id = parsed_url.path.lstrip("/").split("?")[0]
else: else:
path = parsed_url.path.lstrip("/") path = parsed_url.path.lstrip("/")
video_id = path.split("/")[-1].split("?")[0] video_id = path.split("/")[-1]
if len(video_id) != 11: if len(video_id) != 11: # Video IDs are 11 characters long
return None return None
return video_id return video_id
@ -111,19 +109,19 @@ class YoutubeLoader:
# Try each language in order of priority # Try each language in order of priority
for lang in self.language: for lang in self.language:
try: try:
try: transcript = transcript_list.find_transcript([lang])
transcript = transcript_list.find_manually_created_transcript([lang]) if transcript.is_generated:
log.debug(f"Found manual transcript for language '{lang}'") log.debug(f"Found generated transcript for language '{lang}'")
except NoTranscriptFound: try:
transcript = transcript_list.find_generated_transcript([lang]) transcript = transcript_list.find_manually_created_transcript(
log.debug(f"Found auto-generated transcript for language '{lang}'") [lang]
)
log.debug(f"Found transcript for language '{lang}'") log.debug(f"Found manual transcript for language '{lang}'")
try: except NoTranscriptFound:
transcript_pieces: List[Dict[str, Any]] = transcript.fetch() log.debug(
except ParseError: f"No manual transcript found for language '{lang}', using generated"
log.debug(f"Empty or invalid transcript for language '{lang}'") )
continue pass
log.debug(f"Found transcript for language '{lang}'") log.debug(f"Found transcript for language '{lang}'")
try: try: