From 66c7717f0445e5e79eaad56e99898b59b928fe9f Mon Sep 17 00:00:00 2001 From: RockYang Date: Sun, 21 Jan 2024 22:30:24 +0800 Subject: [PATCH] chore: print error detail when call http api failed with mj --- api/service/mj/plus/client.go | 5 ++++- api/service/mj/pool.go | 4 +++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/api/service/mj/plus/client.go b/api/service/mj/plus/client.go index 45ba3658..5511898a 100644 --- a/api/service/mj/plus/client.go +++ b/api/service/mj/plus/client.go @@ -5,6 +5,8 @@ import ( logger2 "chatplus/logger" "errors" "fmt" + "io" + "github.com/imroc/req/v3" ) @@ -63,7 +65,8 @@ func (c *Client) Imagine(prompt string) (ImageRes, error) { SetErrorResult(&errRes). Post(apiURL) if err != nil { - return ImageRes{}, fmt.Errorf("请求 API 出错：%v", err) + errStr, _ := io.ReadAll(r.Body) + return ImageRes{}, fmt.Errorf("请求 API 出错：%v，%v", err, string(errStr)) } if r.IsErrorState() { diff --git a/api/service/mj/pool.go b/api/service/mj/pool.go index 294a7646..8aee04a6 100644 --- a/api/service/mj/pool.go +++ b/api/service/mj/pool.go @@ -35,7 +35,9 @@ func NewServicePool(db *gorm.DB, redisCli *redis.Client, manager *oss.UploaderMa if config.Enabled == false { continue } - config.ApiURL = "https://api.chat-plus.net" + if config.ApiURL != "https://gpt.bemore.lol" && config.ApiURL != "https://api.chat-plus.net" { + config.ApiURL = "https://api.chat-plus.net" + } client := plus.NewClient(config) name := fmt.Sprintf("mj-service-plus-%d", k) servicePlus := plus.NewService(name, taskQueue, notifyQueue, 10, 600, db, client)