correct name of doc/docx loader

BBC-Esq · Jan 7, 2024 · cbd4766 · cbd4766
1 parent 73dd68d
commit cbd4766
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/constants.py b/src/constants.py
@@ -273,7 +273,8 @@
 
 DOCUMENT_LOADERS = {
     ".pdf": "PyMuPDFLoader",
-    ".docx": "Docx2txtLoader",
+    ".docx": "UnstructuredWordDocumentLoader",
+    ".doc": "UnstructuredWordDocumentLoader",
     ".txt": "TextLoader",
     ".enex": "EverNoteLoader",
     ".epub": "UnstructuredEPubLoader",

diff --git a/src/document_processor.py b/src/document_processor.py
@@ -7,7 +7,7 @@
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.document_loaders import (
     PyMuPDFLoader,
-    Docx2txtLoader,
+    UnstructuredWordDocumentLoader,
     TextLoader,
     EverNoteLoader,
     UnstructuredEPubLoader,
@@ -67,7 +67,7 @@ def load_single_document(file_path: Path) -> Document:
         elif file_extension == ".epub":
             loader = UnstructuredEPubLoader(str(file_path), mode="single", strategy="fast")
         elif file_extension == ".docx":
-            loader = Docx2txtLoader(str(file_path), mode="single", strategy="fast")
+            loader = UnstructuredWordDocumentLoader(str(file_path), mode="single", strategy="fast")
         elif file_extension == ".rtf":
             loader = UnstructuredRTFLoader(str(file_path), mode="single", strategy="fast")
         elif file_extension == ".odt":