Compare commits

..

No commits in common. "main" and "v1.2.51" have entirely different histories.

19 changed files with 92 additions and 692 deletions

View File

@ -1,56 +0,0 @@
package app
import (
"fmt"
"github.com/gin-gonic/gin"
"go-wechat/model/vo"
"go-wechat/service"
"go-wechat/utils"
"log"
"net/http"
)
// SendAiSummary
// @description: 发送AI摘要
// @param ctx
func SendAiSummary(ctx *gin.Context) {
// 获取群Id
groupId := ctx.Query("id")
if groupId == "" {
ctx.String(http.StatusForbidden, "群Id不能为空")
return
}
// 取出群名称
groupInfo, err := service.GetFriendInfoById(groupId)
if err != nil {
ctx.String(http.StatusInternalServerError, "获取群信息失败")
return
}
// 获取对话记录
var records []vo.TextMessageItem
if records, err = service.GetTextMessagesById(groupId); err != nil {
log.Printf("获取群[%s]对话记录失败, 错误信息: %v", groupId, err)
ctx.String(http.StatusInternalServerError, "获取群对话记录失败")
return
}
if len(records) < 10 {
ctx.String(http.StatusForbidden, "群对话记录不足10条建议自己看")
return
}
// 组装对话记录为字符串
var replyMsg string
replyMsg, err = utils.GetAiSummary(groupInfo.Nickname, records)
if err != nil {
log.Printf("群聊记录总结失败: %v", err.Error())
ctx.String(http.StatusInternalServerError, "群聊消息总结失败,错误信息: "+err.Error())
return
}
replyMsg = fmt.Sprintf("#昨日消息总结\n又是一天过去了让我们一起来看看昨儿群友们都聊了什么有趣的话题吧~\n\n%s", replyMsg)
log.Printf("群[%s]对话记录总结成功,总结内容: %s", groupInfo.Nickname, replyMsg)
_ = utils.SendMessage(groupId, "", replyMsg, 0)
ctx.String(http.StatusOK, "操作完成")
}

View File

@ -31,7 +31,7 @@ system:
# 微信HOOK配置 # 微信HOOK配置
wechat: wechat:
# 微信HOOK接口地址 # 微信HOOK接口地址
host: 10.0.0.79:19098 host: 10.0.0.79:19088
# 微信容器映射出来的vnc页面地址没有就不填 # 微信容器映射出来的vnc页面地址没有就不填
# vncUrl: http://192.168.1.175:19087/vnc_lite.html # vncUrl: http://192.168.1.175:19087/vnc_lite.html
# 是否在启动的时候自动设置hook服务的回调 # 是否在启动的时候自动设置hook服务的回调
@ -61,9 +61,9 @@ task:
hotTop: hotTop:
enable: true enable: true
cron: '0 */1 * * *' # 每小时一次 cron: '0 */1 * * *' # 每小时一次
syncFriends: # 逻辑已修改,每天大半夜同步一次即可(主要是用以修正数据),平时会根据收到的消息针对性同步 syncFriends:
enable: false enable: false
cron: '0 4 * * *' # 每天4:00 cron: '*/5 * * * *' # 五分钟一次
groupSummary: groupSummary:
enable: false enable: false
cron: '30 0 * * *' # 每天0:30 cron: '30 0 * * *' # 每天0:30
@ -81,7 +81,7 @@ mq:
enable: false enable: false
# RabbitMQ配置 # RabbitMQ配置
rabbitmq: rabbitmq:
host: 10.0.0.31 host: 10.0.0.247
port: 5672 port: 5672
user: wechat user: wechat
password: wechat123 password: wechat123
@ -102,7 +102,6 @@ ai:
# 人设 # 人设
personality: 你的名字叫张三,你是一个百科机器人,你的爱好是看电影,你的性格是开朗的,你的专长是讲故事,你的梦想是当一名童话故事作家。你对政治没有一点儿兴趣,也不会讨论任何与政治相关的话题,你甚至可以拒绝回答这一类话题。 personality: 你的名字叫张三,你是一个百科机器人,你的爱好是看电影,你的性格是开朗的,你的专长是讲故事,你的梦想是当一名童话故事作家。你对政治没有一点儿兴趣,也不会讨论任何与政治相关的话题,你甚至可以拒绝回答这一类话题。
models: models:
- name: 默认
- name: ChatGPT-4 - name: ChatGPT-4
model: gpt-4 model: gpt-4
canManager: false canManager: false
@ -140,7 +139,3 @@ resource:
wordcloud: wordcloud:
type: image type: image
path: D:\Share\wordcloud\%s path: D:\Share\wordcloud\%s
# 邀请入群消息回复
invitation-join-group:
type: text
path: "您的邀请消息已收到啦,正在通知我的主人来同意请求。在我加群之后将会进行初始化操作,直到收到我主动发送的消息就是初始化完成咯,在那之前请耐心等待喔~"

View File

