Function fetchAntigravityUsage

packages/arctic/src/provider/usage.ts:1515–1603

(input: {
    provider: Provider.Info
    sessionID?: string
    timePeriod?: TimePeriod
  })

  }

  async function fetchAntigravityUsage(input: {
    provider: Provider.Info
    sessionID?: string
    timePeriod?: TimePeriod
  }): Promise<Omit<Record, "providerID" | "providerName" | "fetchedAt">> {
    const auth = await Auth.get(input.provider.id)
    if (!auth || auth.type !== "oauth") {
      throw new Error("Antigravity OAuth authentication is required. Run `arctic auth login` and select Antigravity.")
    }

    const accessToken = await ensureAntigravityAccessToken(auth, input.provider.id)

    // Fetch quota information from Antigravity API
    const quotas = await fetchAntigravityModels(accessToken)

    if (quotas.length === 0) {
      return {
        planType: "Antigravity (No quota information available)",
        credits: {
          hasCredits: true,
          unlimited: true,
        },
      }
    }

    // Find the model with the lowest remaining fraction (most constrained)
    let lowestRemaining = 1.0
    let resetTime: string | null = null

    for (const quota of quotas) {
      const remaining = quota.remainingFraction ?? 1.0
      if (remaining < lowestRemaining) {
        lowestRemaining = remaining
        resetTime = quota.resetTime
      }
    }

    // Format quota details for display
    const quotaEntries = quotas
      .map((quota) => {
        const percent = quota.remainingFraction !== null ? Math.round(quota.remainingFraction * 100) : "N/A"
        return `${quota.displayName}: ${percent}% remaining`
      })
      .slice(0, 10) // Limit to first 10 models to avoid clutter

    const quotaDetails = quotaEntries.map((entry) => `- ${entry}`).join("\n")
    const planType = `Antigravity\n${quotaDetails}`

    // Parse reset time if available
    let resetsAt: number | undefined
    if (resetTime) {
      const resetDate = new Date(resetTime)
      if (!Number.isNaN(resetDate.getTime())) {
        resetsAt = Math.floor(resetDate.getTime() / 1000)
      }
    }

    const usedPercent = Math.max(0, Math.min(100, (1 - lowestRemaining) * 100))