siclaw 0.1.2 → 0.1.4

package/dist/core/tool-call-id.js

@@ -0,0 +1,226 @@
+/**
+ * Tool call ID sanitization for provider compatibility.
+ *
+ * Some providers (e.g., Mistral, certain OpenAI-compatible APIs) require
+ * strictly alphanumeric tool call IDs. This module rewrites IDs in the
+ * conversation history while preserving assistant-toolResult pairing.
+ *
+ * Ported from OpenClaw's src/agents/tool-call-id.ts.
+ */
+import { createHash } from "node:crypto";
+import { extractToolCallsFromAssistant } from "./message-utils.js";
+const STRICT9_LEN = 9;
+/**
+ * Sanitize a tool call ID to be compatible with various providers.
+ *
+ * - "strict" mode: only [a-zA-Z0-9]
+ * - "strict9" mode: only [a-zA-Z0-9], length 9 (Mistral tool call requirement)
+ */
+export function sanitizeToolCallId(id, mode = "strict") {
+    if (!id || typeof id !== "string") {
+        if (mode === "strict9") {
+            return "defaultid";
+        }
+        return "defaulttoolid";
+    }
+    if (mode === "strict9") {
+        const alphanumericOnly = id.replace(/[^a-zA-Z0-9]/g, "");
+        if (alphanumericOnly.length >= STRICT9_LEN) {
+            return alphanumericOnly.slice(0, STRICT9_LEN);
+        }
+        if (alphanumericOnly.length > 0) {
+            return shortHash(alphanumericOnly, STRICT9_LEN);
+        }
+        return shortHash("sanitized", STRICT9_LEN);
+    }
+    // Some providers require strictly alphanumeric tool call IDs.
+    const alphanumericOnly = id.replace(/[^a-zA-Z0-9]/g, "");
+    return alphanumericOnly.length > 0 ? alphanumericOnly : "sanitizedtoolid";
+}
+export function isValidCloudCodeAssistToolId(id, mode = "strict") {
+    if (!id || typeof id !== "string") {
+        return false;
+    }
+    if (mode === "strict9") {
+        return /^[a-zA-Z0-9]{9}$/.test(id);
+    }
+    // Strictly alphanumeric for providers with tighter tool ID constraints
+    return /^[a-zA-Z0-9]+$/.test(id);
+}
+function shortHash(text, length = 8) {
+    return createHash("sha256").update(text).digest("hex").slice(0, length);
+}
+function makeUniqueToolId(params) {
+    if (params.mode === "strict9") {
+        const base = sanitizeToolCallId(params.id, params.mode);
+        const candidate = base.length >= STRICT9_LEN ? base.slice(0, STRICT9_LEN) : "";
+        if (candidate && !params.used.has(candidate)) {
+            return candidate;
+        }
+        for (let i = 0; i < 1000; i += 1) {
+            const hashed = shortHash(`${params.id}:${i}`, STRICT9_LEN);
+            if (!params.used.has(hashed)) {
+                return hashed;
+            }
+        }
+        return shortHash(`${params.id}:${Date.now()}`, STRICT9_LEN);
+    }
+    const MAX_LEN = 40;
+    const base = sanitizeToolCallId(params.id, params.mode).slice(0, MAX_LEN);
+    if (!params.used.has(base)) {
+        return base;
+    }
+    const hash = shortHash(params.id);
+    // Use separator based on mode: none for strict, underscore for non-strict variants
+    const separator = params.mode === "strict" ? "" : "_";
+    const maxBaseLen = MAX_LEN - separator.length - hash.length;
+    const clippedBase = base.length > maxBaseLen ? base.slice(0, maxBaseLen) : base;
+    const candidate = `${clippedBase}${separator}${hash}`;
+    if (!params.used.has(candidate)) {
+        return candidate;
+    }
+    for (let i = 2; i < 1000; i += 1) {
+        const suffix = params.mode === "strict" ? `x${i}` : `_${i}`;
+        const next = `${candidate.slice(0, MAX_LEN - suffix.length)}${suffix}`;
+        if (!params.used.has(next)) {
+            return next;
+        }
+    }
+    const ts = params.mode === "strict" ? `t${Date.now()}` : `_${Date.now()}`;
+    return `${candidate.slice(0, MAX_LEN - ts.length)}${ts}`;
+}
+function createOccurrenceAwareResolver(mode) {
+    const used = new Set();
+    const assistantOccurrences = new Map();
+    const orphanToolResultOccurrences = new Map();
+    const pendingByRawId = new Map();
+    const allocate = (seed) => {
+        const next = makeUniqueToolId({ id: seed, used, mode });
+        used.add(next);
+        return next;
+    };
+    const resolveAssistantId = (id) => {
+        const occurrence = (assistantOccurrences.get(id) ?? 0) + 1;
+        assistantOccurrences.set(id, occurrence);
+        const next = allocate(occurrence === 1 ? id : `${id}:${occurrence}`);
+        const pending = pendingByRawId.get(id);
+        if (pending) {
+            pending.push(next);
+        }
+        else {
+            pendingByRawId.set(id, [next]);
+        }
+        return next;
+    };
+    const resolveToolResultId = (id) => {
+        const pending = pendingByRawId.get(id);
+        if (pending && pending.length > 0) {
+            const next = pending.shift();
+            if (pending.length === 0) {
+                pendingByRawId.delete(id);
+            }
+            return next;
+        }
+        const occurrence = (orphanToolResultOccurrences.get(id) ?? 0) + 1;
+        orphanToolResultOccurrences.set(id, occurrence);
+        return allocate(`${id}:tool_result:${occurrence}`);
+    };
+    return { resolveAssistantId, resolveToolResultId };
+}
+function rewriteAssistantToolCallIds(params) {
+    const toolCalls = extractToolCallsFromAssistant(params.message);
+    if (toolCalls.length === 0) {
+        return params.message;
+    }
+    const content = params.message.content;
+    if (!Array.isArray(content)) {
+        return params.message;
+    }
+    let changed = false;
+    const next = content.map((block) => {
+        if (!block || typeof block !== "object") {
+            return block;
+        }
+        const rec = block;
+        const type = rec.type;
+        const id = rec.id;
+        if ((type !== "functionCall" && type !== "toolUse" && type !== "toolCall") ||
+            typeof id !== "string" ||
+            !id) {
+            return block;
+        }
+        const nextId = params.resolveId(id);
+        if (nextId === id) {
+            return block;
+        }
+        changed = true;
+        return { ...block, id: nextId };
+    });
+    if (!changed) {
+        return params.message;
+    }
+    return { ...params.message, content: next };
+}
+function rewriteToolResultIds(params) {
+    const toolCallId = typeof params.message.toolCallId === "string" && params.message.toolCallId
+        ? params.message.toolCallId
+        : undefined;
+    const toolUseId = params.message.toolUseId;
+    const toolUseIdStr = typeof toolUseId === "string" && toolUseId ? toolUseId : undefined;
+    const sharedRawId = toolCallId && toolUseIdStr && toolCallId === toolUseIdStr ? toolCallId : undefined;
+    const sharedResolvedId = sharedRawId ? params.resolveId(sharedRawId) : undefined;
+    const nextToolCallId = sharedResolvedId ?? (toolCallId ? params.resolveId(toolCallId) : undefined);
+    const nextToolUseId = sharedResolvedId ?? (toolUseIdStr ? params.resolveId(toolUseIdStr) : undefined);
+    if (nextToolCallId === toolCallId && nextToolUseId === toolUseIdStr) {
+        return params.message;
+    }
+    return {
+        ...params.message,
+        ...(nextToolCallId && { toolCallId: nextToolCallId }),
+        ...(nextToolUseId && { toolUseId: nextToolUseId }),
+    };
+}
+/**
+ * Sanitize tool call IDs for provider compatibility.
+ *
+ * Sanitization can introduce collisions, and some providers also reject raw
+ * duplicate tool-call IDs. Track assistant occurrences in-order so repeated
+ * raw IDs receive distinct rewritten IDs, while matching tool results consume
+ * the same rewritten IDs in encounter order.
+ *
+ * @param messages - The messages to sanitize
+ * @param mode - "strict" (alphanumeric only) or "strict9" (alphanumeric length 9)
+ */
+export function sanitizeToolCallIdsForCloudCodeAssist(messages, mode = "strict") {
+    const { resolveAssistantId, resolveToolResultId: resolveToolResultIdFn } = createOccurrenceAwareResolver(mode);
+    let changed = false;
+    const out = messages.map((msg) => {
+        if (!msg || typeof msg !== "object") {
+            return msg;
+        }
+        const role = msg.role;
+        if (role === "assistant") {
+            const next = rewriteAssistantToolCallIds({
+                message: msg,
+                resolveId: resolveAssistantId,
+            });
+            if (next !== msg) {
+                changed = true;
+            }
+            return next;
+        }
+        if (role === "toolResult") {
+            const next = rewriteToolResultIds({
+                message: msg,
+                resolveId: resolveToolResultIdFn,
+            });
+            if (next !== msg) {
+                changed = true;
+            }
+            return next;
+        }
+        return msg;
+    });
+    return changed ? out : messages;
+}
+//# sourceMappingURL=tool-call-id.js.map

