优化 Antigravity 配额展示顺序

2026-10-08 18:37:46 +08:00 · 2026-07-08 22:34:29 +08:00
parent accd250226
commit 9ec48ab6b9
6 changed files with 266 additions and 78 deletions
@@ -0,0 +1,77 @@
+import { describe, expect, it } from 'vitest'
+
+import {
+  compareAntigravityQuotaItems,
+  resolveAntigravityQuotaLabel,
+} from '@/features/providers/utils/antigravityQuota'
+
+describe('antigravityQuota', () => {
+  it('uses canonical labels from model ids before stale upstream labels', () => {
+    const opaqueDisplayIndex = { value: 1 }
+
+    expect(
+      resolveAntigravityQuotaLabel(
+        'gemini-2.5-flash',
+        'Gemini 3.1 Flash Lite',
+        opaqueDisplayIndex,
+      ),
+    ).toBe('Gemini 2.5 Flash')
+    expect(
+      resolveAntigravityQuotaLabel(
+        'claude-sonnet-4-6',
+        'Claude Sonnet 4.6 (Thinking)',
+        opaqueDisplayIndex,
+      ),
+    ).toBe('Claude Sonnet 4.6')
+    expect(
+      resolveAntigravityQuotaLabel(
+        'RateLimitResetCredit_05cbb6eeeb9c81918e011d8300f9ebfb',
+        'RateLimitResetCredit_05cbb6eeeb9c81918e011d8300f9ebfb',
+        opaqueDisplayIndex,
+      ),
+    ).toBe('Key-1')
+  })
+
+  it('keeps latest Gemini and Claude quota buckets before low-priority buckets', () => {
+    const items = [
+      {
+        model: 'tab_flash_lite_preview',
+        label: 'Tab Flash Lite Preview',
+        remainingPercent: 1,
+        resetSeconds: 10,
+      },
+      {
+        model: 'gemini-3.5-flash-low',
+        label: 'Gemini 3.5 Flash Low',
+        remainingPercent: 90,
+        resetSeconds: null,
+      },
+      {
+        model: 'chat_20706',
+        label: 'chat_20706',
+        remainingPercent: 0,
+        resetSeconds: 1,
+      },
+      {
+        model: 'claude-opus-4-6-thinking',
+        label: 'Claude Opus 4.6 Thinking',
+        remainingPercent: 100,
+        resetSeconds: null,
+      },
+      {
+        model: 'gemini-2.5-flash-lite',
+        label: 'Gemini 2.5 Flash Lite',
+        remainingPercent: 5,
+        resetSeconds: 5,
+      },
+    ].sort(compareAntigravityQuotaItems)
+
+    expect(items.map(item => item.model)).toEqual([
+      'claude-opus-4-6-thinking',
+      'gemini-3.5-flash-low',
+      'gemini-2.5-flash-lite',
+      'tab_flash_lite_preview',
+      'chat_20706',
+    ])
+  })
+})
@@ -0,0 +1,100 @@
+export interface AntigravityQuotaSortableItem {
+  model: string
+  label: string
+  remainingPercent: number
+  resetSeconds: number | null
+}
+
+const ANTIGRAVITY_MODEL_LABELS: Record<string, string> = {
+  'claude-opus-4-6-thinking': 'Claude Opus 4.6 Thinking',
+  'claude-sonnet-4-6': 'Claude Sonnet 4.6',
+  'gemini-3.5-flash-low': 'Gemini 3.5 Flash Low',
+  'gemini-3.5-flash-extra-low': 'Gemini 3.5 Flash Extra Low',
+  'gemini-3.1-pro-high': 'Gemini 3.1 Pro High',
+  'gemini-3.1-pro-low': 'Gemini 3.1 Pro Low',
+  'gemini-pro-agent': 'Gemini Pro Agent',
+  'gemini-3.1-flash-image': 'Gemini 3.1 Flash Image',
+  'gemini-3.1-flash-lite': 'Gemini 3.1 Flash Lite',
+  'gemini-3-flash-agent': 'Gemini 3 Flash Agent',
+  'gemini-3-flash': 'Gemini 3 Flash',
+  'gemini-2.5-pro': 'Gemini 2.5 Pro',
+  'gemini-2.5-flash-thinking': 'Gemini 2.5 Flash Thinking',
+  'gemini-2.5-flash': 'Gemini 2.5 Flash',
+  'gemini-2.5-flash-lite': 'Gemini 2.5 Flash Lite',
+  'gpt-oss-120b-medium': 'GPT-OSS 120B Medium',
+  'tab_flash_lite_preview': 'Tab Flash Lite Preview',
+  'tab_jump_flash_lite_preview': 'Tab Jump Flash Lite Preview',
+  'models/proactive-observer': 'Proactive Observer',
+}
+
+const ANTIGRAVITY_MODEL_PRIORITY: Record<string, number> = {
+  'claude-opus-4-6-thinking': 10,
+  'claude-sonnet-4-6': 20,
+  'gemini-3.5-flash-low': 100,
+  'gemini-3.5-flash-extra-low': 110,
+  'gemini-3.1-pro-high': 120,
+  'gemini-3.1-pro-low': 130,
+  'gemini-pro-agent': 140,
+  'gemini-3.1-flash-image': 150,
+  'gemini-3.1-flash-lite': 160,
+  'gemini-3-flash-agent': 170,
+  'gemini-3-flash': 180,
+  'gemini-2.5-pro': 300,
+  'gemini-2.5-flash-thinking': 310,
+  'gemini-2.5-flash': 320,
+  'gemini-2.5-flash-lite': 330,
+  'gpt-oss-120b-medium': 700,
+  'models/proactive-observer': 780,
+  'tab_flash_lite_preview': 800,
+  'tab_jump_flash_lite_preview': 810,
+}
+
+export function isOpaqueAntigravityQuotaIdentifier(value: string): boolean {
+  return value.trim().startsWith('RateLimitResetCredit_')
+}
+
+export function resolveAntigravityQuotaLabel(
+  model: string,
+  rawLabel: unknown,
+  opaqueDisplayIndex: { value: number },
+): string {
+  const normalizedModel = model.trim()
+  const canonical = ANTIGRAVITY_MODEL_LABELS[normalizedModel]
+  if (canonical) return canonical
+
+  const candidate = String(rawLabel || '').trim()
+  if (candidate && !isOpaqueAntigravityQuotaIdentifier(candidate)) return candidate
+  if (isOpaqueAntigravityQuotaIdentifier(normalizedModel) || (candidate && isOpaqueAntigravityQuotaIdentifier(candidate))) {
+    const label = `Key-${opaqueDisplayIndex.value}`
+    opaqueDisplayIndex.value += 1
+    return label
+  }
+  return candidate || normalizedModel
+}
+
+function getAntigravityModelPriority(model: string): number {
+  const normalizedModel = model.trim()
+  const explicit = ANTIGRAVITY_MODEL_PRIORITY[normalizedModel]
+  if (explicit !== undefined) return explicit
+  if (normalizedModel.startsWith('claude-')) return 90
+  if (normalizedModel.startsWith('gemini-3.')) return 200
+  if (normalizedModel.startsWith('gemini-2.')) return 390
+  if (normalizedModel.startsWith('gemini-')) return 490
+  if (normalizedModel.startsWith('gpt-oss-')) return 700
+  if (normalizedModel.startsWith('models/')) return 780
+  if (normalizedModel.startsWith('tab_')) return 800
+  if (normalizedModel.startsWith('chat_')) return 900
+  if (isOpaqueAntigravityQuotaIdentifier(normalizedModel)) return 950
+  return 850
+}
+
+export function compareAntigravityQuotaItems<T extends AntigravityQuotaSortableItem>(
+  a: T,
+  b: T,
+): number {
+  return (getAntigravityModelPriority(a.model) - getAntigravityModelPriority(b.model))
+    || ((a.resetSeconds ?? Number.POSITIVE_INFINITY) - (b.resetSeconds ?? Number.POSITIVE_INFINITY))
+    || (a.remainingPercent - b.remainingPercent)
+    || a.label.localeCompare(b.label)
+    || a.model.localeCompare(b.model)
+}