pi-ui-extend 1.0.42 → 1.0.45

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/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