package/dist/core/tool-call-id.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-call-id.js","sourceRoot":"","sources":["../../src/core/tool-call-id.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,EAAE,6BAA6B,EAAuB,MAAM,oBAAoB,CAAC;AAIxF,MAAM,WAAW,GAAG,CAAC,CAAC;AAEtB;;;;;GAKG;AACH,MAAM,UAAU,kBAAkB,CAAC,EAAU,EAAE,OAAuB,QAAQ;IAC5E,IAAI,CAAC,EAAE,IAAI,OAAO,EAAE,KAAK,QAAQ,EAAE,CAAC;QAClC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,OAAO,WAAW,CAAC;QACrB,CAAC;QACD,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,MAAM,gBAAgB,GAAG,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;QACzD,IAAI,gBAAgB,CAAC,MAAM,IAAI,WAAW,EAAE,CAAC;YAC3C,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC;QAChD,CAAC;QACD,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,OAAO,SAAS,CAAC,gBAAgB,EAAE,WAAW,CAAC,CAAC;QAClD,CAAC;QACD,OAAO,SAAS,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;IAC7C,CAAC;IAED,8DAA8D;IAC9D,MAAM,gBAAgB,GAAG,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;IACzD,OAAO,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,iBAAiB,CAAC;AAC5E,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,EAAU,EAAE,OAAuB,QAAQ;IACtF,IAAI,CAAC,EAAE,IAAI,OAAO,EAAE,KAAK,QAAQ,EAAE,CAAC;QAClC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,OAAO,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACrC,CAAC;IACD,uEAAuE;IACvE,OAAO,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,SAAS,CAAC,IAAY,EAAE,MAAM,GAAG,CAAC;IACzC,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;AAC1E,CAAC;AAED,SAAS,gBAAgB,CAAC,MAA+D;IACvF,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;QAC9B,MAAM,IAAI,GAAG,kBAAkB,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;QACxD,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,IAAI,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/E,IAAI,SAAS,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7C,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;YACjC,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,MAAM,CAAC,EAAE,IAAI,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;YAC3D,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC7B,OAAO,MAAM,CAAC;YAChB,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC,GAAG,MAAM,CAAC,EAAE,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,WAAW,CAAC,CAAC;IAC9D,CAAC;IAED,MAAM,OAAO,GAAG,EAAE,CAAC;IAEnB,MAAM,IAAI,GAAG,kBAAkB,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;IAC1E,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;QAC3B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,IAAI,GAAG,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAClC,mFAAmF;IACnF,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;IACtD,MAAM,UAAU,GAAG,OAAO,GAAG,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;IAC5D,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAChF,MAAM,SAAS,GAAG,GAAG,WAAW,GAAG,SAAS,GAAG,IAAI,EAAE,CAAC;IACtD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;QAChC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;QACjC,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;QAC5D,MAAM,IAAI,GAAG,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,EAAE,CAAC;QACvE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3B,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,MAAM,EAAE,GAAG,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;IAC1E,OAAO,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,GAAG,EAAE,CAAC,MAAM,CAAC,GAAG,EAAE,EAAE,CAAC;AAC3D,CAAC;AAED,SAAS,6BAA6B,CAAC,IAAoB;IAIzD,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAkB,CAAC;IACvD,MAAM,2BAA2B,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC9D,MAAM,cAAc,GAAG,IAAI,GAAG,EAAoB,CAAC;IAEnD,MAAM,QAAQ,GAAG,CAAC,IAAY,EAAU,EAAE;QACxC,MAAM,IAAI,GAAG,gBAAgB,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACf,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;IAEF,MAAM,kBAAkB,GAAG,CAAC,EAAU,EAAU,EAAE;QAChD,MAAM,UAAU,GAAG,CAAC,oBAAoB,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QAC3D,oBAAoB,CAAC,GAAG,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QACzC,MAAM,IAAI,GAAG,QAAQ,CAAC,UAAU,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,IAAI,UAAU,EAAE,CAAC,CAAC;QACrE,MAAM,OAAO,GAAG,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvC,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrB,CAAC;aAAM,CAAC;YACN,cAAc,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;QACjC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;IAEF,MAAM,mBAAmB,GAAG,CAAC,EAAU,EAAU,EAAE;QACjD,MAAM,OAAO,GAAG,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvC,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,EAAG,CAAC;YAC9B,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC5B,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,UAAU,GAAG,CAAC,2BAA2B,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QAClE,2BAA2B,CAAC,GAAG,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QAChD,OAAO,QAAQ,CAAC,GAAG,EAAE,gBAAgB,UAAU,EAAE,CAAC,CAAC;IACrD,CAAC,CAAC;IAEF,OAAO,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,CAAC;AACrD,CAAC;AAED,SAAS,2BAA2B,CAAC,MAGpC;IACC,MAAM,SAAS,GAAG,6BAA6B,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAChE,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO,MAAM,CAAC,OAAO,CAAC;IACxB,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;IACvC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,OAAO,MAAM,CAAC,OAAO,CAAC;IACxB,CAAC;IAED,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QACjC,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACxC,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,GAAG,GAAG,KAAyC,CAAC;QACtD,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;QACtB,MAAM,EAAE,GAAG,GAAG,CAAC,EAAE,CAAC;QAClB,IACE,CAAC,IAAI,KAAK,cAAc,IAAI,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,UAAU,CAAC;YACtE,OAAO,EAAE,KAAK,QAAQ;YACtB,CAAC,EAAE,EACH,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACpC,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAClB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,GAAG,IAAI,CAAC;QACf,OAAO,EAAE,GAAI,KAA4C,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,MAAM,CAAC,OAAO,CAAC;IACxB,CAAC;IACD,OAAO,EAAE,GAAG,MAAM,CAAC,OAAO,EAAE,OAAO,EAAE,IAAqC,EAAE,CAAC;AAC/E,CAAC;AAED,SAAS,oBAAoB,CAAC,MAG7B;IACC,MAAM,UAAU,GACd,OAAO,MAAM,CAAC,OAAO,CAAC,UAAU,KAAK,QAAQ,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU;QACxE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU;QAC3B,CAAC,CAAC,SAAS,CAAC;IAChB,MAAM,SAAS,GAAI,MAAM,CAAC,OAAmC,CAAC,SAAS,CAAC;IACxE,MAAM,YAAY,GAAG,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;IACxF,MAAM,WAAW,GACf,UAAU,IAAI,YAAY,IAAI,UAAU,KAAK,YAAY,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;IAErF,MAAM,gBAAgB,GAAG,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACjF,MAAM,cAAc,GAClB,gBAAgB,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAC9E,MAAM,aAAa,GACjB,gBAAgB,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAElF,IAAI,cAAc,KAAK,UAAU,IAAI,aAAa,KAAK,YAAY,EAAE,CAAC;QACpE,OAAO,MAAM,CAAC,OAAO,CAAC;IACxB,CAAC;IAED,OAAO;QACL,GAAG,MAAM,CAAC,OAAO;QACjB,GAAG,CAAC,cAAc,IAAI,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC;QACrD,GAAG,CAAC,aAAa,IAAI,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;KACF,CAAC;AACrD,CAAC;AAED;;;;;;;;;;GAUG;AACH,MAAM,UAAU,qCAAqC,CACnD,QAAwB,EACxB,OAAuB,QAAQ;IAE/B,MAAM,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,qBAAqB,EAAE,GACtE,6BAA6B,CAAC,IAAI,CAAC,CAAC;IAEtC,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QAC/B,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;YACpC,OAAO,GAAG,CAAC;QACb,CAAC;QACD,MAAM,IAAI,GAAI,GAA0B,CAAC,IAAI,CAAC;QAC9C,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;YACzB,MAAM,IAAI,GAAG,2BAA2B,CAAC;gBACvC,OAAO,EAAE,GAAmD;gBAC5D,SAAS,EAAE,kBAAkB;aAC9B,CAAC,CAAC;YACH,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACjB,OAAO,GAAG,IAAI,CAAC;YACjB,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;YAC1B,MAAM,IAAI,GAAG,oBAAoB,CAAC;gBAChC,OAAO,EAAE,GAAoD;gBAC7D,SAAS,EAAE,qBAAqB;aACjC,CAAC,CAAC;YACH,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACjB,OAAO,GAAG,IAAI,CAAC;YACjB,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC;AAClC,CAAC"}

