opencode-antigravity-auth-remix 1.0.7

package/dist/src/plugin/refresh-queue.js

@@ -0,0 +1,244 @@
+/**
+ * Proactive Token Refresh Queue
+ *
+ * Ported from LLM-API-Key-Proxy's BackgroundRefresher.
+ *
+ * This module provides background token refresh to ensure OAuth tokens
+ * remain valid without blocking user requests. It periodically checks
+ * all accounts and refreshes tokens that are approaching expiry.
+ *
+ * Features:
+ * - Non-blocking background refresh (doesn't block requests)
+ * - Configurable refresh buffer (default: 30 minutes before expiry)
+ * - Configurable check interval (default: 5 minutes)
+ * - Serialized refresh to prevent concurrent refresh storms
+ * - Integrates with existing AccountManager and token refresh logic
+ * - Silent operation: no console output, uses structured logger
+ */
+import { refreshAccessToken } from "./token";
+import { createLogger } from "./logger";
+const log = createLogger("refresh-queue");
+export const DEFAULT_PROACTIVE_REFRESH_CONFIG = {
+    enabled: true,
+    bufferSeconds: 1800, // 30 minutes
+    checkIntervalSeconds: 300, // 5 minutes
+};
+/**
+ * Proactive Token Refresh Queue
+ *
+ * Runs in the background and proactively refreshes tokens before they expire.
+ * This ensures that user requests never block on token refresh.
+ *
+ * All logging is silent by default - uses structured logger that only outputs
+ * when OPENCODE_ANTIGRAVITY_CONSOLE_LOG=1 is set or TUI logging is available.
+ */
+export class ProactiveRefreshQueue {
+    config;
+    client;
+    providerId;
+    accountManager = null;
+    state = {
+        isRunning: false,
+        intervalHandle: null,
+        isRefreshing: false,
+        lastCheckTime: 0,
+        lastRefreshTime: 0,
+        refreshCount: 0,
+        errorCount: 0,
+    };
+    constructor(client, providerId, config) {
+        this.client = client;
+        this.providerId = providerId;
+        this.config = {
+            ...DEFAULT_PROACTIVE_REFRESH_CONFIG,
+            ...config,
+        };
+    }
+    /**
+     * Set the account manager to use for refresh operations.
+     * Must be called before start().
+     */
+    setAccountManager(manager) {
+        this.accountManager = manager;
+    }
+    /**
+     * Check if a token needs proactive refresh.
+     * Returns true if the token expires within the buffer period.
+     */
+    needsRefresh(account) {
+        if (!account.expires) {
+            // No expiry set - assume it's fine
+            return false;
+        }
+        const now = Date.now();
+        const bufferMs = this.config.bufferSeconds * 1000;
+        const refreshThreshold = now + bufferMs;
+        return account.expires <= refreshThreshold;
+    }
+    /**
+     * Check if a token is already expired.
+     */
+    isExpired(account) {
+        if (!account.expires) {
+            return false;
+        }
+        return account.expires <= Date.now();
+    }
+    /**
+     * Get all accounts that need proactive refresh.
+     */
+    getAccountsNeedingRefresh() {
+        if (!this.accountManager) {
+            return [];
+        }
+        return this.accountManager.getAccounts().filter((account) => {
+            // Only refresh if not already expired (let the main flow handle expired tokens)
+            if (this.isExpired(account)) {
+                return false;
+            }
+            return this.needsRefresh(account);
+        });
+    }
+    /**
+     * Perform a single refresh check iteration.
+     * This is called periodically by the background interval.
+     */
+    async runRefreshCheck() {
+        if (this.state.isRefreshing) {
+            // Already refreshing - skip this iteration
+            return;
+        }
+        if (!this.accountManager) {
+            return;
+        }
+        this.state.isRefreshing = true;
+        this.state.lastCheckTime = Date.now();
+        try {
+            const accountsToRefresh = this.getAccountsNeedingRefresh();
+            if (accountsToRefresh.length === 0) {
+                return;
+            }
+            log.debug("Found accounts needing refresh", { count: accountsToRefresh.length });
+            // Refresh accounts serially to avoid concurrent refresh storms
+            for (const account of accountsToRefresh) {
+                if (!this.state.isRunning) {
+                    // Queue was stopped - abort
+                    break;
+                }
+                try {
+                    const auth = this.accountManager.toAuthDetails(account);
+                    const refreshed = await this.refreshToken(auth, account);
+                    if (refreshed) {
+                        this.accountManager.updateFromAuth(account, refreshed);
+                        this.state.refreshCount++;
+                        this.state.lastRefreshTime = Date.now();
+                        // Persist the refreshed token
+                        try {
+                            await this.accountManager.saveToDisk();
+                        }
+                        catch {
+                            // Non-fatal - token is refreshed in memory
+                        }
+                    }
+                }
+                catch (error) {
+                    this.state.errorCount++;
+                    // Log but don't throw - continue with other accounts
+                    log.warn("Failed to refresh account", {
+                        accountIndex: account.index,
+                        error: error instanceof Error ? error.message : String(error),
+                    });
+                }
+            }
+        }
+        finally {
+            this.state.isRefreshing = false;
+        }
+    }
+    /**
+     * Refresh a single token.
+     */
+    async refreshToken(auth, account) {
+        const minutesUntilExpiry = account.expires
+            ? Math.round((account.expires - Date.now()) / 60000)
+            : "unknown";
+        log.debug("Proactively refreshing token", {
+            accountIndex: account.index,
+            email: account.email ?? "unknown",
+            minutesUntilExpiry,
+        });
+        return refreshAccessToken(auth, this.client, this.providerId);
+    }
+    /**
+     * Start the background refresh queue.
+     */
+    start() {
+        if (this.state.isRunning) {
+            return;
+        }
+        if (!this.config.enabled) {
+            log.debug("Proactive refresh disabled by config");
+            return;
+        }
+        this.state.isRunning = true;
+        const intervalMs = this.config.checkIntervalSeconds * 1000;
+        log.debug("Started proactive refresh queue", {
+            checkIntervalSeconds: this.config.checkIntervalSeconds,
+            bufferSeconds: this.config.bufferSeconds,
+        });
+        // Run initial check after a short delay (let things settle)
+        setTimeout(() => {
+            if (this.state.isRunning) {
+                this.runRefreshCheck().catch((error) => {
+                    log.error("Initial check failed", {
+                        error: error instanceof Error ? error.message : String(error),
+                    });
+                });
+            }
+        }, 5000);
+        // Set up periodic checks
+        this.state.intervalHandle = setInterval(() => {
+            this.runRefreshCheck().catch((error) => {
+                log.error("Check failed", {
+                    error: error instanceof Error ? error.message : String(error),
+                });
+            });
+        }, intervalMs);
+    }
+    /**
+     * Stop the background refresh queue.
+     */
+    stop() {
+        if (!this.state.isRunning) {
+            return;
+        }
+        this.state.isRunning = false;
+        if (this.state.intervalHandle) {
+            clearInterval(this.state.intervalHandle);
+            this.state.intervalHandle = null;
+        }
+        log.debug("Stopped proactive refresh queue", {
+            refreshCount: this.state.refreshCount,
+            errorCount: this.state.errorCount,
+        });
+    }
+    /**
+     * Get current queue statistics.
+     */
+    getStats() {
+        return { ...this.state };
+    }
+    /**
+     * Check if the queue is currently running.
+     */
+    isRunning() {
+        return this.state.isRunning;
+    }
+}
+/**
+ * Create a proactive refresh queue instance.
+ */
+export function createProactiveRefreshQueue(client, providerId, config) {
+    return new ProactiveRefreshQueue(client, providerId, config);
+}
+//# sourceMappingURL=refresh-queue.js.map

