Skip to content

Commit e36e151

Browse files
committed
prepare for loader class parameter
1 parent eefa8ee commit e36e151

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

base_bot/document_loader.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
from langchain.document_loaders import DirectoryLoader
22
from langchain.document_loaders.base import BaseLoader
3+
from langchain.document_loaders import PyMuPDFLoader
4+
#from langchain.document_loaders import UnstructuredFileLoader
35

46

57
def get_document_loader(docs_path="data/docs") -> BaseLoader:
@@ -11,5 +13,6 @@ def get_document_loader(docs_path="data/docs") -> BaseLoader:
1113
show_progress=True,
1214
use_multithreading=True,
1315
silent_errors=True,
16+
# loader_cls=UnstructuredFileLoader
1417
)
1518
return loader

0 commit comments

Comments
 (0)