package/dist/core/tool-call-repair.d.ts

@@ -0,0 +1,18 @@
+/**
+ * Tool call validation and repair for session history.
+ *
+ * Prevents 400 errors from OpenAI-compatible APIs when session history
+ * contains malformed tool calls (missing/empty id or name) due to
+ * interrupted streams or incomplete API responses.
+ *
+ * Ported from OpenClaw's session-transcript-repair.ts (simplified —
+ * no allowedToolNames, no sessions_spawn handling).
+ */
+import type { AgentMessage } from "@mariozechner/pi-agent-core";
+export type ToolCallInputRepairReport = {
+    messages: AgentMessage[];
+    droppedToolCalls: number;
+    droppedAssistantMessages: number;
+};
+export declare function repairToolCallInputs(messages: AgentMessage[]): ToolCallInputRepairReport;
+export declare function sanitizeToolCallInputs(messages: AgentMessage[]): AgentMessage[];

package/dist/core/tool-call-repair.js

@@ -0,0 +1,73 @@
+const TOOL_CALL_NAME_MAX_CHARS = 64;
+const TOOL_CALL_NAME_RE = /^[A-Za-z0-9_-]+$/;
+function isRawToolCallBlock(block) {
+    if (!block || typeof block !== "object")
+        return false;
+    const type = block.type;
+    return (typeof type === "string" &&
+        (type === "toolCall" || type === "toolUse" || type === "functionCall"));
+}
+function hasToolCallInput(block) {
+    const hasInput = "input" in block ? block.input !== undefined && block.input !== null : false;
+    const hasArguments = "arguments" in block ? block.arguments !== undefined && block.arguments !== null : false;
+    return hasInput || hasArguments;
+}
+function hasToolCallId(block) {
+    return typeof block.id === "string" && block.id.trim().length > 0;
+}
+function hasToolCallName(block) {
+    if (typeof block.name !== "string")
+        return false;
+    const trimmed = block.name.trim();
+    if (!trimmed)
+        return false;
+    if (trimmed.length > TOOL_CALL_NAME_MAX_CHARS || !TOOL_CALL_NAME_RE.test(trimmed))
+        return false;
+    return true;
+}
+export function repairToolCallInputs(messages) {
+    let droppedToolCalls = 0;
+    let droppedAssistantMessages = 0;
+    let changed = false;
+    const out = [];
+    for (const msg of messages) {
+        if (!msg || typeof msg !== "object") {
+            out.push(msg);
+            continue;
+        }
+        if (msg.role !== "assistant" || !Array.isArray(msg.content)) {
+            out.push(msg);
+            continue;
+        }
+        const nextContent = [];
+        let droppedInMessage = 0;
+        for (const block of msg.content) {
+            if (isRawToolCallBlock(block) &&
+                (!hasToolCallInput(block) || !hasToolCallId(block) || !hasToolCallName(block))) {
+                droppedToolCalls += 1;
+                droppedInMessage += 1;
+                changed = true;
+                continue;
+            }
+            nextContent.push(block);
+        }
+        if (droppedInMessage > 0) {
+            if (nextContent.length === 0) {
+                droppedAssistantMessages += 1;
+                continue;
+            }
+            out.push({ ...msg, content: nextContent });
+            continue;
+        }
+        out.push(msg);
+    }
+    return {
+        messages: changed ? out : messages,
+        droppedToolCalls,
+        droppedAssistantMessages,
+    };
+}
+export function sanitizeToolCallInputs(messages) {
+    return repairToolCallInputs(messages).messages;
+}
+//# sourceMappingURL=tool-call-repair.js.map

