From 8b31e00a145552d9f251044666e95936cfb311a4 Mon Sep 17 00:00:00 2001 From: erio Date: Tue, 24 Mar 2026 18:42:28 +0800 Subject: [PATCH] fix: sync test rate limit keys and fix gofmt alignment Update rate limit keys in credits overages tests from claude-sonnet-4-5 to claude-sonnet-4-6 to match the new default mapping. Fix gofmt alignment in constants.go. --- backend/internal/domain/constants.go | 6 +++--- .../service/antigravity_credits_overages_test.go | 12 ++++++------ 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/backend/internal/domain/constants.go b/backend/internal/domain/constants.go index ab07751f8b..911376dd72 100644 --- a/backend/internal/domain/constants.go +++ b/backend/internal/domain/constants.go @@ -76,11 +76,11 @@ var DefaultAntigravityModelMapping = map[string]string{ "claude-opus-4-6": "claude-opus-4-6-thinking", // 简称映射 "claude-opus-4-5-thinking": "claude-opus-4-6-thinking", // 迁移旧模型 "claude-sonnet-4-6": "claude-sonnet-4-6", - "claude-sonnet-4-5": "claude-sonnet-4-6", // Sonnet 4.5 → 4.6 - "claude-sonnet-4-5-thinking": "claude-sonnet-4-6", // Sonnet 4.5 → 4.6 + "claude-sonnet-4-5": "claude-sonnet-4-6", // Sonnet 4.5 → 4.6 + "claude-sonnet-4-5-thinking": "claude-sonnet-4-6", // Sonnet 4.5 → 4.6 // Claude 详细版本 ID 映射 "claude-opus-4-5-20251101": "claude-opus-4-6-thinking", // 迁移旧模型 - "claude-sonnet-4-5-20250929": "claude-sonnet-4-6", // Sonnet 4.5 → 4.6 + "claude-sonnet-4-5-20250929": "claude-sonnet-4-6", // Sonnet 4.5 → 4.6 // Claude Haiku → Sonnet（无 Haiku 支持） "claude-haiku-4-5": "claude-sonnet-4-6", "claude-haiku-4-5-20251001": "claude-sonnet-4-6", diff --git a/backend/internal/service/antigravity_credits_overages_test.go b/backend/internal/service/antigravity_credits_overages_test.go index 7a5224daf3..c3439be5b2 100644 --- a/backend/internal/service/antigravity_credits_overages_test.go +++ b/backend/internal/service/antigravity_credits_overages_test.go @@ -239,7 +239,7 @@ func TestAntigravityRetryLoop_ModelRateLimited_InjectsCredits(t *testing.T) { Extra: map[string]any{ "allow_overages": true, modelRateLimitsKey: map[string]any{ - "claude-sonnet-4-5": map[string]any{ + "claude-sonnet-4-6": map[string]any{ "rate_limited_at": time.Now().UTC().Format(time.RFC3339), "rate_limit_reset_at": time.Now().Add(30 * time.Minute).UTC().Format(time.RFC3339), }, @@ -290,7 +290,7 @@ func TestAntigravityRetryLoop_CreditsExhausted_DoesNotInject(t *testing.T) { Extra: map[string]any{ "allow_overages": true, modelRateLimitsKey: map[string]any{ - "claude-sonnet-4-5": map[string]any{ + "claude-sonnet-4-6": map[string]any{ "rate_limited_at": time.Now().UTC().Format(time.RFC3339), "rate_limit_reset_at": time.Now().Add(30 * time.Minute).UTC().Format(time.RFC3339), }, @@ -355,7 +355,7 @@ func TestAntigravityRetryLoop_CreditErrorMarksExhausted(t *testing.T) { Extra: map[string]any{ "allow_overages": true, modelRateLimitsKey: map[string]any{ - "claude-sonnet-4-5": map[string]any{ + "claude-sonnet-4-6": map[string]any{ "rate_limited_at": time.Now().UTC().Format(time.RFC3339), "rate_limit_reset_at": time.Now().Add(30 * time.Minute).UTC().Format(time.RFC3339), }, @@ -503,7 +503,7 @@ func TestClearCreditsExhausted(t *testing.T) { ID: 1, Extra: map[string]any{ modelRateLimitsKey: map[string]any{ - "claude-sonnet-4-5": map[string]any{ + "claude-sonnet-4-6": map[string]any{ "rate_limited_at": "2026-03-15T00:00:00Z", "rate_limit_reset_at": "2099-03-15T00:00:00Z", }, @@ -520,7 +520,7 @@ func TestClearCreditsExhausted(t *testing.T) { ID: 1, Extra: map[string]any{ modelRateLimitsKey: map[string]any{ - "claude-sonnet-4-5": map[string]any{ + "claude-sonnet-4-6": map[string]any{ "rate_limited_at": "2026-03-15T00:00:00Z", "rate_limit_reset_at": "2099-03-15T00:00:00Z", }, @@ -538,7 +538,7 @@ func TestClearCreditsExhausted(t *testing.T) { _, exists := rawLimits[creditsExhaustedKey] require.False(t, exists, "AICredits key 应被删除") // 普通模型限流应保留 - _, exists = rawLimits["claude-sonnet-4-5"] + _, exists = rawLimits["claude-sonnet-4-6"] require.True(t, exists, "普通模型限流应保留") }) }