@caupulican/pi-ai 0.81.11 → 0.81.13

package/dist/utils/tool-repair/replay.js

@@ -0,0 +1,200 @@
+import { Compile } from "typebox/compile";
+import { analyzeToolArgumentErrors } from "./analyzer.js";
+import { TOOL_REPAIR_MODE_NAMES } from "./registry.js";
+import { repairToolArguments } from "./repairer.js";
+function splitReplayPath(path) {
+    return path === "root" ? [] : path.split(".").filter((segment) => segment.length > 0);
+}
+function schemaForExpectedType(expectedType) {
+    if (!expectedType || expectedType === "unknown")
+        return { type: "null" };
+    if (expectedType === "enum")
+        return { enum: ["expected"] };
+    const types = expectedType.split("|").filter((type) => type.length > 0);
+    return { type: types.length <= 1 ? (types[0] ?? "null") : types };
+}
+function isNumericSegment(segment) {
+    return /^\d+$/.test(segment);
+}
+function ensureObjectProperty(schema, key) {
+    schema.type = "object";
+    schema.properties ??= {};
+    schema.required ??= [];
+    if (!schema.required.includes(key))
+        schema.required.push(key);
+    schema.properties[key] ??= { type: "object", properties: {}, required: [] };
+    return schema.properties[key];
+}
+function ensureArrayItems(schema) {
+    schema.type = "array";
+    schema.items ??= { type: "object", properties: {}, required: [] };
+    return schema.items;
+}
+function setSchemaAtPath(root, path, leaf) {
+    if (path.length === 0) {
+        Object.assign(root, leaf);
+        return;
+    }
+    let cursor = root;
+    for (let index = 0; index < path.length; index++) {
+        const segment = path[index];
+        const isLast = index === path.length - 1;
+        if (isNumericSegment(segment)) {
+            cursor = isLast ? Object.assign(ensureArrayItems(cursor), leaf) : ensureArrayItems(cursor);
+            continue;
+        }
+        if (isLast) {
+            cursor.type = "object";
+            cursor.properties ??= {};
+            cursor.required ??= [];
+            if (!cursor.required.includes(segment))
+                cursor.required.push(segment);
+            cursor.properties[segment] = leaf;
+            continue;
+        }
+        const next = path[index + 1];
+        cursor = isNumericSegment(next)
+            ? ensureArrayItems(ensureObjectProperty(cursor, segment))
+            : ensureObjectProperty(cursor, segment);
+    }
+}
+function sampleValue(receivedType, expectedType) {
+    if (receivedType === "missing")
+        return undefined;
+    if (receivedType === "null")
+        return null;
+    if (receivedType === "array")
+        return [];
+    if (receivedType === "object")
+        return {};
+    if (receivedType === "boolean")
+        return true;
+    if (receivedType === "number" || receivedType === "integer")
+        return 1;
+    if (receivedType === "string") {
+        if (expectedType?.includes("number") || expectedType?.includes("integer"))
+            return "not-a-number";
+        if (expectedType?.includes("boolean"))
+            return "not-a-boolean";
+        if (expectedType?.includes("array") || expectedType?.includes("object"))
+            return "not-json";
+        return "not-a-valid-replay-value";
+    }
+    return {};
+}
+function ensureArgsObject(target, key) {
+    const current = target[key];
+    if (typeof current === "object" && current !== null && !Array.isArray(current))
+        return current;
+    const next = {};
+    target[key] = next;
+    return next;
+}
+function ensureArgsArray(target, key) {
+    const current = target[key];
+    if (Array.isArray(current))
+        return current;
+    const next = [];
+    target[key] = next;
+    return next;
+}
+function setArgumentAtPath(root, path, value) {
+    if (value === undefined)
+        return;
+    if (path.length === 0)
+        return;
+    let objectCursor = root;
+    let arrayCursor;
+    for (let index = 0; index < path.length; index++) {
+        const segment = path[index];
+        const isLast = index === path.length - 1;
+        if (arrayCursor) {
+            const arrayIndex = Number(segment);
+            if (!Number.isInteger(arrayIndex))
+                return;
+            if (isLast) {
+                arrayCursor[arrayIndex] = value;
+                return;
+            }
+            const next = path[index + 1];
+            if (isNumericSegment(next)) {
+                const existing = arrayCursor[arrayIndex];
+                const nextArray = Array.isArray(existing) ? existing : [];
+                arrayCursor[arrayIndex] = nextArray;
+                arrayCursor = nextArray;
+                continue;
+            }
+            const existing = arrayCursor[arrayIndex];
+            const nextObject = typeof existing === "object" && existing !== null && !Array.isArray(existing)
+                ? existing
+                : {};
+            arrayCursor[arrayIndex] = nextObject;
+            objectCursor = nextObject;
+            arrayCursor = undefined;
+            continue;
+        }
+        if (isLast) {
+            objectCursor[segment] = value;
+            return;
+        }
+        const next = path[index + 1];
+        if (isNumericSegment(next)) {
+            arrayCursor = ensureArgsArray(objectCursor, segment);
+        }
+        else {
+            objectCursor = ensureArgsObject(objectCursor, segment);
+        }
+    }
+}
+function buildFixture(record) {
+    const parameters = { type: "object", properties: {}, required: [] };
+    const args = {};
+    for (const shape of record.shape) {
+        const path = splitReplayPath(shape.path);
+        setSchemaAtPath(parameters, path, schemaForExpectedType(shape.expectedType));
+        setArgumentAtPath(args, path, sampleValue(shape.receivedType, shape.expectedType));
+    }
+    return { tool: record.tool, parameters: parameters, arguments: args };
+}
+function validationErrorsForFixture(fixture) {
+    const validator = Compile(fixture.parameters);
+    return [...validator.Errors(fixture.arguments)].map((error) => ({
+        instancePath: error.instancePath,
+        keyword: error.keyword,
+        message: error.message,
+    }));
+}
+const KNOWN_FAILURE_MODES = new Set([...TOOL_REPAIR_MODE_NAMES, "other"]);
+function normalizedFailureModes(modes) {
+    const unique = [...new Set(modes ?? [])]
+        .map((mode) => (KNOWN_FAILURE_MODES.has(mode) ? mode : "other"))
+        .sort();
+    return unique.length > 0 ? unique : ["other"];
+}
+export function isToolRepairCorpusRecord(value) {
+    if (typeof value !== "object" || value === null || Array.isArray(value))
+        return false;
+    const record = value;
+    return record.kind === "tool_validation" && typeof record.tool === "string" && Array.isArray(record.shape);
+}
+export function replayToolRepairCorpus(records) {
+    return records.map((record, index) => {
+        const fixture = buildFixture(record);
+        const errors = validationErrorsForFixture(fixture);
+        const analyzerModes = analyzeToolArgumentErrors(fixture.tool, fixture.parameters, fixture.arguments, errors).flatMap((issue) => issue.modes);
+        const classifiedModes = analyzerModes.length > 0 ? normalizedFailureModes(analyzerModes) : normalizedFailureModes(record.failureModes);
+        const validator = Compile(fixture.parameters);
+        const repaired = repairToolArguments(fixture.tool, fixture.parameters, fixture.arguments, errors, (candidate) => validator.Check(candidate));
+        return {
+            record: index + 1,
+            provider: record.provider,
+            modelId: record.modelId,
+            tool: record.tool,
+            classifiedModes,
+            outcome: repaired ? "would-repair" : "would-bounce",
+            repairsApplied: repaired?.repairsApplied ?? [],
+            fixture,
+        };
+    });
+}
+//# sourceMappingURL=replay.js.map

