@kaitranntt/ccs 8.4.0-dev.1 → 8.4.0-dev.11

package/dist/cliproxy/quota/quota-fetcher/all-accounts-fetcher.js

@@ -0,0 +1,98 @@
+"use strict";
+/**
+ * fetchAllProviderQuotas and findAvailableAccount.
+ *
+ * fetchAllProviderQuotas fans quota fetches out across all accounts of a
+ * provider in parallel and groups them by GCP project id (accounts that share
+ * a project pool quota together, so failover between them won't help).
+ * findAvailableAccount wraps that to pick the first account that still has
+ * remaining quota (used by the auto-switch preflight check).
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.findAvailableAccount = exports.fetchAllProviderQuotas = void 0;
+const account_manager_1 = require("../../accounts/account-manager");
+const account_quota_fetcher_1 = require("./account-quota-fetcher");
+const auth_file_reader_1 = require("./auth-file-reader");
+/**
+ * Fetch quota for all accounts of a provider.
+ * Also detects accounts sharing the same GCP project (failover won't help).
+ *
+ * @param provider - Provider name (only 'agy' supported for quota)
+ * @param verbose - Show detailed diagnostics
+ * @returns Results for all accounts with project grouping
+ */
+async function fetchAllProviderQuotas(provider, verbose = false) {
+    const accounts = (0, account_manager_1.getProviderAccounts)(provider);
+    const results = {
+        provider,
+        accounts: [],
+        projectGroups: {},
+        lastUpdated: Date.now(),
+    };
+    if (accounts.length === 0) {
+        return results;
+    }
+    // Fetch quota for each account in parallel
+    const quotaPromises = accounts.map(async (account) => {
+        const quota = await (0, account_quota_fetcher_1.fetchAccountQuota)(provider, account.id, verbose);
+        // Read project ID from auth file if not in quota result
+        let projectId = quota.projectId;
+        if (!projectId) {
+            projectId = (0, auth_file_reader_1.readProjectIdFromAuthFile)(provider, account.id) || undefined;
+        }
+        return {
+            account,
+            quota: { ...quota, accountId: account.id, projectId },
+        };
+    });
+    const quotaResults = await Promise.all(quotaPromises);
+    // Build project groups for detecting shared projects
+    for (const { account, quota } of quotaResults) {
+        results.accounts.push({ account, quota });
+        if (quota.projectId) {
+            if (!results.projectGroups[quota.projectId]) {
+                results.projectGroups[quota.projectId] = [];
+            }
+            results.projectGroups[quota.projectId].push(account.id);
+        }
+    }
+    return results;
+}
+exports.fetchAllProviderQuotas = fetchAllProviderQuotas;
+/**
+ * Find an available account with remaining quota.
+ * Used by preflight check for auto-switching.
+ *
+ * @param provider - Provider name
+ * @param excludeAccountId - Account to exclude (current exhausted account)
+ * @param verbose - Show detailed diagnostics
+ * @returns Account with available quota, or null if none available
+ */
+async function findAvailableAccount(provider, excludeAccountId, verbose = false) {
+    const allQuotas = await fetchAllProviderQuotas(provider, verbose);
+    // Get excluded account's project ID to avoid switching to same-project accounts
+    const excludedProjectId = allQuotas.accounts.find((a) => a.account.id === excludeAccountId)?.quota
+        .projectId;
+    for (const { account, quota } of allQuotas.accounts) {
+        // Skip excluded account
+        if (excludeAccountId && account.id === excludeAccountId) {
+            continue;
+        }
+        // Skip failed quota fetches
+        if (!quota.success) {
+            continue;
+        }
+        // Skip accounts sharing the same GCP project (quota is pooled)
+        if (excludedProjectId && quota.projectId === excludedProjectId) {
+            continue;
+        }
+        // Check if any model has remaining quota (> 5% to avoid edge cases)
+        const hasQuota = quota.models.some((m) => m.percentage > 5);
+        if (hasQuota) {
+            return { account, quota };
+        }
+    }
+    return null;
+}
+exports.findAvailableAccount = findAvailableAccount;
+//# sourceMappingURL=all-accounts-fetcher.js.map

