提交 55071c79 作者: imClumsyPanda

Add support for folder path as input

上级 19690487
...@@ -55,9 +55,14 @@ def init_knowledge_vector_store(filepath:str): ...@@ -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):
file = os.path.split(filepath)[-1]
try:
loader = UnstructuredFileLoader(filepath, mode="elements") loader = UnstructuredFileLoader(filepath, mode="elements")
docs = loader.load() docs = loader.load()
print(f"{os.path.split(filepath)[-1]} 已成功加载") 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):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论