@ -6,6 +6,7 @@ import (
plugin "go-wechat/plugin" plugin "go-wechat/plugin"
"go-wechat/plugin/plugins" "go-wechat/plugin/plugins"
"go-wechat/service" "go-wechat/service"
"go-wechat/types"
) )
// Plugin // Plugin
@ -30,7 +31,7 @@ func Plugin() {
}, plugins.NotifyInvitationJoinGroup) }, plugins.NotifyInvitationJoinGroup)
// 被移除群聊通知到配置用户 // 被移除群聊通知到配置用户
dispatcher.RegisterHandler(func(m *dto.Message) bool { dispatcher.RegisterHandler(func(m *dto.Message) bool {
return m.IsRemoveFromChatroom() return m.Type == types.MsgTypeSys
}, plugins.NotifyRemoveFromChatroom) }, plugins.NotifyRemoveFromChatroom)
// 响应好友添加成功消息 // 响应好友添加成功消息
dispatcher.RegisterHandler(func(m *dto.Message) bool { dispatcher.RegisterHandler(func(m *dto.Message) bool {

View File

@ -12,8 +12,6 @@ import (
"html/template" "html/template"
"log" "log"
"net/http" "net/http"
"os"
"strconv"
"strings" "strings"
"time" "time"
) )
@ -24,13 +22,6 @@ func init() {
initialization.Plugin() // 注册插件 initialization.Plugin() // 注册插件
tasks.InitTasks() // 初始化定时任务 tasks.InitTasks() // 初始化定时任务
mq.Init() // 初始化MQ mq.Init() // 初始化MQ
if flag, _ := strconv.ParseBool(os.Getenv("DONT_SEND")); flag {
log.Printf("已设置环境变量DONT_SEND不发送消息")
}
if flag, _ := strconv.ParseBool(os.Getenv("DONT_SAVE")); flag {
log.Printf("已设置环境变量DONT_SAVE不保存消息")
}
} }
func main() { func main() {

View File

@ -250,31 +250,5 @@ func (m Message) IsJoinToGroup() (flag bool) {
if m.Type != types.MsgTypeSys { if m.Type != types.MsgTypeSys {
return return
} }
flag = strings.Contains(m.Content, "\"邀请你加入了群聊,群聊参与人还有:") return strings.Contains(m.Content, "\"邀请你加入了群聊,群聊参与人还有:")
if flag {
return
}
return strings.Contains(m.Content, "\"邀请你和\"") && strings.Contains(m.Content, "\"加入了群聊")
}
// IsRemoveFromChatroom
// @description: 是否是被移出群聊消息
// @receiver m
// @return flag
func (m Message) IsRemoveFromChatroom() (flag bool) {
if m.Type != types.MsgTypeSys {
return
}
return strings.HasPrefix(m.Content, "你被\"") && strings.HasSuffix(m.Content, "\"移出群聊")
}
// IsChangeGroupName
// @description: 是否是修改群名称消息
// @receiver m
// @return flag
func (m Message) IsChangeGroupName() (flag bool) {
if m.Type != types.MsgTypeSys {
return
}
return strings.HasPrefix(m.Content, "\"修改群名为“")
} }

View File

@ -43,7 +43,7 @@ func AI(m *plugin.MessageContext) {
if !friendInfo.EnableAi { if !friendInfo.EnableAi {
return return
} }
if friendInfo.AiFreeLimit > 0 && friendInfo.AiUsedToday >= friendInfo.AiFreeLimit { if friendInfo.AiUsedToday > 0 && friendInfo.AiUsedToday >= friendInfo.AiFreeLimit {
if notifyMap[m.FromUser] { if notifyMap[m.FromUser] {
return return
} }

View File

@ -2,41 +2,28 @@ package plugins
import ( import (
"fmt" "fmt"
"go-wechat/client"
"go-wechat/config" "go-wechat/config"
"go-wechat/model/entity"
"go-wechat/plugin" "go-wechat/plugin"
"go-wechat/service" "go-wechat/service"
"go-wechat/utils" "go-wechat/utils"
"strings"
) )
// NotifyInvitationJoinGroup // NotifyInvitationJoinGroup
// @description: 通知邀请入群消息到配置用户 // @description: 通知邀请入群消息到配置用户
// @param m // @param m
func NotifyInvitationJoinGroup(m *plugin.MessageContext) { func NotifyInvitationJoinGroup(m *plugin.MessageContext) {
// 先回复一条固定消息 // 先回复一条固定句子
if conf, e := config.Conf.Resource["invitation-join-group"]; e { utils.SendMessage(m.FromUser, m.GroupUser, "您的邀请消息已收到啦,正在通知我的主人来同意请求。在我加群之后将会进行初始化操作,直到收到我主动发送的消息就是初始化完成咯,在那之前请耐心等待喔~", 0)
// 发送一条新消息
switch conf.Type {
case "text":
// 文字类型
_ = utils.SendMessage(m.FromUser, "", conf.Path, 0)
case "image":
// 图片类型
utils.SendImage(m.FromUser, conf.Path, 0)
case "emotion":
// 表情类型
utils.SendEmotion(m.FromUser, conf.Path, 0)
}
}
// 推送到配置的用户 // 如果是邀请进群,推送到配置的用户
_, dec := m.IsInvitationJoinGroup() if flag, dec := m.IsInvitationJoinGroup(); flag {
for _, user := range config.Conf.System.NewFriendNotify.ToUser { for _, user := range config.Conf.System.NewFriendNotify.ToUser {
if user != "" { if user != "" {
// 发送一条新消息 // 发送一条新消息
dec = fmt.Sprintf("#邀请入群提醒\n\n%s", dec) dec = fmt.Sprintf("#邀请入群提醒\n\n%s", dec)
_ = utils.SendMessage(user, "", dec, 0) utils.SendMessage(user, "", dec, 0)
}
} }
} }
} }
@ -45,6 +32,8 @@ func NotifyInvitationJoinGroup(m *plugin.MessageContext) {
// @description: 被移除群聊通知到配置用户 // @description: 被移除群聊通知到配置用户
// @param m // @param m
func NotifyRemoveFromChatroom(m *plugin.MessageContext) { func NotifyRemoveFromChatroom(m *plugin.MessageContext) {
// 如果是被移出群聊,推送到配置的用户
if strings.HasPrefix(m.Content, "你被\"") && strings.HasSuffix(m.Content, "\"移出群聊") {
// 调用一下退出群聊接口,防止被移出后还能从好友列表接口拉到相关信息 // 调用一下退出群聊接口,防止被移出后还能从好友列表接口拉到相关信息
utils.QuitChatroom(m.FromUser, 0) utils.QuitChatroom(m.FromUser, 0)
@ -59,10 +48,8 @@ func NotifyRemoveFromChatroom(m *plugin.MessageContext) {
for _, user := range config.Conf.System.NewFriendNotify.ToUser { for _, user := range config.Conf.System.NewFriendNotify.ToUser {
if user != "" { if user != "" {
// 发送一条新消息 // 发送一条新消息
_ = utils.SendMessage(user, "", msg, 0) utils.SendMessage(user, "", msg, 0)
}
} }
} }
// 修改is_ok状态
client.MySQL.Model(&entity.Friend{}).Where("wxid = ?", m.FromUser).Update("is_ok", false)
} }

View File

@ -2,9 +2,7 @@ package plugins
import ( import (
"go-wechat/plugin" "go-wechat/plugin"
"go-wechat/service"
"go-wechat/utils" "go-wechat/utils"
"time"
) )
// ReplyNewFriend // ReplyNewFriend
@ -15,17 +13,6 @@ func ReplyNewFriend(m *plugin.MessageContext) {
_ = utils.SendMessage(m.FromUser, m.GroupUser, "AI正在初始化请稍等几分钟初始化完成之后我将主动告知您。", 0) _ = utils.SendMessage(m.FromUser, m.GroupUser, "AI正在初始化请稍等几分钟初始化完成之后我将主动告知您。", 0)
} }
if m.IsOldFriendBack() { if m.IsOldFriendBack() {
_ = utils.SendMessage(m.FromUser, "", "嘿,我的朋友,你为何要离我而去?又为何去而复返?", 0) _ = utils.SendMessage(m.FromUser, m.GroupUser, "嘿,我的朋友,你为何要离我而去?又为何去而复返?", 0)
} }
go func() {
// 等待5秒
time.Sleep(5 * time.Second)
// 刷新好友列表
service.SyncFriend()
// 如果是加入群,刷新群成员列表
if m.IsJoinToGroup() {
service.SyncGroupMembers(m.FromUser)
}
}()
} }

View File

@ -5,7 +5,6 @@ import (
"go-wechat/config" "go-wechat/config"
"go-wechat/model/entity" "go-wechat/model/entity"
"go-wechat/plugin" "go-wechat/plugin"
"go-wechat/service"
"go-wechat/utils" "go-wechat/utils"
) )
@ -29,7 +28,7 @@ func WelcomeNew(m *plugin.MessageContext) {
switch conf.Type { switch conf.Type {
case "text": case "text":
// 文字类型 // 文字类型
_ = utils.SendMessage(m.FromUser, "", conf.Path, 0) utils.SendMessage(m.FromUser, "", conf.Path, 0)
case "image": case "image":
// 图片类型 // 图片类型
utils.SendImage(m.FromUser, conf.Path, 0) utils.SendImage(m.FromUser, conf.Path, 0)
@ -37,7 +36,4 @@ func WelcomeNew(m *plugin.MessageContext) {
// 表情类型 // 表情类型
utils.SendEmotion(m.FromUser, conf.Path, 0) utils.SendEmotion(m.FromUser, conf.Path, 0)
} }
// 刷新群成员列表
go service.SyncGroupMembers(m.FromUser)
} }

View File

@ -28,7 +28,6 @@ func Init(g *gin.Engine) {
api.POST("/ai/model", app.ChangeUseAiModel) // 修改使用的AI模型 api.POST("/ai/model", app.ChangeUseAiModel) // 修改使用的AI模型
api.POST("/ai/free", app.ChangeAiFreeLimit) // 修改AI免费次数 api.POST("/ai/free", app.ChangeAiFreeLimit) // 修改AI免费次数
api.POST("/ai/assistant", app.ChangeUseAiAssistant) // 修改使用的AI助手 api.POST("/ai/assistant", app.ChangeUseAiAssistant) // 修改使用的AI助手
api.POST("/ai/summary", app.SendAiSummary) // 手动发送AI聊天总结
api.PUT("/welcome/status", app.ChangeEnableWelcomeStatus) // 修改是否开启迎新状态 api.PUT("/welcome/status", app.ChangeEnableWelcomeStatus) // 修改是否开启迎新状态
api.PUT("/command/status", app.ChangeEnableCommandStatus) // 修改是否开启指令状态 api.PUT("/command/status", app.ChangeEnableCommandStatus) // 修改是否开启指令状态
api.PUT("/news/status", app.ChangeEnableNewsStatus) // 修改是否开启早报状态 api.PUT("/news/status", app.ChangeEnableNewsStatus) // 修改是否开启早报状态

View File

@ -1,257 +0,0 @@
package service
import (
"go-wechat/client"
"go-wechat/common/types"
"go-wechat/config"
"go-wechat/model/entity"
"go-wechat/utils"
"log"
"strings"
"time"
)
// SyncFriend
// @description: 同步好友列表
func SyncFriend() {
// 获取好友列表
friends, err := utils.GetFriendList()
if err != nil {
log.Printf("获取好友列表失败: %s", err.Error())
return
}
// 当前获取到的成员Id用于后续设置is_ok状态
nowIds := make([]string, 0)
// 取出已存在的成员信息
var oldData []entity.Friend
err = client.MySQL.Find(&oldData).Error
if err != nil {
log.Printf("查询好友列表失败: %s", err.Error())
return
}
// 将历史数据整理成map
oldMap := make(map[string]bool)
for _, item := range oldData {
oldMap[item.Wxid] = item.IsOk
}
// 新增的成员,用于通知给指定的人
var notifyMap = make(map[string]string)
// 开启事务
tx := client.MySQL.Begin()
defer tx.Commit()
// 循环获取到的好友列表
for _, item := range friends {
// 填充当前存在的账号
nowIds = append(nowIds, item.Wxid)
// 判断是否已经存在
if _, ok := oldMap[item.Wxid]; ok {
// 已存在,修改
pm := map[string]any{
"nickname": item.Nickname,
"custom_account": item.CustomAccount,
"pinyin": item.Pinyin,
"pinyin_all": item.PinyinAll,
"is_ok": true,
}
err = tx.Model(&entity.Friend{}).Where("wxid = ?", item.Wxid).Updates(pm).Error
if err != nil {
log.Printf("修改好友失败: %s", err.Error())
continue
}
// 如果已存在但是是已退出的群,也通知一下
if !oldMap[item.Wxid] {
notifyMap[item.Wxid] = item.Nickname + " #秽土转生"
// 通知一下,初始化完成
if conf, e := config.Conf.Resource["introduce"]; e {
// 发送一条新消息
switch conf.Type {
case "text":
// 文字类型
_ = utils.SendMessage(item.Wxid, "", conf.Path, 0)
case "image":
// 图片类型
utils.SendImage(item.Wxid, conf.Path, 0)
case "emotion":
// 表情类型
utils.SendEmotion(item.Wxid, conf.Path, 0)
}
}
// 发送配置网页
if config.Conf.System.Domain != "" {
title := "欢迎使用微信机器人(切勿转发)"
desc := "点我可以配置功能喔,提示非微信官方网页,点击继续访问即可"
url := utils.GetManagerUrl(item.Wxid)
utils.SendPublicMsg(item.Wxid, title, desc, url, 0)
}
}
} else {
// 新增
err = tx.Create(&entity.Friend{
CustomAccount: item.CustomAccount,
Nickname: item.Nickname,
Pinyin: item.Pinyin,
PinyinAll: item.PinyinAll,
Wxid: item.Wxid,
IsOk: true,
EnableAi: config.Conf.System.DefaultRule.Ai,
EnableChatRank: config.Conf.System.DefaultRule.ChatRank,
EnableSummary: config.Conf.System.DefaultRule.Summary,
EnableWelcome: config.Conf.System.DefaultRule.Welcome,
EnableNews: config.Conf.System.DefaultRule.News,
EnableHotTop: config.Conf.System.DefaultRule.HotTop,
AiFreeLimit: config.Conf.System.DefaultRule.AiFreeLimit,
ClearMember: 0,
LastActive: types.DateTime(time.Now().Local()),
}).Error
if err != nil {
log.Printf("新增好友失败: %s", err.Error())
continue
}
notifyMap[item.Wxid] = item.Nickname
if conf, e := config.Conf.Resource["introduce"]; e {
// 发送一条新消息
switch conf.Type {
case "text":
// 文字类型
_ = utils.SendMessage(item.Wxid, "", conf.Path, 0)
case "image":
// 图片类型
utils.SendImage(item.Wxid, conf.Path, 0)
case "emotion":
// 表情类型
utils.SendEmotion(item.Wxid, conf.Path, 0)
}
}
// 发送配置网页
if config.Conf.System.Domain != "" {
title := "欢迎使用微信机器人(切勿转发)"
desc := "点我可以配置功能喔,提示非微信官方网页,点击继续访问即可"
url := utils.GetManagerUrl(item.Wxid)
utils.SendPublicMsg(item.Wxid, title, desc, url, 0)
}
}
}
// 通知有新成员
if len(notifyMap) > 0 && config.Conf.System.NewFriendNotify.Enable {
// 组装成一句话
msg := []string{"#新好友通知\n"}
for wxId, nickname := range notifyMap {
msg = append(msg, "微信Id: "+wxId+"\n昵称: "+nickname)
}
for _, user := range config.Conf.System.NewFriendNotify.ToUser {
if user != "" {
// 发送一条新消息
_ = utils.SendMessage(user, "", strings.Join(msg, "\n-------\n"), 0)
}
}
}
// 清理不在列表中的好友
clearPm := map[string]any{
"is_ok": false,
}
err = tx.Model(&entity.Friend{}).Where("wxid NOT IN (?)", nowIds).Updates(clearPm).Error
if err != nil {
log.Printf("清理好友失败: %s", err.Error())
}
log.Println("同步好友列表完成")
}
// SyncGroupMembers
// @description: 同步群成员
// @param wxId
func SyncGroupMembers(wxId string) {
membersIds, admin, err := utils.GetGroupMembers(wxId)
if err != nil {
return
}
// 修改不在数组的群成员状态为不在
pm := map[string]any{
"is_member": false,
"leave_time": time.Now().Local(),
}
err = client.MySQL.Model(&entity.GroupUser{}).
Where("group_id = ?", wxId).
Where("is_member IS TRUE").
Where("wxid NOT IN (?)", membersIds).
Updates(pm).Error
if err != nil {
log.Printf("修改群成员状态失败: %s", err.Error())
return
}
// 取出当前数据库存在的成员信息
var oldData []entity.GroupUser
err = client.MySQL.Model(&entity.GroupUser{}).
Where("group_id = ?", wxId).
Find(&oldData).Error
if err != nil {
log.Printf("查询群成员失败: %s", err.Error())
return
}
// 将历史数据整理成map
oldMap := make(map[string]bool)
for _, item := range oldData {
oldMap[item.Wxid] = item.IsMember
}
// 循环获取到的群成员列表
for _, wxid := range membersIds {
// 如果历史数据中存在,且是成员,跳过
if isMember, ok := oldMap[wxid]; ok && isMember {
continue
}
// 获取成员信息
cp, e := utils.GetContactProfile(wxid)
if e != nil {
log.Printf("获取成员信息失败: %s", e.Error())
continue
}
if cp.Wxid != "" {
if _, ok := oldMap[wxid]; ok {
// 历史数据中存在,修改
// 修改
gupm := map[string]any{
"account": cp.Account,
"head_image": cp.HeadImage,
"nickname": cp.Nickname,
"is_member": true,
"is_admin": cp.Wxid == admin,
"leave_time": nil,
}
err = client.MySQL.Model(&entity.GroupUser{}).Where("group_id = ?", wxId).Where("wxid = ?", wxid).Updates(gupm).Error
if err != nil {
log.Printf("修改群成员失败: %s", err.Error())
continue
}
} else {
// 新增的
// 新增
err = client.MySQL.Create(&entity.GroupUser{
GroupId: wxId,
Account: cp.Account,
HeadImage: cp.HeadImage,
Nickname: cp.Nickname,
Wxid: cp.Wxid,
IsMember: true,
IsAdmin: cp.Wxid == admin,
JoinTime: time.Now().Local(),
LastActive: time.Now().Local(),
}).Error
if err != nil {
log.Printf("新增群成员失败: %s", err.Error())
continue
}
}
}
}
}

View File

@ -1,11 +1,15 @@
package summary package summary
import ( import (
"context"
"fmt" "fmt"
"github.com/sashabaranov/go-openai"
"go-wechat/config"
"go-wechat/model/vo" "go-wechat/model/vo"
"go-wechat/service" "go-wechat/service"
"go-wechat/utils" "go-wechat/utils"
"log" "log"
"strings"
"time" "time"
) )
@ -33,21 +37,63 @@ func AiSummary() {
continue continue
} }
// 组装对话记录为字符串 // 组装对话记录为字符串
var replyMsg string var content []string
replyMsg, err = utils.GetAiSummary(group.Nickname, records) for _, record := range records {
content = append(content, fmt.Sprintf(`{"%s": "%s"}--end--`, record.Nickname, strings.ReplaceAll(record.Message, "\n", "。。")))
}
msgTmp := `请帮我总结一下一下的群聊内容的梗概生成的梗概需要尽可能详细需要带上一些聊天关键信息并且带上群友名字
注意他们可能是多个话题请仔细甄别
每一行代表一个人的发言每一行的的格式为 {"{nickname}": "{content}"}--end--
群名称: %s
聊天记录如下:
%s
`
msg := fmt.Sprintf(msgTmp, group.Nickname, strings.Join(content, "\n"))
// AI总结
messages := []openai.ChatCompletionMessage{
{
Role: openai.ChatMessageRoleUser,
Content: msg,
},
}
// 默认使用AI回复
conf := openai.DefaultConfig(config.Conf.Ai.ApiKey)
if config.Conf.Ai.BaseUrl != "" {
conf.BaseURL = fmt.Sprintf("%s/v1", config.Conf.Ai.BaseUrl)
}
ai := openai.NewClientWithConfig(conf)
var resp openai.ChatCompletionResponse
resp, err = ai.CreateChatCompletion(
context.Background(),
openai.ChatCompletionRequest{
Model: config.Conf.Ai.SummaryModel,
Messages: messages,
},
)
if err != nil { if err != nil {
log.Printf("群聊记录总结失败: %v", err.Error()) log.Printf("群聊记录总结失败: %v", err.Error())
_ = utils.SendMessage(group.Wxid, "", "#昨日消息总结\n\n群聊消息总结失败错误信息: "+err.Error(), 0) utils.SendMessage(group.Wxid, "", "#昨日消息总结\n\n群聊消息总结失败错误信息: "+err.Error(), 0)
continue continue
} }
replyMsg = fmt.Sprintf("#昨日消息总结\n又是一天过去了让我们一起来看看昨儿群友们都聊了什么有趣的话题吧~\n\n%s", replyMsg) // 返回消息为空
if resp.Choices[0].Message.Content == "" {
utils.SendMessage(group.Wxid, "", "#昨日消息总结\n\n群聊消息总结失败AI返回结果为空", 0)
continue
}
replyMsg := fmt.Sprintf("#昨日消息总结\n又是一天过去了让我们一起来看看昨儿群友们都聊了什么有趣的话题吧~\n\n%s", resp.Choices[0].Message.Content)
//log.Printf("群[%s]对话记录总结成功,总结内容: %s", group.Wxid, replyMsg) //log.Printf("群[%s]对话记录总结成功,总结内容: %s", group.Wxid, replyMsg)
_ = utils.SendMessage(group.Wxid, "", replyMsg, 0) utils.SendMessage(group.Wxid, "", replyMsg, 0)
// 判断耗时是否达到一分钟,不足就等待 // 判断耗时是否达到15秒,不足就等待
if used := time.Now().Sub(start); used < time.Minute { if used := time.Now().Sub(start); used < 15*time.Second {
time.Sleep(time.Minute - used) time.Sleep(15*time.Second - used)
} }
} }
} }

