From e3f66807eeaff7faa9721e920ded7488807af091 Mon Sep 17 00:00:00 2001
From: "1808837298@qq.com" <1808837298@qq.com>
Date: Sun, 16 Jun 2024 15:45:41 +0800
Subject: [PATCH] feat: auto ban 403

---
 service/channel.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/service/channel.go b/service/channel.go
index 69a7d3a..7ee12b2 100644
--- a/service/channel.go
+++ b/service/channel.go
@@ -31,7 +31,7 @@ func ShouldDisableChannel(err *relaymodel.OpenAIError, statusCode int) bool {
 	if err == nil {
 		return false
 	}
-	if statusCode == http.StatusUnauthorized {
+	if statusCode == http.StatusUnauthorized || statusCode == http.StatusForbidden {
 		return true
 	}
 	switch err.Code {
@@ -59,6 +59,8 @@ func ShouldDisableChannel(err *relaymodel.OpenAIError, statusCode int) bool {
 		return true
 	} else if strings.HasPrefix(err.Message, "You exceeded your current quota") {
 		return true
+	} else if strings.HasPrefix(err.Message, "Permission denied") {
+		return true
 	}
 	return false
 }