🎨 优化传递给AI的消息，去掉艾特机器人那一段字符串

handler/at_message.go

@@ -2,10 +2,14 @@ package handler
 
 import (
 	"context"
+	"fmt"
 	"github.com/sashabaranov/go-openai"
 	"go-wechat/config"
 	"go-wechat/entity"
 	"go-wechat/utils"
+	"log"
+	"regexp"
+	"strings"
 )
 
 // handleAtMessage
@@ -15,8 +19,21 @@ func handleAtMessage(m entity.Message) {
 	if !config.Conf.Ai.Enable {
 		return
 	}
+
+	// 预处理一下发送的消息，用正则去掉@机器人的内容
+	re := regexp.MustCompile(`@([^ ]+)`)
+	matches := re.FindStringSubmatch(m.Content)
+	if len(matches) > 0 {
+		// 过滤掉第一个匹配到的
+		m.Content = strings.Replace(m.Content, matches[0], "", 1)
+	}
+
 	// 默认使用AI回复
-	client := openai.NewClient(config.Conf.Ai.ApiKey)
+	conf := openai.DefaultConfig(config.Conf.Ai.ApiKey)
+	if config.Conf.Ai.BaseUrl != "" {
+		conf.BaseURL = fmt.Sprintf("%s/v1", config.Conf.Ai.BaseUrl)
+	}
+	client := openai.NewClientWithConfig(conf)
 	resp, err := client.CreateChatCompletion(
 		context.Background(),
 		openai.ChatCompletionRequest{
@@ -31,10 +48,11 @@ func handleAtMessage(m entity.Message) {
 	)
 
 	if err != nil {
+		log.Printf("OpenAI聊天发起失败: %v", err.Error())
 		utils.SendMessage(m.FromUser, m.GroupUser, "AI炸啦~", 0)
 		return
 	}
 
 	// 发送消息
-	utils.SendMessage(m.FromUser, m.GroupUser, resp.Choices[0].Message.Content, 0)
+	utils.SendMessage(m.FromUser, m.GroupUser, "\n"+resp.Choices[0].Message.Content, 0)
 }