當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 有請求限制，所以我們不能毫無節制的調用該接口哈。