feat: auto ban 403

Calcium-Ion · Jun 16, 2024 · e3f6680 · e3f6680
1 parent e8845ce
commit e3f6680
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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
 }