opencode-antigravity-auth 1.0.3 → 1.0.5

package/dist/src/plugin/debug.js

@@ -0,0 +1,142 @@
+import { createWriteStream } from "node:fs";
+import { join } from "node:path";
+import { cwd, env } from "node:process";
+const DEBUG_FLAG = env.OPENCODE_ANTIGRAVITY_DEBUG ?? "";
+const MAX_BODY_PREVIEW_CHARS = 12000;
+const debugEnabled = DEBUG_FLAG.trim() === "1";
+const logFilePath = debugEnabled ? defaultLogFilePath() : undefined;
+const logWriter = createLogWriter(logFilePath);
+let requestCounter = 0;
+/**
+ * Begins a debug trace for an Antigravity request, logging request metadata when debugging is enabled.
+ */
+export function startAntigravityDebugRequest(meta) {
+    if (!debugEnabled) {
+        return null;
+    }
+    const id = `ANTIGRAVITY-${++requestCounter}`;
+    const method = meta.method ?? "GET";
+    logDebug(`[Antigravity Debug ${id}] ${method} ${meta.resolvedUrl}`);
+    if (meta.originalUrl && meta.originalUrl !== meta.resolvedUrl) {
+        logDebug(`[Antigravity Debug ${id}] Original URL: ${meta.originalUrl}`);
+    }
+    if (meta.projectId) {
+        logDebug(`[Antigravity Debug ${id}] Project: ${meta.projectId}`);
+    }
+    logDebug(`[Antigravity Debug ${id}] Streaming: ${meta.streaming ? "yes" : "no"}`);
+    logDebug(`[Antigravity Debug ${id}] Headers: ${JSON.stringify(maskHeaders(meta.headers))}`);
+    const bodyPreview = formatBodyPreview(meta.body);
+    if (bodyPreview) {
+        logDebug(`[Antigravity Debug ${id}] Body Preview: ${bodyPreview}`);
+    }
+    return { id, streaming: meta.streaming, startedAt: Date.now() };
+}
+/**
+ * Logs response details for a previously started debug trace when debugging is enabled.
+ */
+export function logAntigravityDebugResponse(context, response, meta = {}) {
+    if (!debugEnabled || !context) {
+        return;
+    }
+    const durationMs = Date.now() - context.startedAt;
+    logDebug(`[Antigravity Debug ${context.id}] Response ${response.status} ${response.statusText} (${durationMs}ms)`);
+    logDebug(`[Antigravity Debug ${context.id}] Response Headers: ${JSON.stringify(maskHeaders(meta.headersOverride ?? response.headers))}`);
+    if (meta.note) {
+        logDebug(`[Antigravity Debug ${context.id}] Note: ${meta.note}`);
+    }
+    if (meta.error) {
+        logDebug(`[Antigravity Debug ${context.id}] Error: ${formatError(meta.error)}`);
+    }
+    if (meta.body) {
+        logDebug(`[Antigravity Debug ${context.id}] Response Body Preview: ${truncateForLog(meta.body)}`);
+    }
+}
+/**
+ * Obscures sensitive headers and returns a plain object for logging.
+ */
+function maskHeaders(headers) {
+    if (!headers) {
+        return {};
+    }
+    const result = {};
+    const parsed = headers instanceof Headers ? headers : new Headers(headers);
+    parsed.forEach((value, key) => {
+        if (key.toLowerCase() === "authorization") {
+            result[key] = "[redacted]";
+        }
+        else {
+            result[key] = value;
+        }
+    });
+    return result;
+}
+/**
+ * Produces a short, type-aware preview of a request/response body for logs.
+ */
+function formatBodyPreview(body) {
+    if (body == null) {
+        return undefined;
+    }
+    if (typeof body === "string") {
+        return truncateForLog(body);
+    }
+    if (body instanceof URLSearchParams) {
+        return truncateForLog(body.toString());
+    }
+    if (typeof Blob !== "undefined" && body instanceof Blob) {
+        return `[Blob size=${body.size}]`;
+    }
+    if (typeof FormData !== "undefined" && body instanceof FormData) {
+        return "[FormData payload omitted]";
+    }
+    return `[${body.constructor?.name ?? typeof body} payload omitted]`;
+}
+/**
+ * Truncates long strings to a fixed preview length for logging.
+ */
+function truncateForLog(text) {
+    if (text.length <= MAX_BODY_PREVIEW_CHARS) {
+        return text;
+    }
+    return `${text.slice(0, MAX_BODY_PREVIEW_CHARS)}... (truncated ${text.length - MAX_BODY_PREVIEW_CHARS} chars)`;
+}
+/**
+ * Writes a single debug line using the configured writer.
+ */
+function logDebug(line) {
+    logWriter(line);
+}
+/**
+ * Converts unknown error-like values into printable strings.
+ */
+function formatError(error) {
+    if (error instanceof Error) {
+        return error.stack ?? error.message;
+    }
+    try {
+        return JSON.stringify(error);
+    }
+    catch {
+        return String(error);
+    }
+}
+/**
+ * Builds a timestamped log file path in the current working directory.
+ */
+function defaultLogFilePath() {
+    const timestamp = new Date().toISOString().replace(/[:.]/g, "-");
+    return join(cwd(), `antigravity-debug-${timestamp}.log`);
+}
+/**
+ * Creates a line writer that appends to a file when provided.
+ */
+function createLogWriter(filePath) {
+    if (!filePath) {
+        return () => { };
+    }
+    const stream = createWriteStream(filePath, { flags: "a" });
+    return (line) => {
+        stream.write(`${line}\n`);
+    };
+}
+//# sourceMappingURL=debug.js.map

