✨ 添加设置AI语音模型

Author: guohuiyuan

plugin/aichat/cfg.go

@@ -1,6 +1,7 @@
 package aichat
 
 import (
+	"fmt"
 	"strconv"
 	"strings"
 
@@ -151,3 +152,33 @@ func newextrasetfloat32(ptr *float32) func(ctx *zero.Ctx) {
 		ctx.SendChain(message.Text("成功"))
 	}
 }
+
+func printConfig(cfg config) string {
+	var builder strings.Builder
+	builder.WriteString("当前AI聊天配置：\n")
+	builder.WriteString(fmt.Sprintf("• 模型名：%s\n", cfg.ModelName))
+	builder.WriteString(fmt.Sprintf("• 触发概率：%d%%\n", cfg.Type))
+	builder.WriteString(fmt.Sprintf("• 最大生成长度：%d\n", cfg.MaxN))
+	builder.WriteString(fmt.Sprintf("• TopP采样值：%.1f\n", cfg.TopP))
+	builder.WriteString(fmt.Sprintf("• 系统提示词：%s\n", cfg.SystemP))
+	builder.WriteString(fmt.Sprintf("• 接口地址：%s\n", cfg.API))
+	builder.WriteString(fmt.Sprintf("• 密钥：%s\n", maskKey(cfg.Key)))
+	builder.WriteString(fmt.Sprintf("• 分隔符：%s\n", cfg.Separator))
+	builder.WriteString(fmt.Sprintf("• 响应@：%s\n", yesNo(!cfg.NoReplyAT)))
+	builder.WriteString(fmt.Sprintf("• 支持系统提示词：%s\n", yesNo(!cfg.NoSystemP)))
+	return builder.String()
+}
+
+func maskKey(key string) string {
+	if len(key) <= 4 {
+		return "****"
+	}
+	return key[:2] + strings.Repeat("*", len(key)-4) + key[len(key)-2:]
+}
+
+func yesNo(b bool) string {
+	if b {
+		return "是"
+	}
+	return "否"
+}

plugin/aichat/main.go

@@ -5,6 +5,7 @@ import (
 	"math/rand"
 	"strconv"
 	"strings"
+	"time"
 
 	"github.com/fumiama/deepinfra"
 	"github.com/fumiama/deepinfra/model"
@@ -40,8 +41,9 @@ var (
 			"- 设置AI聊天(不)响应AT\n" +
 			"- 设置AI聊天最大长度4096\n" +
 			"- 设置AI聊天TopP 0.9\n" +
-			"- 设置AI语音群号1048452984	(tips：群里必须有AI声聊应用)\n" +
+			"- 查看AI聊天配置\n" +
 			"- [启用|禁用]AI语音\n" +
+			"- 设置AI语音群号1048452984	(tips：群里必须有AI声聊应用)\n" +
 			"- 设置AI语音模型\n" +
 			"- 发送AI语音xxx",
 		PrivateDataFolder: "aichat",
@@ -278,6 +280,10 @@ func init() {
 		Handle(newextrasetuint(&cfg.MaxN))
 	en.OnPrefix("设置AI聊天TopP", ensureconfig, zero.OnlyPrivate, zero.SuperUserPermission).SetBlock(true).
 		Handle(newextrasetfloat32(&cfg.TopP))
+	en.OnFullMatch("查看AI聊天配置", ensureconfig, zero.SuperUserPermission).SetBlock(true).
+		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))
@@ -291,36 +297,39 @@ func init() {
 		})
 	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 := jsonData.Get("#.characters")
 
 			// 遍历每个角色对象
-			characters.ForEach(func(_, character gjson.Result) bool {
-				// 提取当前角色的三个字段
-				name := character.Get("character_name").String()
-				names = append(names, name)
-				id := character.Get("character_id").String()
-				url := character.Get("preview_url").String()
-
-				// 存入映射表（重复名称会覆盖，保留最后出现的条目）
-				nameToID[name] = id
-				nameToURL[name] = url
+			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")
+			builder.WriteString("请选择语音模型序号：\n")
 
 			// 遍历names数组，拼接序号和名称
 			for i, v := range names {
 				// 将数字转换为字符串（不依赖fmt）
-				numStr := strconv.Itoa(i + 1)
+				numStr := strconv.Itoa(i)
 				// 拼接格式："序号. 名称\n"
 				builder.WriteString(numStr)
 				builder.WriteString(". ")
@@ -329,5 +338,27 @@ func init() {
 			}
 			// 获取最终字符串
 			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
+				}
+			}
 		})
 }