package/dist/cliproxy/quota/quota-fetcher/all-accounts-fetcher.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"all-accounts-fetcher.js","sourceRoot":"","sources":["../../../../src/cliproxy/quota/quota-fetcher/all-accounts-fetcher.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;;AAGH,oEAAuF;AAEvF,mEAA4D;AAC5D,yDAA+D;AAG/D;;;;;;;GAOG;AACI,KAAK,UAAU,sBAAsB,CAC1C,QAA0B,EAC1B,OAAO,GAAG,KAAK;IAEf,MAAM,QAAQ,GAAG,IAAA,qCAAmB,EAAC,QAAQ,CAAC,CAAC;IAC/C,MAAM,OAAO,GAA2B;QACtC,QAAQ;QACR,QAAQ,EAAE,EAAE;QACZ,aAAa,EAAE,EAAE;QACjB,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;KACxB,CAAC;IAEF,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,2CAA2C;IAC3C,MAAM,aAAa,GAAG,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;QACnD,MAAM,KAAK,GAAG,MAAM,IAAA,yCAAiB,EAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAErE,wDAAwD;QACxD,IAAI,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC;QAChC,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,SAAS,GAAG,IAAA,4CAAyB,EAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC,IAAI,SAAS,CAAC;QAC3E,CAAC;QAED,OAAO;YACL,OAAO;YACP,KAAK,EAAE,EAAE,GAAG,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,EAAE,SAAS,EAAE;SACtD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAEtD,qDAAqD;IACrD,KAAK,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,YAAY,EAAE,CAAC;QAC9C,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QAE1C,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC5C,OAAO,CAAC,aAAa,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC;YAC9C,CAAC;YACD,OAAO,CAAC,aAAa,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AA/CD,wDA+CC;AAED;;;;;;;;GAQG;AACI,KAAK,UAAU,oBAAoB,CACxC,QAA0B,EAC1B,gBAAyB,EACzB,OAAO,GAAG,KAAK;IAEf,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAElE,gFAAgF;IAChF,MAAM,iBAAiB,GAAG,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,gBAAgB,CAAC,EAAE,KAAK;SAC/F,SAAS,CAAC;IAEb,KAAK,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,SAAS,CAAC,QAAQ,EAAE,CAAC;QACpD,wBAAwB;QACxB,IAAI,gBAAgB,IAAI,OAAO,CAAC,EAAE,KAAK,gBAAgB,EAAE,CAAC;YACxD,SAAS;QACX,CAAC;QAED,4BAA4B;QAC5B,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;YACnB,SAAS;QACX,CAAC;QAED,+DAA+D;QAC/D,IAAI,iBAAiB,IAAI,KAAK,CAAC,SAAS,KAAK,iBAAiB,EAAE,CAAC;YAC/D,SAAS;QACX,CAAC;QAED,oEAAoE;QACpE,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;QAC5D,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAnCD,oDAmCC"}

package/dist/cliproxy/quota/quota-fetcher/auth-file-reader.d.ts

@@ -0,0 +1,22 @@
+/**
+ * Auth file reader for Antigravity quota fetching.
+ *
+ * Reads the local Antigravity auth file (active or paused directory) and
+ * extracts the access token, refresh token, project id, and expiry state.
+ * Falls back to scanning the directory and matching by the embedded email
+ * field when the canonical sanitized filename is not present.
+ */
+import type { CLIProxyProvider } from '../../types';
+import type { AuthData } from './types';
+/**
+ * Read auth data from the auth file (access token, project_id, expiry state).
+ * Checks both active and paused auth directories (quota is needed for paused
+ * accounts too).
+ */
+export declare function readAuthData(provider: CLIProxyProvider, accountId: string): AuthData | null;
+/**
+ * Read project ID directly from auth file without making an API call.
+ * Used for quick project ID comparison in the doctor command.
+ */
+export declare function readProjectIdFromAuthFile(provider: CLIProxyProvider, accountId: string): string | null;
+//# sourceMappingURL=auth-file-reader.d.ts.map

