@kaitranntt/ccs 7.14.0-dev.1 → 7.14.0-dev.3

package/dist/cliproxy/quota-manager.js

@@ -0,0 +1,278 @@
+"use strict";
+/**
+ * Quota Manager for Hybrid Auto+Manual Account Selection
+ *
+ * Provides pre-flight quota checking with caching, tier-based failover,
+ * and cooldown tracking for exhausted accounts.
+ *
+ * Key features:
+ * - 30-second in-memory cache for quota results
+ * - Tier-priority failover (ultra > pro by default)
+ * - Cooldown tracking for exhausted accounts
+ * - Respects paused accounts from manual config
+ * - Graceful degradation on API failures
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getQuotaStatus = exports.preflightCheck = exports.findHealthyAccount = exports.clearCooldown = exports.applyCooldown = exports.isOnCooldown = exports.clearQuotaCache = exports.setCachedQuota = exports.getCachedQuota = void 0;
+const quota_fetcher_1 = require("./quota-fetcher");
+const account_manager_1 = require("./account-manager");
+const unified_config_loader_1 = require("../config/unified-config-loader");
+const CACHE_TTL_MS = 30000; // 30 seconds
+const quotaCache = new Map();
+// Request deduplication: track in-flight fetch promises to avoid parallel duplicate requests
+const pendingFetches = new Map();
+function getCacheKey(provider, accountId) {
+    return `${provider}:${accountId}`;
+}
+/**
+ * Get cached quota result if still valid
+ */
+function getCachedQuota(provider, accountId) {
+    const key = getCacheKey(provider, accountId);
+    const entry = quotaCache.get(key);
+    if (!entry)
+        return null;
+    if (Date.now() - entry.timestamp > CACHE_TTL_MS) {
+        quotaCache.delete(key);
+        return null;
+    }
+    return entry.result;
+}
+exports.getCachedQuota = getCachedQuota;
+/**
+ * Cache quota result
+ */
+function setCachedQuota(provider, accountId, result) {
+    const key = getCacheKey(provider, accountId);
+    quotaCache.set(key, { result, timestamp: Date.now() });
+}
+exports.setCachedQuota = setCachedQuota;
+/**
+ * Clear all cached quota results
+ */
+function clearQuotaCache() {
+    quotaCache.clear();
+}
+exports.clearQuotaCache = clearQuotaCache;
+/**
+ * Fetch quota with request deduplication
+ * If a fetch for this account is already in progress, return the existing promise
+ */
+async function fetchQuotaWithDedup(provider, accountId) {
+    const key = getCacheKey(provider, accountId);
+    // Check if fetch already in progress
+    const pending = pendingFetches.get(key);
+    if (pending) {
+        return pending;
+    }
+    // Start new fetch and track it
+    const fetchPromise = (0, quota_fetcher_1.fetchAccountQuota)(provider, accountId)
+        .then((result) => {
+        setCachedQuota(provider, accountId, result);
+        return result;
+    })
+        .catch(() => {
+        return { success: false, models: [], lastUpdated: Date.now() };
+    })
+        .finally(() => {
+        pendingFetches.delete(key);
+    });
+    pendingFetches.set(key, fetchPromise);
+    return fetchPromise;
+}
+const cooldownMap = new Map();
+/**
+ * Check if account is on cooldown
+ */
+function isOnCooldown(provider, accountId) {
+    const key = getCacheKey(provider, accountId);
+    const entry = cooldownMap.get(key);
+    if (!entry)
+        return false;
+    if (Date.now() > entry.until) {
+        cooldownMap.delete(key);
+        return false;
+    }
+    return true;
+}
+exports.isOnCooldown = isOnCooldown;
+/**
+ * Apply cooldown to an exhausted account
+ */
+function applyCooldown(provider, accountId, minutes) {
+    const key = getCacheKey(provider, accountId);
+    cooldownMap.set(key, { until: Date.now() + minutes * 60 * 1000 });
+}
+exports.applyCooldown = applyCooldown;
+/**
+ * Clear cooldown for an account
+ */
+function clearCooldown(provider, accountId) {
+    const key = getCacheKey(provider, accountId);
+    cooldownMap.delete(key);
+}
+exports.clearCooldown = clearCooldown;
+/**
+ * Calculate average quota percentage from models
+ */
+function calculateAverageQuota(quota) {
+    if (!quota.success || quota.models.length === 0) {
+        return 100; // Assume OK if no data
+    }
+    const total = quota.models.reduce((sum, m) => sum + m.percentage, 0);
+    return total / quota.models.length;
+}
+/**
+ * Find healthy account with remaining quota
+ * Respects tier priority and skips paused/cooldown accounts
+ */
+async function findHealthyAccount(provider, exclude) {
+    const config = (0, unified_config_loader_1.loadOrCreateUnifiedConfig)();
+    const tierPriority = config.quota_management?.auto?.tier_priority ?? ['paid'];
+    const threshold = config.quota_management?.auto?.exhaustion_threshold ?? 5;
+    const accounts = (0, account_manager_1.getProviderAccounts)(provider);
+    // Filter available accounts
+    const available = accounts.filter((a) => !exclude.includes(a.id) && !(0, account_manager_1.isAccountPaused)(provider, a.id) && !isOnCooldown(provider, a.id));
+    if (available.length === 0)
+        return null;
+    // Fetch quota for each available account (with caching and deduplication)
+    const withQuotas = await Promise.all(available.map(async (account) => {
+        let quota = getCachedQuota(provider, account.id);
+        if (!quota) {
+            quota = await fetchQuotaWithDedup(provider, account.id);
+        }
+        const avgQuota = calculateAverageQuota(quota);
+        return {
+            id: account.id,
+            tier: account.tier || 'paid',
+            lastQuota: avgQuota,
+        };
+    }));
+    // Filter by threshold
+    const healthy = withQuotas.filter((a) => a.lastQuota >= threshold);
+    if (healthy.length === 0)
+        return null;
+    // Sort by tier priority then quota descending
+    healthy.sort((a, b) => {
+        const tierA = tierPriority.indexOf(a.tier);
+        const tierB = tierPriority.indexOf(b.tier);
+        const tierOrderA = tierA === -1 ? 999 : tierA;
+        const tierOrderB = tierB === -1 ? 999 : tierB;
+        if (tierOrderA !== tierOrderB)
+            return tierOrderA - tierOrderB;
+        return b.lastQuota - a.lastQuota;
+    });
+    return healthy[0];
+}
+exports.findHealthyAccount = findHealthyAccount;
+/**
+ * Find and switch to a healthy account
+ */
+async function findAndSwitch(provider, excludeAccountId, reason) {
+    const alternative = await findHealthyAccount(provider, [excludeAccountId]);
+    if (!alternative) {
+        // No alternatives: use original anyway (graceful degradation)
+        return {
+            proceed: true,
+            accountId: excludeAccountId,
+            reason: `${reason}, no alternatives available`,
+        };
+    }
+    // Switch default
+    (0, account_manager_1.setDefaultAccount)(provider, alternative.id);
+    (0, account_manager_1.touchAccount)(provider, alternative.id);
+    return {
+        proceed: true,
+        accountId: alternative.id,
+        switchedFrom: excludeAccountId,
+        reason,
+        quotaPercent: alternative.lastQuota,
+    };
+}
+/**
+ * Perform pre-flight quota check before session start
+ *
+ * Checks if default account has sufficient quota, auto-switches if needed.
+ * Respects paused accounts, tier priority, and cooldown settings.
+ *
+ * @param provider - CLIProxy provider (only 'agy' supports quota)
+ * @returns PreflightResult with account to use and any switch info
+ */
+async function preflightCheck(provider) {
+    // Only Antigravity supports quota checking
+    if (provider !== 'agy') {
+        const defaultAccount = (0, account_manager_1.getDefaultAccount)(provider);
+        return { proceed: true, accountId: defaultAccount?.id || '' };
+    }
+    const config = (0, unified_config_loader_1.loadOrCreateUnifiedConfig)();
+    const quotaConfig = config.quota_management;
+    // Skip if preflight disabled or mode is manual
+    if (!quotaConfig?.auto?.preflight_check || quotaConfig?.mode === 'manual') {
+        const defaultAccount = (0, account_manager_1.getDefaultAccount)(provider);
+        return { proceed: true, accountId: defaultAccount?.id || '' };
+    }
+    const defaultAccount = (0, account_manager_1.getDefaultAccount)(provider);
+    if (!defaultAccount) {
+        return { proceed: false, accountId: '', reason: 'No accounts configured' };
+    }
+    // Check forced_default override (manual mode)
+    const forcedDefault = quotaConfig.manual?.forced_default;
+    if (forcedDefault) {
+        const forcedAccount = (0, account_manager_1.getProviderAccounts)(provider).find((a) => a.id === forcedDefault);
+        if (forcedAccount) {
+            return { proceed: true, accountId: forcedAccount.id, reason: 'Forced default override' };
+        }
+    }
+    // Check if default is paused
+    if ((0, account_manager_1.isAccountPaused)(provider, defaultAccount.id)) {
+        return await findAndSwitch(provider, defaultAccount.id, 'Default account is paused');
+    }
+    // Check cooldown
+    if (isOnCooldown(provider, defaultAccount.id)) {
+        return await findAndSwitch(provider, defaultAccount.id, 'Default account on cooldown');
+    }
+    // Check quota (with cache and deduplication)
+    let quota = getCachedQuota(provider, defaultAccount.id);
+    if (!quota) {
+        quota = await fetchQuotaWithDedup(provider, defaultAccount.id);
+    }
+    // Calculate average quota
+    const avgQuota = calculateAverageQuota(quota);
+    const threshold = quotaConfig.auto?.exhaustion_threshold ?? 5;
+    if (avgQuota < threshold) {
+        // Apply cooldown to exhausted account
+        applyCooldown(provider, defaultAccount.id, quotaConfig.auto?.cooldown_minutes ?? 5);
+        return await findAndSwitch(provider, defaultAccount.id, `Quota exhausted (${avgQuota.toFixed(1)}%)`);
+    }
+    return {
+        proceed: true,
+        accountId: defaultAccount.id,
+        quotaPercent: avgQuota,
+    };
+}
+exports.preflightCheck = preflightCheck;
+/**
+ * Get quota status for all accounts of a provider
+ * Used by CLI status command
+ */
+async function getQuotaStatus(provider) {
+    const accounts = (0, account_manager_1.getProviderAccounts)(provider);
+    const defaultAccount = (0, account_manager_1.getDefaultAccount)(provider);
+    const results = await Promise.all(accounts.map(async (account) => {
+        let quota = getCachedQuota(provider, account.id);
+        if (!quota && provider === 'agy') {
+            quota = await fetchQuotaWithDedup(provider, account.id);
+        }
+        const avgQuota = quota ? calculateAverageQuota(quota) : 100;
+        return {
+            account,
+            quota: avgQuota,
+            paused: (0, account_manager_1.isAccountPaused)(provider, account.id),
+            onCooldown: isOnCooldown(provider, account.id),
+            isDefault: defaultAccount?.id === account.id,
+        };
+    }));
+    return { accounts: results };
+}
+exports.getQuotaStatus = getQuotaStatus;
+//# sourceMappingURL=quota-manager.js.map