package/dist/utils/tool-repair/replay.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"replay.js","sourceRoot":"","sources":["../../../src/utils/tool-repair/replay.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAC;AAE1C,OAAO,EAAE,yBAAyB,EAA4B,MAAM,eAAe,CAAC;AACpF,OAAO,EAAE,sBAAsB,EAA2D,MAAM,eAAe,CAAC;AAChH,OAAO,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AA6CpD,SAAS,eAAe,CAAC,IAAY,EAAY;IAChD,OAAO,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AAAA,CACtF;AAED,SAAS,qBAAqB,CAAC,YAAgC,EAA0B;IACxF,IAAI,CAAC,YAAY,IAAI,YAAY,KAAK,SAAS;QAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;IACzE,IAAI,YAAY,KAAK,MAAM;QAAE,OAAO,EAAE,IAAI,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC;IAC3D,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACxE,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC;AAAA,CAClE;AAED,SAAS,gBAAgB,CAAC,OAAe,EAAW;IACnD,OAAO,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;AAAA,CAC7B;AAED,SAAS,oBAAoB,CAAC,MAA8B,EAAE,GAAW,EAA0B;IAClG,MAAM,CAAC,IAAI,GAAG,QAAQ,CAAC;IACvB,MAAM,CAAC,UAAU,KAAK,EAAE,CAAC;IACzB,MAAM,CAAC,QAAQ,KAAK,EAAE,CAAC;IACvB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC;QAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC9D,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;IAC5E,OAAO,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;AAAA,CAC9B;AAED,SAAS,gBAAgB,CAAC,MAA8B,EAA0B;IACjF,MAAM,CAAC,IAAI,GAAG,OAAO,CAAC;IACtB,MAAM,CAAC,KAAK,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;IAClE,OAAO,MAAM,CAAC,KAAK,CAAC;AAAA,CACpB;AAED,SAAS,eAAe,CAAC,IAA4B,EAAE,IAAuB,EAAE,IAA4B,EAAQ;IACnH,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAC1B,OAAO;IACR,CAAC;IAED,IAAI,MAAM,GAAG,IAAI,CAAC;IAClB,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC;QAClD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,KAAK,KAAK,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;QACzC,IAAI,gBAAgB,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/B,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC3F,SAAS;QACV,CAAC;QACD,IAAI,MAAM,EAAE,CAAC;YACZ,MAAM,CAAC,IAAI,GAAG,QAAQ,CAAC;YACvB,MAAM,CAAC,UAAU,KAAK,EAAE,CAAC;YACzB,MAAM,CAAC,QAAQ,KAAK,EAAE,CAAC;YACvB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC;gBAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACtE,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC;YAClC,SAAS;QACV,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CAAE,CAAC;QAC9B,MAAM,GAAG,gBAAgB,CAAC,IAAI,CAAC;YAC9B,CAAC,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACzD,CAAC,CAAC,oBAAoB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC1C,CAAC;AAAA,CACD;AAED,SAAS,WAAW,CAAC,YAAoB,EAAE,YAAgC,EAAW;IACrF,IAAI,YAAY,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IACjD,IAAI,YAAY,KAAK,MAAM;QAAE,OAAO,IAAI,CAAC;IACzC,IAAI,YAAY,KAAK,OAAO;QAAE,OAAO,EAAE,CAAC;IACxC,IAAI,YAAY,KAAK,QAAQ;QAAE,OAAO,EAAE,CAAC;IACzC,IAAI,YAAY,KAAK,SAAS;QAAE,OAAO,IAAI,CAAC;IAC5C,IAAI,YAAY,KAAK,QAAQ,IAAI,YAAY,KAAK,SAAS;QAAE,OAAO,CAAC,CAAC;IACtE,IAAI,YAAY,KAAK,QAAQ,EAAE,CAAC;QAC/B,IAAI,YAAY,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI,YAAY,EAAE,QAAQ,CAAC,SAAS,CAAC;YAAE,OAAO,cAAc,CAAC;QACjG,IAAI,YAAY,EAAE,QAAQ,CAAC,SAAS,CAAC;YAAE,OAAO,eAAe,CAAC;QAC9D,IAAI,YAAY,EAAE,QAAQ,CAAC,OAAO,CAAC,IAAI,YAAY,EAAE,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,UAAU,CAAC;QAC3F,OAAO,0BAA0B,CAAC;IACnC,CAAC;IACD,OAAO,EAAE,CAAC;AAAA,CACV;AAED,SAAS,gBAAgB,CAAC,MAA+B,EAAE,GAAW,EAA2B;IAChG,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;IAC5B,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;QAC7E,OAAO,OAAkC,CAAC;IAC3C,MAAM,IAAI,GAA4B,EAAE,CAAC;IACzC,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;IACnB,OAAO,IAAI,CAAC;AAAA,CACZ;AAED,SAAS,eAAe,CAAC,MAA+B,EAAE,GAAW,EAAa;IACjF,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;IAC5B,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;QAAE,OAAO,OAAO,CAAC;IAC3C,MAAM,IAAI,GAAc,EAAE,CAAC;IAC3B,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;IACnB,OAAO,IAAI,CAAC;AAAA,CACZ;AAED,SAAS,iBAAiB,CAAC,IAA6B,EAAE,IAAuB,EAAE,KAAc,EAAQ;IACxG,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO;IAChC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IAC9B,IAAI,YAAY,GAAG,IAAI,CAAC;IACxB,IAAI,WAAkC,CAAC;IACvC,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC;QAClD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,KAAK,KAAK,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;QACzC,IAAI,WAAW,EAAE,CAAC;YACjB,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;YACnC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC;gBAAE,OAAO;YAC1C,IAAI,MAAM,EAAE,CAAC;gBACZ,WAAW,CAAC,UAAU,CAAC,GAAG,KAAK,CAAC;gBAChC,OAAO;YACR,CAAC;YACD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CAAE,CAAC;YAC9B,IAAI,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5B,MAAM,QAAQ,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC;gBACzC,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC1D,WAAW,CAAC,UAAU,CAAC,GAAG,SAAS,CAAC;gBACpC,WAAW,GAAG,SAAS,CAAC;gBACxB,SAAS;YACV,CAAC;YACD,MAAM,QAAQ,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC;YACzC,MAAM,UAAU,GACf,OAAO,QAAQ,KAAK,QAAQ,IAAI,QAAQ,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;gBAC5E,CAAC,CAAE,QAAoC;gBACvC,CAAC,CAAC,EAAE,CAAC;YACP,WAAW,CAAC,UAAU,CAAC,GAAG,UAAU,CAAC;YACrC,YAAY,GAAG,UAAU,CAAC;YAC1B,WAAW,GAAG,SAAS,CAAC;YACxB,SAAS;QACV,CAAC;QACD,IAAI,MAAM,EAAE,CAAC;YACZ,YAAY,CAAC,OAAO,CAAC,GAAG,KAAK,CAAC;YAC9B,OAAO;QACR,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CAAE,CAAC;QAC9B,IAAI,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5B,WAAW,GAAG,eAAe,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QACtD,CAAC;aAAM,CAAC;YACP,YAAY,GAAG,gBAAgB,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QACxD,CAAC;IACF,CAAC;AAAA,CACD;AAED,SAAS,YAAY,CAAC,MAA8B,EAA2B;IAC9E,MAAM,UAAU,GAA2B,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;IAC5F,MAAM,IAAI,GAA4B,EAAE,CAAC;IACzC,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QAClC,MAAM,IAAI,GAAG,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACzC,eAAe,CAAC,UAAU,EAAE,IAAI,EAAE,qBAAqB,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;QAC7E,iBAAiB,CAAC,IAAI,EAAE,IAAI,EAAE,WAAW,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;IACpF,CAAC;IACD,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,UAAU,EAAE,UAAgC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;AAAA,CAC5F;AAED,SAAS,0BAA0B,CAAC,OAAgC,EAAyB;IAC5F,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC9C,OAAO,CAAC,GAAG,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC/D,YAAY,EAAE,KAAK,CAAC,YAAY;QAChC,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,OAAO,EAAE,KAAK,CAAC,OAAO;KACtB,CAAC,CAAC,CAAC;AAAA,CACJ;AAED,MAAM,mBAAmB,GAAG,IAAI,GAAG,CAAS,CAAC,GAAG,sBAAsB,EAAE,OAAO,CAAC,CAAC,CAAC;AAElF,SAAS,sBAAsB,CAAC,KAAoC,EAA+B;IAClG,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;SACtC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;SAC/D,IAAI,EAAiC,CAAC;IACxC,OAAO,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;AAAA,CAC9C;AAED,MAAM,UAAU,wBAAwB,CAAC,KAAc,EAAmC;IACzF,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IACtF,MAAM,MAAM,GAAG,KAA4D,CAAC;IAC5E,OAAO,MAAM,CAAC,IAAI,KAAK,iBAAiB,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAAA,CAC3G;AAED,MAAM,UAAU,sBAAsB,CAAC,OAA0C,EAA4B;IAC5G,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC;QACrC,MAAM,OAAO,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,0BAA0B,CAAC,OAAO,CAAC,CAAC;QACnD,MAAM,aAAa,GAAG,yBAAyB,CAC9C,OAAO,CAAC,IAAI,EACZ,OAAO,CAAC,UAAU,EAClB,OAAO,CAAC,SAAS,EACjB,MAAM,CACN,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,eAAe,GACpB,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAChH,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC9C,MAAM,QAAQ,GAAG,mBAAmB,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC,SAAS,EAAE,EAAE,CAC/G,SAAS,CAAC,KAAK,CAAC,SAAS,CAAC,CAC1B,CAAC;QACF,OAAO;YACN,MAAM,EAAE,KAAK,GAAG,CAAC;YACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,eAAe;YACf,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,cAAc;YACnD,cAAc,EAAE,QAAQ,EAAE,cAAc,IAAI,EAAE;YAC9C,OAAO;SACP,CAAC;IAAA,CACF,CAAC,CAAC;AAAA,CACH","sourcesContent":["import { Compile } from \"typebox/compile\";\nimport type { Tool } from \"../../types.ts\";\nimport { analyzeToolArgumentErrors, type ValidationErrorLike } from \"./analyzer.ts\";\nimport { TOOL_REPAIR_MODE_NAMES, type ToolRepairFailureModeName, type ToolRepairModeName } from \"./registry.ts\";\nimport { repairToolArguments } from \"./repairer.ts\";\n\ninterface ReplayJsonSchemaObject {\n\ttype?: string | string[];\n\tproperties?: Record<string, ReplayJsonSchemaObject>;\n\trequired?: string[];\n\titems?: ReplayJsonSchemaObject;\n\tenum?: unknown[];\n\tadditionalProperties?: boolean;\n}\n\nexport interface ToolRepairCorpusShapeEntry {\n\tpath: string;\n\texpectedType?: string;\n\treceivedType: string;\n\tkeyword?: string;\n}\n\nexport interface ToolRepairCorpusRecord {\n\tkind: \"tool_validation\";\n\tprovider?: string;\n\tmodelId?: string;\n\ttool: string;\n\tfailureModes?: readonly string[];\n\tshape: readonly ToolRepairCorpusShapeEntry[];\n\terrorKeywords?: readonly string[];\n}\n\nexport interface ToolRepairReplayFixture {\n\ttool: string;\n\tparameters: Tool[\"parameters\"];\n\targuments: Record<string, unknown>;\n}\n\nexport interface ToolRepairReplayResult {\n\trecord: number;\n\tprovider?: string;\n\tmodelId?: string;\n\ttool: string;\n\tclassifiedModes: ToolRepairFailureModeName[];\n\toutcome: \"would-repair\" | \"would-bounce\";\n\trepairsApplied: ToolRepairModeName[];\n\tfixture: ToolRepairReplayFixture;\n}\n\nfunction splitReplayPath(path: string): string[] {\n\treturn path === \"root\" ? [] : path.split(\".\").filter((segment) => segment.length > 0);\n}\n\nfunction schemaForExpectedType(expectedType: string | undefined): ReplayJsonSchemaObject {\n\tif (!expectedType || expectedType === \"unknown\") return { type: \"null\" };\n\tif (expectedType === \"enum\") return { enum: [\"expected\"] };\n\tconst types = expectedType.split(\"|\").filter((type) => type.length > 0);\n\treturn { type: types.length <= 1 ? (types[0] ?? \"null\") : types };\n}\n\nfunction isNumericSegment(segment: string): boolean {\n\treturn /^\\d+$/.test(segment);\n}\n\nfunction ensureObjectProperty(schema: ReplayJsonSchemaObject, key: string): ReplayJsonSchemaObject {\n\tschema.type = \"object\";\n\tschema.properties ??= {};\n\tschema.required ??= [];\n\tif (!schema.required.includes(key)) schema.required.push(key);\n\tschema.properties[key] ??= { type: \"object\", properties: {}, required: [] };\n\treturn schema.properties[key];\n}\n\nfunction ensureArrayItems(schema: ReplayJsonSchemaObject): ReplayJsonSchemaObject {\n\tschema.type = \"array\";\n\tschema.items ??= { type: \"object\", properties: {}, required: [] };\n\treturn schema.items;\n}\n\nfunction setSchemaAtPath(root: ReplayJsonSchemaObject, path: readonly string[], leaf: ReplayJsonSchemaObject): void {\n\tif (path.length === 0) {\n\t\tObject.assign(root, leaf);\n\t\treturn;\n\t}\n\n\tlet cursor = root;\n\tfor (let index = 0; index < path.length; index++) {\n\t\tconst segment = path[index]!;\n\t\tconst isLast = index === path.length - 1;\n\t\tif (isNumericSegment(segment)) {\n\t\t\tcursor = isLast ? Object.assign(ensureArrayItems(cursor), leaf) : ensureArrayItems(cursor);\n\t\t\tcontinue;\n\t\t}\n\t\tif (isLast) {\n\t\t\tcursor.type = \"object\";\n\t\t\tcursor.properties ??= {};\n\t\t\tcursor.required ??= [];\n\t\t\tif (!cursor.required.includes(segment)) cursor.required.push(segment);\n\t\t\tcursor.properties[segment] = leaf;\n\t\t\tcontinue;\n\t\t}\n\t\tconst next = path[index + 1]!;\n\t\tcursor = isNumericSegment(next)\n\t\t\t? ensureArrayItems(ensureObjectProperty(cursor, segment))\n\t\t\t: ensureObjectProperty(cursor, segment);\n\t}\n}\n\nfunction sampleValue(receivedType: string, expectedType: string | undefined): unknown {\n\tif (receivedType === \"missing\") return undefined;\n\tif (receivedType === \"null\") return null;\n\tif (receivedType === \"array\") return [];\n\tif (receivedType === \"object\") return {};\n\tif (receivedType === \"boolean\") return true;\n\tif (receivedType === \"number\" || receivedType === \"integer\") return 1;\n\tif (receivedType === \"string\") {\n\t\tif (expectedType?.includes(\"number\") || expectedType?.includes(\"integer\")) return \"not-a-number\";\n\t\tif (expectedType?.includes(\"boolean\")) return \"not-a-boolean\";\n\t\tif (expectedType?.includes(\"array\") || expectedType?.includes(\"object\")) return \"not-json\";\n\t\treturn \"not-a-valid-replay-value\";\n\t}\n\treturn {};\n}\n\nfunction ensureArgsObject(target: Record<string, unknown>, key: string): Record<string, unknown> {\n\tconst current = target[key];\n\tif (typeof current === \"object\" && current !== null && !Array.isArray(current))\n\t\treturn current as Record<string, unknown>;\n\tconst next: Record<string, unknown> = {};\n\ttarget[key] = next;\n\treturn next;\n}\n\nfunction ensureArgsArray(target: Record<string, unknown>, key: string): unknown[] {\n\tconst current = target[key];\n\tif (Array.isArray(current)) return current;\n\tconst next: unknown[] = [];\n\ttarget[key] = next;\n\treturn next;\n}\n\nfunction setArgumentAtPath(root: Record<string, unknown>, path: readonly string[], value: unknown): void {\n\tif (value === undefined) return;\n\tif (path.length === 0) return;\n\tlet objectCursor = root;\n\tlet arrayCursor: unknown[] | undefined;\n\tfor (let index = 0; index < path.length; index++) {\n\t\tconst segment = path[index]!;\n\t\tconst isLast = index === path.length - 1;\n\t\tif (arrayCursor) {\n\t\t\tconst arrayIndex = Number(segment);\n\t\t\tif (!Number.isInteger(arrayIndex)) return;\n\t\t\tif (isLast) {\n\t\t\t\tarrayCursor[arrayIndex] = value;\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tconst next = path[index + 1]!;\n\t\t\tif (isNumericSegment(next)) {\n\t\t\t\tconst existing = arrayCursor[arrayIndex];\n\t\t\t\tconst nextArray = Array.isArray(existing) ? existing : [];\n\t\t\t\tarrayCursor[arrayIndex] = nextArray;\n\t\t\t\tarrayCursor = nextArray;\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tconst existing = arrayCursor[arrayIndex];\n\t\t\tconst nextObject =\n\t\t\t\ttypeof existing === \"object\" && existing !== null && !Array.isArray(existing)\n\t\t\t\t\t? (existing as Record<string, unknown>)\n\t\t\t\t\t: {};\n\t\t\tarrayCursor[arrayIndex] = nextObject;\n\t\t\tobjectCursor = nextObject;\n\t\t\tarrayCursor = undefined;\n\t\t\tcontinue;\n\t\t}\n\t\tif (isLast) {\n\t\t\tobjectCursor[segment] = value;\n\t\t\treturn;\n\t\t}\n\t\tconst next = path[index + 1]!;\n\t\tif (isNumericSegment(next)) {\n\t\t\tarrayCursor = ensureArgsArray(objectCursor, segment);\n\t\t} else {\n\t\t\tobjectCursor = ensureArgsObject(objectCursor, segment);\n\t\t}\n\t}\n}\n\nfunction buildFixture(record: ToolRepairCorpusRecord): ToolRepairReplayFixture {\n\tconst parameters: ReplayJsonSchemaObject = { type: \"object\", properties: {}, required: [] };\n\tconst args: Record<string, unknown> = {};\n\tfor (const shape of record.shape) {\n\t\tconst path = splitReplayPath(shape.path);\n\t\tsetSchemaAtPath(parameters, path, schemaForExpectedType(shape.expectedType));\n\t\tsetArgumentAtPath(args, path, sampleValue(shape.receivedType, shape.expectedType));\n\t}\n\treturn { tool: record.tool, parameters: parameters as Tool[\"parameters\"], arguments: args };\n}\n\nfunction validationErrorsForFixture(fixture: ToolRepairReplayFixture): ValidationErrorLike[] {\n\tconst validator = Compile(fixture.parameters);\n\treturn [...validator.Errors(fixture.arguments)].map((error) => ({\n\t\tinstancePath: error.instancePath,\n\t\tkeyword: error.keyword,\n\t\tmessage: error.message,\n\t}));\n}\n\nconst KNOWN_FAILURE_MODES = new Set<string>([...TOOL_REPAIR_MODE_NAMES, \"other\"]);\n\nfunction normalizedFailureModes(modes: readonly string[] | undefined): ToolRepairFailureModeName[] {\n\tconst unique = [...new Set(modes ?? [])]\n\t\t.map((mode) => (KNOWN_FAILURE_MODES.has(mode) ? mode : \"other\"))\n\t\t.sort() as ToolRepairFailureModeName[];\n\treturn unique.length > 0 ? unique : [\"other\"];\n}\n\nexport function isToolRepairCorpusRecord(value: unknown): value is ToolRepairCorpusRecord {\n\tif (typeof value !== \"object\" || value === null || Array.isArray(value)) return false;\n\tconst record = value as { kind?: unknown; tool?: unknown; shape?: unknown };\n\treturn record.kind === \"tool_validation\" && typeof record.tool === \"string\" && Array.isArray(record.shape);\n}\n\nexport function replayToolRepairCorpus(records: readonly ToolRepairCorpusRecord[]): ToolRepairReplayResult[] {\n\treturn records.map((record, index) => {\n\t\tconst fixture = buildFixture(record);\n\t\tconst errors = validationErrorsForFixture(fixture);\n\t\tconst analyzerModes = analyzeToolArgumentErrors(\n\t\t\tfixture.tool,\n\t\t\tfixture.parameters,\n\t\t\tfixture.arguments,\n\t\t\terrors,\n\t\t).flatMap((issue) => issue.modes);\n\t\tconst classifiedModes =\n\t\t\tanalyzerModes.length > 0 ? normalizedFailureModes(analyzerModes) : normalizedFailureModes(record.failureModes);\n\t\tconst validator = Compile(fixture.parameters);\n\t\tconst repaired = repairToolArguments(fixture.tool, fixture.parameters, fixture.arguments, errors, (candidate) =>\n\t\t\tvalidator.Check(candidate),\n\t\t);\n\t\treturn {\n\t\t\trecord: index + 1,\n\t\t\tprovider: record.provider,\n\t\t\tmodelId: record.modelId,\n\t\t\ttool: record.tool,\n\t\t\tclassifiedModes,\n\t\t\toutcome: repaired ? \"would-repair\" : \"would-bounce\",\n\t\t\trepairsApplied: repaired?.repairsApplied ?? [],\n\t\t\tfixture,\n\t\t};\n\t});\n}\n"]}

