使用Python与OpenAI API交互

最近ChatGPT大火,相信OpenAI的大名大家都听过了。

注册OpenAI后可以在服务器不忙的时候使用Chat,繁忙的时候系统就优先照顾付费用户了。对于程序员来说,可以使用API与OpenAI的模拟进行交互,按使用量计费,注册成功赠送5$,够测试了。

这里要说的是使用Python与OpenAI API交互。

步骤:

1.注册OpenAI API账户并获取API密钥。

2.安装OpenAI库

pip3 install openai

3.导入OpenAI库和API密钥

import openai
openai.api_key = "YOUR_API_KEY"

对于国内用户来说,需要使用代理请求,好在OpenAI提供了代理配置

openai.proxy = "http://proxy_ip:port"

4.使用API进行文本生成

prompt = "Today is a beautiful day" 
model = "text-davinci-003" 
response = openai.Completion.create(engine=model, prompt=prompt, max_tokens=30) 
text = response.choices[0].text 
print(text)

其中,“prompt”是生成文本的开头,“model”是选择的生成模型,“max_tokens”是生成的最大长度。生成的文本保存在response.choices[0].text中。

这是GPT3模型的写法,也可以使用新的写法,模型使用gpt-3.5-turbo

model_engine = "gpt-3.5-turbo"
prompt = "怎么看待ChatGPT"

completion = openai.ChatCompletion.create(
  model=model_engine, 
  messages=[{"role": "user", "content": prompt}]
)

print(completion.choices[0].message.content)

返回的数据结构有所不同,打印一下就能看到了。

Leave a Comment

豫ICP备19001387号-1