zhayujie e5e0979bfa feat: bring in openai completion api | 1 år sedan | |
---|---|---|
bot | 1 år sedan | |
bridge | 1 år sedan | |
channel | 1 år sedan | |
common | 1 år sedan | |
docs/images | 1 år sedan | |
.gitignore | 1 år sedan | |
LICENSE | 1 år sedan | |
README.md | 1 år sedan | |
app.py | 1 år sedan | |
config.json | 1 år sedan | |
config.py | 1 år sedan |
本项目是基于ChatGPT模型实现的微信聊天机器人,通过 OpenAI 访提供的API,使用 itchat 实现微信消息的接收和发送。已实现的功能如下:
2022.12.17: 原来的方案是从 ChatGPT页面 获取session_token,使用 revChatGPT 直接访问web接口,但随着ChatGPT接入Cloudflare人机验证,这一方案难以在服务器顺利运行。 所以目前使用的方案是调用 OpenAI 官方提供的 API,劣势是暂不支持有上下文记忆的对话、且回复内容的智能性上相比ChatGPT稍差一些,优势是稳定性和响应速度较好。
本方案中实现微信消息的收发依赖了网页版微信的登录,可以尝试登录 https://wx.qq.com/,如果能够成功登录就可以开始后面的步骤了。
前往 OpenAI注册页面 创建账号,参考这篇 博客 可以通过虚拟手机号来接收验证码。创建完账号则前往 API管理页面 创建一个 API Key 并保存下来,后面需要在项目中配置这个key。
项目中使用的对话模型是 davinci,计费方式是每1k字 (包含请求和回复) 消耗 $0.02,账号创建有免费的 $18 额度,使用完可以更换邮箱重新注册。
支持运行在 Linux、MacOS、Windows 操作系统上,需安装 Python3.6
及以上版本。推荐使用Linux服务器,可以托管在后台长期运行。
克隆项目代码:
https://github.com/zhayujie/chatgpt-on-wechat
安装所需核心依赖:
pip3 install itchat
pip3 install openai
配置文件在根目录的 config.json
中,示例文件及各配置项含义如下:
{
"open_ai_api_key": "${YOUR API KEY}$" # 上面在创建的 API KEY
"single_chat_prefix": ["bot", "@bot"], # 私聊时文本需要包含该前缀才能触发机器人回复
"single_chat_reply_prefix": "[bot] ", # 私聊时自动回复的前缀,用于区分真人
"group_chat_prefix": ["@bot"], # 群聊时包含该前缀则会触发机器人回复
"group_name_white_list": ["ChatGPT测试群", "ChatGPT测试群2"] # 开启自动回复的群名称列表
}
关于OpenAI对话接口的参数配置,可以参考 接口文档 直接在代码 bot\openai\open_ai_bot.py
中进行调整。
1.如果是开发机本地调试,直接执行:
python3 app.py
终端输出二维码后,使用微信进行扫码,当输出 “Start auto replying” 时表示自动回复程序已经成功运行了。
2.如果是服务器部署,则使用nohup在后台运行:
nohup python3 app.py & tail -f nohup.out
同样在扫码后程序即可运行于后台。
默认配置中,个人聊天会以 “bot” 或 “@bot” 为开头的内容触发机器人,对应配置中的 single_chat_prefix
;机器人回复的内容会以 “[bot]” 作为前缀, 以区分真人,对应的配置为 single_chat_reply_prefix
。
群名称需要配置在 group_name_white_list
中才能开启群聊自动回复,默认只要被@就会触发机器人自动回复,另外群聊天中只要检测到以 “@bot” 开头的内容,同样会自动回复,这对应配置 group_chat_prefix
。