package/dist/utils/tool-repair/text-protocol.d.ts

@@ -0,0 +1,16 @@
+import type { Tool, ToolCall } from "../../types.ts";
+export type TextToolProtocolParseFailure = "overlap" | "mixed-prose" | "unrecognized";
+export interface ParsedTextToolCalls {
+    calls: ToolCall[];
+    text: string;
+    attempted: boolean;
+    failure?: TextToolProtocolParseFailure;
+}
+export type TextToolProtocolVariant = "tool-tag" | "tool-call" | "fenced-json";
+export interface TextToolProtocolOptions {
+    variant?: TextToolProtocolVariant;
+}
+export declare function normalizeTextToolProtocolOptions(option: boolean | TextToolProtocolOptions | undefined): TextToolProtocolOptions | undefined;
+export declare function generateTextToolProtocolPrimer(tools: readonly Tool[], options?: TextToolProtocolOptions): string;
+export declare function parseTextToolCalls(text: string, knownTools: readonly Tool[]): ParsedTextToolCalls;
+//# sourceMappingURL=text-protocol.d.ts.map

package/dist/utils/tool-repair/text-protocol.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"text-protocol.d.ts","sourceRoot":"","sources":["../../../src/utils/tool-repair/text-protocol.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAErD,MAAM,MAAM,4BAA4B,GAAG,SAAS,GAAG,aAAa,GAAG,cAAc,CAAC;AAEtF,MAAM,WAAW,mBAAmB;IACnC,KAAK,EAAE,QAAQ,EAAE,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,OAAO,CAAC;IACnB,OAAO,CAAC,EAAE,4BAA4B,CAAC;CACvC;AAED,MAAM,MAAM,uBAAuB,GAAG,UAAU,GAAG,WAAW,GAAG,aAAa,CAAC;AAE/E,MAAM,WAAW,uBAAuB;IACvC,OAAO,CAAC,EAAE,uBAAuB,CAAC;CAClC;AAgLD,wBAAgB,gCAAgC,CAC/C,MAAM,EAAE,OAAO,GAAG,uBAAuB,GAAG,SAAS,GACnD,uBAAuB,GAAG,SAAS,CAIrC;AAQD,wBAAgB,8BAA8B,CAAC,KAAK,EAAE,SAAS,IAAI,EAAE,EAAE,OAAO,CAAC,EAAE,uBAAuB,GAAG,MAAM,CAehH;AAED,wBAAgB,kBAAkB,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,IAAI,EAAE,GAAG,mBAAmB,CAajG","sourcesContent":["import type { Tool, ToolCall } from \"../../types.ts\";\n\nexport type TextToolProtocolParseFailure = \"overlap\" | \"mixed-prose\" | \"unrecognized\";\n\nexport interface ParsedTextToolCalls {\n\tcalls: ToolCall[];\n\ttext: string;\n\tattempted: boolean;\n\tfailure?: TextToolProtocolParseFailure;\n}\n\nexport type TextToolProtocolVariant = \"tool-tag\" | \"tool-call\" | \"fenced-json\";\n\nexport interface TextToolProtocolOptions {\n\tvariant?: TextToolProtocolVariant;\n}\n\ntype EnvelopeKind = \"pi_call\" | \"tool_call\" | \"fenced_json\";\n\ninterface EnvelopeMatch {\n\tkind: EnvelopeKind;\n\tstart: number;\n\tend: number;\n\tname?: string;\n\tbody: string;\n}\n\nconst MAX_SCHEMA_CHARS = 600;\nconst DEFAULT_TEXT_TOOL_PROTOCOL_VARIANT: TextToolProtocolVariant = \"tool-tag\";\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n\treturn typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n\nfunction compactJson(value: unknown, maxChars = MAX_SCHEMA_CHARS): string {\n\tconst json = JSON.stringify(value);\n\tif (json.length <= maxChars) return json;\n\treturn `${json.slice(0, maxChars - 1)}…`;\n}\n\nfunction escapeAttribute(value: string): string {\n\treturn value.replace(/&/g, \"&amp;\").replace(/\"/g, \"&quot;\").replace(/</g, \"&lt;\").replace(/>/g, \"&gt;\");\n}\n\nfunction unescapeAttribute(value: string): string {\n\treturn value\n\t\t.replace(/&quot;/g, '\"')\n\t\t.replace(/&lt;/g, \"<\")\n\t\t.replace(/&gt;/g, \">\")\n\t\t.replace(/&amp;/g, \"&\");\n}\n\nfunction knownToolNames(tools: readonly Tool[]): Set<string> {\n\treturn new Set(tools.map((tool) => tool.name));\n}\n\nfunction findToolEnvelopes(text: string): EnvelopeMatch[] {\n\tconst matches: EnvelopeMatch[] = [];\n\tconst piCallEnvelope = /<pi:call\\s+name=\"([^\"]+)\"\\s*>([\\s\\S]*?)<\\/pi:call>/g;\n\tfor (const match of text.matchAll(piCallEnvelope)) {\n\t\tmatches.push({\n\t\t\tkind: \"pi_call\",\n\t\t\tstart: match.index,\n\t\t\tend: match.index + match[0].length,\n\t\t\tname: unescapeAttribute(match[1] ?? \"\"),\n\t\t\tbody: match[2] ?? \"\",\n\t\t});\n\t}\n\n\tconst toolCallEnvelope = /<tool_call>([\\s\\S]*?)<\\/tool_call>/g;\n\tfor (const match of text.matchAll(toolCallEnvelope)) {\n\t\tmatches.push({\n\t\t\tkind: \"tool_call\",\n\t\t\tstart: match.index,\n\t\t\tend: match.index + match[0].length,\n\t\t\tbody: match[1] ?? \"\",\n\t\t});\n\t}\n\n\tconst fence = /```(?:tool|tool_call)\\s*\\n([\\s\\S]*?)\\n?```/gi;\n\tfor (const match of text.matchAll(fence)) {\n\t\tmatches.push({\n\t\t\tkind: \"fenced_json\",\n\t\t\tstart: match.index,\n\t\t\tend: match.index + match[0].length,\n\t\t\tbody: match[1] ?? \"\",\n\t\t});\n\t}\n\n\treturn matches.sort((a, b) => a.start - b.start);\n}\n\nfunction hasOverlap(matches: readonly EnvelopeMatch[]): boolean {\n\tlet previousEnd = -1;\n\tfor (const match of matches) {\n\t\tif (match.start < previousEnd) return true;\n\t\tpreviousEnd = match.end;\n\t}\n\treturn false;\n}\n\nfunction remainingText(text: string, matches: readonly EnvelopeMatch[]): string {\n\tlet cursor = 0;\n\tconst pieces: string[] = [];\n\tfor (const match of matches) {\n\t\tpieces.push(text.slice(cursor, match.start));\n\t\tcursor = match.end;\n\t}\n\tpieces.push(text.slice(cursor));\n\treturn pieces.join(\"\");\n}\n\nfunction coerceArguments(value: unknown): {\n\targuments: Record<string, unknown>;\n\trawArguments?: Record<string, unknown>;\n} {\n\tif (isRecord(value)) return { arguments: value };\n\treturn { arguments: value as Record<string, unknown>, rawArguments: { value } };\n}\n\nfunction parseJsonValue(raw: string): { ok: true; value: unknown } | { ok: false } {\n\ttry {\n\t\treturn { ok: true, value: JSON.parse(raw) as unknown };\n\t} catch {\n\t\treturn { ok: false };\n\t}\n}\n\nfunction parseJsonObject(raw: string): Record<string, unknown> | undefined {\n\tconst parsed = parseJsonValue(raw);\n\treturn parsed.ok && isRecord(parsed.value) ? parsed.value : undefined;\n}\n\nfunction textToolErrorMessage(name: string, names: readonly string[]): string | undefined {\n\tif (names.includes(name)) return undefined;\n\treturn `Unknown tool \"${name}\". Valid tools: ${names.join(\", \")}.`;\n}\n\nfunction extractNameFromMalformedJson(raw: string): string | undefined {\n\tconst match = /\"(?:name|tool)\"\\s*:\\s*\"([^\"]+)\"/.exec(raw);\n\treturn match?.[1];\n}\n\nfunction parsePiCallEnvelope(match: EnvelopeMatch, names: readonly string[], index: number): ToolCall | undefined {\n\tif (!match.name) return undefined;\n\tconst parsed = parseJsonValue(match.body);\n\tconst args = parsed.ok ? coerceArguments(parsed.value) : coerceArguments(match.body.trim());\n\treturn {\n\t\ttype: \"toolCall\",\n\t\tid: `text-tool-${index}`,\n\t\tname: match.name,\n\t\targuments: args.arguments,\n\t\trawArguments: parsed.ok ? args.rawArguments : { text: match.body.trim() },\n\t\tsource: \"text-protocol\",\n\t\terrorMessage: textToolErrorMessage(match.name, names),\n\t};\n}\n\nfunction parseEnvelope(match: EnvelopeMatch, names: readonly string[], index: number): ToolCall | undefined {\n\tif (match.kind === \"pi_call\") return parsePiCallEnvelope(match, names, index);\n\n\tconst parsed = parseJsonObject(match.body);\n\tif (!parsed) {\n\t\tconst name = extractNameFromMalformedJson(match.body);\n\t\tif (!name) return undefined;\n\t\treturn {\n\t\t\ttype: \"toolCall\",\n\t\t\tid: `text-tool-${index}`,\n\t\t\tname,\n\t\t\targuments: match.body.trim() as unknown as Record<string, unknown>,\n\t\t\trawArguments: { text: match.body.trim() },\n\t\t\tsource: \"text-protocol\",\n\t\t\terrorMessage: textToolErrorMessage(name, names),\n\t\t};\n\t}\n\n\tconst nameValue = parsed.name ?? parsed.tool;\n\tif (typeof nameValue !== \"string\") return undefined;\n\tconst argsValue = parsed.arguments ?? parsed.args ?? {};\n\tconst args = coerceArguments(argsValue);\n\treturn {\n\t\ttype: \"toolCall\",\n\t\tid: `text-tool-${index}`,\n\t\tname: nameValue,\n\t\targuments: args.arguments,\n\t\trawArguments: args.rawArguments,\n\t\tsource: \"text-protocol\",\n\t\terrorMessage: textToolErrorMessage(nameValue, names),\n\t};\n}\n\nexport function normalizeTextToolProtocolOptions(\n\toption: boolean | TextToolProtocolOptions | undefined,\n): TextToolProtocolOptions | undefined {\n\tif (!option) return undefined;\n\tif (option === true) return { variant: DEFAULT_TEXT_TOOL_PROTOCOL_VARIANT };\n\treturn { variant: option.variant ?? DEFAULT_TEXT_TOOL_PROTOCOL_VARIANT };\n}\n\nfunction formatVariantEnvelope(variant: TextToolProtocolVariant, toolName: string, argsJson: string): string {\n\tif (variant === \"tool-call\") return `<tool_call>{\"name\":\"${toolName}\",\"arguments\":${argsJson}}</tool_call>`;\n\tif (variant === \"fenced-json\") return `\\`\\`\\`tool_call\\n{\"name\":\"${toolName}\",\"arguments\":${argsJson}}\\n\\`\\`\\``;\n\treturn `<pi:call name=\"${escapeAttribute(toolName)}\">${argsJson}</pi:call>`;\n}\n\nexport function generateTextToolProtocolPrimer(tools: readonly Tool[], options?: TextToolProtocolOptions): string {\n\tif (tools.length === 0) return \"\";\n\tconst variant = options?.variant ?? DEFAULT_TEXT_TOOL_PROTOCOL_VARIANT;\n\tconst lines = [\n\t\t\"Text tool-call protocol is enabled.\",\n\t\t\"When calling tools, output only one or more envelopes and no prose:\",\n\t\tformatVariantEnvelope(variant, \"TOOL_NAME\", '{\"argument\":\"value\"}'),\n\t\t\"Available tools:\",\n\t];\n\tfor (const tool of tools) {\n\t\tlines.push(`- ${escapeAttribute(tool.name)}: ${tool.description}; args schema ${compactJson(tool.parameters)}`);\n\t}\n\tconst exampleTool = tools[0];\n\tlines.push(\"Examples:\", formatVariantEnvelope(variant, exampleTool.name, \"{}\"));\n\treturn lines.join(\"\\n\");\n}\n\nexport function parseTextToolCalls(text: string, knownTools: readonly Tool[]): ParsedTextToolCalls {\n\tif (knownTools.length === 0) return { calls: [], text, attempted: false };\n\tconst matches = findToolEnvelopes(text);\n\tif (matches.length === 0) return { calls: [], text, attempted: false };\n\tif (hasOverlap(matches)) return { calls: [], text, attempted: true, failure: \"overlap\" };\n\tconst remainder = remainingText(text, matches);\n\tconst names = [...knownToolNames(knownTools)];\n\tconst calls = matches\n\t\t.map((match, index) => parseEnvelope(match, names, index + 1))\n\t\t.filter((call): call is ToolCall => call !== undefined);\n\treturn calls.length > 0\n\t\t? { calls, text: remainder.trim(), attempted: true }\n\t\t: { calls: [], text, attempted: true, failure: \"unrecognized\" };\n}\n"]}