package/dist/src/plugin/debug.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"debug.js","sourceRoot":"","sources":["../../../src/plugin/debug.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAC;AAC5C,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAExC,MAAM,UAAU,GAAG,GAAG,CAAC,0BAA0B,IAAI,EAAE,CAAC;AACxD,MAAM,sBAAsB,GAAG,KAAK,CAAC;AACrC,MAAM,YAAY,GAAG,UAAU,CAAC,IAAI,EAAE,KAAK,GAAG,CAAC;AAC/C,MAAM,WAAW,GAAG,YAAY,CAAC,CAAC,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;AACpE,MAAM,SAAS,GAAG,eAAe,CAAC,WAAW,CAAC,CAAC;AAyB/C,IAAI,cAAc,GAAG,CAAC,CAAC;AAEvB;;GAEG;AACH,MAAM,UAAU,4BAA4B,CAAC,IAAiC;IAC5E,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,EAAE,GAAG,eAAe,EAAE,cAAc,EAAE,CAAC;IAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,KAAK,CAAC;IACpC,QAAQ,CAAC,sBAAsB,EAAE,KAAK,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IACpE,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,WAAW,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;QAC9D,QAAQ,CAAC,sBAAsB,EAAE,mBAAmB,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IAC1E,CAAC;IACD,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;QACnB,QAAQ,CAAC,sBAAsB,EAAE,cAAc,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;IACnE,CAAC;IACD,QAAQ,CAAC,sBAAsB,EAAE,gBAAgB,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IAClF,QAAQ,CAAC,sBAAsB,EAAE,cAAc,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC;IAC5F,MAAM,WAAW,GAAG,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjD,IAAI,WAAW,EAAE,CAAC;QAChB,QAAQ,CAAC,sBAAsB,EAAE,mBAAmB,WAAW,EAAE,CAAC,CAAC;IACrE,CAAC;IAED,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;AAClE,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,2BAA2B,CACzC,OAAmD,EACnD,QAAkB,EAClB,OAAqC,EAAE;IAEvC,IAAI,CAAC,YAAY,IAAI,CAAC,OAAO,EAAE,CAAC;QAC9B,OAAO;IACT,CAAC;IAED,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC;IAClD,QAAQ,CACN,sBAAsB,OAAO,CAAC,EAAE,cAAc,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,KAAK,UAAU,KAAK,CACzG,CAAC;IACF,QAAQ,CACN,sBAAsB,OAAO,CAAC,EAAE,uBAAuB,IAAI,CAAC,SAAS,CACnE,WAAW,CAAC,IAAI,CAAC,eAAe,IAAI,QAAQ,CAAC,OAAO,CAAC,CACtD,EAAE,CACJ,CAAC;IAEF,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;QACd,QAAQ,CAAC,sBAAsB,OAAO,CAAC,EAAE,WAAW,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QACf,QAAQ,CAAC,sBAAsB,OAAO,CAAC,EAAE,YAAY,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAClF,CAAC;IAED,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;QACd,QAAQ,CACN,sBAAsB,OAAO,CAAC,EAAE,4BAA4B,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACxF,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,OAA+B;IAClD,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,MAAM,GAA2B,EAAE,CAAC;IAC1C,MAAM,MAAM,GAAG,OAAO,YAAY,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;IAC3E,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QAC5B,IAAI,GAAG,CAAC,WAAW,EAAE,KAAK,eAAe,EAAE,CAAC;YAC1C,MAAM,CAAC,GAAG,CAAC,GAAG,YAAY,CAAC;QAC7B,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QACtB,CAAC;IACH,CAAC,CAAC,CAAC;IACH,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,IAAsB;IAC/C,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;QACjB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC7B,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC;IAC9B,CAAC;IAED,IAAI,IAAI,YAAY,eAAe,EAAE,CAAC;QACpC,OAAO,cAAc,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;IACzC,CAAC;IAED,IAAI,OAAO,IAAI,KAAK,WAAW,IAAI,IAAI,YAAY,IAAI,EAAE,CAAC;QACxD,OAAO,cAAc,IAAI,CAAC,IAAI,GAAG,CAAC;IACpC,CAAC;IAED,IAAI,OAAO,QAAQ,KAAK,WAAW,IAAI,IAAI,YAAY,QAAQ,EAAE,CAAC;QAChE,OAAO,4BAA4B,CAAC;IACtC,CAAC;IAED,OAAO,IAAI,IAAI,CAAC,WAAW,EAAE,IAAI,IAAI,OAAO,IAAI,mBAAmB,CAAC;AACtE,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,IAAY;IAClC,IAAI,IAAI,CAAC,MAAM,IAAI,sBAAsB,EAAE,CAAC;QAC1C,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,sBAAsB,CAAC,kBAAkB,IAAI,CAAC,MAAM,GAAG,sBAAsB,SAAS,CAAC;AACjH,CAAC;AAED;;GAEG;AACH,SAAS,QAAQ,CAAC,IAAY;IAC5B,SAAS,CAAC,IAAI,CAAC,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,KAAc;IACjC,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC;IACtC,CAAC;IACD,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,kBAAkB;IACzB,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IACjE,OAAO,IAAI,CAAC,GAAG,EAAE,EAAE,qBAAqB,SAAS,MAAM,CAAC,CAAC;AAC3D,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,QAAiB;IACxC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,OAAO,GAAG,EAAE,GAAE,CAAC,CAAC;IAClB,CAAC;IAED,MAAM,MAAM,GAAG,iBAAiB,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3D,OAAO,CAAC,IAAY,EAAE,EAAE;QACtB,MAAM,CAAC,KAAK,CAAC,GAAG,IAAI,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC;AACJ,CAAC"}

package/dist/src/plugin/project.d.ts

