diff --git a/controller/relay-audio.go b/controller/relay-audio.go index 371ed88..4cb74ae 100644 --- a/controller/relay-audio.go +++ b/controller/relay-audio.go @@ -129,6 +129,11 @@ func relayAudioHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode if err != nil { return errorWrapper(err, "close_request_body_failed", http.StatusInternalServerError) } + + if resp.StatusCode != http.StatusOK { + return relayErrorHandler(resp) + } + var audioResponse AudioResponse defer func(ctx context.Context) { diff --git a/controller/relay-image.go b/controller/relay-image.go index c1f837b..f7894be 100644 --- a/controller/relay-image.go +++ b/controller/relay-image.go @@ -147,6 +147,11 @@ func relayImageHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode if err != nil { return errorWrapper(err, "close_request_body_failed", http.StatusInternalServerError) } + + if resp.StatusCode != http.StatusOK { + return relayErrorHandler(resp) + } + var textResponse ImageResponse defer func(ctx context.Context) { if consumeQuota {