package/dist/core/tool-call-repair.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-call-repair.js","sourceRoot":"","sources":["../../src/core/tool-call-repair.ts"],"names":[],"mappings":"AAYA,MAAM,wBAAwB,GAAG,EAAE,CAAC;AACpC,MAAM,iBAAiB,GAAG,kBAAkB,CAAC;AAU7C,SAAS,kBAAkB,CAAC,KAAc;IACxC,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACtD,MAAM,IAAI,GAAI,KAA4B,CAAC,IAAI,CAAC;IAChD,OAAO,CACL,OAAO,IAAI,KAAK,QAAQ;QACxB,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,cAAc,CAAC,CACvE,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAuB;IAC/C,MAAM,QAAQ,GAAG,OAAO,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,KAAK,SAAS,IAAI,KAAK,CAAC,KAAK,KAAK,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC;IAC9F,MAAM,YAAY,GAChB,WAAW,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,KAAK,SAAS,IAAI,KAAK,CAAC,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC;IAC3F,OAAO,QAAQ,IAAI,YAAY,CAAC;AAClC,CAAC;AAED,SAAS,aAAa,CAAC,KAAuB;IAC5C,OAAO,OAAO,KAAK,CAAC,EAAE,KAAK,QAAQ,IAAI,KAAK,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;AACpE,CAAC;AAED,SAAS,eAAe,CAAC,KAAuB;IAC9C,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACjD,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;IAClC,IAAI,CAAC,OAAO;QAAE,OAAO,KAAK,CAAC;IAC3B,IAAI,OAAO,CAAC,MAAM,GAAG,wBAAwB,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC;QAAE,OAAO,KAAK,CAAC;IAChG,OAAO,IAAI,CAAC;AACd,CAAC;AAQD,MAAM,UAAU,oBAAoB,CAClC,QAAwB;IAExB,IAAI,gBAAgB,GAAG,CAAC,CAAC;IACzB,IAAI,wBAAwB,GAAG,CAAC,CAAC;IACjC,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,MAAM,GAAG,GAAmB,EAAE,CAAC;IAE/B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;YACpC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACd,SAAS;QACX,CAAC;QAED,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5D,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACd,SAAS;QACX,CAAC;QAED,MAAM,WAAW,GAAuB,EAAE,CAAC;QAC3C,IAAI,gBAAgB,GAAG,CAAC,CAAC;QAEzB,KAAK,MAAM,KAAK,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YAChC,IACE,kBAAkB,CAAC,KAAK,CAAC;gBACzB,CAAC,CAAC,gBAAgB,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC,EAC9E,CAAC;gBACD,gBAAgB,IAAI,CAAC,CAAC;gBACtB,gBAAgB,IAAI,CAAC,CAAC;gBACtB,OAAO,GAAG,IAAI,CAAC;gBACf,SAAS;YACX,CAAC;YACD,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;QAED,IAAI,gBAAgB,GAAG,CAAC,EAAE,CAAC;YACzB,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC7B,wBAAwB,IAAI,CAAC,CAAC;gBAC9B,SAAS;YACX,CAAC;YACD,GAAG,CAAC,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;YAC3C,SAAS;QACX,CAAC;QAED,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAChB,CAAC;IAED,OAAO;QACL,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ;QAClC,gBAAgB;QAChB,wBAAwB;KACzB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,QAAwB;IAC7D,OAAO,oBAAoB,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC;AACjD,CAAC"}

package/dist/core/tool-result-context-guard.d.ts