@@ -0,0 +1,33 @@
+import type { OAuthAuthDetails, PluginClient, ProjectContextResult } from "./types";
+interface AntigravityUserTier {
+    id?: string;
+    isDefault?: boolean;
+    userDefinedCloudaicompanionProject?: boolean;
+}
+interface LoadCodeAssistPayload {
+    cloudaicompanionProject?: string | {
+        id?: string;
+    };
+    currentTier?: {
+        id?: string;
+    };
+    allowedTiers?: AntigravityUserTier[];
+}
+/**
+ * Clears cached project context results and pending promises, globally or for a refresh key.
+ */
+export declare function invalidateProjectContextCache(refresh?: string): void;
+/**
+ * Loads managed project information for the given access token and optional project.
+ */
+export declare function loadManagedProject(accessToken: string, projectId?: string): Promise<LoadCodeAssistPayload | null>;
+/**
+ * Onboards a managed project for the user, optionally retrying until completion.
+ */
+export declare function onboardManagedProject(accessToken: string, tierId: string, projectId?: string, attempts?: number, delayMs?: number): Promise<string | undefined>;
+/**
+ * Resolves an effective project ID for the current auth state, caching results per refresh token.
+ */
+export declare function ensureProjectContext(auth: OAuthAuthDetails, client: PluginClient, providerId: string): Promise<ProjectContextResult>;
+export {};
+//# sourceMappingURL=project.d.ts.map

package/dist/src/plugin/project.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"project.d.ts","sourceRoot":"","sources":["../../../src/plugin/project.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EACV,gBAAgB,EAChB,YAAY,EACZ,oBAAoB,EACrB,MAAM,SAAS,CAAC;AAWjB,UAAU,mBAAmB;IAC3B,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,kCAAkC,CAAC,EAAE,OAAO,CAAC;CAC9C;AAED,UAAU,qBAAqB;IAC7B,uBAAuB,CAAC,EAAE,MAAM,GAAG;QAAE,EAAE,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IACnD,WAAW,CAAC,EAAE;QACZ,EAAE,CAAC,EAAE,MAAM,CAAC;KACb,CAAC;IACF,YAAY,CAAC,EAAE,mBAAmB,EAAE,CAAC;CACtC;AAqFD;;GAEG;AACH,wBAAgB,6BAA6B,CAAC,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAQpE;AAED;;GAEG;AACH,wBAAsB,kBAAkB,CACtC,WAAW,EAAE,MAAM,EACnB,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC,CAuCvC;AAGD;;GAEG;AACH,wBAAsB,qBAAqB,CACzC,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,MAAM,EACd,SAAS,CAAC,EAAE,MAAM,EAClB,QAAQ,SAAK,EACb,OAAO,SAAO,GACb,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAuD7B;AAED;;GAEG;AACH,wBAAsB,oBAAoB,CACxC,IAAI,EAAE,gBAAgB,EACtB,MAAM,EAAE,YAAY,EACpB,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,oBAAoB,CAAC,CAgF/B"}

package/dist/src/plugin/project.js

