@jsonstudio/llms 0.6.1423 → 0.6.1435

package/dist/conversion/compat/actions/antigravity-thought-signature-cache.d.ts

@@ -0,0 +1,3 @@
+import type { AdapterContext } from '../../hub/types/chat-envelope.js';
+import type { JsonObject } from '../../hub/types/json.js';
+export declare function cacheAntigravityThoughtSignatureFromGeminiResponse(payload: JsonObject, adapterContext?: AdapterContext): JsonObject;

package/dist/conversion/compat/actions/antigravity-thought-signature-cache.js

@@ -0,0 +1,60 @@
+import { cacheAntigravitySessionSignature, getAntigravityRequestSessionId } from '../antigravity-session-signature.js';
+function isRecord(value) {
+    return typeof value === 'object' && value !== null && !Array.isArray(value);
+}
+function shouldEnableForAdapter(adapterContext) {
+    if (!adapterContext) {
+        return false;
+    }
+    const protocol = typeof adapterContext.providerProtocol === 'string' ? adapterContext.providerProtocol.trim().toLowerCase() : '';
+    if (protocol !== 'gemini-chat') {
+        return false;
+    }
+    const providerIdOrKey = typeof adapterContext.providerId === 'string' ? adapterContext.providerId.trim().toLowerCase() : '';
+    const effectiveProviderId = providerIdOrKey.split('.')[0] ?? '';
+    return effectiveProviderId === 'antigravity';
+}
+export function cacheAntigravityThoughtSignatureFromGeminiResponse(payload, adapterContext) {
+    if (!shouldEnableForAdapter(adapterContext)) {
+        return payload;
+    }
+    const ctxAny = adapterContext;
+    const payloadAny = payload;
+    const keyCandidates = [
+        adapterContext?.requestId,
+        typeof ctxAny.clientRequestId === 'string' ? String(ctxAny.clientRequestId) : '',
+        typeof ctxAny.groupRequestId === 'string' ? String(ctxAny.groupRequestId) : '',
+        typeof payloadAny.request_id === 'string' ? String(payloadAny.request_id) : '',
+        typeof payloadAny.requestId === 'string' ? String(payloadAny.requestId) : ''
+    ].filter((k) => typeof k === 'string' && k.trim().length);
+    let sessionId = '';
+    for (const key of keyCandidates) {
+        const resolved = getAntigravityRequestSessionId(key);
+        if (resolved && resolved.trim().length) {
+            sessionId = resolved.trim();
+            break;
+        }
+    }
+    if (!sessionId) {
+        return payload;
+    }
+    const messageCount = 1;
+    const candidatesRaw = payload.candidates;
+    const candidates = Array.isArray(candidatesRaw) ? candidatesRaw : [];
+    for (const candidate of candidates) {
+        const content = isRecord(candidate.content) ? candidate.content : undefined;
+        const partsRaw = content?.parts;
+        const parts = Array.isArray(partsRaw) ? partsRaw : [];
+        for (const part of parts) {
+            if (!isRecord(part.functionCall)) {
+                continue;
+            }
+            const sig = typeof part.thoughtSignature === 'string' ? String(part.thoughtSignature) : '';
+            if (!sig.trim().length) {
+                continue;
+            }
+            cacheAntigravitySessionSignature(sessionId, sig.trim(), messageCount);
+        }
+    }
+    return payload;
+}

package/dist/conversion/compat/actions/gemini-cli-request.js

