Przeglądaj źródła

Merge pull request #18419 from ivanostanin/fix-youtube-proxy-passing

fix: pass youtube_proxy as a `GenericProxyConfig` type object
Tim Baek 5 miesięcy temu
rodzic
commit
680361a88c
1 zmienionych plików z 2 dodań i 4 usunięć
  1. 2 4
      backend/open_webui/retrieval/loaders/youtube.py

+ 2 - 4
backend/open_webui/retrieval/loaders/youtube.py

@@ -83,6 +83,7 @@ class YoutubeLoader:
                 TranscriptsDisabled,
                 YouTubeTranscriptApi,
             )
+            from youtube_transcript_api.proxies import (GenericProxyConfig)
         except ImportError:
             raise ImportError(
                 'Could not import "youtube_transcript_api" Python package. '
@@ -90,10 +91,7 @@ class YoutubeLoader:
             )
 
         if self.proxy_url:
-            youtube_proxies = {
-                "http": self.proxy_url,
-                "https": self.proxy_url,
-            }
+            youtube_proxies = GenericProxyConfig(http_url=self.proxy_url, https_url=self.proxy_url)
             log.debug(f"Using proxy URL: {self.proxy_url[:14]}...")
         else:
             youtube_proxies = None