oadmin/sub2api
1
0
Fork 0
Code Issues Pull Requests 1 Actions 2 Packages Projects Releases Wiki Activity

fix: skip rate limiting when custom error codes don't match upstream status

Browse Source 
Add ShouldHandleErrorCode guard at the entry of handleGeminiUpstreamError
and AntigravityGatewayService.handleUpstreamError so that accounts with
custom error codes (e.g. [599]) are not rate-limited when the upstream
returns a non-matching status (e.g. 429).
This commit is contained in:
erio
2026-02-09 18:53:52 +08:00
parent 73f455745c
commit 6892e84ad2
2 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/internal/service/antigravity_gateway_service.go
View File

@@ -2243,6 +2243,10 @@ func (s *AntigravityGatewayService) handleUpstreamError(
requestedModel string, requestedModel string,
groupID int64, sessionHash string, isStickySession bool, groupID int64, sessionHash string, isStickySession bool,
) *handleModelRateLimitResult { ) *handleModelRateLimitResult {
// 遵守自定义错误码策略：未命中则跳过所有限流处理
if !account.ShouldHandleErrorCode(statusCode) {
return nil
}
// 模型级限流处理（优先） // 模型级限流处理（优先）
result := s.handleModelRateLimit(&handleModelRateLimitParams{ result := s.handleModelRateLimit(&handleModelRateLimitParams{
ctx: ctx, ctx: ctx,

4
backend/internal/service/gemini_messages_compat_service.go
View File

@@ -2597,6 +2597,10 @@ func asInt(v any) (int, bool) {
} }
func (s *GeminiMessagesCompatService) handleGeminiUpstreamError(ctx context.Context, account *Account, statusCode int, headers http.Header, body []byte) { func (s *GeminiMessagesCompatService) handleGeminiUpstreamError(ctx context.Context, account *Account, statusCode int, headers http.Header, body []byte) {
// 遵守自定义错误码策略：未命中则跳过所有限流处理
if !account.ShouldHandleErrorCode(statusCode) {
return
}
if s.rateLimitService != nil && (statusCode == 401 || statusCode == 403 || statusCode == 529) { if s.rateLimitService != nil && (statusCode == 401 || statusCode == 403 || statusCode == 529) {
s.rateLimitService.HandleUpstreamError(ctx, account, statusCode, headers, body) s.rateLimitService.HandleUpstreamError(ctx, account, statusCode, headers, body)
return return