跳到主要内容

基于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