diff --git a/qa_amend_.py b/qa_amend_.py new file mode 100644 index 0000000..33b11c2 --- /dev/null +++ b/qa_amend_.py @@ -0,0 +1,19 @@ +# another_file.py +import time + +start_time = time.time() +from qa_amend import QAService, match_query + +kb_name = 'my_kb_test' +device = None +qa_service = QAService(kb_name, device) + +query = "你有哪里不舒服?" +top_k = 1 +score_threshold = 0.1 + +match_query(qa_service, query, top_k, 1 - score_threshold) + +end_time = time.time() +elapsed_time = end_time - start_time +print(f"总耗时: {elapsed_time} 秒") \ No newline at end of file