This commit is contained in:
parent
997ad806f0
commit
3b4862d5bc
@ -46,3 +46,6 @@ ai:
|
||||
baseUrl: https://sxxx
|
||||
# 人设
|
||||
personality: 你的名字叫张三,你是一个百科机器人,你的爱好是看电影,你的性格是开朗的,你的专长是讲故事,你的梦想是当一名童话故事作家。你对政治没有一点儿兴趣,也不会讨论任何与政治相关的话题,你甚至可以拒绝回答这一类话题。
|
||||
# 禁用的群
|
||||
disableGroup:
|
||||
- 49448748645@chatroom
|
@ -8,4 +8,5 @@ type ai struct {
|
||||
ApiKey string `json:"apiKey" yaml:"apiKey"` // API Key
|
||||
BaseUrl string `json:"baseUrl" yaml:"baseUrl"` // API地址
|
||||
Personality string `json:"personality" yaml:"personality"` // 人设
|
||||
DisableGroup []string `json:"disableGroup" yaml:"disableGroup"` // 禁用群组
|
||||
}
|
||||
|
@ -3,6 +3,7 @@ package handler
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"github.com/duke-git/lancet/v2/slice"
|
||||
"github.com/sashabaranov/go-openai"
|
||||
"go-wechat/config"
|
||||
"go-wechat/entity"
|
||||
@ -20,6 +21,11 @@ func handleAtMessage(m entity.Message) {
|
||||
return
|
||||
}
|
||||
|
||||
// 如果在禁用的群组里面,就不处理
|
||||
if slice.Contain(config.Conf.Ai.DisableGroup, m.FromUser) {
|
||||
return
|
||||
}
|
||||
|
||||
// 预处理一下发送的消息,用正则去掉@机器人的内容
|
||||
re := regexp.MustCompile(`@([^ ]+)`)
|
||||
matches := re.FindStringSubmatch(m.Content)
|
||||
|
Loading…
Reference in New Issue
Block a user