This commit is contained in:
parent
60bfa0e8a0
commit
06a64c5e5a
@ -2,10 +2,12 @@ package handler
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"fmt"
|
||||||
"github.com/sashabaranov/go-openai"
|
"github.com/sashabaranov/go-openai"
|
||||||
"go-wechat/config"
|
"go-wechat/config"
|
||||||
"go-wechat/entity"
|
"go-wechat/entity"
|
||||||
"go-wechat/utils"
|
"go-wechat/utils"
|
||||||
|
"log"
|
||||||
)
|
)
|
||||||
|
|
||||||
// handleAtMessage
|
// handleAtMessage
|
||||||
@ -16,7 +18,11 @@ func handleAtMessage(m entity.Message) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
// 默认使用AI回复
|
// 默认使用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(
|
resp, err := client.CreateChatCompletion(
|
||||||
context.Background(),
|
context.Background(),
|
||||||
openai.ChatCompletionRequest{
|
openai.ChatCompletionRequest{
|
||||||
@ -31,6 +37,7 @@ func handleAtMessage(m entity.Message) {
|
|||||||
)
|
)
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
log.Printf("OpenAI聊天发起失败: %v", err.Error())
|
||||||
utils.SendMessage(m.FromUser, m.GroupUser, "AI炸啦~", 0)
|
utils.SendMessage(m.FromUser, m.GroupUser, "AI炸啦~", 0)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user