diff --git a/internal/runtime/executor/antigravity_executor.go b/internal/runtime/executor/antigravity_executor.go index 0baba498..e113f0a4 100644 --- a/internal/runtime/executor/antigravity_executor.go +++ b/internal/runtime/executor/antigravity_executor.go @@ -45,7 +45,6 @@ const ( defaultAntigravityAgent = "antigravity/1.104.0 darwin/arm64" antigravityAuthType = "antigravity" refreshSkew = 3000 * time.Second - tokenRefreshTimeout = 30 * time.Second ) var ( @@ -951,7 +950,7 @@ func (e *AntigravityExecutor) refreshToken(ctx context.Context, auth *cliproxyau httpReq.Header.Set("User-Agent", defaultAntigravityAgent) httpReq.Header.Set("Content-Type", "application/x-www-form-urlencoded") - httpClient := newProxyAwareHTTPClient(ctx, e.cfg, auth, tokenRefreshTimeout) + httpClient := newProxyAwareHTTPClient(ctx, e.cfg, auth, 0) httpResp, errDo := httpClient.Do(httpReq) if errDo != nil { return auth, errDo diff --git a/sdk/api/management.go b/sdk/api/management.go index 7faaa0d0..66af41ae 100644 --- a/sdk/api/management.go +++ b/sdk/api/management.go @@ -21,6 +21,7 @@ type ManagementTokenRequester interface { RequestIFlowToken(*gin.Context) RequestIFlowCookieToken(*gin.Context) GetAuthStatus(c *gin.Context) + PostOAuthCallback(c *gin.Context) } type managementTokenRequester struct { @@ -65,3 +66,7 @@ func (m *managementTokenRequester) RequestIFlowCookieToken(c *gin.Context) { func (m *managementTokenRequester) GetAuthStatus(c *gin.Context) { m.handler.GetAuthStatus(c) } + +func (m *managementTokenRequester) PostOAuthCallback(c *gin.Context) { + m.handler.PostOAuthCallback(c) +}