package/dist/cliproxy/quota/quota-fetcher/auth-file-reader.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth-file-reader.d.ts","sourceRoot":"","sources":["../../../../src/cliproxy/quota/quota-fetcher/auth-file-reader.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAMH,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAGpD,OAAO,KAAK,EAAuB,QAAQ,EAAE,MAAM,SAAS,CAAC;AAE7D;;;;GAIG;AACH,wBAAgB,YAAY,CAAC,QAAQ,EAAE,gBAAgB,EAAE,SAAS,EAAE,MAAM,GAAG,QAAQ,GAAG,IAAI,CAyD3F;AAED;;;GAGG;AACH,wBAAgB,yBAAyB,CACvC,QAAQ,EAAE,gBAAgB,EAC1B,SAAS,EAAE,MAAM,GAChB,MAAM,GAAG,IAAI,CAGf"}

package/dist/cliproxy/quota/quota-fetcher/auth-file-reader.js

@@ -0,0 +1,111 @@
+"use strict";
+/**
+ * Auth file reader for Antigravity quota fetching.
+ *
+ * Reads the local Antigravity auth file (active or paused directory) and
+ * extracts the access token, refresh token, project id, and expiry state.
+ * Falls back to scanning the directory and matching by the embedded email
+ * field when the canonical sanitized filename is not present.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.readProjectIdFromAuthFile = exports.readAuthData = void 0;
+const fs = __importStar(require("node:fs"));
+const path = __importStar(require("node:path"));
+const config_generator_1 = require("../../config/config-generator");
+const auth_utils_1 = require("../../auth/auth-utils");
+const account_manager_1 = require("../../accounts/account-manager");
+/**
+ * Read auth data from the auth file (access token, project_id, expiry state).
+ * Checks both active and paused auth directories (quota is needed for paused
+ * accounts too).
+ */
+function readAuthData(provider, accountId) {
+    const authDirs = [(0, config_generator_1.getAuthDir)(), (0, account_manager_1.getPausedDir)()];
+    // Sanitize accountId (email) to match auth file naming: @ and . → _
+    const sanitizedId = (0, auth_utils_1.sanitizeEmail)(accountId);
+    const prefix = provider === 'agy' ? 'antigravity-' : `${provider}-`;
+    const expectedFile = `${prefix}${sanitizedId}.json`;
+    for (const authDir of authDirs) {
+        if (!fs.existsSync(authDir))
+            continue;
+        const filePath = path.join(authDir, expectedFile);
+        // Direct file access (most common case)
+        if (fs.existsSync(filePath)) {
+            try {
+                const content = fs.readFileSync(filePath, 'utf-8');
+                const data = JSON.parse(content);
+                if (!data.access_token)
+                    continue;
+                return {
+                    accessToken: data.access_token,
+                    refreshToken: data.refresh_token || null,
+                    projectId: data.project_id || null,
+                    isExpired: (0, auth_utils_1.isTokenExpired)(data.expired),
+                    expiresAt: data.expired || null,
+                };
+            }
+            catch {
+                continue;
+            }
+        }
+        // Fallback: scan directory for matching email in file content
+        const files = fs.readdirSync(authDir);
+        for (const file of files) {
+            if (file.startsWith(prefix) && file.endsWith('.json')) {
+                const candidatePath = path.join(authDir, file);
+                try {
+                    const content = fs.readFileSync(candidatePath, 'utf-8');
+                    const data = JSON.parse(content);
+                    // Match by email field inside the auth file
+                    if (data.email === accountId && data.access_token) {
+                        return {
+                            accessToken: data.access_token,
+                            refreshToken: data.refresh_token || null,
+                            projectId: data.project_id || null,
+                            isExpired: (0, auth_utils_1.isTokenExpired)(data.expired),
+                            expiresAt: data.expired || null,
+                        };
+                    }
+                }
+                catch {
+                    continue;
+                }
+            }
+        }
+    }
+    return null;
+}
+exports.readAuthData = readAuthData;
+/**
+ * Read project ID directly from auth file without making an API call.
+ * Used for quick project ID comparison in the doctor command.
+ */
+function readProjectIdFromAuthFile(provider, accountId) {
+    const authData = readAuthData(provider, accountId);
+    return authData?.projectId || null;
+}
+exports.readProjectIdFromAuthFile = readProjectIdFromAuthFile;
+//# sourceMappingURL=auth-file-reader.js.map

