diff --git a/server/knowledge_base/utils.py b/server/knowledge_base/utils.py index 8582c9c..a8a9bcc 100644 --- a/server/knowledge_base/utils.py +++ b/server/knowledge_base/utils.py @@ -197,7 +197,7 @@ class KnowledgeFile: print(f"{self.document_loader_name} used for {self.filepath}") try: - if self.document_loader_name in []: + if self.document_loader_name in ["RapidOCRPDFLoader", "RapidOCRLoader"]: document_loaders_module = importlib.import_module('document_loaders') else: document_loaders_module = importlib.import_module('langchain.document_loaders')