pi-provider-antigravity 0.15.0 → 0.16.0

package/README.md

@@ -22,7 +22,7 @@ Use Google Antigravity models directly in [Pi](https://pi.dev).
 
 ## Requirements
 
-- Pi >= 0.86.0 (For Pi 0.80.0 ~ 0.85.x, see [Legacy Pi Support](#legacy-pi-support))
+- **Pi >= 0.86.0** (Pi 0.86+ introduced breaking changes; older versions are no longer supported. Please upgrade Pi, or pin to `0.13.0` via `pi install npm:pi-provider-antigravity@0.13.0` if using Pi 0.80 ~ 0.85.)
 - A Google account with Antigravity access
 
 ## Installation
@@ -31,28 +31,6 @@ Use Google Antigravity models directly in [Pi](https://pi.dev).
 pi install npm:pi-provider-antigravity
 ```
 
-Or try it without installing:
-
-```bash
-pi -e npm:pi-provider-antigravity
-```
-
-### Legacy Pi Support
-
-We strongly recommend upgrading to Pi >= 0.86.0 to take advantage of the latest SDK capabilities, streaming reliability improvements, and full feature parity.
-
-However, if you are unable to upgrade your Pi installation just yet (for environment or workflow reasons), you can pin to the compatible `0.13.0` release for Pi 0.80.0 ~ 0.85.x:
-
-```bash
-pi install npm:pi-provider-antigravity@0.13.0
-```
-
-Or try it without installing:
-
-```bash
-pi -e npm:pi-provider-antigravity@0.13.0
-```
-
 ## Quick Start
 
 1. Authenticate:

package/dist/index.js

@@ -1,12 +1,9 @@
-// src/index.ts
-import { Type } from "@earendil-works/pi-ai";
-
 // src/auth.ts
 import { createHash, randomBytes } from "node:crypto";
 
 // src/protocol.ts
 var DEFAULT_ENDPOINT = "https://daily-cloudcode-pa.googleapis.com";
-var DEFAULT_USER_AGENT = "antigravity/cli/1.2.5 (aidev_client; os_type=linux; arch=amd64; cl=982839923; auth_method=consumer)";
+var DEFAULT_USER_AGENT = "antigravity/cli/1.2.7 (aidev_client; os_type=linux; arch=amd64; cl=984112147; auth_method=consumer)";
 var PROVIDER_ID = "antigravity";
 var AntigravityHttpError = class extends Error {
   status;
@@ -787,33 +784,6 @@ function saveSubsystemState(subsystem, state, file = defaultConfigFile()) {
     config.states = states;
   });
 }
-var SettingsRegistry = class {
-  fields = [];
-  register(field) {
-    const existingIndex = this.fields.findIndex((f) => f.key === field.key);
-    if (existingIndex >= 0) {
-      this.fields[existingIndex] = field;
-    } else {
-      this.fields.push(field);
-    }
-  }
-  unregister(key) {
-    this.fields = this.fields.filter((f) => f.key !== key);
-  }
-  getFields() {
-    return [...this.fields];
-  }
-  clear() {
-    this.fields = [];
-  }
-};
-var defaultSettingsRegistry = new SettingsRegistry();
-function registerSettingField(field) {
-  defaultSettingsRegistry.register(field);
-}
-function getRegisteredSettingFields() {
-  return defaultSettingsRegistry.getFields();
-}
 function fieldDisplayValue(config, field) {
   const raw = config?.settings?.[field.key];
   if (typeof raw === "string" && raw.trim()) return raw.trim();
@@ -836,7 +806,7 @@ async function applySettingValue(field, value, file, ctx) {
 function writeFailedMessage(file) {
   return `Failed to write ${file}. If the file is not valid JSON, fix or delete it.`;
 }
-function buildSettingsItems(config, fields = getRegisteredSettingFields()) {
+function buildSettingsItems(config, fields) {
   return fields.map((f) => ({
     id: f.key,
     label: f.label,
@@ -859,15 +829,7 @@ async function settingsListTheme(theme) {
     };
   }
 }
-function resolveFields(registryOrFields) {
-  if (!registryOrFields) return getRegisteredSettingFields();
-  if ("getFields" in registryOrFields && typeof registryOrFields.getFields === "function") {
-    return registryOrFields.getFields();
-  }
-  return registryOrFields;
-}
-async function openSettings(ctx, registryOrFields) {
-  const fields = resolveFields(registryOrFields);
+async function openSettings(ctx, fields) {
   const file = defaultConfigFile();
   if (ctx.mode === "tui" && ctx.hasUI) {
     await openSettingsDialog(ctx, fields, file);
@@ -1007,11 +969,11 @@ function sortQuotaBuckets(buckets) {
     (a, b) => QUOTA_WINDOW_RANK[classifyQuotaWindow(a)] - QUOTA_WINDOW_RANK[classifyQuotaWindow(b)]
   );
 }
-function formatResetRemaining(resetTime) {
+function formatResetRemaining(resetTime, nowFn = Date.now) {
   if (!resetTime) return null;
   const ts = Date.parse(resetTime);
   if (!Number.isFinite(ts)) return null;
-  const delta = ts - Date.now();
+  const delta = ts - nowFn();
   if (delta <= 0) return "ready";
   const totalMin = Math.ceil(delta / 6e4);
   const days = Math.floor(totalMin / (60 * 24));
@@ -1046,7 +1008,7 @@ function parseQuotaSummary(data) {
     description: data.description
   };
 }
-function formatQuotaSummary(summary) {
+function formatQuotaSummary(summary, nowFn = Date.now) {
   const lines = [];
   for (const group of summary.groups) {
     if (lines.length > 0) lines.push("");
@@ -1054,7 +1016,7 @@ function formatQuotaSummary(summary) {
     for (const b of sortQuotaBuckets(group.buckets)) {
       const pct = Math.round(b.remainingFraction * 100);
       const label = shortWindowLabel(b, classifyQuotaWindow(b)).padEnd(3);
-      const reset = formatResetRemaining(b.resetTime);
+      const reset = formatResetRemaining(b.resetTime, nowFn);
       const suffix = reset && pct < 100 ? ` (${reset})` : "";
       lines.push(
         `  ${label} ${renderProgressBar(b.remainingFraction)} ${String(pct).padStart(3)}% left${suffix}`
@@ -1127,22 +1089,22 @@ function selectUrgentBucket(group, weeklyTo5hRatio) {
   }
   return worst;
 }
-function buildQuotaFooter(summary, modelId, weeklyTo5hRatio = DEFAULT_WEEKLY_TO_5H_RATIO) {
+function buildQuotaFooter(summary, modelId, weeklyTo5hRatio = DEFAULT_WEEKLY_TO_5H_RATIO, nowFn = Date.now) {
   const group = selectQuotaGroup(summary.groups, modelId);
   if (!group) return void 0;
   const urgent = selectUrgentBucket(group, weeklyTo5hRatio);
-  return urgent ? formatQuotaWindowPart(urgent) : void 0;
+  return urgent ? formatQuotaWindowPart(urgent, nowFn) : void 0;
 }
-function buildQuotaFooterBoth(summary, modelId) {
+function buildQuotaFooterBoth(summary, modelId, nowFn = Date.now) {
   const group = selectQuotaGroup(summary.groups, modelId);
   if (!group || group.buckets.length === 0) return void 0;
-  return sortQuotaBuckets(group.buckets).map(formatQuotaWindowPart).join(" \xB7 ");
+  return sortQuotaBuckets(group.buckets).map((b) => formatQuotaWindowPart(b, nowFn)).join(" \xB7 ");
 }
-function formatQuotaWindowPart(bucket) {
+function formatQuotaWindowPart(bucket, nowFn = Date.now) {
   const kind = classifyQuotaWindow(bucket);
   const prefix = kind === "5h" ? "5h" : kind === "weekly" ? "Wk" : bucket.displayName || bucket.bucketId;
   const pct = Math.round(bucket.remainingFraction * 100);
-  const short = formatResetRemaining(bucket.resetTime)?.replace(/^in /, "");
+  const short = formatResetRemaining(bucket.resetTime, nowFn)?.replace(/^in /, "");
   if (short && pct < 100) return `${prefix} ${pct}% (${short})`;
   return `${prefix} ${pct}%`;
 }
@@ -1202,25 +1164,22 @@ function createQuotaFooterField(quotaStatus) {
     options: FOOTER_MODE_OPTIONS,
     defaultValue: "off",
     optionNotes: FOOTER_MODE_NOTES,
-    renderPreview: (mode, ctx, theme) => {
+    renderPreview: quotaStatus ? (mode, ctx, theme) => {
       if (mode === "off") return "hidden";
       const activeTheme = theme || ctx.ui?.theme;
       return previewQuotaFooterText(quotaStatus, ctx.model?.id, mode, activeTheme);
-    },
-    prepare: async (ctx) => {
+    } : void 0,
+    prepare: quotaStatus ? async (ctx) => {
       await quotaStatus.ensurePreview(ctx);
-    },
-    onChange: async (ctx, _value) => {
+    } : void 0,
+    onChange: quotaStatus ? async (ctx, _value) => {
       if (quotaStatus.mode() !== "off") {
         await quotaStatus.refresh(ctx, { ignoreMode: true });
       }
       quotaStatus.paint(ctx);
-    }
+    } : void 0
   };
 }
-function registerQuotaSettings(quotaStatus) {
-  registerSettingField(createQuotaFooterField(quotaStatus));
-}
 function isFiniteNumber(value) {
   return typeof value === "number" && Number.isFinite(value);
 }
@@ -1251,18 +1210,18 @@ var FOOTER_MODES = Object.freeze({
   smart: {
     key: "smart",
     note: FOOTER_MODE_NOTES.smart,
-    render: (summary, modelId, ratio = DEFAULT_WEEKLY_TO_5H_RATIO, style) => {
+    render: (summary, modelId, ratio = DEFAULT_WEEKLY_TO_5H_RATIO, style, nowFn = Date.now) => {
       if (!summary) return {};
-      const plain = buildQuotaFooter(summary, modelId, ratio);
+      const plain = buildQuotaFooter(summary, modelId, ratio, nowFn);
       return { plain, colored: colorizeQuotaFooter(plain, style) };
     }
   },
   all: {
     key: "all",
     note: FOOTER_MODE_NOTES.all,
-    render: (summary, modelId, _ratio, style) => {
+    render: (summary, modelId, _ratio, style, nowFn = Date.now) => {
       if (!summary) return {};
-      const plain = buildQuotaFooterBoth(summary, modelId);
+      const plain = buildQuotaFooterBoth(summary, modelId, nowFn);
       return { plain, colored: colorizeQuotaFooterBoth(plain, style) };
     }
   }
@@ -1277,11 +1236,19 @@ var QuotaStatusCoordinator = class {
   weeklyTo5hRatio;
   lastPairs;
   store;
-  constructor(storeOrFile) {
-    if (typeof storeOrFile === "string" || !storeOrFile) {
-      this.store = fileQuotaStatusStore(storeOrFile || defaultConfigFile());
+  fetchSummaryOverride;
+  nowFn;
+  constructor(depsOrStoreOrFile) {
+    if (typeof depsOrStoreOrFile === "string" || !depsOrStoreOrFile) {
+      this.store = fileQuotaStatusStore(depsOrStoreOrFile || defaultConfigFile());
+      this.nowFn = Date.now;
+    } else if ("loadMode" in depsOrStoreOrFile) {
+      this.store = depsOrStoreOrFile;
+      this.nowFn = Date.now;
     } else {
-      this.store = storeOrFile;
+      this.store = depsOrStoreOrFile.store ?? fileQuotaStatusStore(depsOrStoreOrFile.configFile || defaultConfigFile());
+      this.fetchSummaryOverride = depsOrStoreOrFile.fetchQuotaSummary;
+      this.nowFn = depsOrStoreOrFile.now ?? Date.now;
     }
     this.weeklyTo5hRatio = DEFAULT_WEEKLY_TO_5H_RATIO;
     this.loadPersisted();
@@ -1294,13 +1261,13 @@ var QuotaStatusCoordinator = class {
     return this.store.loadMode();
   }
   footerFor(modelId, mode = "smart") {
-    return FOOTER_MODES[mode]?.render(this.summary, modelId, this.weeklyTo5hRatio).plain;
+    return FOOTER_MODES[mode]?.render(this.summary, modelId, this.weeklyTo5hRatio, void 0, this.nowFn).plain;
   }
   renderFooter(modelId, mode = "smart", style) {
-    return FOOTER_MODES[mode]?.render(this.summary, modelId, this.weeklyTo5hRatio, style) ?? {};
+    return FOOTER_MODES[mode]?.render(this.summary, modelId, this.weeklyTo5hRatio, style, this.nowFn) ?? {};
   }
   get isFresh() {
-    return !!this.summary && Date.now() - this.fetchedAt < QUOTA_REFRESH_INTERVAL_MS;
+    return !!this.summary && this.nowFn() - this.fetchedAt < QUOTA_REFRESH_INTERVAL_MS;
   }
   refresh(ctx, opts = {}) {
     if (!isLiveCtx(ctx)) return Promise.resolve(void 0);
@@ -1339,7 +1306,10 @@ var QuotaStatusCoordinator = class {
       throw new Error("Failed to fetch usage.");
     }
     this.paint(ctx);
-    return formatQuotaSummary(summary);
+    return formatQuotaSummary(summary, this.nowFn);
+  }
+  createSettingsField() {
+    return createQuotaFooterField(this);
   }
   // One fetch for preview purposes even when the footer slot is off or the
   // current model belongs to another provider: opening quota settings is an
@@ -1360,22 +1330,29 @@ var QuotaStatusCoordinator = class {
     try {
       const creds = await resolveCredentials(ctx);
       if (!creds) return void 0;
-      const rawSummary = await postAntigravityJson({
-        auth: creds,
-        path: "v1internal:retrieveUserQuotaSummary",
-        body: { project: creds.projectId },
-        signal
-      });
-      const summary = parseQuotaSummary(rawSummary);
+      let rawSummary;
+      if (this.fetchSummaryOverride) {
+        rawSummary = await this.fetchSummaryOverride(creds.token, signal);
+      } else {
+        rawSummary = await postAntigravityJson({
+          auth: creds,
+          path: "v1internal:retrieveUserQuotaSummary",
+          body: { project: creds.projectId },
+          signal
+        });
+      }
+      if (!rawSummary) return void 0;
+      const summary = rawSummary.groups ? parseQuotaSummary(rawSummary) : rawSummary;
       this.ingest(summary);
       return summary;
     } catch {
       return void 0;
     }
   }
-  // Accepts an externally fetched summary into the shared cache: same
+  // Accepts an externally fetched summary or raw JSON into the shared cache: same
   // calibration, persistence, and freshness as fetch().
-  ingest(summary) {
+  ingest(summaryOrRaw) {
+    const summary = summaryOrRaw && summaryOrRaw.groups && typeof summaryOrRaw.groups[0]?.buckets?.[0]?.remainingFraction === "number" ? parseQuotaSummary(summaryOrRaw) : summaryOrRaw;
     const calibrated = calibrateWeeklyTo5hRatio(
       this.lastPairs,
       extractWindowFractionPairs(summary),
@@ -1384,7 +1361,7 @@ var QuotaStatusCoordinator = class {
     this.weeklyTo5hRatio = calibrated;
     this.lastPairs = extractWindowFractionPairs(summary);
     this.summary = summary;
-    this.fetchedAt = Date.now();
+    this.fetchedAt = this.nowFn();
     this.saveState();
   }
   loadPersisted() {
@@ -1394,7 +1371,7 @@ var QuotaStatusCoordinator = class {
       this.weeklyTo5hRatio = ratio;
     }
     const at = entry?.updatedAt;
-    this.lastPairs = typeof at === "number" && Date.now() - at <= MAX_OBSERVATION_AGE_MS ? validPairs(entry?.previousObservation) : void 0;
+    this.lastPairs = typeof at === "number" && this.nowFn() - at <= MAX_OBSERVATION_AGE_MS ? validPairs(entry?.previousObservation) : void 0;
   }
   // Persists calibration state through the injected store. Best-effort: a
   // stale ratio is still usable.
@@ -1403,12 +1380,15 @@ var QuotaStatusCoordinator = class {
       this.store.saveQuotaState({
         weeklyTo5hRatio: this.weeklyTo5hRatio,
         previousObservation: this.lastPairs ?? {},
-        updatedAt: Date.now()
+        updatedAt: this.nowFn()
       });
     } catch {
     }
   }
 };
+function createQuotaStatus(deps) {
+  return new QuotaStatusCoordinator(deps);
+}
 
 // src/stream.ts
 import {
@@ -1808,6 +1788,11 @@ function attachLoneSignature(content, lastThoughtSignature) {
   }
   content.push({ type: "text", text: "", textSignature: lastThoughtSignature });
 }
+function hasAnswerContent(content) {
+  return content.some(
+    (block) => block.type === "toolCall" ? true : block.type === "text" ? block.text.trim().length > 0 : false
+  );
+}
 function closeOpenBlock(state, onEvent) {
   if (state.openType === null) return;
   const contentIndex = state.content.length - 1;
@@ -2025,13 +2010,14 @@ function streamAntigravity(model, context, options, catalog) {
       const effort = options?.reasoning;
       const trajectoryId = options && "trajectoryId" in options && typeof options.trajectoryId === "string" ? options.trajectoryId : void 0;
       const plan = catalog.resolvePlan(model.id, effort);
+      const maxOutputTokens = options?.maxTokens ?? model.maxTokens;
       let requestBody = buildAntigravityRequestBody({
         projectId,
         plan,
         context,
         sessionId: options?.sessionId,
         trajectoryId,
-        maxOutputTokens: options?.maxTokens ?? model.maxTokens,
+        maxOutputTokens,
         toolChoice: options?.toolChoice
       });
       if (options?.onPayload) {
@@ -2079,6 +2065,19 @@ function streamAntigravity(model, context, options, catalog) {
           output.rawStopReason ? `Provider stopped with: ${output.rawStopReason}` : output.errorMessage || "An unknown error occurred"
         );
       }
+      if (output.stopReason === "stop" && !hasAnswerContent(output.content)) {
+        const reasoningTokens = output.usage.reasoning ?? 0;
+        const budgetThreshold = Math.max(1, Math.floor(maxOutputTokens * 0.9));
+        const isDeliberatedToBudget = reasoningTokens >= budgetThreshold;
+        if (isDeliberatedToBudget) {
+          throw new Error(
+            `Antigravity returned no answer: the turn ended after thinking only (reasoning ${reasoningTokens} of ${maxOutputTokens} max output tokens). Lower the thinking level or prompt again.`
+          );
+        }
+        throw new Error(
+          `Antigravity returned an empty response with no answer (reasoning ${reasoningTokens} of ${maxOutputTokens} max output tokens). Please retry your request.`
+        );
+      }
       const doneReason = output.stopReason === "toolUse" || output.stopReason === "length" ? output.stopReason : "stop";
       stream.push({
         type: "done",
@@ -2098,8 +2097,10 @@ function streamAntigravity(model, context, options, catalog) {
 }
 
 // src/search.ts
-var SEARCH_SYSTEM_INSTRUCTION = "You are a search engine bot. You will be given a query from a user. Your task is to search the web for relevant information that will help the user. You MUST perform a web search. Do not respond or interact with the user, please respond as if they typed the query into a search bar.";
+import { Type } from "@earendil-works/pi-ai";
+var WEB_SEARCH_TOOL_NAME = "antigravity_websearch";
 var SEARCH_MODEL = "gemini-3.1-flash-lite";
+var SEARCH_SYSTEM_INSTRUCTION = "You are a search engine bot. You will be given a query from a user. Your task is to search the web for relevant information that will help the user. You MUST perform a web search. Do not respond or interact with the user, please respond as if they typed the query into a search bar.";
 function formatSearchResults(rawText, metadata, query, createdAt, completedAt) {
   const sources = (metadata?.groundingChunks || []).map((chunk, i) => ({
     index: i + 1,
@@ -2192,6 +2193,84 @@ async function performWebSearch(creds, options) {
   const metadata = candidate?.groundingMetadata;
   return formatSearchResults(rawText, metadata, effectiveQuery, createdAt, completedAt);
 }
+function notifyUser(ctx, text, type = "info") {
+  if (ctx.hasUI) {
+    ctx.ui.notify(text, type);
+  } else {
+    if (type === "error" || type === "warning") console.error(text);
+    else console.log(text);
+  }
+}
+function registerWebSearchTool(pi) {
+  if (typeof pi.registerTool !== "function") return;
+  pi.registerTool({
+    name: WEB_SEARCH_TOOL_NAME,
+    label: "Antigravity Web Search",
+    description: "Performs a web search for a given query. Returns a summary of relevant information along with URL citations.",
+    promptSnippet: "Search the web using Google Search Grounding with source citations",
+    promptGuidelines: [
+      "Use antigravity_websearch when you need real-time web search, latest documentation, or up-to-date facts with verified citations."
+    ],
+    parameters: Type.Object({
+      query: Type.String({ description: "The search query string to look up on the web." }),
+      domain: Type.Optional(
+        Type.String({
+          description: "Optional domain to recommend the search prioritize (e.g. 'github.com', 'developer.mozilla.org')."
+        })
+      )
+    }),
+    async execute(_toolCallId, params, signal, _onUpdate, ctx) {
+      const creds = await resolveCredentials(ctx);
+      if (!creds) {
+        return {
+          content: [{ type: "text", text: `Error: ${NOT_LOGGED_IN}` }],
+          details: { error: "not_logged_in" }
+        };
+      }
+      try {
+        const result = await performWebSearch(creds, {
+          query: params.query,
+          domain: params.domain,
+          signal
+        });
+        return {
+          content: [{ type: "text", text: result.formattedOutput }],
+          details: {
+            sources: result.sources,
+            queries: result.queries
+          }
+        };
+      } catch (err) {
+        return {
+          content: [{ type: "text", text: `Search failed: ${err.message}` }],
+          details: { error: err.message }
+        };
+      }
+    }
+  });
+}
+async function executeWebSearchCommand(ctx, query) {
+  const creds = await resolveCredentials(ctx);
+  if (!creds) {
+    notifyUser(ctx, NOT_LOGGED_IN, "warning");
+    return;
+  }
+  const trimmedQuery = (query || "").trim();
+  if (!trimmedQuery) {
+    notifyUser(ctx, "Usage: /antigravity websearch <query>", "warning");
+    return;
+  }
+  try {
+    if (ctx.hasUI) ctx.ui.notify(`Searching: "${trimmedQuery}"\u2026`, "info");
+    const result = await performWebSearch(creds, {
+      query: trimmedQuery,
+      signal: ctx.signal
+    });
+    notifyUser(ctx, result.formattedOutput);
+  } catch (err) {
+    notifyUser(ctx, `Search failed: ${err.message}`, "error");
+  }
+}
 
 // src/commands.ts
 function emitOutput(ctx, text, type = "info") {
@@ -2224,14 +2303,14 @@ var SUBCOMMANDS = Object.freeze([
     description: "Configure provider settings",
     aliases: Object.freeze(["setting"]),
     run: async ({ ctx, quotaStatus }) => {
-      if (quotaStatus) registerQuotaSettings(quotaStatus);
-      await openSettings(ctx);
+      const qs = quotaStatus ?? createQuotaStatus();
+      await openSettings(ctx, [qs.createSettingsField()]);
     }
   },
   {
     name: "websearch",
     description: "Search the web using Google Search Grounding",
-    run: async ({ ctx, subArgs }) => runSearchSubcommand(ctx, subArgs)
+    run: async ({ ctx, subArgs }) => executeWebSearchCommand(ctx, subArgs)
   },
   {
     name: "login",
@@ -2338,28 +2417,6 @@ function runLoginSubcommand(ctx) {
     emitOutput(ctx, "Please run /login antigravity to authenticate.");
   }
 }
-async function runSearchSubcommand(ctx, query) {
-  const creds = await resolveCredentials(ctx);
-  if (!creds) {
-    emitOutput(ctx, NOT_LOGGED_IN, "warning");
-    return;
-  }
-  const trimmedQuery = (query || "").trim();
-  if (!trimmedQuery) {
-    emitOutput(ctx, "Usage: /antigravity websearch <query>", "warning");
-    return;
-  }
-  try {
-    if (ctx.hasUI) ctx.ui.notify(`Searching: "${trimmedQuery}"\u2026`, "info");
-    const result = await performWebSearch(creds, {
-      query: trimmedQuery,
-      signal: ctx.signal
-    });
-    emitOutput(ctx, result.formattedOutput);
-  } catch (err) {
-    emitOutput(ctx, `Search failed: ${err.message}`, "error");
-  }
-}
 async function runAntigravitySubcommand(args, ctx, quotaStatus, catalog) {
   const raw = (args || "").trim();
   const parts = raw.split(/\s+/).filter(Boolean);
@@ -2376,8 +2433,7 @@ async function runAntigravitySubcommand(args, ctx, quotaStatus, catalog) {
 // src/index.ts
 var PROVIDER_NAME = "Antigravity";
 function index_default(pi) {
-  const quotaStatus = new QuotaStatusCoordinator();
-  registerQuotaSettings(quotaStatus);
+  const quotaStatus = createQuotaStatus();
   const catalog = createModelCatalog();
   pi.registerProvider(PROVIDER_ID, {
     name: PROVIDER_NAME,
@@ -2402,53 +2458,7 @@ function index_default(pi) {
       await runAntigravitySubcommand(args, ctx, quotaStatus, catalog);
     }
   });
-  if (typeof pi.registerTool === "function") {
-    pi.registerTool({
-      name: "antigravity_websearch",
-      label: "Antigravity Web Search",
-      description: "Performs a web search for a given query. Returns a summary of relevant information along with URL citations.",
-      promptSnippet: "Search the web using Google Search Grounding with source citations",
-      promptGuidelines: [
-        "Use antigravity_websearch when you need real-time web search, latest documentation, or up-to-date facts with verified citations."
-      ],
-      parameters: Type.Object({
-        query: Type.String({ description: "The search query string to look up on the web." }),
-        domain: Type.Optional(
-          Type.String({
-            description: "Optional domain to recommend the search prioritize (e.g. 'github.com', 'developer.mozilla.org')."
-          })
-        )
-      }),
-      async execute(_toolCallId, params, signal, _onUpdate, ctx) {
-        const creds = await resolveCredentials(ctx);
-        if (!creds) {
-          return {
-            content: [{ type: "text", text: "Error: Not logged in. Run /login antigravity first." }],
-            details: { error: "not_logged_in" }
-          };
-        }
-        try {
-          const result = await performWebSearch(creds, {
-            query: params.query,
-            domain: params.domain,
-            signal
-          });
-          return {
-            content: [{ type: "text", text: result.formattedOutput }],
-            details: {
-              sources: result.sources,
-              queries: result.queries
-            }
-          };
-        } catch (err) {
-          return {
-            content: [{ type: "text", text: `Search failed: ${err.message}` }],
-            details: { error: err.message }
-          };
-        }
-      }
-    });
-  }
+  registerWebSearchTool(pi);
   pi.on("session_start", async (_event, ctx) => {
     await quotaStatus.refreshAndPaint(ctx);
   });