package/dist/cliproxy/quota/quota-fetcher/auth-file-reader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth-file-reader.js","sourceRoot":"","sources":["../../../../src/cliproxy/quota/quota-fetcher/auth-file-reader.ts"],"names":[],"mappings":";AAAA;;;;;;;GAOG;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,4CAA8B;AAC9B,gDAAkC;AAElC,oEAA2D;AAE3D,sDAAsE;AACtE,oEAA8D;AAG9D;;;;GAIG;AACH,SAAgB,YAAY,CAAC,QAA0B,EAAE,SAAiB;IACxE,MAAM,QAAQ,GAAG,CAAC,IAAA,6BAAU,GAAE,EAAE,IAAA,8BAAY,GAAE,CAAC,CAAC;IAEhD,oEAAoE;IACpE,MAAM,WAAW,GAAG,IAAA,0BAAa,EAAC,SAAS,CAAC,CAAC;IAC7C,MAAM,MAAM,GAAG,QAAQ,KAAK,KAAK,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,GAAG,QAAQ,GAAG,CAAC;IACpE,MAAM,YAAY,GAAG,GAAG,MAAM,GAAG,WAAW,OAAO,CAAC;IAEpD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC;YAAE,SAAS;QAEtC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QAElD,wCAAwC;QACxC,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBACnD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAwB,CAAC;gBACxD,IAAI,CAAC,IAAI,CAAC,YAAY;oBAAE,SAAS;gBACjC,OAAO;oBACL,WAAW,EAAE,IAAI,CAAC,YAAY;oBAC9B,YAAY,EAAE,IAAI,CAAC,aAAa,IAAI,IAAI;oBACxC,SAAS,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI;oBAClC,SAAS,EAAE,IAAA,2BAAc,EAAC,IAAI,CAAC,OAAO,CAAC;oBACvC,SAAS,EAAE,IAAI,CAAC,OAAO,IAAI,IAAI;iBAChC,CAAC;YACJ,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;QACH,CAAC;QAED,8DAA8D;QAC9D,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QACtC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBACtD,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;gBAC/C,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;oBACxD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAwB,CAAC;oBACxD,4CAA4C;oBAC5C,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;wBAClD,OAAO;4BACL,WAAW,EAAE,IAAI,CAAC,YAAY;4BAC9B,YAAY,EAAE,IAAI,CAAC,aAAa,IAAI,IAAI;4BACxC,SAAS,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI;4BAClC,SAAS,EAAE,IAAA,2BAAc,EAAC,IAAI,CAAC,OAAO,CAAC;4BACvC,SAAS,EAAE,IAAI,CAAC,OAAO,IAAI,IAAI;yBAChC,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,SAAS;gBACX,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAzDD,oCAyDC;AAED;;;GAGG;AACH,SAAgB,yBAAyB,CACvC,QAA0B,EAC1B,SAAiB;IAEjB,MAAM,QAAQ,GAAG,YAAY,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IACnD,OAAO,QAAQ,EAAE,SAAS,IAAI,IAAI,CAAC;AACrC,CAAC;AAND,8DAMC"}

package/dist/cliproxy/quota/quota-fetcher/available-models-fetcher.d.ts

@@ -0,0 +1,17 @@
+/**
+ * fetchAvailableModels call for Antigravity quota.
+ *
+ * Fetches the model -> remaining-fraction map from the Cloud Code internal
+ * API and projects it into ModelQuota[] percentages (0-100). The projectId
+ * is intentionally NOT sent in the body (CLIProxyAPI sends an empty {} body
+ * for this endpoint); it is accepted only for symmetry with the project
+ * lookup flow.
+ */
+import type { QuotaResult } from './types';
+/**
+ * Fetch available models with quota info.
+ * Note: projectId is kept for potential future use but not sent in body
+ * (CLIProxyAPI sends empty {} body for this endpoint).
+ */
+export declare function fetchAvailableModels(accountId: string, accessToken: string, _projectId: string): Promise<QuotaResult>;
+//# sourceMappingURL=available-models-fetcher.d.ts.map

