diff --git a/backend/apps/rag/main.py b/backend/apps/rag/main.py index 163f1b0fa..1e50ef20b 100644 --- a/backend/apps/rag/main.py +++ b/backend/apps/rag/main.py @@ -21,7 +21,7 @@ from langchain_community.document_loaders import ( TextLoader, PyPDFLoader, CSVLoader, - UnstructuredHTMLLoader, + BSHTMLLoader, Docx2txtLoader, UnstructuredEPubLoader, UnstructuredWordDocumentLoader, @@ -404,7 +404,7 @@ def get_loader(filename: str, file_content_type: str, file_path: str): elif file_ext == "xml": loader = UnstructuredXMLLoader(file_path) elif file_ext in ["htm", "html"]: - loader = UnstructuredHTMLLoader(file_path) + loader = BSHTMLLoader(file_path) elif file_ext == "md": loader = UnstructuredMarkdownLoader(file_path) elif file_content_type == "application/epub+zip":