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, }, },