package/dist/cliproxy/quota/quota-fetcher/available-models-fetcher.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"available-models-fetcher.d.ts","sourceRoot":"","sources":["../../../../src/cliproxy/quota/quota-fetcher/available-models-fetcher.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAMH,OAAO,KAAK,EAA4C,WAAW,EAAE,MAAM,SAAS,CAAC;AAErF;;;;GAIG;AACH,wBAAsB,oBAAoB,CACxC,SAAS,EAAE,MAAM,EACjB,WAAW,EAAE,MAAM,EACnB,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,WAAW,CAAC,CAyEtB"}

package/dist/cliproxy/quota/quota-fetcher/available-models-fetcher.js

@@ -0,0 +1,85 @@
+"use strict";
+/**
+ * fetchAvailableModels call for Antigravity quota.
+ *
+ * Fetches the model -> remaining-fraction map from the Cloud Code internal
+ * API and projects it into ModelQuota[] percentages (0-100). The projectId
+ * is intentionally NOT sent in the body (CLIProxyAPI sends an empty {} body
+ * for this endpoint); it is accepted only for symmetry with the project
+ * lookup flow.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.fetchAvailableModels = void 0;
+const provider_entitlement_evidence_1 = require("../../auth/provider-entitlement-evidence");
+const constants_1 = require("./constants");
+const http_client_1 = require("./http-client");
+const status_classifier_1 = require("./status-classifier");
+/**
+ * Fetch available models with quota info.
+ * Note: projectId is kept for potential future use but not sent in body
+ * (CLIProxyAPI sends empty {} body for this endpoint).
+ */
+async function fetchAvailableModels(accountId, accessToken, _projectId) {
+    const url = `${constants_1.ANTIGRAVITY_API_BASE}/${constants_1.ANTIGRAVITY_API_VERSION}:fetchAvailableModels`;
+    const response = await (0, http_client_1.performAntigravityRequest)(accountId, accessToken, url, constants_1.FETCHMODELS_HEADERS, JSON.stringify({}));
+    if (response.status < 200 || response.status >= 300) {
+        return {
+            success: false,
+            models: [],
+            lastUpdated: Date.now(),
+            ...(0, status_classifier_1.buildAntigravityFailure)(response.status, response.bodyText),
+        };
+    }
+    const data = response.json;
+    if (!data) {
+        return {
+            success: false,
+            models: [],
+            lastUpdated: Date.now(),
+            error: 'Invalid quota response from provider',
+            errorCode: 'provider_unavailable',
+            retryable: true,
+            entitlement: (0, provider_entitlement_evidence_1.buildProviderEntitlementEvidence)({
+                normalizedTier: 'unknown',
+                source: 'runtime_inference',
+                confidence: 'low',
+                accessState: 'unknown',
+                capacityState: 'temporarily_unavailable',
+                notes: 'Provider returned a 2xx response with an empty or invalid quota payload.',
+            }),
+        };
+    }
+    const models = [];
+    if (data.models && typeof data.models === 'object') {
+        for (const [modelId, modelData] of Object.entries(data.models)) {
+            const quotaInfo = modelData.quotaInfo || modelData.quota_info;
+            if (!quotaInfo)
+                continue;
+            const remaining = quotaInfo.remainingFraction ?? quotaInfo.remaining_fraction ?? quotaInfo.remaining;
+            const resetTime = quotaInfo.resetTime || quotaInfo.reset_time || null;
+            let percentage;
+            if (typeof remaining === 'number' && isFinite(remaining)) {
+                percentage = Math.max(0, Math.min(100, Math.round(remaining * 100)));
+            }
+            else if (resetTime) {
+                percentage = 0;
+            }
+            else {
+                continue;
+            }
+            models.push({
+                name: modelId,
+                displayName: modelData.displayName,
+                percentage,
+                resetTime,
+            });
+        }
+    }
+    return {
+        success: true,
+        models,
+        lastUpdated: Date.now(),
+    };
+}
+exports.fetchAvailableModels = fetchAvailableModels;
+//# sourceMappingURL=available-models-fetcher.js.map

