深圳幻海软件技术有限公司 欢迎您!

当ChatGPT遇见Python

2023-02-28

在如火如荼的ChatGPT大潮当中,已经衍生出了各种各样的周边产品。Python作为著名的万金油工具,怎么能没有它的身影呢。今天我们就介绍两种通过Python调用ChatGPT的方法,一起来看看吧!chatgpt-wrapper这是一个开源在GitHub上的项目,通过自动化工具Playwright来

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

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

chatgpt-wrapper

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

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

安装配置

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

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

然后我们安装 Playwright。

pip install playwright
  • 1.

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

playwright install firefox
  • 1.

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

chatgpt install
  • 1.

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

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

QQWbJZRg" data-id="h6f20189-cQPL9YNU">接入 Python

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

from chatgpt_wrapper import ChatGPT

bot = ChatGPT()
response = bot.ask("Hello, world!")
print(response)  # prints the response from chatGPT
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.

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

OpenAI

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

我们先安装 OpenAI 库。

pip install openai
  • 1.

然后需要获取 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)
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.
  • 16.
  • 17.
  • 18.
  • 19.
  • 20.
  • 21.

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

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