把网页版 ChatGPT 封装为一个简单的 API,以便在代码中使用。
pip3 install git+https://github.com/ultrasev/chatrapper.git
环境变量中设置 TOKEN
,然后调用 chat
函数即可。
export TOKEN="eyJhbGci..."
在代码中使用 Rapper
:
import os
from chatrapper import Rapper
token = os.environ.get("TOKEN")
rapper = Rapper(
access_token=token
model="text-davinci-002-render-sha"
)
rapper("鲁迅为什么打周树人?")
或者有异步需求的话,可以使用 AsyncRapper
。这种情况下,最好有多个账号支持,单账号下,同一时间只支持一轮对话。
import os
import asyncio
from chatrapper import AsyncRapper
token = os.environ.get("TOKEN")
rapper = AsyncRapper(
access_token=token
model="text-davinci-002-render-sha"
)
async def main():
print(await rapper("鲁迅为什么打周树人?"))
asyncio.run(main())
Demo:
- 一定要保护好自己的 token,不要泄露给他人。
- 合理使用 API,调用频率不宜过高,树大易招风,避免触发风控。