antigravity-auth 1.6.0

package/dist/plugin/request.d.ts

@@ -0,0 +1,110 @@
+import { type HeaderStyle } from "../constants";
+import { createStreamingTransformer, transformSseLine, transformStreamingPayload } from "./core/streaming";
+import { type AntigravityDebugContext } from "./debug";
+import { type Fingerprint } from "./fingerprint";
+import type { GoogleSearchConfig } from "./transform/types";
+declare function buildSignatureSessionKey(sessionId: string, model?: string, conversationKey?: string, projectKey?: string): string;
+declare function hashConversationSeed(seed: string): string;
+declare function extractTextFromContent(content: unknown): string;
+declare function extractConversationSeedFromMessages(messages: any[]): string;
+declare function extractConversationSeedFromContents(contents: any[]): string;
+declare function resolveConversationKey(requestPayload: Record<string, unknown>): string | undefined;
+declare function resolveProjectKey(candidate?: unknown, fallback?: string): string | undefined;
+declare function isGeminiToolUsePart(part: any): boolean;
+declare function isGeminiThinkingPart(part: any): boolean;
+declare function ensureThoughtSignature(part: any, sessionId: string): any;
+declare function hasSignedThinkingPart(part: any, sessionId?: string): boolean;
+declare function ensureThinkingBeforeToolUseInContents(contents: any[], signatureSessionKey: string): any[];
+declare function hasToolUseInContents(contents: any[]): boolean;
+declare function hasSignedThinkingInContents(contents: any[], sessionId?: string): boolean;
+declare function hasToolUseInMessages(messages: any[]): boolean;
+declare function hasSignedThinkingInMessages(messages: any[], sessionId?: string): boolean;
+declare function ensureThinkingBeforeToolUseInMessages(messages: any[], signatureSessionKey: string): any[];
+/**
+ * Gets the stable session ID for this plugin instance.
+ */
+export declare function getPluginSessionId(): string;
+declare function generateSyntheticProjectId(): string;
+/**
+ * Resolve a fetch() URL from RequestInfo. OpenCode / AI SDK often calls fetch(Request, init)
+ * instead of fetch(string, init); we must inspect the URL the same way in both cases.
+ */
+export declare function requestInfoToUrlString(input: RequestInfo): string | null;
+/**
+ * Detects requests headed to the Google Generative Language API (or Antigravity / Cloud Code PA)
+ * so we can intercept and rewrite them.
+ */
+export declare function isGenerativeLanguageRequest(input: RequestInfo): boolean;
+/**
+ * OpenCode / AI SDK often calls fetch(Request, init) with the JSON body on the Request only.
+ * prepareAntigravityRequest reads init.body as a string; materialize so sanitization always runs.
+ */
+export declare function materializeGenerativeLanguageFetchInput(input: RequestInfo, init?: RequestInit): Promise<{
+    input: RequestInfo;
+    init?: RequestInit;
+}>;
+/**
+ * Options for request preparation.
+ */
+export interface PrepareRequestOptions {
+    /** Enable Claude tool hardening (parameter signatures + system instruction). Default: true */
+    claudeToolHardening?: boolean;
+    /** Enable top-level Claude prompt auto-caching (`cache_control`). Default: false */
+    claudePromptAutoCaching?: boolean;
+    /** Google Search configuration (global default) */
+    googleSearch?: GoogleSearchConfig;
+    /** Per-account fingerprint for rate limit mitigation. Falls back to session fingerprint if not provided. */
+    fingerprint?: Fingerprint;
+    /** Write Gemini payloads to debug log. Default: false */
+    debugGeminiPayloads?: boolean;
+}
+export declare function prepareAntigravityRequest(input: RequestInfo, init: RequestInit | undefined, accessToken: string, projectId: string, endpointOverride?: string, headerStyle?: HeaderStyle, forceThinkingRecovery?: boolean, options?: PrepareRequestOptions): {
+    request: RequestInfo;
+    init: RequestInit;
+    streaming: boolean;
+    requestedModel?: string;
+    effectiveModel?: string;
+    projectId?: string;
+    endpoint?: string;
+    sessionId?: string;
+    toolDebugMissing?: number;
+    toolDebugSummary?: string;
+    toolDebugPayload?: string;
+    needsSignedThinkingWarmup?: boolean;
+    headerStyle: HeaderStyle;
+    thinkingRecoveryMessage?: string;
+};
+export declare function buildThinkingWarmupBody(bodyText: string | undefined, isClaudeThinking: boolean): string | null;
+/**
+ * Normalizes Antigravity responses: applies retry headers, extracts cache usage into headers,
+ * rewrites preview errors, flattens streaming payloads, and logs debug metadata.
+ *
+ * For streaming SSE responses, uses TransformStream for true real-time incremental streaming.
+ * Thinking/reasoning tokens are transformed and forwarded immediately as they arrive.
+ */
+export declare function transformAntigravityResponse(response: Response, streaming: boolean, debugContext?: AntigravityDebugContext | null, requestedModel?: string, projectId?: string, endpoint?: string, effectiveModel?: string, sessionId?: string, toolDebugMissing?: number, toolDebugSummary?: string, toolDebugPayload?: string, debugLines?: string[], onComplete?: () => void, onWatchdogTimeout?: () => Promise<void> | void): Promise<Response>;
+export declare const __testExports: {
+    buildSignatureSessionKey: typeof buildSignatureSessionKey;
+    hashConversationSeed: typeof hashConversationSeed;
+    extractTextFromContent: typeof extractTextFromContent;
+    extractConversationSeedFromMessages: typeof extractConversationSeedFromMessages;
+    extractConversationSeedFromContents: typeof extractConversationSeedFromContents;
+    resolveConversationKey: typeof resolveConversationKey;
+    resolveProjectKey: typeof resolveProjectKey;
+    isGeminiToolUsePart: typeof isGeminiToolUsePart;
+    isGeminiThinkingPart: typeof isGeminiThinkingPart;
+    ensureThoughtSignature: typeof ensureThoughtSignature;
+    hasSignedThinkingPart: typeof hasSignedThinkingPart;
+    hasSignedThinkingInContents: typeof hasSignedThinkingInContents;
+    hasSignedThinkingInMessages: typeof hasSignedThinkingInMessages;
+    hasToolUseInContents: typeof hasToolUseInContents;
+    hasToolUseInMessages: typeof hasToolUseInMessages;
+    ensureThinkingBeforeToolUseInContents: typeof ensureThinkingBeforeToolUseInContents;
+    ensureThinkingBeforeToolUseInMessages: typeof ensureThinkingBeforeToolUseInMessages;
+    generateSyntheticProjectId: typeof generateSyntheticProjectId;
+    MIN_SIGNATURE_LENGTH: number;
+    transformSseLine: typeof transformSseLine;
+    transformStreamingPayload: typeof transformStreamingPayload;
+    createStreamingTransformer: typeof createStreamingTransformer;
+};
+export {};