您现在的位置是:首页 >技术杂谈 >如何在 Python 开发环境中调用 ChatGPT 模型?网站首页技术杂谈
如何在 Python 开发环境中调用 ChatGPT 模型?
本文将演示在本地的 python 项目中调用 ChatGPT 模型。
写在前面
作为一名程序员,在开发过程当中时常需要使用 ChatGPT 来完成一些任务,但总是使用网页交互模式去 Web 端访问 ChatGPT 是很麻烦的,这时候我们可以使用代码来调用 ChatGPT 模型,以实现在本地和 Web 端一样的效果。
本文将演示 Python 开发语言对 ChatGPT 模型的调用,总体分为三步:
- 步骤一:获取 API Key,访问 OpenAI 官网下的 API 网站获取自己的密钥;
- 步骤二:安装 OpenAI 第三方库;
- 步骤三:使用 Python 调用 ChatGPT 模型 API。
第一步:获取 API Key
每一个账号在注册成功之后都会有自己专属的 API key。首先使用我们注册的 ChatGPT 账号密码登录 OpenAI 官网(https://platform.openai.com/overview);
登录后进入以下界面,选择“View API keys”;
如果我们是第一次来到该页面,需要先自己创建 key,选择“Create new secret key”;
创建成功后如下,属于自己的 API key 已生成,复制使用即可。
第二步:安装 OpenAI 第三方库
这里 python 安装第三方库需要使用到 pypi,在浏览器输入“pypi”,就可以看到 Python Package Index,点击进入;
pypi:Python Package Index 首字母简写,表示的是 Python 的 Packag 索引,这是 Python 的官方索引。
随后搜索 openai,可以找到 openai 当前的库;
我们选择 openai 0.27.4 版本,进入。这里面告知了我们 openai 第三方库的安装方式:使用命令 pip install openai
。
接下来使用本地 anaconda 窗口输入命令下载即可。 进入 anaconda 的 prompt 界面;
然后在 prompt 界面中输入命令 pip install openai
进行第三方库的安装;
安装完成之后,再使用命令 pip list
查看已安装的包,能看到 “openai” 说明第三方库安装成功。
第三步:Python 开发环境中调用 ChatGPT 模型
此处基于 python 开发环境(Python 项目)来调用 OpenAI 的第三方库,所以我们本地需要有已经搭建好的 python 开发环境,没有 python 环境的参见此文:
来到开发工具 PyCharm,新建一个项目,界面如下:
映入眼帘的是新项目默认生成的一段入门基础代码,运行后会输出 “Hi, PyCharm”,能运行成功说明我们的 python 环境是没有问题的。
接下来在项目文件夹下创建文件夹和 .py 文件,就可以编写代码了;
# 1.引入必须的包
import os
import openai
# 2.获取api-key
openai.api_key = "你自己的API-key"
# 3.使用OpenAI的API完成ChatGPT模型调用
# model:指的就是ChatGPT模型
# prompt:向ChatGPT提出的问题
# max_tokens:返回的最大字符个数
response = openai.Completion.create(
model="text-davinci-003",
prompt="请用python语言生成一个二分法查找算法",
max_tokens=256,
)
# 4.打印结果
message=response.choices[0].text
print(message)
代码编写完成之后,运行此程序,可以看到已生成二分查找算法;
ChatGPT 模型调用成功。