chuan/CLIProxyAPI
1
0
Fork 0
mirror of https://github.com/router-for-me/CLIProxyAPI.git synced 2026-02-02 20:40:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(auth): handle quota cooldown in retry logic for transient errors

Browse Source
This commit is contained in:
sxjeru
2026-01-21 21:48:23 +08:00
parent ef4508dbc8
commit 30a59168d7
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
sdk/cliproxy/auth/conductor.go
View File

@@ -1371,8 +1371,12 @@ func (m *Manager) MarkResult(ctx context.Context, result Result) {
shouldSuspendModel = true shouldSuspendModel = true
setModelQuota = true setModelQuota = true
case 408, 500, 502, 503, 504: case 408, 500, 502, 503, 504:
next := now.Add(1 * time.Minute) if quotaCooldownDisabled.Load() {
state.NextRetryAfter = next state.NextRetryAfter = time.Time{}
} else {
next := now.Add(1 * time.Minute)
state.NextRetryAfter = next
}
default: default:
state.NextRetryAfter = time.Time{} state.NextRetryAfter = time.Time{}
} }
@@ -1623,7 +1627,11 @@ func applyAuthFailureState(auth *Auth, resultErr *Error, retryAfter *time.Durati
auth.NextRetryAfter = next auth.NextRetryAfter = next
case 408, 500, 502, 503, 504: case 408, 500, 502, 503, 504:
auth.StatusMessage = "transient upstream error" auth.StatusMessage = "transient upstream error"
auth.NextRetryAfter = now.Add(1 * time.Minute) if quotaCooldownDisabled.Load() {
auth.NextRetryAfter = time.Time{}
} else {
auth.NextRetryAfter = now.Add(1 * time.Minute)
}
default: default:
if auth.StatusMessage == "" { if auth.StatusMessage == "" {
auth.StatusMessage = "request failed" auth.StatusMessage = "request failed"