dsh-antigravity-auth 0.1.2 → 0.1.3

package/CHANGELOG.md

@@ -2,6 +2,12 @@
 
 ## [Unreleased]
 
+## [0.1.3] - 2026-08-31
+
+- Restored reliable Claude Opus tool selection by preserving the complete DSH system prompt and applying the audited Claude tool instruction, strict parameter descriptions, and validated function-calling mode.
+- Fixed Claude continuation after parallel tool execution by preserving call/response IDs, applying the safe thought-signature sentinel, dropping unsigned reasoning replay, and grouping correlated function responses.
+- Mapped exact bounded Antigravity context-window overflow responses into DSH compaction recovery while retaining fail-closed parsing, timeout, response-size, frame-size, depth, and redaction limits.
+
 ## [0.1.2] - 2026-08-27
 
 - Kept the Antigravity provider visible in DSH's stock model catalog during explicitly allowlisted transient live-discovery failures by falling back to the audited pinned text snapshot.

package/lib/llm-adapter.js

@@ -2,9 +2,9 @@ import { a as DEFAULT_PRIVATE_RESPONSE_HEADER_TIMEOUT_MS, d as privateStatusErro
 import { ANTIGRAVITY_WIRE_ORIGIN } from "./wire-identity.js";
 import { t as classifyPrivateFailure } from "./private-failure-B9vzUkQ7.js";
 import { antigravityModelFamily, buildFunctionDeclarations, compatibleReplayState, createReplayState } from "./replay.js";
-import { AgyRequestSessionStore, applyClaudeTransforms, applyGeminiTransforms, buildAgyAgentRequestMetadata, fnv1a64Signed, getPublicModelDefinitions, getResolverAliasMap, resolveModelWithTier } from "@cortexkit/antigravity-auth-core";
+import { AgyRequestSessionStore, CLAUDE_DESCRIPTION_PROMPT, CLAUDE_TOOL_SYSTEM_INSTRUCTION, SKIP_THOUGHT_SIGNATURE, applyClaudeTransforms, applyGeminiTransforms, buildAgyAgentRequestMetadata, fnv1a64Signed, getPublicModelDefinitions, getResolverAliasMap, resolveModelWithTier } from "@cortexkit/antigravity-auth-core";
 import { Buffer } from "node:buffer";
-import { CallId, LlmAdapter, LlmError, ReasoningEffortId, resolveRetryPolicy } from "@deepseek-ai/dsh-llm";
+import { CONTEXT_WINDOW_EXCEEDED_CODE, CallId, LlmAdapter, LlmError, ReasoningEffortId, resolveRetryPolicy } from "@deepseek-ai/dsh-llm";
 //#region src/llm-adapter.ts
 /** Public DSH LLM adapter for the single Antigravity provider route. */
 const ANTIGRAVITY_PROVIDER = "google-antigravity";
@@ -14,6 +14,9 @@ const ANTIGRAVITY_AVAILABLE_MODELS_ENDPOINT = `${ANTIGRAVITY_WIRE_ORIGIN}/v1inte
 const ANTIGRAVITY_LLM_ROUTE = ANTIGRAVITY_PROVIDER;
 const MODEL_CATALOG_TTL_MS = 3e4;
 const MAX_MODEL_CATALOG_BYTES = 262144;
+const MAX_PROVIDER_ERROR_BYTES = 65536;
+const MAX_PROVIDER_ERROR_FRAME_BYTES = 16384;
+const MAX_PROVIDER_ERROR_JSON_DEPTH = 8;
 const MAX_PROVIDER_PARTS = 4096;
 /** Adapter that owns exactly one provider route and no fallback route. */
 var AntigravityAdapter = class extends LlmAdapter {
@@ -157,6 +160,14 @@ var AntigravityAdapter = class extends LlmAdapter {
 					replayed = true;
 					continue;
 				}
+				if (response.status === 400 && await responseReportsContextWindowExceeded(response, {
+					...signal === void 0 ? {} : { signal },
+					idleTimeoutMs: this.options.idleTimeoutMs,
+					totalTimeoutMs: this.options.totalTimeoutMs,
+					maxResponseBytes: this.options.maxResponseBytes,
+					maxFrameBytes: this.options.maxFrameBytes
+				})) throw contextWindowExceededError(response.status);
+				await cancelResponse(response);
 				throw toLlmError(statusError);
 			}
 			try {
@@ -242,7 +253,7 @@ var AntigravityAdapter = class extends LlmAdapter {
 			usage
 		};
 		if (eventError !== void 0) {
-			yield finishChunk("error", safeProviderErrorCode(eventError.code), safeProviderStatus(eventError.status));
+			yield finishChunk("error", eventError.contextWindowExceeded === true ? CONTEXT_WINDOW_EXCEEDED_CODE : safeProviderErrorCode(eventError.code), safeProviderStatus(eventError.status));
 			return;
 		}
 		if (isAborted(options.signal)) {
@@ -549,10 +560,35 @@ async function buildAntigravityGeneratePayloadForAdapter(options, credential, se
 		requestType: "agent"
 	};
 }
+function groupClaudeFunctionResponses(contents, model) {
+	if (antigravityModelFamily(model) !== "claude") return contents;
+	const grouped = [];
+	let pendingResponses = [];
+	const flushResponses = () => {
+		if (pendingResponses.length === 0) return;
+		grouped.push({
+			role: "user",
+			parts: pendingResponses
+		});
+		pendingResponses = [];
+	};
+	for (const content of contents) {
+		const rawParts = Array.isArray(content.parts) ? content.parts : [];
+		const responseParts = rawParts.filter((part) => isRecord(part) && isRecord(part.functionResponse));
+		if (content.role === "user" && rawParts.length > 0 && responseParts.length === rawParts.length) {
+			pendingResponses.push(...responseParts);
+			continue;
+		}
+		flushResponses();
+		grouped.push(content);
+	}
+	flushResponses();
+	return grouped;
+}
 function buildPayloadFromContents(options, credential, contents) {
 	const wireModel = resolveWireModel(options.model);
-	const request = { contents };
-	if (options.system !== void 0 && options.system.trim().length > 0) request.systemInstruction = { parts: [{ text: options.system.slice(0, 65536) }] };
+	const request = { contents: groupClaudeFunctionResponses(contents, options.model) };
+	if (options.system !== void 0 && options.system.trim().length > 0) request.systemInstruction = { parts: [{ text: options.system }] };
 	const generationConfig = {};
 	if (options.maxTokens !== void 0) generationConfig.maxOutputTokens = boundedInteger(options.maxTokens, 1, 1e6, "maxTokens");
 	if (options.temperature !== void 0) generationConfig.temperature = boundedNumber(options.temperature, -100, 100, "temperature");
@@ -565,19 +601,21 @@ function buildPayloadFromContents(options, credential, contents) {
 	}
 	const resolved = resolveModelWithTier(options.model, { cli_first: false });
 	const thinkingLevel = normalizeReasoningEffort(options.reasoningEffort) ?? resolved.thinkingLevel;
-	if (wireModel.toLowerCase().includes("claude")) applyClaudeTransforms(request, {
-		model: wireModel,
-		...resolved.thinkingBudget === void 0 ? {} : { tierThinkingBudget: resolved.thinkingBudget },
-		...options.reasoningEffort === void 0 && resolved.thinkingBudget === void 0 ? {} : { normalizedThinking: {
-			includeThoughts: true,
-			...resolved.thinkingBudget === void 0 ? {} : { thinkingBudget: resolved.thinkingBudget }
-		} },
-		cleanJSONSchema: (value) => isRecord(value) ? value : {
-			type: "object",
-			properties: {}
-		}
-	});
-	else applyGeminiTransforms(request, {
+	if (wireModel.toLowerCase().includes("claude")) {
+		applyClaudeToolHardening(request);
+		applyClaudeTransforms(request, {
+			model: wireModel,
+			...resolved.thinkingBudget === void 0 ? {} : { tierThinkingBudget: resolved.thinkingBudget },
+			...options.reasoningEffort === void 0 && resolved.thinkingBudget === void 0 ? {} : { normalizedThinking: {
+				includeThoughts: true,
+				...resolved.thinkingBudget === void 0 ? {} : { thinkingBudget: resolved.thinkingBudget }
+			} },
+			cleanJSONSchema: (value) => isRecord(value) ? value : {
+				type: "object",
+				properties: {}
+			}
+		});
+	} else applyGeminiTransforms(request, {
 		model: wireModel,
 		...thinkingLevel === void 0 ? {} : { tierThinkingLevel: thinkingLevel },
 		...resolved.thinkingBudget === void 0 ? {} : { tierThinkingBudget: resolved.thinkingBudget },
@@ -593,37 +631,115 @@ function buildPayloadFromContents(options, credential, contents) {
 		request
 	};
 }
+function applyClaudeToolHardening(request) {
+	if (!Array.isArray(request.tools) || request.tools.length === 0) return;
+	request.tools = request.tools.map((tool) => {
+		if (!isRecord(tool) || !Array.isArray(tool.functionDeclarations)) return tool;
+		return {
+			...tool,
+			functionDeclarations: tool.functionDeclarations.map((declaration) => hardenClaudeToolDeclaration(declaration))
+		};
+	});
+	const instructionPart = { text: CLAUDE_TOOL_SYSTEM_INSTRUCTION };
+	const existing = request.systemInstruction;
+	if (isRecord(existing) && Array.isArray(existing.parts)) {
+		if (existing.parts.some((part) => isRecord(part) && typeof part.text === "string" && part.text.includes("CRITICAL TOOL USAGE INSTRUCTIONS"))) return;
+		request.systemInstruction = {
+			...existing,
+			parts: [...existing.parts, instructionPart]
+		};
+	} else if (typeof existing === "string") request.systemInstruction = {
+		role: "user",
+		parts: [{ text: existing }, instructionPart]
+	};
+	else request.systemInstruction = {
+		role: "user",
+		parts: [instructionPart]
+	};
+}
+function hardenClaudeToolDeclaration(value) {
+	if (!isRecord(value)) return value;
+	const description = typeof value.description === "string" ? value.description : "";
+	if (description.includes("STRICT PARAMETERS:")) return value;
+	const schema = isRecord(value.parameters) ? value.parameters : void 0;
+	const properties = schema !== void 0 && isRecord(schema.properties) ? schema.properties : void 0;
+	if (properties === void 0 || Object.keys(properties).length === 0) return value;
+	const required = new Set(Array.isArray(schema?.required) ? schema.required.filter((item) => typeof item === "string") : []);
+	const parameters = Object.entries(properties).map(([name, property]) => {
+		const requiredHint = required.has(name) ? ", REQUIRED" : "";
+		return `${name} (${claudeToolTypeHint(property)}${requiredHint})`;
+	});
+	return {
+		...value,
+		description: description + CLAUDE_DESCRIPTION_PROMPT.replace("{params}", parameters.join(", "))
+	};
+}
+function claudeToolTypeHint(value) {
+	if (!isRecord(value)) return "unknown";
+	if (Array.isArray(value.enum)) return value.enum.length <= 5 ? `string ENUM[${value.enum.map((item) => JSON.stringify(item)).join(", ")}]` : `string ENUM[${value.enum.length} options]`;
+	const type = typeof value.type === "string" ? value.type : "unknown";
+	if (type === "array") {
+		if (!isRecord(value.items)) return "ARRAY";
+		const itemType = typeof value.items.type === "string" ? value.items.type : "unknown";
+		if (itemType !== "object") return `ARRAY_OF_${itemType.toUpperCase()}`;
+		if (!isRecord(value.items.properties)) return "ARRAY_OF_OBJECTS";
+		const nestedRequired = new Set(Array.isArray(value.items.required) ? value.items.required.filter((item) => typeof item === "string") : []);
+		return `ARRAY_OF_OBJECTS[${Object.entries(value.items.properties).map(([name, property]) => {
+			return `${name}: ${isRecord(property) && typeof property.type === "string" ? property.type : "unknown"}${nestedRequired.has(name) ? " REQUIRED" : ""}`;
+		}).join(", ")}]`;
+	}
+	if (type === "object" && isRecord(value.properties)) {
+		const nestedRequired = new Set(Array.isArray(value.required) ? value.required.filter((item) => typeof item === "string") : []);
+		return `object{${Object.entries(value.properties).map(([name, property]) => {
+			return `${name}: ${isRecord(property) && typeof property.type === "string" ? property.type : "unknown"}${nestedRequired.has(name) ? " REQUIRED" : ""}`;
+		}).join(", ")}}`;
+	}
+	return type;
+}
 function mapMessage(message, model, toolNames) {
 	const parts = [];
 	const replayBlocks = compatibleReplayState(message, "google-antigravity", model, contentKinds(message))?.blocks ?? [];
+	const isClaude = antigravityModelFamily(model) === "claude";
+	let replayIndex = 0;
+	let sawClaudeFunctionCall = false;
 	for (const block of message.content) {
-		const replayBlock = replayBlocks.find((r) => r.kind === block.type);
-		const blockSignature = block.signature ?? block.thoughtSignature ?? replayBlock?.signature;
+		const replayKind = block.type === "text" || block.type === "reasoning" || block.type === "tool-call" ? block.type : void 0;
+		const replayBlock = replayKind === void 0 ? void 0 : replayBlocks[replayIndex++];
+		const replaySignature = replayBlock !== void 0 && replayBlock.kind === replayKind ? replayBlock.signature : void 0;
+		const blockSignature = block.signature ?? block.thoughtSignature ?? replaySignature;
 		if (block.type === "text") {
 			if (block.text.length === 0 && message.content.length > 1) continue;
 			parts.push({
 				text: block.text,
 				...blockSignature === void 0 ? {} : { thoughtSignature: blockSignature }
 			});
-		} else if (block.type === "reasoning") parts.push({
-			text: block.text,
-			thought: true,
-			...blockSignature === void 0 ? {} : { thoughtSignature: blockSignature }
-		});
-		else if (block.type === "tool-call") {
-			rememberToolName(toolNames, block.id, block.name);
+		} else if (block.type === "reasoning") {
+			if (isClaude && blockSignature === void 0) continue;
+			parts.push({
+				text: block.text,
+				thought: true,
+				...blockSignature === void 0 ? {} : { thoughtSignature: blockSignature }
+			});
+		} else if (block.type === "tool-call") {
+			const callId = rememberToolName(toolNames, block.id, block.name);
+			const signature = isClaude ? sawClaudeFunctionCall ? void 0 : blockSignature ?? SKIP_THOUGHT_SIGNATURE : blockSignature;
+			sawClaudeFunctionCall ||= isClaude;
 			parts.push({
 				functionCall: {
+					...isClaude ? { id: callId } : {},
 					name: block.name,
 					args: parseJsonObject(block.arguments)
 				},
-				...blockSignature === void 0 ? {} : { thoughtSignature: blockSignature }
+				...signature === void 0 ? {} : { thoughtSignature: signature }
 			});
-		} else if (block.type === "tool-result") parts.push({ functionResponse: {
-			name: requireToolName(toolNames, block.toolCallId),
-			response: { content: blocksToText(block.content) }
-		} });
-		else if (block.type === "image") throw new LlmError("Antigravity text requests do not accept unresolved image blocks", "UNSUPPORTED_MODALITY");
+		} else if (block.type === "tool-result") {
+			const callId = requireToolCallId(block.toolCallId);
+			parts.push({ functionResponse: {
+				...isClaude ? { id: callId } : {},
+				name: requireToolName(toolNames, callId),
+				response: { content: blocksToText(block.content) }
+			} });
+		} else if (block.type === "image") throw new LlmError("Antigravity text requests do not accept unresolved image blocks", "UNSUPPORTED_MODALITY");
 	}
 	return {
 		role: message.role === "assistant" ? "model" : "user",
@@ -635,9 +751,14 @@ async function mapMessageWithAttachments(message, model, toolNames, attachments,
 	if (attachments === void 0) throw new LlmError("Antigravity image input requires the Host AttachmentStore", "UNSUPPORTED_MODALITY");
 	const replayBlocks = compatibleReplayState(message, "google-antigravity", model, contentKinds(message))?.blocks ?? [];
 	const parts = [];
+	const isClaude = antigravityModelFamily(model) === "claude";
+	let replayIndex = 0;
+	let sawClaudeFunctionCall = false;
 	for (const block of message.content) {
-		const replayBlock = replayBlocks.find((r) => r.kind === block.type);
-		const blockSignature = block.signature ?? block.thoughtSignature ?? replayBlock?.signature;
+		const replayKind = block.type === "text" || block.type === "reasoning" || block.type === "tool-call" ? block.type : void 0;
+		const replayBlock = replayKind === void 0 ? void 0 : replayBlocks[replayIndex++];
+		const replaySignature = replayBlock !== void 0 && replayBlock.kind === replayKind ? replayBlock.signature : void 0;
+		const blockSignature = block.signature ?? block.thoughtSignature ?? replaySignature;
 		if (block.type === "image") {
 			const stored = await attachments.readImage(block.attachment, signal);
 			parts.push({ inlineData: {
@@ -650,24 +771,33 @@ async function mapMessageWithAttachments(message, model, toolNames, attachments,
 				text: block.text,
 				...blockSignature === void 0 ? {} : { thoughtSignature: blockSignature }
 			});
-		} else if (block.type === "reasoning") parts.push({
-			text: block.text,
-			thought: true,
-			...blockSignature === void 0 ? {} : { thoughtSignature: blockSignature }
-		});
-		else if (block.type === "tool-call") {
-			rememberToolName(toolNames, block.id, block.name);
+		} else if (block.type === "reasoning") {
+			if (isClaude && blockSignature === void 0) continue;
+			parts.push({
+				text: block.text,
+				thought: true,
+				...blockSignature === void 0 ? {} : { thoughtSignature: blockSignature }
+			});
+		} else if (block.type === "tool-call") {
+			const callId = rememberToolName(toolNames, block.id, block.name);
+			const signature = isClaude ? sawClaudeFunctionCall ? void 0 : blockSignature ?? SKIP_THOUGHT_SIGNATURE : blockSignature;
+			sawClaudeFunctionCall ||= isClaude;
 			parts.push({
 				functionCall: {
+					...isClaude ? { id: callId } : {},
 					name: block.name,
 					args: parseJsonObject(block.arguments)
 				},
-				...blockSignature === void 0 ? {} : { thoughtSignature: blockSignature }
+				...signature === void 0 ? {} : { thoughtSignature: signature }
 			});
-		} else if (block.type === "tool-result") parts.push({ functionResponse: {
-			name: requireToolName(toolNames, block.toolCallId),
-			response: { content: blocksToText(block.content) }
-		} });
+		} else if (block.type === "tool-result") {
+			const callId = requireToolCallId(block.toolCallId);
+			parts.push({ functionResponse: {
+				...isClaude ? { id: callId } : {},
+				name: requireToolName(toolNames, callId),
+				response: { content: blocksToText(block.content) }
+			} });
+		}
 	}
 	return {
 		role: message.role === "assistant" ? "model" : "user",
@@ -676,16 +806,21 @@ async function mapMessageWithAttachments(message, model, toolNames, attachments,
 }
 function rememberToolName(toolNames, callId, name) {
 	if (name.length === 0 || name.length > 256 || containsControl(name)) throw new LlmError("The tool call name is invalid", "INVALID_ARGS");
-	const id = String(callId);
+	const id = requireToolCallId(callId);
 	const existing = toolNames.get(id);
 	if (existing !== void 0 && existing !== name) throw new LlmError("A tool call id was reused with a different name", "INVALID_ARGS");
 	toolNames.set(id, name);
+	return id;
 }
 function requireToolName(toolNames, callId) {
-	const name = toolNames.get(String(callId));
+	const name = toolNames.get(requireToolCallId(callId));
 	if (name === void 0) throw new LlmError("A tool result did not match a prior tool call", "INVALID_ARGS");
 	return name;
 }
+function requireToolCallId(value) {
+	if (typeof value !== "string" || value.length === 0 || value.length > 512 || containsControl(value)) throw new LlmError("The tool call id is invalid", "INVALID_ARGS");
+	return value;
+}
 function normalizeReasoningEffort(value) {
 	if (value === void 0) return void 0;
 	const normalized = String(value).toLowerCase();
@@ -794,11 +929,83 @@ function parsePart(value) {
 function errorDetails(value) {
 	const status = numberValue(value.status);
 	const code = stringValue(value.code);
+	const contextWindowExceeded = providerErrorReportsContextWindowExceeded(value, 0);
 	return {
 		...status === void 0 ? {} : { status },
-		...code === void 0 ? {} : { code }
+		...code === void 0 ? {} : { code },
+		...contextWindowExceeded ? { contextWindowExceeded: true } : {}
 	};
 }
+async function responseReportsContextWindowExceeded(response, options) {
+	try {
+		for await (const event of iteratePrivateSse(response, {
+			...options.signal === void 0 ? {} : { signal: options.signal },
+			idleTimeoutMs: options.idleTimeoutMs,
+			totalTimeoutMs: options.totalTimeoutMs,
+			maxBytes: Math.min(MAX_PROVIDER_ERROR_BYTES, options.maxResponseBytes),
+			maxFrameBytes: Math.min(MAX_PROVIDER_ERROR_FRAME_BYTES, options.maxFrameBytes)
+		})) if (providerErrorEnvelopeReportsContextWindowExceeded(event.data.replace(/^\)\]\}'(?:\r?\n)?/u, ""), 0)) return true;
+		return false;
+	} catch (error) {
+		if (error instanceof PrivateTransportError && error.code === "cancelled") throw toLlmError(error);
+		if (isAborted(options.signal)) throw new LlmError("The Antigravity request was cancelled", "CANCELLED");
+		return false;
+	}
+}
+function providerErrorReportsContextWindowExceeded(value, depth) {
+	if (depth > MAX_PROVIDER_ERROR_JSON_DEPTH) return false;
+	if (typeof value.message === "string") {
+		if (isExactContextWindowExceededMessage(value.message)) return true;
+		if (providerErrorEnvelopeReportsContextWindowExceeded(value.message, depth + 1)) return true;
+	}
+	return isRecord(value.error) && providerErrorReportsContextWindowExceeded(value.error, depth + 1);
+}
+function providerErrorEnvelopeReportsContextWindowExceeded(value, depth) {
+	if (depth > MAX_PROVIDER_ERROR_JSON_DEPTH || value.length === 0 || value.length > MAX_PROVIDER_ERROR_BYTES) return false;
+	const json = value.trim();
+	if (!json.startsWith("{") || !jsonDepthIsBounded(json, MAX_PROVIDER_ERROR_JSON_DEPTH)) return false;
+	try {
+		const parsed = JSON.parse(json);
+		return isRecord(parsed) && isRecord(parsed.error) ? providerErrorReportsContextWindowExceeded(parsed.error, depth + 1) : false;
+	} catch {
+		return false;
+	}
+}
+function isExactContextWindowExceededMessage(value) {
+	const match = /^prompt is too long: ([0-9]{1,16}) tokens > ([0-9]{1,16}) maximum$/u.exec(value);
+	const actual = match?.[1];
+	const maximum = match?.[2];
+	return actual !== void 0 && maximum !== void 0 && BigInt(actual) > BigInt(maximum);
+}
+function jsonDepthIsBounded(value, maxDepth) {
+	let depth = 0;
+	let inString = false;
+	let escaped = false;
+	for (const character of value) {
+		if (inString) {
+			if (escaped) escaped = false;
+			else if (character === "\\") escaped = true;
+			else if (character === "\"") inString = false;
+			continue;
+		}
+		if (character === "\"") {
+			inString = true;
+			continue;
+		}
+		if (character === "{" || character === "[") {
+			depth += 1;
+			if (depth > maxDepth) return false;
+		} else if (character === "}" || character === "]") {
+			depth -= 1;
+			if (depth < 0) return false;
+		}
+	}
+	return depth === 0 && !inString && !escaped;
+}
+function contextWindowExceededError(status) {
+	const message = "The Antigravity request exceeded the model context window";
+	return status === void 0 ? new LlmError(message, CONTEXT_WINDOW_EXCEEDED_CODE) : new LlmError(message, CONTEXT_WINDOW_EXCEEDED_CODE, { status });
+}
 function parseUsage(value) {
 	if (!isRecord(value)) return void 0;
 	const input = numberValue(value.promptTokenCount ?? value.inputTokenCount);

package/lib/replay.js

@@ -24,7 +24,7 @@ function createReplayState(model, family, finish, blocks) {
 	};
 }
 /** Validate replay metadata before it can affect a later private request. */
-function compatibleReplayState(message, provider, model, _blockKinds) {
+function compatibleReplayState(message, provider, model, blockKinds) {
 	if (provider !== "google-antigravity" || message.role !== "assistant") return void 0;
 	const provenance = isRecord(message.source) ? message.source : void 0;
 	if (provenance !== void 0 && provenance.kind === "model" && (provenance.provider !== provider || provenance.model !== model)) return void 0;
@@ -42,6 +42,7 @@ function compatibleReplayState(message, provider, model, _blockKinds) {
 			...signature === void 0 ? {} : { signature }
 		});
 	}
+	if (blockKinds !== void 0 && (blocks.length !== blockKinds.length || blocks.some((block, index) => block.kind !== blockKinds[index]))) return void 0;
 	const finish = value.response.finish === void 0 ? void 0 : safeFinish(value.response.finish);
 	if (value.response.finish !== void 0 && finish === void 0) return void 0;
 	return {

package/lib/types/llm-adapter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"llm-adapter.d.ts","sourceRoot":"","sources":["../../src/llm-adapter.ts"],"names":[],"mappings":"AAAA,wEAAwE;AAaxE,OAAO,EAEL,UAAU,EAGV,kBAAkB,EACnB,MAAM,sBAAsB,CAAA;AAC7B,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,6BAA6B,CAAA;AAClE,OAAO,KAAK,EAEV,eAAe,EACf,YAAY,EACZ,eAAe,EACf,oBAAoB,EAGpB,WAAW,EAEZ,MAAM,sBAAsB,CAAA;AAC7B,OAAO,KAAK,EAAE,qBAAqB,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAA;AACxF,OAAO,EAWL,KAAK,gBAAgB,EACtB,MAAM,wBAAwB,CAAA;AAU/B,OAAO,KAAK,EAAgC,2BAA2B,EAAE,MAAM,oBAAoB,CAAA;AAEnG,eAAO,MAAM,oBAAoB,EAAG,oBAA6B,CAAA;AACjE,eAAO,MAAM,2BAA2B,sDAAiF,CAAA;AACzH,eAAO,MAAM,6BAA6B,wCAAmE,CAAA;AAC7G,eAAO,MAAM,qCAAqC,6CAAwE,CAAA;AAC1H,eAAO,MAAM,qBAAqB,sBAAuB,CAAA;AAKzD,MAAM,WAAW,+BAA+B;IAC9C,UAAU,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,OAAO,CAAC,EAAE;QAAE,QAAQ,CAAC,YAAY,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,OAAO,CAAC,cAAc,GAAG,SAAS,CAAC,CAAA;CACrH;AAED,MAAM,WAAW,yBAAyB;IACxC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,qBAAqB,EAAE,YAAY,CAAC,GAAG,+BAA+B,CAAA;IAC1F,uEAAuE;IACvE,QAAQ,CAAC,SAAS,CAAC,EAAE,gBAAgB,CAAA;IACrC,QAAQ,CAAC,uBAAuB,CAAC,EAAE,MAAM,CAAA;IACzC,QAAQ,CAAC,aAAa,CAAC,EAAE,MAAM,CAAA;IAC/B,QAAQ,CAAC,cAAc,CAAC,EAAE,MAAM,CAAA;IAChC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,MAAM,CAAA;IAClC,QAAQ,CAAC,aAAa,CAAC,EAAE,MAAM,CAAA;IAC/B,QAAQ,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,eAAe,EAAE,WAAW,CAAC,CAAA;CAC1D;AAyBD,MAAM,WAAW,0BAA0B;IACzC,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAA;IAC1B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAA;IAC1B,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IACvC,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAA;CAC/B;AAED,0EAA0E;AAC1E,qBAAa,kBAAmB,SAAQ,UAAU;IAapC,OAAO,CAAC,QAAQ,CAAC,cAAc;IAZ3C,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAkB;IAC5C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAqD;IAC9E,OAAO,CAAC,QAAQ,CAAC,OAAO,CAEvB;IACD,OAAO,CAAC,QAAQ,CAAC,WAAW,CAA8B;IAC1D,OAAO,CAAC,gBAAgB,CAAoB;IAC5C,OAAO,CAAC,gBAAgB,CAAI;IAC5B,OAAO,CAAC,eAAe,CAAoB;IAC3C,OAAO,CAAC,kBAAkB,CAAoB;IAC9C,OAAO,CAAC,WAAW,CAA6B;gBAEnB,cAAc,EAAE,yBAAyB;IAe7D,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,eAAe;IAK/C,mBAAmB,IAAI,UAAU,CAAC,OAAO,kBAAkB,CAAC;IAMrE,mFAAmF;IACnF,sBAAsB,IAAI,IAAI;IAQ9B,+EAA+E;IAC/E,eAAe,IAAI,2BAA2B;IAI9C,kFAAkF;IAC5E,YAAY,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,YAAY,UAAQ,GAAG,OAAO,CAAC,2BAA2B,CAAC;IAWrF,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,SAAS,YAAY,EAAE,CAAC;IAkBnG,OAAO,CAAC,oBAAoB;IAWb,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAqBlG,MAAM,CAAC,OAAO,EAAE,eAAe,GAAG,aAAa,CAAC,WAAW,CAAC;YA+D7D,cAAc;YAkFf,gBAAgB;IA0F9B,OAAO,CAAC,sBAAsB;YAQhB,cAAc;CAO7B;AA8MD,wBAAgB,+BAA+B,CAAC,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,cAAc,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAM7H"}
+{"version":3,"file":"llm-adapter.d.ts","sourceRoot":"","sources":["../../src/llm-adapter.ts"],"names":[],"mappings":"AAAA,wEAAwE;AAgBxE,OAAO,EAGL,UAAU,EAGV,kBAAkB,EACnB,MAAM,sBAAsB,CAAA;AAC7B,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,6BAA6B,CAAA;AAClE,OAAO,KAAK,EAEV,eAAe,EACf,YAAY,EACZ,eAAe,EACf,oBAAoB,EAGpB,WAAW,EAEZ,MAAM,sBAAsB,CAAA;AAC7B,OAAO,KAAK,EAAE,qBAAqB,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAA;AACxF,OAAO,EAWL,KAAK,gBAAgB,EACtB,MAAM,wBAAwB,CAAA;AAU/B,OAAO,KAAK,EAAgC,2BAA2B,EAAE,MAAM,oBAAoB,CAAA;AAEnG,eAAO,MAAM,oBAAoB,EAAG,oBAA6B,CAAA;AACjE,eAAO,MAAM,2BAA2B,sDAAiF,CAAA;AACzH,eAAO,MAAM,6BAA6B,wCAAmE,CAAA;AAC7G,eAAO,MAAM,qCAAqC,6CAAwE,CAAA;AAC1H,eAAO,MAAM,qBAAqB,sBAAuB,CAAA;AAQzD,MAAM,WAAW,+BAA+B;IAC9C,UAAU,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,OAAO,CAAC,EAAE;QAAE,QAAQ,CAAC,YAAY,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,OAAO,CAAC,cAAc,GAAG,SAAS,CAAC,CAAA;CACrH;AAED,MAAM,WAAW,yBAAyB;IACxC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,qBAAqB,EAAE,YAAY,CAAC,GAAG,+BAA+B,CAAA;IAC1F,uEAAuE;IACvE,QAAQ,CAAC,SAAS,CAAC,EAAE,gBAAgB,CAAA;IACrC,QAAQ,CAAC,uBAAuB,CAAC,EAAE,MAAM,CAAA;IACzC,QAAQ,CAAC,aAAa,CAAC,EAAE,MAAM,CAAA;IAC/B,QAAQ,CAAC,cAAc,CAAC,EAAE,MAAM,CAAA;IAChC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,MAAM,CAAA;IAClC,QAAQ,CAAC,aAAa,CAAC,EAAE,MAAM,CAAA;IAC/B,QAAQ,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,eAAe,EAAE,WAAW,CAAC,CAAA;CAC1D;AAyBD,MAAM,WAAW,0BAA0B;IACzC,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAA;IAC1B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAA;IAC1B,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IACvC,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAA;CAC/B;AAED,0EAA0E;AAC1E,qBAAa,kBAAmB,SAAQ,UAAU;IAapC,OAAO,CAAC,QAAQ,CAAC,cAAc;IAZ3C,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAkB;IAC5C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAqD;IAC9E,OAAO,CAAC,QAAQ,CAAC,OAAO,CAEvB;IACD,OAAO,CAAC,QAAQ,CAAC,WAAW,CAA8B;IAC1D,OAAO,CAAC,gBAAgB,CAAoB;IAC5C,OAAO,CAAC,gBAAgB,CAAI;IAC5B,OAAO,CAAC,eAAe,CAAoB;IAC3C,OAAO,CAAC,kBAAkB,CAAoB;IAC9C,OAAO,CAAC,WAAW,CAA6B;gBAEnB,cAAc,EAAE,yBAAyB;IAe7D,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,eAAe;IAK/C,mBAAmB,IAAI,UAAU,CAAC,OAAO,kBAAkB,CAAC;IAMrE,mFAAmF;IACnF,sBAAsB,IAAI,IAAI;IAQ9B,+EAA+E;IAC/E,eAAe,IAAI,2BAA2B;IAI9C,kFAAkF;IAC5E,YAAY,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,YAAY,UAAQ,GAAG,OAAO,CAAC,2BAA2B,CAAC;IAWrF,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,SAAS,YAAY,EAAE,CAAC;IAkBnG,OAAO,CAAC,oBAAoB;IAWb,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAqBlG,MAAM,CAAC,OAAO,EAAE,eAAe,GAAG,aAAa,CAAC,WAAW,CAAC;YAyE7D,cAAc;YAsFf,gBAAgB;IA0F9B,OAAO,CAAC,sBAAsB;YAQhB,cAAc;CAO7B;AA8MD,wBAAgB,+BAA+B,CAAC,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,cAAc,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAM7H"}

package/lib/types/replay.d.ts

@@ -21,7 +21,7 @@ export interface AntigravityReplayState extends ReplayEnvelope {
 /** Keep only provider-issued signatures and bounded response facts. */
 export declare function createReplayState(model: string, family: AntigravityReplayResponse['family'], finish: string | undefined, blocks: readonly AntigravityReplayBlock[]): AntigravityReplayState;
 /** Validate replay metadata before it can affect a later private request. */
-export declare function compatibleReplayState(message: Message, provider: string, model: string, _blockKinds?: readonly ReplayBlockKind[]): AntigravityReplayState | undefined;
+export declare function compatibleReplayState(message: Message, provider: string, model: string, blockKinds?: readonly ReplayBlockKind[]): AntigravityReplayState | undefined;
 /** Return the block family without allowing a model alias to cross families. */
 export declare function antigravityModelFamily(model: string): AntigravityReplayResponse['family'];
 /** Convert DSH schemas to the small function-declaration subset accepted privately. */

package/lib/types/replay.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"replay.d.ts","sourceRoot":"","sources":["../../src/replay.ts"],"names":[],"mappings":"AAAA,0EAA0E;AAE1E,OAAO,KAAK,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAA;AACtE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAA;AAEnD,eAAO,MAAM,0BAA0B,EAAG,CAAU,CAAA;AAIpD,MAAM,MAAM,eAAe,GAAG,MAAM,GAAG,WAAW,GAAG,WAAW,CAAA;AAEhE,MAAM,WAAW,sBAAsB;IACrC,QAAQ,CAAC,IAAI,EAAE,eAAe,CAAA;IAC9B,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAC5B;AAED,MAAM,WAAW,yBAAyB;IACxC,QAAQ,CAAC,OAAO,EAAE,OAAO,0BAA0B,CAAA;IACnD,QAAQ,CAAC,QAAQ,EAAE,oBAAoB,CAAA;IACvC,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAA;IACtB,QAAQ,CAAC,MAAM,EAAE,QAAQ,GAAG,QAAQ,GAAG,SAAS,GAAG,SAAS,CAAA;IAC5D,QAAQ,CAAC,MAAM,CAAC,EAAE,MAAM,CAAA;CACzB;AAED,MAAM,WAAW,sBAAuB,SAAQ,cAAc;IAC5D,QAAQ,CAAC,QAAQ,EAAE,yBAAyB,CAAA;IAC5C,QAAQ,CAAC,MAAM,EAAE,SAAS,sBAAsB,EAAE,CAAA;CACnD;AAED,uEAAuE;AACvE,wBAAgB,iBAAiB,CAC/B,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,yBAAyB,CAAC,QAAQ,CAAC,EAC3C,MAAM,EAAE,MAAM,GAAG,SAAS,EAC1B,MAAM,EAAE,SAAS,sBAAsB,EAAE,GACxC,sBAAsB,CAgBxB;AAED,6EAA6E;AAC7E,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,OAAO,EAChB,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,SAAS,eAAe,EAAE,GACvC,sBAAsB,GAAG,SAAS,CAmCpC;AAED,gFAAgF;AAChF,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,MAAM,GAAG,yBAAyB,CAAC,QAAQ,CAAC,CAMzF;AAED,uFAAuF;AACvF,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,SAAS,UAAU,EAAE,GAAG,SAAS,GAAG,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAOhH;AAED,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,SAAS,UAAU,EAAE,GAAG,SAAS,GAAG,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAMtH"}
+{"version":3,"file":"replay.d.ts","sourceRoot":"","sources":["../../src/replay.ts"],"names":[],"mappings":"AAAA,0EAA0E;AAE1E,OAAO,KAAK,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAA;AACtE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAA;AAEnD,eAAO,MAAM,0BAA0B,EAAG,CAAU,CAAA;AAIpD,MAAM,MAAM,eAAe,GAAG,MAAM,GAAG,WAAW,GAAG,WAAW,CAAA;AAEhE,MAAM,WAAW,sBAAsB;IACrC,QAAQ,CAAC,IAAI,EAAE,eAAe,CAAA;IAC9B,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAC5B;AAED,MAAM,WAAW,yBAAyB;IACxC,QAAQ,CAAC,OAAO,EAAE,OAAO,0BAA0B,CAAA;IACnD,QAAQ,CAAC,QAAQ,EAAE,oBAAoB,CAAA;IACvC,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAA;IACtB,QAAQ,CAAC,MAAM,EAAE,QAAQ,GAAG,QAAQ,GAAG,SAAS,GAAG,SAAS,CAAA;IAC5D,QAAQ,CAAC,MAAM,CAAC,EAAE,MAAM,CAAA;CACzB;AAED,MAAM,WAAW,sBAAuB,SAAQ,cAAc;IAC5D,QAAQ,CAAC,QAAQ,EAAE,yBAAyB,CAAA;IAC5C,QAAQ,CAAC,MAAM,EAAE,SAAS,sBAAsB,EAAE,CAAA;CACnD;AAED,uEAAuE;AACvE,wBAAgB,iBAAiB,CAC/B,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,yBAAyB,CAAC,QAAQ,CAAC,EAC3C,MAAM,EAAE,MAAM,GAAG,SAAS,EAC1B,MAAM,EAAE,SAAS,sBAAsB,EAAE,GACxC,sBAAsB,CAgBxB;AAED,6EAA6E;AAC7E,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,OAAO,EAChB,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,UAAU,CAAC,EAAE,SAAS,eAAe,EAAE,GACtC,sBAAsB,GAAG,SAAS,CAqCpC;AAED,gFAAgF;AAChF,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,MAAM,GAAG,yBAAyB,CAAC,QAAQ,CAAC,CAMzF;AAED,uFAAuF;AACvF,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,SAAS,UAAU,EAAE,GAAG,SAAS,GAAG,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAOhH;AAED,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,SAAS,UAAU,EAAE,GAAG,SAAS,GAAG,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAMtH"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-antigravity-auth",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "Private, single-account, unofficial Antigravity OAuth capability bundle for DeepSeek Harness",
   "type": "module",
   "engines": {