@@ -0,0 +1,36 @@
+/**
+ * Tool result context guard — preemptive context budget enforcement.
+ *
+ * Wraps the agent's `transformContext` to:
+ * 1. Truncate individual oversized tool results
+ * 2. Compact oldest tool results when total context exceeds budget
+ * 3. Throw preemptive overflow error when context is still too large
+ *
+ * Ported from OpenClaw's src/agents/pi-embedded-runner/tool-result-context-guard.ts
+ * with inlined char estimation (no separate module).
+ */
+import type { AgentMessage } from "@mariozechner/pi-agent-core";
+export declare const PREEMPTIVE_CONTEXT_OVERFLOW_MESSAGE = "Preemptive context overflow: estimated context size exceeds safe threshold during tool loop";
+type GuardableAgent = object;
+export declare function getToolResultText(msg: AgentMessage): string;
+export declare function estimateMessageChars(msg: AgentMessage): number;
+export declare function truncateTextToBudget(text: string, maxChars: number): string;
+export declare function enforceToolResultContextBudgetInPlace(params: {
+    messages: AgentMessage[];
+    contextBudgetChars: number;
+    maxSingleToolResultChars: number;
+}): void;
+/**
+ * Install a context guard that enforces tool result size budgets.
+ *
+ * Wraps `agent.transformContext` (private in pi-coding-agent, accessed via
+ * runtime cast) to truncate/compact tool results and trigger preemptive
+ * overflow when context is still too large.
+ *
+ * Returns an uninstall function that restores the original transformContext.
+ */
+export declare function installToolResultContextGuard(params: {
+    agent: GuardableAgent;
+    contextWindowTokens: number;
+}): () => void;
+export {};

package/dist/core/tool-result-context-guard.js

