本文将演示在本地的 python 项目中调用 ChatGPT 模型。
- 写在前面
- 第一步:获取 API Key
- 第二步:安装 OpenAI 第三方库
- 第三步: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 环境的参见此文:
一文掌握 Python、Anaconda、PyCharm 的安装和使用
来到开发工具 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 模型调用成功。