@benjamolina/pi-antigravity-guard 0.4.8 → 0.4.10

package/dist/context.d.ts

@@ -26,6 +26,7 @@ export interface GenerationRequest {
     request: {
         contents: Content[];
         systemInstruction?: {
+            role?: "user";
             parts: Part[];
         };
         tools?: {
@@ -47,6 +48,8 @@ export interface GenerationRequest {
                 includeThoughts: boolean;
             };
         };
+        sessionId?: string;
+        labels?: Record<string, string>;
     };
     requestType: "agent";
     userAgent: "antigravity";

package/dist/context.js

@@ -1,5 +1,6 @@
 import { CLAUDE_CUSTOM_PARAMETERS_PROFILE, getCatalogEntry, resolveGenerationSelection } from "./catalog.js";
 import { normalizeCustomToolSchema } from "./tool-schema.js";
+import { buildAntigravityEnvelope } from "./envelope.js";
 import { hasToolContext, prepareToolContext, replayToolHistory } from "./tool-context.js";
 const MAX_TEXT_BYTES = 8 * 1024 * 1024;
 const PROVIDER = "antigravity-guard";
@@ -36,6 +37,13 @@ export function serializeContext(input, injectedSelection, onDiagnostics) {
     } : undefined;
     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;
+    const envelope = buildAntigravityEnvelope({
+        wireModelId: selection.route.wireModel,
+        family: entry.response.family,
+        contentsCount: contents.length,
+        messages: isDenseArray(messages),
+        injectedRequestId: input.requestId,
+    });
     return { ...text, request: {
             contents,
             ...(systemInstruction ? { systemInstruction } : {}),
@@ -47,10 +55,16 @@ export function serializeContext(input, injectedSelection, onDiagnostics) {
                                 ? { parameters: normalizeCustomToolSchema(parameters) }
                                 : { parametersJsonSchema: parameters }),
                         })) }],
-                ...(prepared.mode === "NONE" ? { toolConfig: { functionCallingConfig: { mode: "NONE" } } } : {}),
+                ...(prepared.mode === "NONE"
+                    ? { toolConfig: { functionCallingConfig: { mode: "NONE" } } }
+                    : (entry.response.family === "claude" || entry.response.family === "gpt-oss"
+                        ? { toolConfig: { functionCallingConfig: { mode: "AUTO" } } }
+                        : {})),
             } : {}),
             generationConfig,
-        } };
+            sessionId: envelope.sessionId,
+            labels: envelope.labels,
+        }, requestId: envelope.requestId };
 }
 export function serializeTextContext(input) {
     try {
@@ -94,9 +108,16 @@ export function serializeTextContext(input) {
         }
         if (!contents.length)
             fail("A text conversation is required.");
-        const systemInstruction = systemPrompt === undefined ? undefined : { parts: [{ text: systemPrompt }] };
+        const systemInstruction = systemPrompt === undefined ? undefined : { role: "user", parts: [{ text: systemPrompt }] };
         const temperature = option(options, "temperature");
         const maxTokens = option(options, "maxTokens");
+        const envelope = buildAntigravityEnvelope({
+            wireModelId: route.wireModel,
+            family: entry.response.family,
+            contentsCount: contents.length,
+            messages: isDenseArray(field(context, "messages", true)),
+            injectedRequestId: requestId,
+        });
         return { project, model: route.wireModel, request: { contents, ...(systemInstruction ? { systemInstruction } : {}), generationConfig: {
                     temperature: typeof temperature === "number" ? temperature : 1,
                     maxOutputTokens: resolveOutputTokens(maxTokens, entry.descriptor.maxTokens, route.thinking),
@@ -104,7 +125,7 @@ export function serializeTextContext(input) {
                         const thinkingConfig = serializeThinkingConfig(route.thinking);
                         return thinkingConfig ? { thinkingConfig } : {};
                     })(),
-                } }, requestType: "agent", userAgent: "antigravity", requestId };
+                }, sessionId: envelope.sessionId, labels: envelope.labels }, requestType: "agent", userAgent: "antigravity", requestId: envelope.requestId };
     }
     catch (error) {
         if (error instanceof ContextSerializationError)
@@ -175,7 +196,8 @@ function messagePart(part, assistant, sameProviderAndModel) {
             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 } };
