用PYTHON代码写一个调用GPT的程序
时间: 2023-05-27 13:05:43 浏览: 138
利用PYTHON编写的小程序
以下是一个用Python调用GPT的示例代码:
```python
import openai
import os
# 设置OpenAI API密钥
openai.api_key = os.environ["OPENAI_API_KEY"]
# 生成文本的函数
def generate_text(prompt):
# GPT-3模型ID
model_engine = "text-davinci-002"
# 最大生成长度
max_length = 100
# 生成的数量
num_responses = 1
# 生成的温度,控制生成的随机程度
temperature = 0.5
# 调用OpenAI生成API
response = openai.Completion.create(
engine=model_engine,
prompt=prompt,
max_tokens=max_length,
n=num_responses,
temperature=temperature
)
# 返回生成的文本
return response.choices[0].text
# 调用生成文本的函数
generated_text = generate_text("Hello, GPT!")
print(generated_text)
```
在这个示例中,我们首先设置了OpenAI API密钥,然后定义了一个生成文本的函数`generate_text`。这个函数接受一个`prompt`参数,表示要生成的文本的开头,然后使用`openai.Completion.create`方法调用OpenAI的生成API,传递给它模型ID、最大生成长度、生成的数量和温度等参数,然后返回生成的文本。最后,我们调用`generate_text`函数并打印生成的文本。
阅读全文