pi-provider-antigravity 0.16.0 → 0.17.0

package/dist/index.js

@@ -1,77 +1,25 @@
 // 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.7 (aidev_client; os_type=linux; arch=amd64; cl=984112147; auth_method=consumer)";
+// src/constants.ts
 var PROVIDER_ID = "antigravity";
-var AntigravityHttpError = class extends Error {
-  status;
-  path;
-  errorBody;
-  constructor(status, path2, errorBody) {
-    super(`Antigravity request to ${path2} failed (${status}): ${errorBody}`);
-    this.name = "AntigravityHttpError";
-    this.status = status;
-    this.path = path2;
-    this.errorBody = errorBody;
-  }
-};
-function resolveToken(auth) {
-  if (typeof auth === "string") return auth;
-  if (auth && typeof auth.token === "string") return auth.token;
-  throw new Error("Invalid auth parameter: missing token string");
-}
-async function postAntigravityRaw({
-  auth,
-  path: path2,
-  endpoint = DEFAULT_ENDPOINT,
-  headers,
-  body,
-  signal
-}) {
-  const token = resolveToken(auth);
-  return fetch(`${endpoint}/${path2}`, {
-    method: "POST",
-    headers: { ...buildAntigravityHeaders(token), ...headers },
-    body: JSON.stringify(body),
-    signal
-  });
-}
-async function postAntigravityJson(params) {
-  const res = await postAntigravityRaw(params);
-  if (!res.ok) {
-    const errText = await res.text();
-    throw new AntigravityHttpError(res.status, params.path, errText);
-  }
-  return await res.json();
-}
-async function postAntigravityStream(params) {
-  const res = await postAntigravityRaw(params);
-  if (!res.ok) {
-    const errText = await res.text();
-    throw new AntigravityHttpError(res.status, params.path, errText);
-  }
-  if (!res.body) {
-    throw new Error(`No response stream received from Antigravity for ${params.path}.`);
-  }
-  return { response: res, stream: res.body };
-}
-function buildAntigravityHeaders(token, userAgent = DEFAULT_USER_AGENT) {
-  return {
-    Host: "daily-cloudcode-pa.googleapis.com",
-    "User-Agent": userAgent,
-    "Content-Type": "application/json",
-    Authorization: `Bearer ${token}`
-  };
-}
+var PROVIDER_NAME = "Antigravity";
+var DEFAULT_HOST = "daily-cloudcode-pa.googleapis.com";
+var DEFAULT_ENDPOINT = `https://${DEFAULT_HOST}`;
+var DEFAULT_USER_AGENT = "antigravity/cli/1.2.9 (aidev_client; os_type=linux; arch=amd64; cl=986394137; auth_method=consumer)";
+var FALLBACK_PROJECT_ID = "aicode-consumers";
 
 // src/auth.ts
+var AntigravityAuthError = class extends Error {
+  constructor(message = NOT_LOGGED_IN) {
+    super(message);
+    this.name = "AntigravityAuthError";
+  }
+};
 var REDIRECT_URI = "https://antigravity.google/oauth-callback";
 var AUTH_URL = "https://accounts.google.com/o/oauth2/auth";
 var TOKEN_URL = "https://oauth2.googleapis.com/token";
 var OAUTH_CALLBACK_TIMEOUT_MS = 5 * 60 * 1e3;
-var FALLBACK_PROJECT_ID = "aicode-consumers";
 var SCOPES = [
   "https://www.googleapis.com/auth/cloud-platform",
   "https://www.googleapis.com/auth/userinfo.email",
@@ -112,14 +60,22 @@ function generatePKCE() {
 }
 async function fetchProjectId(token, signal) {
   try {
-    const data = await postAntigravityJson({
-      auth: token,
-      path: "v1internal:loadCodeAssist",
-      body: { metadata: { ideType: "ANTIGRAVITY" } },
+    const res = await fetch(`${DEFAULT_ENDPOINT}/v1internal:loadCodeAssist`, {
+      method: "POST",
+      headers: {
+        Host: DEFAULT_HOST,
+        "User-Agent": DEFAULT_USER_AGENT,
+        "Content-Type": "application/json",
+        Authorization: `Bearer ${token}`
+      },
+      body: JSON.stringify({ metadata: { ideType: "ANTIGRAVITY" } }),
       signal
     });
-    if (data.cloudaicompanionProject) {
-      return data.cloudaicompanionProject;
+    if (res.ok) {
+      const data = await res.json();
+      if (data.cloudaicompanionProject) {
+        return data.cloudaicompanionProject;
+      }
     }
   } catch (err) {
     if (signal?.aborted) throw err;
@@ -262,7 +218,7 @@ var NOT_LOGGED_IN = "Not logged in. Run /login antigravity first.";
 async function requireCredentials(source, errorMessage = NOT_LOGGED_IN) {
   const creds = await resolveCredentials(source);
   if (!creds) {
-    throw new Error(errorMessage);
+    throw new AntigravityAuthError(errorMessage);
   }
   return creds;
 }
@@ -277,20 +233,158 @@ function parseStoredCredentials(rawApiKey) {
   return { token: rawApiKey, projectId: FALLBACK_PROJECT_ID };
 }
 
+// src/protocol.ts
+var AntigravityHttpError = class extends Error {
+  status;
+  path;
+  errorBody;
+  constructor(status, path2, errorBody) {
+    super(`Antigravity request to ${path2} failed (${status}): ${errorBody}`);
+    this.name = "AntigravityHttpError";
+    this.status = status;
+    this.path = path2;
+    this.errorBody = errorBody;
+  }
+};
+function resolveToken(auth) {
+  if (typeof auth === "string") return auth;
+  if (auth && typeof auth.token === "string") return auth.token;
+  throw new Error("Invalid auth parameter: missing token string");
+}
+async function postAntigravityRaw({
+  auth,
+  path: path2,
+  endpoint = DEFAULT_ENDPOINT,
+  headers,
+  body,
+  signal
+}) {
+  const token = resolveToken(auth);
+  return fetch(`${endpoint}/${path2}`, {
+    method: "POST",
+    headers: { ...buildAntigravityHeaders(token), ...headers },
+    body: JSON.stringify(body),
+    signal
+  });
+}
+async function postAntigravityJson(params) {
+  const res = await postAntigravityRaw(params);
+  if (!res.ok) {
+    const errText = await res.text();
+    throw new AntigravityHttpError(res.status, params.path, errText);
+  }
+  return await res.json();
+}
+async function postAntigravityStream(params) {
+  const res = await postAntigravityRaw(params);
+  if (!res.ok) {
+    const errText = await res.text();
+    throw new AntigravityHttpError(res.status, params.path, errText);
+  }
+  if (!res.body) {
+    throw new Error(`No response stream received from Antigravity for ${params.path}.`);
+  }
+  return { response: res, stream: res.body };
+}
+function buildAntigravityHeaders(token, userAgent = DEFAULT_USER_AGENT) {
+  return {
+    Host: DEFAULT_HOST,
+    "User-Agent": userAgent,
+    "Content-Type": "application/json",
+    Authorization: `Bearer ${token}`
+  };
+}
+function createAntigravityClient(options) {
+  const endpoint = options?.endpoint || DEFAULT_ENDPOINT;
+  const userAgent = options?.userAgent || DEFAULT_USER_AGENT;
+  return {
+    endpoint,
+    userAgent,
+    async fetchAvailableModels(source, signal) {
+      const creds = await requireCredentials(source);
+      return postAntigravityJson({
+        auth: creds.token,
+        endpoint,
+        path: "v1internal:fetchAvailableModels",
+        body: { project: creds.projectId },
+        signal
+      });
+    },
+    async retrieveQuotaSummary(source, signal) {
+      const creds = await resolveCredentials(source);
+      if (!creds) return void 0;
+      return postAntigravityJson({
+        auth: creds.token,
+        endpoint,
+        path: "v1internal:retrieveUserQuotaSummary",
+        body: { project: creds.projectId },
+        signal
+      });
+    },
+    async streamGenerateContent(params) {
+      const creds = await requireCredentials(params.source);
+      return postAntigravityStream({
+        auth: creds.token,
+        endpoint: params.endpoint || endpoint,
+        path: "v1internal:streamGenerateContent?alt=sse",
+        headers: params.headers,
+        body: params.body,
+        signal: params.signal
+      });
+    },
+    async postJson(source, path2, body = {}, signal, headers) {
+      const creds = await requireCredentials(source);
+      const effectiveBody = body && typeof body === "object" && !Array.isArray(body) ? { project: creds.projectId, ...body } : body;
+      return postAntigravityJson({
+        auth: creds.token,
+        endpoint,
+        path: path2,
+        body: effectiveBody,
+        headers,
+        signal
+      });
+    }
+  };
+}
+var defaultAntigravityClient = createAntigravityClient();
+
 // src/model-identity.ts
-function classifyModelFamily(modelId) {
+function getModelProfile(modelId) {
   const bare = ((modelId || "").split("/").pop() || "").toLowerCase();
-  if (bare.startsWith("gemini-")) return "gemini";
-  if (bare.startsWith("claude-")) return "claude";
-  if (bare.startsWith("gpt-")) return "gpt";
-  return "unknown";
-}
-function isCompatibleFamily(msgModel, targetModelId) {
-  if (!msgModel || !targetModelId) return true;
-  if (msgModel === targetModelId) return true;
-  const msgFamily = classifyModelFamily(msgModel);
-  if (msgFamily !== "unknown" && msgFamily === classifyModelFamily(targetModelId)) return true;
-  return false;
+  let family = "unknown";
+  if (bare.startsWith("gemini-")) family = "gemini";
+  else if (bare.startsWith("claude-")) family = "claude";
+  else if (bare.startsWith("gpt-")) family = "gpt";
+  const isGemini = family === "gemini";
+  const isClaude = family === "claude";
+  const isGpt = family === "gpt";
+  const isNonGemini = family !== "gemini";
+  const quotaPoolKind = isClaude || isGpt ? "3p" : "gemini";
+  const replaysReasoningHistory = family !== "gpt";
+  const isFlash = bare.includes("flash");
+  const defaultContextWindow = isFlash ? 1048576 : isClaude ? 25e4 : isGpt ? 128e3 : 1048576;
+  const defaultMaxOutputTokens = isClaude ? 64e3 : isGpt ? 32768 : 65536;
+  const promptCache = isClaude ? { short: 300, long: 3600 } : { short: 300 };
+  return {
+    id: modelId,
+    family,
+    isGemini,
+    isClaude,
+    isGpt,
+    isNonGemini,
+    quotaPoolKind,
+    replaysReasoningHistory,
+    defaultContextWindow,
+    defaultMaxOutputTokens,
+    promptCache,
+    isReplayCompatible(candidateModel) {
+      if (!candidateModel || !modelId) return true;
+      if (candidateModel === modelId) return true;
+      const other = getModelProfile(candidateModel);
+      if (other.family !== "unknown" && other.family === family) return true;
+      return false;
+    }
+  };
 }
 
 // src/model-catalog.ts
@@ -432,13 +526,8 @@ function parseAvailableModels(data) {
     ...Object.keys(deprecated).length > 0 ? { deprecated } : {}
   };
 }
-async function fetchAvailableModelsCatalog(token, projectId, signal) {
-  const json = await postAntigravityJson({
-    auth: token,
-    path: "v1internal:fetchAvailableModels",
-    body: { project: projectId },
-    signal
-  });
+async function fetchAvailableModelsCatalog(source, signal, client = defaultAntigravityClient) {
+  const json = await client.fetchAvailableModels(source, signal);
   return parseAvailableModels(json);
 }
 function toPersistedSnapshot(snapshot) {
@@ -509,13 +598,13 @@ function resolveModelPlan(publicModelId, effort, snapshot) {
       `Unknown model "${runtimeModelId}". Run /antigravity refresh, then pick a current model.`
     );
   }
-  const family = classifyModelFamily(runtimeModelId);
+  const profile = getModelProfile(runtimeModelId);
   return {
     runtimeModelId,
     modelEnum,
     thinkingConfig: resolveThinkingConfig(runtimeModelId, snapshot),
-    isNonGemini: family !== "gemini",
-    isClaude: family === "claude"
+    isNonGemini: profile.isNonGemini,
+    isClaude: profile.isClaude
   };
 }
 function estimateModelCost(_baseId) {
@@ -544,13 +633,9 @@ function synthesizeDynamicModel(baseId, items) {
     ...hasVariant(tierSpellings("medium")) ? {} : { medium: null },
     ...hasVariant(tierSpellings("high")) || items.some((it) => it.id === baseId) ? {} : { high: null }
   };
-  const family = classifyModelFamily(baseId);
-  const isFlash = baseId.includes("flash");
-  const isClaude = family === "claude";
-  const isGpt = family === "gpt";
-  const defaultContext = isFlash ? 1048576 : isClaude ? 25e4 : isGpt ? 128e3 : 1048576;
-  const defaultMaxOutput = isClaude ? 64e3 : isGpt ? 32768 : 65536;
-  const promptCache = isClaude ? { short: 300, long: 3600 } : { short: 300 };
+  const profile = getModelProfile(baseId);
+  const defaultContext = profile.defaultContextWindow;
+  const defaultMaxOutput = profile.defaultMaxOutputTokens;
   return {
     id: baseId,
     name: formatModelDisplayName(baseId, repItem?.displayName),
@@ -561,7 +646,7 @@ function synthesizeDynamicModel(baseId, items) {
     thinkingLevelMap,
     input: repItem?.supportsImages ? ["text", "image"] : ["text"],
     cost: estimateModelCost(baseId),
-    promptCache,
+    promptCache: profile.promptCache,
     contextWindow: repItem?.maxTokens || defaultContext,
     maxTokens: repItem?.maxOutputTokens || defaultMaxOutput
   };
@@ -639,7 +724,8 @@ function formatModelsList(catalog) {
   }
   return lines.join("\n");
 }
-function createModelCatalog() {
+function createModelCatalog(deps) {
+  const client = deps?.client ?? defaultAntigravityClient;
   let generation = { snapshot: EMPTY_SNAPSHOT(), version: 0 };
   const record = (catalogOrRaw) => {
     let catalog;
@@ -678,9 +764,7 @@ function createModelCatalog() {
     const storedModels = () => [...context.stored?.models ?? []];
     if (!context.allowNetwork) return storedModels();
     try {
-      const creds = await resolveCredentials(context);
-      if (!creds) return storedModels();
-      const catalog = await fetchAvailableModelsCatalog(creds, creds.projectId, context.signal);
+      const catalog = await fetchAvailableModelsCatalog(context, context.signal, client);
       record(catalog);
       const dynamicModels = buildDynamicPublicModels(catalog);
       if (context.publish) {
@@ -1030,8 +1114,8 @@ function isGeminiQuotaGroup(group) {
 }
 function selectQuotaGroup(groups, modelId) {
   if (groups.length === 0) return void 0;
-  const family = classifyModelFamily(modelId);
-  if (family === "claude" || family === "gpt") {
+  const profile = getModelProfile(modelId);
+  if (profile.quotaPoolKind === "3p") {
     const found = groups.find((g) => !isGeminiQuotaGroup(g));
     if (found) return found;
   }
@@ -1237,17 +1321,21 @@ var QuotaStatusCoordinator = class {
   lastPairs;
   store;
   fetchSummaryOverride;
+  client;
   nowFn;
   constructor(depsOrStoreOrFile) {
     if (typeof depsOrStoreOrFile === "string" || !depsOrStoreOrFile) {
       this.store = fileQuotaStatusStore(depsOrStoreOrFile || defaultConfigFile());
+      this.client = defaultAntigravityClient;
       this.nowFn = Date.now;
     } else if ("loadMode" in depsOrStoreOrFile) {
       this.store = depsOrStoreOrFile;
+      this.client = defaultAntigravityClient;
       this.nowFn = Date.now;
     } else {
       this.store = depsOrStoreOrFile.store ?? fileQuotaStatusStore(depsOrStoreOrFile.configFile || defaultConfigFile());
       this.fetchSummaryOverride = depsOrStoreOrFile.fetchQuotaSummary;
+      this.client = depsOrStoreOrFile.client ?? defaultAntigravityClient;
       this.nowFn = depsOrStoreOrFile.now ?? Date.now;
     }
     this.weeklyTo5hRatio = DEFAULT_WEEKLY_TO_5H_RATIO;
@@ -1260,6 +1348,26 @@ var QuotaStatusCoordinator = class {
   mode() {
     return this.store.loadMode();
   }
+  bind(pi) {
+    if (typeof pi?.on !== "function") return;
+    pi.on("session_start", async (_event, ctx) => {
+      await this.refreshAndPaint(ctx);
+    });
+    pi.on("model_select", async (event, ctx) => {
+      const modelCtx = {
+        ui: ctx.ui,
+        modelRegistry: ctx.modelRegistry,
+        model: event.model || ctx.model
+      };
+      await this.refreshAndPaint(modelCtx);
+    });
+    pi.on("agent_settled", async (_event, ctx) => {
+      await this.refreshAndPaint(ctx);
+    });
+  }
+  formatUsage(ctx) {
+    return this.inspectUsage(ctx);
+  }
   footerFor(modelId, mode = "smart") {
     return FOOTER_MODES[mode]?.render(this.summary, modelId, this.weeklyTo5hRatio, void 0, this.nowFn).plain;
   }
@@ -1328,18 +1436,13 @@ var QuotaStatusCoordinator = class {
   }
   async fetch(ctx, signal) {
     try {
-      const creds = await resolveCredentials(ctx);
-      if (!creds) return void 0;
       let rawSummary;
       if (this.fetchSummaryOverride) {
+        const creds = await resolveCredentials(ctx);
+        if (!creds) return void 0;
         rawSummary = await this.fetchSummaryOverride(creds.token, signal);
       } else {
-        rawSummary = await postAntigravityJson({
-          auth: creds,
-          path: "v1internal:retrieveUserQuotaSummary",
-          body: { project: creds.projectId },
-          signal
-        });
+        rawSummary = await this.client.retrieveQuotaSummary(ctx, signal);
       }
       if (!rawSummary) return void 0;
       const summary = rawSummary.groups ? parseQuotaSummary(rawSummary) : rawSummary;
@@ -1516,8 +1619,9 @@ function convertTools(tools) {
 function translateTurnTrace(conversation, runtimeModelId) {
   const contents = [];
   const messages = conversation || [];
-  const isGeminiRequest = classifyModelFamily(runtimeModelId) === "gemini";
-  const isGptRequest = classifyModelFamily(runtimeModelId) === "gpt";
+  const runtimeProfile = getModelProfile(runtimeModelId);
+  const isGeminiRequest = runtimeProfile.isGemini;
+  const isGptRequest = runtimeProfile.isGpt;
   for (const msg of messages) {
     if (msg.role === "user") {
       const parts = [];
@@ -1545,7 +1649,7 @@ function translateTurnTrace(conversation, runtimeModelId) {
         continue;
       }
       const isSameProvider = !msg.provider || msg.provider === PROVIDER_ID;
-      const isSameModel = isCompatibleFamily(msg.model, runtimeModelId);
+      const isSameModel = runtimeProfile.isReplayCompatible(msg.model);
       const isSameProviderAndModel = isSameProvider && isSameModel;
       const parts = [];
       let pendingThinkingSig;
@@ -1984,7 +2088,7 @@ async function consumeAntigravityStream(streamBody, output, stream, model, signa
     reader.releaseLock();
   }
 }
-function streamAntigravity(model, context, options, catalog) {
+function streamAntigravity(model, context, options, catalog, client = defaultAntigravityClient) {
   const stream = createAssistantMessageEventStream();
   (async () => {
     const output = {
@@ -2038,10 +2142,9 @@ function streamAntigravity(model, context, options, catalog) {
           if (typeof v === "string") mergedHeaders[k] = v;
         }
       }
-      const { response, stream: streamBody } = await postAntigravityStream({
-        auth: token,
+      const { response, stream: streamBody } = await client.streamGenerateContent({
+        source: options,
         endpoint,
-        path: "v1internal:streamGenerateContent?alt=sse",
         headers: mergedHeaders,
         body: requestBody,
         signal: options?.signal
@@ -2140,7 +2243,7 @@ ${sourceLines.join("\n")}`;
     completedAt
   };
 }
-async function performWebSearch(creds, options) {
+async function performWebSearch(source, options, client = defaultAntigravityClient) {
   const createdAt = (/* @__PURE__ */ new Date()).toISOString();
   let effectiveQuery = options.query.trim();
   if (options.domain && options.domain.trim()) {
@@ -2150,7 +2253,6 @@ async function performWebSearch(creds, options) {
     }
   }
   const reqBody = {
-    project: creds.projectId || FALLBACK_PROJECT_ID,
     request: {
       contents: [
         {
@@ -2181,12 +2283,12 @@ async function performWebSearch(creds, options) {
     userAgent: "antigravity",
     requestType: "web_search"
   };
-  const resp = await postAntigravityJson({
-    auth: creds.token,
-    path: "v1internal:generateContent",
-    body: reqBody,
-    signal: options.signal
-  });
+  const resp = await client.postJson(
+    source,
+    "v1internal:generateContent",
+    reqBody,
+    options.signal
+  );
   const completedAt = (/* @__PURE__ */ new Date()).toISOString();
   const candidate = resp.response?.candidates?.[0];
   const rawText = candidate?.content?.parts?.filter((p) => typeof p.text === "string").map((p) => p.text).join("") || "";
@@ -2220,15 +2322,8 @@ function registerWebSearchTool(pi) {
       )
     }),
     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, {
+        const result = await performWebSearch(ctx, {
           query: params.query,
           domain: params.domain,
           signal
@@ -2241,6 +2336,12 @@ function registerWebSearchTool(pi) {
           }
         };
       } catch (err) {
+        if (err instanceof AntigravityAuthError) {
+          return {
+            content: [{ type: "text", text: `Error: ${NOT_LOGGED_IN}` }],
+            details: { error: "not_logged_in" }
+          };
+        }
         return {
           content: [{ type: "text", text: `Search failed: ${err.message}` }],
           details: { error: err.message }
@@ -2250,11 +2351,6 @@ function registerWebSearchTool(pi) {
   });
 }
 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");
@@ -2262,12 +2358,16 @@ async function executeWebSearchCommand(ctx, query) {
   }
   try {
     if (ctx.hasUI) ctx.ui.notify(`Searching: "${trimmedQuery}"\u2026`, "info");
-    const result = await performWebSearch(creds, {
+    const result = await performWebSearch(ctx, {
       query: trimmedQuery,
       signal: ctx.signal
     });
     notifyUser(ctx, result.formattedOutput);
   } catch (err) {
+    if (err instanceof AntigravityAuthError) {
+      notifyUser(ctx, NOT_LOGGED_IN, "warning");
+      return;
+    }
     notifyUser(ctx, `Search failed: ${err.message}`, "error");
   }
 }
@@ -2386,7 +2486,7 @@ async function runUsageSubcommand(ctx, quotaStatus) {
   }
   try {
     if (ctx.hasUI) ctx.ui.notify("Fetching quota summary\u2026", "info");
-    const text = await quotaStatus.inspectUsage(ctx);
+    const text = await quotaStatus.formatUsage(ctx);
     emitOutput(ctx, text);
   } catch (err) {
     emitOutput(ctx, `Failed to fetch usage: ${err.message}`, "error");
@@ -2417,24 +2517,36 @@ function runLoginSubcommand(ctx) {
     emitOutput(ctx, "Please run /login antigravity to authenticate.");
   }
 }
-async function runAntigravitySubcommand(args, ctx, quotaStatus, catalog) {
-  const raw = (args || "").trim();
-  const parts = raw.split(/\s+/).filter(Boolean);
-  const sub = parseAntigravitySubcommand(parts[0] || "");
-  const subArgs = raw.slice(parts[0]?.length || 0).trim();
-  const cmd = SUBCOMMANDS.find((c) => c.name === sub);
-  if (cmd) {
-    await cmd.run({ ctx, quotaStatus, catalog, subArgs });
-    return;
-  }
-  emitOutput(ctx, USAGE_TEXT);
+function createAntigravityCommands(deps) {
+  return {
+    async handle(args, ctx) {
+      const raw = (args || "").trim();
+      const parts = raw.split(/\s+/).filter(Boolean);
+      const sub = parseAntigravitySubcommand(parts[0] || "");
+      const subArgs = raw.slice(parts[0]?.length || 0).trim();
+      const cmd = SUBCOMMANDS.find((c) => c.name === sub);
+      if (cmd) {
+        await cmd.run({
+          ctx,
+          quotaStatus: deps.quotaStatus,
+          catalog: deps.catalog,
+          subArgs
+        });
+        return;
+      }
+      emitOutput(ctx, USAGE_TEXT);
+    },
+    complete(prefix) {
+      return completeSubcommands(prefix);
+    }
+  };
 }
 
 // src/index.ts
-var PROVIDER_NAME = "Antigravity";
 function index_default(pi) {
   const quotaStatus = createQuotaStatus();
   const catalog = createModelCatalog();
+  const commands = createAntigravityCommands({ quotaStatus, catalog });
   pi.registerProvider(PROVIDER_ID, {
     name: PROVIDER_NAME,
     baseUrl: DEFAULT_ENDPOINT,
@@ -2453,26 +2565,13 @@ function index_default(pi) {
   });
   pi.registerCommand("antigravity", {
     description: "Antigravity quota, models and settings",
-    getArgumentCompletions: (prefix) => completeSubcommands(prefix),
+    getArgumentCompletions: (prefix) => commands.complete(prefix),
     handler: async (args, ctx) => {
-      await runAntigravitySubcommand(args, ctx, quotaStatus, catalog);
+      await commands.handle(args, ctx);
     }
   });
   registerWebSearchTool(pi);
-  pi.on("session_start", async (_event, ctx) => {
-    await quotaStatus.refreshAndPaint(ctx);
-  });
-  pi.on("model_select", async (event, ctx) => {
-    const modelCtx = {
-      ui: ctx.ui,
-      modelRegistry: ctx.modelRegistry,
-      model: event.model || ctx.model
-    };
-    await quotaStatus.refreshAndPaint(modelCtx);
-  });
-  pi.on("agent_settled", async (_event, ctx) => {
-    await quotaStatus.refreshAndPaint(ctx);
-  });
+  quotaStatus.bind(pi);
 }
 export {
   PROVIDER_ID,