package/dist/src/plugin/refresh-queue.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"refresh-queue.js","sourceRoot":"","sources":["../../../src/plugin/refresh-queue.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAIH,OAAO,EAAE,kBAAkB,EAAE,MAAM,SAAS,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAExC,MAAM,GAAG,GAAG,YAAY,CAAC,eAAe,CAAC,CAAC;AAY1C,MAAM,CAAC,MAAM,gCAAgC,GAA2B;IACtE,OAAO,EAAE,IAAI;IACb,aAAa,EAAE,IAAI,EAAE,aAAa;IAClC,oBAAoB,EAAE,GAAG,EAAE,YAAY;CACxC,CAAC;AAaF;;;;;;;;GAQG;AACH,MAAM,OAAO,qBAAqB;IACf,MAAM,CAAyB;IAC/B,MAAM,CAAe;IACrB,UAAU,CAAS;IAC5B,cAAc,GAA0B,IAAI,CAAC;IAE7C,KAAK,GAAsB;QACjC,SAAS,EAAE,KAAK;QAChB,cAAc,EAAE,IAAI;QACpB,YAAY,EAAE,KAAK;QACnB,aAAa,EAAE,CAAC;QAChB,eAAe,EAAE,CAAC;QAClB,YAAY,EAAE,CAAC;QACf,UAAU,EAAE,CAAC;KACd,CAAC;IAEF,YACE,MAAoB,EACpB,UAAkB,EAClB,MAAwC;QAExC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG;YACZ,GAAG,gCAAgC;YACnC,GAAG,MAAM;SACV,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,iBAAiB,CAAC,OAAuB;QACvC,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC;IAChC,CAAC;IAED;;;OAGG;IACH,YAAY,CAAC,OAAuB;QAClC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YACrB,mCAAmC;YACnC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,GAAG,IAAI,CAAC;QAClD,MAAM,gBAAgB,GAAG,GAAG,GAAG,QAAQ,CAAC;QAExC,OAAO,OAAO,CAAC,OAAO,IAAI,gBAAgB,CAAC;IAC7C,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,OAAuB;QAC/B,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YACrB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,yBAAyB;QACvB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,OAAO,IAAI,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;YAC1D,gFAAgF;YAChF,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC5B,OAAO,KAAK,CAAC;YACf,CAAC;YACD,OAAO,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,eAAe;QAC3B,IAAI,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;YAC5B,2CAA2C;YAC3C,OAAO;QACT,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;QAC/B,IAAI,CAAC,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEtC,IAAI,CAAC;YACH,MAAM,iBAAiB,GAAG,IAAI,CAAC,yBAAyB,EAAE,CAAC;YAE3D,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACnC,OAAO;YACT,CAAC;YAED,GAAG,CAAC,KAAK,CAAC,gCAAgC,EAAE,EAAE,KAAK,EAAE,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC;YAEjF,+DAA+D;YAC/D,KAAK,MAAM,OAAO,IAAI,iBAAiB,EAAE,CAAC;gBACxC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;oBAC1B,4BAA4B;oBAC5B,MAAM;gBACR,CAAC;gBAED,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;oBACxD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;oBAEzD,IAAI,SAAS,EAAE,CAAC;wBACd,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;wBACvD,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;wBAC1B,IAAI,CAAC,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;wBAExC,8BAA8B;wBAC9B,IAAI,CAAC;4BACH,MAAM,IAAI,CAAC,cAAc,CAAC,UAAU,EAAE,CAAC;wBACzC,CAAC;wBAAC,MAAM,CAAC;4BACP,2CAA2C;wBAC7C,CAAC;oBACH,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;oBACxB,qDAAqD;oBACrD,GAAG,CAAC,IAAI,CAAC,2BAA2B,EAAE;wBACpC,YAAY,EAAE,OAAO,CAAC,KAAK;wBAC3B,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;qBAC9D,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,KAAK,CAAC;QAClC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,YAAY,CACxB,IAAsB,EACtB,OAAuB;QAEvB,MAAM,kBAAkB,GAAG,OAAO,CAAC,OAAO;YACxC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,GAAG,KAAK,CAAC;YACpD,CAAC,CAAC,SAAS,CAAC;QAEd,GAAG,CAAC,KAAK,CAAC,8BAA8B,EAAE;YACxC,YAAY,EAAE,OAAO,CAAC,KAAK;YAC3B,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,SAAS;YACjC,kBAAkB;SACnB,CAAC,CAAC;QAEH,OAAO,kBAAkB,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;IAChE,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;YACzB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACzB,GAAG,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;YAClD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;QAC5B,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,GAAG,IAAI,CAAC;QAE3D,GAAG,CAAC,KAAK,CAAC,iCAAiC,EAAE;YAC3C,oBAAoB,EAAE,IAAI,CAAC,MAAM,CAAC,oBAAoB;YACtD,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;SACzC,CAAC,CAAC;QAEH,4DAA4D;QAC5D,UAAU,CAAC,GAAG,EAAE;YACd,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;gBACzB,IAAI,CAAC,eAAe,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBACrC,GAAG,CAAC,KAAK,CAAC,sBAAsB,EAAE;wBAChC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;qBAC9D,CAAC,CAAC;gBACL,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC,EAAE,IAAI,CAAC,CAAC;QAET,yBAAyB;QACzB,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;YAC3C,IAAI,CAAC,eAAe,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACrC,GAAG,CAAC,KAAK,CAAC,cAAc,EAAE;oBACxB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC9D,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,UAAU,CAAC,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,IAAI;QACF,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;YAC1B,OAAO;QACT,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,KAAK,CAAC;QAE7B,IAAI,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC;YAC9B,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;YACzC,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,IAAI,CAAC;QACnC,CAAC;QAED,GAAG,CAAC,KAAK,CAAC,iCAAiC,EAAE;YAC3C,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;YACrC,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU;SAClC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,QAAQ;QAQN,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC;IAC9B,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,2BAA2B,CACzC,MAAoB,EACpB,UAAkB,EAClB,MAAwC;IAExC,OAAO,IAAI,qBAAqB,CAAC,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;AAC/D,CAAC"}

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

