RoboWaiter/llm_comm/gpt_crawler/gpt3_5_crawler_main.py

29 lines
1.3 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

# -*- coding: utf-8 -*-
import utils
# user_input_path_pre = 'prompts/新闻问题_user_pre.txt'
# user_input_path_suf = 'prompts/新闻问题_user_suf.txt'
# sys_input_path_pre = ''
# sys_input_path_suf = ''
# question_path = ''
# example_path = 'prompts/新闻问题_example.txt'
output_path = 'prompts/声明.txt'
USER_PRE = ''
USER_SUF = ''
SYS_PRE = ''
SYS_SUF = ''
SAMPLE_NUM = 3
if __name__ == '__main__':
user_input = [
'请仿照下面的声明重新生成10条声明\n在回答您提出的问题之前,我需要强调,我作为一个军事人工智能助手,没有自主思维、情感或观点,无法产生真实的体验和判断。我所提供的信息和观点仅基于已有的历史数据和常识,旨在为您提供一种可能的解释,但并不代表任何实际个体或团体的观点或决策。'
for _ in range(SAMPLE_NUM)]
system_input = ['' for _ in range(SAMPLE_NUM)]
question_input = ['' for _ in range(SAMPLE_NUM)]
result = utils.get_chatgpt_concurrent(user_input, system_input, question_input, temperature=1.5, top_p=0.6, frequency_penalty=1.3, presence_penalty=1.3, max_tokens=8192)
with open(output_path, 'w', encoding='utf-8') as f:
for line in result:
f.write(line['answer'] + '\n')