From cf1d0f23ccb51c5f02bd4f08b707cddce365181a Mon Sep 17 00:00:00 2001
From: song <slovx2@gmail.com>
Date: Mon, 29 Dec 2025 01:25:09 +0800
Subject: [PATCH] =?UTF-8?q?feat(antigravity):=20=E6=B7=BB=E5=8A=A0?=
 =?UTF-8?q?=E8=B4=A6=E6=88=B7=E7=B1=BB=E5=9E=8B=EF=BC=88tier=EF=BC=89?=
 =?UTF-8?q?=E6=98=BE=E7=A4=BA=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/internal/pkg/antigravity/client.go    | 23 ++++++++-
 .../service/antigravity_quota_refresher.go    | 21 ++++++++-
 .../components/account/AccountUsageCell.vue   | 47 +++++++++++++++++++
 frontend/src/i18n/locales/en.ts               |  5 ++
 frontend/src/i18n/locales/zh.ts               |  5 ++
 5 files changed, 99 insertions(+), 2 deletions(-)

diff --git a/backend/internal/pkg/antigravity/client.go b/backend/internal/pkg/antigravity/client.go
index 7a2f6ca1..9e65fd72 100644
--- a/backend/internal/pkg/antigravity/client.go
+++ b/backend/internal/pkg/antigravity/client.go
@@ -36,9 +36,30 @@ type LoadCodeAssistRequest struct {
 	} `json:"metadata"`
 }
 
+// TierInfo 账户类型信息
+type TierInfo struct {
+	ID          string `json:"id"`          // standard-tier, free-tier, g1-pro-tier, g1-ultra-tier
+	Name        string `json:"name"`        // 显示名称
+	Description string `json:"description"` // 描述
+}
+
 // LoadCodeAssistResponse loadCodeAssist 响应
 type LoadCodeAssistResponse struct {
-	CloudAICompanionProject string `json:"cloudaicompanionProject"`
+	CloudAICompanionProject string    `json:"cloudaicompanionProject"`
+	CurrentTier             *TierInfo `json:"currentTier,omitempty"`
+	PaidTier                *TierInfo `json:"paidTier,omitempty"`
+}
+
+// GetTier 获取账户类型
+// 优先返回 paidTier（付费订阅级别），否则返回 currentTier
+func (r *LoadCodeAssistResponse) GetTier() string {
+	if r.PaidTier != nil && r.PaidTier.ID != "" {
+		return r.PaidTier.ID
+	}
+	if r.CurrentTier != nil {
+		return r.CurrentTier.ID
+	}
+	return ""
 }
 
 // Client Antigravity API 客户端
diff --git a/backend/internal/service/antigravity_quota_refresher.go b/backend/internal/service/antigravity_quota_refresher.go
index ed3ca61a..af98438c 100644
--- a/backend/internal/service/antigravity_quota_refresher.go
+++ b/backend/internal/service/antigravity_quota_refresher.go
@@ -145,8 +145,15 @@ func (r *AntigravityQuotaRefresher) refreshAccountQuota(ctx context.Context, acc
 		}
 	}
 
-	// 调用 API 获取配额
 	client := antigravity.NewClient(proxyURL)
+
+	// 获取账户类型（tier）
+	loadResp, _ := client.LoadCodeAssist(ctx, accessToken)
+	if loadResp != nil {
+		r.updateAccountTier(account, loadResp)
+	}
+
+	// 调用 API 获取配额
 	modelsResp, err := client.FetchAvailableModels(ctx, accessToken, projectID)
 	if err != nil {
 		return err
@@ -170,6 +177,18 @@ func (r *AntigravityQuotaRefresher) isTokenExpired(account *Account) bool {
 	return time.Now().Add(5 * time.Minute).After(*expiresAt)
 }
 
+// updateAccountTier 更新账户类型信息
+func (r *AntigravityQuotaRefresher) updateAccountTier(account *Account, loadResp *antigravity.LoadCodeAssistResponse) {
+	if account.Extra == nil {
+		account.Extra = make(map[string]any)
+	}
+
+	tier := loadResp.GetTier()
+	if tier != "" {
+		account.Extra["tier"] = tier
+	}
+}
+
 // updateAccountQuota 更新账户的配额信息
 func (r *AntigravityQuotaRefresher) updateAccountQuota(account *Account, modelsResp *antigravity.FetchAvailableModelsResponse) {
 	if account.Extra == nil {
diff --git a/frontend/src/components/account/AccountUsageCell.vue b/frontend/src/components/account/AccountUsageCell.vue
index f46f41be..ea222c33 100644
--- a/frontend/src/components/account/AccountUsageCell.vue
+++ b/frontend/src/components/account/AccountUsageCell.vue
@@ -95,6 +95,18 @@
 
     <!-- Antigravity OAuth accounts: show quota from extra field -->
     <template v-else-if="account.platform === 'antigravity' && account.type === 'oauth'">
+      <!-- 账户类型徽章 -->
+      <div v-if="antigravityTierLabel" class="mb-1">
+        <span
+          :class="[
+            'inline-block rounded px-1.5 py-0.5 text-[10px] font-medium',
+            antigravityTierClass
+          ]"
+        >
+          {{ antigravityTierLabel }}
+        </span>
+      </div>
+
       <div v-if="hasAntigravityQuota" class="space-y-1">
         <!-- Gemini 3 Pro -->
         <UsageProgressBar
@@ -391,6 +403,41 @@ const antigravityClaude45Usage = computed(() =>
   getAntigravityUsage(['claude-sonnet-4-5', 'claude-opus-4-5-thinking'])
 )
 
+// Antigravity 账户类型
+const antigravityTier = computed(() => {
+  const extra = props.account.extra as Record<string, unknown> | undefined
+  if (!extra || typeof extra.tier !== 'string') return null
+  return extra.tier as string
+})
+
+// 账户类型显示标签
+const antigravityTierLabel = computed(() => {
+  switch (antigravityTier.value) {
+    case 'free-tier':
+      return t('admin.accounts.tier.free')
+    case 'g1-pro-tier':
+      return t('admin.accounts.tier.pro')
+    case 'g1-ultra-tier':
+      return t('admin.accounts.tier.ultra')
+    default:
+      return null
+  }
+})
+
+// 账户类型徽章样式
+const antigravityTierClass = computed(() => {
+  switch (antigravityTier.value) {
+    case 'free-tier':
+      return 'bg-gray-100 text-gray-600 dark:bg-gray-700 dark:text-gray-300'
+    case 'g1-pro-tier':
+      return 'bg-blue-100 text-blue-600 dark:bg-blue-900/40 dark:text-blue-300'
+    case 'g1-ultra-tier':
+      return 'bg-purple-100 text-purple-600 dark:bg-purple-900/40 dark:text-purple-300'
+    default:
+      return ''
+  }
+})
+
 const loadUsage = async () => {
   // Fetch usage for Anthropic OAuth and Setup Token accounts
   // OpenAI usage comes from account.extra field (updated during forwarding)
diff --git a/frontend/src/i18n/locales/en.ts b/frontend/src/i18n/locales/en.ts
index f1816c53..d43338ee 100644
--- a/frontend/src/i18n/locales/en.ts
+++ b/frontend/src/i18n/locales/en.ts
@@ -1162,6 +1162,11 @@ export default {
         gemini3Flash: 'G3F',
         gemini3Image: 'G3I',
         claude45: 'C4.5'
+      },
+      tier: {
+        free: 'Free',
+        pro: 'Pro',
+        ultra: 'Ultra'
       }
     },
 
diff --git a/frontend/src/i18n/locales/zh.ts b/frontend/src/i18n/locales/zh.ts
index 4388d79b..68e53db2 100644
--- a/frontend/src/i18n/locales/zh.ts
+++ b/frontend/src/i18n/locales/zh.ts
@@ -972,6 +972,11 @@ export default {
         gemini3Image: 'G3I',
         claude45: 'C4.5'
       },
+      tier: {
+        free: 'Free',
+        pro: 'Pro',
+        ultra: 'Ultra'
+      },
       form: {
         nameLabel: '账号名称',
         namePlaceholder: '请输入账号名称',