From 9301b3fed310629df055dd0c30844d3e715cb546 Mon Sep 17 00:00:00 2001 From: JustSong Date: Thu, 8 Jun 2023 14:09:39 +0800 Subject: [PATCH] chore: update test logic --- controller/channel-test.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/controller/channel-test.go b/controller/channel-test.go index 0ae256f9..b1a43eff 100644 --- a/controller/channel-test.go +++ b/controller/channel-test.go @@ -56,7 +56,7 @@ func testChannel(channel *model.Channel, request *ChatRequest) error { if err != nil { return err } - if response.Error.Message != "" || response.Error.Code != "" { + if response.Usage.CompletionTokens == 0 { return errors.New(fmt.Sprintf("type %s, code %s, message %s", response.Error.Type, response.Error.Code, response.Error.Message)) } return nil