pi-ui-extend 1.0.41 → 1.0.45

package/external/pi-tools-suite/src/antigravity-auth/headers.ts

@@ -1,4 +1,44 @@
 import type { AntigravityModel, HeaderStyle } from "./types";
+import { AGY_CLI_CHANGE_LIST, AGY_CLI_VERSION } from "./constants";
+
+function normalizePlatform(platform = process.platform): string {
+	return platform === "win32" ? "windows" : platform || "unknown";
+}
+
+function normalizeArch(arch = process.arch): string {
+	switch (arch) {
+		case "x64":
+			return "amd64";
+		case "ia32":
+			return "386";
+		default:
+			return arch || "unknown";
+	}
+}
+
+/** Native agy CLI 1.1.24 content-request identity captured by cortexkit 2.2.1. */
+export function buildAntigravityHarnessUserAgent(
+	version = AGY_CLI_VERSION,
+	platform = process.platform,
+	arch = process.arch,
+	authMethod = "consumer",
+): string {
+	const changeList = version === AGY_CLI_VERSION ? `; cl=${AGY_CLI_CHANGE_LIST}` : "";
+	return `antigravity/cli/${version} (aidev_client; os_type=${normalizePlatform(platform)}; arch=${normalizeArch(arch)}${changeList}; auth_method=${authMethod})`;
+}
+
+export function getAntigravityBootstrapHeaders(accessToken: string): Record<string, string> {
+	return {
+		"User-Agent": buildAntigravityHarnessUserAgent(),
+		Authorization: `Bearer ${accessToken}`,
+		"Content-Type": "application/json",
+		"Accept-Encoding": "gzip",
+	};
+}
+
+export function getAntigravityLoadCodeAssistMetadata(): Record<string, string> {
+	return { ideType: "ANTIGRAVITY" };
+}
 
 export function getAntigravityHeaders(style: "antigravity" | "gemini-cli" = "antigravity"): Record<string, string> {
 	if (style === "gemini-cli") {
@@ -9,9 +49,7 @@ export function getAntigravityHeaders(style: "antigravity" | "gemini-cli" = "ant
 		};
 	}
 	return {
-		"User-Agent": "antigravity/1.18.3 darwin/arm64",
-		"X-Goog-Api-Client": "google-cloud-sdk vscode_cloudshelleditor/0.1",
-		"Client-Metadata": `{"ideType":"ANTIGRAVITY","platform":"${process.platform === "win32" ? "WINDOWS" : "MACOS"}","pluginType":"GEMINI"}`,
+		"User-Agent": buildAntigravityHarnessUserAgent(),
 	};
 }
 

package/external/pi-tools-suite/src/antigravity-auth/models.ts

