1
0
forked from lxh/go-wxhelper

🎨 AI新增指定群不启用

This commit is contained in:
李寻欢 2023-11-24 09:44:13 +08:00
parent 997ad806f0
commit 3b4862d5bc
3 changed files with 16 additions and 6 deletions

View File

@ -45,4 +45,7 @@ ai:
# 接口代理域名不填默认ChatGPT官方地址 # 接口代理域名不填默认ChatGPT官方地址
baseUrl: https://sxxx baseUrl: https://sxxx
# 人设 # 人设
personality: 你的名字叫张三,你是一个百科机器人,你的爱好是看电影,你的性格是开朗的,你的专长是讲故事,你的梦想是当一名童话故事作家。你对政治没有一点儿兴趣,也不会讨论任何与政治相关的话题,你甚至可以拒绝回答这一类话题。 personality: 你的名字叫张三,你是一个百科机器人,你的爱好是看电影,你的性格是开朗的,你的专长是讲故事,你的梦想是当一名童话故事作家。你对政治没有一点儿兴趣,也不会讨论任何与政治相关的话题,你甚至可以拒绝回答这一类话题。
# 禁用的群
disableGroup:
- 49448748645@chatroom

View File

@ -3,9 +3,10 @@ package config
// ai // ai
// @description: AI配置 // @description: AI配置
type ai struct { type ai struct {
Enable bool `json:"enable" yaml:"enable"` // 是否启用AI Enable bool `json:"enable" yaml:"enable"` // 是否启用AI
Model string `json:"model" yaml:"model"` // 模型 Model string `json:"model" yaml:"model"` // 模型
ApiKey string `json:"apiKey" yaml:"apiKey"` // API Key ApiKey string `json:"apiKey" yaml:"apiKey"` // API Key
BaseUrl string `json:"baseUrl" yaml:"baseUrl"` // API地址 BaseUrl string `json:"baseUrl" yaml:"baseUrl"` // API地址
Personality string `json:"personality" yaml:"personality"` // 人设 Personality string `json:"personality" yaml:"personality"` // 人设
DisableGroup []string `json:"disableGroup" yaml:"disableGroup"` // 禁用群组
} }

View File

@ -3,6 +3,7 @@ package handler
import ( import (
"context" "context"
"fmt" "fmt"
"github.com/duke-git/lancet/v2/slice"
"github.com/sashabaranov/go-openai" "github.com/sashabaranov/go-openai"
"go-wechat/config" "go-wechat/config"
"go-wechat/entity" "go-wechat/entity"
@ -20,6 +21,11 @@ func handleAtMessage(m entity.Message) {
return return
} }
// 如果在禁用的群组里面,就不处理
if slice.Contain(config.Conf.Ai.DisableGroup, m.FromUser) {
return
}
// 预处理一下发送的消息,用正则去掉@机器人的内容 // 预处理一下发送的消息,用正则去掉@机器人的内容
re := regexp.MustCompile(`@([^]+)`) re := regexp.MustCompile(`@([^]+)`)
matches := re.FindStringSubmatch(m.Content) matches := re.FindStringSubmatch(m.Content)