Преглед на файлове

[Bugfix] fix import youtube allowed netlocks by defining them locally (#1191)

Co-authored-by: Deven Patel <deven298@yahoo.com>
Deven Patel преди 1 година
родител
ревизия
e572b5a3dc
променени са 1 файла, в които са добавени 9 реда и са изтрити 3 реда
  1. 9 3
      embedchain/utils/misc.py

+ 9 - 3
embedchain/utils/misc.py

@@ -201,10 +201,16 @@ def detect_datatype(source: Any) -> DataType:
     formatted_source = format_source(str(source), 30)
 
     if url:
-        from langchain.document_loaders.youtube import \
-            ALLOWED_NETLOCK as YOUTUBE_ALLOWED_NETLOCS
+        YOUTUBE_ALLOWED_NETLOCKS = {
+            "www.youtube.com",
+            "m.youtube.com",
+            "youtu.be",
+            "youtube.com",
+            "vid.plus",
+            "www.youtube-nocookie.com",
+        }
 
-        if url.netloc in YOUTUBE_ALLOWED_NETLOCS:
+        if url.netloc in YOUTUBE_ALLOWED_NETLOCKS:
             logging.debug(f"Source of `{formatted_source}` detected as `youtube_video`.")
             return DataType.YOUTUBE_VIDEO