@@ -0,0 +1,272 @@
+// ── Constants ────────────────────────────────────────────────────────────
+const CHARS_PER_TOKEN = 4;
+const TOOL_RESULT_CHARS_PER_TOKEN = 2;
+const IMAGE_CHAR_ESTIMATE = 8_000;
+const CONTEXT_INPUT_HEADROOM_RATIO = 0.75;
+const SINGLE_TOOL_RESULT_CONTEXT_SHARE = 0.5;
+const PREEMPTIVE_OVERFLOW_RATIO = 0.9;
+const TRUNCATION_NOTICE = "[truncated: output exceeded context limit]";
+const TRUNCATION_SUFFIX = `\n${TRUNCATION_NOTICE}`;
+const COMPACTION_PLACEHOLDER = "[compacted: tool output removed to free context]";
+export const PREEMPTIVE_CONTEXT_OVERFLOW_MESSAGE = "Preemptive context overflow: estimated context size exceeds safe threshold during tool loop";
+// ── Char estimation (inlined from tool-result-char-estimator) ────────────
+function isTextBlock(block) {
+    return !!block && typeof block === "object" && block.type === "text";
+}
+function isImageBlock(block) {
+    return !!block && typeof block === "object" && block.type === "image";
+}
+function estimateUnknownChars(value) {
+    if (typeof value === "string")
+        return value.length;
+    if (value === undefined)
+        return 0;
+    try {
+        const serialized = JSON.stringify(value);
+        return typeof serialized === "string" ? serialized.length : 0;
+    }
+    catch {
+        return 256;
+    }
+}
+function isToolResultMessage(msg) {
+    const role = msg.role;
+    const type = msg.type;
+    return role === "toolResult" || role === "tool" || type === "toolResult";
+}
+function getToolResultContent(msg) {
+    if (!isToolResultMessage(msg))
+        return [];
+    const content = msg.content;
+    if (typeof content === "string")
+        return [{ type: "text", text: content }];
+    return Array.isArray(content) ? content : [];
+}
+function estimateContentBlockChars(content) {
+    let chars = 0;
+    for (const block of content) {
+        if (isTextBlock(block)) {
+            chars += block.text.length;
+        }
+        else if (isImageBlock(block)) {
+            chars += IMAGE_CHAR_ESTIMATE;
+        }
+        else {
+            chars += estimateUnknownChars(block);
+        }
+    }
+    return chars;
+}
+export function getToolResultText(msg) {
+    const content = getToolResultContent(msg);
+    const chunks = [];
+    for (const block of content) {
+        if (isTextBlock(block)) {
+            chunks.push(block.text);
+        }
+    }
+    return chunks.join("\n");
+}
+export function estimateMessageChars(msg) {
+    if (!msg || typeof msg !== "object")
+        return 0;
+    if (msg.role === "user") {
+        const content = msg.content;
+        if (typeof content === "string")
+            return content.length;
+        if (Array.isArray(content))
+            return estimateContentBlockChars(content);
+        return 0;
+    }
+    if (msg.role === "assistant") {
+        let chars = 0;
+        const content = msg.content;
+        if (Array.isArray(content)) {
+            for (const block of content) {
+                if (!block || typeof block !== "object")
+                    continue;
+                const typed = block;
+                if (typed.type === "text" && typeof typed.text === "string") {
+                    chars += typed.text.length;
+                }
+                else if (typed.type === "thinking" && typeof typed.thinking === "string") {
+                    chars += typed.thinking.length;
+                }
+                else if (typed.type === "toolCall") {
+                    try {
+                        chars += JSON.stringify(typed.arguments ?? {}).length;
+                    }
+                    catch {
+                        chars += 128;
+                    }
+                }
+                else {
+                    chars += estimateUnknownChars(block);
+                }
+            }
+        }
+        return chars;
+    }
+    if (isToolResultMessage(msg)) {
+        const content = getToolResultContent(msg);
+        let chars = estimateContentBlockChars(content);
+        const details = msg.details;
+        chars += estimateUnknownChars(details);
+        const weightedChars = Math.ceil(chars * (CHARS_PER_TOKEN / TOOL_RESULT_CHARS_PER_TOKEN));
+        return Math.max(chars, weightedChars);
+    }
+    return 256;
+}
+function createMessageCharEstimateCache() {
+    return new WeakMap();
+}
+function estimateMessageCharsCached(msg, cache) {
+    const hit = cache.get(msg);
+    if (hit !== undefined)
+        return hit;
+    const estimated = estimateMessageChars(msg);
+    cache.set(msg, estimated);
+    return estimated;
+}
+function estimateContextChars(messages, cache) {
+    return messages.reduce((sum, msg) => sum + estimateMessageCharsCached(msg, cache), 0);
+}
+function invalidateMessageCharsCacheEntry(cache, msg) {
+    cache.delete(msg);
+}
+// ── Truncation / compaction helpers ──────────────────────────────────────
+export function truncateTextToBudget(text, maxChars) {
+    if (text.length <= maxChars)
+        return text;
+    if (maxChars <= 0)
+        return TRUNCATION_NOTICE;
+    const bodyBudget = Math.max(0, maxChars - TRUNCATION_SUFFIX.length);
+    if (bodyBudget <= 0)
+        return TRUNCATION_NOTICE;
+    let cutPoint = bodyBudget;
+    const newline = text.lastIndexOf("\n", bodyBudget);
+    if (newline > bodyBudget * 0.7) {
+        cutPoint = newline;
+    }
+    return text.slice(0, cutPoint) + TRUNCATION_SUFFIX;
+}
+function replaceToolResultText(msg, text) {
+    const content = msg.content;
+    const replacementContent = typeof content === "string" || content === undefined ? text : [{ type: "text", text }];
+    const sourceRecord = msg;
+    const { details: _details, ...rest } = sourceRecord;
+    return {
+        ...rest,
+        content: replacementContent,
+    };
+}
+function truncateToolResultToChars(msg, maxChars, cache) {
+    if (!isToolResultMessage(msg))
+        return msg;
+    const estimatedChars = estimateMessageCharsCached(msg, cache);
+    if (estimatedChars <= maxChars)
+        return msg;
+    const rawText = getToolResultText(msg);
+    if (!rawText)
+        return replaceToolResultText(msg, TRUNCATION_NOTICE);
+    const truncatedText = truncateTextToBudget(rawText, maxChars);
+    return replaceToolResultText(msg, truncatedText);
+}
+function applyMessageMutationInPlace(target, source, cache) {
+    if (target === source)
+        return;
+    const targetRecord = target;
+    const sourceRecord = source;
+    for (const key of Object.keys(targetRecord)) {
+        if (!(key in sourceRecord)) {
+            delete targetRecord[key];
+        }
+    }
+    Object.assign(targetRecord, sourceRecord);
+    if (cache) {
+        invalidateMessageCharsCacheEntry(cache, target);
+    }
+}
+function compactExistingToolResultsInPlace(params) {
+    const { messages, charsNeeded, cache } = params;
+    if (charsNeeded <= 0)
+        return 0;
+    let reduced = 0;
+    for (let i = 0; i < messages.length; i++) {
+        const msg = messages[i];
+        if (!isToolResultMessage(msg))
+            continue;
+        const before = estimateMessageCharsCached(msg, cache);
+        if (before <= COMPACTION_PLACEHOLDER.length)
+            continue;
+        const compacted = replaceToolResultText(msg, COMPACTION_PLACEHOLDER);
+        applyMessageMutationInPlace(msg, compacted, cache);
+        const after = estimateMessageCharsCached(msg, cache);
+        if (after >= before)
+            continue;
+        reduced += before - after;
+        if (reduced >= charsNeeded)
+            break;
+    }
+    return reduced;
+}
+// ── Main enforcement function ────────────────────────────────────────────
+export function enforceToolResultContextBudgetInPlace(params) {
+    const { messages, contextBudgetChars, maxSingleToolResultChars } = params;
+    const estimateCache = createMessageCharEstimateCache();
+    // Ensure each tool result has an upper bound before considering total context usage.
+    for (const message of messages) {
+        if (!isToolResultMessage(message))
+            continue;
+        const truncated = truncateToolResultToChars(message, maxSingleToolResultChars, estimateCache);
+        applyMessageMutationInPlace(message, truncated, estimateCache);
+    }
+    let currentChars = estimateContextChars(messages, estimateCache);
+    if (currentChars <= contextBudgetChars)
+        return;
+    // Compact oldest tool outputs first until the context is back under budget.
+    compactExistingToolResultsInPlace({
+        messages,
+        charsNeeded: currentChars - contextBudgetChars,
+        cache: estimateCache,
+    });
+}
+// ── Public API ───────────────────────────────────────────────────────────
+/**
+ * Install a context guard that enforces tool result size budgets.
+ *
+ * Wraps `agent.transformContext` (private in pi-coding-agent, accessed via
+ * runtime cast) to truncate/compact tool results and trigger preemptive
+ * overflow when context is still too large.
+ *
+ * Returns an uninstall function that restores the original transformContext.
+ */
+export function installToolResultContextGuard(params) {
+    const contextWindowTokens = Math.max(1, Math.floor(params.contextWindowTokens));
+    const contextBudgetChars = Math.max(1_024, Math.floor(contextWindowTokens * CHARS_PER_TOKEN * CONTEXT_INPUT_HEADROOM_RATIO));
+    const maxSingleToolResultChars = Math.max(1_024, Math.floor(contextWindowTokens * TOOL_RESULT_CHARS_PER_TOKEN * SINGLE_TOOL_RESULT_CONTEXT_SHARE));
+    const preemptiveOverflowChars = Math.max(contextBudgetChars, Math.floor(contextWindowTokens * CHARS_PER_TOKEN * PREEMPTIVE_OVERFLOW_RATIO));
+    const mutableAgent = params.agent;
+    const originalTransformContext = mutableAgent.transformContext;
+    mutableAgent.transformContext = (async (messages, signal) => {
+        const transformed = originalTransformContext
+            ? await originalTransformContext.call(mutableAgent, messages, signal)
+            : messages;
+        const contextMessages = Array.isArray(transformed) ? transformed : messages;
+        enforceToolResultContextBudgetInPlace({
+            messages: contextMessages,
+            contextBudgetChars,
+            maxSingleToolResultChars,
+        });
+        // After tool-result compaction, check if context still exceeds the high-water mark.
+        const postEnforcementChars = estimateContextChars(contextMessages, createMessageCharEstimateCache());
+        if (postEnforcementChars > preemptiveOverflowChars) {
+            throw new Error(PREEMPTIVE_CONTEXT_OVERFLOW_MESSAGE);
+        }
+        return contextMessages;
+    });
+    return () => {
+        mutableAgent.transformContext = originalTransformContext;
+    };
+}
+//# sourceMappingURL=tool-result-context-guard.js.map