+        const cleanedData = rawData.replace(/^data:[^;]+;base64,/s, "").trim();
+        return { inlineData: { mimeType, data: cleanedData } };
     }
     fail("Only text context is supported by this provider.");
 }

package/dist/envelope.d.ts

@@ -0,0 +1,20 @@
+export declare const ANTIGRAVITY_MODEL_ENUM: Readonly<Record<string, string>>;
+export declare function stableUuid(seed: string): string;
+export declare function resolveSessionTrajectory(messages?: readonly unknown[]): {
+    conversationId: string;
+    trajectoryId: string;
+};
+export declare function clearSessionTrajectoryMap(): void;
+export interface AntigravityEnvelope {
+    readonly requestId: string;
+    readonly sessionId: string;
+    readonly labels: Record<string, string>;
+}
+export declare function buildAntigravityEnvelope(options: {
+    wireModelId: string;
+    family: "gemini" | "claude" | "gpt-oss";
+    contentsCount: number;
+    messages?: readonly unknown[];
+    explicitSessionId?: string;
+    injectedRequestId?: string;
+}): AntigravityEnvelope;

package/dist/envelope.js

@@ -0,0 +1,112 @@
+import { createHash } from "node:crypto";
+export const ANTIGRAVITY_MODEL_ENUM = Object.freeze({
+    // Gemini 3.8 Flash
+    "gemini-3.8-flash": "MODEL_PLACEHOLDER_M318",
+    "gemini-3.8-flash-high": "MODEL_PLACEHOLDER_M318",
+    "gemini-3.8-flash-medium": "MODEL_PLACEHOLDER_M319",
+    "gemini-3.8-flash-low": "MODEL_PLACEHOLDER_M320",
+    "gemini-3.8-flash-tiered": "MODEL_PLACEHOLDER_M322",
+    // Gemini 3.7 Flash
+    "gemini-3.7-flash": "MODEL_PLACEHOLDER_M298",
+    "gemini-3.7-flash-high": "MODEL_PLACEHOLDER_M298",
+    "gemini-3.7-flash-medium": "MODEL_PLACEHOLDER_M299",
+    "gemini-3.7-flash-low": "MODEL_PLACEHOLDER_M300",
+    "gemini-3.7-flash-tiered": "MODEL_PLACEHOLDER_M301",
+    // Gemini 3.6 Flash
+    "gemini-3.6-flash": "MODEL_PLACEHOLDER_M71",
+    "gemini-3.6-flash-high": "MODEL_PLACEHOLDER_M71",
+    "gemini-3.6-flash-medium": "MODEL_PLACEHOLDER_M72",
+    "gemini-3.6-flash-low": "MODEL_PLACEHOLDER_M73",
+    "gemini-3.6-flash-tiered": "MODEL_PLACEHOLDER_M196",
+    // Gemini 3.5 Flash
+    "gemini-3.5-flash": "MODEL_PLACEHOLDER_M20",
+    "gemini-3.5-flash-extra-low": "MODEL_PLACEHOLDER_M187",
+    "gemini-3.5-flash-low": "MODEL_PLACEHOLDER_M20",
+    "gemini-3-flash-agent": "MODEL_PLACEHOLDER_M84",
+    // Gemini 3.1 Pro
+    "gemini-3.1-pro": "MODEL_PLACEHOLDER_M36",
+    "gemini-3.1-pro-low": "MODEL_PLACEHOLDER_M36",
+    "gemini-3.1-pro-high": "MODEL_PLACEHOLDER_M37",
+    "gemini-pro-agent": "MODEL_PLACEHOLDER_M16",
+    // Claude
+    "claude-sonnet-4-6": "MODEL_PLACEHOLDER_M35",
+    "claude-opus-4-6": "MODEL_PLACEHOLDER_M26",
+    "claude-opus-4-6-thinking": "MODEL_PLACEHOLDER_M26",
+    // GPT-OSS
+    "gpt-oss-120b": "MODEL_OPENAI_GPT_OSS_120B_MEDIUM",
+    "gpt-oss-120b-medium": "MODEL_OPENAI_GPT_OSS_120B_MEDIUM",
+});
+export function stableUuid(seed) {
+    const bytes = createHash("sha1").update(seed).digest().subarray(0, 16);
+    bytes[6] = (bytes[6] & 0x0f) | 0x50;
+    bytes[8] = (bytes[8] & 0x3f) | 0x80;
+    const hex = [...bytes].map((b) => b.toString(16).padStart(2, "0")).join("");
+    return `${hex.slice(0, 8)}-${hex.slice(8, 12)}-${hex.slice(12, 16)}-${hex.slice(16, 20)}-${hex.slice(20)}`;
+}
+const sessionTrajectoryMap = new Map();
+export function resolveSessionTrajectory(messages) {
+    const firstMsg = Array.isArray(messages) ? messages[0] : undefined;
+    if (!isRecord(firstMsg)) {
+        return { conversationId: crypto.randomUUID(), trajectoryId: crypto.randomUUID() };
+    }
+    const role = typeof firstMsg.role === "string" ? firstMsg.role : "user";
+    const timestamp = typeof firstMsg.timestamp === "number" ? String(firstMsg.timestamp) : "";
+    const rawContent = firstMsg.content;
+    const contentSeed = typeof rawContent === "string"
+        ? rawContent.slice(0, 64)
+        : Array.isArray(rawContent)
+            ? JSON.stringify(rawContent[0] ?? "").slice(0, 64)
+            : "";
+    const seed = `${role}:${timestamp}:${contentSeed}`;
+    let entry = sessionTrajectoryMap.get(seed);
+    if (!entry) {
+        entry = {
+            conversationId: stableUuid(`antigravity:conv:${seed}`),
+            trajectoryId: stableUuid(`antigravity:traj:${seed}`),
+        };
+        sessionTrajectoryMap.set(seed, entry);
+        if (sessionTrajectoryMap.size > 64) {
+            const oldestKey = sessionTrajectoryMap.keys().next().value;
+            if (oldestKey !== undefined)
+                sessionTrajectoryMap.delete(oldestKey);
+        }
+    }
+    return entry;
+}
+export function clearSessionTrajectoryMap() {
+    sessionTrajectoryMap.clear();
+}
+const UUID_AGENT_REGEX = /^agent-[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i;
+export function buildAntigravityEnvelope(options) {
+    const isClaude = options.family === "claude";
+    const isNonGemini = options.family === "claude" || options.family === "gpt-oss";
+    const step = Math.max(1, options.contentsCount);
+    const lastStepIndex = String(Math.max(0, step - 1));
+    const assistantMsgCount = Array.isArray(options.messages)
+        ? options.messages.filter((m) => isRecord(m) && m.role === "assistant" && m.stopReason !== "error" && m.stopReason !== "aborted").length
+        : 0;
+    const requestIndex = Math.max(0, assistantMsgCount);
+    const { conversationId, trajectoryId } = resolveSessionTrajectory(options.messages);
+    const sessionHash = createHash("sha1").update(`session:${trajectoryId}`).digest().subarray(0, 8);
+    const generatedSessionId = String(new DataView(sessionHash.buffer, sessionHash.byteOffset, 8).getBigInt64(0, true));
+    const sessionId = options.explicitSessionId || generatedSessionId;
+    const labels = {
+        last_step_index: lastStepIndex,
+        request_id: `${trajectoryId}-${requestIndex}`,
+        trajectory_id: trajectoryId,
+        used_claude: isClaude ? "true" : "false",
+        used_claude_conservative: isClaude ? "true" : "false",
+        used_non_gemini_model: isNonGemini ? "true" : "false",
+    };
+    const modelEnum = ANTIGRAVITY_MODEL_ENUM[options.wireModelId];
+    if (modelEnum) {
+        labels.model_enum = modelEnum;
+    }
+    const requestId = options.injectedRequestId && !UUID_AGENT_REGEX.test(options.injectedRequestId)
+        ? options.injectedRequestId
+        : `agent/${conversationId}/${trajectoryId}/${step}`;
+    return { requestId, sessionId, labels };
+}
+function isRecord(value) {
+    return typeof value === "object" && value !== null;
+}

package/dist/tool-context.js

@@ -105,7 +105,8 @@ export function replayToolHistory(messages, serializePart, onDiagnostics, policy
                     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 } });
+                        const cleanedData = rawData.replace(/^data:[^;]+;base64,/s, "").trim();
+                        imageParts.push({ inlineData: { mimeType, data: cleanedData } });
                     }
                 }
                 else {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@benjamolina/pi-antigravity-guard",
-  "version": "0.4.8",
+  "version": "0.4.10",
   "description": "Pi adapter for Antigravity Guard",
   "type": "module",
   "license": "MIT",