package/dist/cliproxy/quota/quota-fetcher/available-models-fetcher.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"available-models-fetcher.js","sourceRoot":"","sources":["../../../../src/cliproxy/quota/quota-fetcher/available-models-fetcher.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;;AAEH,4FAA4F;AAC5F,2CAAiG;AACjG,+CAA0D;AAC1D,2DAA8D;AAG9D;;;;GAIG;AACI,KAAK,UAAU,oBAAoB,CACxC,SAAiB,EACjB,WAAmB,EACnB,UAAkB;IAElB,MAAM,GAAG,GAAG,GAAG,gCAAoB,IAAI,mCAAuB,uBAAuB,CAAC;IACtF,MAAM,QAAQ,GAAG,MAAM,IAAA,uCAAyB,EAC9C,SAAS,EACT,WAAW,EACX,GAAG,EACH,+BAAmB,EACnB,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CACnB,CAAC;IAEF,IAAI,QAAQ,CAAC,MAAM,GAAG,GAAG,IAAI,QAAQ,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;QACpD,OAAO;YACL,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,EAAE;YACV,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;YACvB,GAAG,IAAA,2CAAuB,EAAC,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,QAAQ,CAAC;SAC/D,CAAC;IACJ,CAAC;IAED,MAAM,IAAI,GAAG,QAAQ,CAAC,IAA2C,CAAC;IAClE,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO;YACL,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,EAAE;YACV,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;YACvB,KAAK,EAAE,sCAAsC;YAC7C,SAAS,EAAE,sBAAsB;YACjC,SAAS,EAAE,IAAI;YACf,WAAW,EAAE,IAAA,gEAAgC,EAAC;gBAC5C,cAAc,EAAE,SAAS;gBACzB,MAAM,EAAE,mBAAmB;gBAC3B,UAAU,EAAE,KAAK;gBACjB,WAAW,EAAE,SAAS;gBACtB,aAAa,EAAE,yBAAyB;gBACxC,KAAK,EAAE,0EAA0E;aAClF,CAAC;SACH,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAiB,EAAE,CAAC;IAEhC,IAAI,IAAI,CAAC,MAAM,IAAI,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QACnD,KAAK,MAAM,CAAC,OAAO,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAC/D,MAAM,SAAS,GAAG,SAAS,CAAC,SAAS,IAAI,SAAS,CAAC,UAAU,CAAC;YAC9D,IAAI,CAAC,SAAS;gBAAE,SAAS;YAEzB,MAAM,SAAS,GACb,SAAS,CAAC,iBAAiB,IAAI,SAAS,CAAC,kBAAkB,IAAI,SAAS,CAAC,SAAS,CAAC;YACrF,MAAM,SAAS,GAAG,SAAS,CAAC,SAAS,IAAI,SAAS,CAAC,UAAU,IAAI,IAAI,CAAC;YAEtE,IAAI,UAAkB,CAAC;YACvB,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBACzD,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;YACvE,CAAC;iBAAM,IAAI,SAAS,EAAE,CAAC;gBACrB,UAAU,GAAG,CAAC,CAAC;YACjB,CAAC;iBAAM,CAAC;gBACN,SAAS;YACX,CAAC;YAED,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,SAAS,CAAC,WAAW;gBAClC,UAAU;gBACV,SAAS;aACV,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO;QACL,OAAO,EAAE,IAAI;QACb,MAAM;QACN,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;KACxB,CAAC;AACJ,CAAC;AA7ED,oDA6EC"}

package/dist/cliproxy/quota/quota-fetcher/constants.d.ts