View File

@ -1,105 +0,0 @@
package utils
import (
"encoding/json"
"github.com/go-resty/resty/v2"
"go-wechat/common/constant"
"go-wechat/config"
"go-wechat/model/dto"
"log"
"slices"
"strings"
)
// http客户端
var hc = resty.New()
// GetFriendList
// @description: 获取好友列表
// @return friends
// @return err
func GetFriendList() (friends []dto.FriendItem, err error) {
var base dto.Response[[]dto.FriendItem]
resp, err := hc.R().
SetHeader("Content-Type", "application/json;chartset=utf-8").
SetResult(&base).
Post(config.Conf.Wechat.GetURL("/api/getContactList"))
if err != nil {
log.Printf("获取好友列表失败: %s", err.Error())
return
}
log.Printf("获取好友列表结果: %s", resp.String())
// 循环获取到的好友列表
for _, item := range base.Data {
// 跳过特殊账号
// 跳过公众号和企业微信好友
if strings.Contains(item.Wxid, "gh_") || strings.Contains(item.Wxid, "@openim") {
continue
}
// 特殊Id跳过
if slices.Contains(constant.SpecialId, item.Wxid) {
continue
}
// 添加到待返回列表
friends = append(friends, item)
}
return
}
// GetGroupMembers
// @description: 获取指定群成员
// @param wxId 群Id
// @return ids 群成员id数组
// @return members 群成员信息
// @return err 错误信息
func GetGroupMembers(wxId string) (ids []string, admin string, err error) {
var base dto.Response[dto.GroupUser]
// 组装参数
param := map[string]any{
"chatRoomId": wxId, // 群Id
}
pbs, _ := json.Marshal(param)
_, err = hc.R().
SetHeader("Content-Type", "application/json;chartset=utf-8").
SetBody(string(pbs)).
SetResult(&base).
Post(config.Conf.Wechat.GetURL("/api/getMemberFromChatRoom"))
if err != nil {
log.Printf("获取群成员信息失败: %s", err.Error())
return
}
// 昵称Id
ids = strings.Split(base.Data.Members, "^G")
admin = base.Data.Admin
return
}
// GetContactProfile
// @description: 获取联系人信息
// @param wxId 微信Id
// @return info 联系人信息
// @return err 错误信息
func GetContactProfile(wxId string) (info dto.ContactProfile, err error) {
var baseResp dto.Response[dto.ContactProfile]
// 组装参数
param := map[string]any{
"wxid": wxId, // 群Id
}
pbs, _ := json.Marshal(param)
_, err = hc.R().
SetHeader("Content-Type", "application/json;chartset=utf-8").
SetBody(string(pbs)).
SetResult(&baseResp).
Post(config.Conf.Wechat.GetURL("/api/getContactProfile"))
if err != nil {
log.Printf("获取成员详情失败: %s", err.Error())
return
}
info = baseResp.Data
return
}

