From fed1a1d6a3ba6fedda6e3161ab3489fd13b55155 Mon Sep 17 00:00:00 2001
From: CaIon <1808837298@qq.com>
Date: Thu, 4 Apr 2024 21:21:44 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E8=B6=85=E6=97=B6=E7=8A=B6=E6=80=81?=
 =?UTF-8?q?=E7=A0=81=E4=B8=8D=E9=87=8D=E8=AF=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 controller/relay.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/controller/relay.go b/controller/relay.go
index 5c89f91..c6d850d 100644
--- a/controller/relay.go
+++ b/controller/relay.go
@@ -93,6 +93,10 @@ func shouldRetry(c *gin.Context, channelId int, openaiErr *dto.OpenAIErrorWithSt
 		return true
 	}
 	if openaiErr.StatusCode/100 == 5 {
+		// 超时不重试
+		if openaiErr.StatusCode == 504 || openaiErr.StatusCode == 524 {
+			return false
+		}
 		return true
 	}
 	if openaiErr.StatusCode == http.StatusBadRequest {