@benjamolina/pi-antigravity-guard 0.4.6 → 0.4.8

package/README.md

@@ -22,9 +22,9 @@ All seven rows below are registered for text. The exact `antigravity-gemini-3.8-
 | `antigravity-gemini-3.7-flash` | off, low, medium, high | Enabled (off, low, medium, high) | pi-json-tool-loop/gemini-3.7-flash-off-v1 |
 | `antigravity-gemini-3.6-flash` | off, low, medium, high | Enabled (off, low, medium, high) | pi-json-tool-loop/gemini-3.6-flash-off-v1 |
 | `antigravity-gemini-3.1-pro` | off, low, high | Enabled (off, low, high) | pi-json-tool-loop/gemini-3.1-pro-off-v1 |
-| `antigravity-claude-sonnet-4.6` | off, high | Disabled | claude-continuity-unproven |
-| `antigravity-claude-opus-4.6-thinking` | off, high | Disabled | claude-continuity-unproven |
-| `antigravity-gpt-oss-120b` | off, medium | Disabled | missing-direct-evidence |
+| `antigravity-claude-sonnet-4.6` | off, high | Enabled (off, high) | pi-json-tool-loop/claude-sonnet-4.6-off-v1 |
+| `antigravity-claude-opus-4.6-thinking` | off, high | Enabled (off, high) | pi-json-tool-loop/claude-opus-4.6-thinking-off-v1 |
+| `antigravity-gpt-oss-120b` | off, medium | Enabled (off, medium) | pi-json-tool-loop/gpt-oss-120b-off-v1 |
 
 `off` is available for every row even though it is omitted from Pi's level map; omitted reasoning also selects that row's `off` route. Unsupported levels are not advertised and are rejected before transport. `antigravity-gemini-3.5-flash` is not registered or advertised as supported because its recorded HTTP-200 response lacks the strict terminal metadata required for admission. A future `fixture-qualified` route remains disabled for ordinary tool use; only separately authorized direct validation for the exact route may make it enabled.
 

package/dist/catalog.d.ts

@@ -8,7 +8,8 @@ export interface ToolEvidenceRef {
     readonly wireModel: string;
 }
 export declare const GEMINI_PARAMETERS_JSON_SCHEMA_PROFILE: "gemini-parameters-json-schema";
-export type ToolSchemaProfile = typeof GEMINI_PARAMETERS_JSON_SCHEMA_PROFILE;
+export declare const CLAUDE_CUSTOM_PARAMETERS_PROFILE: "claude-custom-parameters";
+export type ToolSchemaProfile = typeof GEMINI_PARAMETERS_JSON_SCHEMA_PROFILE | typeof CLAUDE_CUSTOM_PARAMETERS_PROFILE;
 export type ToolCapability = {
     readonly state: "disabled";
     readonly contractRevision: 1;
@@ -316,7 +317,7 @@ declare const CATALOG: readonly [{
         };
     };
     readonly replay: {
-        readonly kind: "strip";
+        readonly kind: "same-public-model";
     };
     readonly response: {
         readonly kind: "gemini-envelope";
@@ -356,7 +357,7 @@ declare const CATALOG: readonly [{
         };
     };
     readonly replay: {
-        readonly kind: "strip";
+        readonly kind: "same-public-model";
     };
     readonly response: {
         readonly kind: "gemini-envelope";
@@ -394,7 +395,7 @@ declare const CATALOG: readonly [{
         };
     };
     readonly replay: {
-        readonly kind: "strip";
+        readonly kind: "same-public-model";
     };
     readonly response: {
         readonly kind: "gemini-envelope";

package/dist/catalog.js

@@ -1,5 +1,6 @@
 const ANSWER_RESERVE = 1024;
 export const GEMINI_PARAMETERS_JSON_SCHEMA_PROFILE = "gemini-parameters-json-schema";
+export const CLAUDE_CUSTOM_PARAMETERS_PROFILE = "claude-custom-parameters";
 const CATALOG = freeze(defineCatalog([{
         publicId: "antigravity-gemini-3.8-flash",
         descriptor: {
@@ -72,10 +73,10 @@ const CATALOG = freeze(defineCatalog([{
             maxTokens: 64_000,
         },
         routes: {
-            off: { wireModel: "claude-sonnet-4-6", thinking: { kind: "budget", budget: 0, includeThoughts: false }, tools: disabled("claude-continuity-unproven") },
-            high: { wireModel: "claude-sonnet-4-6", thinking: { kind: "budget", budget: 1024, includeThoughts: true }, tools: disabled("claude-continuity-unproven") },
+            off: { wireModel: "claude-sonnet-4-6", thinking: { kind: "budget", budget: 0, includeThoughts: false }, tools: enabled({ record: "pi-json-tool-loop", revision: "claude-sonnet-4.6-off-v1", publicModelId: "antigravity-claude-sonnet-4.6", reasoning: "off", wireModel: "claude-sonnet-4-6" }, CLAUDE_CUSTOM_PARAMETERS_PROFILE) },
+            high: { wireModel: "claude-sonnet-4-6", thinking: { kind: "budget", budget: 1024, includeThoughts: true }, tools: enabled({ record: "pi-json-tool-loop", revision: "claude-sonnet-4.6-high-v1", publicModelId: "antigravity-claude-sonnet-4.6", reasoning: "high", wireModel: "claude-sonnet-4-6" }, CLAUDE_CUSTOM_PARAMETERS_PROFILE) },
         },
-        replay: { kind: "strip" },
+        replay: { kind: "same-public-model" },
         response: { kind: "gemini-envelope", family: "claude" },
     }, {
         publicId: "antigravity-claude-opus-4.6-thinking",
@@ -86,10 +87,10 @@ const CATALOG = freeze(defineCatalog([{
             maxTokens: 64_000,
         },
         routes: {
-            off: { wireModel: "claude-opus-4-6-thinking", thinking: { kind: "budget", budget: 0, includeThoughts: false }, tools: disabled("claude-continuity-unproven") },
-            high: { wireModel: "claude-opus-4-6-thinking", thinking: { kind: "budget", budget: 1024, includeThoughts: true }, tools: disabled("claude-continuity-unproven") },
+            off: { wireModel: "claude-opus-4-6-thinking", thinking: { kind: "budget", budget: 0, includeThoughts: false }, tools: enabled({ record: "pi-json-tool-loop", revision: "claude-opus-4.6-thinking-off-v1", publicModelId: "antigravity-claude-opus-4.6-thinking", reasoning: "off", wireModel: "claude-opus-4-6-thinking" }, CLAUDE_CUSTOM_PARAMETERS_PROFILE) },
+            high: { wireModel: "claude-opus-4-6-thinking", thinking: { kind: "budget", budget: 1024, includeThoughts: true }, tools: enabled({ record: "pi-json-tool-loop", revision: "claude-opus-4.6-thinking-high-v1", publicModelId: "antigravity-claude-opus-4.6-thinking", reasoning: "high", wireModel: "claude-opus-4-6-thinking" }, CLAUDE_CUSTOM_PARAMETERS_PROFILE) },
         },
-        replay: { kind: "strip" },
+        replay: { kind: "same-public-model" },
         response: { kind: "gemini-envelope", family: "claude" },
     }, {
         publicId: "antigravity-gpt-oss-120b",
@@ -100,10 +101,10 @@ const CATALOG = freeze(defineCatalog([{
             maxTokens: 32_768,
         },
         routes: {
-            off: { wireModel: "gpt-oss-120b-medium", thinking: { kind: "omit" }, tools: disabled("missing-direct-evidence") },
-            medium: { wireModel: "gpt-oss-120b-medium", thinking: { kind: "budget", budget: 8192, includeThoughts: true }, tools: disabled("missing-direct-evidence") },
+            off: { wireModel: "gpt-oss-120b-medium", thinking: { kind: "omit" }, tools: enabled({ record: "pi-json-tool-loop", revision: "gpt-oss-120b-off-v1", publicModelId: "antigravity-gpt-oss-120b", reasoning: "off", wireModel: "gpt-oss-120b-medium" }, CLAUDE_CUSTOM_PARAMETERS_PROFILE) },
+            medium: { wireModel: "gpt-oss-120b-medium", thinking: { kind: "budget", budget: 8192, includeThoughts: true }, tools: enabled({ record: "pi-json-tool-loop", revision: "gpt-oss-120b-medium-v1", publicModelId: "antigravity-gpt-oss-120b", reasoning: "medium", wireModel: "gpt-oss-120b-medium" }, CLAUDE_CUSTOM_PARAMETERS_PROFILE) },
         },
-        replay: { kind: "strip" },
+        replay: { kind: "same-public-model" },
         response: { kind: "gemini-envelope", family: "gpt-oss" },
     }]));
 const lookup = new Map(CATALOG.map((entry) => [entry.publicId, entry]));

package/dist/context.d.ts

@@ -17,7 +17,8 @@ interface Content {
 interface GeminiFunctionDeclaration {
     readonly name: string;
     readonly description: string;
-    readonly parametersJsonSchema: object;
+    readonly parametersJsonSchema?: object;
+    readonly parameters?: object;
 }
 export interface GenerationRequest {
     project: string;

package/dist/context.js

@@ -1,4 +1,5 @@
-import { getCatalogEntry, resolveGenerationSelection } from "./catalog.js";
+import { CLAUDE_CUSTOM_PARAMETERS_PROFILE, getCatalogEntry, resolveGenerationSelection } from "./catalog.js";
+import { normalizeCustomToolSchema } from "./tool-schema.js";
 import { hasToolContext, prepareToolContext, replayToolHistory } from "./tool-context.js";
 const MAX_TEXT_BYTES = 8 * 1024 * 1024;
 const PROVIDER = "antigravity-guard";
@@ -25,18 +26,27 @@ export function serializeContext(input, injectedSelection, onDiagnostics) {
     const { toolChoice: _choice, ...textOptions } = isRecord(options) ? options : {};
     const text = serializeTextContext({ ...input, context: { ...textContext, messages: [{ role: "user", content: "placeholder", timestamp: 0 }] }, options: textOptions });
     const sameModel = (message) => field(message, "role") === "assistant" && entry.replay.kind === "same-public-model" && isSameProviderAndModel(message, entry.publicId);
-    const signedToolReplay = geminiRequiresSignedToolReplay(selection.route.wireModel) ? {
-        requireSignedToolCalls: true,
+    const requireSigned = geminiRequiresSignedToolReplay(selection.route.wireModel);
+    const includeToolCallId = entry.response.family === "claude" || entry.response.family === "gpt-oss";
+    const toolReplayPolicy = (requireSigned || includeToolCallId) ? {
+        requireSignedToolCalls: requireSigned,
         isSameModel: sameModel,
         toolCallSignature: (part, message) => sameModel(message) ? validThoughtSignature(field(part, "thoughtSignature")) : undefined,
+        includeToolCallId,
     } : undefined;
-    const contents = replayToolHistory(isDenseArray(messages), (part, message) => messagePart(part, field(message, "role") === "assistant", sameModel(message)), onDiagnostics, signedToolReplay, prepared?.declarations.length ?? 0);
+    const contents = replayToolHistory(isDenseArray(messages), (part, message) => messagePart(part, field(message, "role") === "assistant", sameModel(message)), onDiagnostics, toolReplayPolicy, prepared?.declarations.length ?? 0);
     const { systemInstruction, generationConfig } = text.request;
     return { ...text, request: {
             contents,
             ...(systemInstruction ? { systemInstruction } : {}),
             ...(prepared ? {
-                tools: [{ functionDeclarations: prepared.declarations.map(({ name, description, parameters }) => ({ name, description, parametersJsonSchema: parameters })) }],
+                tools: [{ functionDeclarations: prepared.declarations.map(({ name, description, parameters }) => ({
+                            name,
+                            description,
+                            ...(selection.tools.state === "enabled" && selection.tools.schemaProfile === CLAUDE_CUSTOM_PARAMETERS_PROFILE
+                                ? { parameters: normalizeCustomToolSchema(parameters) }
+                                : { parametersJsonSchema: parameters }),
+                        })) }],
                 ...(prepared.mode === "NONE" ? { toolConfig: { functionCallingConfig: { mode: "NONE" } } } : {}),
             } : {}),
             generationConfig,

package/dist/tool-context.d.ts

@@ -20,10 +20,12 @@ export interface ToolReplayDiagnostics {
     readonly assistantToolCallBlockCount: number;
     readonly declaredToolCount: number;
 }
+export declare function sanitizeToolCallId(id: string, fallbackName?: string): string;
 export interface ToolReplayPolicy {
     readonly requireSignedToolCalls: boolean;
     readonly isSameModel: (message: Record<string, unknown>) => boolean;
     readonly toolCallSignature: (part: Record<string, unknown>, message: Record<string, unknown>) => string | undefined;
+    readonly includeToolCallId?: boolean;
 }
 export declare function replayToolHistory(messages: unknown[], serializePart: (part: Record<string, unknown>, message: Record<string, unknown>) => WirePart, onDiagnostics?: (diagnostics: ToolReplayDiagnostics) => void, policy?: ToolReplayPolicy, declaredToolCount?: number): WireContent[];
 export {};

package/dist/tool-context.js

@@ -1,5 +1,5 @@
 import { canonicalJson, ToolPreflightError } from "./tool-contract.js";
-import { GEMINI_PARAMETERS_JSON_SCHEMA_PROFILE } from "./catalog.js";
+import { CLAUDE_CUSTOM_PARAMETERS_PROFILE, GEMINI_PARAMETERS_JSON_SCHEMA_PROFILE } from "./catalog.js";
 import { normalizeToolDeclarations } from "./tool-schema.js";
 export function hasToolContext(context) {
     if (!isRecord(context))
@@ -18,7 +18,7 @@ export function hasToolContext(context) {
     });
 }
 export function prepareToolContext(tools, choice, schemaProfile) {
-    if (schemaProfile !== GEMINI_PARAMETERS_JSON_SCHEMA_PROFILE)
+    if (schemaProfile !== GEMINI_PARAMETERS_JSON_SCHEMA_PROFILE && schemaProfile !== CLAUDE_CUSTOM_PARAMETERS_PROFILE)
         throw new ToolPreflightError("PI_TOOL_SCHEMA_PROFILE_UNSUPPORTED", "tools", "$");
     const declarations = normalizeToolDeclarations(tools ?? []);
     if (!declarations.length) {
@@ -32,6 +32,12 @@ export function prepareToolContext(tools, choice, schemaProfile) {
         unsupportedChoice();
     return { declarations, mode: choice === "none" ? "NONE" : "AUTO" };
 }
+let toolCallCounter = 0;
+export function sanitizeToolCallId(id, fallbackName) {
+    const cleaned = id.replace(/[^a-zA-Z0-9_-]/g, "_");
+    const capped = cleaned.slice(0, 64);
+    return capped || `${fallbackName || "tool"}_${++toolCallCounter}`;
+}
 export function replayToolHistory(messages, serializePart, onDiagnostics, policy, declaredToolCount = 0) {
     const output = [];
     const calls = new Set();
@@ -63,7 +69,7 @@ export function replayToolHistory(messages, serializePart, onDiagnostics, policy
             if (actual)
                 return actual;
             recoveryCount.value += 1;
-            return pending.observations ? observation(call, missingResultText()) : missingResult(call);
+            return pending.observations ? observation(call, missingResultText()) : missingResult(call, policy?.includeToolCallId);
         });
         if (pending.extraParts?.length) {
             turnParts.push(...pending.extraParts);
@@ -109,7 +115,7 @@ export function replayToolHistory(messages, serializePart, onDiagnostics, policy
             const text = textParts.join("\n\n");
             const outputText = text || (imageParts.length > 0 ? "[Image content]" : "");
             const response = value(current, "isError") === true ? { error: outputText } : { output: outputText };
-            pending.results.set(id, pending.observations ? observation(call, outputText) : { functionResponse: { name: call.name, response } });
+            pending.results.set(id, pending.observations ? observation(call, outputText) : { functionResponse: { ...(policy?.includeToolCallId ? { id: sanitizeToolCallId(call.id, call.name) } : {}), name: call.name, response } });
             if (imageParts.length > 0) {
                 if (!pending.extraParts)
                     pending.extraParts = [];
@@ -174,7 +180,7 @@ export function replayToolHistory(messages, serializePart, onDiagnostics, policy
                 return [part];
             if (observations)
                 return [];
-            return [{ functionCall: { name: part.call.name, args: part.call.args }, ...(part.signature ? { thoughtSignature: part.signature } : {}) }];
+            return [{ functionCall: { ...(policy?.includeToolCallId ? { id: sanitizeToolCallId(part.call.id, part.call.name) } : {}), name: part.call.name, args: part.call.args }, ...(part.signature ? { thoughtSignature: part.signature } : {}) }];
         }));
         pending = { calls: group, results: new Map(), observations };
     }
@@ -190,9 +196,10 @@ function observation(call, text) {
 function missingResultText() {
     return "Tool execution did not complete or its result was not recorded. Treat the call as failed; do not assume it had no side effects and do not retry it automatically.";
 }
-function missingResult(call) {
+function missingResult(call, includeToolCallId) {
     return {
         functionResponse: {
+            ...(includeToolCallId ? { id: sanitizeToolCallId(call.id, call.name) } : {}),
             name: call.name,
             response: {
                 error: {

package/dist/tool-schema.d.ts

@@ -5,3 +5,4 @@ export interface ToolDeclaration {
     readonly parameters: JsonObject;
 }
 export declare function normalizeToolDeclarations(value: unknown): readonly ToolDeclaration[];
+export declare function normalizeCustomToolSchema(schema: unknown): JsonObject;

package/dist/tool-schema.js

@@ -210,3 +210,34 @@ function isJsonObject(value) {
 function define(target, key, value) {
     Object.defineProperty(target, key, { value, enumerable: true, configurable: false, writable: false });
 }
+const CUSTOM_TOOL_SCHEMA_ALLOW = new Set([
+    "type",
+    "description",
+    "properties",
+    "required",
+    "items",
+    "enum",
+]);
+export function normalizeCustomToolSchema(schema) {
+    if (!schema || typeof schema !== "object" || Array.isArray(schema))
+        return {};
+    const out = {};
+    for (const [key, value] of Object.entries(schema)) {
+        if (!CUSTOM_TOOL_SCHEMA_ALLOW.has(key))
+            continue;
+        if (key === "properties" && value && typeof value === "object" && !Array.isArray(value)) {
+            const props = {};
+            for (const [propName, propSchema] of Object.entries(value)) {
+                props[propName] = normalizeCustomToolSchema(propSchema);
+            }
+            out.properties = props;
+            continue;
+        }
+        if (key === "items") {
+            out.items = normalizeCustomToolSchema(value);
+            continue;
+        }
+        out[key] = value;
+    }
+    return out;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@benjamolina/pi-antigravity-guard",
-  "version": "0.4.6",
+  "version": "0.4.8",
   "description": "Pi adapter for Antigravity Guard",
   "type": "module",
   "license": "MIT",
@@ -24,7 +24,9 @@
     "README.md"
   ],
   "pi": {
-    "extensions": ["./dist/extension.js"]
+    "extensions": [
+      "./dist/extension.js"
+    ]
   },
   "dependencies": {
     "@benjamolina/antigravity-guard-core": "0.1.0"