Update main.go

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
This commit is contained in:
Icereed 2025-01-27 13:33:53 +01:00 committed by GitHub
parent b0737aab50
commit 3bcc05aaa8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -387,7 +387,11 @@ func validateOrDefaultEnvVars() {
// Initialize token limit from environment variable // Initialize token limit from environment variable
if limit := os.Getenv("TOKEN_LIMIT"); limit != "" { if limit := os.Getenv("TOKEN_LIMIT"); limit != "" {
if parsed, err := strconv.Atoi(limit); err == nil { if parsed, err := strconv.Atoi(limit); err == nil {
if parsed < 0 {
log.Fatalf("TOKEN_LIMIT must be non-negative, got: %d", parsed)
}
tokenLimit = parsed tokenLimit = parsed
log.Infof("Using token limit: %d", tokenLimit)
} }
} }
} }