From d43ea7df1d55bd747ec5c26bb76a03680429e8ae Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=96=B9=E6=9F=B3=E7=85=9C?= <101934327+fangliuyu@users.noreply.github.com> Date: Mon, 24 Oct 2022 23:36:28 +0800 Subject: [PATCH] =?UTF-8?q?=E6=89=A9=E5=A2=9E=E6=96=87=E5=BF=83AI=E7=9A=84?= =?UTF-8?q?=E5=8A=9F=E8=83=BD=EF=BC=8C=E4=BF=AE=E5=A4=8D=E7=8C=9C=E6=AD=8C?= =?UTF-8?q?=E7=9A=84=E6=9D=83=E9=99=90=E9=97=AE=E9=A2=98=20(#471)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 40 +- go.mod | 2 +- go.sum | 4 +- plugin/guessmusic/main.go | 2 +- plugin/wenxinAI/{ernievilg.go => wenxinAI.go} | 347 ++++++++++++++++-- 5 files changed, 352 insertions(+), 43 deletions(-) rename plugin/wenxinAI/{ernievilg.go => wenxinAI.go} (50%) diff --git a/README.md b/README.md index ef04b758..697bc4c8 100644 --- a/README.md +++ b/README.md @@ -1137,21 +1137,45 @@ print("run[CQ:image,file="+j["img"]+"]")
- 百度文心AI画图 + 百度文心AI `import _ "github.com/FloatTech/ZeroBot-Plugin/plugin/wenxinAI"` - - 基于百度文心的免费AI画图插件(因为是免费的,图片质量你懂的) + 基于百度文心API的一些功能 - - key申请链接：https://wenxin.baidu.com/moduleApi/key - - - [x] 为[自己/本群/QQ号/群+群号]设置AI画图key [API Key] [Secret Key] - - - 例：“为10086设置AI画图key 123 456”；“为群10010设置AI画图key 789 101” + key申请链接：https://wenxin.baidu.com/moduleApi/key + + - [x] 为[自己/本群/QQ号/群+群号]设置文心key [API Key] [Secret Key] + + - [x] 为[自己/本群/QQ号/群+群号]设置画图key [API Key] [Secret Key] + + 例：“为10086设置画图key 123 456”；“为群10010设置画图key 789 101” + + 文心key和画图key的API key 可以是相同的，只是文心key日限为200，画图日限为50，以此作区别。 + + - [x] 文心作文 (x字的)[作文题目] + + - [x] 文心提案 (x字的)[文案标题] + + - [x] 文心摘要 (x字的)[文章内容] + + - [x] 文心小说 (x字的)[小说上文] + + - [x] 文心对联 [上联] + + - [x] 文心问答 [问题] + + - [x] 文心补全 [带“_”的填空题] + + - [x] 文心自定义 [prompt] - [x] [bot名称]画几张[图片描述]的[图片类型][图片尺寸] - - 指令示例： + 指令示例： + + - 文心作文 我的椛椛机器人 + + - 文心作文 300字的我的椛椛机器人 - 椛椛帮我画几张金凤凰，背景绚烂，高饱和，古风，仙境，高清，4K，古风的油画方图 diff --git a/go.mod b/go.mod index 2a43dd0f..72863f00 100644 --- a/go.mod +++ b/go.mod @@ -4,7 +4,7 @@ go 1.19 require ( github.com/Coloured-glaze/gg v1.3.4 - github.com/FloatTech/AnimeAPI v1.5.2-0.20221015060924-fe2f85a3cf45 + github.com/FloatTech/AnimeAPI v1.5.2-0.20221023084913-bd1ff35e91ed github.com/FloatTech/floatbox v0.0.0-20221011153549-68005767c531 github.com/FloatTech/sqlite v0.5.0 github.com/FloatTech/ttl v0.0.0-20220715042055-15612be72f5b diff --git a/go.sum b/go.sum index 3cdc1e84..0862375e 100644 --- a/go.sum +++ b/go.sum @@ -2,8 +2,8 @@ github.com/BurntSushi/xgb v0.0.0-20160522181843-27f122750802/go.mod h1:IVnqGOEym github.com/Coloured-glaze/gg v1.3.4 h1:l31zIF/HaVwkzjrj+A56RGQoSKyKuR1IWtIrqXGFStI= github.com/Coloured-glaze/gg v1.3.4/go.mod h1:Ih5NLNNDHOy3RJbB0EPqGTreIzq/H02TGThIagh8HJg= github.com/DATA-DOG/go-sqlmock v1.3.3/go.mod h1:f/Ixk793poVmq4qj/V1dPUg2JEAKC73Q5eFN3EC/SaM= -github.com/FloatTech/AnimeAPI v1.5.2-0.20221015060924-fe2f85a3cf45 h1:XbNlD0irJELgdR304TvqmFrxdH7hsxA/Ah9xLHFP3eQ= -github.com/FloatTech/AnimeAPI v1.5.2-0.20221015060924-fe2f85a3cf45/go.mod h1:RgcMDA1S7C81bq7HQjygMEEo+EXwAlsutvKMv7DafgY= +github.com/FloatTech/AnimeAPI v1.5.2-0.20221023084913-bd1ff35e91ed h1:Jve+3koem2QAFWoB9c2qsDmhOdLQOAFfbfzG7y47tYQ= +github.com/FloatTech/AnimeAPI v1.5.2-0.20221023084913-bd1ff35e91ed/go.mod h1:RgcMDA1S7C81bq7HQjygMEEo+EXwAlsutvKMv7DafgY= github.com/FloatTech/floatbox v0.0.0-20221011153549-68005767c531 h1:Z0yn6LFhEyC12hj+TBXc2P7/kWlCd/jlwv4JFndgpnw= github.com/FloatTech/floatbox v0.0.0-20221011153549-68005767c531/go.mod h1:4tfIeB74L1RzhNp3nNjaTw8m3IEnc+q/k6k/MhL07ks= github.com/FloatTech/sqlite v0.5.0 h1:U7J5Omc534PqmH6csfu+ypCo3DS8L91l5lTsxUu3b/U= diff --git a/plugin/guessmusic/main.go b/plugin/guessmusic/main.go index f1b7999a..2f7b4ee7 100644 --- a/plugin/guessmusic/main.go +++ b/plugin/guessmusic/main.go @@ -165,7 +165,7 @@ func init() { // 插件主体 ctx.SendChain(message.Text(serviceErr, "ERROR:\n", err)) } }) - engine.OnRegex(`^猜歌(开启|关闭)(歌单|歌词)自动下载`).SetBlock(true). + engine.OnRegex(`^猜歌(开启|关闭)(歌单|歌词)自动下载`, zero.SuperUserPermission).SetBlock(true). Handle(func(ctx *zero.Ctx) { swtich := ctx.State["regex_matched"].([]string)[1] option := ctx.State["regex_matched"].([]string)[1] diff --git a/plugin/wenxinAI/ernievilg.go b/plugin/wenxinAI/wenxinAI.go similarity index 50% rename from plugin/wenxinAI/ernievilg.go rename to plugin/wenxinAI/wenxinAI.go index 06c57241..4e16699f 100644 --- a/plugin/wenxinAI/ernievilg.go +++ b/plugin/wenxinAI/wenxinAI.go @@ -1,5 +1,5 @@ -// Package ernie AI画图 -package ernie +// Package wenxin 百度文心AI +package wenxin import ( "errors" @@ -19,13 +19,17 @@ import ( // 数据库 sql "github.com/FloatTech/sqlite" + // 百度文心大模型 + model "github.com/FloatTech/AnimeAPI/wenxinAI/erniemodle" // 百度文心AI画图API wenxin "github.com/FloatTech/AnimeAPI/wenxinAI/ernievilg" ) const ( - serviceName = "AIdraw" + serviceName = "wenxinvilg" serviceErr = "[" + serviceName + "]ERROR:\n" + modelName = "wenxinmodel" + modelErr = "[" + modelName + "]ERROR:\n" ) type keydb struct { @@ -46,24 +50,34 @@ type apikey struct { } var ( - groupinfo = &keydb{ + name = "椛椛" + limit int + vilginfo = &keydb{ + db: &sql.Sqlite{}, + } + modelinfo = &keydb{ db: &sql.Sqlite{}, } - limit = 50 dtype = [...]string{ "古风", "油画", "水彩画", "卡通画", "二次元", "浮世绘", "蒸汽波艺术", "low poly", "像素风格", "概念艺术", "未来主义", "赛博朋克", "写实风格", "洛丽塔风格", "巴洛克风格", "超现实主义", } ) func init() { // 插件主体 + go func() { + process.GlobalInitMutex.Lock() + defer process.GlobalInitMutex.Unlock() + name = zero.BotConfig.NickName[0] + }() engine := control.Register(serviceName, &ctrl.Options[*zero.Ctx]{ DisableOnDefault: false, - Help: "AI画图\n" + + Help: "文心AI画图\n" + "基于百度文心的免费AI画图插件,\n因为是免费的,图片质量你懂的。\n" + - "key申请链接：https://wenxin.baidu.com/moduleApi/key\n" + - "注意：每个apikey每日上限50次,总上限500次请求。次数超过了请自行更新apikey\n" + - "- 为[自己/本群/QQ号/群+群号]设置AI画图key [API Key] [Secret Key]\n" + - "例：\n[为10086设置AI画图key 123 456]\n[为群10010设置AI画图key 789 101]\n" + + "key申请链接:https://wenxin.baidu.com/moduleApi/key\n" + + "key和erniemodel插件的key相同。\n" + + "注意:每个apikey每日上限50次,总上限500次请求。次数超过了请自行更新apikey\n" + + "- 为[自己/本群/QQ号/群+群号]设置画图key [API Key] [Secret Key]\n" + + "例：\n为自己设置画图key 123 456\n为10086设置画图key 123 456\n为群10010设置画图key 789 101\n" + "- [bot名称]画几张[图片描述]的[图片类型][图片尺寸]\n" + "————————————————————\n" + "图片描述指南:\n图片主体，细节词(请用逗号连接)\n官方prompt指南:https://wenxin.baidu.com/wenxin/docs#Ol7ece95m\n" + @@ -73,8 +87,8 @@ func init() { // 插件主体 "图片尺寸当前只支持：方图/长图/横图\n" + "————————————————————\n" + "指令示例：\n" + - "椛椛帮我画几张金凤凰，背景绚烂，高饱和，古风，仙境，高清，4K，古风的油画方图", - PrivateDataFolder: "ernievilg", + name + "帮我画几张金凤凰，背景绚烂，高饱和，古风，仙境，高清，4K，古风的油画方图", + PrivateDataFolder: "wenxinAI", }).ApplySingle(single.New( single.WithKeyFn(func(ctx *zero.Ctx) int64 { return ctx.Event.GroupID }), single.WithPostFn[int64](func(ctx *zero.Ctx) { @@ -87,8 +101,8 @@ func init() { // 插件主体 }), )) getdb := fcext.DoOnceOnSuccess(func(ctx *zero.Ctx) bool { - groupinfo.db.DBPath = engine.DataFolder() + "keydb.db" - err := groupinfo.db.Open(time.Hour * 24) + vilginfo.db.DBPath = engine.DataFolder() + "ernieVilg.db" + err := vilginfo.db.Open(time.Hour * 24) if err != nil { ctx.SendChain(message.Text(serviceErr, err)) return false @@ -101,8 +115,8 @@ func init() { // 插件主体 uid := -ctx.Event.UserID gid := ctx.Event.GroupID // 获取个人和群的key - userinfo, err1 := groupinfo.checkGroup(uid) - info, err2 := groupinfo.checkGroup(gid) + userinfo, err1 := vilginfo.checkGroup(uid, "vilg") + info, err2 := vilginfo.checkGroup(gid, "vilg") switch { // 如果是个人请求且报错 case gid == 0 && err1 != nil: @@ -189,7 +203,8 @@ func init() { // 插件主体 return } if status == "0" { - msg := message.Message{ctxext.FakeSenderForwardNode(ctx, message.Text("我画好了！"))} + lastTime := time.Duration(i * 10 * int(time.Second)) + msg := message.Message{ctxext.FakeSenderForwardNode(ctx, message.Text("我画好了！\n本次绘画用了", lastTime))} for _, imginfo := range picURL { msg = append(msg, ctxext.FakeSenderForwardNode(ctx, @@ -201,14 +216,14 @@ func init() { // 插件主体 break } } - err = groupinfo.update(gid) + err = vilginfo.update(gid, 1) if err != nil { ctx.SendChain(message.Text(serviceErr, err)) } process.SleepAbout1sTo2s() ctx.SendChain(message.Text("累死了，今天我最多只能画", info.DayLimit-1, "张图哦")) }) - engine.OnRegex(`^为(群)?(自己|本群|\d+)设置AI画图key\s(.*[^\s$])\s(.+)$`, getdb).SetBlock(true). + engine.OnRegex(`^为(群)?(自己|本群|\d+)设置画图key\s(.*[^\s$])\s(.+)$`, getdb).SetBlock(true). Handle(func(ctx *zero.Ctx) { mode := ctx.State["regex_matched"].([]string)[1] user := ctx.State["regex_matched"].([]string)[2] @@ -238,17 +253,267 @@ func init() { // 插件主体 } dbID = -uid } - err := groupinfo.insert(dbID, aKey, sKey) + err := vilginfo.insert(dbID, "vilg", aKey, sKey) if err != nil { ctx.SendChain(message.Text(serviceErr, err)) return } ctx.SendChain(message.Text("成功!")) }) + /*********************************************************/ + en := control.Register(modelName, &ctrl.Options[*zero.Ctx]{ + DisableOnDefault: false, + Help: "文心AI文本处理\n" + + "基于百度文心AI的API文本处理\n" + + "key申请链接:https://wenxin.baidu.com/moduleApi/key\n" + + "key和ernievilg插件的key相同。\n" + + "注意:每个apikey每日上限200条,总上限2000条。次数超过了请自行更新apikey\n" + + "- 为[自己/本群/QQ号/群+群号]设置文心key [API Key] [Secret Key]\n" + + "例：\n为自己设置文心key 123 456\n为10086设置文心key 123 456\n为群10010设置文心key 789 101\n" + + "————————————————————\n" + + "- 文心作文 (x字的)[作文题目]\n" + + "————————————————————\n" + + "- 文心提案 (x字的)[文案标题]\n" + + "————————————————————\n" + + "- 文心摘要 (x字的)[文章内容]\n" + + "————————————————————\n" + + "- 文心小说 (x字的)[小说上文]" + + "————————————————————\n" + + "- 文心对联 [上联]\n" + + "————————————————————\n" + + "- 文心问答 [问题]" + + "————————————————————\n" + + "- 文心补全 [带“_”的填空题]" + + "————————————————————\n" + + "- 文心自定义 [prompt]\n" + + "prompt: [问题描述] [问题类型]:[题目] [解答类型]:[解题必带内容]\n" + + "指令示例:\n" + + "文心自定义 请写出下面这道题的解题过程。 题目:养殖场养鸭376只,养鸡的只数比鸭多258只,这个养殖场一共养鸭和鸡多少只？ 解：\n" + + "文心自定义 1+1=?\n" + + "文心自定义 歌曲名：大风车转啊转 歌词：", + }).ApplySingle(single.New( + single.WithKeyFn(func(ctx *zero.Ctx) int64 { return ctx.Event.GroupID }), + single.WithPostFn[int64](func(ctx *zero.Ctx) { + ctx.Break() + ctx.Send( + message.ReplyWithMessage(ctx.Event.MessageID, + message.Text(zero.BotConfig.NickName[0], "正在给别人编辑，请不要打扰哦"), + ), + ) + }), + )) + getmodeldb := fcext.DoOnceOnSuccess(func(ctx *zero.Ctx) bool { + modelinfo.db.DBPath = engine.DataFolder() + "ernieModel.db" + err := modelinfo.db.Open(time.Hour * 24) + if err != nil { + ctx.SendChain(message.Text(modelErr, err)) + return false + } + return true + }) + en.OnRegex(`^为(群)?(自己|本群|\d+)设置文心key\s(.*[^\s$])\s(.+)$`, getmodeldb).SetBlock(true). + Handle(func(ctx *zero.Ctx) { + mode := ctx.State["regex_matched"].([]string)[1] + user := ctx.State["regex_matched"].([]string)[2] + aKey := ctx.State["regex_matched"].([]string)[3] + sKey := ctx.State["regex_matched"].([]string)[4] + dbID := -ctx.Event.UserID // 默认给自己 + switch { + case mode != "": // 指定群的话 + gid, err := strconv.ParseInt(user, 10, 64) + if err != nil { + ctx.SendChain(message.Text(modelErr, err)) + return + } + dbID = gid + case user == "本群": // 用于本群 + gid := ctx.Event.GroupID + if gid == 0 { + ctx.SendChain(message.Text(modelErr, "请指定群聊，或者使用指令；\n为群xxx设置AI画图key xxx xxx")) + return + } + dbID = gid + case user != "自己": // 给别人开key + uid, err := strconv.ParseInt(user, 10, 64) + if err != nil { + ctx.SendChain(message.Text(modelErr, err)) + return + } + dbID = -uid + } + err := modelinfo.insert(dbID, "model", aKey, sKey) + if err != nil { + ctx.SendChain(message.Text(modelErr, err)) + return + } + ctx.SendChain(message.Text("成功!")) + }) + + var erniemodel = map[string]int{ + "作文": 1, + "提案": 2, + "摘要": 3, + "对联": 4, + "问答": 5, + "小说": 6, + "补全": 7, + "自定义": 8} + var erniePrompt = map[string]string{ + "作文": "zuowen", + "提案": "adtext", + "摘要": "Summarization", + "对联": "couplet", + "问答": "Dialogue", + "小说": "novel", + "补全": "cloze"} + en.OnRegex(`^文心(作文|提案|摘要|小说)\s?((\d+)字的)?(.*)$`, getmodeldb).SetBlock(true). + Handle(func(ctx *zero.Ctx) { + uid := -ctx.Event.UserID + gid := ctx.Event.GroupID + // 获取个人和群的key + userinfo, err1 := modelinfo.checkGroup(uid, "model") + info, err2 := modelinfo.checkGroup(gid, "model") + switch { + // 如果是个人请求且报错 + case gid == 0 && err1 != nil: + ctx.SendChain(message.Text(modelErr, err1)) + return + // 如果群报错而个人没有,就切换成个人的 + case err2 != nil && err1 == nil: + gid = uid + info = userinfo + // 如果都报错就以群为优先级 + case err1 != nil && err2 != nil: + ctx.SendChain(message.Text(modelErr, err2)) + return + } + // 判断使用次数 + check := false + switch { + // 群和个人都没有次数了 + case info.DayLimit == 0 && userinfo.DayLimit == 0: + ctx.SendChain(message.Text("今日请求次数已到200次了,明天在玩吧")) + return + // 个人还有次数的话 + case info.DayLimit == 0 && userinfo.DayLimit != 0: + check = true + } + switch { + // 群和个人都没有总次数了 + case info.MaxLimit == 0 && userinfo.MaxLimit == 0: + ctx.SendChain(message.Text("设置的key使用次数超过了限额,请更换key。")) + return + // 个人还有总次数的话 + case info.MaxLimit == 0 && userinfo.MaxLimit != 0: + check = true + } + if check { // 如果只有个人有次数就切换回个人key + gid = uid + info = userinfo + } + // 调用API + modelStr := ctx.State["regex_matched"].([]string)[1] + mun := ctx.State["regex_matched"].([]string)[3] + minlen := 1 + maxlen := 128 + if mun != "" { + max, err := strconv.Atoi(mun) + if err != nil { + ctx.SendChain(message.Text(modelErr, err)) + return + } + minlen = max + if max > 128 { + maxlen = max + } + } + keyword := ctx.State["regex_matched"].([]string)[4] + if len([]rune(keyword)) >= 1000 { // 描述不能超过1000 + ctx.SendChain(message.Text("是你写作文还是我写？减少点！")) + return + } + result, err := model.GetResult(info.Token, erniemodel[modelStr], keyword, minlen, maxlen, erniePrompt[modelStr]) + if err != nil { + ctx.SendChain(message.Text(modelErr, err)) + return + } + if id := ctx.Send(message.ReplyWithMessage(ctx.Event.MessageID, message.Text(keyword, "，", result))); id.ID() == 0 { + ctx.SendChain(message.Text("ERROR: 请求超时!")) + } + err = modelinfo.update(gid, 1) + if err != nil { + ctx.SendChain(message.Text(modelErr, err)) + } + }) + en.OnRegex(`^文心(对联|问答|补全|自定义)\s?(.*)$`, getmodeldb).SetBlock(true). + Handle(func(ctx *zero.Ctx) { + uid := -ctx.Event.UserID + gid := ctx.Event.GroupID + // 获取个人和群的key + userinfo, err1 := modelinfo.checkGroup(uid, "model") + info, err2 := modelinfo.checkGroup(gid, "model") + switch { + // 如果是个人请求且报错 + case gid == 0 && err1 != nil: + ctx.SendChain(message.Text(modelErr, err1)) + return + // 如果群报错而个人没有,就切换成个人的 + case err2 != nil && err1 == nil: + gid = uid + info = userinfo + // 如果都报错就以群为优先级 + case err1 != nil && err2 != nil: + ctx.SendChain(message.Text(modelErr, err2)) + return + } + // 判断使用次数 + check := false + switch { + // 群和个人都没有次数了 + case info.DayLimit == 0 && userinfo.DayLimit == 0: + ctx.SendChain(message.Text("今日请求次数已到200次了,明天在玩吧")) + return + // 个人还有次数的话 + case info.DayLimit == 0 && userinfo.DayLimit != 0: + check = true + } + switch { + // 群和个人都没有总次数了 + case info.MaxLimit == 0 && userinfo.MaxLimit == 0: + ctx.SendChain(message.Text("设置的key使用次数超过了限额,请更换key。")) + return + // 个人还有总次数的话 + case info.MaxLimit == 0 && userinfo.MaxLimit != 0: + check = true + } + if check { // 如果只有个人有次数就切换回个人key + gid = uid + info = userinfo + } + // 创建任务 + modelStr := ctx.State["regex_matched"].([]string)[1] + keyword := ctx.State["regex_matched"].([]string)[2] + if len([]rune(keyword)) >= 1000 { // 描述不能超过1000 + ctx.SendChain(message.Text("你在写作文吗？减少点！")) + return + } + result, err := model.GetResult(info.Token, erniemodel[modelStr], keyword, 1, 128, erniePrompt[modelStr]) + if err != nil { + ctx.SendChain(message.Text(modelErr, err)) + return + } + if id := ctx.Send(message.ReplyWithMessage(ctx.Event.MessageID, message.Text(result))); id.ID() == 0 { + ctx.SendChain(message.Text("ERROR: 请求超时!")) + } + err = modelinfo.update(gid, 1) + if err != nil { + ctx.SendChain(message.Text(modelErr, err)) + } + }) } // 登记group的key -func (sql *keydb) insert(gid int64, akey, skey string) error { +func (sql *keydb) insert(gid int64, model, akey, skey string) error { sql.Lock() defer sql.Unlock() // 给db文件创建表格(没有才创建)，表格名称groupinfo，表格结构apikey @@ -270,7 +535,12 @@ func (sql *keydb) insert(gid int64, akey, skey string) error { ID: gid, APIKey: akey, SecretKey: skey, - MaxLimit: 500, + } + switch model { + case "vilg": + groupinfo.MaxLimit = 500 + case "model": + groupinfo.MaxLimit = 2000 } } return sql.db.Insert("groupinfo", &groupinfo) @@ -278,12 +548,19 @@ func (sql *keydb) insert(gid int64, akey, skey string) error { // 进行更新 groupinfo.APIKey = akey groupinfo.SecretKey = skey - groupinfo.MaxLimit = 500 + groupinfo.Token = "" + groupinfo.Updatetime = 0 + switch model { + case "vilg": + groupinfo.MaxLimit = 500 + case "model": + groupinfo.MaxLimit = 2000 + } return sql.db.Insert("groupinfo", &groupinfo) } // 获取group信息 -func (sql *keydb) checkGroup(gid int64) (groupinfo apikey, err error) { +func (sql *keydb) checkGroup(gid int64, model string) (groupinfo apikey, err error) { sql.Lock() defer sql.Unlock() // 给db文件创建表格(没有才创建)，表格名称groupinfo，表格结构apikey @@ -291,12 +568,20 @@ func (sql *keydb) checkGroup(gid int64) (groupinfo apikey, err error) { if err != nil { return } + switch model { + case "vilg": + limit = 50 + model = "画图" + case "model": + limit = 200 + model = "文心" + } // 先判断该群是否已经设置过key了 if ok := sql.db.CanFind("groupinfo", "where ID is "+strconv.FormatInt(gid, 10)); !ok { if gid > 0 { - err = errors.New("该群没有设置过apikey，请前往https://wenxin.baidu.com/moduleApi/key获取key值后，发送指令:\n为本群设置AI画图key [API Key] [Secret Key]") + err = errors.New("该群没有设置过apikey，请前往https://wenxin.baidu.com/moduleApi/key获取key值后，发送指令:\n为本群设置" + model + "key [API Key] [Secret Key]\n或\n为自己设置" + model + "key [API Key] [Secret Key]") } else { - err = errors.New("你没有设置过apikey，请前往https://wenxin.baidu.com/moduleApi/key获取key值后，发送指令:\n为自己设置AI画图key [API Key] [Secret Key]") + err = errors.New("你没有设置过apikey，请前往https://wenxin.baidu.com/moduleApi/key获取key值后，发送指令:\n为自己设置" + model + "key [API Key] [Secret Key]") } return } @@ -314,7 +599,7 @@ func (sql *keydb) checkGroup(gid int64) (groupinfo apikey, err error) { return } // 如果token有效期过期 - if time.Since(time.Unix(groupinfo.Updatetime, 0)).Hours() > 24 { + if time.Since(time.Unix(groupinfo.Updatetime, 0)).Hours() > 24 || groupinfo.Token == "" { token, err1 := wenxin.GetToken(groupinfo.APIKey, groupinfo.SecretKey) if err1 != nil { err = err1 @@ -350,8 +635,8 @@ func (sql *keydb) checkGroup(gid int64) (groupinfo apikey, err error) { return } -// 记录次数(-1) -func (sql *keydb) update(gid int64) error { +// 记录次数(-sub) +func (sql *keydb) update(gid int64, sub int) error { sql.Lock() defer sql.Unlock() // 给db文件创建表格(没有才创建)，表格名称groupinfo，表格结构apikey @@ -365,8 +650,8 @@ func (sql *keydb) update(gid int64) error { if err != nil { return err } - groupinfo.MaxLimit-- - groupinfo.DayLimit-- + groupinfo.MaxLimit -= sub + groupinfo.DayLimit -= sub err = sql.db.Insert("groupinfo", &groupinfo) if err != nil { return err