优化 Antigravity 配额展示顺序

2026-10-09 18:59: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,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',
+    ])
+  })
+})