diff --git a/cognee/infrastructure/loaders/external/__init__.py b/cognee/infrastructure/loaders/external/__init__.py index 6bf9f9200..785338c09 100644 --- a/cognee/infrastructure/loaders/external/__init__.py +++ b/cognee/infrastructure/loaders/external/__init__.py @@ -27,3 +27,10 @@ try: __all__.append("AdvancedPdfLoader") except ImportError: pass + +try: + from .beautiful_soup_loader import BeautifulSoupLoader + + __all__.append("BeautifulSoupLoader") +except ImportError: + pass diff --git a/cognee/infrastructure/loaders/external/bs4_loader.py b/cognee/infrastructure/loaders/external/beautiful_soup_loader.py similarity index 100% rename from cognee/infrastructure/loaders/external/bs4_loader.py rename to cognee/infrastructure/loaders/external/beautiful_soup_loader.py diff --git a/cognee/infrastructure/loaders/supported_loaders.py b/cognee/infrastructure/loaders/supported_loaders.py index d103babe3..156253b53 100644 --- a/cognee/infrastructure/loaders/supported_loaders.py +++ b/cognee/infrastructure/loaders/supported_loaders.py @@ -23,3 +23,10 @@ try: supported_loaders[AdvancedPdfLoader.loader_name] = AdvancedPdfLoader except ImportError: pass + +try: + from cognee.infrastructure.loaders.external import BeautifulSoupLoader + + supported_loaders[BeautifulSoupLoader.loader_name] = BeautifulSoupLoader +except ImportError: + pass