package/dist/cliproxy/quota-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota-manager.js","sourceRoot":"","sources":["../../src/cliproxy/quota-manager.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;GAYG;;;AAGH,mDAAiE;AACjE,uDAO2B;AAC3B,2EAA4E;AAW5E,MAAM,YAAY,GAAG,KAAM,CAAC,CAAC,aAAa;AAC1C,MAAM,UAAU,GAAG,IAAI,GAAG,EAAsB,CAAC;AAEjD,6FAA6F;AAC7F,MAAM,cAAc,GAAG,IAAI,GAAG,EAAgC,CAAC;AAE/D,SAAS,WAAW,CAAC,QAA0B,EAAE,SAAiB;IAChE,OAAO,GAAG,QAAQ,IAAI,SAAS,EAAE,CAAC;AACpC,CAAC;AAED;;GAEG;AACH,SAAgB,cAAc,CAAC,QAA0B,EAAE,SAAiB;IAC1E,MAAM,GAAG,GAAG,WAAW,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAC7C,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAElC,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IAExB,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,SAAS,GAAG,YAAY,EAAE,CAAC;QAChD,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACvB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC,MAAM,CAAC;AACtB,CAAC;AAZD,wCAYC;AAED;;GAEG;AACH,SAAgB,cAAc,CAC5B,QAA0B,EAC1B,SAAiB,EACjB,MAAmB;IAEnB,MAAM,GAAG,GAAG,WAAW,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAC7C,UAAU,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;AACzD,CAAC;AAPD,wCAOC;AAED;;GAEG;AACH,SAAgB,eAAe;IAC7B,UAAU,CAAC,KAAK,EAAE,CAAC;AACrB,CAAC;AAFD,0CAEC;AAED;;;GAGG;AACH,KAAK,UAAU,mBAAmB,CAChC,QAA0B,EAC1B,SAAiB;IAEjB,MAAM,GAAG,GAAG,WAAW,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAE7C,qCAAqC;IACrC,MAAM,OAAO,GAAG,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IACxC,IAAI,OAAO,EAAE,CAAC;QACZ,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,+BAA+B;IAC/B,MAAM,YAAY,GAAG,IAAA,iCAAiB,EAAC,QAAQ,EAAE,SAAS,CAAC;SACxD,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;QACf,cAAc,CAAC,QAAQ,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;QAC5C,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;SACD,KAAK,CAAC,GAAgB,EAAE;QACvB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;IACjE,CAAC,CAAC;SACD,OAAO,CAAC,GAAG,EAAE;QACZ,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEL,cAAc,CAAC,GAAG,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;IACtC,OAAO,YAAY,CAAC;AACtB,CAAC;AAUD,MAAM,WAAW,GAAG,IAAI,GAAG,EAAyB,CAAC;AAErD;;GAEG;AACH,SAAgB,YAAY,CAAC,QAA0B,EAAE,SAAiB;IACxE,MAAM,GAAG,GAAG,WAAW,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAC7C,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAEnC,IAAI,CAAC,KAAK;QAAE,OAAO,KAAK,CAAC;IAEzB,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;QAC7B,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACxB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAZD,oCAYC;AAED;;GAEG;AACH,SAAgB,aAAa,CAC3B,QAA0B,EAC1B,SAAiB,EACjB,OAAe;IAEf,MAAM,GAAG,GAAG,WAAW,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAC7C,WAAW,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,GAAG,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC;AACpE,CAAC;AAPD,sCAOC;AAED;;GAEG;AACH,SAAgB,aAAa,CAAC,QAA0B,EAAE,SAAiB;IACzE,MAAM,GAAG,GAAG,WAAW,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAC7C,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;AAC1B,CAAC;AAHD,sCAGC;AAsBD;;GAEG;AACH,SAAS,qBAAqB,CAAC,KAAkB;IAC/C,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAChD,OAAO,GAAG,CAAC,CAAC,uBAAuB;IACrC,CAAC;IACD,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;IACrE,OAAO,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC;AACrC,CAAC;AAED;;;GAGG;AACI,KAAK,UAAU,kBAAkB,CACtC,QAA0B,EAC1B,OAAiB;IAEjB,MAAM,MAAM,GAAG,IAAA,iDAAyB,GAAE,CAAC;IAC3C,MAAM,YAAY,GAAG,MAAM,CAAC,gBAAgB,EAAE,IAAI,EAAE,aAAa,IAAI,CAAC,MAAM,CAAC,CAAC;IAC9E,MAAM,SAAS,GAAG,MAAM,CAAC,gBAAgB,EAAE,IAAI,EAAE,oBAAoB,IAAI,CAAC,CAAC;IAE3E,MAAM,QAAQ,GAAG,IAAA,qCAAmB,EAAC,QAAQ,CAAC,CAAC;IAE/C,4BAA4B;IAC5B,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAC/B,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,IAAA,iCAAe,EAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,CAC/F,CAAC;IAEF,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAExC,0EAA0E;IAC1E,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,GAAG,CAClC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;QAC9B,IAAI,KAAK,GAAG,cAAc,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;QACjD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,KAAK,GAAG,MAAM,mBAAmB,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,QAAQ,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;QAE9C,OAAO;YACL,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,MAAM;YAC5B,SAAS,EAAE,QAAQ;SACpB,CAAC;IACJ,CAAC,CAAC,CACH,CAAC;IAEF,sBAAsB;IACtB,MAAM,OAAO,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC,CAAC;IACnE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAEtC,8CAA8C;IAC9C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACpB,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC3C,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC3C,MAAM,UAAU,GAAG,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC;QAC9C,MAAM,UAAU,GAAG,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC;QAE9C,IAAI,UAAU,KAAK,UAAU;YAAE,OAAO,UAAU,GAAG,UAAU,CAAC;QAC9D,OAAO,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC;AACpB,CAAC;AApDD,gDAoDC;AAED;;GAEG;AACH,KAAK,UAAU,aAAa,CAC1B,QAA0B,EAC1B,gBAAwB,EACxB,MAAc;IAEd,MAAM,WAAW,GAAG,MAAM,kBAAkB,CAAC,QAAQ,EAAE,CAAC,gBAAgB,CAAC,CAAC,CAAC;IAE3E,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,8DAA8D;QAC9D,OAAO;YACL,OAAO,EAAE,IAAI;YACb,SAAS,EAAE,gBAAgB;YAC3B,MAAM,EAAE,GAAG,MAAM,6BAA6B;SAC/C,CAAC;IACJ,CAAC;IAED,iBAAiB;IACjB,IAAA,mCAAiB,EAAC,QAAQ,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;IAC5C,IAAA,8BAAY,EAAC,QAAQ,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;IAEvC,OAAO;QACL,OAAO,EAAE,IAAI;QACb,SAAS,EAAE,WAAW,CAAC,EAAE;QACzB,YAAY,EAAE,gBAAgB;QAC9B,MAAM;QACN,YAAY,EAAE,WAAW,CAAC,SAAS;KACpC,CAAC;AACJ,CAAC;AAED;;;;;;;;GAQG;AACI,KAAK,UAAU,cAAc,CAAC,QAA0B;IAC7D,2CAA2C;IAC3C,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;QACvB,MAAM,cAAc,GAAG,IAAA,mCAAiB,EAAC,QAAQ,CAAC,CAAC;QACnD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,cAAc,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC;IAChE,CAAC;IAED,MAAM,MAAM,GAAG,IAAA,iDAAyB,GAAE,CAAC;IAC3C,MAAM,WAAW,GAAG,MAAM,CAAC,gBAAgB,CAAC;IAE5C,+CAA+C;IAC/C,IAAI,CAAC,WAAW,EAAE,IAAI,EAAE,eAAe,IAAI,WAAW,EAAE,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC1E,MAAM,cAAc,GAAG,IAAA,mCAAiB,EAAC,QAAQ,CAAC,CAAC;QACnD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,cAAc,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC;IAChE,CAAC;IAED,MAAM,cAAc,GAAG,IAAA,mCAAiB,EAAC,QAAQ,CAAC,CAAC;IACnD,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAC;IAC7E,CAAC;IAED,8CAA8C;IAC9C,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,EAAE,cAAc,CAAC;IACzD,IAAI,aAAa,EAAE,CAAC;QAClB,MAAM,aAAa,GAAG,IAAA,qCAAmB,EAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,aAAa,CAAC,CAAC;QACxF,IAAI,aAAa,EAAE,CAAC;YAClB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,aAAa,CAAC,EAAE,EAAE,MAAM,EAAE,yBAAyB,EAAE,CAAC;QAC3F,CAAC;IACH,CAAC;IAED,6BAA6B;IAC7B,IAAI,IAAA,iCAAe,EAAC,QAAQ,EAAE,cAAc,CAAC,EAAE,CAAC,EAAE,CAAC;QACjD,OAAO,MAAM,aAAa,CAAC,QAAQ,EAAE,cAAc,CAAC,EAAE,EAAE,2BAA2B,CAAC,CAAC;IACvF,CAAC;IAED,iBAAiB;IACjB,IAAI,YAAY,CAAC,QAAQ,EAAE,cAAc,CAAC,EAAE,CAAC,EAAE,CAAC;QAC9C,OAAO,MAAM,aAAa,CAAC,QAAQ,EAAE,cAAc,CAAC,EAAE,EAAE,6BAA6B,CAAC,CAAC;IACzF,CAAC;IAED,6CAA6C;IAC7C,IAAI,KAAK,GAAG,cAAc,CAAC,QAAQ,EAAE,cAAc,CAAC,EAAE,CAAC,CAAC;IACxD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,KAAK,GAAG,MAAM,mBAAmB,CAAC,QAAQ,EAAE,cAAc,CAAC,EAAE,CAAC,CAAC;IACjE,CAAC;IAED,0BAA0B;IAC1B,MAAM,QAAQ,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;IAC9C,MAAM,SAAS,GAAG,WAAW,CAAC,IAAI,EAAE,oBAAoB,IAAI,CAAC,CAAC;IAE9D,IAAI,QAAQ,GAAG,SAAS,EAAE,CAAC;QACzB,sCAAsC;QACtC,aAAa,CAAC,QAAQ,EAAE,cAAc,CAAC,EAAE,EAAE,WAAW,CAAC,IAAI,EAAE,gBAAgB,IAAI,CAAC,CAAC,CAAC;QACpF,OAAO,MAAM,aAAa,CACxB,QAAQ,EACR,cAAc,CAAC,EAAE,EACjB,oBAAoB,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAC5C,CAAC;IACJ,CAAC;IAED,OAAO;QACL,OAAO,EAAE,IAAI;QACb,SAAS,EAAE,cAAc,CAAC,EAAE;QAC5B,YAAY,EAAE,QAAQ;KACvB,CAAC;AACJ,CAAC;AAjED,wCAiEC;AAED;;;GAGG;AACI,KAAK,UAAU,cAAc,CAAC,QAA0B;IAS7D,MAAM,QAAQ,GAAG,IAAA,qCAAmB,EAAC,QAAQ,CAAC,CAAC;IAC/C,MAAM,cAAc,GAAG,IAAA,mCAAiB,EAAC,QAAQ,CAAC,CAAC;IAEnD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;QAC7B,IAAI,KAAK,GAAG,cAAc,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;QACjD,IAAI,CAAC,KAAK,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;YACjC,KAAK,GAAG,MAAM,mBAAmB,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;QAE5D,OAAO;YACL,OAAO;YACP,KAAK,EAAE,QAAQ;YACf,MAAM,EAAE,IAAA,iCAAe,EAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC;YAC7C,UAAU,EAAE,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC;YAC9C,SAAS,EAAE,cAAc,EAAE,EAAE,KAAK,OAAO,CAAC,EAAE;SAC7C,CAAC;IACJ,CAAC,CAAC,CACH,CAAC;IAEF,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;AAC/B,CAAC;AAhCD,wCAgCC"}

