From 9ddafba7b8c7750662e497a96d7c325b2ae52ea4 Mon Sep 17 00:00:00 2001
From: CaIon <1808837298@qq.com>
Date: Mon, 27 Nov 2023 19:21:16 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dtts=E5=BC=82=E5=B8=B8?=
 =?UTF-8?q?=E6=97=B6=E4=BB=8D=E7=84=B6=E6=89=A3=E8=B4=B9=E7=9A=84=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 controller/relay-audio.go | 5 +++++
 controller/relay-image.go | 5 +++++
 2 files changed, 10 insertions(+)

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 {