当ChatGPT遇见Python

Posted 2023-03-14 一枚务实的码农

在如火如荼的 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 愉快的玩耍了

接入 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 有请求限制，所以我们不能毫无节制的调用该接口哈~

好了，这就是今天分享的全部内容，喜欢就点个赞吧！