loks666/go-wxhelper
1
0
Fork 0
forked from lxh/go-wxhelper
Code Pull Requests Activity

Merge pull request '🐛 Fix a bug.' (#60) from hotfix into main

Browse Source 
Reviewed-on: lxh/go-wxhelper#60
This commit is contained in:
李寻欢 2024-06-18 17:30:27 +08:00
commit 44b8e4a162
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
plugin/plugins/ai.go
View File

@ -48,8 +48,23 @@ func AI(m *plugin.MessageContext) {
// 处理预设角色，默认是配置文件里的，如果数据库配置不为空，则使用数据库配置
prompt := config.Conf.Ai.Personality
var dbPrompt entity.AiAssistant
if friendInfo.Prompt != "" {
prompt = friendInfo.Prompt
// 取出配置的角色
client.MySQL.First(&dbPrompt, "id = ?", friendInfo.Prompt)
if dbPrompt.Id != "" {
prompt = dbPrompt.Personality
}
}
// 配置模型
chatModel := openai.GPT3Dot5Turbo0613
if friendInfo.AiModel != "" {
chatModel = friendInfo.AiModel
} else if dbPrompt.Model != "" {
chatModel = dbPrompt.Model
} else if config.Conf.Ai.Model != "" {
chatModel = config.Conf.Ai.Model
}
// 组装消息体
@ -101,14 +116,6 @@ func AI(m *plugin.MessageContext) {
Content: m.Content,
})
// 配置模型
chatModel := openai.GPT3Dot5Turbo0613
if friendInfo.AiModel != "" {
chatModel = friendInfo.AiModel
} else if config.Conf.Ai.Model != "" {
chatModel = config.Conf.Ai.Model
}
// 默认使用AI回复
conf := openai.DefaultConfig(config.Conf.Ai.ApiKey)
if config.Conf.Ai.BaseUrl != "" {