fix: treat prolite plan as pro (closes #111)

OpenAI reports the $100 Pro tier as "prolite" — functionally a Pro
plan with a smaller usage cap, but the exact-match plan gating meant
these accounts fell into the default branch for premium 5h rate limit
handling, scheduler score bias, 429 cooldown window detection, and the
Spark model's pro-only account filter, so gpt-5.3-codex-spark requests
could not be served by prolite accounts.

Fold prolite/pro_lite/pro-lite into "pro" via a single NormalizePlanType
helper and route existing callers through it. Raw PlanType is preserved
on the account for UI display and downstream diagnostics.

The frontend usage/filter decisions are normalized in parallel so
prolite accounts render Pro-style 5h+7d usage bars and match the "pro"
plan filter, while the table still shows the raw label.

Author: james-6-23

auth/premium_rate_limit.go

@@ -10,8 +10,24 @@ import (
 
 const premium5hFallbackWindow = 5 * time.Hour
 
+// NormalizePlanType canonicalizes a plan string for behavior-level comparisons.
+// OpenAI reports the $100 Pro tier as "prolite"; functionally it is a Pro plan
+// with a smaller usage cap, so we fold it into "pro" so that downstream plan
+// gating (premium 5h rate-limit, Spark routing, scheduler bias, 429 cooldown
+// window) treats it identically. The raw value is kept in Account.PlanType so
+// the UI can still render "prolite" for operator visibility.
+func NormalizePlanType(plan string) string {
+	normalized := strings.ToLower(strings.TrimSpace(plan))
+	switch normalized {
+	case "prolite", "pro_lite", "pro-lite":
+		return "pro"
+	default:
+		return normalized
+	}
+}
+
 func normalizePlanType(plan string) string {
-	return strings.ToLower(strings.TrimSpace(plan))
+	return NormalizePlanType(plan)
 }
 
 func isPremium5hPlan(plan string) bool {

auth/premium_rate_limit_test.go

@@ -70,6 +70,38 @@ func TestPremium5hRateLimitedSkipsUsageProbeBeforeReset(t *testing.T) {
 	}
 }
 
+func TestNormalizePlanTypeFoldsProliteIntoPro(t *testing.T) {
+	cases := map[string]string{
+		"prolite":   "pro",
+		"ProLite":   "pro",
+		" prolite ": "pro",
+		"pro_lite":  "pro",
+		"pro-lite":  "pro",
+		"pro":       "pro",
+		"plus":      "plus",
+		"free":      "free",
+		"":          "",
+	}
+	for input, want := range cases {
+		if got := NormalizePlanType(input); got != want {
+			t.Errorf("NormalizePlanType(%q) = %q, want %q", input, got, want)
+		}
+	}
+}
+
+func TestProliteIsTreatedAsPremium5hPlan(t *testing.T) {
+	acc := newPremium5hTestAccount("prolite", time.Now().Add(30*time.Minute))
+	if !acc.IsPremium5hPlan() {
+		t.Fatal("prolite should be recognized as a premium 5h plan")
+	}
+	if !IsPlusOrHigherPlan("prolite") {
+		t.Fatal("prolite should qualify as plus-or-higher for image routing")
+	}
+	if got := defaultScoreBiasForPlan("prolite"); got != 50 {
+		t.Fatalf("defaultScoreBiasForPlan(prolite) = %d, want 50", got)
+	}
+}
+
 func TestCleanByRuntimeStatusSkipsPremium5hRateLimitedAccount(t *testing.T) {
 	acc := newPremium5hTestAccount("plus", time.Now().Add(20*time.Minute))
 	store := &Store{

auth/store.go

@@ -308,7 +308,7 @@ func concurrencyLimitForTier(baseLimit int64, tier AccountHealthTier) int64 {
 }
 
 func defaultScoreBiasForPlan(planType string) int64 {
-	switch strings.ToLower(strings.TrimSpace(planType)) {
+	switch NormalizePlanType(planType) {
 	case "pro", "plus", "team":
 		return 50
 	default:

frontend/src/pages/Accounts.tsx

@@ -119,7 +119,7 @@ export default function Accounts() {
         return false
       }
 
-      const plan = (account.plan_type || '').toLowerCase()
+      const plan = normalizePlanType(account.plan_type)
       const has7d = account.usage_percent_7d !== null && account.usage_percent_7d !== undefined
       const has5h = account.usage_percent_5h !== null && account.usage_percent_5h !== undefined
 
@@ -191,7 +191,7 @@ export default function Accounts() {
     }
     // 套餐过滤
     if (planFilter !== 'all') {
-      const plan = (account.plan_type || '').toLowerCase()
+      const plan = normalizePlanType(account.plan_type)
       if (plan !== planFilter) return false
     }
     // 搜索过滤
@@ -2243,8 +2243,17 @@ function getDispatchScore(account: AccountRow): number {
   return account.dispatch_score ?? account.scheduler_score ?? 0
 }
 
+// OpenAI reports the $100 Pro tier as "prolite" — functionally a Pro plan with
+// a smaller usage cap. Keep behavioral comparisons (usage windows, plan filter,
+// scheduler bias) aligned with the Go side by folding it into "pro".
+function normalizePlanType(planType?: string): string {
+  const raw = (planType || '').toLowerCase().trim()
+  if (raw === 'prolite' || raw === 'pro_lite' || raw === 'pro-lite') return 'pro'
+  return raw
+}
+
 function getDefaultScoreBias(planType?: string): number {
-  switch ((planType || '').toLowerCase()) {
+  switch (normalizePlanType(planType)) {
     case 'pro':
     case 'plus':
     case 'team':
@@ -2817,7 +2826,7 @@ function UsageWindowStat({ label, detail }: { label: string; detail?: AccountRow
 
 // 用量列组件
 function UsageCell({ account }: { account: AccountRow }) {
-  const plan = (account.plan_type || '').toLowerCase()
+  const plan = normalizePlanType(account.plan_type)
   const has7d = account.usage_percent_7d !== null && account.usage_percent_7d !== undefined
   const has5h = account.usage_percent_5h !== null && account.usage_percent_5h !== undefined
   const has7dDetail = hasUsageWindowDetail(account.usage_7d_detail)

proxy/handler.go

@@ -158,7 +158,7 @@ func accountFilterForModel(model string) auth.AccountFilter {
 			return false
 		}
 		if isProOnlyModel(model) {
-			return strings.EqualFold(strings.TrimSpace(account.GetPlanType()), "pro")
+			return auth.NormalizePlanType(account.GetPlanType()) == "pro"
 		}
 		return true
 	}
@@ -2194,7 +2194,7 @@ func compute429Cooldown(account *auth.Account, body []byte, resp *http.Response)
 	}
 
 	// 2. 没有精确重置时间，根据套餐类型 + 用量窗口推断
-	planType := strings.ToLower(account.GetPlanType())
+	planType := auth.NormalizePlanType(account.GetPlanType())
 
 	switch planType {
 	case "free":

proxy/handler_test.go

@@ -349,6 +349,9 @@ func TestAccountFilterForSparkRequiresPro(t *testing.T) {
 	if !filter(&auth.Account{PlanType: "pro"}) {
 		t.Fatal("spark filter should allow pro accounts")
 	}
+	if !filter(&auth.Account{PlanType: "prolite"}) {
+		t.Fatal("spark filter should treat prolite as pro")
+	}
 	if filter(&auth.Account{PlanType: "plus"}) {
 		t.Fatal("spark filter should reject non-pro accounts")
 	}