diff --git a/articles_directory_predict.py b/articles_directory_predict.py
index 0113b9b..fb9274f 100644
--- a/articles_directory_predict.py
+++ b/articles_directory_predict.py
@@ -23,7 +23,8 @@ def get_host_ip():
chatgpt_url_predict = "http://{}:12000/predict".format(str(get_host_ip()))
chatgpt_url_search = "http://{}:12000/search".format(str(get_host_ip()))
# prompt = "<|role|>user<|says|>{}<|end|>\n<|role|>assistant<|says|>"
-prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n"
+qwen_prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n"
+qwen_think_prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n\n\n\n\n"
app = Flask(__name__)
app.config["JSON_AS_ASCII"] = False
@@ -165,7 +166,7 @@ def get_multiple_urls(urls):
def articles_directory():
text = request.json["texts"] # 获取用户query中的文本 例如"I love you"
nums = request.json["nums"]
- text = prompt.format(text)
+ text = qwen_think_prompt.format(text)
nums = int(nums)
input_data = []
diff --git a/articles_directory_predict_qikan_word.py b/articles_directory_predict_qikan_word.py
index 61416b4..a784e6d 100644
--- a/articles_directory_predict_qikan_word.py
+++ b/articles_directory_predict_qikan_word.py
@@ -7,6 +7,7 @@ import socket
import os
+
class log:
def __init__(self):
pass
@@ -42,34 +43,37 @@ def get_host_ip():
chatgpt_url_predict = "http://{}:12000/predict".format(str(get_host_ip()))
chatgpt_url_search = "http://{}:12000/search".format(str(get_host_ip()))
+qwen_prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n"
+qwen_think_prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n\n\n\n\n"
+
prompt = {
- "mulu_title_Level_2_1": "<|im_start|>user\n任务：生成目录\n请帮我生成一个期刊论文的目录，论文题目是“%s”，要求只有一级标题，二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题<|im_end|>\n<|im_start|>assistant\n",
- "mulu_title_Level_3_1": "<|im_start|>user\n任务：生成目录\n请帮我生成一个期刊论文的目录，论文题目是“%s”，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数适中<|im_end|>\n<|im_start|>assistant\n",
+ "mulu_title_Level_2_1": "任务：生成目录\n请帮我生成一个期刊论文的目录，论文题目是“%s”，要求只有一级标题，二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题",
+ "mulu_title_Level_3_1": "任务：生成目录\n请帮我生成一个期刊论文的目录，论文题目是“%s”，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数适中",
}
nums_word_dict = {
"3000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "3",
"title_2_nums": ["0","3"],
"title_small_nums": ["3", "0","3"]
},
"5000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "4",
"title_2_nums": ["0","5"],
"title_small_nums": ["4", "0","5"]
},
"8000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "5",
"title_2_nums": ["0", "6"],
"title_small_nums": ["5", "0", "6"]
},
"10000": {
- "prompt": prompt["mulu_title_Level_3_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_3_1"]),
"title_1_nums": "6",
"title_2_nums": ["0", "5"],
"title_small_nums": ["6", "0", "5"],
diff --git a/articles_directory_predict_qikan_word_online.py b/articles_directory_predict_qikan_word_online.py
index 2a7ab0d..8460a0b 100644
--- a/articles_directory_predict_qikan_word_online.py
+++ b/articles_directory_predict_qikan_word_online.py
@@ -6,7 +6,6 @@ import requests
import socket
import os
-
class log:
def __init__(self):
pass
@@ -42,34 +41,36 @@ def get_host_ip():
chatgpt_url_predict = "http://{}:12001/predict".format(str(get_host_ip()))
chatgpt_url_search = "http://{}:12001/search".format(str(get_host_ip()))
+qwen_prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n"
+qwen_think_prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n\n\n\n\n"
prompt = {
- "mulu_title_Level_2_1": "<|im_start|>user\n任务：生成目录\n请帮我生成一个期刊论文的目录，论文题目是“%s”，要求只有一级标题，二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题<|im_end|>\n<|im_start|>assistant\n",
- "mulu_title_Level_3_1": "<|im_start|>user\n任务：生成目录\n请帮我生成一个期刊论文的目录，论文题目是“%s”，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数适中<|im_end|>\n<|im_start|>assistant\n",
+ "mulu_title_Level_2_1": "任务：生成目录\n请帮我生成一个期刊论文的目录，论文题目是“%s”，要求只有一级标题，二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题",
+ "mulu_title_Level_3_1": "任务：生成目录\n请帮我生成一个期刊论文的目录，论文题目是“%s”，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数适中",
}
nums_word_dict = {
"3000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "3",
"title_2_nums": ["0","3"],
"title_small_nums": ["3", "0","3"]
},
"5000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "4",
"title_2_nums": ["0","5"],
"title_small_nums": ["4", "0","5"]
},
"8000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "5",
"title_2_nums": ["0", "6"],
"title_small_nums": ["5", "0", "6"]
},
"10000": {
- "prompt": prompt["mulu_title_Level_3_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_3_1"]),
"title_1_nums": "6",
"title_2_nums": ["0", "5"],
"title_small_nums": ["6", "0", "5"],
diff --git a/articles_directory_predict_qikan_word_table_formula_kongzhi.py b/articles_directory_predict_qikan_word_table_formula_kongzhi.py
index 36529f8..a936c72 100644
--- a/articles_directory_predict_qikan_word_table_formula_kongzhi.py
+++ b/articles_directory_predict_qikan_word_table_formula_kongzhi.py
@@ -44,34 +44,37 @@ def get_host_ip():
chatgpt_url_predict = "http://{}:12000/predict".format(str(get_host_ip()))
chatgpt_url_search = "http://{}:12000/search".format(str(get_host_ip()))
+qwen_prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n"
+qwen_think_prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n\n\n\n\n"
+
prompt = {
- "mulu_title_Level_2_1": "<|im_start|>user\n任务：生成目录\n请帮我生成一个期刊论文的目录，论文题目是“%s”，要求只有一级标题，二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题<|im_end|>\n<|im_start|>assistant\n",
- "mulu_title_Level_3_1": "<|im_start|>user\n任务：生成目录\n请帮我生成一个期刊论文的目录，论文题目是“%s”，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数适中<|im_end|>\n<|im_start|>assistant\n",
+ "mulu_title_Level_2_1": "任务：生成目录\n请帮我生成一个期刊论文的目录，论文题目是“%s”，要求只有一级标题，二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题",
+ "mulu_title_Level_3_1": "任务：生成目录\n请帮我生成一个期刊论文的目录，论文题目是“%s”，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数适中",
}
nums_word_dict = {
"3000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "3",
"title_2_nums": ["0","3"],
"title_small_nums": ["3", "0","3"]
},
"5000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "4",
"title_2_nums": ["0","5"],
"title_small_nums": ["4", "0","5"]
},
"8000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "5",
"title_2_nums": ["0", "6"],
"title_small_nums": ["5", "0", "6"]
},
"10000": {
- "prompt": prompt["mulu_title_Level_3_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_3_1"]),
"title_1_nums": "6",
"title_2_nums": ["0", "5"],
"title_small_nums": ["6", "0", "5"],
@@ -424,7 +427,8 @@ def biaogegongshi_kongzhi_prompt(title, mulu_list, table_and_formulas_list):
"，".join(prompt_2_kongzhi),
prompt_biaogekongzhi["prompt3_multiple_tags"].format("&".join(lable_en), "&".join(lable_en)),
])
- prompt = "<|im_start|>user\n任务：公式和表格生成控制\n{}<|im_end|>\n<|im_start|>assistant\n".format(prompt)
+ prompt = "任务：公式和表格生成控制\n{}".format(prompt)
+ prompt = qwen_think_prompt.format(prompt)
prompt_list.append(prompt)
return prompt_list
diff --git a/articles_directory_predict_word.py b/articles_directory_predict_word.py
index 1b10255..9aa631d 100644
--- a/articles_directory_predict_word.py
+++ b/articles_directory_predict_word.py
@@ -42,60 +42,62 @@ def get_host_ip():
chatgpt_url_predict = "http://{}:12000/predict".format(str(get_host_ip()))
chatgpt_url_search = "http://{}:12000/search".format(str(get_host_ip()))
+qwen_prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n"
+qwen_think_prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n\n\n\n\n"
prompt = {
- "mulu_title_Level_2": "<|im_start|>user\n为论文题目“%s”生成中文目录，要求只有一级标题和二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s个二级标题<|im_end|>\n<|im_start|>assistant\n",
- "mulu_title_Level_2_1": "<|im_start|>user\n为论文题目“%s”生成中文目录，要求只有一级标题和二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题<|im_end|>\n<|im_start|>assistant\n",
- "mulu_title_Level_3": "<|im_start|>user\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s个二级标题；三级标题个数适中<|im_end|>\n<|im_start|>assistant\n",
- "mulu_title_Level_3_1": "<|im_start|>user\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数适中<|im_end|>\n<|im_start|>assistant\n",
- "mulu_title_Level_3_2": "<|im_start|>user\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数越多越好<|im_end|>\n<|im_start|>assistant\n"
+ "mulu_title_Level_2": "任务：生成目录\n为论文题目“%s”生成中文目录，要求只有一级标题和二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s个二级标题",
+ "mulu_title_Level_2_1": "任务：生成目录\n为论文题目“%s”生成中文目录，要求只有一级标题和二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题",
+ "mulu_title_Level_3": "任务：生成目录\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s个二级标题；三级标题个数适中",
+ "mulu_title_Level_3_1": "任务：生成目录\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数适中",
+ "mulu_title_Level_3_2": "任务：生成目录\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数越多越好"
}
nums_word_dict = {
"3000": {
- "prompt": prompt["mulu_title_Level_2"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2"]),
"title_1_nums": "3",
"title_2_nums": ["2"],
"title_small_nums": ["3", "2"]
},
"5000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "3",
"title_2_nums": ["2","4"],
"title_small_nums": ["3", "2","4"]
},
"8000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "4",
"title_2_nums": ["2", "4"],
"title_small_nums": ["4", "2", "4"],
},
"10000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "6",
"title_2_nums": ["2", "5"],
"title_small_nums": ["6", "2", "5"],
},
"15000": {
- "prompt": prompt["mulu_title_Level_3_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_3_1"]),
"title_1_nums": "5",
"title_2_nums": ["2", "3"],
"title_small_nums": ["5", "2", "3"],
},
"20000": {
- "prompt": prompt["mulu_title_Level_3_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_3_1"]),
"title_1_nums": "6",
"title_2_nums": ["2", "4"],
"title_small_nums": ["6", "2", "4"],
},
"30000": {
- "prompt": prompt["mulu_title_Level_3_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_3_1"]),
"title_1_nums": "7",
"title_2_nums": ["3", "5"],
"title_small_nums": ["7", "3", "5"],
},
"50000": {
- "prompt": prompt["mulu_title_Level_3_2"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_3_2"]),
"title_1_nums": "8",
"title_2_nums": ["3", "8"],
"title_small_nums": ["8", "3", "8"],
diff --git a/articles_directory_predict_word_online.py b/articles_directory_predict_word_online.py
index e6fc539..a0227a3 100644
--- a/articles_directory_predict_word_online.py
+++ b/articles_directory_predict_word_online.py
@@ -42,60 +42,62 @@ def get_host_ip():
chatgpt_url_predict = "http://{}:12001/predict".format(str(get_host_ip()))
chatgpt_url_search = "http://{}:12001/search".format(str(get_host_ip()))
+qwen_prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n"
+qwen_think_prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n\n\n\n\n"
prompt = {
- "mulu_title_Level_2": "<|im_start|>user\n为论文题目“%s”生成中文目录，要求只有一级标题和二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s个二级标题<|im_end|>\n<|im_start|>assistant\n",
- "mulu_title_Level_2_1": "<|im_start|>user\n为论文题目“%s”生成中文目录，要求只有一级标题和二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题<|im_end|>\n<|im_start|>assistant\n",
- "mulu_title_Level_3": "<|im_start|>user\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s个二级标题；三级标题个数适中<|im_end|>\n<|im_start|>assistant\n",
- "mulu_title_Level_3_1": "<|im_start|>user\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数适中<|im_end|>\n<|im_start|>assistant\n",
- "mulu_title_Level_3_2": "<|im_start|>user\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数越多越好<|im_end|>\n<|im_start|>assistant\n"
+ "mulu_title_Level_2": "任务：生成目录\n为论文题目“%s”生成中文目录，要求只有一级标题和二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s个二级标题",
+ "mulu_title_Level_2_1": "任务：生成目录\n为论文题目“%s”生成中文目录，要求只有一级标题和二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题",
+ "mulu_title_Level_3": "任务：生成目录\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s个二级标题；三级标题个数适中",
+ "mulu_title_Level_3_1": "任务：生成目录\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数适中",
+ "mulu_title_Level_3_2": "任务：生成目录\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数越多越好"
}
nums_word_dict = {
"3000": {
- "prompt": prompt["mulu_title_Level_2"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2"]),
"title_1_nums": "3",
"title_2_nums": ["2"],
"title_small_nums": ["3", "2"]
},
"5000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "3",
"title_2_nums": ["2","4"],
"title_small_nums": ["3", "2","4"]
},
"8000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "4",
"title_2_nums": ["2", "4"],
"title_small_nums": ["4", "2", "4"],
},
"10000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "6",
"title_2_nums": ["2", "5"],
"title_small_nums": ["6", "2", "5"],
},
"15000": {
- "prompt": prompt["mulu_title_Level_3_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_3_1"]),
"title_1_nums": "5",
"title_2_nums": ["2", "3"],
"title_small_nums": ["5", "2", "3"],
},
"20000": {
- "prompt": prompt["mulu_title_Level_3_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_3_1"]),
"title_1_nums": "6",
"title_2_nums": ["2", "4"],
"title_small_nums": ["6", "2", "4"],
},
"30000": {
- "prompt": prompt["mulu_title_Level_3_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_3_1"]),
"title_1_nums": "7",
"title_2_nums": ["3", "5"],
"title_small_nums": ["7", "3", "5"],
},
"50000": {
- "prompt": prompt["mulu_title_Level_3_2"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_3_2"]),
"title_1_nums": "8",
"title_2_nums": ["3", "8"],
"title_small_nums": ["8", "3", "8"],
diff --git a/articles_directory_predict_word_table_formula_kongzhi.py b/articles_directory_predict_word_table_formula_kongzhi.py
index 9843e3d..2d602f2 100644
--- a/articles_directory_predict_word_table_formula_kongzhi.py
+++ b/articles_directory_predict_word_table_formula_kongzhi.py
@@ -42,15 +42,20 @@ def get_host_ip():
return ip
-chatgpt_url_predict = "http://{}:12000/predict".format(str(get_host_ip()))
-chatgpt_url_search = "http://{}:12000/search".format(str(get_host_ip()))
+# chatgpt_url_predict = "http://{}:12000/predict".format(str(get_host_ip()))
+# chatgpt_url_search = "http://{}:12000/search".format(str(get_host_ip()))
+chatgpt_url_predict = "http://{}:12000/predict".format("192.168.31.149")
+chatgpt_url_search = "http://{}:12000/search".format("192.168.31.149")
+
+qwen_prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n"
+qwen_think_prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n\n\n\n\n"
prompt = {
- "mulu_title_Level_2": "<|im_start|>user\n为论文题目“%s”生成中文目录，要求只有一级标题和二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s个二级标题<|im_end|>\n<|im_start|>assistant\n",
- "mulu_title_Level_2_1": "<|im_start|>user\n为论文题目“%s”生成中文目录，要求只有一级标题和二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题<|im_end|>\n<|im_start|>assistant\n",
- "mulu_title_Level_3": "<|im_start|>user\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s个二级标题；三级标题个数适中<|im_end|>\n<|im_start|>assistant\n",
- "mulu_title_Level_3_1": "<|im_start|>user\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数适中<|im_end|>\n<|im_start|>assistant\n",
- "mulu_title_Level_3_2": "<|im_start|>user\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数越多越好<|im_end|>\n<|im_start|>assistant\n"
+ "mulu_title_Level_2": "任务：生成目录\n为论文题目“%s”生成中文目录，要求只有一级标题和二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s个二级标题",
+ "mulu_title_Level_2_1": "任务：生成目录\n为论文题目“%s”生成中文目录，要求只有一级标题和二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题",
+ "mulu_title_Level_3": "任务：生成目录\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s个二级标题；三级标题个数适中",
+ "mulu_title_Level_3_1": "任务：生成目录\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数适中",
+ "mulu_title_Level_3_2": "任务：生成目录\n为论文题目“%s”生成中文目录，要求只有一级标题，二级标题和三级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；三级标题使用阿拉伯数字 例如1.1.2 xxx；一级标题生成%s个；每个一级标题包含%s-%s个二级标题；三级标题个数越多越好"
}
prompt_biaogekongzhi = {
@@ -84,49 +89,49 @@ id_to_lable = {
nums_word_dict = {
"3000": {
- "prompt": prompt["mulu_title_Level_2"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2"]),
"title_1_nums": "3",
"title_2_nums": ["2"],
"title_small_nums": ["3", "2"]
},
"5000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "3",
"title_2_nums": ["2","4"],
"title_small_nums": ["3", "2","4"]
},
"8000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "4",
"title_2_nums": ["2", "4"],
"title_small_nums": ["4", "2", "4"],
},
"10000": {
- "prompt": prompt["mulu_title_Level_2_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_2_1"]),
"title_1_nums": "6",
"title_2_nums": ["2", "5"],
"title_small_nums": ["6", "2", "5"],
},
"15000": {
- "prompt": prompt["mulu_title_Level_3_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_3_1"]),
"title_1_nums": "5",
"title_2_nums": ["2", "3"],
"title_small_nums": ["5", "2", "3"],
},
"20000": {
- "prompt": prompt["mulu_title_Level_3_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_3_1"]),
"title_1_nums": "6",
"title_2_nums": ["2", "4"],
"title_small_nums": ["6", "2", "4"],
},
"30000": {
- "prompt": prompt["mulu_title_Level_3_1"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_3_1"]),
"title_1_nums": "7",
"title_2_nums": ["3", "5"],
"title_small_nums": ["7", "3", "5"],
},
"50000": {
- "prompt": prompt["mulu_title_Level_3_2"],
+ "prompt": qwen_think_prompt.format(prompt["mulu_title_Level_3_2"]),
"title_1_nums": "8",
"title_2_nums": ["3", "8"],
"title_small_nums": ["8", "3", "8"],
@@ -450,7 +455,8 @@ def biaogegongshi_kongzhi_prompt(title, mulu_list, table_and_formulas_list):
"，".join(prompt_2_kongzhi),
prompt_biaogekongzhi["prompt3_multiple_tags"].format("&".join(lable_en), "&".join(lable_en)),
])
- prompt = "<|im_start|>user\n任务：公式和表格生成控制\n{}<|im_end|>\n<|im_start|>assistant\n".format(prompt)
+ prompt = "任务：公式和表格生成控制\n{}".format(prompt)
+ prompt = qwen_think_prompt.format(prompt)
prompt_list.append(prompt)
return prompt_list