@@ -1,4 +1,4 @@
-import { extractAntigravityGeminiSessionId, getAntigravitySessionSignature, shouldTreatAsMissingThoughtSignature } from '../antigravity-session-signature.js';
+import { cacheAntigravityRequestSessionId, extractAntigravityGeminiSessionId, getAntigravitySessionSignature, shouldTreatAsMissingThoughtSignature } from '../antigravity-session-signature.js';
 const REQUEST_FIELDS = [
     'contents',
     'systemInstruction',
@@ -515,9 +515,29 @@ export function wrapGeminiCliRequest(payload, adapterContext) {
     normalizeToolDeclarations(requestNode);
     normalizeFunctionCallArgs(requestNode);
     const enableSignature = shouldEnableAntigravitySignature(adapterContext) ||
+        (typeof requestNode.userAgent === 'string' && requestNode.userAgent.trim().toLowerCase() === 'antigravity') ||
+        (typeof requestNode.requestId === 'string' && requestNode.requestId.trim().toLowerCase().startsWith('agent-')) ||
         (typeof root.userAgent === 'string' && root.userAgent.trim().toLowerCase() === 'antigravity') ||
         (typeof root.requestId === 'string' && root.requestId.trim().toLowerCase().startsWith('agent-'));
     if (enableSignature) {
+        // Antigravity-Manager alignment:
+        if (adapterContext) {
+            try {
+                const sessionId = extractAntigravityGeminiSessionId(requestNode);
+                const ctxAny = adapterContext;
+                const keys = [
+                    adapterContext.requestId,
+                    typeof ctxAny.clientRequestId === 'string' ? String(ctxAny.clientRequestId) : '',
+                    typeof ctxAny.groupRequestId === 'string' ? String(ctxAny.groupRequestId) : ''
+                ].filter((k) => typeof k === 'string' && k.trim().length);
+                for (const key of keys) {
+                    cacheAntigravityRequestSessionId(key, sessionId);
+                }
+            }
+            catch {
+                // best-effort only
+            }
+        }
         injectAntigravityThoughtSignature(requestNode);
     }
     // Cloud Code Assist request wrapper should not carry metadata/action/web_search/stream.

package/dist/conversion/compat/antigravity-session-signature.d.ts

@@ -1,4 +1,6 @@
 export declare const DUMMY_THOUGHT_SIGNATURE = "skip_thought_signature_validator";
+export declare function cacheAntigravityRequestSessionId(requestId: string, sessionId: string): void;
+export declare function getAntigravityRequestSessionId(requestId: string): string | undefined;
 /**
  * Antigravity-Manager alignment: derive a stable session fingerprint for Gemini native requests.
  * - sha256(first user text parts joined), if len>10 and no "<system-reminder>"

package/dist/conversion/compat/antigravity-session-signature.js

@@ -5,6 +5,7 @@ const SIGNATURE_TTL_MS = 2 * 60 * 60 * 1000;
 const MIN_SIGNATURE_LENGTH = 50;
 const SESSION_CACHE_LIMIT = 1000;
 const GLOBAL_SIGNATURE_CACHE_KEY = '__LLMSWITCH_ANTIGRAVITY_SESSION_SIGNATURE_CACHE__';
+const GLOBAL_REQUEST_SESSION_CACHE_KEY = '__LLMSWITCH_ANTIGRAVITY_REQUEST_SESSION_ID_CACHE__';
 function getGlobalSignatureCache() {
     const g = globalThis;
     const existing = g[GLOBAL_SIGNATURE_CACHE_KEY];
@@ -16,6 +17,17 @@ function getGlobalSignatureCache() {
     return created;
 }
 const sessionSignatures = getGlobalSignatureCache();
+function getGlobalRequestSessionCache() {
+    const g = globalThis;
+    const existing = g[GLOBAL_REQUEST_SESSION_CACHE_KEY];
+    if (existing instanceof Map) {
+        return existing;
+    }
+    const created = new Map();
+    g[GLOBAL_REQUEST_SESSION_CACHE_KEY] = created;
+    return created;
+}
+const requestSessionIds = getGlobalRequestSessionCache();
 function nowMs() {
     return Date.now();
 }
@@ -53,6 +65,9 @@ function sha256Hex(value) {
 function isExpired(entry, ts) {
     return ts - entry.timestamp > SIGNATURE_TTL_MS;
 }
+function isRequestSessionExpired(entry, ts) {
+    return ts - entry.timestamp > SIGNATURE_TTL_MS;
+}
 function pruneExpired() {
     const ts = nowMs();
     for (const [key, entry] of sessionSignatures.entries()) {
@@ -79,6 +94,43 @@ function ensureCacheLimit() {
         }
     }
 }
+export function cacheAntigravityRequestSessionId(requestId, sessionId) {
+    const rid = typeof requestId === 'string' ? requestId.trim() : '';
+    const sid = typeof sessionId === 'string' ? sessionId.trim() : '';
+    if (!rid || !sid) {
+        return;
+    }
+    const ts = nowMs();
+    requestSessionIds.set(rid, { sessionId: sid, timestamp: ts });
+    if (requestSessionIds.size <= SESSION_CACHE_LIMIT) {
+        return;
+    }
+    // Best-effort cleanup to avoid unbounded growth.
+    const entries = Array.from(requestSessionIds.entries()).sort((a, b) => a[1].timestamp - b[1].timestamp);
+    const overflow = requestSessionIds.size - SESSION_CACHE_LIMIT;
+    for (let i = 0; i < overflow; i++) {
+        const key = entries[i]?.[0];
+        if (key) {
+            requestSessionIds.delete(key);
+        }
+    }
+}
+export function getAntigravityRequestSessionId(requestId) {
+    const rid = typeof requestId === 'string' ? requestId.trim() : '';
+    if (!rid) {
+        return undefined;
+    }
+    const entry = requestSessionIds.get(rid);
+    if (!entry) {
+        return undefined;
+    }
+    const ts = nowMs();
+    if (isRequestSessionExpired(entry, ts)) {
+        requestSessionIds.delete(rid);
+        return undefined;
+    }
+    return entry.sessionId;
+}
 function findGeminiContentsNode(payload) {
     if (!isRecord(payload)) {
         return undefined;

package/dist/conversion/compat/profiles/chat-gemini.json

@@ -29,6 +29,8 @@
     ]
   },
   "response": {
-    "mappings": []
+    "mappings": [
+      { "action": "antigravity_thought_signature_cache" }
+    ]
   }
 }

package/dist/conversion/hub/pipeline/compat/compat-pipeline-executor.js

@@ -18,6 +18,7 @@ import { applyGlmImageContentTransform } from '../../../compat/actions/glm-image
 import { applyGlmVisionPromptTransform } from '../../../compat/actions/glm-vision-prompt.js';
 import { applyClaudeThinkingToolSchemaCompat } from '../../../compat/actions/claude-thinking-tools.js';
 import { wrapGeminiCliRequest } from '../../../compat/actions/gemini-cli-request.js';
+import { cacheAntigravityThoughtSignatureFromGeminiResponse } from '../../../compat/actions/antigravity-thought-signature-cache.js';
 const RATE_LIMIT_ERROR = 'ERR_COMPAT_RATE_LIMIT_DETECTED';
 const INTERNAL_STATE = Symbol('compat.internal_state');
 export function runRequestCompatPipeline(profileId, payload, options) {
@@ -206,6 +207,11 @@ function applyMapping(root, mapping, state) {
                 replaceRoot(root, wrapGeminiCliRequest(root, state.adapterContext));
             }
             break;
+        case 'antigravity_thought_signature_cache':
+            if (state.direction === 'response') {
+                replaceRoot(root, cacheAntigravityThoughtSignatureFromGeminiResponse(root, state.adapterContext));
+            }
+            break;
         case 'glm_image_content':
             if (state.direction === 'request') {
                 replaceRoot(root, applyGlmImageContentTransform(root));

package/dist/conversion/hub/pipeline/compat/compat-types.d.ts

@@ -118,6 +118,8 @@ export type MappingInstruction = {
     action: 'claude_thinking_tool_schema';
 } | {
     action: 'gemini_cli_request_wrap';
+} | {
+    action: 'antigravity_thought_signature_cache';
 };
 export type FilterInstruction = {
     action: 'rate_limit_text';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@jsonstudio/llms",
-  "version": "0.6.1423",
+  "version": "0.6.1435",
   "type": "module",
   "main": "dist/index.js",
   "module": "dist/index.js",