package/dist/utils/tool-repair/text-protocol.js

@@ -0,0 +1,198 @@
+const MAX_SCHEMA_CHARS = 600;
+const DEFAULT_TEXT_TOOL_PROTOCOL_VARIANT = "tool-tag";
+function isRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function compactJson(value, maxChars = MAX_SCHEMA_CHARS) {
+    const json = JSON.stringify(value);
+    if (json.length <= maxChars)
+        return json;
+    return `${json.slice(0, maxChars - 1)}…`;
+}
+function escapeAttribute(value) {
+    return value.replace(/&/g, "&amp;").replace(/"/g, "&quot;").replace(/</g, "&lt;").replace(/>/g, "&gt;");
+}
+function unescapeAttribute(value) {
+    return value
+        .replace(/&quot;/g, '"')
+        .replace(/&lt;/g, "<")
+        .replace(/&gt;/g, ">")
+        .replace(/&amp;/g, "&");
+}
+function knownToolNames(tools) {
+    return new Set(tools.map((tool) => tool.name));
+}
+function findToolEnvelopes(text) {
+    const matches = [];
+    const piCallEnvelope = /<pi:call\s+name="([^"]+)"\s*>([\s\S]*?)<\/pi:call>/g;
+    for (const match of text.matchAll(piCallEnvelope)) {
+        matches.push({
+            kind: "pi_call",
+            start: match.index,
+            end: match.index + match[0].length,
+            name: unescapeAttribute(match[1] ?? ""),
+            body: match[2] ?? "",
+        });
+    }
+    const toolCallEnvelope = /<tool_call>([\s\S]*?)<\/tool_call>/g;
+    for (const match of text.matchAll(toolCallEnvelope)) {
+        matches.push({
+            kind: "tool_call",
+            start: match.index,
+            end: match.index + match[0].length,
+            body: match[1] ?? "",
+        });
+    }
+    const fence = /```(?:tool|tool_call)\s*\n([\s\S]*?)\n?```/gi;
+    for (const match of text.matchAll(fence)) {
+        matches.push({
+            kind: "fenced_json",
+            start: match.index,
+            end: match.index + match[0].length,
+            body: match[1] ?? "",
+        });
+    }
+    return matches.sort((a, b) => a.start - b.start);
+}
+function hasOverlap(matches) {
+    let previousEnd = -1;
+    for (const match of matches) {
+        if (match.start < previousEnd)
+            return true;
+        previousEnd = match.end;
+    }
+    return false;
+}
+function remainingText(text, matches) {
+    let cursor = 0;
+    const pieces = [];
+    for (const match of matches) {
+        pieces.push(text.slice(cursor, match.start));
+        cursor = match.end;
+    }
+    pieces.push(text.slice(cursor));
+    return pieces.join("");
+}
+function coerceArguments(value) {
+    if (isRecord(value))
+        return { arguments: value };
+    return { arguments: value, rawArguments: { value } };
+}
+function parseJsonValue(raw) {
+    try {
+        return { ok: true, value: JSON.parse(raw) };
+    }
+    catch {
+        return { ok: false };
+    }
+}
+function parseJsonObject(raw) {
+    const parsed = parseJsonValue(raw);
+    return parsed.ok && isRecord(parsed.value) ? parsed.value : undefined;
+}
+function textToolErrorMessage(name, names) {
+    if (names.includes(name))
+        return undefined;
+    return `Unknown tool "${name}". Valid tools: ${names.join(", ")}.`;
+}
+function extractNameFromMalformedJson(raw) {
+    const match = /"(?:name|tool)"\s*:\s*"([^"]+)"/.exec(raw);
+    return match?.[1];
+}
+function parsePiCallEnvelope(match, names, index) {
+    if (!match.name)
+        return undefined;
+    const parsed = parseJsonValue(match.body);
+    const args = parsed.ok ? coerceArguments(parsed.value) : coerceArguments(match.body.trim());
+    return {
+        type: "toolCall",
+        id: `text-tool-${index}`,
+        name: match.name,
+        arguments: args.arguments,
+        rawArguments: parsed.ok ? args.rawArguments : { text: match.body.trim() },
+        source: "text-protocol",
+        errorMessage: textToolErrorMessage(match.name, names),
+    };
+}
+function parseEnvelope(match, names, index) {
+    if (match.kind === "pi_call")
+        return parsePiCallEnvelope(match, names, index);
+    const parsed = parseJsonObject(match.body);
+    if (!parsed) {
+        const name = extractNameFromMalformedJson(match.body);
+        if (!name)
+            return undefined;
+        return {
+            type: "toolCall",
+            id: `text-tool-${index}`,
+            name,
+            arguments: match.body.trim(),
+            rawArguments: { text: match.body.trim() },
+            source: "text-protocol",
+            errorMessage: textToolErrorMessage(name, names),
+        };
+    }
+    const nameValue = parsed.name ?? parsed.tool;
+    if (typeof nameValue !== "string")
+        return undefined;
+    const argsValue = parsed.arguments ?? parsed.args ?? {};
+    const args = coerceArguments(argsValue);
+    return {
+        type: "toolCall",
+        id: `text-tool-${index}`,
+        name: nameValue,
+        arguments: args.arguments,
+        rawArguments: args.rawArguments,
+        source: "text-protocol",
+        errorMessage: textToolErrorMessage(nameValue, names),
+    };
+}
+export function normalizeTextToolProtocolOptions(option) {
+    if (!option)
+        return undefined;
+    if (option === true)
+        return { variant: DEFAULT_TEXT_TOOL_PROTOCOL_VARIANT };
+    return { variant: option.variant ?? DEFAULT_TEXT_TOOL_PROTOCOL_VARIANT };
+}
+function formatVariantEnvelope(variant, toolName, argsJson) {
+    if (variant === "tool-call")
+        return `<tool_call>{"name":"${toolName}","arguments":${argsJson}}</tool_call>`;
+    if (variant === "fenced-json")
+        return `\`\`\`tool_call\n{"name":"${toolName}","arguments":${argsJson}}\n\`\`\``;
+    return `<pi:call name="${escapeAttribute(toolName)}">${argsJson}</pi:call>`;
+}
+export function generateTextToolProtocolPrimer(tools, options) {
+    if (tools.length === 0)
+        return "";
+    const variant = options?.variant ?? DEFAULT_TEXT_TOOL_PROTOCOL_VARIANT;
+    const lines = [
+        "Text tool-call protocol is enabled.",
+        "When calling tools, output only one or more envelopes and no prose:",
+        formatVariantEnvelope(variant, "TOOL_NAME", '{"argument":"value"}'),
+        "Available tools:",
+    ];
+    for (const tool of tools) {
+        lines.push(`- ${escapeAttribute(tool.name)}: ${tool.description}; args schema ${compactJson(tool.parameters)}`);
+    }
+    const exampleTool = tools[0];
+    lines.push("Examples:", formatVariantEnvelope(variant, exampleTool.name, "{}"));
+    return lines.join("\n");
+}
+export function parseTextToolCalls(text, knownTools) {
+    if (knownTools.length === 0)
+        return { calls: [], text, attempted: false };
+    const matches = findToolEnvelopes(text);
+    if (matches.length === 0)
+        return { calls: [], text, attempted: false };
+    if (hasOverlap(matches))
+        return { calls: [], text, attempted: true, failure: "overlap" };
+    const remainder = remainingText(text, matches);
+    const names = [...knownToolNames(knownTools)];
+    const calls = matches
+        .map((match, index) => parseEnvelope(match, names, index + 1))
+        .filter((call) => call !== undefined);
+    return calls.length > 0
+        ? { calls, text: remainder.trim(), attempted: true }
+        : { calls: [], text, attempted: true, failure: "unrecognized" };
+}
+//# sourceMappingURL=text-protocol.js.map

