1
0
forked from lxh/go-wxhelper

🎨 优化AI对话角色设置,支持为群或者好友单独设置角色(需手动修改数据库)

This commit is contained in:
李寻欢 2024-04-16 17:19:08 +08:00
parent bb92c67069
commit daa36f308b
2 changed files with 8 additions and 1 deletions

View File

@ -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"` // 是否启用总结

View File

@ -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,
})
}