@@ -0,0 +1,25 @@
+/**
+ * Constants for the Antigravity quota fetcher.
+ *
+ * Google Cloud Code internal API endpoints, fixed headers used by the
+ * CLIProxyAPIPlus control-plane requests, and the shared timeout applied to
+ * every Antigravity management API call.
+ */
+/** Google Cloud Code API endpoints */
+export declare const ANTIGRAVITY_DAILY_API_BASE = "https://daily-cloudcode-pa.googleapis.com";
+export declare const ANTIGRAVITY_API_BASE = "https://cloudcode-pa.googleapis.com";
+export declare const ANTIGRAVITY_API_VERSION = "v1internal";
+export declare const ANTIGRAVITY_LOADCODEASSIST_BASE_URLS: readonly ["https://daily-cloudcode-pa.googleapis.com", "https://cloudcode-pa.googleapis.com"];
+export declare const MANAGEMENT_API_TIMEOUT_MS = 5000;
+/** Headers for loadCodeAssist (matches current CLIProxyAPIPlus control-plane requests) */
+export declare const LOADCODEASSIST_HEADERS: {
+    'Content-Type': string;
+    'User-Agent': string;
+    'X-Goog-Api-Client': string;
+};
+/** Headers for fetchAvailableModels (matches CLIProxyAPI antigravity_executor.go) */
+export declare const FETCHMODELS_HEADERS: {
+    'Content-Type': string;
+    'User-Agent': string;
+};
+//# sourceMappingURL=constants.d.ts.map

package/dist/cliproxy/quota/quota-fetcher/constants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../../../../src/cliproxy/quota/quota-fetcher/constants.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,sCAAsC;AACtC,eAAO,MAAM,0BAA0B,8CAA8C,CAAC;AACtF,eAAO,MAAM,oBAAoB,wCAAwC,CAAC;AAC1E,eAAO,MAAM,uBAAuB,eAAe,CAAC;AACpD,eAAO,MAAM,oCAAoC,+FAGvC,CAAC;AACX,eAAO,MAAM,yBAAyB,OAAO,CAAC;AAE9C,0FAA0F;AAC1F,eAAO,MAAM,sBAAsB;;;;CAIlC,CAAC;AAEF,qFAAqF;AACrF,eAAO,MAAM,mBAAmB;;;CAG/B,CAAC"}

package/dist/cliproxy/quota/quota-fetcher/constants.js

@@ -0,0 +1,31 @@
+"use strict";
+/**
+ * Constants for the Antigravity quota fetcher.
+ *
+ * Google Cloud Code internal API endpoints, fixed headers used by the
+ * CLIProxyAPIPlus control-plane requests, and the shared timeout applied to
+ * every Antigravity management API call.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FETCHMODELS_HEADERS = exports.LOADCODEASSIST_HEADERS = exports.MANAGEMENT_API_TIMEOUT_MS = exports.ANTIGRAVITY_LOADCODEASSIST_BASE_URLS = exports.ANTIGRAVITY_API_VERSION = exports.ANTIGRAVITY_API_BASE = exports.ANTIGRAVITY_DAILY_API_BASE = void 0;
+/** Google Cloud Code API endpoints */
+exports.ANTIGRAVITY_DAILY_API_BASE = 'https://daily-cloudcode-pa.googleapis.com';
+exports.ANTIGRAVITY_API_BASE = 'https://cloudcode-pa.googleapis.com';
+exports.ANTIGRAVITY_API_VERSION = 'v1internal';
+exports.ANTIGRAVITY_LOADCODEASSIST_BASE_URLS = [
+    exports.ANTIGRAVITY_DAILY_API_BASE,
+    exports.ANTIGRAVITY_API_BASE,
+];
+exports.MANAGEMENT_API_TIMEOUT_MS = 5000;
+/** Headers for loadCodeAssist (matches current CLIProxyAPIPlus control-plane requests) */
+exports.LOADCODEASSIST_HEADERS = {
+    'Content-Type': 'application/json',
+    'User-Agent': 'antigravity/1.21.9 darwin/arm64 google-api-nodejs-client/10.3.0',
+    'X-Goog-Api-Client': 'gl-node/22.21.1',
+};
+/** Headers for fetchAvailableModels (matches CLIProxyAPI antigravity_executor.go) */
+exports.FETCHMODELS_HEADERS = {
+    'Content-Type': 'application/json',
+    'User-Agent': 'antigravity/1.104.0 darwin/arm64',
+};
+//# sourceMappingURL=constants.js.map