package/dist/utils/tool-repair/text-protocol.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"text-protocol.js","sourceRoot":"","sources":["../../../src/utils/tool-repair/text-protocol.ts"],"names":[],"mappings":"AA2BA,MAAM,gBAAgB,GAAG,GAAG,CAAC;AAC7B,MAAM,kCAAkC,GAA4B,UAAU,CAAC;AAE/E,SAAS,QAAQ,CAAC,KAAc,EAAoC;IACnE,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAAA,CAC5E;AAED,SAAS,WAAW,CAAC,KAAc,EAAE,QAAQ,GAAG,gBAAgB,EAAU;IACzE,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IACnC,IAAI,IAAI,CAAC,MAAM,IAAI,QAAQ;QAAE,OAAO,IAAI,CAAC;IACzC,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,GAAG,CAAC,CAAC,KAAG,CAAC;AAAA,CACzC;AAED,SAAS,eAAe,CAAC,KAAa,EAAU;IAC/C,OAAO,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;AAAA,CACxG;AAED,SAAS,iBAAiB,CAAC,KAAa,EAAU;IACjD,OAAO,KAAK;SACV,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC;SACvB,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC;SACrB,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC;SACrB,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;AAAA,CACzB;AAED,SAAS,cAAc,CAAC,KAAsB,EAAe;IAC5D,OAAO,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;AAAA,CAC/C;AAED,SAAS,iBAAiB,CAAC,IAAY,EAAmB;IACzD,MAAM,OAAO,GAAoB,EAAE,CAAC;IACpC,MAAM,cAAc,GAAG,qDAAqD,CAAC;IAC7E,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;QACnD,OAAO,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,SAAS;YACf,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,GAAG,EAAE,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM;YAClC,IAAI,EAAE,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YACvC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE;SACpB,CAAC,CAAC;IACJ,CAAC;IAED,MAAM,gBAAgB,GAAG,qCAAqC,CAAC;IAC/D,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;QACrD,OAAO,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,GAAG,EAAE,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM;YAClC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE;SACpB,CAAC,CAAC;IACJ,CAAC;IAED,MAAM,KAAK,GAAG,8CAA8C,CAAC;IAC7D,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1C,OAAO,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,aAAa;YACnB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,GAAG,EAAE,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM;YAClC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE;SACpB,CAAC,CAAC;IACJ,CAAC;IAED,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;AAAA,CACjD;AAED,SAAS,UAAU,CAAC,OAAiC,EAAW;IAC/D,IAAI,WAAW,GAAG,CAAC,CAAC,CAAC;IACrB,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC7B,IAAI,KAAK,CAAC,KAAK,GAAG,WAAW;YAAE,OAAO,IAAI,CAAC;QAC3C,WAAW,GAAG,KAAK,CAAC,GAAG,CAAC;IACzB,CAAC;IACD,OAAO,KAAK,CAAC;AAAA,CACb;AAED,SAAS,aAAa,CAAC,IAAY,EAAE,OAAiC,EAAU;IAC/E,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC7B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;QAC7C,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC;IACpB,CAAC;IACD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAChC,OAAO,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAAA,CACvB;AAED,SAAS,eAAe,CAAC,KAAc,EAGrC;IACD,IAAI,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;IACjD,OAAO,EAAE,SAAS,EAAE,KAAgC,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC;AAAA,CAChF;AAED,SAAS,cAAc,CAAC,GAAW,EAAgD;IAClF,IAAI,CAAC;QACJ,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAY,EAAE,CAAC;IACxD,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;IACtB,CAAC;AAAA,CACD;AAED,SAAS,eAAe,CAAC,GAAW,EAAuC;IAC1E,MAAM,MAAM,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;IACnC,OAAO,MAAM,CAAC,EAAE,IAAI,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AAAA,CACtE;AAED,SAAS,oBAAoB,CAAC,IAAY,EAAE,KAAwB,EAAsB;IACzF,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC;QAAE,OAAO,SAAS,CAAC;IAC3C,OAAO,iBAAiB,IAAI,mBAAmB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;AAAA,CACnE;AAED,SAAS,4BAA4B,CAAC,GAAW,EAAsB;IACtE,MAAM,KAAK,GAAG,iCAAiC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC1D,OAAO,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;AAAA,CAClB;AAED,SAAS,mBAAmB,CAAC,KAAoB,EAAE,KAAwB,EAAE,KAAa,EAAwB;IACjH,IAAI,CAAC,KAAK,CAAC,IAAI;QAAE,OAAO,SAAS,CAAC;IAClC,MAAM,MAAM,GAAG,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC1C,MAAM,IAAI,GAAG,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IAC5F,OAAO;QACN,IAAI,EAAE,UAAU;QAChB,EAAE,EAAE,aAAa,KAAK,EAAE;QACxB,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,YAAY,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE;QACzE,MAAM,EAAE,eAAe;QACvB,YAAY,EAAE,oBAAoB,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC;KACrD,CAAC;AAAA,CACF;AAED,SAAS,aAAa,CAAC,KAAoB,EAAE,KAAwB,EAAE,KAAa,EAAwB;IAC3G,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS;QAAE,OAAO,mBAAmB,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;IAE9E,MAAM,MAAM,GAAG,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC3C,IAAI,CAAC,MAAM,EAAE,CAAC;QACb,MAAM,IAAI,GAAG,4BAA4B,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACtD,IAAI,CAAC,IAAI;YAAE,OAAO,SAAS,CAAC;QAC5B,OAAO;YACN,IAAI,EAAE,UAAU;YAChB,EAAE,EAAE,aAAa,KAAK,EAAE;YACxB,IAAI;YACJ,SAAS,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,EAAwC;YAClE,YAAY,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE;YACzC,MAAM,EAAE,eAAe;YACvB,YAAY,EAAE,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC;SAC/C,CAAC;IACH,CAAC;IAED,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC;IAC7C,IAAI,OAAO,SAAS,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IACpD,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;IACxD,MAAM,IAAI,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC;IACxC,OAAO;QACN,IAAI,EAAE,UAAU;QAChB,EAAE,EAAE,aAAa,KAAK,EAAE;QACxB,IAAI,EAAE,SAAS;QACf,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,YAAY,EAAE,IAAI,CAAC,YAAY;QAC/B,MAAM,EAAE,eAAe;QACvB,YAAY,EAAE,oBAAoB,CAAC,SAAS,EAAE,KAAK,CAAC;KACpD,CAAC;AAAA,CACF;AAED,MAAM,UAAU,gCAAgC,CAC/C,MAAqD,EACf;IACtC,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAC9B,IAAI,MAAM,KAAK,IAAI;QAAE,OAAO,EAAE,OAAO,EAAE,kCAAkC,EAAE,CAAC;IAC5E,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,kCAAkC,EAAE,CAAC;AAAA,CACzE;AAED,SAAS,qBAAqB,CAAC,OAAgC,EAAE,QAAgB,EAAE,QAAgB,EAAU;IAC5G,IAAI,OAAO,KAAK,WAAW;QAAE,OAAO,uBAAuB,QAAQ,iBAAiB,QAAQ,eAAe,CAAC;IAC5G,IAAI,OAAO,KAAK,aAAa;QAAE,OAAO,6BAA6B,QAAQ,iBAAiB,QAAQ,WAAW,CAAC;IAChH,OAAO,kBAAkB,eAAe,CAAC,QAAQ,CAAC,KAAK,QAAQ,YAAY,CAAC;AAAA,CAC5E;AAED,MAAM,UAAU,8BAA8B,CAAC,KAAsB,EAAE,OAAiC,EAAU;IACjH,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAClC,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,kCAAkC,CAAC;IACvE,MAAM,KAAK,GAAG;QACb,qCAAqC;QACrC,qEAAqE;QACrE,qBAAqB,CAAC,OAAO,EAAE,WAAW,EAAE,sBAAsB,CAAC;QACnE,kBAAkB;KAClB,CAAC;IACF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QAC1B,KAAK,CAAC,IAAI,CAAC,KAAK,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,WAAW,iBAAiB,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IACjH,CAAC;IACD,MAAM,WAAW,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IAC7B,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,qBAAqB,CAAC,OAAO,EAAE,WAAW,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;IAChF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAAA,CACxB;AAED,MAAM,UAAU,kBAAkB,CAAC,IAAY,EAAE,UAA2B,EAAuB;IAClG,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;IAC1E,MAAM,OAAO,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;IACxC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;IACvE,IAAI,UAAU,CAAC,OAAO,CAAC;QAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;IACzF,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAC/C,MAAM,KAAK,GAAG,CAAC,GAAG,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC;IAC9C,MAAM,KAAK,GAAG,OAAO;SACnB,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,aAAa,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;SAC7D,MAAM,CAAC,CAAC,IAAI,EAAoB,EAAE,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;IACzD,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC;QACtB,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE;QACpD,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC;AAAA,CACjE","sourcesContent":["import type { Tool, ToolCall } from \"../../types.ts\";\n\nexport type TextToolProtocolParseFailure = \"overlap\" | \"mixed-prose\" | \"unrecognized\";\n\nexport interface ParsedTextToolCalls {\n\tcalls: ToolCall[];\n\ttext: string;\n\tattempted: boolean;\n\tfailure?: TextToolProtocolParseFailure;\n}\n\nexport type TextToolProtocolVariant = \"tool-tag\" | \"tool-call\" | \"fenced-json\";\n\nexport interface TextToolProtocolOptions {\n\tvariant?: TextToolProtocolVariant;\n}\n\ntype EnvelopeKind = \"pi_call\" | \"tool_call\" | \"fenced_json\";\n\ninterface EnvelopeMatch {\n\tkind: EnvelopeKind;\n\tstart: number;\n\tend: number;\n\tname?: string;\n\tbody: string;\n}\n\nconst MAX_SCHEMA_CHARS = 600;\nconst DEFAULT_TEXT_TOOL_PROTOCOL_VARIANT: TextToolProtocolVariant = \"tool-tag\";\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n\treturn typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n\nfunction compactJson(value: unknown, maxChars = MAX_SCHEMA_CHARS): string {\n\tconst json = JSON.stringify(value);\n\tif (json.length <= maxChars) return json;\n\treturn `${json.slice(0, maxChars - 1)}…`;\n}\n\nfunction escapeAttribute(value: string): string {\n\treturn value.replace(/&/g, \"&amp;\").replace(/\"/g, \"&quot;\").replace(/</g, \"&lt;\").replace(/>/g, \"&gt;\");\n}\n\nfunction unescapeAttribute(value: string): string {\n\treturn value\n\t\t.replace(/&quot;/g, '\"')\n\t\t.replace(/&lt;/g, \"<\")\n\t\t.replace(/&gt;/g, \">\")\n\t\t.replace(/&amp;/g, \"&\");\n}\n\nfunction knownToolNames(tools: readonly Tool[]): Set<string> {\n\treturn new Set(tools.map((tool) => tool.name));\n}\n\nfunction findToolEnvelopes(text: string): EnvelopeMatch[] {\n\tconst matches: EnvelopeMatch[] = [];\n\tconst piCallEnvelope = /<pi:call\\s+name=\"([^\"]+)\"\\s*>([\\s\\S]*?)<\\/pi:call>/g;\n\tfor (const match of text.matchAll(piCallEnvelope)) {\n\t\tmatches.push({\n\t\t\tkind: \"pi_call\",\n\t\t\tstart: match.index,\n\t\t\tend: match.index + match[0].length,\n\t\t\tname: unescapeAttribute(match[1] ?? \"\"),\n\t\t\tbody: match[2] ?? \"\",\n\t\t});\n\t}\n\n\tconst toolCallEnvelope = /<tool_call>([\\s\\S]*?)<\\/tool_call>/g;\n\tfor (const match of text.matchAll(toolCallEnvelope)) {\n\t\tmatches.push({\n\t\t\tkind: \"tool_call\",\n\t\t\tstart: match.index,\n\t\t\tend: match.index + match[0].length,\n\t\t\tbody: match[1] ?? \"\",\n\t\t});\n\t}\n\n\tconst fence = /```(?:tool|tool_call)\\s*\\n([\\s\\S]*?)\\n?```/gi;\n\tfor (const match of text.matchAll(fence)) {\n\t\tmatches.push({\n\t\t\tkind: \"fenced_json\",\n\t\t\tstart: match.index,\n\t\t\tend: match.index + match[0].length,\n\t\t\tbody: match[1] ?? \"\",\n\t\t});\n\t}\n\n\treturn matches.sort((a, b) => a.start - b.start);\n}\n\nfunction hasOverlap(matches: readonly EnvelopeMatch[]): boolean {\n\tlet previousEnd = -1;\n\tfor (const match of matches) {\n\t\tif (match.start < previousEnd) return true;\n\t\tpreviousEnd = match.end;\n\t}\n\treturn false;\n}\n\nfunction remainingText(text: string, matches: readonly EnvelopeMatch[]): string {\n\tlet cursor = 0;\n\tconst pieces: string[] = [];\n\tfor (const match of matches) {\n\t\tpieces.push(text.slice(cursor, match.start));\n\t\tcursor = match.end;\n\t}\n\tpieces.push(text.slice(cursor));\n\treturn pieces.join(\"\");\n}\n\nfunction coerceArguments(value: unknown): {\n\targuments: Record<string, unknown>;\n\trawArguments?: Record<string, unknown>;\n} {\n\tif (isRecord(value)) return { arguments: value };\n\treturn { arguments: value as Record<string, unknown>, rawArguments: { value } };\n}\n\nfunction parseJsonValue(raw: string): { ok: true; value: unknown } | { ok: false } {\n\ttry {\n\t\treturn { ok: true, value: JSON.parse(raw) as unknown };\n\t} catch {\n\t\treturn { ok: false };\n\t}\n}\n\nfunction parseJsonObject(raw: string): Record<string, unknown> | undefined {\n\tconst parsed = parseJsonValue(raw);\n\treturn parsed.ok && isRecord(parsed.value) ? parsed.value : undefined;\n}\n\nfunction textToolErrorMessage(name: string, names: readonly string[]): string | undefined {\n\tif (names.includes(name)) return undefined;\n\treturn `Unknown tool \"${name}\". Valid tools: ${names.join(\", \")}.`;\n}\n\nfunction extractNameFromMalformedJson(raw: string): string | undefined {\n\tconst match = /\"(?:name|tool)\"\\s*:\\s*\"([^\"]+)\"/.exec(raw);\n\treturn match?.[1];\n}\n\nfunction parsePiCallEnvelope(match: EnvelopeMatch, names: readonly string[], index: number): ToolCall | undefined {\n\tif (!match.name) return undefined;\n\tconst parsed = parseJsonValue(match.body);\n\tconst args = parsed.ok ? coerceArguments(parsed.value) : coerceArguments(match.body.trim());\n\treturn {\n\t\ttype: \"toolCall\",\n\t\tid: `text-tool-${index}`,\n\t\tname: match.name,\n\t\targuments: args.arguments,\n\t\trawArguments: parsed.ok ? args.rawArguments : { text: match.body.trim() },\n\t\tsource: \"text-protocol\",\n\t\terrorMessage: textToolErrorMessage(match.name, names),\n\t};\n}\n\nfunction parseEnvelope(match: EnvelopeMatch, names: readonly string[], index: number): ToolCall | undefined {\n\tif (match.kind === \"pi_call\") return parsePiCallEnvelope(match, names, index);\n\n\tconst parsed = parseJsonObject(match.body);\n\tif (!parsed) {\n\t\tconst name = extractNameFromMalformedJson(match.body);\n\t\tif (!name) return undefined;\n\t\treturn {\n\t\t\ttype: \"toolCall\",\n\t\t\tid: `text-tool-${index}`,\n\t\t\tname,\n\t\t\targuments: match.body.trim() as unknown as Record<string, unknown>,\n\t\t\trawArguments: { text: match.body.trim() },\n\t\t\tsource: \"text-protocol\",\n\t\t\terrorMessage: textToolErrorMessage(name, names),\n\t\t};\n\t}\n\n\tconst nameValue = parsed.name ?? parsed.tool;\n\tif (typeof nameValue !== \"string\") return undefined;\n\tconst argsValue = parsed.arguments ?? parsed.args ?? {};\n\tconst args = coerceArguments(argsValue);\n\treturn {\n\t\ttype: \"toolCall\",\n\t\tid: `text-tool-${index}`,\n\t\tname: nameValue,\n\t\targuments: args.arguments,\n\t\trawArguments: args.rawArguments,\n\t\tsource: \"text-protocol\",\n\t\terrorMessage: textToolErrorMessage(nameValue, names),\n\t};\n}\n\nexport function normalizeTextToolProtocolOptions(\n\toption: boolean | TextToolProtocolOptions | undefined,\n): TextToolProtocolOptions | undefined {\n\tif (!option) return undefined;\n\tif (option === true) return { variant: DEFAULT_TEXT_TOOL_PROTOCOL_VARIANT };\n\treturn { variant: option.variant ?? DEFAULT_TEXT_TOOL_PROTOCOL_VARIANT };\n}\n\nfunction formatVariantEnvelope(variant: TextToolProtocolVariant, toolName: string, argsJson: string): string {\n\tif (variant === \"tool-call\") return `<tool_call>{\"name\":\"${toolName}\",\"arguments\":${argsJson}}</tool_call>`;\n\tif (variant === \"fenced-json\") return `\\`\\`\\`tool_call\\n{\"name\":\"${toolName}\",\"arguments\":${argsJson}}\\n\\`\\`\\``;\n\treturn `<pi:call name=\"${escapeAttribute(toolName)}\">${argsJson}</pi:call>`;\n}\n\nexport function generateTextToolProtocolPrimer(tools: readonly Tool[], options?: TextToolProtocolOptions): string {\n\tif (tools.length === 0) return \"\";\n\tconst variant = options?.variant ?? DEFAULT_TEXT_TOOL_PROTOCOL_VARIANT;\n\tconst lines = [\n\t\t\"Text tool-call protocol is enabled.\",\n\t\t\"When calling tools, output only one or more envelopes and no prose:\",\n\t\tformatVariantEnvelope(variant, \"TOOL_NAME\", '{\"argument\":\"value\"}'),\n\t\t\"Available tools:\",\n\t];\n\tfor (const tool of tools) {\n\t\tlines.push(`- ${escapeAttribute(tool.name)}: ${tool.description}; args schema ${compactJson(tool.parameters)}`);\n\t}\n\tconst exampleTool = tools[0];\n\tlines.push(\"Examples:\", formatVariantEnvelope(variant, exampleTool.name, \"{}\"));\n\treturn lines.join(\"\\n\");\n}\n\nexport function parseTextToolCalls(text: string, knownTools: readonly Tool[]): ParsedTextToolCalls {\n\tif (knownTools.length === 0) return { calls: [], text, attempted: false };\n\tconst matches = findToolEnvelopes(text);\n\tif (matches.length === 0) return { calls: [], text, attempted: false };\n\tif (hasOverlap(matches)) return { calls: [], text, attempted: true, failure: \"overlap\" };\n\tconst remainder = remainingText(text, matches);\n\tconst names = [...knownToolNames(knownTools)];\n\tconst calls = matches\n\t\t.map((match, index) => parseEnvelope(match, names, index + 1))\n\t\t.filter((call): call is ToolCall => call !== undefined);\n\treturn calls.length > 0\n\t\t? { calls, text: remainder.trim(), attempted: true }\n\t\t: { calls: [], text, attempted: true, failure: \"unrecognized\" };\n}\n"]}

