|
@@ -31,16 +31,16 @@ class DataGenerator:
|
|
|
|
|
|
# 当 human 和 bot 都取到后,拼装一个新的 JSON
|
|
|
target_data = {
|
|
|
- "question": case_data[1]+" 请予以故障诊断?",
|
|
|
+ "question": case_data[1],
|
|
|
"answer": "".join([case_data[2],case_data[3],case_data[4]]),
|
|
|
"prompt": [
|
|
|
{
|
|
|
- "content": f"\nRespond in the following format:\n<reasoning>\n {case_data[2]} \n</reasoning>\n<answer>\n {case_data[3]} {case_data[4]} \n</answer>\n",
|
|
|
- # "content": "\nRespond in the following format:\n<reasoning>\n...\n</reasoning>\n<answer>\n...\n</answer>\n",
|
|
|
+ # "content": f"\nRespond in the following format:\n<reasoning>\n {case_data[2]} \n</reasoning>\n<answer>\n {case_data[3]} {case_data[4]} \n</answer>\n",
|
|
|
+ "content": "\nRespond in the following format:\n<reasoning>\n...\n</reasoning>\n<answer>\n...\n</answer>\n",
|
|
|
"role": "system"
|
|
|
},
|
|
|
{
|
|
|
- "content": case_data[1]+" 请予以故障诊断?",
|
|
|
+ "content": case_data[1],
|
|
|
"role": "user"
|
|
|
}
|
|
|
]
|