diff --git a/articles_directory_predict.py b/articles_directory_predict.py index a794d69..0113b9b 100644 --- a/articles_directory_predict.py +++ b/articles_directory_predict.py @@ -22,7 +22,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 = "<|role|>user<|says|>{}<|end|>\n<|role|>assistant<|says|>" +prompt = "<|im_start|>user\n{}<|im_end|>\n<|im_start|>assistant\n" app = Flask(__name__) app.config["JSON_AS_ASCII"] = False @@ -70,7 +71,7 @@ def request_api_chatgpt(prompt): "content": prompt, "model": "gpt-4-turbo", "top_p": 0.95, - "temperature": 0.9 + "temperature": 0.7 } response = requests.post( chatgpt_url_predict, diff --git a/articles_directory_predict_word.py b/articles_directory_predict_word.py index 77ac9f4..1b10255 100644 --- a/articles_directory_predict_word.py +++ b/articles_directory_predict_word.py @@ -44,11 +44,11 @@ chatgpt_url_predict = "http://{}:12000/predict".format(str(get_host_ip())) chatgpt_url_search = "http://{}:12000/search".format(str(get_host_ip())) prompt = { - "mulu_title_Level_2": "<|role|>user<|says|>为论文题目“%s”生成中文目录,要求只有一级标题和二级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;一级标题生成%s个;每个一级标题包含%s个二级标题<|end|>\n<|role|>assistant<|says|>", - "mulu_title_Level_2_1": "<|role|>user<|says|>为论文题目“%s”生成中文目录,要求只有一级标题和二级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;一级标题生成%s个;每个一级标题包含%s-%s个二级标题<|end|>\n<|role|>assistant<|says|>", - "mulu_title_Level_3": "<|role|>user<|says|>为论文题目“%s”生成中文目录,要求只有一级标题,二级标题和三级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;三级标题使用阿拉伯数字 例如1.1.2 xxx;一级标题生成%s个;每个一级标题包含%s个二级标题;三级标题个数适中<|end|>\n<|role|>assistant<|says|>", - "mulu_title_Level_3_1": "<|role|>user<|says|>为论文题目“%s”生成中文目录,要求只有一级标题,二级标题和三级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;三级标题使用阿拉伯数字 例如1.1.2 xxx;一级标题生成%s个;每个一级标题包含%s-%s个二级标题;三级标题个数适中<|end|>\n<|role|>assistant<|says|>", - "mulu_title_Level_3_2": "<|role|>user<|says|>为论文题目“%s”生成中文目录,要求只有一级标题,二级标题和三级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;三级标题使用阿拉伯数字 例如1.1.2 xxx;一级标题生成%s个;每个一级标题包含%s-%s个二级标题;三级标题个数越多越好<|end|>\n<|role|>assistant<|says|>" + "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" } nums_word_dict = { @@ -147,7 +147,7 @@ def request_api_chatgpt(prompt): "content": prompt, "model": "gpt-4-turbo", "top_p": 0.95, - "temperature": 0.9 + "temperature": 0.7 } response = requests.post( chatgpt_url_predict, diff --git a/articles_directory_predict_word_online.py b/articles_directory_predict_word_online.py index de68306..e6fc539 100644 --- a/articles_directory_predict_word_online.py +++ b/articles_directory_predict_word_online.py @@ -44,11 +44,11 @@ chatgpt_url_predict = "http://{}:12001/predict".format(str(get_host_ip())) chatgpt_url_search = "http://{}:12001/search".format(str(get_host_ip())) prompt = { - "mulu_title_Level_2": "<|role|>user<|says|>为论文题目“%s”生成中文目录,要求只有一级标题和二级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;一级标题生成%s个;每个一级标题包含%s个二级标题<|end|>\n<|role|>assistant<|says|>", - "mulu_title_Level_2_1": "<|role|>user<|says|>为论文题目“%s”生成中文目录,要求只有一级标题和二级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;一级标题生成%s个;每个一级标题包含%s-%s个二级标题<|end|>\n<|role|>assistant<|says|>", - "mulu_title_Level_3": "<|role|>user<|says|>为论文题目“%s”生成中文目录,要求只有一级标题,二级标题和三级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;三级标题使用阿拉伯数字 例如1.1.2 xxx;一级标题生成%s个;每个一级标题包含%s个二级标题;三级标题个数适中<|end|>\n<|role|>assistant<|says|>", - "mulu_title_Level_3_1": "<|role|>user<|says|>为论文题目“%s”生成中文目录,要求只有一级标题,二级标题和三级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;三级标题使用阿拉伯数字 例如1.1.2 xxx;一级标题生成%s个;每个一级标题包含%s-%s个二级标题;三级标题个数适中<|end|>\n<|role|>assistant<|says|>", - "mulu_title_Level_3_2": "<|role|>user<|says|>为论文题目“%s”生成中文目录,要求只有一级标题,二级标题和三级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;三级标题使用阿拉伯数字 例如1.1.2 xxx;一级标题生成%s个;每个一级标题包含%s-%s个二级标题;三级标题个数越多越好<|end|>\n<|role|>assistant<|says|>" + "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" } nums_word_dict = { @@ -146,7 +146,7 @@ def request_api_chatgpt(prompt): "content": prompt, "model": "gpt-4-turbo", "top_p": 0.95, - "temperature": 0.9 + "temperature": 0.7 } response = requests.post( chatgpt_url_predict, diff --git a/articles_directory_predict_word_table_formula_kongzhi.py b/articles_directory_predict_word_table_formula_kongzhi.py index be5c3b8..8fa487d 100644 --- a/articles_directory_predict_word_table_formula_kongzhi.py +++ b/articles_directory_predict_word_table_formula_kongzhi.py @@ -46,11 +46,11 @@ chatgpt_url_predict = "http://{}:12000/predict".format(str(get_host_ip())) chatgpt_url_search = "http://{}:12000/search".format(str(get_host_ip())) prompt = { - "mulu_title_Level_2": "<|role|>user<|says|>为论文题目“%s”生成中文目录,要求只有一级标题和二级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;一级标题生成%s个;每个一级标题包含%s个二级标题<|end|>\n<|role|>assistant<|says|>", - "mulu_title_Level_2_1": "<|role|>user<|says|>为论文题目“%s”生成中文目录,要求只有一级标题和二级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;一级标题生成%s个;每个一级标题包含%s-%s个二级标题<|end|>\n<|role|>assistant<|says|>", - "mulu_title_Level_3": "<|role|>user<|says|>为论文题目“%s”生成中文目录,要求只有一级标题,二级标题和三级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;三级标题使用阿拉伯数字 例如1.1.2 xxx;一级标题生成%s个;每个一级标题包含%s个二级标题;三级标题个数适中<|end|>\n<|role|>assistant<|says|>", - "mulu_title_Level_3_1": "<|role|>user<|says|>为论文题目“%s”生成中文目录,要求只有一级标题,二级标题和三级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;三级标题使用阿拉伯数字 例如1.1.2 xxx;一级标题生成%s个;每个一级标题包含%s-%s个二级标题;三级标题个数适中<|end|>\n<|role|>assistant<|says|>", - "mulu_title_Level_3_2": "<|role|>user<|says|>为论文题目“%s”生成中文目录,要求只有一级标题,二级标题和三级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;三级标题使用阿拉伯数字 例如1.1.2 xxx;一级标题生成%s个;每个一级标题包含%s-%s个二级标题;三级标题个数越多越好<|end|>\n<|role|>assistant<|says|>" + "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" } prompt_biaogekongzhi = { @@ -178,7 +178,7 @@ def request_api_chatgpt(prompt): "content": prompt, "model": "gpt-4-turbo", "top_p": 0.95, - "temperature": 0.9 + "temperature": 0.7 } response = requests.post( chatgpt_url_predict,