From 8d5e470e1fc0661a9434ca85c34eda90a5631e8c Mon Sep 17 00:00:00 2001 From: rensumo Date: Sat, 4 Apr 2026 14:52:59 +0800 Subject: [PATCH] feat: dynamically fetch antigravity UA version from releases API Fetch the latest version from the antigravity auto-updater releases endpoint and cache it for 6 hours. Falls back to 1.21.9 if the API is unreachable or returns unexpected data. --- cmd/fetch_antigravity_models/main.go | 3 +- internal/misc/antigravity_version.go | 97 +++++++++++++++++++ .../runtime/executor/antigravity_executor.go | 5 +- 3 files changed, 102 insertions(+), 3 deletions(-) create mode 100644 internal/misc/antigravity_version.go diff --git a/cmd/fetch_antigravity_models/main.go b/cmd/fetch_antigravity_models/main.go index 54ec16ca..d4328eb3 100644 --- a/cmd/fetch_antigravity_models/main.go +++ b/cmd/fetch_antigravity_models/main.go @@ -26,6 +26,7 @@ import ( "time" "github.com/router-for-me/CLIProxyAPI/v6/internal/logging" + "github.com/router-for-me/CLIProxyAPI/v6/internal/misc" sdkauth "github.com/router-for-me/CLIProxyAPI/v6/sdk/auth" coreauth "github.com/router-for-me/CLIProxyAPI/v6/sdk/cliproxy/auth" "github.com/router-for-me/CLIProxyAPI/v6/sdk/proxyutil" @@ -188,7 +189,7 @@ func fetchModels(ctx context.Context, auth *coreauth.Auth) []modelEntry { httpReq.Close = true httpReq.Header.Set("Content-Type", "application/json") httpReq.Header.Set("Authorization", "Bearer "+accessToken) - httpReq.Header.Set("User-Agent", "antigravity/1.21.9 darwin/arm64") + httpReq.Header.Set("User-Agent", misc.AntigravityUserAgent()) httpClient := &http.Client{Timeout: 30 * time.Second} if transport, _, errProxy := proxyutil.BuildHTTPTransport(auth.ProxyURL); errProxy == nil && transport != nil { diff --git a/internal/misc/antigravity_version.go b/internal/misc/antigravity_version.go new file mode 100644 index 00000000..c882269f --- /dev/null +++ b/internal/misc/antigravity_version.go @@ -0,0 +1,97 @@ +// Package misc provides miscellaneous utility functions for the CLI Proxy API server. +package misc + +import ( + "encoding/json" + "fmt" + "net/http" + "sync" + "time" + + log "github.com/sirupsen/logrus" +) + +const ( + antigravityReleasesURL = "https://antigravity-auto-updater-974169037036.us-central1.run.app/releases" + antigravityFallbackVersion = "1.21.9" + antigravityVersionCacheTTL = 6 * time.Hour + antigravityFetchTimeout = 10 * time.Second +) + +type antigravityRelease struct { + Version string `json:"version"` + ExecutionID string `json:"execution_id"` +} + +var ( + cachedAntigravityVersion string + antigravityVersionMu sync.RWMutex + antigravityVersionExpiry time.Time +) + +// AntigravityLatestVersion returns the latest antigravity version from the releases API. +// It caches the result for antigravityVersionCacheTTL and falls back to antigravityFallbackVersion +// if the fetch fails. +func AntigravityLatestVersion() string { + antigravityVersionMu.RLock() + if cachedAntigravityVersion != "" && time.Now().Before(antigravityVersionExpiry) { + v := cachedAntigravityVersion + antigravityVersionMu.RUnlock() + return v + } + antigravityVersionMu.RUnlock() + + antigravityVersionMu.Lock() + defer antigravityVersionMu.Unlock() + + // Double-check after acquiring write lock. + if cachedAntigravityVersion != "" && time.Now().Before(antigravityVersionExpiry) { + return cachedAntigravityVersion + } + + version := fetchAntigravityLatestVersion() + cachedAntigravityVersion = version + antigravityVersionExpiry = time.Now().Add(antigravityVersionCacheTTL) + return version +} + +// AntigravityUserAgent returns the User-Agent string for antigravity requests +// using the latest version fetched from the releases API. +func AntigravityUserAgent() string { + return fmt.Sprintf("antigravity/%s darwin/arm64", AntigravityLatestVersion()) +} + +func fetchAntigravityLatestVersion() string { + client := &http.Client{Timeout: antigravityFetchTimeout} + resp, err := client.Get(antigravityReleasesURL) + if err != nil { + log.WithError(err).Warn("failed to fetch antigravity releases, using fallback version") + return antigravityFallbackVersion + } + defer resp.Body.Close() + + if resp.StatusCode != http.StatusOK { + log.WithField("status", resp.StatusCode).Warn("antigravity releases API returned non-200, using fallback version") + return antigravityFallbackVersion + } + + var releases []antigravityRelease + if err := json.NewDecoder(resp.Body).Decode(&releases); err != nil { + log.WithError(err).Warn("failed to decode antigravity releases response, using fallback version") + return antigravityFallbackVersion + } + + if len(releases) == 0 { + log.Warn("antigravity releases API returned empty list, using fallback version") + return antigravityFallbackVersion + } + + version := releases[0].Version + if version == "" { + log.Warn("antigravity releases API returned empty version, using fallback version") + return antigravityFallbackVersion + } + + log.WithField("version", version).Info("fetched latest antigravity version") + return version +} diff --git a/internal/runtime/executor/antigravity_executor.go b/internal/runtime/executor/antigravity_executor.go index b9bf4842..ecab3c87 100644 --- a/internal/runtime/executor/antigravity_executor.go +++ b/internal/runtime/executor/antigravity_executor.go @@ -24,6 +24,7 @@ import ( "github.com/google/uuid" "github.com/router-for-me/CLIProxyAPI/v6/internal/config" + "github.com/router-for-me/CLIProxyAPI/v6/internal/misc" "github.com/router-for-me/CLIProxyAPI/v6/internal/runtime/executor/helps" "github.com/router-for-me/CLIProxyAPI/v6/internal/thinking" "github.com/router-for-me/CLIProxyAPI/v6/internal/util" @@ -45,7 +46,7 @@ const ( antigravityGeneratePath = "/v1internal:generateContent" antigravityClientID = "1071006060591-tmhssin2h21lcre235vtolojh4g403ep.apps.googleusercontent.com" antigravityClientSecret = "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" - defaultAntigravityAgent = "antigravity/1.21.9 darwin/arm64" + defaultAntigravityAgent = "antigravity/1.21.9 darwin/arm64" // fallback only; overridden at runtime by misc.AntigravityUserAgent() antigravityAuthType = "antigravity" refreshSkew = 3000 * time.Second antigravityCreditsRetryTTL = 5 * time.Hour @@ -1739,7 +1740,7 @@ func resolveUserAgent(auth *cliproxyauth.Auth) string { } } } - return defaultAntigravityAgent + return misc.AntigravityUserAgent() } func antigravityRetryAttempts(auth *cliproxyauth.Auth, cfg *config.Config) int {