opencode-antigravity-auth 1.0.2 → 1.0.4

package/src/plugin/request-helpers.ts

@@ -1,197 +0,0 @@
-const ANTIGRAVITY_PREVIEW_LINK = "https://goo.gle/enable-preview-features"; // TODO: Update to Antigravity link if available
-
-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 function normalizeThinkingConfig(config: unknown): ThinkingConfig | undefined {
-  if (!config || typeof config !== "object") {
-    return undefined;
-  }
-
-  const record = config as Record<string, unknown>;
-  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: ThinkingConfig = {};
-  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: string): AntigravityApiBody | null {
-  try {
-    const parsed = JSON.parse(rawText);
-    if (Array.isArray(parsed)) {
-      const firstObject = parsed.find((item: unknown) => typeof item === "object" && item !== null);
-      if (firstObject && typeof firstObject === "object") {
-        return firstObject as AntigravityApiBody;
-      }
-      return null;
-    }
-
-    if (parsed && typeof parsed === "object") {
-      return parsed as AntigravityApiBody;
-    }
-
-    return null;
-  } catch {
-    return null;
-  }
-}
-
-/**
- * Extracts usageMetadata from a response object, guarding types.
- */
-export function extractUsageMetadata(body: AntigravityApiBody): AntigravityUsageMetadata | null {
-  const usage = (body.response && typeof body.response === "object"
-    ? (body.response as { usageMetadata?: unknown }).usageMetadata
-    : undefined) as AntigravityUsageMetadata | undefined;
-
-  if (!usage || typeof usage !== "object") {
-    return null;
-  }
-
-  const asRecord = usage as Record<string, unknown>;
-  const toNumber = (value: unknown): number | undefined =>
-    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: string): AntigravityUsageMetadata | null {
-  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 as Record<string, unknown>).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: AntigravityApiBody,
-  status: number,
-  requestedModel?: string,
-): AntigravityApiBody | null {
-  if (!needsPreviewAccessOverride(status, body, requestedModel)) {
-    return null;
-  }
-
-  const error: AntigravityApiError = 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: number,
-  body: AntigravityApiBody,
-  requestedModel?: string,
-): boolean {
-  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?: string): boolean {
-  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);
-}