按照review建议进行修改
This commit is contained in:
parent
ba336440aa
commit
43fae98bb5
|
|
@ -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):
|
||||
|
|
|
|||
Loading…
Reference in New Issue