View File

@ -8,8 +8,6 @@ import (
"go-wechat/common/current" "go-wechat/common/current"
"go-wechat/config" "go-wechat/config"
"log" "log"
"os"
"strconv"
"time" "time"
) )
@ -19,10 +17,6 @@ import (
// @param atId // @param atId
// @param msg // @param msg
func SendMessage(toId, atId, msg string, retryCount int) (err error) { func SendMessage(toId, atId, msg string, retryCount int) (err error) {
if flag, _ := strconv.ParseBool(os.Getenv("DONT_SEND")); flag {
return
}
if retryCount > 5 { if retryCount > 5 {
log.Printf("重试五次失败,停止发送") log.Printf("重试五次失败,停止发送")
err = errors.New("重试五次失败,停止发送") err = errors.New("重试五次失败,停止发送")
@ -68,10 +62,6 @@ func SendMessage(toId, atId, msg string, retryCount int) (err error) {
// @param imgPath string 图片路径 // @param imgPath string 图片路径
// @param retryCount int 重试次数 // @param retryCount int 重试次数
func SendImage(toId, imgPath string, retryCount int) { func SendImage(toId, imgPath string, retryCount int) {
if flag, _ := strconv.ParseBool(os.Getenv("DONT_SEND")); flag {
return
}
if retryCount > 5 { if retryCount > 5 {
log.Printf("重试五次失败,停止发送") log.Printf("重试五次失败,停止发送")
return return
@ -104,10 +94,6 @@ func SendImage(toId, imgPath string, retryCount int) {
// @param emotionHash string 表情包hash(md5值) // @param emotionHash string 表情包hash(md5值)
// @param retryCount int 重试次数 // @param retryCount int 重试次数
func SendEmotion(toId, emotionHash string, retryCount int) { func SendEmotion(toId, emotionHash string, retryCount int) {
if flag, _ := strconv.ParseBool(os.Getenv("DONT_SEND")); flag {
return
}
if retryCount > 5 { if retryCount > 5 {
log.Printf("重试五次失败,停止发送") log.Printf("重试五次失败,停止发送")
return return
@ -214,10 +200,6 @@ func QuitChatroom(chatRoomId string, retryCount int) {
// @param url string 链接 // @param url string 链接
// @param retryCount int 重试次数 // @param retryCount int 重试次数
func SendPublicMsg(wxId, title, digest, url string, retryCount int) { func SendPublicMsg(wxId, title, digest, url string, retryCount int) {
if flag, _ := strconv.ParseBool(os.Getenv("DONT_SEND")); flag {
return
}
if retryCount > 5 { if retryCount > 5 {
log.Printf("重试五次失败,停止发送") log.Printf("重试五次失败,停止发送")
return return

View File

@ -1,108 +0,0 @@
package utils
import (
"context"
"errors"
"fmt"
"github.com/sashabaranov/go-openai"
"go-wechat/config"
"go-wechat/model/vo"
"log"
"strings"
)
// GetAiSummary
// @description: AI总结群聊记录
// @param groupName
// @param records
// @return replyMsg
// @return err
func GetAiSummary(groupName string, records []vo.TextMessageItem) (replyMsg string, err error) {
// 组装对话记录为字符串
var content []string
for _, record := range records {
content = append(content, fmt.Sprintf(`{"%s": "%s"}--end--`, record.Nickname, strings.ReplaceAll(record.Message, "\n", "。。")))
}
prompt := `# Role: 数据分析师和信息整理专家
# Background: 用户需要从群聊记录中提取有价值的信息包括技术话题讨论工具分享网站或资源链接
# Profile: 你是一位专业的数据分析师擅长从大量文本中提取关键信息并能够高效地整理和总结内容
# Skills: 数据分析信息提取内容总结网络资源评估
# Goals: 设计一个流程帮助用户识别群聊中的技术话题共享的工具和资源并高效地总结每个话题的要点
# Constrains: 该流程需要简洁明了易于用户理解和操作同时确保不遗漏任何重要信息
# OutputFormat: 结构化的文本总结包含关键讨论点结论等
# Workflow:
## 1. 快速浏览群聊记录
- 通过快速浏览聊天记录初步识别出主要话题
## 2. 识别和分类主要话题
- 识别群聊中的主要话题分类讨论内容
- 每个话题独立提取避免混淆
- 话题识别尽可能准确确保不遗漏重要信息
## 3. 编写总结概括
- 对每个话题进行总结包含以下内容
- 关键讨论点主要的讨论内容和观点
- 参与成员参与讨论的成员的昵称
- 结论得出的结论或共识内容要尽可能详细
## 4. 组织信息
- 以逻辑清晰易于理解的方式组织信息
- 使用结构化文本使总结内容易于阅读
## 5. 生成最终总结
- 根据上述步骤生成每个话题的详细总结
- 确保每个总结完整且有条理
- 返回数据格式为结构化文本方便用户查阅
## 6. 群聊记录消息格式
格式: {"{nickname}": "{content}"}--end--
# Examples:
- 话题技术讨论 - AI在医疗领域的应用
- 关键讨论点讨论了AI在医疗诊断中的应用前景和现有挑战
- 成员张三李四王五
- 结论AI技术有望提高医疗诊断的准确性和效率但数据隐私和模型解释性仍需解决
- 话题美食讨论 - 杭州到底有没有美食
- 关键讨论点讨论了杭州的美食有哪些分布地区评价如何
- 成员张三李四王五
- 结论杭州好像没有美食
`
msg := fmt.Sprintf("群名称: %s\n聊天记录如下:\n%s", groupName, strings.Join(content, "\n"))
// AI总结
messages := []openai.ChatCompletionMessage{
{
Role: openai.ChatMessageRoleSystem,
Content: prompt,
},
{
Role: openai.ChatMessageRoleUser,
Content: msg,
},
}
// 默认使用AI回复
conf := openai.DefaultConfig(config.Conf.Ai.ApiKey)
if config.Conf.Ai.BaseUrl != "" {
conf.BaseURL = fmt.Sprintf("%s/v1", config.Conf.Ai.BaseUrl)
}
ai := openai.NewClientWithConfig(conf)
var resp openai.ChatCompletionResponse
resp, err = ai.CreateChatCompletion(
context.Background(),
openai.ChatCompletionRequest{
Model: config.Conf.Ai.SummaryModel,
Messages: messages,
},
)
if err != nil {
log.Printf("群聊记录总结失败: %v", err.Error())
return
}
// 返回消息为空
if resp.Choices[0].Message.Content == "" {
err = errors.New("AI返回结果为空")
return
}
replyMsg = resp.Choices[0].Message.Content
return
}

View File

@ -16,6 +16,6 @@ func GetManagerUrl(wxId string) (url string) {
// base64一下 // base64一下
encodeString := base64.StdEncoding.EncodeToString([]byte(url)) encodeString := base64.StdEncoding.EncodeToString([]byte(url))
// 拼接新链接(这个是一个已备案的域名) // 拼接新链接(这个是一个已备案的域名)
url = "https://redirect.gitee.ltd/?s=" + encodeString url = "https://redirect.wjg95.cn/?s=" + encodeString
return return
} }

View File

@ -57,9 +57,6 @@
<!-- 消息总结 --> <!-- 消息总结 -->
{{define "summary"}} {{define "summary"}}
{{ if eq .EnableSummary true }}
<button type="button" class="btn-link float-end text-blue-600" onclick="sendAiSummary({{.Wxid}})">手动发送总结</button>
{{ end }}
<button type="button" <button type="button"
class="{{ if eq .EnableSummary true }}bg-green-600{{ else }}bg-gray-200{{ end }} relative inline-flex h-6 w-11 flex-shrink-0 cursor-pointer rounded-full border-2 border-transparent transition-colors duration-200 ease-in-out focus:outline-none focus:ring-2 focus:ring-green-600 focus:ring-offset-2" class="{{ if eq .EnableSummary true }}bg-green-600{{ else }}bg-gray-200{{ end }} relative inline-flex h-6 w-11 flex-shrink-0 cursor-pointer rounded-full border-2 border-transparent transition-colors duration-200 ease-in-out focus:outline-none focus:ring-2 focus:ring-green-600 focus:ring-offset-2"
role="switch" aria-checked="false" onclick="changeSummaryEnableStatus({{.Wxid}})"> role="switch" aria-checked="false" onclick="changeSummaryEnableStatus({{.Wxid}})">

View File

@ -56,7 +56,6 @@
</div> </div>
{{ if eq .EnableAi true }} {{ if eq .EnableAi true }}
<div class="float-end"> <div class="float-end">
<!-- 模型 -->
<div> <div>
<label> <label>
<select class="block w-full rounded-md border-0 py-1.5 pl-3 pr-10 text-gray-900 ring-1 ring-inset ring-gray-300 focus:ring-2 focus:ring-green-600 sm:text-sm sm:leading-6" onchange="aiModelChange(event, {{.Wxid}})"> <select class="block w-full rounded-md border-0 py-1.5 pl-3 pr-10 text-gray-900 ring-1 ring-inset ring-gray-300 focus:ring-2 focus:ring-green-600 sm:text-sm sm:leading-6" onchange="aiModelChange(event, {{.Wxid}})">
@ -69,7 +68,6 @@
</select> </select>
</label> </label>
</div> </div>
<!-- AI角色 -->
<div class="float-end mt-1"> <div class="float-end mt-1">
<label> <label>
<select <select

View File

@ -317,30 +317,3 @@ function changeAiFreeLimit(wxid, limitNumber) {
window.location.reload(); window.location.reload();
}) })
} }
// 发送AI群聊总结
function sendAiSummary(wxid) {
if (wxid === '') {
alert("wxid不能为空")
return
}
if (!confirm("确定发送AI群聊总结吗?")) {
return
}
// 请求接口
axios({
method: 'post',
url: '/api/ai/summary',
params: {
id: wxid,
}
}).then(function (response) {
console.log(`返回结果: ${JSON.stringify(response)}`);
alert(`${response.data}`)
}).catch(function (error) {
console.log(`错误信息: ${error}`);
alert(`${response.data}`)
}).finally(function () {
window.location.reload();
})
}