|
|
@ -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, |
|
|
|