update cli_demo.py

This commit is contained in:
hzg0601 2023-06-15 13:46:11 +08:00
parent b262612248
commit c42c0cbf87
1 changed files with 2 additions and 0 deletions

View File

@ -38,6 +38,8 @@ def main():
temp = local_doc_qa.init_knowledge_vector_store(filepath) temp = local_doc_qa.init_knowledge_vector_store(filepath)
if temp is not None: if temp is not None:
vs_path,loaded_files = temp vs_path,loaded_files = temp
# 如果loaded_files和len(filepath)不一致,则说明部分文件没有加载成功
# 如果是路径错误,则应该支持重新加载
if len(loaded_files) != len(filepath): if len(loaded_files) != len(filepath):
reload_flag = eval(input("部分文件加载失败若提示路径不存在可重新加载是否重新加载输入True或False: ")) reload_flag = eval(input("部分文件加载失败若提示路径不存在可重新加载是否重新加载输入True或False: "))
if reload_flag: if reload_flag: