Compare commits

..

2 Commits

Author SHA1 Message Date
李寻欢
91d2fc50e2 🎨 优化传递给AI的消息,去掉艾特机器人那一段字符串 2023-11-17 09:44:14 +08:00
李寻欢
119c2a5359 🎨 优化AI返回内容格式,加个换行 2023-11-13 16:58:52 +08:00

View File

@@ -8,6 +8,8 @@ import (
"go-wechat/entity"
"go-wechat/utils"
"log"
"regexp"
"strings"
)
// handleAtMessage
@@ -17,6 +19,15 @@ func handleAtMessage(m entity.Message) {
if !config.Conf.Ai.Enable {
return
}
// 预处理一下发送的消息,用正则去掉@机器人的内容
re := regexp.MustCompile(`@([^]+)`)
matches := re.FindStringSubmatch(m.Content)
if len(matches) > 0 {
// 过滤掉第一个匹配到的
m.Content = strings.Replace(m.Content, matches[0], "", 1)
}
// 默认使用AI回复
conf := openai.DefaultConfig(config.Conf.Ai.ApiKey)
if config.Conf.Ai.BaseUrl != "" {
@@ -43,5 +54,5 @@ func handleAtMessage(m entity.Message) {
}
// 发送消息
utils.SendMessage(m.FromUser, m.GroupUser, resp.Choices[0].Message.Content, 0)
utils.SendMessage(m.FromUser, m.GroupUser, "\n"+resp.Choices[0].Message.Content, 0)
}