From 8388aa537ff4876ac443ac3dceef72ac3d54a304 Mon Sep 17 00:00:00 2001 From: JustSong Date: Sun, 10 Mar 2024 14:59:57 +0800 Subject: [PATCH] chore: able to search channel now --- relay/controller/text.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/relay/controller/text.go b/relay/controller/text.go index 59c5f637..781170f4 100644 --- a/relay/controller/text.go +++ b/relay/controller/text.go @@ -83,11 +83,12 @@ func RelayTextHelper(c *gin.Context) *model.ErrorWithStatusCode { logger.Errorf(ctx, "DoRequest failed: %s", err.Error()) return openai.ErrorWrapper(err, "do_request_failed", http.StatusInternalServerError) } - meta.IsStream = meta.IsStream || strings.HasPrefix(resp.Header.Get("Content-Type"), "text/event-stream") - if resp.StatusCode != http.StatusOK { + errorHappened := (resp.StatusCode != http.StatusOK) || (meta.IsStream && resp.Header.Get("Content-Type") == "application/json") + if errorHappened { util.ReturnPreConsumedQuota(ctx, preConsumedQuota, meta.TokenId) return util.RelayErrorHandler(resp) } + meta.IsStream = meta.IsStream || strings.HasPrefix(resp.Header.Get("Content-Type"), "text/event-stream") // do response usage, respErr := adaptor.DoResponse(c, resp, meta)