python 批量调用openai每个prompt生成一个txt文件每次执行5个请求执行完后在执行下一个

您可以使用以下代码批量调用openai来生成txt文件，并在每次执行5个请求后暂停，等待用户按下任意键继续执行下一个批处理。请注意，您需要在代码中填写您自己的OpenAI API密钥和prompt列表。

import openai
import time

# 在此处添加您的OpenAI API密钥
openai.api_key = "YOUR_API_KEY"

# 在此处添加您想要生成的prompt列表
prompt_list = [
    "Prompt 1",
    "Prompt 2",
    "Prompt 3",
    "Prompt 4",
    "Prompt 5",
    "Prompt 6",
    "Prompt 7",
    "Prompt 8"
]

# 每次处理的请求数量
batch_size = 5

# 处理请求
for i in range(0, len(prompt_list), batch_size):
    # 获取批处理的prompt
    prompt_batch = prompt_list[i:i+batch_size]
    # 处理每个prompt
    for prompt in prompt_batch:
        # 调用OpenAI生成文本
        response = openai.Completion.create(
            engine="davinci",
            prompt=prompt,
            max_tokens=1024,
            n=1,
            stop=None,
            temperature=0.5
        )
        # 将结果写入txt文件
        with open(prompt.replace(" ", "_") + ".txt", "w") as f:
            f.write(response.choices[0].text)
        # 打印处理完成的提示
        print("Processed prompt: " + prompt)
    # 暂停等待用户按下任意键继续
    if i + batch_size < len(prompt_list):
        input("Press any key to continue...")