DBT/clawgo
1
0
Fork 0
mirror of https://github.com/YspCoder/clawgo.git synced 2026-05-07 20:27:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: disable revoked codex oauth accounts

Browse Source
This commit is contained in:
lpf
2026-03-13 11:08:28 +08:00
parent 06e3599e45
commit 80f728d8b6
4 changed files with 197 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
pkg/providers/codex_provider.go
View File

@@ -545,6 +545,16 @@ func (p *CodexProvider) postWebsocketStream(ctx context.Context, endpoint string
}
func (p *CodexProvider) handleAttemptFailure(attempt authAttempt, status int, body []byte) {
if reason, detail, disabled := classifyCodexPermanentDisable(status, body); disabled {
if attempt.kind == "oauth" && attempt.session != nil && p.base != nil && p.base.oauth != nil {
p.base.oauth.disableSession(attempt.session, reason, detail)
recordProviderOAuthError(p.base.providerName, attempt.session, reason)
}
if attempt.kind == "api_key" && p.base != nil {
p.base.markAPIKeyFailure(reason)
}
return
}
reason, retry := classifyOAuthFailure(status, body)
if !retry {
return
@@ -558,6 +568,21 @@ func (p *CodexProvider) handleAttemptFailure(attempt authAttempt, status int, bo
}
}
func classifyCodexPermanentDisable(status int, body []byte) (oauthFailureReason, string, bool) {
if status != http.StatusUnauthorized && status != http.StatusPaymentRequired {
return "", "", false
}
lower := strings.ToLower(strings.TrimSpace(string(body)))
switch {
case strings.Contains(lower, "token_revoked"), strings.Contains(lower, "invalidated oauth token"):
return oauthFailureRevoked, "oauth token revoked", true
case strings.Contains(lower, "deactivated_workspace"):
return oauthFailureDisabled, "workspace deactivated", true
default:
return "", "", false
}
}
func (p *CodexProvider) doWebsocketAttempt(ctx context.Context, endpoint string, payload map[string]interface{}, attempt authAttempt, options map[string]interface{}, onDelta func(string)) ([]byte, int, string, error) {
wsURL, err := buildCodexResponsesWebsocketURL(endpoint)
if err != nil {