@@ -0,0 +1,241 @@
+import { ANTIGRAVITY_HEADERS, ANTIGRAVITY_ENDPOINT_FALLBACKS, ANTIGRAVITY_LOAD_ENDPOINTS, ANTIGRAVITY_DEFAULT_PROJECT_ID, } from "../constants";
+import { formatRefreshParts, parseRefreshParts } from "./auth";
+const projectContextResultCache = new Map();
+const projectContextPendingCache = new Map();
+const CODE_ASSIST_METADATA = {
+    ideType: "IDE_UNSPECIFIED",
+    platform: "PLATFORM_UNSPECIFIED",
+    pluginType: "GEMINI",
+};
+class ProjectIdRequiredError extends Error {
+    /**
+     * Error raised when a required Google Cloud project is missing during Antigravity onboarding.
+     */
+    constructor() {
+        super("Google Antigravity requires a Google Cloud project. Enable the Antigravity API on a project you control, rerun `opencode auth login`, and supply that project ID when prompted.");
+    }
+}
+/**
+ * Builds metadata headers required by the Code Assist API.
+ */
+function buildMetadata(projectId) {
+    const metadata = {
+        ideType: CODE_ASSIST_METADATA.ideType,
+        platform: CODE_ASSIST_METADATA.platform,
+        pluginType: CODE_ASSIST_METADATA.pluginType,
+    };
+    if (projectId) {
+        metadata.duetProject = projectId;
+    }
+    return metadata;
+}
+/**
+ * Selects the default tier ID from the allowed tiers list.
+ */
+function getDefaultTierId(allowedTiers) {
+    if (!allowedTiers || allowedTiers.length === 0) {
+        return undefined;
+    }
+    for (const tier of allowedTiers) {
+        if (tier?.isDefault) {
+            return tier.id;
+        }
+    }
+    return allowedTiers[0]?.id;
+}
+/**
+ * Promise-based delay utility.
+ */
+function wait(ms) {
+    return new Promise(function (resolve) {
+        setTimeout(resolve, ms);
+    });
+}
+/**
+ * Extracts the cloudaicompanion project id from loadCodeAssist responses.
+ */
+function extractManagedProjectId(payload) {
+    if (!payload) {
+        return undefined;
+    }
+    if (typeof payload.cloudaicompanionProject === "string") {
+        return payload.cloudaicompanionProject;
+    }
+    if (payload.cloudaicompanionProject && typeof payload.cloudaicompanionProject.id === "string") {
+        return payload.cloudaicompanionProject.id;
+    }
+    return undefined;
+}
+/**
+ * Generates a cache key for project context based on refresh token.
+ */
+function getCacheKey(auth) {
+    const refresh = auth.refresh?.trim();
+    return refresh ? refresh : undefined;
+}
+/**
+ * Clears cached project context results and pending promises, globally or for a refresh key.
+ */
+export function invalidateProjectContextCache(refresh) {
+    if (!refresh) {
+        projectContextPendingCache.clear();
+        projectContextResultCache.clear();
+        return;
+    }
+    projectContextPendingCache.delete(refresh);
+    projectContextResultCache.delete(refresh);
+}
+/**
+ * Loads managed project information for the given access token and optional project.
+ */
+export async function loadManagedProject(accessToken, projectId) {
+    const metadata = buildMetadata(projectId);
+    const requestBody = { metadata };
+    const loadHeaders = {
+        "Content-Type": "application/json",
+        Authorization: `Bearer ${accessToken}`,
+        "User-Agent": "google-api-nodejs-client/9.15.1",
+        "X-Goog-Api-Client": "google-cloud-sdk vscode_cloudshelleditor/0.1",
+        "Client-Metadata": ANTIGRAVITY_HEADERS["Client-Metadata"],
+    };
+    const loadEndpoints = Array.from(new Set([...ANTIGRAVITY_LOAD_ENDPOINTS, ...ANTIGRAVITY_ENDPOINT_FALLBACKS]));
+    for (const baseEndpoint of loadEndpoints) {
+        try {
+            const response = await fetch(`${baseEndpoint}/v1internal:loadCodeAssist`, {
+                method: "POST",
+                headers: loadHeaders,
+                body: JSON.stringify(requestBody),
+            });
+            if (!response.ok) {
+                continue;
+            }
+            return (await response.json());
+        }
+        catch (error) {
+            console.error(`Failed to load Antigravity managed project via ${baseEndpoint}:`, error);
+            continue;
+        }
+    }
+    return null;
+}
+/**
+ * Onboards a managed project for the user, optionally retrying until completion.
+ */
+export async function onboardManagedProject(accessToken, tierId, projectId, attempts = 10, delayMs = 5000) {
+    const metadata = buildMetadata(projectId);
+    const requestBody = {
+        tierId,
+        metadata,
+    };
+    if (tierId !== "FREE") {
+        if (!projectId) {
+            throw new ProjectIdRequiredError();
+        }
+        requestBody.cloudaicompanionProject = projectId;
+    }
+    for (const baseEndpoint of ANTIGRAVITY_ENDPOINT_FALLBACKS) {
+        for (let attempt = 0; attempt < attempts; attempt += 1) {
+            try {
+                const response = await fetch(`${baseEndpoint}/v1internal:onboardUser`, {
+                    method: "POST",
+                    headers: {
+                        "Content-Type": "application/json",
+                        Authorization: `Bearer ${accessToken}`,
+                        ...ANTIGRAVITY_HEADERS,
+                    },
+                    body: JSON.stringify(requestBody),
+                });
+                if (!response.ok) {
+                    break;
+                }
+                const payload = (await response.json());
+                const managedProjectId = payload.response?.cloudaicompanionProject?.id;
+                if (payload.done && managedProjectId) {
+                    return managedProjectId;
+                }
+                if (payload.done && projectId) {
+                    return projectId;
+                }
+            }
+            catch (error) {
+                console.error(`Failed to onboard Antigravity managed project via ${baseEndpoint}:`, error);
+                break;
+            }
+            await wait(delayMs);
+        }
+    }
+    return undefined;
+}
+/**
+ * Resolves an effective project ID for the current auth state, caching results per refresh token.
+ */
+export async function ensureProjectContext(auth, client, providerId) {
+    const accessToken = auth.access;
+    if (!accessToken) {
+        return { auth, effectiveProjectId: "" };
+    }
+    const cacheKey = getCacheKey(auth);
+    if (cacheKey) {
+        const cached = projectContextResultCache.get(cacheKey);
+        if (cached) {
+            return cached;
+        }
+        const pending = projectContextPendingCache.get(cacheKey);
+        if (pending) {
+            return pending;
+        }
+    }
+    const resolveContext = async () => {
+        const parts = parseRefreshParts(auth.refresh);
+        if (parts.managedProjectId) {
+            return { auth, effectiveProjectId: parts.managedProjectId };
+        }
+        const fallbackProjectId = ANTIGRAVITY_DEFAULT_PROJECT_ID;
+        const persistManagedProject = async (managedProjectId) => {
+            const updatedAuth = {
+                ...auth,
+                refresh: formatRefreshParts({
+                    refreshToken: parts.refreshToken,
+                    projectId: parts.projectId,
+                    managedProjectId,
+                }),
+            };
+            await client.auth.set({
+                path: { id: providerId },
+                body: updatedAuth,
+            });
+            return { auth: updatedAuth, effectiveProjectId: managedProjectId };
+        };
+        // Try to resolve a managed project from Antigravity if possible.
+        const loadPayload = await loadManagedProject(accessToken, parts.projectId ?? fallbackProjectId);
+        const resolvedManagedProjectId = extractManagedProjectId(loadPayload);
+        if (resolvedManagedProjectId) {
+            return persistManagedProject(resolvedManagedProjectId);
+        }
+        if (parts.projectId) {
+            return { auth, effectiveProjectId: parts.projectId };
+        }
+        // No project id present in auth; fall back to the hardcoded id for requests.
+        return { auth, effectiveProjectId: fallbackProjectId };
+    };
+    if (!cacheKey) {
+        return resolveContext();
+    }
+    const promise = resolveContext()
+        .then((result) => {
+        const nextKey = getCacheKey(result.auth) ?? cacheKey;
+        projectContextPendingCache.delete(cacheKey);
+        projectContextResultCache.set(nextKey, result);
+        if (nextKey !== cacheKey) {
+            projectContextResultCache.delete(cacheKey);
+        }
+        return result;
+    })
+        .catch((error) => {
+        projectContextPendingCache.delete(cacheKey);
+        throw error;
+    });
+    projectContextPendingCache.set(cacheKey, promise);
+    return promise;
+}
+//# sourceMappingURL=project.js.map

