From b3ed0fcc6fb62f2c93f507d23254faeec3560656 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=AF=BB=E6=AC=A2?= Date: Fri, 12 Apr 2024 11:37:21 +0800 Subject: [PATCH] =?UTF-8?q?:new:=20=E6=96=B0=E5=A2=9E=E7=BE=A4=E8=81=8A?= =?UTF-8?q?=E5=AF=B9=E8=AF=9D=E8=AE=B0=E5=BD=95=E6=80=BB=E7=BB=93?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/friend.go | 23 ++++++++++++ config.yaml | 25 ++++++++----- config/ai.go | 13 ++++--- config/task.go | 7 ++-- entity/friend.go | 1 + router/router.go | 1 + service/friend.go | 9 +++++ service/message.go | 20 ++++++++++ tasks/summary/summary.go | 81 ++++++++++++++++++++++++++++++++++++++++ tasks/tasks.go | 8 ++++ views/index.html | 10 +++++ views/static/js/index.js | 23 ++++++++++++ vo/friend.go | 1 + vo/message.go | 8 ++++ 14 files changed, 211 insertions(+), 19 deletions(-) create mode 100644 tasks/summary/summary.go create mode 100644 vo/message.go diff --git a/app/friend.go b/app/friend.go index d131d7a..e9d28f4 100644 --- a/app/friend.go +++ b/app/friend.go @@ -90,6 +90,29 @@ func ChangeEnableGroupRankStatus(ctx *gin.Context) { ctx.String(http.StatusOK, "操作成功") } +// ChangeEnableSummaryStatus +// @description: 修改是否开启聊天记录总结 +// @param ctx +func ChangeEnableSummaryStatus(ctx *gin.Context) { + var p changeStatusParam + if err := ctx.ShouldBindJSON(&p); err != nil { + ctx.String(http.StatusBadRequest, "参数错误") + return + } + log.Printf("待修改的群Id:%s", p.WxId) + + err := client.MySQL.Model(&entity.Friend{}). + Where("wxid = ?", p.WxId). + Update("`enable_summary`", gorm.Expr(" !`enable_summary`")).Error + if err != nil { + log.Printf("修改开启聊天记录总结失败:%s", err) + ctx.String(http.StatusInternalServerError, "操作失败: %s", err) + return + } + + ctx.String(http.StatusOK, "操作成功") +} + // ChangeEnableWelcomeStatus // @description: 修改是否开启迎新 // @param ctx diff --git a/config.yaml b/config.yaml index b5ff71e..044a642 100644 --- a/config.yaml +++ b/config.yaml @@ -3,7 +3,7 @@ wechat: # 微信HOOK接口地址 host: 10.0.0.73:19088 # 微信容器映射出来的vnc页面地址,没有就不填 -# vncUrl: http://192.168.1.175:19087/vnc_lite.html + # vncUrl: http://192.168.1.175:19087/vnc_lite.html # 是否在启动的时候自动设置hook服务的回调 autoSetCallback: false # 回调IP,如果是Docker运行,本参数必填(填auto表示自动,不适用于 docker 环境),如果Docker修改了映射,格式为 ip:port @@ -23,17 +23,20 @@ mysql: schema: public # postgres 专用 task: - enable: false + enable: true syncFriends: - enable: false - cron: '*/5 * * * *' # 五分钟一次 + enable: false + cron: '*/5 * * * *' # 五分钟一次 + groupSummary: + enable: true + cron: '30 0 * * *' # 每天0:30 waterGroup: - enable: true - cron: - yesterday: '30 9 * * *' # 每天9:30 - week: '30 9 * * 1' # 每周一9:30 - month: '30 9 1 * *' # 每月1号9:30 - year: '0 9 1 1 *' # 每年1月1号9:30 + enable: false + cron: + yesterday: '30 9 * * *' # 每天9:30 + week: '30 9 * * 1' # 每周一9:30 + month: '30 9 1 * *' # 每月1号9:30 + year: '0 9 1 1 *' # 每年1月1号9:30 # MQ配置 mq: @@ -53,6 +56,8 @@ ai: enable: false # 模型,不填默认gpt-3.5-turbo-0613 model: gpt-3.5-turbo-0613 + # 群聊总结模型 + summaryModel: gpt-4-0613 # OpenAI Api key apiKey: sk-xxxx # 接口代理域名,不填默认ChatGPT官方地址 diff --git a/config/ai.go b/config/ai.go index 1aeff19..fb05374 100644 --- a/config/ai.go +++ b/config/ai.go @@ -3,12 +3,13 @@ package config // ai // @description: AI配置 type ai struct { - Enable bool `json:"enable" yaml:"enable"` // 是否启用AI - Model string `json:"model" yaml:"model"` // 模型 - ApiKey string `json:"apiKey" yaml:"apiKey"` // API Key - BaseUrl string `json:"baseUrl" yaml:"baseUrl"` // API地址 - Personality string `json:"personality" yaml:"personality"` // 人设 - Models []aiModel `json:"models" yaml:"models"` // 模型列表 + Enable bool `json:"enable" yaml:"enable"` // 是否启用AI + Model string `json:"model" yaml:"model"` // 模型 + SummaryModel string `json:"summaryModel" yaml:"summaryModel"` // 总结模型 + ApiKey string `json:"apiKey" yaml:"apiKey"` // API Key + BaseUrl string `json:"baseUrl" yaml:"baseUrl"` // API地址 + Personality string `json:"personality" yaml:"personality"` // 人设 + Models []aiModel `json:"models" yaml:"models"` // 模型列表 } // aiModel diff --git a/config/task.go b/config/task.go index 9db7ed3..ce46e7d 100644 --- a/config/task.go +++ b/config/task.go @@ -3,9 +3,10 @@ package config // task // @description: 定时任务 type task struct { - Enable bool `json:"enable" yaml:"enable"` // 是否启用 - SyncFriends syncFriends `json:"syncFriends" yaml:"syncFriends"` // 同步好友 - WaterGroup waterGroup `json:"waterGroup" yaml:"waterGroup"` // 水群排行榜 + Enable bool `json:"enable" yaml:"enable"` // 是否启用 + SyncFriends syncFriends `json:"syncFriends" yaml:"syncFriends"` // 同步好友 + WaterGroup waterGroup `json:"waterGroup" yaml:"waterGroup"` // 水群排行榜 + GroupSummary syncFriends `json:"groupSummary" yaml:"groupSummary"` // 群聊总结 } // syncFriends diff --git a/entity/friend.go b/entity/friend.go index 4261784..0cb57be 100644 --- a/entity/friend.go +++ b/entity/friend.go @@ -17,6 +17,7 @@ type Friend struct { AiModel string `json:"aiModel"` // AI模型 EnableChatRank bool `json:"enableChatRank" gorm:"type:tinyint(1) default 0 not null"` // 是否使用聊天排行 EnableWelcome bool `json:"enableWelcome" gorm:"type:tinyint(1) default 0 not null"` // 是否启用迎新 + EnableSummary bool `json:"enableSummary" gorm:"type:tinyint(1) default 0 not null"` // 是否启用总结 IsOk bool `json:"isOk" gorm:"type:tinyint(1) default 0 not null"` // 是否正常 } diff --git a/router/router.go b/router/router.go index 5213f82..33848da 100644 --- a/router/router.go +++ b/router/router.go @@ -28,4 +28,5 @@ func Init(g *gin.Engine) { api.PUT("/grouprank/status", app.ChangeEnableGroupRankStatus) // 修改是否开启水群排行榜状态 api.PUT("/grouprank/skip", app.ChangeSkipGroupRankStatus) // 修改是否跳过水群排行榜状态 api.GET("/group/users", app.GetGroupUsers) // 获取群成员列表 + api.PUT("/summary/status", app.ChangeEnableSummaryStatus) // 修改是否开启群聊总结状态 } diff --git a/service/friend.go b/service/friend.go index 3ba8cc2..856120c 100644 --- a/service/friend.go +++ b/service/friend.go @@ -53,6 +53,15 @@ func GetAllEnableChatRank() (records []entity.Friend, err error) { return } +// GetAllEnableSummary +// @description: 取出所有启用了总结的群组 +// @return records +// @return err +func GetAllEnableSummary() (records []entity.Friend, err error) { + err = client.MySQL.Where("enable_summary = ?", 1).Where("wxid LIKE '%@chatroom'").Find(&records).Error + return +} + // CheckIsEnableCommand // @description: 检查用户是否启用了指令 // @param userId diff --git a/service/message.go b/service/message.go index 2e0c9cf..343fd0e 100644 --- a/service/message.go +++ b/service/message.go @@ -3,6 +3,7 @@ package service import ( "go-wechat/client" "go-wechat/entity" + "go-wechat/vo" "log" "os" "strconv" @@ -39,3 +40,22 @@ func SaveMessage(msg entity.Message) { go updateLastActive(msg) } } + +// GetTextMessagesById +// @description: 根据群id或者用户Id获取消息 +// @param id +// @return records +// @return err +func GetTextMessagesById(id string) (records []vo.TextMessageItem, err error) { + tx := client.MySQL. + Table("`t_message` AS tm"). + Joins("LEFT JOIN t_group_user AS tgu ON tm.group_user = tgu.wxid AND tgu.group_id = tm.from_user"). + Select("tgu.nickname", "IF( tm.type = 49, EXTRACTVALUE ( tm.content, \"/msg/appmsg/title\" ), tm.content ) AS message"). + Where("tm.`from_user` = ?", id). + Where(`(tm.type = 1 OR ( tm.type = 49 AND EXTRACTVALUE ( tm.content, "/msg/appmsg/type" ) = '57' ))`). + Where("DATE ( tm.create_at ) = DATE ( CURDATE() - INTERVAL 1 DAY )"). + Order("tm.create_at ASC") + + err = tx.Find(&records).Error + return +} diff --git a/tasks/summary/summary.go b/tasks/summary/summary.go new file mode 100644 index 0000000..0f27f82 --- /dev/null +++ b/tasks/summary/summary.go @@ -0,0 +1,81 @@ +package summary + +import ( + "context" + "fmt" + "github.com/sashabaranov/go-openai" + "go-wechat/config" + "go-wechat/service" + "go-wechat/utils" + "go-wechat/vo" + "log" + "strings" +) + +// AiSummary +// @description: AI总结群聊记录 +func AiSummary() { + groups, err := service.GetAllEnableSummary() + if err != nil { + log.Printf("获取启用了聊天排行榜的群组失败, 错误信息: %v", err) + return + } + + for _, group := range groups { + // 获取对话记录 + var records []vo.TextMessageItem + if records, err = service.GetTextMessagesById(group.Wxid); err != nil { + log.Printf("获取群[%s]对话记录失败, 错误信息: %v", group.Wxid, err) + continue + } + //if len(records) < 100 { + // log.Printf("群[%s]对话记录不足100条,跳过总结", group.Wxid) + // continue + //} + // 组装对话记录为字符串 + var content []string + for _, record := range records { + content = append(content, fmt.Sprintf("%s: %s\n-----end-----", record.Nickname, record.Message)) + } + + msg := fmt.Sprintf("请帮我总结一下一下的群聊内容的梗概(内容尽可能详细一些)。\n"+ + "注意,他们可能是多个话题,请仔细甄别。\n"+ + "每一行代表一个人的发言,每一行的的格式为: \n{nickname}: {content}\n-----end-----"+ + "\n\n聊天记录如下: \n%s", 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 { + log.Printf("群聊记录总结失败: %v", err.Error()) + continue + } + + // 返回消息为空 + if resp.Choices[0].Message.Content == "" { + continue + } + replyMsg := fmt.Sprintf("#昨日消息总结\n\n%s", resp.Choices[0].Message.Content) + utils.SendMessage(group.Wxid, "", replyMsg, 0) + } +} diff --git a/tasks/tasks.go b/tasks/tasks.go index d6ef579..630f643 100644 --- a/tasks/tasks.go +++ b/tasks/tasks.go @@ -4,6 +4,7 @@ import ( "github.com/go-co-op/gocron" "go-wechat/config" "go-wechat/tasks/friends" + "go-wechat/tasks/summary" "go-wechat/tasks/watergroup" "log" "time" @@ -37,6 +38,13 @@ func InitTasks() { } } + // 群聊总结 + if config.Conf.Task.GroupSummary.Enable { + log.Printf("群聊总结任务已启用,执行表达式: %s", config.Conf.Task.GroupSummary.Cron) + _, _ = s.Cron(config.Conf.Task.GroupSummary.Cron).Do(summary.AiSummary) + + } + // 更新好友列表 if config.Conf.Task.SyncFriends.Enable { log.Printf("更新好友列表任务已启用,执行表达式: %s", config.Conf.Task.SyncFriends.Cron) diff --git a/views/index.html b/views/index.html index e0e72f6..06fb6df 100644 --- a/views/index.html +++ b/views/index.html @@ -108,6 +108,7 @@ 是否在通讯录 是否启用AI 是否启用水群排行榜 + 是否启用聊天记录总结 是否启用迎新 是否启用指令 操作 @@ -164,6 +165,15 @@
❌已禁用
+ + + +