From d6dce956825b8fe96643ca4f56af6ce9e79d4182 Mon Sep 17 00:00:00 2001 From: imClumsyPanda Date: Fri, 1 Sep 2023 14:15:13 +0800 Subject: [PATCH] bug fix --- server/knowledge_base/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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')