forked from lxh/go-wxhelper
🎨 AI新增指定群不启用
This commit is contained in:
parent
997ad806f0
commit
3b4862d5bc
@ -45,4 +45,7 @@ ai:
|
|||||||
# 接口代理域名,不填默认ChatGPT官方地址
|
# 接口代理域名,不填默认ChatGPT官方地址
|
||||||
baseUrl: https://sxxx
|
baseUrl: https://sxxx
|
||||||
# 人设
|
# 人设
|
||||||
personality: 你的名字叫张三,你是一个百科机器人,你的爱好是看电影,你的性格是开朗的,你的专长是讲故事,你的梦想是当一名童话故事作家。你对政治没有一点儿兴趣,也不会讨论任何与政治相关的话题,你甚至可以拒绝回答这一类话题。
|
personality: 你的名字叫张三,你是一个百科机器人,你的爱好是看电影,你的性格是开朗的,你的专长是讲故事,你的梦想是当一名童话故事作家。你对政治没有一点儿兴趣,也不会讨论任何与政治相关的话题,你甚至可以拒绝回答这一类话题。
|
||||||
|
# 禁用的群
|
||||||
|
disableGroup:
|
||||||
|
- 49448748645@chatroom
|
11
config/ai.go
11
config/ai.go
@ -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"` // 禁用群组
|
||||||
}
|
}
|
||||||
|
@ -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)
|
||||||
|
Loading…
Reference in New Issue
Block a user