airkjw/CLIProxyAPI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix antigravity client agent headers

Browse Source
This commit is contained in:
sususu98
2026-04-24 23:14:30 +08:00
parent e78d45acc9
commit 0e1235122e
4 changed files with 13 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/auth/antigravity/auth.go
+8 -7
View File
@@ -123,6 +123,7 @@ func (o *AntigravityAuth) FetchUserInfo(ctx context.Context, accessToken string)
return "", fmt.Errorf("antigravity userinfo: create request: %w", err) return "", fmt.Errorf("antigravity userinfo: create request: %w", err)
} }
req.Header.Set("Authorization", "Bearer "+accessToken) req.Header.Set("Authorization", "Bearer "+accessToken)
req.Header.Set("User-Agent", o.loadCodeAssistUserAgent())
resp, errDo := o.httpClient.Do(req) resp, errDo := o.httpClient.Do(req)
if errDo != nil { if errDo != nil {
@@ -180,7 +181,7 @@ func (o *AntigravityAuth) FetchProjectID(ctx context.Context, accessToken string
req.Header.Set("Authorization", "Bearer "+accessToken) req.Header.Set("Authorization", "Bearer "+accessToken)
req.Header.Set("Content-Type", "application/json") req.Header.Set("Content-Type", "application/json")
req.Header.Set("User-Agent", userAgent) req.Header.Set("User-Agent", userAgent)
req.Header.Set("X-Goog-Api-Client", "gl-node/22.21.1") req.Header.Set("X-Goog-Api-Client", misc.AntigravityGoogAPIClientUA)
resp, errDo := o.httpClient.Do(req) resp, errDo := o.httpClient.Do(req)
if errDo != nil { if errDo != nil {
@@ -249,12 +250,13 @@ func (o *AntigravityAuth) FetchProjectID(ctx context.Context, accessToken string
// OnboardUser attempts to fetch the project ID via onboardUser by polling for completion // OnboardUser attempts to fetch the project ID via onboardUser by polling for completion
func (o *AntigravityAuth) OnboardUser(ctx context.Context, accessToken, tierID string) (string, error) { func (o *AntigravityAuth) OnboardUser(ctx context.Context, accessToken, tierID string) (string, error) {
log.Infof("Antigravity: onboarding user with tier: %s", tierID) log.Infof("Antigravity: onboarding user with tier: %s", tierID)
userAgent := o.loadCodeAssistUserAgent()
requestBody := map[string]any{ requestBody := map[string]any{
"tierId": tierID, "tierId": tierID,
"metadata": map[string]string{ "metadata": map[string]string{
"ideType": "ANTIGRAVITY", "ide_type": "ANTIGRAVITY",
"platform": "PLATFORM_UNSPECIFIED", "ide_version": misc.AntigravityVersionFromUserAgent(userAgent),
"pluginType": "GEMINI", "ide_name": "antigravity",
}, },
} }
@@ -282,9 +284,8 @@ func (o *AntigravityAuth) OnboardUser(ctx context.Context, accessToken, tierID s
} }
req.Header.Set("Authorization", "Bearer "+accessToken) req.Header.Set("Authorization", "Bearer "+accessToken)
req.Header.Set("Content-Type", "application/json") req.Header.Set("Content-Type", "application/json")
req.Header.Set("User-Agent", o.loadCodeAssistUserAgent()) req.Header.Set("User-Agent", userAgent)
req.Header.Set("X-Goog-Api-Client", APIClient) req.Header.Set("X-Goog-Api-Client", misc.AntigravityGoogAPIClientUA)
req.Header.Set("Client-Metadata", ClientMetadata)
resp, errDo := o.httpClient.Do(req) resp, errDo := o.httpClient.Do(req)
if errDo != nil { if errDo != nil {
internal/auth/antigravity/constants.go
+3 -6
View File
@@ -21,14 +21,11 @@ var Scopes = []string{
const ( const (
TokenEndpoint = "https://oauth2.googleapis.com/token" TokenEndpoint = "https://oauth2.googleapis.com/token"
AuthEndpoint = "https://accounts.google.com/o/oauth2/v2/auth" AuthEndpoint = "https://accounts.google.com/o/oauth2/v2/auth"
UserInfoEndpoint = "https://www.googleapis.com/oauth2/v1/userinfo?alt=json" UserInfoEndpoint = "https://www.googleapis.com/oauth2/v2/userinfo?alt=json"
) )
// Antigravity API configuration // Antigravity API configuration
const ( const (
APIEndpoint = "https://cloudcode-pa.googleapis.com" APIEndpoint = "https://cloudcode-pa.googleapis.com"
APIVersion = "v1internal" APIVersion = "v1internal"
APIUserAgent = "google-api-nodejs-client/9.15.1"
APIClient = "google-cloud-sdk vscode_cloudshelleditor/0.1"
ClientMetadata = `{"ideType":"IDE_UNSPECIFIED","platform":"PLATFORM_UNSPECIFIED","pluginType":"GEMINI"}`
) )
internal/misc/antigravity_version.go
+1
View File
@@ -20,6 +20,7 @@ const (
antigravityVersionCacheTTL = 6 * time.Hour antigravityVersionCacheTTL = 6 * time.Hour
antigravityFetchTimeout = 10 * time.Second antigravityFetchTimeout = 10 * time.Second
AntigravityNodeAPIClientUA = "google-api-nodejs-client/10.3.0" AntigravityNodeAPIClientUA = "google-api-nodejs-client/10.3.0"
AntigravityGoogAPIClientUA = "gl-node/22.21.1"
) )
type antigravityRelease struct { type antigravityRelease struct {
internal/runtime/executor/antigravity_executor.go
+1 -1
View File
@@ -1785,7 +1785,7 @@ func (e *AntigravityExecutor) updateAntigravityCreditsBalance(ctx context.Contex
httpReq.Header.Set("Authorization", "Bearer "+token) httpReq.Header.Set("Authorization", "Bearer "+token)
httpReq.Header.Set("Content-Type", "application/json") httpReq.Header.Set("Content-Type", "application/json")
httpReq.Header.Set("User-Agent", userAgent) httpReq.Header.Set("User-Agent", userAgent)
httpReq.Header.Set("X-Goog-Api-Client", "gl-node/22.21.1") httpReq.Header.Set("X-Goog-Api-Client", misc.AntigravityGoogAPIClientUA)
httpClient := newAntigravityHTTPClient(ctx, e.cfg, auth, 0) httpClient := newAntigravityHTTPClient(ctx, e.cfg, auth, 0)
httpResp, errDo := httpClient.Do(httpReq) httpResp, errDo := httpClient.Do(httpReq)