formagent-sdk 0.3.3 → 0.4.0

package/dist/cli/index.js

@@ -2116,6 +2116,111 @@ class SessionManagerImpl {
   }
 }
 
+// src/utils/retry.ts
+var DEFAULT_RETRY_OPTIONS = {
+  maxAttempts: 3,
+  initialDelay: 1000,
+  maxDelay: 30000,
+  backoffMultiplier: 2,
+  jitter: true,
+  onRetry: () => {},
+  signal: undefined
+};
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+function calculateDelay(attempt, initialDelay, maxDelay, backoffMultiplier, jitter) {
+  const delay = Math.min(initialDelay * Math.pow(backoffMultiplier, attempt), maxDelay);
+  if (jitter) {
+    return delay * (0.5 + Math.random() * 0.5);
+  }
+  return delay;
+}
+function isRetryableStatus(status) {
+  return status >= 500 || status === 429;
+}
+function isRetryableError(error) {
+  if (error instanceof TypeError) {
+    return error.message.includes("fetch") || error.message.includes("network") || error.message.includes("ECONNREFUSED") || error.message.includes("ETIMEDOUT") || error.message.includes("ECONNRESET");
+  }
+  if (error instanceof Error) {
+    const message = error.message.toLowerCase();
+    if (message.includes("rate_limit") || message.includes("rate limit")) {
+      return true;
+    }
+    if (message.includes("usage_limit") || message.includes("usage limit")) {
+      return true;
+    }
+    if (message.includes("timeout") || message.includes("timed out")) {
+      return true;
+    }
+    if (message.includes("5") || message.includes("502") || message.includes("503") || message.includes("504")) {
+      return true;
+    }
+  }
+  return false;
+}
+function extractStatusCode(error) {
+  const match = error.message.match(/(\d{3})/);
+  return match ? parseInt(match[1], 10) : null;
+}
+async function withRetry(fn, options = {}) {
+  const opts = { ...DEFAULT_RETRY_OPTIONS, ...options };
+  let lastError = null;
+  for (let attempt = 0;attempt < opts.maxAttempts; attempt++) {
+    if (opts.signal?.aborted) {
+      throw new Error("Request aborted");
+    }
+    try {
+      return await fn();
+    } catch (error) {
+      lastError = error instanceof Error ? error : new Error(String(error));
+      if (attempt >= opts.maxAttempts - 1) {
+        throw lastError;
+      }
+      const statusCode = extractStatusCode(lastError);
+      const isRetryable = statusCode !== null ? isRetryableStatus(statusCode) : isRetryableError(lastError);
+      if (!isRetryable) {
+        throw lastError;
+      }
+      const delay = calculateDelay(attempt, opts.initialDelay, opts.maxDelay, opts.backoffMultiplier, opts.jitter);
+      opts.onRetry(attempt + 1, lastError);
+      await sleep(delay);
+    }
+  }
+  throw lastError || new Error("Max retries exceeded");
+}
+async function fetchWithRetry(url, init, retryOptions) {
+  return withRetry(async () => {
+    const response = await fetch(url, init);
+    if (!response.ok) {
+      const errorText = await response.text().catch(() => "");
+      let errorMessage = `HTTP ${response.status}`;
+      try {
+        const errorJson = JSON.parse(errorText);
+        if (errorJson.error?.type) {
+          errorMessage += `: ${errorJson.error.type}`;
+        }
+        if (errorJson.error?.message) {
+          errorMessage += ` - ${errorJson.error.message}`;
+        }
+      } catch {
+        if (errorText) {
+          errorMessage += ` ${errorText}`;
+        }
+      }
+      const error = new Error(errorMessage);
+      error.status = response.status;
+      error.responseText = errorText;
+      throw error;
+    }
+    return response;
+  }, {
+    signal: init.signal,
+    ...retryOptions
+  });
+}
+
 // src/llm/anthropic.ts
 class AnthropicProvider {
   id = "anthropic";
@@ -2128,6 +2233,7 @@ class AnthropicProvider {
     /^claude-instant/
   ];
   config;
+  defaultRetryOptions;
   constructor(config = {}) {
     const apiKey = config.apiKey ?? process.env.ANTHROPIC_API_KEY;
     if (!apiKey) {
@@ -2137,7 +2243,15 @@ class AnthropicProvider {
       apiKey,
       baseUrl: config.baseUrl ?? process.env.ANTHROPIC_BASE_URL ?? "https://api.anthropic.com",
       apiVersion: config.apiVersion ?? "2023-06-01",
-      defaultMaxTokens: config.defaultMaxTokens ?? 4096
+      defaultMaxTokens: config.defaultMaxTokens ?? 4096,
+      retry: config.retry
+    };
+    this.defaultRetryOptions = {
+      maxAttempts: 3,
+      initialDelay: 1000,
+      maxDelay: 30000,
+      backoffMultiplier: 2,
+      jitter: true
     };
   }
   supportsModel(model) {
@@ -2145,31 +2259,25 @@ class AnthropicProvider {
   }
   async complete(request) {
     const anthropicRequest = this.buildRequest(request, false);
-    const response = await fetch(`${this.config.baseUrl}/v1/messages`, {
+    const retryOptions = this.config.retry ?? this.defaultRetryOptions;
+    const response = await fetchWithRetry(`${this.config.baseUrl}/v1/messages`, {
       method: "POST",
       headers: this.getHeaders(),
       body: JSON.stringify(anthropicRequest),
       signal: request.abortSignal
-    });
-    if (!response.ok) {
-      const error = await response.text();
-      throw new Error(`Anthropic API error: ${response.status} ${error}`);
-    }
+    }, retryOptions);
     const data = await response.json();
     return this.convertResponse(data);
   }
   async stream(request, options) {
     const anthropicRequest = this.buildRequest(request, true);
-    const response = await fetch(`${this.config.baseUrl}/v1/messages`, {
+    const retryOptions = this.config.retry ?? this.defaultRetryOptions;
+    const response = await fetchWithRetry(`${this.config.baseUrl}/v1/messages`, {
       method: "POST",
       headers: this.getHeaders(),
       body: JSON.stringify(anthropicRequest),
       signal: request.abortSignal
-    });
-    if (!response.ok) {
-      const error = await response.text();
-      throw new Error(`Anthropic API error: ${response.status} ${error}`);
-    }
+    }, retryOptions);
     return this.createStreamIterator(response.body, options);
   }
   buildRequest(request, stream) {
@@ -2390,6 +2498,7 @@ class OpenAIProvider {
     /^chatgpt/
   ];
   config;
+  defaultRetryOptions;
   constructor(config = {}) {
     const apiKey = config.apiKey ?? process.env.OPENAI_API_KEY;
     if (!apiKey) {
@@ -2399,96 +2508,88 @@ class OpenAIProvider {
       apiKey,
       baseUrl: this.normalizeBaseUrl(config.baseUrl ?? process.env.OPENAI_BASE_URL ?? "https://api.openai.com/v1"),
       organization: config.organization,
-      defaultMaxTokens: config.defaultMaxTokens ?? 4096
+      defaultMaxTokens: config.defaultMaxTokens ?? 4096,
+      retry: config.retry
+    };
+    this.defaultRetryOptions = {
+      maxAttempts: 3,
+      initialDelay: 1000,
+      maxDelay: 30000,
+      backoffMultiplier: 2,
+      jitter: true
     };
   }
   supportsModel(model) {
     return this.supportedModels.some((pattern) => pattern.test(model));
   }
   async complete(request) {
+    const retryOptions = this.config.retry ?? this.defaultRetryOptions;
     if (this.usesResponsesApi(request.config.model)) {
       const openaiRequest2 = this.buildResponsesRequest(request, false);
-      const response2 = await fetch(`${this.config.baseUrl}/responses`, {
+      const response2 = await fetchWithRetry(`${this.config.baseUrl}/responses`, {
         method: "POST",
         headers: this.getHeaders(),
         body: JSON.stringify(openaiRequest2),
         signal: request.abortSignal
-      });
-      if (!response2.ok) {
-        const error = await response2.text();
-        throw new Error(`OpenAI API error: ${response2.status} ${error}`);
-      }
+      }, retryOptions);
       const data2 = await response2.json();
       return this.convertResponsesResponse(data2);
     }
     const openaiRequest = this.buildRequest(request, false);
-    let response = await fetch(`${this.config.baseUrl}/chat/completions`, {
+    const response = await fetchWithRetry(`${this.config.baseUrl}/chat/completions`, {
       method: "POST",
       headers: this.getHeaders(),
       body: JSON.stringify(openaiRequest),
       signal: request.abortSignal
-    });
-    if (!response.ok) {
-      const error = await response.text();
-      if (this.shouldFallbackToResponses(response.status, error)) {
+    }, retryOptions);
+    if (response.status === 404) {
+      const errorText = await response.clone().text();
+      if (this.shouldFallbackToResponses(404, errorText)) {
         const fallbackRequest = this.buildResponsesRequest(request, false);
-        response = await fetch(`${this.config.baseUrl}/responses`, {
+        const fallbackResponse = await fetchWithRetry(`${this.config.baseUrl}/responses`, {
           method: "POST",
           headers: this.getHeaders(),
           body: JSON.stringify(fallbackRequest),
           signal: request.abortSignal
-        });
-        if (!response.ok) {
-          const fallbackError = await response.text();
-          throw new Error(`OpenAI API error: ${response.status} ${fallbackError}`);
-        }
-        const data2 = await response.json();
+        }, retryOptions);
+        const data2 = await fallbackResponse.json();
         return this.convertResponsesResponse(data2);
       }
-      throw new Error(`OpenAI API error: ${response.status} ${error}`);
     }
     const data = await response.json();
     return this.convertResponse(data);
   }
   async stream(request, options) {
+    const retryOptions = this.config.retry ?? this.defaultRetryOptions;
     if (this.usesResponsesApi(request.config.model)) {
       const openaiRequest2 = this.buildResponsesRequest(request, true);
-      const response2 = await fetch(`${this.config.baseUrl}/responses`, {
+      const response2 = await fetchWithRetry(`${this.config.baseUrl}/responses`, {
         method: "POST",
         headers: this.getHeaders(),
         body: JSON.stringify(openaiRequest2),
         signal: request.abortSignal
-      });
-      if (!response2.ok) {
-        const error = await response2.text();
-        throw new Error(`OpenAI API error: ${response2.status} ${error}`);
-      }
+      }, retryOptions);
       return this.createResponsesStreamIterator(response2.body, options);
     }
     const openaiRequest = this.buildRequest(request, true);
-    let response = await fetch(`${this.config.baseUrl}/chat/completions`, {
+    const response = await fetchWithRetry(`${this.config.baseUrl}/chat/completions`, {
       method: "POST",
       headers: this.getHeaders(),
       body: JSON.stringify(openaiRequest),
       signal: request.abortSignal
-    });
-    if (!response.ok) {
-      const error = await response.text();
-      if (this.shouldFallbackToResponses(response.status, error)) {
+    }, retryOptions);
+    if (response.status === 404) {
+      const errorText = await response.clone().text();
+      if (this.shouldFallbackToResponses(404, errorText)) {
         const fallbackRequest = this.buildResponsesRequest(request, true);
-        response = await fetch(`${this.config.baseUrl}/responses`, {
+        const fallbackResponse = await fetchWithRetry(`${this.config.baseUrl}/responses`, {
           method: "POST",
           headers: this.getHeaders(),
           body: JSON.stringify(fallbackRequest),
           signal: request.abortSignal
-        });
-        if (!response.ok) {
-          const fallbackError = await response.text();
-          throw new Error(`OpenAI API error: ${response.status} ${fallbackError}`);
-        }
-        return this.createResponsesStreamIterator(response.body, options);
+        }, retryOptions);
+        return this.createResponsesStreamIterator(fallbackResponse.body, options);
       }
-      throw new Error(`OpenAI API error: ${response.status} ${error}`);
     }
     return this.createStreamIterator(response.body, options);
   }
@@ -3203,6 +3304,7 @@ class GeminiProvider {
   name = "Gemini";
   supportedModels = [/^gemini-/, /^models\/gemini-/];
   config;
+  defaultRetryOptions;
   constructor(config = {}) {
     const apiKey = config.apiKey ?? process.env.GEMINI_API_KEY ?? process.env.GOOGLE_API_KEY;
     if (!apiKey) {
@@ -3211,7 +3313,15 @@ class GeminiProvider {
     this.config = {
       apiKey,
       baseUrl: config.baseUrl ?? process.env.GEMINI_BASE_URL ?? "https://generativelanguage.googleapis.com/v1beta",
-      defaultMaxTokens: config.defaultMaxTokens ?? 4096
+      defaultMaxTokens: config.defaultMaxTokens ?? 4096,
+      retry: config.retry
+    };
+    this.defaultRetryOptions = {
+      maxAttempts: 3,
+      initialDelay: 1000,
+      maxDelay: 30000,
+      backoffMultiplier: 2,
+      jitter: true
     };
   }
   supportsModel(model) {
@@ -3220,16 +3330,13 @@ class GeminiProvider {
   async complete(request) {
     const geminiRequest = this.buildRequest(request);
     const url = this.buildUrl(this.getModelPath(request.config.model) + ":generateContent");
-    const response = await fetch(url, {
+    const retryOptions = this.config.retry ?? this.defaultRetryOptions;
+    const response = await fetchWithRetry(url, {
       method: "POST",
       headers: this.getHeaders(),
       body: JSON.stringify(geminiRequest),
       signal: request.abortSignal
-    });
-    if (!response.ok) {
-      const error = await response.text();
-      throw new Error(`Gemini API error: ${response.status} ${error}`);
-    }
+    }, retryOptions);
     const data = await response.json();
     return this.convertResponse(data, request.config.model);
   }
@@ -3238,16 +3345,13 @@ class GeminiProvider {
     const url = this.buildUrl(this.getModelPath(request.config.model) + ":streamGenerateContent", {
       alt: "sse"
     });
-    const response = await fetch(url, {
+    const retryOptions = this.config.retry ?? this.defaultRetryOptions;
+    const response = await fetchWithRetry(url, {
       method: "POST",
       headers: this.getHeaders(),
       body: JSON.stringify(geminiRequest),
       signal: request.abortSignal
-    });
-    if (!response.ok) {
-      const error = await response.text();
-      throw new Error(`Gemini API error: ${response.status} ${error}`);
-    }
+    }, retryOptions);
     const contentType = response.headers.get("content-type") ?? "";
     if (!contentType.includes("text/event-stream")) {
       const data = await response.json();
@@ -6236,14 +6340,17 @@ ${c.bold("Interactive Commands:")}
   ${c.cyan("/exit")}     Exit the CLI
 
 ${c.bold("Environment:")}
-  ${c.cyan("ANTHROPIC_API_KEY")}   Anthropic API key (for Claude models)
-  ${c.cyan("ANTHROPIC_MODEL")}     Optional. Claude model (default: claude-sonnet-4-20250514)
-  ${c.cyan("GEMINI_API_KEY")}      Gemini API key (for Gemini models)
-  ${c.cyan("GEMINI_MODEL")}        Optional. Gemini model (default: gemini-1.5-pro)
-  ${c.cyan("GEMINI_BASE_URL")}     Optional. Custom Gemini API base URL
-  ${c.cyan("OPENAI_API_KEY")}      OpenAI API key (for GPT models)
-  ${c.cyan("OPENAI_MODEL")}        Optional. OpenAI model (default: gpt-5.2)
-  ${c.cyan("OPENAI_BASE_URL")}     Optional. Custom OpenAI-compatible API URL
+  ${c.cyan("ANTHROPIC_API_KEY")}     Anthropic API key (for Claude models)
+  ${c.cyan("ANTHROPIC_MODEL")}       Optional. Claude model (default: claude-sonnet-4-20250514)
+  ${c.cyan("GEMINI_API_KEY")}        Gemini API key (for Gemini models)
+  ${c.cyan("GEMINI_MODEL")}          Optional. Gemini model (default: gemini-1.5-pro)
+  ${c.cyan("GEMINI_BASE_URL")}       Optional. Custom Gemini API base URL
+  ${c.cyan("OPENAI_API_KEY")}        OpenAI API key (for GPT models)
+  ${c.cyan("OPENAI_MODEL")}          Optional. OpenAI model (default: gpt-5.2)
+  ${c.cyan("OPENAI_BASE_URL")}       Optional. Custom OpenAI-compatible API URL
+  ${c.cyan("ANTIGRAVITY_API_KEY")}   Antigravity API key (default: sk-antigravity)
+  ${c.cyan("ANTIGRAVITY_MODEL")}     Optional. Antigravity model (default: gemini-3-flash)
+  ${c.cyan("ANTIGRAVITY_BASE_URL")}  Optional. Antigravity gateway URL (default: http://127.0.0.1:8045)
 
 ${c.bold("Examples:")}
   ${c.dim("# Start interactive mode")}
@@ -6379,6 +6486,7 @@ function printModelsHelp() {
   console.log(`  ${c.cyan("/models")} openai gpt-5-mini`);
   console.log(`  ${c.cyan("/models")} anthropic claude-sonnet-4-20250514`);
   console.log(`  ${c.cyan("/models")} gemini gemini-1.5-pro`);
+  console.log(`  ${c.cyan("/models")} antigravity gemini-3-flash`);
   console.log(`  ${c.cyan("/models")} gpt-5.2`);
   console.log(`  ${c.cyan("/models")} reset`);
   console.log();
@@ -6419,7 +6527,7 @@ async function handleModelsCommand(args) {
   const provider = parseProvider(args[0]);
   if (!provider) {
     console.log(c.yellow(`
-  Unknown provider: ${args[0]}. Use "openai", "anthropic", or "gemini".
+  Unknown provider: ${args[0]}. Use "openai", "anthropic", "gemini", or "antigravity".
 `));
     return;
   }
@@ -6563,6 +6671,34 @@ async function listGeminiModels() {
   }
   console.log();
 }
+async function listAntigravityModels() {
+  const config = getAntigravityConfig();
+  const baseUrl = `${config.baseUrl}/v1`;
+  console.log(c.bold("Antigravity Models:"));
+  console.log(c.dim("  API Type: openai-compatible (Antigravity Gateway)"));
+  console.log(c.dim(`  Base URL: ${baseUrl}`));
+  console.log(c.dim(`  Default Model: ${config.model}`));
+  const res = await fetch(`${baseUrl}/models`, {
+    headers: {
+      Authorization: `Bearer ${config.apiKey}`,
+      "x-api-key": config.apiKey
+    }
+  });
+  if (!res.ok) {
+    console.log(c.red(`  ✗ Failed to fetch models (${res.status})`));
+    console.log(c.dim(`  URL: ${baseUrl}/models`));
+    console.log(c.dim("  Make sure Antigravity Manager is running"));
+    console.log();
+    return;
+  }
+  const payload = await res.json();
+  const items = payload.data ?? [];
+  for (const item of items) {
+    const owner = item.owned_by ? ` (${item.owned_by})` : "";
+    console.log(`  ${c.green("●")} ${item.id}${owner}`);
+  }
+  console.log();
+}
 async function listModelsSummary() {
   const provider = getActiveProviderId();
   const apiType = provider ?? "auto";
@@ -6583,6 +6719,12 @@ async function listModelsSummary() {
     console.log(c.red(`  ✗ Gemini: ${error instanceof Error ? error.message : String(error)}`));
     console.log();
   }
+  try {
+    await listAntigravityModels();
+  } catch (error) {
+    console.log(c.red(`  ✗ Antigravity: ${error instanceof Error ? error.message : String(error)}`));
+    console.log();
+  }
   await listAnthropicModels();
 }
 function printDebug() {
@@ -6602,16 +6744,20 @@ function printDebug() {
   console.log(`  ${c.cyan("ANTHROPIC_MODEL:")} ${process.env.ANTHROPIC_MODEL || c.dim("(not set)")}`);
   console.log(`  ${c.cyan("GEMINI_MODEL:")}    ${process.env.GEMINI_MODEL || c.dim("(not set)")}`);
   console.log(`  ${c.cyan("GEMINI_BASE_URL:")} ${process.env.GEMINI_BASE_URL || c.dim("(not set)")}`);
-  console.log(`  ${c.cyan("OPENAI_MODEL:")}    ${process.env.OPENAI_MODEL || c.dim("(not set)")}`);
-  console.log(`  ${c.cyan("OPENAI_BASE_URL:")} ${process.env.OPENAI_BASE_URL || c.dim("(not set)")}`);
+  console.log(`  ${c.cyan("OPENAI_MODEL:")}       ${process.env.OPENAI_MODEL || c.dim("(not set)")}`);
+  console.log(`  ${c.cyan("OPENAI_BASE_URL:")}    ${process.env.OPENAI_BASE_URL || c.dim("(not set)")}`);
+  console.log(`  ${c.cyan("ANTIGRAVITY_MODEL:")}  ${process.env.ANTIGRAVITY_MODEL || c.dim("(not set)")}`);
+  console.log(`  ${c.cyan("ANTIGRAVITY_BASE_URL:")} ${process.env.ANTIGRAVITY_BASE_URL || c.dim("(not set)")}`);
   console.log();
   console.log(c.bold("API Keys:"));
   const anthropicKey = process.env.ANTHROPIC_API_KEY;
   const geminiKey = process.env.GEMINI_API_KEY ?? process.env.GOOGLE_API_KEY;
   const openaiKey = process.env.OPENAI_API_KEY;
-  console.log(`  ${c.cyan("ANTHROPIC_API_KEY:")} ${anthropicKey ? c.green("✓ set") + c.dim(` (${anthropicKey.slice(0, 8)}...${anthropicKey.slice(-4)})`) : c.red("✗ not set")}`);
-  console.log(`  ${c.cyan("GEMINI_API_KEY:")}    ${geminiKey ? c.green("✓ set") + c.dim(` (${geminiKey.slice(0, 8)}...${geminiKey.slice(-4)})`) : c.red("✗ not set")}`);
-  console.log(`  ${c.cyan("OPENAI_API_KEY:")}    ${openaiKey ? c.green("✓ set") + c.dim(` (${openaiKey.slice(0, 8)}...${openaiKey.slice(-4)})`) : c.red("✗ not set")}`);
+  const antigravityKey = process.env.ANTIGRAVITY_API_KEY;
+  console.log(`  ${c.cyan("ANTHROPIC_API_KEY:")}   ${anthropicKey ? c.green("✓ set") + c.dim(` (${anthropicKey.slice(0, 8)}...${anthropicKey.slice(-4)})`) : c.red("✗ not set")}`);
+  console.log(`  ${c.cyan("GEMINI_API_KEY:")}      ${geminiKey ? c.green("✓ set") + c.dim(` (${geminiKey.slice(0, 8)}...${geminiKey.slice(-4)})`) : c.red("✗ not set")}`);
+  console.log(`  ${c.cyan("OPENAI_API_KEY:")}      ${openaiKey ? c.green("✓ set") + c.dim(` (${openaiKey.slice(0, 8)}...${openaiKey.slice(-4)})`) : c.red("✗ not set")}`);
+  console.log(`  ${c.cyan("ANTIGRAVITY_API_KEY:")} ${antigravityKey ? c.green("✓ set") + c.dim(` (${antigravityKey.slice(0, 8)}...${antigravityKey.slice(-4)})`) : c.dim("(default: sk-antigravity)")}`);
   console.log();
   console.log(c.bold("Environment:"));
   console.log(`  ${c.cyan("Working dir:")}  ${cwd}`);
@@ -6675,6 +6821,9 @@ function formatToolInput(name, input) {
   }
 }
 function getDefaultProviderFromEnv() {
+  if (process.env.ANTIGRAVITY_API_KEY || process.env.ANTIGRAVITY_BASE_URL) {
+    return "antigravity";
+  }
   if (process.env.ANTHROPIC_API_KEY) {
     return "anthropic";
   }
@@ -6706,6 +6855,9 @@ function getDefaultModelForProvider(providerId) {
   if (providerId === "gemini") {
     return process.env.GEMINI_MODEL || "gemini-1.5-pro";
   }
+  if (providerId === "antigravity") {
+    return process.env.ANTIGRAVITY_MODEL || "gemini-3-flash";
+  }
   return process.env.OPENAI_MODEL || "gpt-5.2";
 }
 function getActiveProviderId() {
@@ -6738,8 +6890,17 @@ function parseProvider(arg) {
   if (normalized === "gemini" || normalized === "google") {
     return "gemini";
   }
+  if (normalized === "antigravity" || normalized === "ag") {
+    return "antigravity";
+  }
   return null;
 }
+function getAntigravityConfig() {
+  const baseUrl = (process.env.ANTIGRAVITY_BASE_URL || "http://127.0.0.1:8045").replace(/\/+$/, "");
+  const apiKey = process.env.ANTIGRAVITY_API_KEY || "sk-antigravity";
+  const model = process.env.ANTIGRAVITY_MODEL || "gemini-3-flash";
+  return { baseUrl, apiKey, model };
+}
 function createProvider(providerId) {
   if (providerId === "anthropic") {
     return new AnthropicProvider;
@@ -6750,6 +6911,13 @@ function createProvider(providerId) {
       baseUrl: process.env.GEMINI_BASE_URL
     });
   }
+  if (providerId === "antigravity") {
+    const config = getAntigravityConfig();
+    return new OpenAIProvider({
+      apiKey: config.apiKey,
+      baseUrl: `${config.baseUrl}/v1`
+    });
+  }
   return new OpenAIProvider({
     apiKey: process.env.OPENAI_API_KEY,
     baseUrl: process.env.OPENAI_BASE_URL

package/dist/index.js

@@ -480,6 +480,117 @@ function getProviderEnv() {
     }
   };
 }
+// src/utils/retry.ts
+var DEFAULT_RETRY_OPTIONS = {
+  maxAttempts: 3,
+  initialDelay: 1000,
+  maxDelay: 30000,
+  backoffMultiplier: 2,
+  jitter: true,
+  onRetry: () => {},
+  signal: undefined
+};
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+function calculateDelay(attempt, initialDelay, maxDelay, backoffMultiplier, jitter) {
+  const delay = Math.min(initialDelay * Math.pow(backoffMultiplier, attempt), maxDelay);
+  if (jitter) {
+    return delay * (0.5 + Math.random() * 0.5);
+  }
+  return delay;
+}
+function isRetryableStatus(status) {
+  return status >= 500 || status === 429;
+}
+function isRetryableError(error) {
+  if (error instanceof TypeError) {
+    return error.message.includes("fetch") || error.message.includes("network") || error.message.includes("ECONNREFUSED") || error.message.includes("ETIMEDOUT") || error.message.includes("ECONNRESET");
+  }
+  if (error instanceof Error) {
+    const message = error.message.toLowerCase();
+    if (message.includes("rate_limit") || message.includes("rate limit")) {
+      return true;
+    }
+    if (message.includes("usage_limit") || message.includes("usage limit")) {
+      return true;
+    }
+    if (message.includes("timeout") || message.includes("timed out")) {
+      return true;
+    }
+    if (message.includes("5") || message.includes("502") || message.includes("503") || message.includes("504")) {
+      return true;
+    }
+  }
+  return false;
+}
+function extractStatusCode(error) {
+  const match = error.message.match(/(\d{3})/);
+  return match ? parseInt(match[1], 10) : null;
+}
+async function withRetry(fn, options2 = {}) {
+  const opts = { ...DEFAULT_RETRY_OPTIONS, ...options2 };
+  let lastError = null;
+  for (let attempt = 0;attempt < opts.maxAttempts; attempt++) {
+    if (opts.signal?.aborted) {
+      throw new Error("Request aborted");
+    }
+    try {
+      return await fn();
+    } catch (error) {
+      lastError = error instanceof Error ? error : new Error(String(error));
+      if (attempt >= opts.maxAttempts - 1) {
+        throw lastError;
+      }
+      const statusCode = extractStatusCode(lastError);
+      const isRetryable = statusCode !== null ? isRetryableStatus(statusCode) : isRetryableError(lastError);
+      if (!isRetryable) {
+        throw lastError;
+      }
+      const delay = calculateDelay(attempt, opts.initialDelay, opts.maxDelay, opts.backoffMultiplier, opts.jitter);
+      opts.onRetry(attempt + 1, lastError);
+      await sleep(delay);
+    }
+  }
+  throw lastError || new Error("Max retries exceeded");
+}
+async function fetchWithRetry(url, init, retryOptions) {
+  return withRetry(async () => {
+    const response = await fetch(url, init);
+    if (!response.ok) {
+      const errorText = await response.text().catch(() => "");
+      let errorMessage = `HTTP ${response.status}`;
+      try {
+        const errorJson = JSON.parse(errorText);
+        if (errorJson.error?.type) {
+          errorMessage += `: ${errorJson.error.type}`;
+        }
+        if (errorJson.error?.message) {
+          errorMessage += ` - ${errorJson.error.message}`;
+        }
+      } catch {
+        if (errorText) {
+          errorMessage += ` ${errorText}`;
+        }
+      }
+      const error = new Error(errorMessage);
+      error.status = response.status;
+      error.responseText = errorText;
+      throw error;
+    }
+    return response;
+  }, {
+    signal: init.signal,
+    ...retryOptions
+  });
+}
+function getRetryOptionsFromEnv() {
+  return {
+    maxAttempts: Number.parseInt(process.env.FORMAGENT_RETRY_MAX_ATTEMPTS || "3", 10),
+    initialDelay: Number.parseInt(process.env.FORMAGENT_RETRY_INITIAL_DELAY || "1000", 10),
+    maxDelay: Number.parseInt(process.env.FORMAGENT_RETRY_MAX_DELAY || "30000", 10)
+  };
+}
 // src/utils/truncation.ts
 import * as fs from "node:fs/promises";
 import * as path from "node:path";
@@ -2785,6 +2896,7 @@ class AnthropicProvider {
     /^claude-instant/
   ];
   config;
+  defaultRetryOptions;
   constructor(config = {}) {
     const apiKey = config.apiKey ?? process.env.ANTHROPIC_API_KEY;
     if (!apiKey) {
@@ -2794,7 +2906,15 @@ class AnthropicProvider {
       apiKey,
       baseUrl: config.baseUrl ?? process.env.ANTHROPIC_BASE_URL ?? "https://api.anthropic.com",
       apiVersion: config.apiVersion ?? "2023-06-01",
-      defaultMaxTokens: config.defaultMaxTokens ?? 4096
+      defaultMaxTokens: config.defaultMaxTokens ?? 4096,
+      retry: config.retry
+    };
+    this.defaultRetryOptions = {
+      maxAttempts: 3,
+      initialDelay: 1000,
+      maxDelay: 30000,
+      backoffMultiplier: 2,
+      jitter: true
     };
   }
   supportsModel(model) {
@@ -2802,31 +2922,25 @@ class AnthropicProvider {
   }
   async complete(request) {
     const anthropicRequest = this.buildRequest(request, false);
-    const response = await fetch(`${this.config.baseUrl}/v1/messages`, {
+    const retryOptions = this.config.retry ?? this.defaultRetryOptions;
+    const response = await fetchWithRetry(`${this.config.baseUrl}/v1/messages`, {
       method: "POST",
       headers: this.getHeaders(),
       body: JSON.stringify(anthropicRequest),
       signal: request.abortSignal
-    });
-    if (!response.ok) {
-      const error = await response.text();
-      throw new Error(`Anthropic API error: ${response.status} ${error}`);
-    }
+    }, retryOptions);
     const data = await response.json();
     return this.convertResponse(data);
   }
   async stream(request, options2) {
     const anthropicRequest = this.buildRequest(request, true);
-    const response = await fetch(`${this.config.baseUrl}/v1/messages`, {
+    const retryOptions = this.config.retry ?? this.defaultRetryOptions;
+    const response = await fetchWithRetry(`${this.config.baseUrl}/v1/messages`, {
       method: "POST",
       headers: this.getHeaders(),
       body: JSON.stringify(anthropicRequest),
       signal: request.abortSignal
-    });
-    if (!response.ok) {
-      const error = await response.text();
-      throw new Error(`Anthropic API error: ${response.status} ${error}`);
-    }
+    }, retryOptions);
     return this.createStreamIterator(response.body, options2);
   }
   buildRequest(request, stream) {
@@ -3050,6 +3164,7 @@ class OpenAIProvider {
     /^chatgpt/
   ];
   config;
+  defaultRetryOptions;
   constructor(config = {}) {
     const apiKey = config.apiKey ?? process.env.OPENAI_API_KEY;
     if (!apiKey) {
@@ -3059,96 +3174,88 @@ class OpenAIProvider {
       apiKey,
       baseUrl: this.normalizeBaseUrl(config.baseUrl ?? process.env.OPENAI_BASE_URL ?? "https://api.openai.com/v1"),
       organization: config.organization,
-      defaultMaxTokens: config.defaultMaxTokens ?? 4096
+      defaultMaxTokens: config.defaultMaxTokens ?? 4096,
+      retry: config.retry
+    };
+    this.defaultRetryOptions = {
+      maxAttempts: 3,
+      initialDelay: 1000,
+      maxDelay: 30000,
+      backoffMultiplier: 2,
+      jitter: true
     };
   }
   supportsModel(model) {
     return this.supportedModels.some((pattern) => pattern.test(model));
   }
   async complete(request) {
+    const retryOptions = this.config.retry ?? this.defaultRetryOptions;
     if (this.usesResponsesApi(request.config.model)) {
       const openaiRequest2 = this.buildResponsesRequest(request, false);
-      const response2 = await fetch(`${this.config.baseUrl}/responses`, {
+      const response2 = await fetchWithRetry(`${this.config.baseUrl}/responses`, {
         method: "POST",
         headers: this.getHeaders(),
         body: JSON.stringify(openaiRequest2),
         signal: request.abortSignal
-      });
-      if (!response2.ok) {
-        const error = await response2.text();
-        throw new Error(`OpenAI API error: ${response2.status} ${error}`);
-      }
+      }, retryOptions);
       const data2 = await response2.json();
       return this.convertResponsesResponse(data2);
     }
     const openaiRequest = this.buildRequest(request, false);
-    let response = await fetch(`${this.config.baseUrl}/chat/completions`, {
+    const response = await fetchWithRetry(`${this.config.baseUrl}/chat/completions`, {
       method: "POST",
       headers: this.getHeaders(),
       body: JSON.stringify(openaiRequest),
       signal: request.abortSignal
-    });
-    if (!response.ok) {
-      const error = await response.text();
-      if (this.shouldFallbackToResponses(response.status, error)) {
+    }, retryOptions);
+    if (response.status === 404) {
+      const errorText = await response.clone().text();
+      if (this.shouldFallbackToResponses(404, errorText)) {
         const fallbackRequest = this.buildResponsesRequest(request, false);
-        response = await fetch(`${this.config.baseUrl}/responses`, {
+        const fallbackResponse = await fetchWithRetry(`${this.config.baseUrl}/responses`, {
           method: "POST",
           headers: this.getHeaders(),
           body: JSON.stringify(fallbackRequest),
           signal: request.abortSignal
-        });
-        if (!response.ok) {
-          const fallbackError = await response.text();
-          throw new Error(`OpenAI API error: ${response.status} ${fallbackError}`);
-        }
-        const data2 = await response.json();
+        }, retryOptions);
+        const data2 = await fallbackResponse.json();
         return this.convertResponsesResponse(data2);
       }
-      throw new Error(`OpenAI API error: ${response.status} ${error}`);
     }
     const data = await response.json();
     return this.convertResponse(data);
   }
   async stream(request, options2) {
+    const retryOptions = this.config.retry ?? this.defaultRetryOptions;
     if (this.usesResponsesApi(request.config.model)) {
       const openaiRequest2 = this.buildResponsesRequest(request, true);
-      const response2 = await fetch(`${this.config.baseUrl}/responses`, {
+      const response2 = await fetchWithRetry(`${this.config.baseUrl}/responses`, {
         method: "POST",
         headers: this.getHeaders(),
         body: JSON.stringify(openaiRequest2),
         signal: request.abortSignal
-      });
-      if (!response2.ok) {
-        const error = await response2.text();
-        throw new Error(`OpenAI API error: ${response2.status} ${error}`);
-      }
+      }, retryOptions);
       return this.createResponsesStreamIterator(response2.body, options2);
     }
     const openaiRequest = this.buildRequest(request, true);
-    let response = await fetch(`${this.config.baseUrl}/chat/completions`, {
+    const response = await fetchWithRetry(`${this.config.baseUrl}/chat/completions`, {
       method: "POST",
       headers: this.getHeaders(),
       body: JSON.stringify(openaiRequest),
       signal: request.abortSignal
-    });
-    if (!response.ok) {
-      const error = await response.text();
-      if (this.shouldFallbackToResponses(response.status, error)) {
+    }, retryOptions);
+    if (response.status === 404) {
+      const errorText = await response.clone().text();
+      if (this.shouldFallbackToResponses(404, errorText)) {
         const fallbackRequest = this.buildResponsesRequest(request, true);
-        response = await fetch(`${this.config.baseUrl}/responses`, {
+        const fallbackResponse = await fetchWithRetry(`${this.config.baseUrl}/responses`, {
           method: "POST",
           headers: this.getHeaders(),
           body: JSON.stringify(fallbackRequest),
           signal: request.abortSignal
-        });
-        if (!response.ok) {
-          const fallbackError = await response.text();
-          throw new Error(`OpenAI API error: ${response.status} ${fallbackError}`);
-        }
-        return this.createResponsesStreamIterator(response.body, options2);
+        }, retryOptions);
+        return this.createResponsesStreamIterator(fallbackResponse.body, options2);
       }
-      throw new Error(`OpenAI API error: ${response.status} ${error}`);
     }
     return this.createStreamIterator(response.body, options2);
   }
@@ -3866,6 +3973,7 @@ class GeminiProvider {
   name = "Gemini";
   supportedModels = [/^gemini-/, /^models\/gemini-/];
   config;
+  defaultRetryOptions;
   constructor(config = {}) {
     const apiKey = config.apiKey ?? process.env.GEMINI_API_KEY ?? process.env.GOOGLE_API_KEY;
     if (!apiKey) {
@@ -3874,7 +3982,15 @@ class GeminiProvider {
     this.config = {
       apiKey,
       baseUrl: config.baseUrl ?? process.env.GEMINI_BASE_URL ?? "https://generativelanguage.googleapis.com/v1beta",
-      defaultMaxTokens: config.defaultMaxTokens ?? 4096
+      defaultMaxTokens: config.defaultMaxTokens ?? 4096,
+      retry: config.retry
+    };
+    this.defaultRetryOptions = {
+      maxAttempts: 3,
+      initialDelay: 1000,
+      maxDelay: 30000,
+      backoffMultiplier: 2,
+      jitter: true
     };
   }
   supportsModel(model) {
@@ -3883,16 +3999,13 @@ class GeminiProvider {
   async complete(request) {
     const geminiRequest = this.buildRequest(request);
     const url = this.buildUrl(this.getModelPath(request.config.model) + ":generateContent");
-    const response = await fetch(url, {
+    const retryOptions = this.config.retry ?? this.defaultRetryOptions;
+    const response = await fetchWithRetry(url, {
       method: "POST",
       headers: this.getHeaders(),
       body: JSON.stringify(geminiRequest),
       signal: request.abortSignal
-    });
-    if (!response.ok) {
-      const error = await response.text();
-      throw new Error(`Gemini API error: ${response.status} ${error}`);
-    }
+    }, retryOptions);
     const data = await response.json();
     return this.convertResponse(data, request.config.model);
   }
@@ -3901,16 +4014,13 @@ class GeminiProvider {
     const url = this.buildUrl(this.getModelPath(request.config.model) + ":streamGenerateContent", {
       alt: "sse"
     });
-    const response = await fetch(url, {
+    const retryOptions = this.config.retry ?? this.defaultRetryOptions;
+    const response = await fetchWithRetry(url, {
       method: "POST",
       headers: this.getHeaders(),
       body: JSON.stringify(geminiRequest),
       signal: request.abortSignal
-    });
-    if (!response.ok) {
-      const error = await response.text();
-      throw new Error(`Gemini API error: ${response.status} ${error}`);
-    }
+    }, retryOptions);
     const contentType = response.headers.get("content-type") ?? "";
     if (!contentType.includes("text/event-stream")) {
       const data = await response.json();
@@ -4778,10 +4888,13 @@ function isZodSchema(schema) {
   return typeof schema === "object" && schema !== null && "parse" in schema && "safeParse" in schema && typeof schema.parse === "function" && typeof schema.safeParse === "function";
 }
 function zodToJsonSchema(zodSchema) {
-  const shape = zodSchema._def?.shape;
-  const typeName = zodSchema._def?.typeName;
-  if (typeName === "ZodObject" && shape) {
-    return objectShapeToJsonSchema(shape);
+  const def = zodSchema._def;
+  const typeName = def?.typeName;
+  if (typeName === "ZodObject") {
+    const shape = typeof def.shape === "function" ? def.shape() : def.shape;
+    if (shape) {
+      return objectShapeToJsonSchema(shape);
+    }
   }
   return {
     type: "object",
@@ -4794,13 +4907,35 @@ function objectShapeToJsonSchema(shape) {
   const required = [];
   for (const [key, value] of Object.entries(shape)) {
     const def = value?._def;
-    const typeName = def?.typeName;
-    const description = def?.description;
+    let typeName = def?.typeName;
+    let description = def?.description;
     let innerDef = def;
     let isOptional = false;
-    if (typeName === "ZodOptional") {
-      isOptional = true;
-      innerDef = def.innerType?._def;
+    while (innerDef) {
+      const innerTypeName = innerDef.typeName;
+      if (innerTypeName === "ZodOptional") {
+        isOptional = true;
+        if (!description && innerDef.description) {
+          description = innerDef.description;
+        }
+        innerDef = innerDef.innerType?._def;
+      } else if (innerTypeName === "ZodDefault") {
+        isOptional = true;
+        if (!description && innerDef.description) {
+          description = innerDef.description;
+        }
+        innerDef = innerDef.innerType?._def;
+      } else if (innerTypeName === "ZodNullable") {
+        if (!description && innerDef.description) {
+          description = innerDef.description;
+        }
+        innerDef = innerDef.innerType?._def;
+      } else {
+        if (!description && innerDef.description) {
+          description = innerDef.description;
+        }
+        break;
+      }
     }
     const prop = zodDefToJsonSchema(innerDef);
     if (description) {
@@ -4833,8 +4968,10 @@ function zodDefToJsonSchema(def) {
         type: "array",
         items: zodDefToJsonSchema(def.type?._def)
       };
-    case "ZodObject":
-      return objectShapeToJsonSchema(def.shape);
+    case "ZodObject": {
+      const shape = typeof def.shape === "function" ? def.shape() : def.shape;
+      return objectShapeToJsonSchema(shape);
+    }
     case "ZodLiteral":
       return { type: typeof def.value, enum: [def.value] };
     case "ZodUnion":
@@ -7938,6 +8075,7 @@ function createAgent(config) {
 }
 export {
   zodToJsonSchema,
+  withRetry,
   webTools,
   waitForEvent,
   utilityTools,
@@ -7969,6 +8107,7 @@ export {
   interactiveTools,
   globalCostTracker,
   getTodos,
+  getRetryOptionsFromEnv,
   getProviderEnv,
   getProjectSkillsPath,
   getPreset,
@@ -7985,6 +8124,7 @@ export {
   generateEnvContext,
   forkSession,
   fileTools,
+  fetchWithRetry,
   extractTitle,
   extractDescription,
   estimateTokens,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "formagent-sdk",
-  "version": "0.3.3",
+  "version": "0.4.0",
   "description": "FormAgent SDK - AI Agent framework with streaming support, tool execution, skill management, and hooks system",
   "type": "module",
   "main": "./dist/index.js",