package/dist/src/plugin/project.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"project.js","sourceRoot":"","sources":["../../../src/plugin/project.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,mBAAmB,EACnB,8BAA8B,EAC9B,0BAA0B,EAC1B,8BAA8B,GAC/B,MAAM,cAAc,CAAC;AACtB,OAAO,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAO/D,MAAM,yBAAyB,GAAG,IAAI,GAAG,EAAgC,CAAC;AAC1E,MAAM,0BAA0B,GAAG,IAAI,GAAG,EAAyC,CAAC;AAEpF,MAAM,oBAAoB,GAAG;IAC3B,OAAO,EAAE,iBAAiB;IAC1B,QAAQ,EAAE,sBAAsB;IAChC,UAAU,EAAE,QAAQ;CACZ,CAAC;AAyBX,MAAM,sBAAuB,SAAQ,KAAK;IACxC;;OAEG;IACH;QACE,KAAK,CACH,iLAAiL,CAClL,CAAC;IACJ,CAAC;CACF;AAED;;GAEG;AACH,SAAS,aAAa,CAAC,SAAkB;IACvC,MAAM,QAAQ,GAA2B;QACvC,OAAO,EAAE,oBAAoB,CAAC,OAAO;QACrC,QAAQ,EAAE,oBAAoB,CAAC,QAAQ;QACvC,UAAU,EAAE,oBAAoB,CAAC,UAAU;KAC5C,CAAC;IACF,IAAI,SAAS,EAAE,CAAC;QACd,QAAQ,CAAC,WAAW,GAAG,SAAS,CAAC;IACnC,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,SAAS,gBAAgB,CAAC,YAAoC;IAC5D,IAAI,CAAC,YAAY,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/C,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;QAChC,IAAI,IAAI,EAAE,SAAS,EAAE,CAAC;YACpB,OAAO,IAAI,CAAC,EAAE,CAAC;QACjB,CAAC;IACH,CAAC;IACD,OAAO,YAAY,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;AAC7B,CAAC;AAED;;GAEG;AACH,SAAS,IAAI,CAAC,EAAU;IACtB,OAAO,IAAI,OAAO,CAAC,UAAU,OAAO;QAClC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,SAAS,uBAAuB,CAAC,OAAqC;IACpE,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,OAAO,OAAO,CAAC,uBAAuB,KAAK,QAAQ,EAAE,CAAC;QACxD,OAAO,OAAO,CAAC,uBAAuB,CAAC;IACzC,CAAC;IACD,IAAI,OAAO,CAAC,uBAAuB,IAAI,OAAO,OAAO,CAAC,uBAAuB,CAAC,EAAE,KAAK,QAAQ,EAAE,CAAC;QAC9F,OAAO,OAAO,CAAC,uBAAuB,CAAC,EAAE,CAAC;IAC5C,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,IAAsB;IACzC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC;IACrC,OAAO,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;AACvC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,6BAA6B,CAAC,OAAgB;IAC5D,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,0BAA0B,CAAC,KAAK,EAAE,CAAC;QACnC,yBAAyB,CAAC,KAAK,EAAE,CAAC;QAClC,OAAO;IACT,CAAC;IACD,0BAA0B,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC3C,yBAAyB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;AAC5C,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,WAAmB,EACnB,SAAkB;IAElB,MAAM,QAAQ,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;IAC1C,MAAM,WAAW,GAA4B,EAAE,QAAQ,EAAE,CAAC;IAE1D,MAAM,WAAW,GAA2B;QAC1C,cAAc,EAAE,kBAAkB;QAClC,aAAa,EAAE,UAAU,WAAW,EAAE;QACtC,YAAY,EAAE,iCAAiC;QAC/C,mBAAmB,EAAE,8CAA8C;QACnE,iBAAiB,EAAE,mBAAmB,CAAC,iBAAiB,CAAC;KAC1D,CAAC;IAEF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAC9B,IAAI,GAAG,CAAS,CAAC,GAAG,0BAA0B,EAAE,GAAG,8BAA8B,CAAC,CAAC,CACpF,CAAC;IAEF,KAAK,MAAM,YAAY,IAAI,aAAa,EAAE,CAAC;QACzC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,GAAG,YAAY,4BAA4B,EAC3C;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,WAAW;gBACpB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;aAClC,CACF,CAAC;YAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,SAAS;YACX,CAAC;YAED,OAAO,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA0B,CAAC;QAC1D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,kDAAkD,YAAY,GAAG,EAAE,KAAK,CAAC,CAAC;YACxF,SAAS;QACX,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAGD;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,WAAmB,EACnB,MAAc,EACd,SAAkB,EAClB,QAAQ,GAAG,EAAE,EACb,OAAO,GAAG,IAAI;IAEd,MAAM,QAAQ,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;IAC1C,MAAM,WAAW,GAA4B;QAC3C,MAAM;QACN,QAAQ;KACT,CAAC;IAEF,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;QACtB,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,sBAAsB,EAAE,CAAC;QACrC,CAAC;QACD,WAAW,CAAC,uBAAuB,GAAG,SAAS,CAAC;IAClD,CAAC;IAED,KAAK,MAAM,YAAY,IAAI,8BAA8B,EAAE,CAAC;QAC1D,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,QAAQ,EAAE,OAAO,IAAI,CAAC,EAAE,CAAC;YACvD,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,GAAG,YAAY,yBAAyB,EACxC;oBACE,MAAM,EAAE,MAAM;oBACd,OAAO,EAAE;wBACP,cAAc,EAAE,kBAAkB;wBAClC,aAAa,EAAE,UAAU,WAAW,EAAE;wBACtC,GAAG,mBAAmB;qBACvB;oBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;iBAClC,CACF,CAAC;gBAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACjB,MAAM;gBACR,CAAC;gBAED,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAuB,CAAC;gBAC9D,MAAM,gBAAgB,GAAG,OAAO,CAAC,QAAQ,EAAE,uBAAuB,EAAE,EAAE,CAAC;gBACvE,IAAI,OAAO,CAAC,IAAI,IAAI,gBAAgB,EAAE,CAAC;oBACrC,OAAO,gBAAgB,CAAC;gBAC1B,CAAC;gBACD,IAAI,OAAO,CAAC,IAAI,IAAI,SAAS,EAAE,CAAC;oBAC9B,OAAO,SAAS,CAAC;gBACnB,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CACX,qDAAqD,YAAY,GAAG,EACpE,KAAK,CACN,CAAC;gBACF,MAAM;YACR,CAAC;YAED,MAAM,IAAI,CAAC,OAAO,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,IAAsB,EACtB,MAAoB,EACpB,UAAkB;IAElB,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC;IAChC,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,EAAE,EAAE,CAAC;IAC1C,CAAC;IAED,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IACnC,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,MAAM,GAAG,yBAAyB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACvD,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,MAAM,OAAO,GAAG,0BAA0B,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACzD,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,OAAO,CAAC;QACjB,CAAC;IACH,CAAC;IAED,MAAM,cAAc,GAAG,KAAK,IAAmC,EAAE;QAC/D,MAAM,KAAK,GAAG,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9C,IAAI,KAAK,CAAC,gBAAgB,EAAE,CAAC;YAC3B,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,KAAK,CAAC,gBAAgB,EAAE,CAAC;QAC9D,CAAC;QAED,MAAM,iBAAiB,GAAG,8BAA8B,CAAC;QACzD,MAAM,qBAAqB,GAAG,KAAK,EAAE,gBAAwB,EAAiC,EAAE;YAC9F,MAAM,WAAW,GAAqB;gBACpC,GAAG,IAAI;gBACP,OAAO,EAAE,kBAAkB,CAAC;oBAC1B,YAAY,EAAE,KAAK,CAAC,YAAY;oBAChC,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,gBAAgB;iBACjB,CAAC;aACH,CAAC;YAEF,MAAM,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC;gBACpB,IAAI,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE;gBACxB,IAAI,EAAE,WAAW;aAClB,CAAC,CAAC;YAEH,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,CAAC;QACrE,CAAC,CAAC;QAEF,iEAAiE;QACjE,MAAM,WAAW,GAAG,MAAM,kBAAkB,CAAC,WAAW,EAAE,KAAK,CAAC,SAAS,IAAI,iBAAiB,CAAC,CAAC;QAChG,MAAM,wBAAwB,GAAG,uBAAuB,CAAC,WAAW,CAAC,CAAC;QAEtE,IAAI,wBAAwB,EAAE,CAAC;YAC7B,OAAO,qBAAqB,CAAC,wBAAwB,CAAC,CAAC;QACzD,CAAC;QAED,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACpB,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC;QACvD,CAAC;QAED,6EAA6E;QAC7E,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,CAAC;IACzD,CAAC,CAAC;IAEF,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,OAAO,cAAc,EAAE,CAAC;IAC1B,CAAC;IAED,MAAM,OAAO,GAAG,cAAc,EAAE;SAC7B,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;QACf,MAAM,OAAO,GAAG,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,QAAQ,CAAC;QACrD,0BAA0B,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5C,yBAAyB,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC/C,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;YACzB,yBAAyB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7C,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;SACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;QACf,0BAA0B,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,KAAK,CAAC;IACd,CAAC,CAAC,CAAC;IAEL,0BAA0B,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAClD,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/dist/src/plugin/request-helpers.d.ts

@@ -0,0 +1,51 @@
+export interface AntigravityApiError {
+    code?: number;
+    message?: string;
+    status?: string;
+    [key: string]: unknown;
+}
+/**
+ * Minimal representation of Antigravity API responses we touch.
+ */
+export interface AntigravityApiBody {
+    response?: unknown;
+    error?: AntigravityApiError;
+    [key: string]: unknown;
+}
+/**
+ * Usage metadata exposed by Antigravity responses. Fields are optional to reflect partial payloads.
+ */
+export interface AntigravityUsageMetadata {
+    totalTokenCount?: number;
+    promptTokenCount?: number;
+    candidatesTokenCount?: number;
+    cachedContentTokenCount?: number;
+}
+/**
+ * Normalized thinking configuration accepted by Antigravity.
+ */
+export interface ThinkingConfig {
+    thinkingBudget?: number;
+    includeThoughts?: boolean;
+}
+/**
+ * Ensures thinkingConfig is valid: includeThoughts only allowed when budget > 0.
+ */
+export declare function normalizeThinkingConfig(config: unknown): ThinkingConfig | undefined;
+/**
+ * Parses an Antigravity API body; handles array-wrapped responses the API sometimes returns.
+ */
+export declare function parseAntigravityApiBody(rawText: string): AntigravityApiBody | null;
+/**
+ * Extracts usageMetadata from a response object, guarding types.
+ */
+export declare function extractUsageMetadata(body: AntigravityApiBody): AntigravityUsageMetadata | null;
+/**
+ * Walks SSE lines to find a usage-bearing response chunk.
+ */
+export declare function extractUsageFromSsePayload(payload: string): AntigravityUsageMetadata | null;
+/**
+ * Enhances 404 errors for Antigravity models with a direct preview-access message.
+ */
+export declare function rewriteAntigravityPreviewAccessError(body: AntigravityApiBody, status: number, requestedModel?: string): AntigravityApiBody | null;
+//# sourceMappingURL=request-helpers.d.ts.map

package/dist/src/plugin/request-helpers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"request-helpers.d.ts","sourceRoot":"","sources":["../../../src/plugin/request-helpers.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,mBAAmB;IAClC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,KAAK,CAAC,EAAE,mBAAmB,CAAC;IAC5B,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED;;GAEG;AACH,MAAM,WAAW,wBAAwB;IACvC,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,uBAAuB,CAAC,EAAE,MAAM,CAAC;CAClC;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,eAAe,CAAC,EAAE,OAAO,CAAC;CAC3B;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,OAAO,GAAG,cAAc,GAAG,SAAS,CA2BnF;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,kBAAkB,GAAG,IAAI,CAmBlF;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,kBAAkB,GAAG,wBAAwB,GAAG,IAAI,CAmB9F;AAED;;GAEG;AACH,wBAAgB,0BAA0B,CAAC,OAAO,EAAE,MAAM,GAAG,wBAAwB,GAAG,IAAI,CAuB3F;AAED;;GAEG;AACH,wBAAgB,oCAAoC,CAClD,IAAI,EAAE,kBAAkB,EACxB,MAAM,EAAE,MAAM,EACd,cAAc,CAAC,EAAE,MAAM,GACtB,kBAAkB,GAAG,IAAI,CAmB3B"}

package/dist/src/plugin/request-helpers.js

@@ -0,0 +1,135 @@
+const ANTIGRAVITY_PREVIEW_LINK = "https://goo.gle/enable-preview-features"; // TODO: Update to Antigravity link if available
+/**
+ * Ensures thinkingConfig is valid: includeThoughts only allowed when budget > 0.
+ */
+export function normalizeThinkingConfig(config) {
+    if (!config || typeof config !== "object") {
+        return undefined;
+    }
+    const record = config;
+    const budgetRaw = record.thinkingBudget ?? record.thinking_budget;
+    const includeRaw = record.includeThoughts ?? record.include_thoughts;
+    const thinkingBudget = typeof budgetRaw === "number" && Number.isFinite(budgetRaw) ? budgetRaw : undefined;
+    const includeThoughts = typeof includeRaw === "boolean" ? includeRaw : undefined;
+    const enableThinking = thinkingBudget !== undefined && thinkingBudget > 0;
+    const finalInclude = enableThinking ? includeThoughts ?? false : false;
+    if (!enableThinking && finalInclude === false && thinkingBudget === undefined && includeThoughts === undefined) {
+        return undefined;
+    }
+    const normalized = {};
+    if (thinkingBudget !== undefined) {
+        normalized.thinkingBudget = thinkingBudget;
+    }
+    if (finalInclude !== undefined) {
+        normalized.includeThoughts = finalInclude;
+    }
+    return normalized;
+}
+/**
+ * Parses an Antigravity API body; handles array-wrapped responses the API sometimes returns.
+ */
+export function parseAntigravityApiBody(rawText) {
+    try {
+        const parsed = JSON.parse(rawText);
+        if (Array.isArray(parsed)) {
+            const firstObject = parsed.find((item) => typeof item === "object" && item !== null);
+            if (firstObject && typeof firstObject === "object") {
+                return firstObject;
+            }
+            return null;
+        }
+        if (parsed && typeof parsed === "object") {
+            return parsed;
+        }
+        return null;
+    }
+    catch {
+        return null;
+    }
+}
+/**
+ * Extracts usageMetadata from a response object, guarding types.
+ */
+export function extractUsageMetadata(body) {
+    const usage = (body.response && typeof body.response === "object"
+        ? body.response.usageMetadata
+        : undefined);
+    if (!usage || typeof usage !== "object") {
+        return null;
+    }
+    const asRecord = usage;
+    const toNumber = (value) => typeof value === "number" && Number.isFinite(value) ? value : undefined;
+    return {
+        totalTokenCount: toNumber(asRecord.totalTokenCount),
+        promptTokenCount: toNumber(asRecord.promptTokenCount),
+        candidatesTokenCount: toNumber(asRecord.candidatesTokenCount),
+        cachedContentTokenCount: toNumber(asRecord.cachedContentTokenCount),
+    };
+}
+/**
+ * Walks SSE lines to find a usage-bearing response chunk.
+ */
+export function extractUsageFromSsePayload(payload) {
+    const lines = payload.split("\n");
+    for (const line of lines) {
+        if (!line.startsWith("data:")) {
+            continue;
+        }
+        const jsonText = line.slice(5).trim();
+        if (!jsonText) {
+            continue;
+        }
+        try {
+            const parsed = JSON.parse(jsonText);
+            if (parsed && typeof parsed === "object") {
+                const usage = extractUsageMetadata({ response: parsed.response });
+                if (usage) {
+                    return usage;
+                }
+            }
+        }
+        catch {
+            continue;
+        }
+    }
+    return null;
+}
+/**
+ * Enhances 404 errors for Antigravity models with a direct preview-access message.
+ */
+export function rewriteAntigravityPreviewAccessError(body, status, requestedModel) {
+    if (!needsPreviewAccessOverride(status, body, requestedModel)) {
+        return null;
+    }
+    const error = body.error ?? {};
+    const trimmedMessage = typeof error.message === "string" ? error.message.trim() : "";
+    const messagePrefix = trimmedMessage.length > 0
+        ? trimmedMessage
+        : "Antigravity preview features are not enabled for this account.";
+    const enhancedMessage = `${messagePrefix} Request preview access at ${ANTIGRAVITY_PREVIEW_LINK} before using this model.`;
+    return {
+        ...body,
+        error: {
+            ...error,
+            message: enhancedMessage,
+        },
+    };
+}
+function needsPreviewAccessOverride(status, body, requestedModel) {
+    if (status !== 404) {
+        return false;
+    }
+    if (isAntigravityModel(requestedModel)) {
+        return true;
+    }
+    const errorMessage = typeof body.error?.message === "string" ? body.error.message : "";
+    return isAntigravityModel(errorMessage);
+}
+function isAntigravityModel(target) {
+    if (!target) {
+        return false;
+    }
+    // Check for Antigravity models instead of Gemini 3
+    return /antigravity/i.test(target) || /opus/i.test(target) || /claude/i.test(target);
+}
+//# sourceMappingURL=request-helpers.js.map

package/dist/src/plugin/request-helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"request-helpers.js","sourceRoot":"","sources":["../../../src/plugin/request-helpers.ts"],"names":[],"mappings":"AAAA,MAAM,wBAAwB,GAAG,yCAAyC,CAAC,CAAC,gDAAgD;AAoC5H;;GAEG;AACH,MAAM,UAAU,uBAAuB,CAAC,MAAe;IACrD,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC1C,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,MAAM,GAAG,MAAiC,CAAC;IACjD,MAAM,SAAS,GAAG,MAAM,CAAC,cAAc,IAAI,MAAM,CAAC,eAAe,CAAC;IAClE,MAAM,UAAU,GAAG,MAAM,CAAC,eAAe,IAAI,MAAM,CAAC,gBAAgB,CAAC;IAErE,MAAM,cAAc,GAAG,OAAO,SAAS,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;IAC3G,MAAM,eAAe,GAAG,OAAO,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;IAEjF,MAAM,cAAc,GAAG,cAAc,KAAK,SAAS,IAAI,cAAc,GAAG,CAAC,CAAC;IAC1E,MAAM,YAAY,GAAG,cAAc,CAAC,CAAC,CAAC,eAAe,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;IAEvE,IAAI,CAAC,cAAc,IAAI,YAAY,KAAK,KAAK,IAAI,cAAc,KAAK,SAAS,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;QAC/G,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,UAAU,GAAmB,EAAE,CAAC;IACtC,IAAI,cAAc,KAAK,SAAS,EAAE,CAAC;QACjC,UAAU,CAAC,cAAc,GAAG,cAAc,CAAC;IAC7C,CAAC;IACD,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;QAC/B,UAAU,CAAC,eAAe,GAAG,YAAY,CAAC;IAC5C,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,uBAAuB,CAAC,OAAe;IACrD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACnC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1B,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,IAAa,EAAE,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,CAAC,CAAC;YAC9F,IAAI,WAAW,IAAI,OAAO,WAAW,KAAK,QAAQ,EAAE,CAAC;gBACnD,OAAO,WAAiC,CAAC;YAC3C,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YACzC,OAAO,MAA4B,CAAC;QACtC,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAAC,IAAwB;IAC3D,MAAM,KAAK,GAAG,CAAC,IAAI,CAAC,QAAQ,IAAI,OAAO,IAAI,CAAC,QAAQ,KAAK,QAAQ;QAC/D,CAAC,CAAE,IAAI,CAAC,QAAwC,CAAC,aAAa;QAC9D,CAAC,CAAC,SAAS,CAAyC,CAAC;IAEvD,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,QAAQ,GAAG,KAAgC,CAAC;IAClD,MAAM,QAAQ,GAAG,CAAC,KAAc,EAAsB,EAAE,CACtD,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;IAE1E,OAAO;QACL,eAAe,EAAE,QAAQ,CAAC,QAAQ,CAAC,eAAe,CAAC;QACnD,gBAAgB,EAAE,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,CAAC;QACrD,oBAAoB,EAAE,QAAQ,CAAC,QAAQ,CAAC,oBAAoB,CAAC;QAC7D,uBAAuB,EAAE,QAAQ,CAAC,QAAQ,CAAC,uBAAuB,CAAC;KACpE,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,0BAA0B,CAAC,OAAe;IACxD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAClC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YAC9B,SAAS;QACX,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,SAAS;QACX,CAAC;QACD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YACpC,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;gBACzC,MAAM,KAAK,GAAG,oBAAoB,CAAC,EAAE,QAAQ,EAAG,MAAkC,CAAC,QAAQ,EAAE,CAAC,CAAC;gBAC/F,IAAI,KAAK,EAAE,CAAC;oBACV,OAAO,KAAK,CAAC;gBACf,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,SAAS;QACX,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oCAAoC,CAClD,IAAwB,EACxB,MAAc,EACd,cAAuB;IAEvB,IAAI,CAAC,0BAA0B,CAAC,MAAM,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,CAAC;QAC9D,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,KAAK,GAAwB,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;IACpD,MAAM,cAAc,GAAG,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACrF,MAAM,aAAa,GAAG,cAAc,CAAC,MAAM,GAAG,CAAC;QAC7C,CAAC,CAAC,cAAc;QAChB,CAAC,CAAC,gEAAgE,CAAC;IACrE,MAAM,eAAe,GAAG,GAAG,aAAa,8BAA8B,wBAAwB,2BAA2B,CAAC;IAE1H,OAAO;QACL,GAAG,IAAI;QACP,KAAK,EAAE;YACL,GAAG,KAAK;YACR,OAAO,EAAE,eAAe;SACzB;KACF,CAAC;AACJ,CAAC;AAED,SAAS,0BAA0B,CACjC,MAAc,EACd,IAAwB,EACxB,cAAuB;IAEvB,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;QACnB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,kBAAkB,CAAC,cAAc,CAAC,EAAE,CAAC;QACvC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,YAAY,GAAG,OAAO,IAAI,CAAC,KAAK,EAAE,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;IACvF,OAAO,kBAAkB,CAAC,YAAY,CAAC,CAAC;AAC1C,CAAC;AAED,SAAS,kBAAkB,CAAC,MAAe;IACzC,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,KAAK,CAAC;IACf,CAAC;IAED,mDAAmD;IACnD,OAAO,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AACvF,CAAC"}