@@ -0,0 +1,270 @@
+/**
+ * Cleans a JSON schema for Antigravity API compatibility.
+ * Transforms unsupported features into description hints while preserving semantic information.
+ *
+ * Ported from CLIProxyAPI's CleanJSONSchemaForAntigravity (gemini_schema.go)
+ */
+export declare function cleanJSONSchemaForAntigravity(schema: any): any;
+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;
+    thoughtsTokenCount?: number;
+}
+/**
+ * Normalized thinking configuration accepted by Antigravity.
+ */
+export interface ThinkingConfig {
+    thinkingBudget?: number;
+    includeThoughts?: boolean;
+}
+/**
+ * Default token budget for thinking/reasoning. 16000 tokens provides sufficient
+ * space for complex reasoning while staying within typical model limits.
+ */
+export declare const DEFAULT_THINKING_BUDGET = 16000;
+/**
+ * Checks if a model name indicates thinking/reasoning capability.
+ * Models with "thinking", "gemini-3", or "opus" in their name support extended thinking.
+ */
+export declare function isThinkingCapableModel(modelName: string): boolean;
+/**
+ * Extracts thinking configuration from various possible request locations.
+ * Supports both Gemini-style thinkingConfig and Anthropic-style thinking options.
+ */
+export declare function extractThinkingConfig(requestPayload: Record<string, unknown>, rawGenerationConfig: Record<string, unknown> | undefined, extraBody: Record<string, unknown> | undefined): ThinkingConfig | undefined;
+/**
+ * Determines the final thinking configuration based on model capabilities and user settings.
+ * For Claude thinking models, we keep thinking enabled even in multi-turn conversations.
+ * The filterUnsignedThinkingBlocks function will handle signature validation/restoration.
+ */
+export declare function resolveThinkingConfig(userConfig: ThinkingConfig | undefined, isThinkingModel: boolean, _isClaudeModel: boolean, _hasAssistantHistory: boolean): ThinkingConfig | undefined;
+/**
+ * Filters thinking blocks from contents unless the signature matches our cache.
+ * Attempts to restore signatures from cache for thinking blocks that lack signatures.
+ *
+ * @param contents - The contents array from the request
+ * @param sessionId - Optional session ID for signature cache lookup
+ * @param getCachedSignatureFn - Optional function to retrieve cached signatures
+ */
+export declare function filterUnsignedThinkingBlocks(contents: any[], sessionId?: string, getCachedSignatureFn?: (sessionId: string, text: string) => string | undefined, isClaudeModel?: boolean): any[];
+/**
+ * Filters thinking blocks from Anthropic-style messages[] payloads using cached signatures.
+ */
+export declare function filterMessagesThinkingBlocks(messages: any[], sessionId?: string, getCachedSignatureFn?: (sessionId: string, text: string) => string | undefined, isClaudeModel?: boolean): any[];
+export declare function deepFilterThinkingBlocks(payload: unknown, sessionId?: string, getCachedSignatureFn?: (sessionId: string, text: string) => string | undefined, isClaudeModel?: boolean): unknown;
+/**
+ * Transforms thinking/reasoning content in response parts to OpenCode's expected format.
+ * Handles both Gemini-style (thought: true) and Anthropic-style (type: "thinking") formats.
+ * Also extracts reasoning_content for Anthropic-style responses.
+ */
+export declare function transformThinkingParts(response: unknown): unknown;
+/**
+ * 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;
+/**
+ * Checks if a JSON response body represents an empty response.
+ *
+ * Empty responses occur when:
+ * - No candidates in Gemini format
+ * - No choices in OpenAI format
+ * - Candidates/choices exist but have no content
+ *
+ * @param text - The response body text (should be valid JSON)
+ * @returns true if the response is empty
+ */
+export declare function isEmptyResponseBody(text: string): boolean;
+/**
+ * Checks if a streaming SSE response yielded zero meaningful chunks.
+ *
+ * This is used after consuming a streaming response to determine if retry is needed.
+ */
+export interface StreamingChunkCounter {
+    increment: () => void;
+    getCount: () => number;
+    hasContent: () => boolean;
+}
+export declare function createStreamingChunkCounter(): StreamingChunkCounter;
+/**
+ * Checks if an SSE line contains meaningful content.
+ *
+ * @param line - A single SSE line (e.g., "data: {...}")
+ * @returns true if the line contains content worth counting
+ */
+export declare function isMeaningfulSseLine(line: string): boolean;
+/**
+ * Recursively parses JSON strings in nested data structures.
+ *
+ * This is a port of LLM-API-Key-Proxy's _recursively_parse_json_strings() function.
+ *
+ * Handles:
+ * - JSON-stringified values: {"files": "[{...}]"} → {"files": [{...}]}
+ * - Malformed double-encoded JSON (extra trailing chars)
+ * - Escaped control characters (\\n → \n, \\t → \t)
+ *
+ * This is useful because Antigravity sometimes returns JSON-stringified values
+ * in tool arguments, which can cause downstream parsing issues.
+ *
+ * @param obj - The object to recursively parse
+ * @returns The parsed object with JSON strings expanded
+ */
+export declare function recursivelyParseJsonStrings(obj: unknown): unknown;
+/**
+ * Groups function calls with their responses, handling ID mismatches.
+ *
+ * This is a port of LLM-API-Key-Proxy's _fix_tool_response_grouping() function.
+ *
+ * When context compaction or other processes strip tool responses, the tool call
+ * IDs become orphaned. This function attempts to recover by:
+ *
+ * 1. Pass 1: Match by exact ID (normal case)
+ * 2. Pass 2: Match by function name (for ID mismatches)
+ * 3. Pass 3: Match "unknown_function" orphans or take first available
+ * 4. Fallback: Create placeholder responses for missing tool results
+ *
+ * @param contents - Array of Gemini-style content messages
+ * @returns Fixed contents array with matched tool responses
+ */
+export declare function fixToolResponseGrouping(contents: any[]): any[];
+/**
+ * Checks if contents have any tool call/response ID mismatches.
+ *
+ * @param contents - Array of Gemini-style content messages
+ * @returns Object with mismatch details
+ */
+export declare function detectToolIdMismatches(contents: any[]): {
+    hasMismatches: boolean;
+    expectedIds: string[];
+    foundIds: string[];
+    missingIds: string[];
+    orphanIds: string[];
+};
+/**
+ * Find orphaned tool_use IDs (tool_use without matching tool_result).
+ * Works on Claude format messages.
+ */
+export declare function findOrphanedToolUseIds(messages: any[]): Set<string>;
+/**
+ * Fix orphaned tool_use blocks in Claude format messages.
+ * Mirrors fixToolResponseGrouping() but for Claude's messages[] format.
+ *
+ * Claude format:
+ * - assistant message with content[]: { type: 'tool_use', id, name, input }
+ * - user message with content[]: { type: 'tool_result', tool_use_id, content }
+ *
+ * @param messages - Claude format messages array
+ * @returns Fixed messages with placeholder tool_results for orphans
+ */
+export declare function fixClaudeToolPairing(messages: any[]): any[];
+/**
+ * Validate and fix tool pairing with fallback nuclear option.
+ * Defense in depth: tries gentle fix first, then nuclear removal.
+ */
+export declare function validateAndFixClaudeToolPairing(messages: any[]): any[];
+/**
+ * Injects parameter signatures into tool descriptions.
+ * Port of LLM-API-Key-Proxy's _inject_signature_into_descriptions()
+ *
+ * This helps prevent tool hallucination by explicitly listing parameters
+ * in the description, making it harder for the model to hallucinate
+ * parameters from its training data.
+ *
+ * @param tools - Array of tool definitions (Gemini format)
+ * @param promptTemplate - Template for the signature (default: "\\n\\nSTRICT PARAMETERS: {params}.")
+ * @returns Modified tools array with signatures injected
+ */
+export declare function injectParameterSignatures(tools: any[], promptTemplate?: string): any[];
+/**
+ * Injects a tool hardening system instruction into the request payload.
+ * Port of LLM-API-Key-Proxy's _inject_tool_hardening_instruction()
+ *
+ * @param payload - The Gemini request payload
+ * @param instructionText - The instruction text to inject
+ */
+export declare function injectToolHardeningInstruction(payload: Record<string, unknown>, instructionText: string): void;
+/**
+ * Assigns IDs to functionCall parts and returns the pending call IDs by name.
+ * This is the first pass of tool ID assignment.
+ *
+ * @param contents - Gemini-style contents array
+ * @returns Object with modified contents and pending call IDs map
+ */
+export declare function assignToolIdsToContents(contents: any[]): {
+    contents: any[];
+    pendingCallIdsByName: Map<string, string[]>;
+    toolCallCounter: number;
+};
+/**
+ * Matches functionResponse IDs to their corresponding functionCall IDs.
+ * This is the second pass of tool ID assignment.
+ *
+ * @param contents - Gemini-style contents array
+ * @param pendingCallIdsByName - Map of function names to pending call IDs
+ * @returns Modified contents with matched response IDs
+ */
+export declare function matchResponseIdsToContents(contents: any[], pendingCallIdsByName: Map<string, string[]>): any[];
+/**
+ * Applies all tool fixes to a request payload for Claude models.
+ * This includes:
+ * 1. Tool ID assignment for functionCalls
+ * 2. Response ID matching for functionResponses
+ * 3. Orphan recovery via fixToolResponseGrouping
+ * 4. Claude format pairing fix via validateAndFixClaudeToolPairing
+ *
+ * @param payload - Request payload object
+ * @param isClaude - Whether this is a Claude model request
+ * @returns Object with fix applied status
+ */
+export declare function applyToolPairingFixes(payload: Record<string, unknown>, isClaude: boolean): {
+    contentsFixed: boolean;
+    messagesFixed: boolean;
+};
+/**
+ * Creates a synthetic Claude SSE streaming response with error content.
+ *
+ * When returning HTTP 400/500 errors to OpenCode, the session becomes locked
+ * and the user cannot use /compact or other commands. This function creates
+ * a fake "successful" SSE response (200 OK) with the error message as text content,
+ * allowing the user to continue using the session.
+ *
+ * @param errorMessage - The error message to include in the response
+ * @param requestedModel - The model that was requested
+ * @returns A Response object with synthetic SSE stream
+ */
+export declare function createSyntheticErrorResponse(errorMessage: string, requestedModel?: string): Response;
+//# 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":"AAwoBA;;;;;GAKG;AACH,wBAAgB,6BAA6B,CAAC,MAAM,EAAE,GAAG,GAAG,GAAG,CA2B9D;AAMD,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;IACjC,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,eAAe,CAAC,EAAE,OAAO,CAAC;CAC3B;AAED;;;GAGG;AACH,eAAO,MAAM,uBAAuB,QAAQ,CAAC;AAE7C;;;GAGG;AACH,wBAAgB,sBAAsB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAKjE;AAED;;;GAGG;AACH,wBAAgB,qBAAqB,CACnC,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACvC,mBAAmB,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACxD,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,GAC7C,cAAc,GAAG,SAAS,CA0B5B;AAED;;;;GAIG;AACH,wBAAgB,qBAAqB,CACnC,UAAU,EAAE,cAAc,GAAG,SAAS,EACtC,eAAe,EAAE,OAAO,EACxB,cAAc,EAAE,OAAO,EACvB,oBAAoB,EAAE,OAAO,GAC5B,cAAc,GAAG,SAAS,CAQ5B;AA4SD;;;;;;;GAOG;AACH,wBAAgB,4BAA4B,CAC1C,QAAQ,EAAE,GAAG,EAAE,EACf,SAAS,CAAC,EAAE,MAAM,EAClB,oBAAoB,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,KAAK,MAAM,GAAG,SAAS,EAC9E,aAAa,CAAC,EAAE,OAAO,GACtB,GAAG,EAAE,CAgDP;AAED;;GAEG;AACH,wBAAgB,4BAA4B,CAC1C,QAAQ,EAAE,GAAG,EAAE,EACf,SAAS,CAAC,EAAE,MAAM,EAClB,oBAAoB,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,KAAK,MAAM,GAAG,SAAS,EAC9E,aAAa,CAAC,EAAE,OAAO,GACtB,GAAG,EAAE,CA6BP;AAED,wBAAgB,wBAAwB,CACtC,OAAO,EAAE,OAAO,EAChB,SAAS,CAAC,EAAE,MAAM,EAClB,oBAAoB,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,KAAK,MAAM,GAAG,SAAS,EAC9E,aAAa,CAAC,EAAE,OAAO,GACtB,OAAO,CA4CT;AAoED;;;;GAIG;AACH,wBAAgB,sBAAsB,CAAC,QAAQ,EAAE,OAAO,GAAG,OAAO,CAwCjE;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,CAoB9F;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;AAgCD;;;;;;;;;;GAUG;AACH,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAkFzD;AAED;;;;GAIG;AACH,MAAM,WAAW,qBAAqB;IACpC,SAAS,EAAE,MAAM,IAAI,CAAC;IACtB,QAAQ,EAAE,MAAM,MAAM,CAAC;IACvB,UAAU,EAAE,MAAM,OAAO,CAAC;CAC3B;AAED,wBAAgB,2BAA2B,IAAI,qBAAqB,CAWnE;AAED;;;;;GAKG;AACH,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAwCzD;AAMD;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,2BAA2B,CAAC,GAAG,EAAE,OAAO,GAAG,OAAO,CAwFjE;AAMD;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,uBAAuB,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,GAAG,EAAE,CA4K9D;AAED;;;;;GAKG;AACH,wBAAgB,sBAAsB,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG;IACvD,aAAa,EAAE,OAAO,CAAC;IACvB,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,UAAU,EAAE,MAAM,EAAE,CAAC;IACrB,SAAS,EAAE,MAAM,EAAE,CAAC;CACrB,CA8BA;AAMD;;;GAGG;AACH,wBAAgB,sBAAsB,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC,CAkBnE;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,GAAG,EAAE,CAyF3D;AA0BD;;;GAGG;AACH,wBAAgB,+BAA+B,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,GAAG,EAAE,CAsBtE;AAiED;;;;;;;;;;;GAWG;AACH,wBAAgB,yBAAyB,CACvC,KAAK,EAAE,GAAG,EAAE,EACZ,cAAc,SAAwC,GACrD,GAAG,EAAE,CAqCP;AAED;;;;;;GAMG;AACH,wBAAgB,8BAA8B,CAC5C,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAChC,eAAe,EAAE,MAAM,GACtB,IAAI,CAqCN;AAOD;;;;;;GAMG;AACH,wBAAgB,uBAAuB,CACrC,QAAQ,EAAE,GAAG,EAAE,GACd;IAAE,QAAQ,EAAE,GAAG,EAAE,CAAC;IAAC,oBAAoB,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IAAC,eAAe,EAAE,MAAM,CAAA;CAAE,CAgC3F;AAED;;;;;;;GAOG;AACH,wBAAgB,0BAA0B,CACxC,QAAQ,EAAE,GAAG,EAAE,EACf,oBAAoB,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,GAC1C,GAAG,EAAE,CA2BP;AAED;;;;;;;;;;;GAWG;AACH,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAChC,QAAQ,EAAE,OAAO,GAChB;IAAE,aAAa,EAAE,OAAO,CAAC;IAAC,aAAa,EAAE,OAAO,CAAA;CAAE,CAsCpD;AAQD;;;;;;;;;;;GAWG;AACH,wBAAgB,4BAA4B,CAC1C,YAAY,EAAE,MAAM,EACpB,cAAc,GAAE,MAAkB,GACjC,QAAQ,CAkFV"}