From 018ad7b9b87b654bbe2e10309c3c4d79b52b5106 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E6=98=9F=E6=BC=A9?= Date: Thu, 30 May 2024 12:04:59 +0800 Subject: [PATCH] udpate code for domain selection --- run.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/run.py b/run.py index f1bc31d..d00c0c5 100644 --- a/run.py +++ b/run.py @@ -19,6 +19,15 @@ def s1_reasoning_preparation(dataset, data_point, model, threshold): print("****************** Start stage 1: reasoning preparation ...") question = dataset.get_question(data_point) print("****** Question:", question) + + ### Domain selection + domain_selection_prompt = domain_selection_demonstration + "Q: " + question.strip() + "\nRelevant domains: " + domain_selection_response = call_openai_api(model, domain_selection_prompt, max_tokens=256, temperature=0) + + if domain_selection_response is not None: + domain_selection_text_response = domain_selection_response[1].strip() + print("****** Relevant domains:", domain_selection_text_response) + data_point["s1_domains"] = [x.strip() for x in domain_selection_text_response.split(",")] ### CoT generation cot_prompt = dataset.get_s1_prompt(question)