@benjamolina/pi-antigravity-guard 0.4.5 → 0.4.7

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";
@@ -436,7 +437,7 @@ export declare function toPiModelDescriptor(entry: CatalogEntry): {
         medium: string | null;
         minimal: string | null;
     };
-    input: "text"[];
+    input: ("text" | "image")[];
     cost: {
         input: number;
         output: number;

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]));
@@ -179,12 +180,13 @@ export function resolveToolCapability(capability, publicModelId, reasoning, wire
     return capability;
 }
 export function toPiModelDescriptor(entry) {
+    const isMultimodal = entry.response.family === "gemini" || entry.response.family === "claude";
     return {
         id: entry.publicId,
-        name: entry.descriptor.name,
+        name: entry.descriptor.name.replace(", text only", ""),
         reasoning: true,
         thinkingLevelMap: { ...entry.descriptor.thinkingLevelMap },
-        input: ["text"],
+        input: (isMultimodal ? ["text", "image"] : ["text"]),
         cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
         contextWindow: entry.descriptor.contextWindow,
         maxTokens: entry.descriptor.maxTokens,

package/dist/context.d.ts

@@ -2,9 +2,13 @@ import type { Context, Model, SimpleStreamOptions } from "@earendil-works/pi-ai"
 import { type GenerationSelection } from "./catalog.ts";
 import { type ToolReplayDiagnostics } from "./tool-context.ts";
 interface Part {
-    text: string;
+    text?: string;
     thought?: true;
     thoughtSignature?: string;
+    inlineData?: {
+        mimeType: string;
+        data: string;
+    };
 }
 interface Content {
     role: "user" | "model";
@@ -13,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";
@@ -36,7 +37,13 @@ export function serializeContext(input, injectedSelection, onDiagnostics) {
             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,
@@ -77,7 +84,7 @@ export function serializeTextContext(input) {
                 fail("Unsupported context role for this text-only provider.");
             const assistant = role === "assistant";
             const parts = messageParts(field(message, "content", true), assistant, assistant && entry.replay.kind === "same-public-model" && isSameProviderAndModel(message, entry.publicId));
-            textBytes += parts.reduce((total, part) => total + byteLength(part.text), 0);
+            textBytes += parts.reduce((total, part) => total + (part.text ? byteLength(part.text) : (part.inlineData ? byteLength(part.inlineData.data) : 0)), 0);
             if (textBytes > MAX_TEXT_BYTES)
                 fail("Text context is too large.");
             contents.push({ role: role === "assistant" ? "model" : "user", parts });
@@ -159,6 +166,14 @@ function messagePart(part, assistant, sameProviderAndModel) {
             fail("Only text context is supported by this provider.");
         return { text };
     }
+    if (type === "image") {
+        const rawData = field(part, "data") ?? (isRecord(field(part, "source")) ? field(field(part, "source"), "data") : undefined);
+        if (typeof rawData !== "string" || !rawData)
+            fail("Invalid image data.");
+        const rawMime = field(part, "mimeType") ?? (isRecord(field(part, "source")) ? field(field(part, "source"), "media_type") : undefined);
+        const mimeType = typeof rawMime === "string" && rawMime ? rawMime : "image/jpeg";
+        return { inlineData: { mimeType, data: rawData } };
+    }
     fail("Only text context is supported by this provider.");
 }
 function isSameProviderAndModel(message, publicId) {

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) {
@@ -58,13 +58,17 @@ export function replayToolHistory(messages, serializePart, onDiagnostics, policy
     const finalize = () => {
         if (!pending)
             return;
-        appendTurn("user", pending.calls.map((call) => {
+        const turnParts = pending.calls.map((call) => {
             const actual = pending.results.get(call.id);
             if (actual)
                 return actual;
             recoveryCount.value += 1;
             return pending.observations ? observation(call, missingResultText()) : missingResult(call);
-        }));
+        });
+        if (pending.extraParts?.length) {
+            turnParts.push(...pending.extraParts);
+        }
+        appendTurn("user", turnParts);
         pending = undefined;
     };
     for (const message of messages) {
@@ -83,14 +87,34 @@ export function replayToolHistory(messages, serializePart, onDiagnostics, policy
                 history("PI_TOOL_RESULT_NAME_MISMATCH");
             if (value(current, "addedToolNames") !== undefined && dense(value(current, "addedToolNames")).length)
                 history("PI_TOOL_CALL_INVALID");
-            const text = dense(value(current, "content")).map((item) => {
+            const textParts = [];
+            const imageParts = [];
+            for (const item of dense(value(current, "content"))) {
                 const part = record(item);
-                if (value(part, "type") !== "text" || typeof value(part, "text") !== "string")
+                if (value(part, "type") === "text" && typeof value(part, "text") === "string") {
+                    textParts.push(value(part, "text"));
+                }
+                else if (value(part, "type") === "image") {
+                    const rawData = value(part, "data") ?? (isRecord(value(part, "source")) ? value(record(value(part, "source")), "data") : undefined);
+                    if (typeof rawData === "string" && rawData) {
+                        const rawMime = value(part, "mimeType") ?? (isRecord(value(part, "source")) ? value(record(value(part, "source")), "media_type") : undefined);
+                        const mimeType = typeof rawMime === "string" && rawMime ? rawMime : "image/jpeg";
+                        imageParts.push({ inlineData: { mimeType, data: rawData } });
+                    }
+                }
+                else {
                     history("PI_TOOL_RESULT_MEDIA_UNSUPPORTED");
-                return value(part, "text");
-            }).join("\n\n");
-            const response = value(current, "isError") === true ? { error: text } : { output: text };
-            pending.results.set(id, pending.observations ? observation(call, text) : { functionResponse: { name: call.name, response } });
+                }
+            }
+            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 } });
+            if (imageParts.length > 0) {
+                if (!pending.extraParts)
+                    pending.extraParts = [];
+                pending.extraParts.push(...imageParts);
+            }
             results.add(id);
             continue;
         }

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.5",
+  "version": "0.4.7",
   "description": "Pi adapter for Antigravity Guard",
   "type": "module",
   "license": "MIT",