✨ 添加以语音输出

Author: guohuiyuan

plugin/aichat/cfg.go

@@ -29,6 +29,7 @@ type config struct {
 	Separator string
 	NoReplyAT bool
 	NoSystemP bool
+	NoRecord  bool
 }
 
 func newconfig() config {
@@ -155,19 +156,30 @@ func newextrasetfloat32(ptr *float32) func(ctx *zero.Ctx) {
 	}
 }
 
-func printConfig(cfg config) string {
+func printConfig(rate int64, temperature int64, cfg config) string {
+	maxn := cfg.MaxN
+	if maxn == 0 {
+		maxn = 4096
+	}
+	topp := cfg.TopP
+	if topp == 0 {
+		topp = 0.9
+	}
 	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("• 接口类型：%d(%s)\n", cfg.Type, apilist[cfg.Type]))
+	builder.WriteString(fmt.Sprintf("• 触发概率：%d%%\n", rate))
+	builder.WriteString(fmt.Sprintf("• 温度：%.2f\n", float32(temperature)/100))
+	builder.WriteString(fmt.Sprintf("• 最大长度：%d\n", maxn))
+	builder.WriteString(fmt.Sprintf("• TopP：%.1f\n", 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)))
+	builder.WriteString(fmt.Sprintf("• 以AI语音输出：%s\n", yesNo(!cfg.NoRecord)))
 	return builder.String()
 }
 

plugin/aichat/main.go

@@ -2,7 +2,6 @@
 package aichat
 
 import (
-	"fmt"
 	"math/rand"
 	"strconv"
 	"strings"
@@ -41,8 +40,8 @@ var (
 			"- 设置AI聊天(不)响应AT\n" +
 			"- 设置AI聊天最大长度4096\n" +
 			"- 设置AI聊天TopP 0.9\n" +
-			"- 查看AI聊天配置\n" +
-			"- [启用|禁用]AI语音\n",
+			"- 设置AI聊天(不)以AI语音输出\n" +
+			"- 查看AI聊天配置\n",
 		PrivateDataFolder: "aichat",
 	})
 )
@@ -53,6 +52,7 @@ var (
 		"OLLaMA": 1,
 		"GenAI":  2,
 	}
+	apilist = [3]string{"OpenAI", "OLLaMA", "GenAI"}
 )
 
 func init() {
@@ -141,12 +141,25 @@ func init() {
 				if t == "" {
 					continue
 				}
+				logrus.Infoln("[aichat] 回复内容:", t)
 				recCfg := airecord.GetRecordConfig()
-				fmt.Println(recCfg)
-				if id != nil {
-					id = ctx.SendChain(message.Reply(id), message.Text(t))
+				if !cfg.NoRecord {
+					record := ctx.GetAIRecord(recCfg.ModelID, recCfg.Customgid, t).String()
+					if record != "" {
+						ctx.SendChain(message.Record(record))
+					} else {
+						if id != nil {
+							id = ctx.SendChain(message.Reply(id), message.Text(t))
+						} else {
+							id = ctx.SendChain(message.Text(t))
+						}
+					}
 				} else {
-					id = ctx.SendChain(message.Text(t))
+					if id != nil {
+						id = ctx.SendChain(message.Reply(id), message.Text(t))
+					} else {
+						id = ctx.SendChain(message.Text(t))
+					}
 				}
 				process.SleepAbout1sTo2s()
 			}
@@ -277,8 +290,24 @@ func init() {
 		Handle(newextrasetuint(&cfg.MaxN))
 	en.OnPrefix("设置AI聊天TopP", ensureconfig, zero.OnlyPrivate, zero.SuperUserPermission).SetBlock(true).
 		Handle(newextrasetfloat32(&cfg.TopP))
+	en.OnRegex("^设置AI聊天(不)?以AI语音输出$", ensureconfig, zero.OnlyPrivate, zero.SuperUserPermission).SetBlock(true).
+		Handle(newextrasetbool(&cfg.NoRecord))
 	en.OnFullMatch("查看AI聊天配置", ensureconfig).SetBlock(true).
 		Handle(func(ctx *zero.Ctx) {
-			ctx.SendChain(message.Text(printConfig(cfg)))
+			c, ok := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
+			if !ok {
+				ctx.SendChain(message.Text("ERROR: no such plugin"))
+				return
+			}
+			gid := ctx.Event.GroupID
+			rate := c.GetData(gid) & 0xff
+			temp := (c.GetData(gid) >> 8) & 0xff
+			if temp <= 0 {
+				temp = 70 // default setting
+			}
+			if temp > 100 {
+				temp = 100
+			}
+			ctx.SendChain(message.Text(printConfig(rate, temp, cfg)))
 		})
 }