/chatgpt-wecom

企业微信接入 GPT3

Primary LanguageGoMIT LicenseMIT

chatgpt-wecom

企业微信接入 GPT3 接口。可以创建企业微信应用,体验 ChatGPT。

相较于官方提供的 CreateCompletion 接口,该项目增加了会话管理功能,能够较好地提供多轮对话能力。

快速安装

0. 前置步骤

  • 登录企业微信的管理后台创建机器人,名字随意取
    • 点击【应用管理】-【创建应用】,填写完毕保存。

1. 配置

修改 conf/chatgpt.conf 文件,主要涉及企业微信应用配置、GPT3 API Key、会话管理数据库配置等。

  • 企业微信应用配置
    • corp_id:在企业微信后台【我的企业】-【企业信息】处获取【企业ID】
    • corp_secret:在企业微信后台【应用管理】处获取【Secret】
    • agent_id:在企业微信后台【应用管理】处获取【AgentId】
    • encoding_aes_key:企业微信后台 【接收消息】- 【API 接收消息】获取【EncodingAESKey】,可以随机生成
    • token:企业微信后台 【接收消息】- 【API 接收消息】获取【Token】,可以随机生成
  • OpenAI Key
    • 需要自行申请
  • 数据库
    • 数据库需要自行创建,数据表的创建可以通过命令行方式执行。
    • 数据库支持 sqlite3,可以通过修改配置使用。如果使用 MySQL,需要自行创建数据库。
    • 数据表在程序启动时自动创建。

2. 运行

  • 选择1:Docker运行(sqlite3版,推荐)
git clone https://github.com/fanchunke/chatgpt-wecom.git
cd chatgpt-wecom

# 构建镜像
make dockerenv

# 运行带sqlite的镜像,运行前确认chatgpt.conf修改完毕
docker run -it -d --name chatgpt --restart=always \
  -v $(pwd)/conf/chatgpt.conf:/home/works/program/conf/chatgpt.conf \
  -p 0.0.0.0:8000:8000 chatgpt-wecom:0.1.1
  • 选择2:Docker运行(MySQL版)
# 运行前确认chatgpt.conf修改完毕
docker compose up -d
  • 选择3:本地运行
  • 下载对应的二进制,chatgpt-wecom
  • 执行命令 ./chatgpt-wecom -conf=conf/chatgpt.conf 即可,同理需要确认chatgpt.conf配置完毕

3. 配置企业微信

  • URL 配置格式
    • text-davinci-003 版本:http://ip:port/wecom/receive
    • gpt-3.5-turbo 版本:http://ip:port/wecom/receive/v2
  • 在企业微信后台,添加可信IP地址

FAQ

怎么创建数据库

  • 新版本支持 sqlite3 数据库,只需要修改配置文件的配置,程序启动后便会初始化数据库和数据表,不需要额外的操作。

  • 如果使用的是 MySQL,则需要自行创建数据库,建库 SQL 可直接使用命令:init.sql,之后程序启动,便可以自动创建数据表。

数据库连接失败

  • 首先检查数据库配置是否正确
  • 如果使用 docker 部署服务,需要确认容器内能否连接到数据库。最常见的一个问题是,在宿主机部署了 MySQL,但是在容器内配置 127.0.0.1,这种情况需要配置宿主机的 IP

数据库配置说明

新版本可以支持 MySQL、SQLite、PostgreSQL。常见的配置如下:

MySQL:

[database]
# mysql
driver="mysql"
dataSource="root:12345678@tcp(127.0.0.1:3306)/chatgpt?parseTime=True&loc=Local"

SQLite

[database]
# sqlite3
driver="sqlite3"
dataSource="file:chatgpt?_fk=1&parseTime=True&loc=Local"

dataSource 字段有两个参数需要配置:

  • parseTime=True
  • loc=Local: 以本地时间存储时间类型的字段

Changelog

v0.1.5

  • 修复跨平台编译包问题

v0.1.4

  • 增加 gpt-3.5-turbo API 的支持

v0.1.3

  • 添加makefile,docker-compose自动build MySQL等 #14

v0.1.2

  • 修复MySQL数据库迁移问题

v0.1.1

  • 修复 prompt 过长导致接口调用失败问题
  • 支持 sqlite3
  • 自动初始化数据库
  • 支持企业微信进入事件,可以配置进入事件回复语。需要先在企业微信上配置【上报进入应用事件】
  • 支持关闭会话功能

v0.1.0

  • 项目初始化

Happy Chatting

img