package/dist/cliproxy/quota/quota-fetcher/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../../src/cliproxy/quota/quota-fetcher/constants.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;AAEH,sCAAsC;AACzB,QAAA,0BAA0B,GAAG,2CAA2C,CAAC;AACzE,QAAA,oBAAoB,GAAG,qCAAqC,CAAC;AAC7D,QAAA,uBAAuB,GAAG,YAAY,CAAC;AACvC,QAAA,oCAAoC,GAAG;IAClD,kCAA0B;IAC1B,4BAAoB;CACZ,CAAC;AACE,QAAA,yBAAyB,GAAG,IAAI,CAAC;AAE9C,0FAA0F;AAC7E,QAAA,sBAAsB,GAAG;IACpC,cAAc,EAAE,kBAAkB;IAClC,YAAY,EAAE,iEAAiE;IAC/E,mBAAmB,EAAE,iBAAiB;CACvC,CAAC;AAEF,qFAAqF;AACxE,QAAA,mBAAmB,GAAG;IACjC,cAAc,EAAE,kBAAkB;IAClC,YAAY,EAAE,kCAAkC;CACjD,CAAC"}

package/dist/cliproxy/quota/quota-fetcher/http-client.d.ts

@@ -0,0 +1,29 @@
+/**
+ * HTTP transport for Antigravity quota requests.
+ *
+ * Wraps fetch() with three fallback strategies:
+ *  1. Direct call with the local access token.
+ *  2. If that returns 401 (token rejected), retry through CLIProxy management
+ *     auth using the proxy's stored token.
+ *  3. For loadCodeAssist, fall back across the daily then prod Cloud Code hosts.
+ *
+ * Every call is bounded by MANAGEMENT_API_TIMEOUT_MS via an AbortController.
+ * Network errors become synthetic 503 responses; abort timeouts become 408.
+ */
+import type { ManagedResponse } from './types';
+/** Best-effort JSON.parse; returns null on failure. */
+export declare function safeParseJson(bodyText: string): unknown;
+/**
+ * Perform a single Antigravity POST. Tries direct with the local access token
+ * first; on 401, retries through CLIProxy management auth. Network errors map
+ * to synthetic 503 responses so the caller's status-based classifier still
+ * works; abort timeouts map to 408.
+ */
+export declare function performAntigravityRequest(accountId: string, accessToken: string, url: string, headers: Record<string, string>, body: string): Promise<ManagedResponse>;
+/**
+ * Run a request against each base URL in order, returning the first 2xx
+ * response. If none succeed, return the last failure (or a synthetic 503 if
+ * no URLs were attempted).
+ */
+export declare function performAntigravityRequestWithBaseUrlFallback(accountId: string, accessToken: string, baseUrls: readonly string[], apiPath: string, headers: Record<string, string>, body: string): Promise<ManagedResponse>;
+//# sourceMappingURL=http-client.d.ts.map

package/dist/cliproxy/quota/quota-fetcher/http-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"http-client.d.ts","sourceRoot":"","sources":["../../../../src/cliproxy/quota/quota-fetcher/http-client.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AASH,OAAO,KAAK,EAAE,eAAe,EAAiD,MAAM,SAAS,CAAC;AAE9F,uDAAuD;AACvD,wBAAgB,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAMvD;AA6HD;;;;;GAKG;AACH,wBAAsB,yBAAyB,CAC7C,SAAS,EAAE,MAAM,EACjB,WAAW,EAAE,MAAM,EACnB,GAAG,EAAE,MAAM,EACX,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAC/B,IAAI,EAAE,MAAM,GACX,OAAO,CAAC,eAAe,CAAC,CA0C1B;AAED;;;;GAIG;AACH,wBAAsB,4CAA4C,CAChE,SAAS,EAAE,MAAM,EACjB,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,SAAS,MAAM,EAAE,EAC3B,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAC/B,IAAI,EAAE,MAAM,GACX,OAAO,CAAC,eAAe,CAAC,CAyB1B"}