提交 43fae98b 作者: hzg0601

按照review建议进行修改

上级 ba336440
......@@ -197,9 +197,8 @@ class LocalDocQA:
return vs_path, loaded_files
else:
logger.info("文件均未成功加载,请检查依赖包或替换为其他文件再次上传。")
# 若len(docs) !> 0,必然是所有文件均未加载成功,loaded_files必然为[],返回没有实际意义
# 而若只返回None,可以跟上文的异常返回值保持一致,更便于下游任务判断
return None
return None, loaded_files
def one_knowledge_add(self, vs_path, one_title, one_conent, one_content_segmentation, sentence_size):
try:
......
......@@ -23,7 +23,7 @@ def main():
top_k=VECTOR_SEARCH_TOP_K)
vs_path = None
while not vs_path:
print("注意输入的路径是完整的文件路径,例如content/`knowledge_base_id`/file.md,多个路径用英文逗号分割")
print("注意输入的路径是完整的文件路径,例如knowledge_base/`knowledge_base_id`/content/file.md,多个路径用英文逗号分割")
filepath = input("Input your local knowledge file path 请输入本地知识文件路径:")
# 判断 filepath 是否为空,如果为空的话,重新让用户输入,防止用户误触回车
......@@ -35,9 +35,9 @@ def main():
# filepath错误的返回为None, 如果直接用原先的vs_path,_ = local_doc_qa.init_knowledge_vector_store(filepath)
# 会直接导致TypeError: cannot unpack non-iterable NoneType object而使得程序直接退出
# 因此需要先加一层判断,保证程序能继续运行
temp = local_doc_qa.init_knowledge_vector_store(filepath)
temp,loaded_files = local_doc_qa.init_knowledge_vector_store(filepath)
if temp is not None:
vs_path,loaded_files = temp
vs_path = temp
# 如果loaded_files和len(filepath)不一致,则说明部分文件没有加载成功
# 如果是路径错误,则应该支持重新加载
if len(loaded_files) != len(filepath):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论