@@ -1,4 +1,52 @@
+// Antigravity model catalog, synced with the upstream cortexkit registry
+// (packages/core/src/model-registry.ts + transform/model-resolver.ts at
+// commit fa48c66f2d931a9182d9c2fa38cdb1d9c4d61f10).
+//
+// Only models the Pi protocol can represent are advertised: the upstream
+// image-output model is intentionally NOT registered because Pi has no way to
+// express image-only output. Live route resolution for these public IDs lives
+// in payload.ts (resolveActualModel).
 export const modelDefinitions = [
+	{
+		id: "antigravity-gemini-3.8-flash",
+		name: "Gemini 3.8 Flash (Antigravity)",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 1_048_576,
+		maxTokens: 65_536,
+		thinkingLevelMap: { minimal: "low", low: "low", medium: "medium", high: "high", xhigh: "high" },
+	},
+	{
+		id: "antigravity-gemini-3.7-flash",
+		name: "Gemini 3.7 Flash (Antigravity)",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 1_048_576,
+		maxTokens: 65_536,
+		thinkingLevelMap: { minimal: "low", low: "low", medium: "medium", high: "high", xhigh: "high" },
+	},
+	{
+		id: "antigravity-gemini-3.6-flash",
+		name: "Gemini 3.6 Flash (Antigravity)",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 1_048_576,
+		maxTokens: 65_536,
+		thinkingLevelMap: { minimal: "low", low: "low", medium: "medium", high: "high", xhigh: "high" },
+	},
+	{
+		id: "antigravity-gemini-3.5-flash",
+		name: "Gemini 3.5 Flash (Antigravity)",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 1_048_576,
+		maxTokens: 65_536,
+		thinkingLevelMap: { minimal: "low", low: "low", medium: "medium", high: "high", xhigh: "high" },
+	},
 	{
 		id: "antigravity-gemini-3.1-pro",
 		name: "Gemini 3.1 Pro (Antigravity)",
@@ -7,11 +55,46 @@ export const modelDefinitions = [
 		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
 		contextWindow: 1_048_576,
 		maxTokens: 65_535,
-		thinkingLevelMap: { minimal: null, low: "low", medium: null, high: null, xhigh: null },
+		thinkingLevelMap: { minimal: null, low: "low", medium: null, high: "high", xhigh: "high" },
 	},
+	{
+		id: "antigravity-claude-sonnet-4-6-thinking",
+		name: "Claude Sonnet 4.6 Thinking (Antigravity)",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 200_000,
+		maxTokens: 64_000,
+		thinkingLevelMap: { minimal: null, low: "low", medium: "medium", high: "high", xhigh: "high" },
+	},
+	{
+		id: "antigravity-claude-opus-4-6-thinking",
+		name: "Claude Opus 4.6 Thinking (Antigravity)",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 200_000,
+		maxTokens: 64_000,
+		thinkingLevelMap: { minimal: null, low: "low", medium: "medium", high: "high", xhigh: "high" },
+	},
+	{
+		// GPT-OSS 120B with reasoning effort "medium" baked into the live route.
+		// Pi gets no thinkingConfig for it; the map pins every level to medium
+		// so the UI stays honest without changing routing.
+		id: "antigravity-gpt-oss-120b-medium",
+		name: "GPT-OSS 120B Medium (Antigravity)",
+		reasoning: true,
+		input: ["text"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 131_072,
+		maxTokens: 32_768,
+		thinkingLevelMap: { minimal: "medium", low: "medium", medium: "medium", high: "medium", xhigh: "medium" },
+	},
+	// Legacy Antigravity entries kept as aliases; both live routes still exist
+	// upstream (gemini-3-flash is also the target of the 3.5-flash high tier).
 	{
 		id: "antigravity-gemini-3-flash",
-		name: "Gemini 3 Flash (Antigravity)",
+		name: "Gemini 3 Flash (Antigravity, legacy)",
 		reasoning: true,
 		input: ["text", "image"],
 		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
@@ -19,6 +102,15 @@ export const modelDefinitions = [
 		maxTokens: 65_536,
 		thinkingLevelMap: { minimal: "minimal", low: "low", medium: "medium", high: "high", xhigh: null },
 	},
+	{
+		id: "antigravity-claude-sonnet-4-6",
+		name: "Claude Sonnet 4.6 (Antigravity, legacy)",
+		reasoning: false,
+		input: ["text", "image"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 200_000,
+		maxTokens: 64_000,
+	},
 	// These Gemini CLI-named models mirror opencode's provider surface, but opencode
 	// routes them Antigravity-first unless cli_first is enabled.
 	{
@@ -64,23 +156,4 @@ export const modelDefinitions = [
 		thinkingLevelMap: { minimal: null, low: "low", medium: null, high: null, xhigh: null },
 		antigravityHeaderStyle: "antigravity",
 	},
-	{
-		id: "antigravity-claude-sonnet-4-6",
-		name: "Claude Sonnet 4.6 (Antigravity)",
-		reasoning: false,
-		input: ["text", "image"],
-		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 200_000,
-		maxTokens: 64_000,
-	},
-	{
-		id: "antigravity-claude-opus-4-6-thinking",
-		name: "Claude Opus 4.6 Thinking (Antigravity)",
-		reasoning: true,
-		input: ["text", "image"],
-		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 200_000,
-		maxTokens: 64_000,
-		thinkingLevelMap: { minimal: null, low: "low", medium: "medium", high: "high", xhigh: "high" },
-	},
 ];

package/external/pi-tools-suite/src/antigravity-auth/oauth.ts

@@ -2,7 +2,7 @@ import { createHash, randomBytes } from "node:crypto";
 import type { OAuthCredentials } from "@earendil-works/pi-ai";
 import { DEFAULT_PROJECT_ID, LOAD_ENDPOINTS, REDIRECT_URI, SCOPES, TOKEN_EXPIRY_SKEW_MS, PROVIDER_ID } from "./constants";
 import { accountFromCredential, clampAccountIndex, encodeApiKey, findMatchingAccountIndex, getAccountProjectId, getAccountRefreshToken, getGoogleOAuthClientCredentials, getPiAuthPath, getStoredAccounts, joinRefresh, readJsonFile, splitRefresh, writeJsonFileSecure } from "./auth-store";
-import { getAntigravityHeaders } from "./headers";
+import { getAntigravityBootstrapHeaders, getAntigravityHeaders, getAntigravityLoadCodeAssistMetadata } from "./headers";
 import { notifyAntigravityLoginFailure } from "./status";
 import type { AntigravityAddAccountResult, AntigravityFailoverCredential, AntigravityLoginCallbacks, GoogleOAuthClientCredentials, OpencodeAntigravityAccount, PiAuthCredential, PiAuthData, RefreshedAntigravityAccount } from "./types";
 
@@ -33,12 +33,7 @@ function assertGoogleOAuthCredentialsConfigured(credentials?: GoogleOAuthClientC
 }
 
 async function fetchProjectId(accessToken: string): Promise<string | undefined> {
-	const headers = {
-		Authorization: `Bearer ${accessToken}`,
-		"Content-Type": "application/json",
-		...getAntigravityHeaders("gemini-cli"),
-		"Client-Metadata": getAntigravityHeaders()["Client-Metadata"],
-	};
+	const headers = getAntigravityBootstrapHeaders(accessToken);
 
 	for (const endpoint of LOAD_ENDPOINTS) {
 		try {
@@ -46,11 +41,7 @@ async function fetchProjectId(accessToken: string): Promise<string | undefined>
 				method: "POST",
 				headers,
 				body: JSON.stringify({
-					metadata: {
-						ideType: "ANTIGRAVITY",
-						platform: process.platform === "win32" ? "WINDOWS" : "MACOS",
-						pluginType: "GEMINI",
-					},
+					metadata: getAntigravityLoadCodeAssistMetadata(),
 				}),
 			});
 			if (!response.ok) continue;
@@ -130,6 +121,7 @@ export async function loginAntigravity(callbacks: AntigravityLoginCallbacks, opt
 			headers: {
 				"Content-Type": "application/x-www-form-urlencoded;charset=UTF-8",
 				Accept: "*/*",
+				"Accept-Encoding": "gzip, deflate, br",
 				"User-Agent": getAntigravityHeaders("gemini-cli")["User-Agent"],
 			},
 			body: new URLSearchParams({
@@ -235,11 +227,7 @@ async function refreshAccountToken(
 	const response = await fetch("https://oauth2.googleapis.com/token", {
 		method: "POST",
 		signal,
-		headers: {
-			"Content-Type": "application/x-www-form-urlencoded;charset=UTF-8",
-			Accept: "*/*",
-			"User-Agent": getAntigravityHeaders("gemini-cli")["User-Agent"],
-		},
+		headers: { "Content-Type": "application/x-www-form-urlencoded" },
 		body: new URLSearchParams({
 			client_id: clientCredentials.clientId,
 			...(clientCredentials.clientSecret ? { client_secret: clientCredentials.clientSecret } : {}),

package/external/pi-tools-suite/src/antigravity-auth/payload.ts

@@ -22,7 +22,10 @@ function supportsMultimodalFunctionResponse(modelId: string): boolean {
 }
 
 function isGemini3Model(modelId: string): boolean {
-	return /^gemini(?:-live)?-3(?:[.-]|$)/i.test(modelId);
+	// "gemini-pro-agent" is the Antigravity agent-mode route for Gemini 3.1 Pro;
+	// it needs the same thought-signature sentinel as the versioned gemini-3
+	// routes.
+	return /^gemini(?:-live)?-3(?:[.-]|$)/i.test(modelId) || /^gemini-pro-agent$/i.test(modelId);
 }
 
 function isValidThoughtSignature(signature: unknown): signature is string {
@@ -333,13 +336,51 @@ function geminiProThinkingLevel(modelId: string, headerStyle: HeaderStyle, level
 	return "low";
 }
 
+type FlashTier = "low" | "medium" | "high";
+
+/**
+ * Live Antigravity routes for the versioned flash generations, keyed by the
+ * public (prefix-stripped) model id. Mirrors the upstream cortexkit
+ * model-resolver: each thinking tier maps to a distinct live route plus a
+ * numeric thinking budget (-1 = dynamic for 3.7/3.8). 3.5-flash has no
+ * per-tier routes of its own and rides the older gemini-3 routes instead.
+ */
+const ANTIGRAVITY_FLASH_ROUTES: Record<string, {
+	routes: Record<FlashTier, string>;
+	budgets: Record<FlashTier, number>;
+}> = {
+	"gemini-3.8-flash": {
+		routes: { low: "gemini-3.8-flash-low", medium: "gemini-3.8-flash-medium", high: "gemini-3.8-flash-high" },
+		budgets: { low: -1, medium: -1, high: -1 },
+	},
+	"gemini-3.7-flash": {
+		routes: { low: "gemini-3.7-flash-low", medium: "gemini-3.7-flash-medium", high: "gemini-3.7-flash-high" },
+		budgets: { low: -1, medium: -1, high: -1 },
+	},
+	"gemini-3.6-flash": {
+		routes: { low: "gemini-3.6-flash-low", medium: "gemini-3.6-flash-medium", high: "gemini-3.6-flash-high" },
+		budgets: { low: 1_000, medium: 4_000, high: 10_000 },
+	},
+	"gemini-3.5-flash": {
+		routes: { low: "gemini-3.5-flash-extra-low", medium: "gemini-3.5-flash-low", high: "gemini-3-flash-agent" },
+		budgets: { low: 1_000, medium: 4_000, high: 10_000 },
+	},
+};
+
+function flashTier(level: "minimal" | "low" | "medium" | "high" | undefined): FlashTier {
+	if (level === "high") return "high";
+	if (level === undefined || level === "medium") return "medium";
+	return "low";
+}
+
 function resolveActualModel(
 	model: AntigravityModel,
 	options?: SimpleStreamOptions,
 ): { actualModel: string; thinkingConfig?: Record<string, unknown> } {
 	const headerStyle = getModelHeaderStyle(model);
 	const requested = model.id.replace(/^antigravity-/i, "");
-	const level = reasoningLevel(options) ?? "low";
+	const requestedLevel = reasoningLevel(options);
+	const level = requestedLevel ?? "low";
 	let effective = requested;
 	const requestedLower = requested.toLowerCase();
 
@@ -354,6 +395,27 @@ function resolveActualModel(
 
 	const lower = effective.toLowerCase();
 
+	if (headerStyle === "antigravity") {
+		// Current Antigravity catalog: versioned flash generations and 3.1 Pro
+		// route to per-tier live models with numeric thinking budgets. Requests
+		// without an explicit level default to the medium tier for flash.
+		const flash = ANTIGRAVITY_FLASH_ROUTES[lower];
+		if (flash) {
+			const tier = flashTier(requestedLevel);
+			return {
+				actualModel: flash.routes[tier],
+				thinkingConfig: { thinkingBudget: flash.budgets[tier], includeThoughts: true },
+			};
+		}
+		if (/^gemini-3\.1-pro$/.test(lower)) {
+			const high = level === "high";
+			return {
+				actualModel: high ? "gemini-pro-agent" : "gemini-3.1-pro-low",
+				thinkingConfig: { thinkingBudget: high ? 10_001 : 1_001, includeThoughts: true },
+			};
+		}
+	}
+
 	if (/^gemini-3(?:\.\d+)?-pro/.test(lower)) {
 		// Live Antigravity currently rejects gemini-3.1-pro-high with a generic
 		// 400 INVALID_ARGUMENT, while gemini-3.1-pro-low works. Keep the public
@@ -370,7 +432,11 @@ function resolveActualModel(
 	if (lower.includes("claude") && lower.includes("thinking")) {
 		const budgets = options?.thinkingBudgets ?? {};
 		const budget = budgetForLevel(level, budgets);
-		return { actualModel: effective, thinkingConfig: { thinking_budget: budget, include_thoughts: true } };
+		// Sonnet 4.6 thinking is served by the base claude-sonnet-4-6 route with
+		// thinking enabled via the budget, while Opus keeps a dedicated
+		// -thinking route.
+		const actualModel = /^claude-sonnet/.test(lower) ? effective.replace(/-thinking$/i, "") : effective;
+		return { actualModel, thinkingConfig: { thinking_budget: budget, include_thoughts: true } };
 	}
 	if (lower.includes("gemini-2.5")) {
 		const budgets = options?.thinkingBudgets ?? {};
@@ -415,9 +481,8 @@ export function buildPayload(model: AntigravityModel, context: Context, options?
 }
 
 export function extraHeadersForPayload(payload: Record<string, unknown>): Record<string, string> {
-	const model = typeof payload.model === "string" ? payload.model.toLowerCase() : "";
-	if (model.includes("claude") && model.includes("thinking")) {
-		return { "anthropic-beta": "interleaved-thinking-2025-05-14" };
-	}
+	void payload;
+	// Native agy CLI content requests do not send anthropic-beta or the old
+	// Cloud SDK metadata headers. Thinking is encoded in generationConfig only.
 	return {};
 }

package/external/pi-tools-suite/src/antigravity-auth/stream.ts

@@ -71,7 +71,7 @@ async function sendAntigravityRequest(
 		const headers = new Headers({
 			Authorization: `Bearer ${apiKey}`,
 			"Content-Type": "application/json",
-			Accept: "text/event-stream",
+			...(headerStyle === "antigravity" ? { "Accept-Encoding": "gzip" } : { Accept: "text/event-stream" }),
 			...getAntigravityHeaders(headerStyle),
 			...requestHeaders,
 		});
@@ -79,6 +79,18 @@ async function sendAntigravityRequest(
 			if (value === null) headers.delete(name);
 			else headers.set(name, value);
 		}
+		if (headerStyle === "antigravity") {
+			// Match native agy content traffic and prevent host/provider headers from
+			// leaking into the Cloud Code proxy. x-goog-user-project in particular is
+			// known to turn otherwise valid requests into 403 responses.
+			headers.set("User-Agent", getAntigravityHeaders("antigravity")["User-Agent"]);
+			headers.set("Accept-Encoding", "gzip");
+			headers.delete("X-Goog-Api-Client");
+			headers.delete("Client-Metadata");
+			headers.delete("X-Goog-User-Project");
+			headers.delete("anthropic-beta");
+			headers.delete("Accept");
+		}
 		response = await fetch(`${endpoint}/v1internal:streamGenerateContent?alt=sse`, {
 			method: "POST",
 			signal: options?.signal,

package/external/pi-tools-suite/src/async-subagents/agents/frontier-review.md

@@ -0,0 +1,23 @@
+---
+description: Independent frontier-model code review gate for substantive code changes made by a non-frontier parent. Use after implementation and targeted checks, before finalizing. Read-only review of the actual diff and surrounding code; use research for ordinary evidence gathering and oracle for architecture or high-stakes second opinions.
+icon: eye
+models: [openai-codex/gpt-5.6-sol, zai/glm-5.3]
+notForParentModels: [openai-codex/gpt-5.6-sol*, zai/glm-5.3]
+thinking: high
+tools: [read, grep, bash]
+---
+
+# Frontier code review
+
+Review the implementation independently. Inspect the actual diff and the
+surrounding contracts; do not accept the implementer's summary as evidence.
+
+Look for correctness bugs, regressions, broken invariants, unsafe edge cases,
+missing error handling, concurrency/state mistakes, security issues, and tests
+that fail to cover a material behavior change. Prefer a small number of
+actionable findings over stylistic commentary or speculative rewrites.
+
+Do not edit source or tests. Run narrow read-only checks when they materially
+increase confidence. For each finding, give severity, rationale, and precise
+file/line evidence. If there are no findings, say so explicitly and name any
+important residual risk or verification gap.

package/external/pi-tools-suite/src/async-subagents/agents/implement.md

@@ -13,7 +13,7 @@ not broaden scope or make product/architecture decisions for the parent.
 
 For UI work, preserve the existing design language and inspect supplied visual
 references with an image-capable model. Report unavailable capabilities instead
-of claiming visual verification. Actual browser QA belongs to browser-qa.
+of claiming visual verification. Actual user-interface QA belongs to ui-qa.
 
 Run relevant targeted checks and report changed paths plus their results. If
 requirements conflict or the task exceeds your capabilities, stop with a

package/external/pi-tools-suite/src/async-subagents/agents/presets.jsonc

@@ -0,0 +1,16 @@
+{
+  // Presets are available-model pools. Agent candidate order still comes from
+  // each agent's Markdown frontmatter; pool order never reorders candidates.
+  "cheap": {
+    "description": "GLM pool: economical workers plus GLM-5.3 for frontier-review/oracle.",
+    "models": ["zai/glm-5-turbo", "zai/glm-5.3-flash", "zai/glm-5.3"]
+  },
+  "gpt": {
+    "description": "GPT pool: Luna/Terra workers; Sol is available to frontier-review/oracle.",
+    "models": ["openai-codex/gpt-5.6-luna", "openai-codex/gpt-5.6-terra", "openai-codex/gpt-5.6-sol"]
+  },
+  "deep": {
+    "description": "Mixed pool; workers retain economical candidate order and frontier-review/oracle use strong models.",
+    "models": ["zai/glm-5-turbo", "zai/glm-5.3-flash", "openai-codex/gpt-5.6-luna", "openai-codex/gpt-5.6-terra", "openai-codex/gpt-5.6-sol", "zai/glm-5.3"]
+  }
+}

package/external/pi-tools-suite/src/async-subagents/agents/research.md

@@ -1,5 +1,5 @@
 ---
-description: Read-only evidence gathering - search files, trace behavior, investigate hypotheses, or independently review a diff. Return findings with paths, not raw source. Use verify for running checks and implement for changes.
+description: Read-only evidence gathering - search files, trace behavior, investigate hypotheses, or analyze a focused review question. Return findings with paths, not raw source. Use frontier-review for the post-implementation code-review gate when available, verify for running checks, and implement for changes.
 icon: search
 models: [zai/glm-5-turbo, openai-codex/gpt-5.6-luna]
 thinking: low
@@ -9,8 +9,10 @@ tools: [read, grep]
 # Research
 
 Investigate the assigned question within scope. Read and compare evidence; do
-not edit files or invent missing facts. A review is a fresh investigation of
-the supplied diff, not approval based on the implementer's summary.
+not edit files or invent missing facts. When assigned a review question, make
+it a fresh investigation of the relevant code, not approval based on another
+agent's summary. The broad post-implementation review gate belongs to
+frontier-review when that role is available.
 
 Return the answer and supporting file:line references, distinguishing confirmed
 findings from hypotheses. Report gaps or a concrete blocker when the available