From fbe6cd75b1531311dc1438fb78a64ffe8b56e083 Mon Sep 17 00:00:00 2001
From: CalciumIon <1808837298@qq.com>
Date: Sat, 3 Aug 2024 17:07:14 +0800
Subject: [PATCH] =?UTF-8?q?chore:=20=E4=BC=98=E5=8C=96relay=E4=BB=A3?=
 =?UTF-8?q?=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 controller/relay.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/controller/relay.go b/controller/relay.go
index 66339f4..4726d7e 100644
--- a/controller/relay.go
+++ b/controller/relay.go
@@ -47,8 +47,14 @@ func Relay(c *gin.Context) {
 	for i := 0; i <= common.RetryTimes; i++ {
 		channel, err := getChannel(c, group, originalModel, i)
 		if err != nil {
-			common.LogError(c, fmt.Sprintf("Failed to get channel: %s", err.Error()))
-			break
+			errMsg := fmt.Sprintf("获取渠道出错: %s", err.Error())
+			common.LogError(c, errMsg)
+			openaiErr = service.OpenAIErrorWrapperLocal(err, "get_channel_failed", http.StatusInternalServerError)
+			openaiErr.Error.Message = common.MessageWithRequestId(errMsg, requestId)
+			c.JSON(openaiErr.StatusCode, gin.H{
+				"error": openaiErr.Error,
+			})
+			return
 		}
 
 		openaiErr = relayRequest(c, relayMode, channel)