From a2d8fe5e79e6e31dd602bc2527a374a590fc9b7f Mon Sep 17 00:00:00 2001 From: lixiangwuxian Date: Thu, 9 Jan 2025 02:03:47 +0800 Subject: [PATCH] chore: update go-openai dependency to v1.36.1 and improve OpenAI client configuration in headmasterHandler --- go.mod | 2 +- go.sum | 4 ++-- handler/headmaster/headmaster.go | 5 ++++- 3 files changed, 7 insertions(+), 4 deletions(-) diff --git a/go.mod b/go.mod index 4dbba47..8937bb6 100644 --- a/go.mod +++ b/go.mod @@ -13,7 +13,7 @@ require ( github.com/gorilla/websocket v1.5.3 github.com/jmoiron/sqlx v1.4.0 github.com/mattn/go-sqlite3 v1.14.23 - github.com/sashabaranov/go-openai v1.30.3 + github.com/sashabaranov/go-openai v1.36.1 golang.org/x/exp v0.0.0-20241009180824-f66d83c29e7c golang.org/x/image v0.21.0 golang.org/x/net v0.30.0 diff --git a/go.sum b/go.sum index fe961fb..67a4cf6 100644 --- a/go.sum +++ b/go.sum @@ -138,8 +138,8 @@ github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZb github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= github.com/rogpeppe/go-internal v1.8.0 h1:FCbCCtXNOY3UtUuHUYaghJg4y7Fd14rXifAYUAtL9R8= github.com/rogpeppe/go-internal v1.8.0/go.mod h1:WmiCO8CzOY8rg0OYDC4/i/2WRWAB6poM+XZ2dLUbcbE= -github.com/sashabaranov/go-openai v1.30.3 h1:TEdRP3otRXX2A7vLoU+kI5XpoSo7VUUlM/rEttUqgek= -github.com/sashabaranov/go-openai v1.30.3/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg= +github.com/sashabaranov/go-openai v1.36.1 h1:EVfRXwIlW2rUzpx6vR+aeIKCK/xylSrVYAx1TMTSX3g= +github.com/sashabaranov/go-openai v1.36.1/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg= github.com/sirupsen/logrus v1.4.1/go.mod h1:ni0Sbl8bgC9z8RoU9G6nDWqqs/fq4eDPysMBDgk/93Q= github.com/sirupsen/logrus v1.9.3 h1:dueUQJ1C2q9oE3F7wvmSGAaVtTmUizReu6fjN8uqzbQ= github.com/sirupsen/logrus v1.9.3/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ= diff --git a/handler/headmaster/headmaster.go b/handler/headmaster/headmaster.go index ce8ae79..cb4ca14 100644 --- a/handler/headmaster/headmaster.go +++ b/handler/headmaster/headmaster.go @@ -46,7 +46,10 @@ func headmasterHandler(msg model.Message) (reply model.Reply) { } func ask(from string, question string) (reply string) { - client := openai.NewClientWithConfig(openai.DefaultAzureConfig(config.ConfigManager.GetProperty("openai_api_key"), config.ConfigManager.GetProperty("openai_api_base_url"))) + llmConfig := openai.DefaultAzureConfig(config.ConfigManager.GetProperty("openai_api_key"), config.ConfigManager.GetProperty("openai_api_base_url")) + llmConfig.APIType = openai.APITypeOpenAI + llmConfig.APIVersion = "" + client := openai.NewClientWithConfig(llmConfig) resp, err := client.CreateChatCompletion( context.Background(), openai.ChatCompletionRequest{