diff --git a/chains/local_doc_qa.py b/chains/local_doc_qa.py index a454931..d4cdcb0 100644 --- a/chains/local_doc_qa.py +++ b/chains/local_doc_qa.py @@ -39,7 +39,11 @@ def load_file(filepath, sentence_size=SENTENCE_SIZE): def write_check_file(filepath, docs): - fout = open('load_file.txt', 'a') + folder_path = os.path.join(os.path.dirname(filepath), "tmp_files") + if not os.path.exists(folder_path): + os.makedirs(folder_path) + fp = os.path.join(folder_path, 'load_file.txt') + fout = open(fp, 'a') fout.write("filepath=%s,len=%s" % (filepath, len(docs))) fout.write('\n') for i in docs: diff --git a/cli_demo.py b/cli_demo.py index eb0f7e2..ea5a895 100644 --- a/cli_demo.py +++ b/cli_demo.py @@ -31,13 +31,13 @@ if __name__ == "__main__": chat_history=history, streaming=STREAMING): if STREAMING: - logger.info(resp["result"][last_print_len:]) + print(resp["result"][last_print_len:], end="", flush=True) last_print_len = len(resp["result"]) else: - logger.info(resp["result"]) + print(resp["result"]) if REPLY_WITH_SOURCE: source_text = [f"""出处 [{inum + 1}] {os.path.split(doc.metadata['source'])[-1]}:\n\n{doc.page_content}\n\n""" # f"""相关度:{doc.metadata['score']}\n\n""" for inum, doc in enumerate(resp["source_documents"])] - logger.info("\n\n" + "\n\n".join(source_text)) + print("\n\n" + "\n\n".join(source_text))