From e435a46db528f655399f3dd2a2572182d224a137 Mon Sep 17 00:00:00 2001 From: shaw Date: Tue, 24 Feb 2026 14:44:57 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=20antigravity=20UserA?= =?UTF-8?q?gent=20=E9=87=8D=E6=9E=84=E9=81=97=E7=95=99=E7=9A=84=E7=BC=96?= =?UTF-8?q?=E8=AF=91=E9=94=99=E8=AF=AF=E5=92=8C=E6=B5=8B=E8=AF=95=E4=B8=8D?= =?UTF-8?q?=E5=8C=B9=E9=85=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - oauth.go: GetUserAgent() 缺少闭合大括号导致语法错误 - client_test.go/oauth_test.go: UserAgent 变量已重构为 GetUserAgent()，更新测试引用 - model_rate_limit_test.go: gemini-3-pro-preview 映射目标已更新为 gemini-3.1-pro-high，同步测试 --- backend/internal/pkg/antigravity/client_test.go | 8 ++++---- backend/internal/pkg/antigravity/oauth.go | 3 ++- backend/internal/pkg/antigravity/oauth_test.go | 4 ++-- backend/internal/service/model_rate_limit_test.go | 4 ++-- 4 files changed, 10 insertions(+), 9 deletions(-) diff --git a/backend/internal/pkg/antigravity/client_test.go b/backend/internal/pkg/antigravity/client_test.go index 7e8d3a2a..d3e2fd94 100644 --- a/backend/internal/pkg/antigravity/client_test.go +++ b/backend/internal/pkg/antigravity/client_test.go @@ -49,8 +49,8 @@ func TestNewAPIRequestWithURL_普通请求(t *testing.T) { if auth := req.Header.Get("Authorization"); auth != "Bearer test-token" { t.Errorf("Authorization 不匹配: got %s", auth) } - if ua := req.Header.Get("User-Agent"); ua != UserAgent { - t.Errorf("User-Agent 不匹配: got %s, want %s", ua, UserAgent) + if ua := req.Header.Get("User-Agent"); ua != GetUserAgent() { + t.Errorf("User-Agent 不匹配: got %s, want %s", ua, GetUserAgent()) } } @@ -1191,7 +1191,7 @@ func TestClient_LoadCodeAssist_Success_RealCall(t *testing.T) { if ct := r.Header.Get("Content-Type"); ct != "application/json" { t.Errorf("Content-Type 不匹配: got %s", ct) } - if ua := r.Header.Get("User-Agent"); ua != UserAgent { + if ua := r.Header.Get("User-Agent"); ua != GetUserAgent() { t.Errorf("User-Agent 不匹配: got %s", ua) } @@ -1380,7 +1380,7 @@ func TestClient_FetchAvailableModels_Success_RealCall(t *testing.T) { if ct := r.Header.Get("Content-Type"); ct != "application/json" { t.Errorf("Content-Type 不匹配: got %s", ct) } - if ua := r.Header.Get("User-Agent"); ua != UserAgent { + if ua := r.Header.Get("User-Agent"); ua != GetUserAgent() { t.Errorf("User-Agent 不匹配: got %s", ua) } diff --git a/backend/internal/pkg/antigravity/oauth.go b/backend/internal/pkg/antigravity/oauth.go index cdda9be6..ba84a247 100644 --- a/backend/internal/pkg/antigravity/oauth.go +++ b/backend/internal/pkg/antigravity/oauth.go @@ -64,7 +64,8 @@ func init() { // GetUserAgent 返回当前配置的 User-Agent func GetUserAgent() string { return fmt.Sprintf("antigravity/%s windows/amd64", defaultUserAgentVersion) - +} + func getClientSecret() (string, error) { if v := strings.TrimSpace(ClientSecret); v != "" { return v, nil diff --git a/backend/internal/pkg/antigravity/oauth_test.go b/backend/internal/pkg/antigravity/oauth_test.go index 67731c06..78184941 100644 --- a/backend/internal/pkg/antigravity/oauth_test.go +++ b/backend/internal/pkg/antigravity/oauth_test.go @@ -676,8 +676,8 @@ func TestConstants_值正确(t *testing.T) { if RedirectURI != "http://localhost:8085/callback" { t.Errorf("RedirectURI 不匹配: got %s", RedirectURI) } - if UserAgent != "antigravity/1.15.8 windows/amd64" { - t.Errorf("UserAgent 不匹配: got %s", UserAgent) + if GetUserAgent() != "antigravity/1.84.2 windows/amd64" { + t.Errorf("UserAgent 不匹配: got %s", GetUserAgent()) } if SessionTTL != 30*time.Minute { t.Errorf("SessionTTL 不匹配: got %v", SessionTTL) diff --git a/backend/internal/service/model_rate_limit_test.go b/backend/internal/service/model_rate_limit_test.go index b79b9688..c8f78ce3 100644 --- a/backend/internal/service/model_rate_limit_test.go +++ b/backend/internal/service/model_rate_limit_test.go @@ -107,12 +107,12 @@ func TestIsModelRateLimited(t *testing.T) { expected: true, }, { - name: "antigravity platform - gemini-3-pro-preview mapped to gemini-3-pro-high", + name: "antigravity platform - gemini-3-pro-preview mapped to gemini-3.1-pro-high", account: &Account{ Platform: PlatformAntigravity, Extra: map[string]any{ modelRateLimitsKey: map[string]any{ - "gemini-3-pro-high": map[string]any{ + "gemini-3.1-pro-high": map[string]any{ "rate_limit_reset_at": future, }, },