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

🔧 新增数据库配置模型

Browse Source
This commit is contained in:
李寻欢 2024-01-22 08:31:38 +08:00
parent 0a421dbb9d
commit 497e17ec62
3 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
entity/friend.go
View File

@ -13,6 +13,7 @@ type Friend struct {
Pinyin string `json:"pinyin"` // 昵称拼音大写首字母 Pinyin string `json:"pinyin"` // 昵称拼音大写首字母
PinyinAll string `json:"pinyinAll"` // 昵称全拼 PinyinAll string `json:"pinyinAll"` // 昵称全拼
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模型
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"` // 是否启用迎新
IsOk bool `json:"isOk" gorm:"type:tinyint(1) default 0 not null"` // 是否正常 IsOk bool `json:"isOk" gorm:"type:tinyint(1) default 0 not null"` // 是否正常

10
plugin/plugins/ai.go
View File

@ -28,9 +28,9 @@ func AI(m *plugin.MessageContext) {
} }
// 取出所有启用了AI的好友或群组 // 取出所有启用了AI的好友或群组
var count int64 var friendInfo entity.Friend
client.MySQL.Model(&entity.Friend{}).Where("enable_ai IS TRUE").Where("wxid = ?", m.FromUser).Count(&count) client.MySQL.Where("wxid = ?", m.FromUser).First(&friendInfo)
if count < 1 { if friendInfo.Wxid == "" {
return return
} }
@ -93,7 +93,9 @@ func AI(m *plugin.MessageContext) {
// 配置模型 // 配置模型
chatModel := openai.GPT3Dot5Turbo0613 chatModel := openai.GPT3Dot5Turbo0613
if config.Conf.Ai.Model != "" { if friendInfo.AiModel != "" {
chatModel = friendInfo.AiModel
} else if config.Conf.Ai.Model != "" {
chatModel = config.Conf.Ai.Model chatModel = config.Conf.Ai.Model
} }

1
vo/friend.go
View File

@ -13,6 +13,7 @@ type FriendItem struct {
PinyinAll string // 昵称全拼 PinyinAll string // 昵称全拼
Wxid string // 微信原始Id Wxid string // 微信原始Id
EnableAi bool // 是否使用AI EnableAi bool // 是否使用AI
AiModel string // AI模型
EnableChatRank bool // 是否使用聊天排行 EnableChatRank bool // 是否使用聊天排行
EnableWelcome bool // 是否使用迎新 EnableWelcome bool // 是否使用迎新
EnableCommand bool // 是否启用指令 EnableCommand bool // 是否启用指令