优化 Antigravity 配额展示顺序

2026-10-11 11:49:50 +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,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)
+}