diff --git a/chains/local_doc_qa.py b/chains/local_doc_qa.py index 7755ef1..cf4281e 100644 --- a/chains/local_doc_qa.py +++ b/chains/local_doc_qa.py @@ -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: diff --git a/cli_demo.py b/cli_demo.py index c9ca558..a445e14 100644 --- a/cli_demo.py +++ b/cli_demo.py @@ -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):