package/dist/core/tool-result-context-guard.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-result-context-guard.js","sourceRoot":"","sources":["../../src/core/tool-result-context-guard.ts"],"names":[],"mappings":"AAaA,4EAA4E;AAE5E,MAAM,eAAe,GAAG,CAAC,CAAC;AAC1B,MAAM,2BAA2B,GAAG,CAAC,CAAC;AACtC,MAAM,mBAAmB,GAAG,KAAK,CAAC;AAClC,MAAM,4BAA4B,GAAG,IAAI,CAAC;AAC1C,MAAM,gCAAgC,GAAG,GAAG,CAAC;AAC7C,MAAM,yBAAyB,GAAG,GAAG,CAAC;AAEtC,MAAM,iBAAiB,GAAG,4CAA4C,CAAC;AACvE,MAAM,iBAAiB,GAAG,KAAK,iBAAiB,EAAE,CAAC;AACnD,MAAM,sBAAsB,GAAG,kDAAkD,CAAC;AAElF,MAAM,CAAC,MAAM,mCAAmC,GAC9C,6FAA6F,CAAC;AAiBhG,4EAA4E;AAE5E,SAAS,WAAW,CAAC,KAAc;IACjC,OAAO,CAAC,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAK,KAA4B,CAAC,IAAI,KAAK,MAAM,CAAC;AAC/F,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAClC,OAAO,CAAC,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAK,KAA4B,CAAC,IAAI,KAAK,OAAO,CAAC;AAChG,CAAC;AAED,SAAS,oBAAoB,CAAC,KAAc;IAC1C,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC,MAAM,CAAC;IACnD,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,CAAC,CAAC;IAClC,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACzC,OAAO,OAAO,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAChE,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,GAAG,CAAC;IACb,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,GAAiB;IAC5C,MAAM,IAAI,GAAI,GAA0B,CAAC,IAAI,CAAC;IAC9C,MAAM,IAAI,GAAI,GAA0B,CAAC,IAAI,CAAC;IAC9C,OAAO,IAAI,KAAK,YAAY,IAAI,IAAI,KAAK,MAAM,IAAI,IAAI,KAAK,YAAY,CAAC;AAC3E,CAAC;AAED,SAAS,oBAAoB,CAAC,GAAiB;IAC7C,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC;QAAE,OAAO,EAAE,CAAC;IACzC,MAAM,OAAO,GAAI,GAA6B,CAAC,OAAO,CAAC;IACvD,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IAC1E,OAAO,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;AAC/C,CAAC;AAED,SAAS,yBAAyB,CAAC,OAAkB;IACnD,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,IAAI,WAAW,CAAC,KAAK,CAAC,EAAE,CAAC;YACvB,KAAK,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC;QAC7B,CAAC;aAAM,IAAI,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/B,KAAK,IAAI,mBAAmB,CAAC;QAC/B,CAAC;aAAM,CAAC;YACN,KAAK,IAAI,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,GAAiB;IACjD,MAAM,OAAO,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,IAAI,WAAW,CAAC,KAAK,CAAC,EAAE,CAAC;YACvB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,GAAiB;IACpD,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,OAAO,CAAC,CAAC;IAE9C,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QACxB,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;QAC5B,IAAI,OAAO,OAAO,KAAK,QAAQ;YAAE,OAAO,OAAO,CAAC,MAAM,CAAC;QACvD,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;YAAE,OAAO,yBAAyB,CAAC,OAAO,CAAC,CAAC;QACtE,OAAO,CAAC,CAAC;IACX,CAAC;IAED,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;QAC7B,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,MAAM,OAAO,GAAI,GAA6B,CAAC,OAAO,CAAC;QACvD,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;oBAAE,SAAS;gBAClD,MAAM,KAAK,GAAG,KAKb,CAAC;gBACF,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;oBAC5D,KAAK,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC;gBAC7B,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,OAAO,KAAK,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;oBAC3E,KAAK,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;gBACjC,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBACrC,IAAI,CAAC;wBACH,KAAK,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;oBACxD,CAAC;oBAAC,MAAM,CAAC;wBACP,KAAK,IAAI,GAAG,CAAC;oBACf,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,KAAK,IAAI,oBAAoB,CAAC,KAAK,CAAC,CAAC;gBACvC,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,mBAAmB,CAAC,GAAG,CAAC,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC;QAC1C,IAAI,KAAK,GAAG,yBAAyB,CAAC,OAAO,CAAC,CAAC;QAC/C,MAAM,OAAO,GAAI,GAA6B,CAAC,OAAO,CAAC;QACvD,KAAK,IAAI,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACvC,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,eAAe,GAAG,2BAA2B,CAAC,CAAC,CAAC;QACzF,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;IACxC,CAAC;IAED,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,8BAA8B;IACrC,OAAO,IAAI,OAAO,EAAwB,CAAC;AAC7C,CAAC;AAED,SAAS,0BAA0B,CAAC,GAAiB,EAAE,KAA+B;IACpF,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC3B,IAAI,GAAG,KAAK,SAAS;QAAE,OAAO,GAAG,CAAC;IAClC,MAAM,SAAS,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC;IAC5C,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;IAC1B,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,oBAAoB,CAAC,QAAwB,EAAE,KAA+B;IACrF,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,0BAA0B,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;AACxF,CAAC;AAED,SAAS,gCAAgC,CAAC,KAA+B,EAAE,GAAiB;IAC1F,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;AACpB,CAAC;AAED,4EAA4E;AAE5E,MAAM,UAAU,oBAAoB,CAAC,IAAY,EAAE,QAAgB;IACjE,IAAI,IAAI,CAAC,MAAM,IAAI,QAAQ;QAAE,OAAO,IAAI,CAAC;IACzC,IAAI,QAAQ,IAAI,CAAC;QAAE,OAAO,iBAAiB,CAAC;IAE5C,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;IACpE,IAAI,UAAU,IAAI,CAAC;QAAE,OAAO,iBAAiB,CAAC;IAE9C,IAAI,QAAQ,GAAG,UAAU,CAAC;IAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;IACnD,IAAI,OAAO,GAAG,UAAU,GAAG,GAAG,EAAE,CAAC;QAC/B,QAAQ,GAAG,OAAO,CAAC;IACrB,CAAC;IAED,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,GAAG,iBAAiB,CAAC;AACrD,CAAC;AAED,SAAS,qBAAqB,CAAC,GAAiB,EAAE,IAAY;IAC5D,MAAM,OAAO,GAAI,GAA6B,CAAC,OAAO,CAAC;IACvD,MAAM,kBAAkB,GACtB,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IAEzF,MAAM,YAAY,GAAG,GAAyC,CAAC;IAC/D,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,IAAI,EAAE,GAAG,YAAY,CAAC;IACpD,OAAO;QACL,GAAG,IAAI;QACP,OAAO,EAAE,kBAAkB;KACZ,CAAC;AACpB,CAAC;AAED,SAAS,yBAAyB,CAChC,GAAiB,EACjB,QAAgB,EAChB,KAA+B;IAE/B,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC;QAAE,OAAO,GAAG,CAAC;IAE1C,MAAM,cAAc,GAAG,0BAA0B,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IAC9D,IAAI,cAAc,IAAI,QAAQ;QAAE,OAAO,GAAG,CAAC;IAE3C,MAAM,OAAO,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC;IACvC,IAAI,CAAC,OAAO;QAAE,OAAO,qBAAqB,CAAC,GAAG,EAAE,iBAAiB,CAAC,CAAC;IAEnE,MAAM,aAAa,GAAG,oBAAoB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IAC9D,OAAO,qBAAqB,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;AACnD,CAAC;AAED,SAAS,2BAA2B,CAClC,MAAoB,EACpB,MAAoB,EACpB,KAAgC;IAEhC,IAAI,MAAM,KAAK,MAAM;QAAE,OAAO;IAE9B,MAAM,YAAY,GAAG,MAA4C,CAAC;IAClE,MAAM,YAAY,GAAG,MAA4C,CAAC;IAClE,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;QAC5C,IAAI,CAAC,CAAC,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC;YAC3B,OAAO,YAAY,CAAC,GAAG,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IACD,MAAM,CAAC,MAAM,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;IAC1C,IAAI,KAAK,EAAE,CAAC;QACV,gCAAgC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;AACH,CAAC;AAED,SAAS,iCAAiC,CAAC,MAI1C;IACC,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;IAChD,IAAI,WAAW,IAAI,CAAC;QAAE,OAAO,CAAC,CAAC;IAE/B,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QACxB,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC;YAAE,SAAS;QAExC,MAAM,MAAM,GAAG,0BAA0B,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QACtD,IAAI,MAAM,IAAI,sBAAsB,CAAC,MAAM;YAAE,SAAS;QAEtD,MAAM,SAAS,GAAG,qBAAqB,CAAC,GAAG,EAAE,sBAAsB,CAAC,CAAC;QACrE,2BAA2B,CAAC,GAAG,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QACnD,MAAM,KAAK,GAAG,0BAA0B,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QACrD,IAAI,KAAK,IAAI,MAAM;YAAE,SAAS;QAE9B,OAAO,IAAI,MAAM,GAAG,KAAK,CAAC;QAC1B,IAAI,OAAO,IAAI,WAAW;YAAE,MAAM;IACpC,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,4EAA4E;AAE5E,MAAM,UAAU,qCAAqC,CAAC,MAIrD;IACC,MAAM,EAAE,QAAQ,EAAE,kBAAkB,EAAE,wBAAwB,EAAE,GAAG,MAAM,CAAC;IAC1E,MAAM,aAAa,GAAG,8BAA8B,EAAE,CAAC;IAEvD,qFAAqF;IACrF,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC;YAAE,SAAS;QAC5C,MAAM,SAAS,GAAG,yBAAyB,CAAC,OAAO,EAAE,wBAAwB,EAAE,aAAa,CAAC,CAAC;QAC9F,2BAA2B,CAAC,OAAO,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IACjE,CAAC;IAED,IAAI,YAAY,GAAG,oBAAoB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;IACjE,IAAI,YAAY,IAAI,kBAAkB;QAAE,OAAO;IAE/C,4EAA4E;IAC5E,iCAAiC,CAAC;QAChC,QAAQ;QACR,WAAW,EAAE,YAAY,GAAG,kBAAkB;QAC9C,KAAK,EAAE,aAAa;KACrB,CAAC,CAAC;AACL,CAAC;AAED,4EAA4E;AAE5E;;;;;;;;GAQG;AACH,MAAM,UAAU,6BAA6B,CAAC,MAG7C;IACC,MAAM,mBAAmB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,CAAC;IAChF,MAAM,kBAAkB,GAAG,IAAI,CAAC,GAAG,CACjC,KAAK,EACL,IAAI,CAAC,KAAK,CAAC,mBAAmB,GAAG,eAAe,GAAG,4BAA4B,CAAC,CACjF,CAAC;IACF,MAAM,wBAAwB,GAAG,IAAI,CAAC,GAAG,CACvC,KAAK,EACL,IAAI,CAAC,KAAK,CACR,mBAAmB,GAAG,2BAA2B,GAAG,gCAAgC,CACrF,CACF,CAAC;IACF,MAAM,uBAAuB,GAAG,IAAI,CAAC,GAAG,CACtC,kBAAkB,EAClB,IAAI,CAAC,KAAK,CAAC,mBAAmB,GAAG,eAAe,GAAG,yBAAyB,CAAC,CAC9E,CAAC;IAEF,MAAM,YAAY,GAAG,MAAM,CAAC,KAA6B,CAAC;IAC1D,MAAM,wBAAwB,GAAG,YAAY,CAAC,gBAAgB,CAAC;IAE/D,YAAY,CAAC,gBAAgB,GAAG,CAAC,KAAK,EAAE,QAAwB,EAAE,MAAmB,EAAE,EAAE;QACvF,MAAM,WAAW,GAAG,wBAAwB;YAC1C,CAAC,CAAC,MAAM,wBAAwB,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE,MAAM,CAAC;YACrE,CAAC,CAAC,QAAQ,CAAC;QAEb,MAAM,eAAe,GAAG,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC;QAC5E,qCAAqC,CAAC;YACpC,QAAQ,EAAE,eAAe;YACzB,kBAAkB;YAClB,wBAAwB;SACzB,CAAC,CAAC;QAEH,oFAAoF;QACpF,MAAM,oBAAoB,GAAG,oBAAoB,CAC/C,eAAe,EACf,8BAA8B,EAAE,CACjC,CAAC;QACF,IAAI,oBAAoB,GAAG,uBAAuB,EAAE,CAAC;YACnD,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QAED,OAAO,eAAe,CAAC;IACzB,CAAC,CAAuB,CAAC;IAEzB,OAAO,GAAG,EAAE;QACV,YAAY,CAAC,gBAAgB,GAAG,wBAAwB,CAAC;IAC3D,CAAC,CAAC;AACJ,CAAC"}