Kaynağa Gözat

Add support for folder path as input

imClumsyPanda 2 yıl önce
ebeveyn
işleme
55071c79ef
1 değiştirilmiş dosya ile 8 ekleme ve 3 silme
  1. 8 3
      knowledge_based_chatglm.py

+ 8 - 3
knowledge_based_chatglm.py

@@ -55,9 +55,14 @@ def init_knowledge_vector_store(filepath:str):
         print("路径不存在")
         print("路径不存在")
         return None
         return None
     elif os.path.isfile(filepath):
     elif os.path.isfile(filepath):
-        loader = UnstructuredFileLoader(filepath, mode="elements")
-        docs = loader.load()
-        print(f"{os.path.split(filepath)[-1]} 已成功加载")
+        file = os.path.split(filepath)[-1]
+        try:
+            loader = UnstructuredFileLoader(filepath, mode="elements")
+            docs = loader.load()
+            print(f"{file} 已成功加载")
+        except:
+            print(f"{file} 未能成功加载")
+            return None
     elif os.path.isdir(filepath):
     elif os.path.isdir(filepath):
         docs = []
         docs = []
         for file in os.listdir(filepath):
         for file in os.listdir(filepath):