按照review建议进行修改
This commit is contained in:
parent
ba336440aa
commit
43fae98bb5
|
|
@ -197,9 +197,8 @@ class LocalDocQA:
|
||||||
return vs_path, loaded_files
|
return vs_path, loaded_files
|
||||||
else:
|
else:
|
||||||
logger.info("文件均未成功加载,请检查依赖包或替换为其他文件再次上传。")
|
logger.info("文件均未成功加载,请检查依赖包或替换为其他文件再次上传。")
|
||||||
# 若len(docs) !> 0,必然是所有文件均未加载成功,loaded_files必然为[],返回没有实际意义
|
|
||||||
# 而若只返回None,可以跟上文的异常返回值保持一致,更便于下游任务判断
|
return None, loaded_files
|
||||||
return None
|
|
||||||
|
|
||||||
def one_knowledge_add(self, vs_path, one_title, one_conent, one_content_segmentation, sentence_size):
|
def one_knowledge_add(self, vs_path, one_title, one_conent, one_content_segmentation, sentence_size):
|
||||||
try:
|
try:
|
||||||
|
|
|
||||||
|
|
@ -23,7 +23,7 @@ def main():
|
||||||
top_k=VECTOR_SEARCH_TOP_K)
|
top_k=VECTOR_SEARCH_TOP_K)
|
||||||
vs_path = None
|
vs_path = None
|
||||||
while not vs_path:
|
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 = input("Input your local knowledge file path 请输入本地知识文件路径:")
|
||||||
|
|
||||||
# 判断 filepath 是否为空,如果为空的话,重新让用户输入,防止用户误触回车
|
# 判断 filepath 是否为空,如果为空的话,重新让用户输入,防止用户误触回车
|
||||||
|
|
@ -35,9 +35,9 @@ def main():
|
||||||
# filepath错误的返回为None, 如果直接用原先的vs_path,_ = local_doc_qa.init_knowledge_vector_store(filepath)
|
# filepath错误的返回为None, 如果直接用原先的vs_path,_ = local_doc_qa.init_knowledge_vector_store(filepath)
|
||||||
# 会直接导致TypeError: cannot unpack non-iterable NoneType object而使得程序直接退出
|
# 会直接导致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:
|
if temp is not None:
|
||||||
vs_path,loaded_files = temp
|
vs_path = temp
|
||||||
# 如果loaded_files和len(filepath)不一致,则说明部分文件没有加载成功
|
# 如果loaded_files和len(filepath)不一致,则说明部分文件没有加载成功
|
||||||
# 如果是路径错误,则应该支持重新加载
|
# 如果是路径错误,则应该支持重新加载
|
||||||
if len(loaded_files) != len(filepath):
|
if len(loaded_files) != len(filepath):
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue