pi-provider-antigravity 0.20.0 → 0.21.0

package/dist/index.js

@@ -6,7 +6,7 @@ var PROVIDER_ID = "antigravity";
 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.14 (aidev_client; os_type=linux; arch=amd64; cl=990662481; auth_method=consumer)";
+var DEFAULT_USER_AGENT = "antigravity/cli/1.2.16 (aidev_client; os_type=linux; arch=amd64; cl=992658124; auth_method=consumer)";
 var FALLBACK_PROJECT_ID = "aicode-consumers";
 
 // src/auth.ts
@@ -392,12 +392,16 @@ var PRIVATE_SNAPSHOT_KEY = "pi-provider-antigravity";
 var CANONICAL_TIER_SUFFIXES = Object.freeze({
   low: "-low",
   medium: "-medium",
-  high: "-high"
+  high: "-high",
+  xhigh: "-xhigh",
+  max: "-max"
 });
 var TIER_ALIASES = Object.freeze({
   low: Object.freeze(["-extra-low"]),
   medium: Object.freeze([]),
-  high: Object.freeze(["-thinking", "-agent"])
+  high: Object.freeze(["-thinking", "-agent"]),
+  xhigh: Object.freeze([]),
+  max: Object.freeze([])
 });
 var SPECIAL_TIER_SUFFIXES = Object.freeze(["-tiered"]);
 var ALL_TIER_SUFFIXES = Object.freeze([
@@ -425,7 +429,8 @@ var TIER_FALLBACKS = Object.freeze({
   // Gemini 3.1 Pro lists no -medium: up to high, never down
   high: Object.freeze(["-thinking", "-agent", ""]),
   // Claude's high tier is -thinking
-  xhigh: Object.freeze(["-high", "-thinking", "-agent", ""]),
+  xhigh: Object.freeze(["-max", "-high", "-thinking", "-agent", ""]),
+  // up to max, never down
   max: Object.freeze(["-high", "-thinking", "-agent", ""])
 });
 var DEFAULT_TIER_ORDER = Object.freeze([
@@ -583,9 +588,21 @@ function followRenames(runtimeModelId, snapshot) {
   }
   return current;
 }
+function thinkingLevelFor(runtimeModelId) {
+  for (const tier of Object.keys(CANONICAL_TIER_SUFFIXES)) {
+    if (tierSpellings(tier).some((suffix) => runtimeModelId.endsWith(suffix))) {
+      return tier.toUpperCase();
+    }
+  }
+  return void 0;
+}
 function resolveThinkingConfig(runtimeModelId, snapshot) {
   const budget = snapshot.thinking[runtimeModelId]?.budget;
-  return typeof budget === "number" ? { includeThoughts: true, thinkingBudget: budget } : { includeThoughts: false, thinkingBudget: 0 };
+  if (typeof budget === "number") {
+    return { includeThoughts: true, thinkingBudget: budget };
+  }
+  const thinkingLevel = thinkingLevelFor(runtimeModelId);
+  return thinkingLevel ? { includeThoughts: true, thinkingBudget: 0, thinkingLevel } : { includeThoughts: false, thinkingBudget: 0 };
 }
 function resolveModelPlan(publicModelId, effort, snapshot) {
   const runtimeModelId = followRenames(
@@ -631,7 +648,9 @@ function synthesizeDynamicModel(baseId, items) {
     minimal: null,
     ...hasVariant(tierSpellings("low")) ? {} : { low: null },
     ...hasVariant(tierSpellings("medium")) ? {} : { medium: null },
-    ...hasVariant(tierSpellings("high")) || items.some((it) => it.id === baseId) ? {} : { high: null }
+    ...hasVariant(tierSpellings("high")) || items.some((it) => it.id === baseId) ? {} : { high: null },
+    ...hasVariant(tierSpellings("xhigh")) ? { xhigh: "xhigh" } : {},
+    ...hasVariant(tierSpellings("max")) ? { max: "max" } : {}
   };
   const profile = getModelProfile(baseId);
   const defaultContext = profile.defaultContextWindow;
@@ -1534,13 +1553,19 @@ function deriveSessionId(conversation, explicitSessionId) {
   const hex = [...hash].map((b) => b.toString(16).padStart(2, "0")).join("");
   return `${hex.slice(0, 8)}-${hex.slice(8, 12)}-${hex.slice(12, 16)}-${hex.slice(16, 20)}-${hex.slice(20)}`;
 }
-function resolveSessionTrajectory(sessionId) {
-  const bytes = createHash2("sha1").update(`antigravity:${sessionId}`).digest().subarray(0, 16);
+function v5Uuid(seed) {
+  const bytes = createHash2("sha1").update(seed).digest().subarray(0, 16);
   bytes[6] = bytes[6] & 15 | 80;
   bytes[8] = bytes[8] & 63 | 128;
   const hex = [...bytes].map((b) => b.toString(16).padStart(2, "0")).join("");
   return `${hex.slice(0, 8)}-${hex.slice(8, 12)}-${hex.slice(12, 16)}-${hex.slice(16, 20)}-${hex.slice(20)}`;
 }
+function resolveSessionTrajectory(sessionId) {
+  return v5Uuid(`antigravity:${sessionId}`);
+}
+function resolveSessionCascade(sessionId) {
+  return v5Uuid(`antigravity:cascade:${sessionId}`);
+}
 var DEFAULT_SESSION_ID = "-3750763034362895579";
 function resolveNumericSessionId(sessionId) {
   if (/^-?\d+$/.test(sessionId)) {
@@ -1652,6 +1677,7 @@ function translateTurnTrace(conversation, runtimeModelId) {
       const isSameModel = runtimeProfile.isReplayCompatible(msg.model);
       const isSameProviderAndModel = isSameProvider && isSameModel;
       const parts = [];
+      const isToolTurn = Array.isArray(msg.content) && msg.content.some((item) => item.type === "tool_use" || item.type === "toolCall");
       let pendingThinkingSig;
       if (typeof msg.content === "string") {
         parts.push({ text: msg.content });
@@ -1677,15 +1703,13 @@ function translateTurnTrace(conversation, runtimeModelId) {
             const isForeignReasoning = !isSameProviderAndModel || typeof candidateSig === "string" && candidateSig.trim().startsWith("{");
             if (isGptRequest && !isForeignReasoning) continue;
             if (!isForeignReasoning) {
-              const part = {
-                thought: true,
-                text: item.thinking || ""
-              };
               const sig = resolveThoughtSignature(true, candidateSig);
               if (sig) {
                 pendingThinkingSig = sig;
               }
-              parts.push(part);
+              if (!isToolTurn) {
+                parts.push({ thought: true, text: item.thinking || "" });
+              }
             } else {
               let text = item.thinking || "";
               if (!text && typeof candidateSig === "string" && candidateSig.trim().startsWith("{")) {
@@ -1815,17 +1839,22 @@ function buildAntigravityRequestBody(params) {
   const tools = convertTools(getCurrentTools(transcript.messages));
   const conversation = withoutInitialSystemMessage(transcript.messages);
   const sessionId = deriveSessionId(conversation, params.sessionId);
+  const cascadeId = resolveSessionCascade(sessionId);
   const trajectoryId = params.trajectoryId && typeof params.trajectoryId === "string" && params.trajectoryId.trim().length > 0 ? params.trajectoryId.trim() : resolveSessionTrajectory(sessionId);
   const contents = translateTurnTrace(conversation, runtimeModelId);
   const requestIndex = contents.filter(
     (c) => c.role === "model" && !c.parts?.some((p) => p.functionResponse)
   ).length;
-  const requestId = `agent/${sessionId}/${Date.now()}/${trajectoryId}/${contents.length}`;
+  const requestId = `agent/${cascadeId}/${Date.now()}/${trajectoryId}/${contents.length}`;
   const modelEnum = plan.modelEnum;
   const labels = {
+    // agy names the conversation twice on 1.2.16 (cascade_id and its root) and the
+    // per-turn trajectory once; the root is the same cascade for a plain session.
+    cascade_id: cascadeId,
     last_step_index: String(Math.max(0, contents.length - 1)),
     model_enum: modelEnum,
     request_id: `${trajectoryId}-${requestIndex}`,
+    root_cascade_id: cascadeId,
     trajectory_id: trajectoryId,
     used_claude: String(plan.isClaude),
     used_claude_conservative: String(plan.isClaude),