Compare commits

..

No commits in common. "21a6f75ed34452385b7da00d76ca0a0ef14c1b89" and "e78f6a6d6e8f7ab00abff55404ab53d1e4cb8f74" have entirely different histories.

2 changed files with 1 additions and 8 deletions

View File

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

View File

@ -46,19 +46,13 @@ func AI(m *plugin.MessageContext) {
m.Content = strings.Replace(m.Content, matches[0], "", 1) 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) messages := make([]openai.ChatCompletionMessage, 0)
if config.Conf.Ai.Personality != "" { if config.Conf.Ai.Personality != "" {
// 填充人设 // 填充人设
messages = append(messages, openai.ChatCompletionMessage{ messages = append(messages, openai.ChatCompletionMessage{
Role: openai.ChatMessageRoleSystem, Role: openai.ChatMessageRoleSystem,
Content: prompt, Content: config.Conf.Ai.Personality,
}) })
} }