logging: mark antigravity credits requests

This commit is contained in:
sususu98
2026-04-26 23:04:06 +08:00
parent c5bea6f6f8
commit 6fc23568df
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

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