fix(antigravity): strike-break optimistic quota readings that keep 429ing
Google's quota API can report remaining quota while generation endpoints keep returning 429 (sprint/weekly dual-pool mismatch). handleAntigravityQuotaError trusted remainingPercentage > 0 as healthy and returned null, causing 429 retry loops across multi-account pools. Add a strike-based circuit breaker to the optimistic and unavailable quota paths: - After 3 strikes (429/409) within 60s for the same connection+model, cache-block that pair for 15 minutes by synthesizing an entry in the shared RAM quota cache. - Re-assert active strike blocks across refreshes so optimistic readings cannot resurrect a broken pair prematurely. - Reset strikes and clear synthesized cache entry upon successful request. - Keep exact-resetAt handling for genuine 0% exhausted readings. Closes #3681
This commit is contained in:
@@ -7,7 +7,7 @@ import {
|
||||
extractApiKey,
|
||||
isValidApiKey,
|
||||
} from "../services/auth.js";
|
||||
import { handleAntigravityQuotaError } from "../services/antigravityQuota.js";
|
||||
import { handleAntigravityQuotaError, clearAntigravityStrikes } from "../services/antigravityQuota.js";
|
||||
import { getSettings } from "@/lib/localDb";
|
||||
import { getModelInfo, getComboModels } from "../services/model.js";
|
||||
import { handleChatCore } from "open-sse/handlers/chatCore.js";
|
||||
@@ -302,6 +302,8 @@ async function handleSingleModelChat(body, modelStr, clientRawRequest = null, re
|
||||
},
|
||||
onRequestSuccess: async () => {
|
||||
await clearAccountError(credentials.connectionId, credentials, model);
|
||||
// "Consecutive" strikes: a success clears the breaker for this pair.
|
||||
clearAntigravityStrikes(credentials.connectionId, model);
|
||||
}
|
||||
});
|
||||
|
||||
|
||||
@@ -17,6 +17,56 @@ const inflightRefresh = new Map();
|
||||
|
||||
const MIN_REFRESH_INTERVAL_MS = 30_000; // 30s between refreshes per connection
|
||||
|
||||
// Strike-based circuit breaker (#3681): Google's quota API can report remaining
|
||||
// quota while generation endpoints keep returning 429 (sprint/weekly dual-pool
|
||||
// mismatch). After STRIKE_THRESHOLD 429s within the window for the same
|
||||
// connection+model, treat the optimistic quota reading as untrusted and
|
||||
// cache-block that pair instead of retry-storming upstream.
|
||||
const STRIKE_WINDOW_MS = 60_000; // strikes older than this reset the count
|
||||
const STRIKE_THRESHOLD = 3;
|
||||
const STRIKE_BLOCK_MS = 15 * 60_000;
|
||||
const strikeCounts = new Map(); // "connectionId|model" → { count, windowStart (anchored at first strike) }
|
||||
const strikeBlocks = new Map(); // "connectionId|model" → blockedUntil ms
|
||||
|
||||
/**
|
||||
* Re-apply active strike blocks onto a fresh quotas snapshot so the auth
|
||||
* pre-filter (which reads this cache) keeps skipping the blocked pair across
|
||||
* requests until the block expires — same channel as the exhausted-0% path.
|
||||
*/
|
||||
function applyActiveStrikeBlocks(connectionId, quotas) {
|
||||
const now = Date.now();
|
||||
for (const [key, until] of strikeBlocks) {
|
||||
if (!key.startsWith(`${connectionId}|`)) continue;
|
||||
if (until <= now) {
|
||||
strikeBlocks.delete(key);
|
||||
continue;
|
||||
}
|
||||
quotas[key.slice(connectionId.length + 1)] = {
|
||||
remainingPercentage: 0,
|
||||
resetAt: new Date(until).toISOString(),
|
||||
};
|
||||
}
|
||||
return quotas;
|
||||
}
|
||||
|
||||
/**
|
||||
* Clear strike state for a connection|model after a successful request, so
|
||||
* "consecutive" strikes means consecutive. Only removes a synthesized cache
|
||||
* entry (resetAt == our block deadline); a real upstream 0% reading stays.
|
||||
*/
|
||||
export function clearAntigravityStrikes(connectionId, model) {
|
||||
const key = `${connectionId}|${model}`;
|
||||
strikeCounts.delete(key);
|
||||
const until = strikeBlocks.get(key);
|
||||
if (until === undefined) return;
|
||||
strikeBlocks.delete(key);
|
||||
const cached = quotaCache.get(connectionId);
|
||||
if (cached?.[model]?.resetAt === new Date(until).toISOString()) {
|
||||
delete cached[model];
|
||||
quotaCache.set(connectionId, cached);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the quota cache (read-only reference for auth.js pre-filter).
|
||||
*/
|
||||
@@ -69,7 +119,9 @@ async function _doRefresh(connectionId, accessToken, providerSpecificData, now)
|
||||
if (!usage?.quotas || usage.message) return null;
|
||||
|
||||
// Update in-memory cache. Caller logs CACHE_BLOCK only if requested model is exhausted.
|
||||
quotaCache.set(connectionId, usage.quotas);
|
||||
// Strike blocks are re-asserted after every refresh so an optimistic
|
||||
// upstream reading cannot resurrect a pair we just circuit-broke.
|
||||
quotaCache.set(connectionId, applyActiveStrikeBlocks(connectionId, usage.quotas));
|
||||
|
||||
return usage.quotas;
|
||||
} catch (e) {
|
||||
@@ -89,7 +141,43 @@ export async function handleAntigravityQuotaError(connectionId, status, model, a
|
||||
// Throttle applies to error paths too: one quota request per account/30s.
|
||||
// The first 409/429 populates cache; concurrent or repeated errors reuse it.
|
||||
const quota = (await refreshAntigravityQuota(connectionId, accessToken, providerSpecificData))?.[model];
|
||||
if (!quota || quota.remainingPercentage > 0 || !quota.resetAt) return null;
|
||||
|
||||
// Strike breaker: count every 429 whose quota reading is either optimistic
|
||||
// (remaining > 0) or unavailable (quota API 403/error). 3 within the window
|
||||
// => the pair is unhealthy regardless of what the API claims; block 15m.
|
||||
// 409 counts too by design: Antigravity signals pool exhaustion with 409 as
|
||||
// well (see #3561 — "skip exhausted account/model quota before upstream
|
||||
// retry" was motivated by 409/429 pairs), and poisoning by transient 409s
|
||||
// requires 3 of them inside 60 seconds on the same pair.
|
||||
if (!quota || quota.remainingPercentage > 0) {
|
||||
const key = `${connectionId}|${model}`;
|
||||
const now = Date.now();
|
||||
const strike = strikeCounts.get(key);
|
||||
// Fixed window anchored at the FIRST qualifying strike: three 429s must
|
||||
// all land within 60s of that first one, not within 60s of each other.
|
||||
const windowStart = strike && now - strike.windowStart <= STRIKE_WINDOW_MS ? strike.windowStart : now;
|
||||
const count = strike && windowStart === strike.windowStart ? strike.count + 1 : 1;
|
||||
strikeCounts.set(key, { count, windowStart });
|
||||
if (count >= STRIKE_THRESHOLD) {
|
||||
strikeCounts.delete(key);
|
||||
const blockedUntil = now + STRIKE_BLOCK_MS;
|
||||
const reading = quota ? `${Math.round(quota.remainingPercentage)}%` : "unknown";
|
||||
log.warn("AG_QUOTA", `${connectionId.slice(0, 8)} | STRIKE_${status} ${model} — ${count}x 429 (quota ${reading}); CACHE_BLOCK 15m`);
|
||||
// Synthesize a 0% entry in the shared cache so the auth pre-filter skips
|
||||
// this pair on subsequent requests too, not just the current retry loop
|
||||
// (the chat handler does not persist modelLock_* for this path).
|
||||
const cached = quotaCache.get(connectionId) || {};
|
||||
cached[model] = { remainingPercentage: 0, resetAt: new Date(blockedUntil).toISOString() };
|
||||
quotaCache.set(connectionId, cached);
|
||||
strikeBlocks.set(key, blockedUntil);
|
||||
return blockedUntil;
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
// Healthy-but-exhausted reading: clear strikes and use the exact resetAt.
|
||||
strikeCounts.delete(`${connectionId}|${model}`);
|
||||
if (!quota.resetAt) return null;
|
||||
|
||||
const resetMs = new Date(quota.resetAt).getTime();
|
||||
if (resetMs <= Date.now()) return null;
|
||||
|
||||
Reference in New Issue
Block a user