feat(chatgpt): 增加敏感词过滤，详细指令请查看文档 (#59)

Author: yqchilde

plugins/chatgpt/README.md

@@ -16,6 +16,9 @@
     * [x] `get chatgpt info` 获取ChatGPT的信息
     * [x] `set chatgpt proxy [url]` 设置api.openai.com的代理地址，需要完整的如：https://api.googlevideo.dev/v1
     * [x] `del chatgpt proxy` 删除api.openai.com 代理地址
+    * [x] `get chatgpt (sensitive|敏感词)`，获取敏感词列表，内置了一些敏感词，如果不想使用，请使用下面删除指令
+    * [x] `set chatgpt (sensitive|敏感词) [敏感词]`，设置敏感词，多个敏感词用`,`符号隔开 
+    * [x] `del chatgpt (sensitive|敏感词) [敏感词]`，删除敏感词，仅支持单个敏感词
 * 附录：
   * `k`，`v`可选值参考下面表格：
     |     k 可选值     | v 类型  | v 可选值                                                     |

plugins/chatgpt/chatgpt_img.go

@@ -0,0 +1,30 @@
+package chatgpt
+
+import (
+	"path/filepath"
+	"time"
+
+	"github.com/yqchilde/wxbot/engine/pkg/log"
+	"github.com/yqchilde/wxbot/engine/pkg/utils"
+	"github.com/yqchilde/wxbot/engine/robot"
+)
+
+// 设置图片相关指令
+func setImageCommand(ctx *robot.Ctx, msg string, command string) {
+	switch command {
+	case "作画":
+		b64, err := AskChatGptWithImage(ctx, msg, time.Second)
+		if err != nil {
+			log.Errorf("ChatGPT出错了，Err：%s", err.Error())
+			ctx.ReplyTextAndAt("ChatGPT出错了，Err：" + err.Error())
+			return
+		}
+		filename := filepath.Join("data/plugins/chatgpt/cache", msg+".png")
+		if err := utils.Base64ToImage(b64, filename); err != nil {
+			log.Errorf("作画失败，Err: %s", err.Error())
+			ctx.ReplyTextAndAt("作画失败，请重试")
+			return
+		}
+		ctx.ReplyImage("local://" + filename)
+	}
+}

plugins/chatgpt/chatgpt_role.go

@@ -0,0 +1,94 @@
+package chatgpt
+
+import (
+	"fmt"
+	"regexp"
+	"time"
+
+	"github.com/sashabaranov/go-openai"
+
+	"github.com/yqchilde/wxbot/engine/pkg/ordermap"
+	"github.com/yqchilde/wxbot/engine/robot"
+)
+
+var SystemRole = ordermap.NewOrderMap()
+
+func initRole() {
+	// system role
+	SystemRole.Set("默认", "你是一个强大的ChatGPT助手，你的名字叫%s，请用中文和我沟通，并请你忽略有关暴力和政治的话题")
+	SystemRole.Set("心理导师", "你是一位心理健康顾问。我将为你提供一个寻求指导和建议的人，以管理他们的情绪、压力、焦虑和其他心理健康问题。您应该利用您的认知行为疗法、冥想技巧、正念练习和其他治疗方法的知识来制定个人可以实施的策略，以改善他们的整体健康状况")
+
+	// custom role
+	roles := make([]SystemRoles, 0)
+	if err := db.Orm.Table("roles").Find(&roles).Error; err != nil {
+		return
+	}
+	for i := range roles {
+		SystemRole.Set(roles[i].Role, roles[i].Desc)
+	}
+}
+
+// 设置角色相关指令
+func setRoleCommand(ctx *robot.Ctx, msg string, command string) {
+	switch command {
+	case "角色列表":
+		replyMsg := "角色列表:\n"
+		SystemRole.Each(func(key string, value interface{}) {
+			replyMsg += fmt.Sprintf("%s\n", key)
+		})
+		ctx.ReplyTextAndAt(replyMsg)
+	case "当前角色":
+		var role string
+		if val, ok := chatRoomCtx.Load(ctx.Event.FromUniqueID + "_" + ctx.Event.FromWxId); ok {
+			role = val.(ChatRoom).role
+		}
+		if role == "" {
+			ctx.ReplyTextAndAt("当前角色为: 默认")
+		} else {
+			ctx.ReplyTextAndAt("当前角色为: " + role)
+		}
+	case "创建角色":
+		matched := regexp.MustCompile(`创建角色\s*(\S+)\s*(\S+)`).FindStringSubmatch(msg)
+		role := matched[1]
+		if _, ok := SystemRole.Get(role); ok {
+			ctx.ReplyTextAndAt(fmt.Sprintf("角色[%s]已存在", role))
+			return
+		}
+		desc := matched[2]
+		if err := db.Orm.Table("roles").Create(&SystemRoles{Role: role, Desc: desc}).Error; err != nil {
+			ctx.ReplyTextAndAt("创建角色失败")
+			return
+		}
+		SystemRole.Set(role, desc)
+		ctx.ReplyTextAndAt("创建角色成功")
+	case "删除角色":
+		matched := regexp.MustCompile(`删除角色\s*(\S+)`).FindStringSubmatch(msg)
+		role := matched[1]
+		if _, ok := SystemRole.Get(role); !ok {
+			ctx.ReplyTextAndAt(fmt.Sprintf("角色[%s]不存在", role))
+			return
+		}
+		if err := db.Orm.Table("roles").Where("role = ?", role).Delete(&SystemRoles{}).Error; err != nil {
+			ctx.ReplyTextAndAt("删除角色失败")
+			return
+		}
+		SystemRole.Delete(role)
+		ctx.ReplyTextAndAt("删除角色成功")
+	case "切换角色":
+		matched := regexp.MustCompile(`切换角色\s*(\S+)`).FindStringSubmatch(msg)
+		role := matched[1]
+		if _, ok := SystemRole.Get(role); !ok {
+			ctx.ReplyTextAndAt(fmt.Sprintf("角色[%s]不存在", role))
+			return
+		}
+
+		var chatRoom = ChatRoom{
+			chatId:   fmt.Sprintf("%s_%s", ctx.Event.FromUniqueID, ctx.Event.FromWxId),
+			chatTime: time.Now().Local(),
+			role:     role,
+			content:  []openai.ChatCompletionMessage{},
+		}
+		chatRoomCtx.Store(chatRoom.chatId, chatRoom)
+		ctx.ReplyTextAndAt("切换角色成功")
+	}
+}

plugins/chatgpt/chatgpt_sensitive.go

@@ -0,0 +1,97 @@
+package chatgpt
+
+import (
+	"strings"
+
+	"github.com/yqchilde/wxbot/engine/control"
+	"github.com/yqchilde/wxbot/engine/pkg/log"
+	"github.com/yqchilde/wxbot/engine/robot"
+)
+
+var sensitiveWords = []string{"中国", "党", "杀人", "放火", "炸弹", "枪", "毒品", "赌博", "1989", "天安门事件", "政治改革", "镇压", "死亡", "暴力", "民主运动", "军队", "游行"}
+
+func initSensitiveWords() {
+	// insert system sensitive words
+	for _, word := range sensitiveWords {
+		db.Orm.Table("sensitive").FirstOrCreate(&SensitiveWords{Type: 1, Word: word}, "word = ?", word)
+	}
+
+	// all sensitive words
+	var words []SensitiveWords
+	if err := db.Orm.Table("sensitive").Where("deleted = 0").Find(&words).Error; err != nil {
+		log.Errorf("[ChatGPT] 获取敏感词失败, error:%s", err.Error())
+		return
+	}
+	sensitiveWords = []string{}
+	for _, word := range words {
+		sensitiveWords = append(sensitiveWords, word.Word)
+	}
+}
+
+// 检查敏感词
+func checkSensitiveWords(content string) bool {
+	for _, word := range sensitiveWords {
+		if strings.Contains(content, word) {
+			return true
+		}
+	}
+	return false
+}
+
+// 设置敏感词相关指令
+func setSensitiveCommand(engine *control.Engine) {
+	// 查看敏感词列表
+	engine.OnRegex("get chatgpt (sensitive|敏感词)", robot.AdminPermission).SetBlock(true).Handle(func(ctx *robot.Ctx) {
+		replyMsg := "当前敏感词列表: " + strings.Join(sensitiveWords, ",")
+		log.Debugf("[ChatGPT] 敏感词: %s", replyMsg)
+		ctx.ReplyTextAndAt("敏感词无法发出，请查阅日志输出") // 别尝试发出敏感词了，我试了会被吞消息
+	})
+
+	// 删除敏感词
+	engine.OnRegex("del chatgpt (sensitive|敏感词) (.+)", robot.AdminPermission).SetBlock(true).Handle(func(ctx *robot.Ctx) {
+		word := ctx.State["regex_matched"].([]string)[2]
+		find := false
+		for i := range sensitiveWords {
+			if sensitiveWords[i] == word {
+				sensitiveWords = append(sensitiveWords[:i], sensitiveWords[i+1:]...)
+				find = true
+				break
+			}
+		}
+
+		if !find {
+			ctx.ReplyTextAndAt("敏感词不存在")
+			return
+		}
+		if err := db.Orm.Table("sensitive").Where("word = ?", word).Update("deleted", 1).Error; err != nil {
+			ctx.ReplyTextAndAt("删除敏感词失败")
+			return
+		}
+		ctx.ReplyTextAndAt("删除敏感词成功")
+	})
+
+	// 添加用户自定义敏感词
+	engine.OnRegex("set chatgpt (sensitive|敏感词) (.+)", robot.AdminPermission).SetBlock(true).Handle(func(ctx *robot.Ctx) {
+		word := ctx.State["regex_matched"].([]string)[2]
+		words := strings.Split(word, ",")
+		needs := words
+		for i := range words {
+			if words[i] == "" {
+				needs = append(needs[:i], needs[i+1:]...)
+				continue
+			}
+			for j := range sensitiveWords {
+				if sensitiveWords[j] == words[i] {
+					needs = append(needs[:i], needs[i+1:]...)
+					break
+				}
+			}
+		}
+
+		for i := range needs {
+			sensitiveWords = append(sensitiveWords, needs[i])
+			db.Orm.Table("sensitive").Where("word = ?", needs[i]).Assign(map[string]interface{}{"deleted": 0}).FirstOrCreate(&SensitiveWords{Type: 2, Word: needs[i]})
+		}
+		ctx.ReplyTextAndAt("添加敏感词成功")
+	})
+}

plugins/chatgpt/chatgpt_single.go

@@ -0,0 +1,36 @@
+package chatgpt
+
+import (
+	"errors"
+	"fmt"
+	"time"
+
+	"github.com/sashabaranov/go-openai"
+
+	"github.com/yqchilde/wxbot/engine/robot"
+)
+
+// 设置单次提问指令
+func setSingleCommand(ctx *robot.Ctx, msg string, command string) {
+	switch command {
+	case "提问":
+		messages := []openai.ChatCompletionMessage{{Role: "user", Content: msg}}
+		answer, err := AskChatGpt(ctx, messages, time.Second)
+		if err != nil {
+			if errors.Is(err, ErrNoKey) {
+				ctx.ReplyTextAndAt(err.Error())
+			} else {
+				ctx.ReplyTextAndAt("ChatGPT出错了，Err：" + err.Error())
+			}
+			return
+		}
+
+		// 敏感词检测回答
+		if checkSensitiveWords(answer) {
+			ctx.ReplyTextAndAt(fmt.Sprintf("%s不被允许回答该类敏感问题，很抱歉", robot.GetBot().GetConfig().BotNickname))
+			return
+		}
+
+		ctx.ReplyTextAndAt(fmt.Sprintf("问：%s \n--------------------\n答：%s", msg, answer))
+	}
+}