package/dist/src/plugin/request.d.ts

@@ -0,0 +1,27 @@
+import { type AntigravityDebugContext } from "./debug";
+/**
+ * Detects requests headed to the Google Generative Language API so we can intercept them.
+ */
+export declare function isGenerativeLanguageRequest(input: RequestInfo): input is string;
+/**
+ * Rewrites OpenAI-style requests into Antigravity shape, normalizing model, headers,
+ * optional cached_content, and thinking config. Also toggles streaming mode for SSE actions.
+ */
+export declare function prepareAntigravityRequest(input: RequestInfo, init: RequestInit | undefined, accessToken: string, projectId: string, endpointOverride?: string): {
+    request: RequestInfo;
+    init: RequestInit;
+    streaming: boolean;
+    requestedModel?: string;
+    effectiveModel?: string;
+    projectId?: string;
+    endpoint?: string;
+    toolDebugMissing?: number;
+    toolDebugSummary?: string;
+    toolDebugPayload?: string;
+};
+/**
+ * Normalizes Antigravity responses: applies retry headers, extracts cache usage into headers,
+ * rewrites preview errors, flattens streaming payloads, and logs debug metadata.
+ */
+export declare function transformAntigravityResponse(response: Response, streaming: boolean, debugContext?: AntigravityDebugContext | null, requestedModel?: string, projectId?: string, endpoint?: string, effectiveModel?: string, toolDebugMissing?: number, toolDebugSummary?: string, toolDebugPayload?: string): Promise<Response>;
+//# sourceMappingURL=request.d.ts.map