xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-20 17:26:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: auto ban 403

Browse Source
This commit is contained in:
1808837298@qq.com 2024-06-16 15:45:41 +08:00
parent e8845ce1de
commit e3f66807ee
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
service/channel.go
View File

@ -31,7 +31,7 @@ func ShouldDisableChannel(err *relaymodel.OpenAIError, statusCode int) bool {
if err == nil { if err == nil {
return false return false
} }
if statusCode == http.StatusUnauthorized { if statusCode == http.StatusUnauthorized || statusCode == http.StatusForbidden {
return true return true
} }
switch err.Code { switch err.Code {
@ -59,6 +59,8 @@ func ShouldDisableChannel(err *relaymodel.OpenAIError, statusCode int) bool {
return true return true
} else if strings.HasPrefix(err.Message, "You exceeded your current quota") { } else if strings.HasPrefix(err.Message, "You exceeded your current quota") {
return true return true
} else if strings.HasPrefix(err.Message, "Permission denied") {
return true
} }
return false return false
} }