From d8b5f4e52b6a2ff650757f2b48f93491647a4b13 Mon Sep 17 00:00:00 2001 From: Benny Date: Mon, 14 Aug 2023 17:32:32 +0800 Subject: [PATCH] fix: empty completion issue caused by bad status code from upstream channel --- controller/relay-text.go | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/controller/relay-text.go b/controller/relay-text.go index e8dab514..88e807d8 100644 --- a/controller/relay-text.go +++ b/controller/relay-text.go @@ -313,6 +313,11 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode { isStream = isStream || strings.HasPrefix(resp.Header.Get("Content-Type"), "text/event-stream") } + if resp.StatusCode != http.StatusOK { + return errorWrapper( + fmt.Errorf("bad status code: %d", resp.StatusCode), "bad_status_code", resp.StatusCode) + } + var textResponse TextResponse tokenName := c.GetString("token_name") channelId := c.GetInt("channel_id")