当ChatGPT遇见Python

在如火如荼的 ChatGPT 大潮当中，已经衍生出了各种各样的周边产品。

Python 作为著名的万金油工具，怎么能没有它的身影呢。今天我们就介绍两种通过 Python 调用 ChatGPT 的方法，一起来看看吧！

chatgpt-wrapper

这是一个开源在 GitHub 上的项目，通过自动化工具 Playwright 来驱动调用 CHatGPT。

我们配置好这个工具之后，就可以在命令行以及 Python 代码中使用 ChatGPT了。

安装配置

首先当然是安装克隆该项目了。

pip install git+https://github.com/mmabrouk/chatgpt-wrapper

然后我们安装 Playwright。

pip install playwright

接下来我们在 Playwright 当中安装浏览器，比如 firefox。

playwright install firefox

以上都完成后，我们在终端执行如下命令：

chatgpt install

接下来会弹出一个 ChatGPT 的登录页面，输入账号密码，点击登陆即可，如果还没有 ChatGPT 账号密码，可以在公众号后台发送「chatgpt」获取共享测试账号，先到先得哈！

登陆完成之后我们就可以重启终端，在里面输入 chatgpt，此时就可以开始和 CHatGPT 愉快的玩耍了。

接入 Python

当然我们必须可以把这一切都植入 Python 代码。

from chatgpt_wrapper import ChatGPT

bot = ChatGPT()
response = bot.ask("Hello, world!")
print(response)  # prints the response from chatGPT

区区三行代码，我们仿佛拥有了整个世界！

OpenAI

第二种方法是通过 OpenAI 开放的接口。

我们先安装 OpenAI 库。

pip install openai

然后需要获取 api key，需要访问如下网站 https://platform.openai.com/account/api-key，添加对应的 key 即可，保存该 key。

接下来我们就可以编写代码了，也非常简单。

import openai

# 设置 API Key
openai.api_key = "上面保存的api key"

# 设置请求参数
model_engine = "text-davinci-002"
prompt = "Hello World"

completions = openai.Completion.create(
    engine=model_engine,
    prompt=prompt,
    max_tokens=1024,
    n=1,
    stop=None,
    temperature=0.5,
)

# 获取 ChatGPT 的回复
message = completions.choices[0].text
print(message)

在上面的代码中，我们注意到，可以通过更改代码中的请求参数，如提示、模型、温度等，来调整 ChatGPT 的响应。

不过当前需要注意，OpenAI 的 API 有请求限制，所以我们不能毫无节制的调用该接口哈。