import openai
openai.api_key = "YOUR_API_KEY"
# 调用OpenAI API
def generate_text(prompt, model, temperature):
response = openai.Completion.create(
engine=model,
prompt=prompt,
max_tokens=60,
temperature=temperature
)
text = response.choices[0].text
return text.strip()
# 测试API调用
prompt = "Hello, I'm ChatGPT!"
model = "text-davinci-002"
temperature = 0.7
generated_text = generate_text(prompt, model, temperature)
print(generated_text)
在上述代码中,你需要替换 YOUR_API_KEY
为你的API密钥。generate_text
函数使用 openai
库来调用 GPT-3 模型,生成一段文本,返回一个字符串。在本例中,使用 “text-davinci-002” 作为 GPT-3 模型,最大令牌为60,temperature为0.7。
当调用 generate_text
函数时,会向 OpenAI API 发送一个请求,以生成一段文本,然后将其作为字符串返回。这个例子仅仅是给出了一个最基本的 API 调用示例,你可以根据自己的需求和用例,使用更复杂的API请求。