🔥 把ai声聊移动到zbputils

Author: guohuiyuan

plugin/aichat/cfg.go

@@ -14,7 +14,9 @@ import (
 	"github.com/wdvxdr1123/ZeroBot/message"
 )
 
-var cfg = newconfig()
+var (
+	cfg = newconfig()
+)
 
 type config struct {
 	ModelName string

plugin/aichat/main.go

@@ -5,12 +5,10 @@ import (
 	"math/rand"
 	"strconv"
 	"strings"
-	"time"
 
 	"github.com/fumiama/deepinfra"
 	"github.com/fumiama/deepinfra/model"
 	"github.com/sirupsen/logrus"
-	"github.com/tidwall/gjson"
 
 	zero "github.com/wdvxdr1123/ZeroBot"
 	"github.com/wdvxdr1123/ZeroBot/message"
@@ -42,10 +40,7 @@ var (
 			"- 设置AI聊天最大长度4096\n" +
 			"- 设置AI聊天TopP 0.9\n" +
 			"- 查看AI聊天配置\n" +
-			"- [启用|禁用]AI语音\n" +
-			"- 设置AI语音群号1048452984	(tips：群里必须有AI声聊应用)\n" +
-			"- 设置AI语音模型\n" +
-			"- 发送AI语音xxx",
+			"- [启用|禁用]AI语音\n",
 		PrivateDataFolder: "aichat",
 	})
 )
@@ -56,8 +51,6 @@ var (
 		"OLLaMA": 1,
 		"GenAI":  2,
 	}
-	customgid = int64(1048452984)
-	modelName = "lucy-voice-xueling"
 )
 
 func init() {
@@ -284,81 +277,4 @@ func init() {
 		Handle(func(ctx *zero.Ctx) {
 			ctx.SendChain(message.Text(printConfig(cfg)))
 		})
-	en.OnPrefix("设置AI语音群号", zero.SuperUserPermission).SetBlock(true).
-		Handle(func(ctx *zero.Ctx) {
-			u := strings.TrimSpace(ctx.State["args"].(string))
-			num, err := strconv.ParseInt(u, 10, 64)
-			if err != nil {
-				ctx.SendChain(message.Text("ERROR: parse gid err: ", err))
-				return
-			}
-			ctx.SendChain(message.Text("设置AI语音群号为", num))
-			customgid = num
-		})
-	en.OnFullMatch("设置AI语音模型", zero.SuperUserPermission).SetBlock(true).
-		Handle(func(ctx *zero.Ctx) {
-			next := zero.NewFutureEvent("message", 999, false, ctx.CheckSession())
-			recv, cancel := next.Repeat()
-			defer cancel()
-			jsonData := ctx.GetAICharacters(customgid, 1)
-
-			// 转换为字符串数组
-			var names []string
-			// 初始化两个映射表
-			nameToID := make(map[string]string)
-			nameToURL := make(map[string]string)
-			characters := jsonData.Get("#.characters")
-
-			// 遍历每个角色对象
-			characters.ForEach(func(_, group gjson.Result) bool {
-				group.ForEach(func(_, character gjson.Result) bool {
-					// 提取当前角色的三个字段
-					name := character.Get("character_name").String()
-					names = append(names, name)
-					// 存入映射表（重复名称会覆盖，保留最后出现的条目）
-					nameToID[name] = character.Get("character_id").String()
-					nameToURL[name] = character.Get("preview_url").String()
-					return true // 继续遍历
-				})
-				return true // 继续遍历
-			})
-			var builder strings.Builder
-			// 写入开头文本
-			builder.WriteString("请选择语音模型序号：\n")
-
-			// 遍历names数组，拼接序号和名称
-			for i, v := range names {
-				// 将数字转换为字符串（不依赖fmt）
-				numStr := strconv.Itoa(i)
-				// 拼接格式："序号. 名称\n"
-				builder.WriteString(numStr)
-				builder.WriteString(". ")
-				builder.WriteString(v)
-				builder.WriteString("\n")
-			}
-			// 获取最终字符串
-			ctx.SendChain(message.Text(builder.String()))
-			for {
-				select {
-				case <-time.After(time.Second * 120):
-					ctx.SendChain(message.Text("设置AI语音模型指令过期"))
-					return
-				case c := <-recv:
-					msg := c.Event.Message.ExtractPlainText()
-					num, err := strconv.Atoi(msg)
-					if err != nil {
-						ctx.SendChain(message.Text("请输入数字!"))
-						continue
-					}
-					if num < 0 || num >= len(names) {
-						ctx.SendChain(message.Text("序号非法!"))
-						continue
-					}
-					modelName = nameToID[names[num]]
-					ctx.SendChain(message.Text("已选择语音模型: ", names[num]))
-					ctx.SendChain(message.Record(nameToURL[names[num]]))
-					return
-				}
-			}
-		})
 }