Merge pull request '🎨 优化AI对话角色提示词' (#40) from hotfix into main

Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/40

entity/friend.go

@@ -15,6 +15,7 @@ type Friend struct {
 	LastActive     time.Time `json:"lastActive"`                                               // 最后活跃时间
 	EnableAi       bool      `json:"enableAI" gorm:"type:tinyint(1) default 0 not null"`       // 是否使用AI
 	AiModel        string    `json:"aiModel"`                                                  // AI模型
+	Prompt         string    `json:"prompt"`                                                   // 提示词
 	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"`  // 是否启用总结

plugin/plugins/ai.go

@@ -46,13 +46,19 @@ func AI(m *plugin.MessageContext) {
 		m.Content = strings.Replace(m.Content, matches[0], "", 1)
 	}
 
+	// 处理预设角色，默认是配置文件里的，如果数据库配置不为空，则使用数据库配置
+	prompt := config.Conf.Ai.Personality
+	if friendInfo.Prompt != "" {
+		prompt = friendInfo.Prompt
+	}
+
 	// 组装消息体
 	messages := make([]openai.ChatCompletionMessage, 0)
 	if config.Conf.Ai.Personality != "" {
 		// 填充人设
 		messages = append(messages, openai.ChatCompletionMessage{
 			Role:    openai.ChatMessageRoleSystem,
-			Content: config.Conf.Ai.Personality,
+			Content: prompt,
 		})
 	}
 

tasks/summary/summary.go

@@ -35,13 +35,18 @@ func AiSummary() {
 		// 组装对话记录为字符串
 		var content []string
 		for _, record := range records {
-			content = append(content, fmt.Sprintf("%s: %s\n-----end-----", record.Nickname, record.Message))
+			content = append(content, fmt.Sprintf(`{"%s": "%s"}--end--`, record.Nickname, strings.ReplaceAll(record.Message, "\n", "。。")))
 		}
 
-		msg := fmt.Sprintf("请帮我总结一下一下的群聊内容的梗概，生成的梗概尽可能详细，并且最好带上名字。\n"+
+		msgTmp := `请帮我总结一下一下的群聊内容的梗概，生成的梗概需要尽可能详细，需要带上一些聊天关键信息，并且带上群友名字。
-			"注意，他们可能是多个话题，请仔细甄别。\n"+
+注意，他们可能是多个话题，请仔细甄别。
-			"每一行代表一个人的发言，每一行的的格式为： \n{nickname}: {content}\n-----end-----"+
+每一行代表一个人的发言，每一行的的格式为： {"{nickname}": "{content}"}--end--
-			"\n\n聊天记录如下: \n%s", strings.Join(content, "\n"))
+
+聊天记录如下: 
+%s
+`
+
+		msg := fmt.Sprintf(msgTmp, strings.Join(content, "\n"))
 
 		// AI总结
 		messages := []openai.ChatCompletionMessage{
@@ -75,7 +80,7 @@ func AiSummary() {
 		if resp.Choices[0].Message.Content == "" {
 			continue
 		}
-		replyMsg := fmt.Sprintf("#昨日消息总结\n\n%s", resp.Choices[0].Message.Content)
+		replyMsg := fmt.Sprintf("#昨日消息总结\n又是一天过去了，让我们一起来看看昨儿群友们都聊了什么有趣的话题吧~\n\n%s", resp.Choices[0].Message.Content)
 		utils.SendMessage(group.Wxid, "", replyMsg, 0)
 	}
 }