Merge pull request #3071 from sususu98/fix/antigravity-credits-log

Mark Antigravity credits requests in access logs
This commit is contained in:
Luis Pater
2026-04-29 22:56:41 +08:00
committed by GitHub
2 changed files with 21 additions and 3 deletions

View File

@@ -27,7 +27,10 @@ var aiAPIPrefixes = []string{
"/api/provider/",
}
const skipGinLogKey = "__gin_skip_request_logging__"
const (
skipGinLogKey = "__gin_skip_request_logging__"
creditsUsedKey = "__antigravity_credits_used__"
)
// GinLogrusLogger returns a Gin middleware handler that logs HTTP requests and responses
// using logrus. It captures request details including method, path, status code, latency,
@@ -79,6 +82,9 @@ func GinLogrusLogger() gin.HandlerFunc {
requestID = "--------"
}
logLine := fmt.Sprintf("%3d | %13v | %15s | %-7s \"%s\"", statusCode, latency, clientIP, method, path)
if creditsUsed(c) {
logLine += " [credits]"
}
if errorMessage != "" {
logLine = logLine + " | " + errorMessage
}
@@ -149,3 +155,15 @@ func shouldSkipGinRequestLogging(c *gin.Context) bool {
flag, ok := val.(bool)
return ok && flag
}
func creditsUsed(c *gin.Context) bool {
if c == nil {
return false
}
val, exists := c.Get(creditsUsedKey)
if !exists {
return false
}
flag, ok := val.(bool)
return ok && flag
}

View File

@@ -2268,9 +2268,9 @@ var antigravityBaseURLFallbackOrder = func(auth *cliproxyauth.Auth) []string {
return []string{base}
}
return []string{
antigravityBaseURLProd,
antigravityBaseURLDaily,
antigravitySandboxBaseURLDaily,
antigravityBaseURLProd,
// antigravitySandboxBaseURLDaily,
}
}