跳到主要内容

基于OpenIM 实现聊天机器人功能

简要描述

使用 OpenIM 中的 Webhook 机制实现聊天机器人功能。发送文本消息或图片消息给聊天机器人后,机器人会返回相同的消息。开发者可以替换此逻辑,调用LLM接口,以实现智能问答功能。

1. 修改配置文件

参照下面的模板修改 open-im-server 中的 config/webhooks.yml 配置。

PC Web Interface

提示:

  • url 为回调 URL。
  • afterSendSingleMsg.enable 设置为 true 时,启用该回调。

2. 创建聊天机器人账号

  1. 登录管理后台,参考 此文档
  2. 在用户管理中创建聊天机器人账号,并记录该账号的 userID
  3. 为方便体验,可以将此 userID 设置为默认好友。

3. 编写 afterSendSingleMsg 接口

参考以下示例代码。

提示:

  1. 将例子中的 robotics 替换为第 2 步中获取的 userID
func (m *ChatApi) CallbackExample(c *gin.Context) {
// 1. Handling callbacks after sending a single chat message
msgInfo, err := handlingCallbackAfterSendMsg(c)
if err != nil {
apiresp.GinError(c, err)
return
}

// 2. If the user receiving the message is a customer service bot, return the message.
// 2.1 UserID of the robot account
robotics := "robotics"

// 2.2 ChatRobot account validation and determining if messages are text and images
if msgInfo.SendID == robotics || msgInfo.RecvID != robotics {
return
}
if msgInfo.ContentType != constant.Picture && msgInfo.ContentType != constant.Text {
return
}

// 2.3 Get administrator token
adminToken, err := getAdminToken(c)
if err != nil {
apiresp.GinError(c, err)
return
}

// 2.4 Get RobotAccount info
robUser, err := getRobotAccountInfo(c, adminToken.AdminToken, robotics)
if err != nil {
apiresp.GinError(c, err)
return
}

// 2.5 Constructing the contents of the message field or invoking an LLM to implement AI-driven question answering.
mapStruct, err := contextToMap(c, msgInfo)
if err != nil {
apiresp.GinError(c, err)
return
}

// 2.6 Send Message
err = sendMessage(c, adminToken.ImToken, robotics, msgInfo, robUser, mapStruct)
if err != nil {
apiresp.GinError(c, err)
return
}
}

4. 效果展示

PC Web Interface