Browse Source

feat: auto ban 403

[email protected] 1 year ago
parent
commit
e3f66807ee
1 changed files with 3 additions and 1 deletions
  1. 3 1
      service/channel.go

+ 3 - 1
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
 }