RYDE-WORK/Langchain-Chatchat
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/Langchain-Chatchat.git synced 2026-01-19 13:23:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add support for folder path as input

Browse Source
This commit is contained in:
imClumsyPanda 2023-04-11 23:02:24 +08:00
parent 196904872c
commit 55071c79ef
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
knowledge_based_chatglm.py
View File

@ -55,9 +55,14 @@ def init_knowledge_vector_store(filepath:str):
print("路径不存在")
return None
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):
docs = []
for file in os.listdir(filepath):