Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: disable channel when error message contain credit or balance

Browse Source
This commit is contained in:
JustSong 2024-04-05 00:31:41 +08:00
parent 1994256bac
commit 76569bb0b6
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
relay/util/common.go
View File

@ -49,6 +49,12 @@ func ShouldDisableChannel(err *relaymodel.Error, statusCode int) bool {
if strings.Contains(err.Message, "quota") { if strings.Contains(err.Message, "quota") {
return true return true
} }
if strings.Contains(err.Message, "credit") {
return true
}
if strings.Contains(err.Message, "balance") {
return true
}
return false return false
} }