package/dist/utils/validation.d.ts

@@ -1,4 +1,42 @@
 import type { Tool, ToolCall } from "../types.ts";
+import { type ToolRepairFailureModeName, type ToolRepairModeName } from "./tool-repair/registry.ts";
+export type ToolArgumentValidationOutcome = "clean" | "repaired" | "bounced";
+export type ToolArgumentTeachState = "none" | "note" | "rule";
+export type ToolArgumentExecutionOutcome = "not_run" | "succeeded" | "failed";
+export interface ToolArgumentFailureShapeEntry {
+    path: string;
+    expectedType: string;
+    receivedType: string;
+    keyword?: string;
+}
+export interface ToolArgumentValidationTelemetryEvent {
+    outcome: ToolArgumentValidationOutcome;
+    provider?: string;
+    model?: string;
+    tool: string;
+    failureModes: ToolRepairFailureModeName[];
+    repairsApplied: ToolRepairModeName[];
+    failureShape?: ToolArgumentFailureShapeEntry[];
+    errorKeywords?: string[];
+    taught: ToolArgumentTeachState;
+    executionOutcome: ToolArgumentExecutionOutcome;
+}
+export interface ToolArgumentValidationOptions {
+    model?: string;
+    provider?: string;
+    telemetry?: (event: ToolArgumentValidationTelemetryEvent) => void;
+    repairEnabled?: boolean;
+}
+export declare class ToolArgumentValidationError extends Error {
+    readonly toolName: string;
+    readonly signature: string;
+    readonly enrichment: string;
+    constructor(message: string, options: {
+        toolName: string;
+        signature: string;
+        enrichment: string;
+    });
+}
 /**
  * Finds a tool by name and validates the tool call arguments against its TypeBox schema
  * @param tools Array of tool definitions
@@ -6,13 +44,18 @@ import type { Tool, ToolCall } from "../types.ts";
  * @returns The validated arguments
  * @throws Error if tool is not found or validation fails
  */
-export declare function validateToolCall(tools: Tool[], toolCall: ToolCall): any;
+export declare function validateToolCall(tools: Tool[], toolCall: ToolCall, options?: ToolArgumentValidationOptions): Record<string, unknown>;
 /**
- * Validates tool call arguments against the tool's TypeBox schema
+ * Validates tool call arguments against the tool's TypeBox schema.
+ *
+ * The hot path is validate-first and allocation-free: valid arguments return the exact
+ * argument object emitted by the model. Only invalid calls enter the deterministic
+ * repair layer, which applies named, guard-checked shape repairs on a clone.
+ *
  * @param tool The tool definition with TypeBox schema
  * @param toolCall The tool call from the LLM
- * @returns The validated (and potentially coerced) arguments
+ * @returns The validated (and potentially repaired) arguments
  * @throws Error with formatted message if validation fails
  */
-export declare function validateToolArguments(tool: Tool, toolCall: ToolCall): any;
+export declare function validateToolArguments(tool: Tool, toolCall: ToolCall, options?: ToolArgumentValidationOptions): Record<string, unknown>;
 //# sourceMappingURL=validation.d.ts.map