package/dist/commands/cliproxy-command.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"cliproxy-command.d.ts","sourceRoot":"","sources":["../../src/commands/cliproxy-command.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;GAkBG;AAuqBH,wBAAsB,qBAAqB,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CA4DzE"}
+{"version":3,"file":"cliproxy-command.d.ts","sourceRoot":"","sources":["../../src/commands/cliproxy-command.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;GAkBG;AA23BH,wBAAsB,qBAAqB,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAiFzE"}

package/dist/commands/cliproxy-command.js

@@ -47,6 +47,7 @@ const path = __importStar(require("path"));
 const auth_handler_1 = require("../cliproxy/auth-handler");
 const account_manager_1 = require("../cliproxy/account-manager");
 const quota_fetcher_1 = require("../cliproxy/quota-fetcher");
+const quota_manager_1 = require("../cliproxy/quota-manager");
 const platform_detector_1 = require("../cliproxy/platform-detector");
 const profile_detector_1 = require("../auth/profile-detector");
 const model_catalog_1 = require("../cliproxy/model-catalog");
@@ -478,6 +479,15 @@ async function showHelp() {
                 ['remove <name>', 'Remove a CLIProxy variant profile'],
             ],
         ],
+        [
+            'Quota Management:',
+            [
+                ['default <account>', 'Set default account for rotation'],
+                ['pause <account>', 'Pause account (skip in rotation)'],
+                ['resume <account>', 'Resume paused account'],
+                ['quota', 'Show quota status for all accounts'],
+            ],
+        ],
         [
             'Proxy Lifecycle:',
             [
@@ -598,6 +608,170 @@ function formatQuotaBar(percentage) {
     return `[${filledChar.repeat(filled)}${' '.repeat(empty)}]`;
 }
 // ============================================================================
+// QUOTA MANAGEMENT COMMANDS
+// ============================================================================
+async function handleSetDefault(args) {
+    await (0, ui_1.initUI)();
+    const parsed = parseProfileArgs(args);
+    if (!parsed.name) {
+        console.log((0, ui_1.fail)('Usage: ccs cliproxy default <account> [--provider <provider>]'));
+        console.log('');
+        console.log('Examples:');
+        console.log('  ccs cliproxy default ultra@gmail.com');
+        console.log('  ccs cliproxy default john --provider agy');
+        process.exit(1);
+    }
+    const provider = (parsed.provider || 'agy');
+    const account = (0, account_manager_1.findAccountByQuery)(provider, parsed.name);
+    if (!account) {
+        console.log((0, ui_1.fail)(`Account not found: ${parsed.name}`));
+        console.log('');
+        const accounts = (0, account_manager_1.getProviderAccounts)(provider);
+        if (accounts.length > 0) {
+            console.log('Available accounts:');
+            for (const acc of accounts) {
+                const badge = acc.isDefault ? (0, ui_1.color)(' (current default)', 'info') : '';
+                console.log(`  - ${acc.email || acc.id}${badge}`);
+            }
+        }
+        else {
+            console.log(`No accounts found for provider: ${provider}`);
+            console.log(`Run: ccs ${provider} --auth`);
+        }
+        process.exit(1);
+    }
+    const success = (0, account_manager_1.setDefaultAccount)(provider, account.id);
+    if (success) {
+        console.log((0, ui_1.ok)(`Default account set to: ${account.email || account.id}`));
+        console.log((0, ui_1.info)(`Provider: ${provider}`));
+    }
+    else {
+        console.log((0, ui_1.fail)('Failed to set default account'));
+        process.exit(1);
+    }
+}
+async function handlePauseAccount(args) {
+    await (0, ui_1.initUI)();
+    const parsed = parseProfileArgs(args);
+    if (!parsed.name) {
+        console.log((0, ui_1.fail)('Usage: ccs cliproxy pause <account> [--provider <provider>]'));
+        console.log('');
+        console.log('Pauses an account so it will be skipped in quota rotation.');
+        process.exit(1);
+    }
+    const provider = (parsed.provider || 'agy');
+    const account = (0, account_manager_1.findAccountByQuery)(provider, parsed.name);
+    if (!account) {
+        console.log((0, ui_1.fail)(`Account not found: ${parsed.name}`));
+        process.exit(1);
+    }
+    if (account.paused) {
+        console.log((0, ui_1.warn)(`Account already paused: ${account.email || account.id}`));
+        console.log((0, ui_1.info)(`Paused at: ${account.pausedAt || 'unknown'}`));
+        return;
+    }
+    const success = (0, account_manager_1.pauseAccount)(provider, account.id);
+    if (success) {
+        console.log((0, ui_1.ok)(`Account paused: ${account.email || account.id}`));
+        console.log((0, ui_1.info)('Account will be skipped in quota rotation'));
+    }
+    else {
+        console.log((0, ui_1.fail)('Failed to pause account'));
+        process.exit(1);
+    }
+}
+async function handleResumeAccount(args) {
+    await (0, ui_1.initUI)();
+    const parsed = parseProfileArgs(args);
+    if (!parsed.name) {
+        console.log((0, ui_1.fail)('Usage: ccs cliproxy resume <account> [--provider <provider>]'));
+        console.log('');
+        console.log('Resumes a paused account for quota rotation.');
+        process.exit(1);
+    }
+    const provider = (parsed.provider || 'agy');
+    const account = (0, account_manager_1.findAccountByQuery)(provider, parsed.name);
+    if (!account) {
+        console.log((0, ui_1.fail)(`Account not found: ${parsed.name}`));
+        process.exit(1);
+    }
+    if (!account.paused) {
+        console.log((0, ui_1.warn)(`Account is not paused: ${account.email || account.id}`));
+        return;
+    }
+    const success = (0, account_manager_1.resumeAccount)(provider, account.id);
+    if (success) {
+        console.log((0, ui_1.ok)(`Account resumed: ${account.email || account.id}`));
+        console.log((0, ui_1.info)('Account is now active in quota rotation'));
+    }
+    else {
+        console.log((0, ui_1.fail)('Failed to resume account'));
+        process.exit(1);
+    }
+}
+async function handleQuotaStatus() {
+    await (0, ui_1.initUI)();
+    console.log((0, ui_1.header)('Quota Status'));
+    console.log('');
+    const provider = 'agy';
+    const accounts = (0, account_manager_1.getProviderAccounts)(provider);
+    if (accounts.length === 0) {
+        console.log((0, ui_1.info)('No Antigravity accounts configured'));
+        console.log(`    Run: ${(0, ui_1.color)('ccs agy --auth', 'command')} to authenticate`);
+        return;
+    }
+    console.log((0, ui_1.dim)('Fetching quotas...'));
+    const quotaResult = await (0, quota_fetcher_1.fetchAllProviderQuotas)(provider);
+    // Build table rows
+    const rows = [];
+    for (const account of accounts) {
+        const quotaData = quotaResult.accounts.find((q) => q.account.id === account.id);
+        const quota = quotaData?.quota;
+        // Calculate average quota
+        let avgQuota = 'N/A';
+        if (quota?.success && quota.models.length > 0) {
+            const avg = Math.round(quota.models.reduce((sum, m) => sum + m.percentage, 0) / quota.models.length);
+            avgQuota = `${avg}%`;
+        }
+        // Build status badges
+        const statusParts = [];
+        if (account.paused)
+            statusParts.push((0, ui_1.color)('PAUSED', 'warning'));
+        if ((0, quota_manager_1.isOnCooldown)(provider, account.id))
+            statusParts.push((0, ui_1.color)('COOLDOWN', 'warning'));
+        const defaultMark = account.isDefault ? (0, ui_1.color)('*', 'success') : ' ';
+        const tier = account.tier || 'unknown';
+        const status = statusParts.join(', ');
+        rows.push([
+            defaultMark,
+            account.nickname || account.email || account.id,
+            tier,
+            avgQuota,
+            status,
+        ]);
+    }
+    console.log('');
+    console.log((0, ui_1.table)(rows, {
+        head: ['', 'Account', 'Tier', 'Quota', 'Status'],
+        colWidths: [3, 30, 10, 10, 20],
+    }));
+    console.log('');
+    console.log((0, ui_1.info)(`Default account marked with ${(0, ui_1.color)('*', 'success')}`));
+    console.log('');
+    // Show summary of paused/cooldown accounts
+    const pausedCount = accounts.filter((a) => a.paused).length;
+    const cooldownCount = accounts.filter((a) => (0, quota_manager_1.isOnCooldown)(provider, a.id)).length;
+    if (pausedCount > 0) {
+        console.log((0, ui_1.warn)(`${pausedCount} account(s) paused - use 'ccs cliproxy resume <account>' to re-enable`));
+    }
+    if (cooldownCount > 0) {
+        console.log((0, ui_1.info)(`${cooldownCount} account(s) on cooldown (exhausted recently)`));
+    }
+    if (pausedCount > 0 || cooldownCount > 0) {
+        console.log('');
+    }
+}
+// ============================================================================
 // MAIN ROUTER
 // ============================================================================
 async function handleCliproxyCommand(args) {
@@ -631,6 +805,23 @@ async function handleCliproxyCommand(args) {
         await handleDoctor();
         return;
     }
+    // Quota management commands
+    if (command === 'default') {
+        await handleSetDefault(args.slice(1));
+        return;
+    }
+    if (command === 'pause') {
+        await handlePauseAccount(args.slice(1));
+        return;
+    }
+    if (command === 'resume') {
+        await handleResumeAccount(args.slice(1));
+        return;
+    }
+    if (command === 'quota') {
+        await handleQuotaStatus();
+        return;
+    }
     const installIdx = args.indexOf('--install');
     if (installIdx !== -1) {
         let version = args[installIdx + 1];