mirror of
https://gitee.com/wanwujie/sub2api
synced 2026-04-03 06:52:13 +08:00
8a260defc2
Replace process-memory sync.Map + per-model runtime state with a single
"AICredits" key in model_rate_limits, making credits exhaustion fully
isomorphic with model-level rate limiting.
Scheduler: rate-limited accounts with overages enabled + credits available
are now scheduled instead of excluded.
Forwarding: when model is rate-limited + credits available, inject credits
proactively without waiting for a 429 round trip.
Storage: credits exhaustion stored as model_rate_limits["AICredits"] with
5h duration, reusing SetModelRateLimit/isRateLimitActiveForKey.
Frontend: show credits_active (yellow ⚡) when model rate-limited but
credits available, credits_exhausted (red) when AICredits key active.
Tests: add unit tests for shouldMarkCreditsExhausted, injectEnabledCreditTypes,
clearCreditsExhausted, and update existing overages tests.
58 lines
1.8 KiB
Go
58 lines
1.8 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
func normalizeAntigravityModelName(model string) string {
|
|
normalized := strings.ToLower(strings.TrimSpace(model))
|
|
normalized = strings.TrimPrefix(normalized, "models/")
|
|
return normalized
|
|
}
|
|
|
|
// resolveAntigravityModelKey 根据请求的模型名解析限流 key
|
|
// 返回空字符串表示无法解析
|
|
func resolveAntigravityModelKey(requestedModel string) string {
|
|
return normalizeAntigravityModelName(requestedModel)
|
|
}
|
|
|
|
// IsSchedulableForModel 结合模型级限流判断是否可调度。
|
|
// 保持旧签名以兼容既有调用方；默认使用 context.Background()。
|
|
func (a *Account) IsSchedulableForModel(requestedModel string) bool {
|
|
return a.IsSchedulableForModelWithContext(context.Background(), requestedModel)
|
|
}
|
|
|
|
func (a *Account) IsSchedulableForModelWithContext(ctx context.Context, requestedModel string) bool {
|
|
if a == nil {
|
|
return false
|
|
}
|
|
if !a.IsSchedulable() {
|
|
return false
|
|
}
|
|
if a.isModelRateLimitedWithContext(ctx, requestedModel) {
|
|
// Antigravity + overages 启用 + 积分未耗尽 → 放行（有积分可用）
|
|
if a.Platform == PlatformAntigravity && a.IsOveragesEnabled() && !a.isCreditsExhausted() {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
// GetRateLimitRemainingTime 获取限流剩余时间（模型级限流）
|
|
// 返回 0 表示未限流或已过期
|
|
func (a *Account) GetRateLimitRemainingTime(requestedModel string) time.Duration {
|
|
return a.GetRateLimitRemainingTimeWithContext(context.Background(), requestedModel)
|
|
}
|
|
|
|
// GetRateLimitRemainingTimeWithContext 获取限流剩余时间（模型级限流）
|
|
// 返回 0 表示未限流或已过期
|
|
func (a *Account) GetRateLimitRemainingTimeWithContext(ctx context.Context, requestedModel string) time.Duration {
|
|
if a == nil {
|
|
return 0
|
|
}
|
|
return a.GetModelRateLimitRemainingTimeWithContext(ctx, requestedModel)
|
|
}
|