llmz 0.9.0 → 0.9.2

package/dist/{common-BnvrDnIT.cjs → common-DJkSpeqw.cjs}

@@ -1,4 +1,4 @@
-const require_compiler = require("./compiler-C-OI9sim.cjs");
+const require_compiler = require("./compiler-BJJUmNS9.cjs");
 let json5 = require("json5");
 json5 = require_compiler.__toESM(json5, 1);
 let jsonrepair = require("jsonrepair");

package/dist/{common-BlRFtdf2.js → common-DuEhn6u3.js}

@@ -1,5 +1,5 @@
 import { n as __exportAll } from "./rolldown-runtime-EAsCxrDo.js";
-import { r as hasTopLevelReturn } from "./compiler-CW1VjQsn.js";
+import { r as hasTopLevelReturn } from "./compiler-BPl-5CXh.js";
 import JSON5 from "json5";
 import { jsonrepair } from "jsonrepair";
 //#region src/message-stream/types.ts

package/dist/compiler/plugins/variable-extraction.d.ts

@@ -1,10 +1,4 @@
 import { type Ctx } from '../ast.js';
 export declare const VariableTrackingFnIdentifier = "__var__";
-/**
- * Tracks user variables so the VM can read their values between iterations:
- * - after each variable declaration: `__var__("a", () => eval("a"));`
- * - at the top of function bodies, for their parameters
- *
- * The `eval` indirection captures the variable lazily in its own scope.
- */
+/** Retain getters only for bindings visible between iterations, never local shadows. */
 export declare function applyVariableTracking(ctx: Ctx, variables: Set<string>): void;

package/dist/{compiler-C-OI9sim.cjs → compiler-BJJUmNS9.cjs}

@@ -349,77 +349,43 @@ function applyToolCallTracking(ctx, calls) {
 //#endregion
 //#region src/compiler/plugins/variable-extraction.ts
 const VariableTrackingFnIdentifier = "__var__";
-/**
-* Tracks user variables so the VM can read their values between iterations:
-* - after each variable declaration: `__var__("a", () => eval("a"));`
-* - at the top of function bodies, for their parameters
-*
-* The `eval` indirection captures the variable lazily in its own scope.
-*/
+const FUNCTION_TYPES$1 = /* @__PURE__ */ new Set([
+	"FunctionDeclaration",
+	"FunctionExpression",
+	"ArrowFunctionExpression"
+]);
+function declaredNames(id) {
+	if (!id) return [];
+	if (id.type === "Identifier") return [id.name];
+	if (id.type === "RestElement") return declaredNames(id.argument);
+	if (id.type === "AssignmentPattern") return declaredNames(id.left);
+	if (id.type === "ArrayPattern") return id.elements.flatMap(declaredNames);
+	if (id.type === "ObjectPattern") return id.properties.flatMap((prop) => declaredNames(prop.type === "RestElement" ? prop.argument : prop.value));
+	return [];
+}
+/** Retain getters only for bindings visible between iterations, never local shadows. */
 function applyVariableTracking(ctx, variables) {
-	const trackerFor = (name) => {
-		if (name.startsWith("__")) return null;
-		variables.add(name);
-		return `${VariableTrackingFnIdentifier}(${JSON.stringify(name)}, () => eval(${JSON.stringify(name)}));`;
-	};
-	const track = (names) => names.map(trackerFor).filter((t) => !!t).join("");
-	const declaredNames = (id) => {
-		if (id.type === "Identifier") return [id.name];
-		if (id.type === "ObjectPattern") {
-			const names = [];
-			for (const prop of id.properties) {
-				if (prop.type !== "Property") continue;
-				if (prop.value.type === "Identifier") names.push(prop.value.name);
-				else if (prop.value.type === "ObjectPattern") names.push(...declaredNames(prop.value));
-				else if (prop.value.type === "AssignmentPattern" && prop.value.left.type === "Identifier") names.push(prop.value.left.name);
-			}
-			return names;
-		}
-		if (id.type === "ArrayPattern") {
-			const names = [];
-			for (const element of id.elements) {
-				if (!element) continue;
-				if (element.type === "Identifier") names.push(element.name);
-				else if (element.type === "RestElement" && element.argument.type === "Identifier") names.push(element.argument.name);
-				else if (element.type === "AssignmentPattern" && element.left.type === "Identifier") names.push(element.left.name);
-			}
-			return names;
-		}
-		return [];
-	};
-	const paramNames = (params) => {
-		const names = [];
-		for (const param of params) if (param.type === "Identifier") names.push(param.name);
-		else if (param.type === "AssignmentPattern" && param.left.type === "Identifier") names.push(param.left.name);
-		else if (param.type === "ObjectPattern") {
-			for (const prop of param.properties) if (prop.type === "Property" && prop.value.type === "Identifier") names.push(prop.value.name);
-		} else if (param.type === "ArrayPattern") {
-			for (const element of param.elements) if ((element === null || element === void 0 ? void 0 : element.type) === "Identifier") names.push(element.name);
-		}
-		return names;
-	};
-	walk(ctx.ast, (node, parent) => {
-		if (node.type === "VariableDeclaration") {
-			if (parent && (parent.type === "ForStatement" || parent.type === "ForInStatement" || parent.type === "ForOfStatement")) return;
-			const trackers = track(node.declarations.flatMap((d) => declaredNames(d.id)));
-			if (trackers) {
-				const semi = ctx.code[node.end - 1] === ";" ? "" : ";";
-				ctx.ms.appendRight(node.end, `${semi}${trackers}`);
-			}
-			return;
-		}
-		if (node.type === "FunctionDeclaration" || node.type === "ArrowFunctionExpression") {
-			const trackers = track(paramNames(node.params));
-			if (!trackers) return;
-			const body = node.body;
-			if (body.type === "BlockStatement") ctx.ms.appendRight(body.start + 1, trackers);
-			else {
-				const expressions = trackers.slice(0, -1).replaceAll(";", ",");
-				ctx.ms.prependLeft(body.start, `(${expressions}, (`);
-				ctx.ms.appendRight(body.end, "))");
-			}
+	const wrapper = ctx.ast.body.find((node) => {
+		var _node$id;
+		return node.type === "FunctionDeclaration" && ((_node$id = node.id) === null || _node$id === void 0 ? void 0 : _node$id.name) === "__fn__";
+	});
+	const root = (wrapper === null || wrapper === void 0 ? void 0 : wrapper.body) ?? ctx.ast;
+	const varBindings = /* @__PURE__ */ new Set();
+	const track = (names) => names.map((name) => `${VariableTrackingFnIdentifier}(${JSON.stringify(name)}, () => eval(${JSON.stringify(name)}));`).join("");
+	walk(root, (node, parent, ancestors) => {
+		if (node.type !== "VariableDeclaration" || ancestors.some((ancestor) => FUNCTION_TYPES$1.has(ancestor.type) || ancestor.type === "StaticBlock")) return;
+		if (node.kind !== "var" && parent !== root) return;
+		const names = node.declarations.flatMap((declaration) => declaredNames(declaration.id)).filter((name) => !name.startsWith("__"));
+		for (const name of names) variables.add(name);
+		if (node.kind === "var" && parent !== root) for (const name of names) varBindings.add(name);
+		else if (node.declarations.length > 1) ctx.ms.appendLeft(node.start, track(names));
+		else {
+			const semi = ctx.code[node.end - 1] === ";" ? "" : ";";
+			ctx.ms.appendRight(node.end, semi + track(names));
 		}
 	});
+	const entry = root.body.find((node) => !node.directive);
+	if (entry && varBindings.size) ctx.ms.appendLeft(entry.start, track([...varBindings]));
 }
 //#endregion
 //#region src/compiler/compiler.ts

package/dist/{compiler-CW1VjQsn.js → compiler-BPl-5CXh.js}

@@ -316,77 +316,43 @@ function applyToolCallTracking(ctx, calls) {
 //#endregion
 //#region src/compiler/plugins/variable-extraction.ts
 const VariableTrackingFnIdentifier = "__var__";
-/**
-* Tracks user variables so the VM can read their values between iterations:
-* - after each variable declaration: `__var__("a", () => eval("a"));`
-* - at the top of function bodies, for their parameters
-*
-* The `eval` indirection captures the variable lazily in its own scope.
-*/
+const FUNCTION_TYPES$1 = /* @__PURE__ */ new Set([
+	"FunctionDeclaration",
+	"FunctionExpression",
+	"ArrowFunctionExpression"
+]);
+function declaredNames(id) {
+	if (!id) return [];
+	if (id.type === "Identifier") return [id.name];
+	if (id.type === "RestElement") return declaredNames(id.argument);
+	if (id.type === "AssignmentPattern") return declaredNames(id.left);
+	if (id.type === "ArrayPattern") return id.elements.flatMap(declaredNames);
+	if (id.type === "ObjectPattern") return id.properties.flatMap((prop) => declaredNames(prop.type === "RestElement" ? prop.argument : prop.value));
+	return [];
+}
+/** Retain getters only for bindings visible between iterations, never local shadows. */
 function applyVariableTracking(ctx, variables) {
-	const trackerFor = (name) => {
-		if (name.startsWith("__")) return null;
-		variables.add(name);
-		return `${VariableTrackingFnIdentifier}(${JSON.stringify(name)}, () => eval(${JSON.stringify(name)}));`;
-	};
-	const track = (names) => names.map(trackerFor).filter((t) => !!t).join("");
-	const declaredNames = (id) => {
-		if (id.type === "Identifier") return [id.name];
-		if (id.type === "ObjectPattern") {
-			const names = [];
-			for (const prop of id.properties) {
-				if (prop.type !== "Property") continue;
-				if (prop.value.type === "Identifier") names.push(prop.value.name);
-				else if (prop.value.type === "ObjectPattern") names.push(...declaredNames(prop.value));
-				else if (prop.value.type === "AssignmentPattern" && prop.value.left.type === "Identifier") names.push(prop.value.left.name);
-			}
-			return names;
-		}
-		if (id.type === "ArrayPattern") {
-			const names = [];
-			for (const element of id.elements) {
-				if (!element) continue;
-				if (element.type === "Identifier") names.push(element.name);
-				else if (element.type === "RestElement" && element.argument.type === "Identifier") names.push(element.argument.name);
-				else if (element.type === "AssignmentPattern" && element.left.type === "Identifier") names.push(element.left.name);
-			}
-			return names;
-		}
-		return [];
-	};
-	const paramNames = (params) => {
-		const names = [];
-		for (const param of params) if (param.type === "Identifier") names.push(param.name);
-		else if (param.type === "AssignmentPattern" && param.left.type === "Identifier") names.push(param.left.name);
-		else if (param.type === "ObjectPattern") {
-			for (const prop of param.properties) if (prop.type === "Property" && prop.value.type === "Identifier") names.push(prop.value.name);
-		} else if (param.type === "ArrayPattern") {
-			for (const element of param.elements) if ((element === null || element === void 0 ? void 0 : element.type) === "Identifier") names.push(element.name);
-		}
-		return names;
-	};
-	walk(ctx.ast, (node, parent) => {
-		if (node.type === "VariableDeclaration") {
-			if (parent && (parent.type === "ForStatement" || parent.type === "ForInStatement" || parent.type === "ForOfStatement")) return;
-			const trackers = track(node.declarations.flatMap((d) => declaredNames(d.id)));
-			if (trackers) {
-				const semi = ctx.code[node.end - 1] === ";" ? "" : ";";
-				ctx.ms.appendRight(node.end, `${semi}${trackers}`);
-			}
-			return;
-		}
-		if (node.type === "FunctionDeclaration" || node.type === "ArrowFunctionExpression") {
-			const trackers = track(paramNames(node.params));
-			if (!trackers) return;
-			const body = node.body;
-			if (body.type === "BlockStatement") ctx.ms.appendRight(body.start + 1, trackers);
-			else {
-				const expressions = trackers.slice(0, -1).replaceAll(";", ",");
-				ctx.ms.prependLeft(body.start, `(${expressions}, (`);
-				ctx.ms.appendRight(body.end, "))");
-			}
+	const wrapper = ctx.ast.body.find((node) => {
+		var _node$id;
+		return node.type === "FunctionDeclaration" && ((_node$id = node.id) === null || _node$id === void 0 ? void 0 : _node$id.name) === "__fn__";
+	});
+	const root = (wrapper === null || wrapper === void 0 ? void 0 : wrapper.body) ?? ctx.ast;
+	const varBindings = /* @__PURE__ */ new Set();
+	const track = (names) => names.map((name) => `${VariableTrackingFnIdentifier}(${JSON.stringify(name)}, () => eval(${JSON.stringify(name)}));`).join("");
+	walk(root, (node, parent, ancestors) => {
+		if (node.type !== "VariableDeclaration" || ancestors.some((ancestor) => FUNCTION_TYPES$1.has(ancestor.type) || ancestor.type === "StaticBlock")) return;
+		if (node.kind !== "var" && parent !== root) return;
+		const names = node.declarations.flatMap((declaration) => declaredNames(declaration.id)).filter((name) => !name.startsWith("__"));
+		for (const name of names) variables.add(name);
+		if (node.kind === "var" && parent !== root) for (const name of names) varBindings.add(name);
+		else if (node.declarations.length > 1) ctx.ms.appendLeft(node.start, track(names));
+		else {
+			const semi = ctx.code[node.end - 1] === ";" ? "" : ";";
+			ctx.ms.appendRight(node.end, semi + track(names));
 		}
 	});
+	const entry = root.body.find((node) => !node.directive);
+	if (entry && varBindings.size) ctx.ms.appendLeft(entry.start, track([...varBindings]));
 }
 //#endregion
 //#region src/compiler/compiler.ts

package/dist/{component-C6Agl2ns.cjs → component-CPKXe6Ca.cjs}

@@ -1,5 +1,5 @@
-const require_compiler = require("./compiler-C-OI9sim.cjs");
-const require_jsx = require("./jsx-owqxrquU.cjs");
+const require_compiler = require("./compiler-BJJUmNS9.cjs");
+const require_jsx = require("./jsx-BJ6GWXI2.cjs");
 //#region src/component.ts
 var component_exports = /* @__PURE__ */ require_compiler.__exportAll({
 	Component: () => Component,

package/dist/{custom-client-DAsGRnYc.cjs → custom-client-B_o56CB4.cjs}

@@ -1,14 +1,14 @@
-const require_dual_modes = require("./dual-modes-DPXiuvNe.cjs");
+const require_dual_modes = require("./dual-modes-BNHlj2qM.cjs");
 const require__overArg = require("./_overArg-C5Hy7oCi.cjs");
 const require__baseIteratee = require("./_baseIteratee-CF4qP2a8.cjs");
-const require_utils = require("./utils-3ESwP3yg.cjs");
+const require_utils = require("./utils-DGQ4sRke.cjs");
 const require__getPrototype = require("./_getPrototype-DQDlNVVM.cjs");
 const require_isPlainObject = require("./isPlainObject-CSxXVKCk.cjs");
-const require_tool = require("./tool-Dy6AJpJH.cjs");
+const require_tool = require("./tool-BLTrA-sL.cjs");
 const require_errors = require("./errors-C962uJdS.cjs");
-const require_exit = require("./exit-DUM96DN1.cjs");
-const require_component = require("./component-C6Agl2ns.cjs");
-const require_truncator = require("./truncator-CLkouTCO.cjs");
+const require_exit = require("./exit-CGLXI0BH.cjs");
+const require_component = require("./component-CPKXe6Ca.cjs");
+const require_truncator = require("./truncator-DmCu9M4m.cjs");
 let _bpinternal_zui = require("@bpinternal/zui");
 let ulid = require("ulid");
 //#region ../../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/_arrayEach.js

package/dist/{custom-client-ChuZ63zp.js → custom-client-C-nDKG37.js}

@@ -1,14 +1,14 @@
-import { c as baseCreate, i as inspect, o as flatRest, r as extractType, s as copyArray, t as DualModePrompt } from "./dual-modes-DDkoCp3R.js";
+import { c as baseCreate, i as inspect, o as flatRest, r as extractType, s as copyArray, t as DualModePrompt } from "./dual-modes-BNIiNLrR.js";
 import { a as Symbol, n as isArray, o as root, r as isObjectLike } from "./_overArg-CRYbj1NH.js";
 import { D as isBuffer, E as baseUnary, M as baseAssignValue, R as isObject, T as nodeUtil, _ as baseGet, c as _getTag_default, f as getSymbols, h as Stack, k as isPrototype, s as Uint8Array, u as getAllKeys, v as toKey, x as keys, y as castPath, z as arrayMap } from "./_baseIteratee-P_firB7b.js";
 import { _ as getSymbolsIn, b as keysIn, g as getAllKeysIn, i as getErrorMessage, o as getTokenizer, x as assignValue, y as baseSlice } from "./utils-UBeAFeEI.js";
 import { t as getPrototype } from "./_getPrototype-CG7Dxrrz.js";
 import { t as isPlainObject } from "./isPlainObject-D39AzPO2.js";
-import { t as Tool } from "./tool-Bv8N4IPj.js";
+import { t as Tool } from "./tool-B6Pmuk6K.js";
 import { l as ThinkSignal, o as LoopExceededError } from "./errors-Cg1G0Awu.js";
 import { t as Exit } from "./exit-CDdxDBRT.js";
 import { n as assertValidComponent } from "./component-DGG7d_Mz.js";
-import { i as wrapContent, t as stripTruncationTags } from "./truncator-DZqEqnpD.js";
+import { i as wrapContent, t as stripTruncationTags } from "./truncator-CD7QtE84.js";
 import { z } from "@bpinternal/zui";
 import { ulid } from "ulid";
 //#region ../../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/_arrayEach.js

package/dist/{dual-modes-DPXiuvNe.cjs → dual-modes-BNHlj2qM.cjs}

@@ -1,14 +1,14 @@
-const require_compiler = require("./compiler-C-OI9sim.cjs");
-const require_common = require("./common-BnvrDnIT.cjs");
+const require_compiler = require("./compiler-BJJUmNS9.cjs");
+const require_common = require("./common-DJkSpeqw.cjs");
 const require__overArg = require("./_overArg-C5Hy7oCi.cjs");
 const require__baseIteratee = require("./_baseIteratee-CF4qP2a8.cjs");
 const require_uniq = require("./uniq-G872RWmE.cjs");
-const require_utils = require("./utils-3ESwP3yg.cjs");
+const require_utils = require("./utils-DGQ4sRke.cjs");
 const require_isPlainObject = require("./isPlainObject-CSxXVKCk.cjs");
 const require__baseForOwn = require("./_baseForOwn-BfWDTL8L.cjs");
 const require_isEmpty = require("./isEmpty-LY533Wmh.cjs");
 const require_stack_traces = require("./stack-traces-ntw_qhoy.cjs");
-const require_truncator = require("./truncator-CLkouTCO.cjs");
+const require_truncator = require("./truncator-DmCu9M4m.cjs");
 let acorn = require("acorn");
 let _bpinternal_zui = require("@bpinternal/zui");
 let bytes = require("bytes");
@@ -755,7 +755,7 @@ const renderExamples = async (examples, components, exits) => {
 	if (!examples.length) return "";
 	const registry = new ComponentRegistry(components.map(componentToProtocolDefinition));
 	const exitDefinitions = exits.map(exitToProtocolDefinition);
-	const { parseAssistantResponse } = await Promise.resolve().then(() => require("./common-BnvrDnIT.cjs")).then((n) => n.common_exports);
+	const { parseAssistantResponse } = await Promise.resolve().then(() => require("./common-DJkSpeqw.cjs")).then((n) => n.common_exports);
 	const render = (output) => {
 		var _response$diagnostics;
 		const response = parseAssistantResponse(`■start\n${output}\n■end`);
@@ -2114,6 +2114,7 @@ const SECTION_SEP = "\n";
 const NUMBER_LOCALE = new Intl.NumberFormat("en-US");
 const LONG_TEXT_LENGTH = 4096;
 const DEFAULT_OPTIONS = { tokens: 1e5 };
+const stringify = (value) => JSON.stringify(value, (_key, item) => typeof item === "bigint" ? item.toString() + "n" : item);
 function printLimitedJson(obj, maxDepth, maxLength, maxKeys) {
 	const indent = 2;
 	let currentLength = 0;
@@ -2132,7 +2133,7 @@ function printLimitedJson(obj, maxDepth, maxLength, maxKeys) {
 			}
 		}
 		if (typeof currentObj !== "object" || currentObj === null) {
-			const value = JSON.stringify(currentObj);
+			const value = stringify(currentObj);
 			currentLength += require_utils.getTokenizer().count(value);
 			return value;
 		}
@@ -2204,10 +2205,10 @@ function previewValue(value, length = LONG_TEXT_LENGTH) {
 	};
 	const previewObj = (obj) => {
 		const mapped = mapKeys(obj, (_value, key) => previewStr(key));
-		return JSON.stringify(mapped);
+		return stringify(mapped);
 	};
 	if (typeof value === "string") return "<string> " + previewStr(value);
-	if (typeof value === "object" && Array.isArray(value)) return "<array> " + previewStr(JSON.stringify(value));
+	if (typeof value === "object" && Array.isArray(value)) return "<array> " + previewStr(stringify(value));
 	if (typeof value === "object" && value instanceof Date) return "<date> " + value.toISOString();
 	if (typeof value === "object" && value instanceof RegExp) return "<regexp> " + value.toString();
 	if (typeof value === "object") return "<object> " + previewObj(value);
@@ -2247,15 +2248,15 @@ function previewArray(arr, maxStringLength = LONG_TEXT_LENGTH) {
 	lines.push("// Analysis Summary");
 	lines.push(SUBTITLE_LN);
 	const typesCount = countBy(arr, (item) => extractType(item, false));
-	const ordered = orderBy(arr.filter((item) => !isNil(item) && JSON.stringify(item).length < 100), (item) => JSON.stringify(item), "asc");
+	const ordered = orderBy(arr.filter((item) => !isNil(item) && stringify(item).length < 100), (item) => stringify(item), "asc");
 	const minValues = ordered.slice(0, 3).map((item) => previewValue(item, 10));
 	const maxValues = ordered.slice(-3).map((item) => previewValue(item, 10));
 	const uniqueItems = uniqWith(arr, isEqual);
 	const nullValues = filter(arr, isNil).length;
-	const memoryUsage = (0, bytes.default)(JSON.stringify(arr).length);
+	const memoryUsage = (0, bytes.default)(stringify(arr).length);
 	lines.push(`Total Items:     ${arr.length}`);
 	lines.push(`Unique Items:    ${uniqueItems.length}`);
-	lines.push(`Types:           ${JSON.stringify(typesCount)}`);
+	lines.push(`Types:           ${stringify(typesCount)}`);
 	lines.push(`Minimum Values:  [${minValues.join(", ")}]`);
 	lines.push(`Maximum Values:  [${maxValues.join(", ")}]`);
 	lines.push(`Memory Usage:    ${memoryUsage}`);
@@ -2278,7 +2279,7 @@ function previewObject(obj, options) {
 		const keys = Object.keys(obj);
 		const uniqueEntries = require_uniq.uniq(Object.values(obj));
 		const nilValues = filter(entries, ([, value]) => isNil(value)).length;
-		const memoryUsage = (0, bytes.default)(JSON.stringify(obj).length);
+		const memoryUsage = (0, bytes.default)(stringify(obj).length);
 		lines.push(`Total Entries:   ${NUMBER_LOCALE.format(entries.length)}`);
 		lines.push(`Keys:            ${previewValue(keys)}`);
 		lines.push(`Popular Types:   ${previewValue(typesCount)}`);
@@ -2336,7 +2337,7 @@ const inspect = (value, name, options = DEFAULT_OPTIONS) => {
 		let header = "";
 		if (name) header = `// const ${name}: ${genericType}\n`;
 		if (genericType === "Array") return header + previewArray(value, options.maxStringLength);
-		else if (genericType === "error") return header + previewError(value);
+		else if (genericType === "error") return require_utils.getTokenizer().truncate(header + previewError(value), options.tokens);
 		else if (genericType === "object") return header + previewObject(value, options);
 		else if (genericType === "boolean") return header + previewValue(value);
 		else if (typeof value === "string") {
@@ -2345,7 +2346,7 @@ const inspect = (value, name, options = DEFAULT_OPTIONS) => {
 		}
 		return header + previewValue(value);
 	} catch (err) {
-		return `Error: ${(err === null || err === void 0 ? void 0 : err.message) ?? "Unknown Error"}`;
+		return require_utils.getTokenizer().truncate(`Error: ${(err === null || err === void 0 ? void 0 : err.message) ?? "Unknown Error"}`, options.tokens);
 	}
 };
 //#endregion

package/dist/{dual-modes-DDkoCp3R.js → dual-modes-BNIiNLrR.js}

@@ -1,5 +1,5 @@
 import { n as __exportAll } from "./rolldown-runtime-EAsCxrDo.js";
-import { l as NAME_REGEX, n as parseAssistantResponse, r as replacePlaceholders } from "./common-BlRFtdf2.js";
+import { l as NAME_REGEX, n as parseAssistantResponse, r as replacePlaceholders } from "./common-DuEhn6u3.js";
 import { a as Symbol, n as isArray, r as isObjectLike } from "./_overArg-CRYbj1NH.js";
 import { A as isArrayLike, B as isSymbol, E as baseUnary, F as WeakMap, L as identity, M as baseAssignValue, O as isArguments, P as defineProperty, R as isObject, _ as baseGet, c as _getTag_default, g as arrayPush, i as mapToArray, m as arrayFilter, n as baseIsEqual, t as baseIteratee, x as keys, z as arrayMap } from "./_baseIteratee-P_firB7b.js";
 import { n as baseUniq, r as noop, t as uniq } from "./uniq-BYQ7GWwb.js";
@@ -8,7 +8,7 @@ import { t as isPlainObject } from "./isPlainObject-D39AzPO2.js";
 import { t as baseForOwn } from "./_baseForOwn-CFhWe_W6.js";
 import { t as isEmpty } from "./isEmpty-BhJRCqIF.js";
 import { t as cleanStackTrace } from "./stack-traces-BIE_5Xwf.js";
-import { i as wrapContent } from "./truncator-DZqEqnpD.js";
+import { i as wrapContent } from "./truncator-CD7QtE84.js";
 import { parse } from "acorn";
 import { transforms } from "@bpinternal/zui";
 import bytes from "bytes";
@@ -754,7 +754,7 @@ const renderExamples = async (examples, components, exits) => {
 	if (!examples.length) return "";
 	const registry = new ComponentRegistry(components.map(componentToProtocolDefinition));
 	const exitDefinitions = exits.map(exitToProtocolDefinition);
-	const { parseAssistantResponse } = await import("./common-BlRFtdf2.js").then((n) => n.t);
+	const { parseAssistantResponse } = await import("./common-DuEhn6u3.js").then((n) => n.t);
 	const render = (output) => {
 		var _response$diagnostics;
 		const response = parseAssistantResponse(`■start\n${output}\n■end`);
@@ -2113,6 +2113,7 @@ const SECTION_SEP = "\n";
 const NUMBER_LOCALE = new Intl.NumberFormat("en-US");
 const LONG_TEXT_LENGTH = 4096;
 const DEFAULT_OPTIONS = { tokens: 1e5 };
+const stringify = (value) => JSON.stringify(value, (_key, item) => typeof item === "bigint" ? item.toString() + "n" : item);
 function printLimitedJson(obj, maxDepth, maxLength, maxKeys) {
 	const indent = 2;
 	let currentLength = 0;
@@ -2131,7 +2132,7 @@ function printLimitedJson(obj, maxDepth, maxLength, maxKeys) {
 			}
 		}
 		if (typeof currentObj !== "object" || currentObj === null) {
-			const value = JSON.stringify(currentObj);
+			const value = stringify(currentObj);
 			currentLength += getTokenizer().count(value);
 			return value;
 		}
@@ -2203,10 +2204,10 @@ function previewValue(value, length = LONG_TEXT_LENGTH) {
 	};
 	const previewObj = (obj) => {
 		const mapped = mapKeys(obj, (_value, key) => previewStr(key));
-		return JSON.stringify(mapped);
+		return stringify(mapped);
 	};
 	if (typeof value === "string") return "<string> " + previewStr(value);
-	if (typeof value === "object" && Array.isArray(value)) return "<array> " + previewStr(JSON.stringify(value));
+	if (typeof value === "object" && Array.isArray(value)) return "<array> " + previewStr(stringify(value));
 	if (typeof value === "object" && value instanceof Date) return "<date> " + value.toISOString();
 	if (typeof value === "object" && value instanceof RegExp) return "<regexp> " + value.toString();
 	if (typeof value === "object") return "<object> " + previewObj(value);
@@ -2246,15 +2247,15 @@ function previewArray(arr, maxStringLength = LONG_TEXT_LENGTH) {
 	lines.push("// Analysis Summary");
 	lines.push(SUBTITLE_LN);
 	const typesCount = countBy(arr, (item) => extractType(item, false));
-	const ordered = orderBy(arr.filter((item) => !isNil(item) && JSON.stringify(item).length < 100), (item) => JSON.stringify(item), "asc");
+	const ordered = orderBy(arr.filter((item) => !isNil(item) && stringify(item).length < 100), (item) => stringify(item), "asc");
 	const minValues = ordered.slice(0, 3).map((item) => previewValue(item, 10));
 	const maxValues = ordered.slice(-3).map((item) => previewValue(item, 10));
 	const uniqueItems = uniqWith(arr, isEqual);
 	const nullValues = filter(arr, isNil).length;
-	const memoryUsage = bytes(JSON.stringify(arr).length);
+	const memoryUsage = bytes(stringify(arr).length);
 	lines.push(`Total Items:     ${arr.length}`);
 	lines.push(`Unique Items:    ${uniqueItems.length}`);
-	lines.push(`Types:           ${JSON.stringify(typesCount)}`);
+	lines.push(`Types:           ${stringify(typesCount)}`);
 	lines.push(`Minimum Values:  [${minValues.join(", ")}]`);
 	lines.push(`Maximum Values:  [${maxValues.join(", ")}]`);
 	lines.push(`Memory Usage:    ${memoryUsage}`);
@@ -2277,7 +2278,7 @@ function previewObject(obj, options) {
 		const keys = Object.keys(obj);
 		const uniqueEntries = uniq(Object.values(obj));
 		const nilValues = filter(entries, ([, value]) => isNil(value)).length;
-		const memoryUsage = bytes(JSON.stringify(obj).length);
+		const memoryUsage = bytes(stringify(obj).length);
 		lines.push(`Total Entries:   ${NUMBER_LOCALE.format(entries.length)}`);
 		lines.push(`Keys:            ${previewValue(keys)}`);
 		lines.push(`Popular Types:   ${previewValue(typesCount)}`);
@@ -2335,7 +2336,7 @@ const inspect = (value, name, options = DEFAULT_OPTIONS) => {
 		let header = "";
 		if (name) header = `// const ${name}: ${genericType}\n`;
 		if (genericType === "Array") return header + previewArray(value, options.maxStringLength);
-		else if (genericType === "error") return header + previewError(value);
+		else if (genericType === "error") return getTokenizer().truncate(header + previewError(value), options.tokens);
 		else if (genericType === "object") return header + previewObject(value, options);
 		else if (genericType === "boolean") return header + previewValue(value);
 		else if (typeof value === "string") {
@@ -2344,7 +2345,7 @@ const inspect = (value, name, options = DEFAULT_OPTIONS) => {
 		}
 		return header + previewValue(value);
 	} catch (err) {
-		return `Error: ${(err === null || err === void 0 ? void 0 : err.message) ?? "Unknown Error"}`;
+		return getTokenizer().truncate(`Error: ${(err === null || err === void 0 ? void 0 : err.message) ?? "Unknown Error"}`, options.tokens);
 	}
 };
 //#endregion

package/dist/{execute-DM2-7y86.cjs → execute-0bvqakr7.cjs}

@@ -1,13 +1,13 @@
-const require_compiler = require("./compiler-C-OI9sim.cjs");
-const require_dual_modes = require("./dual-modes-DPXiuvNe.cjs");
-const require_common = require("./common-BnvrDnIT.cjs");
-const require_custom_client = require("./custom-client-DAsGRnYc.cjs");
-const require_utils = require("./utils-3ESwP3yg.cjs");
-const require_jsx = require("./jsx-owqxrquU.cjs");
+const require_compiler = require("./compiler-BJJUmNS9.cjs");
+const require_dual_modes = require("./dual-modes-BNHlj2qM.cjs");
+const require_common = require("./common-DJkSpeqw.cjs");
+const require_custom_client = require("./custom-client-B_o56CB4.cjs");
+const require_utils = require("./utils-DGQ4sRke.cjs");
+const require_jsx = require("./jsx-BJ6GWXI2.cjs");
 const require_stack_traces = require("./stack-traces-ntw_qhoy.cjs");
 const require_errors = require("./errors-C962uJdS.cjs");
-const require_truncator = require("./truncator-CLkouTCO.cjs");
-const require_vm = require("./vm-D6s1zF6M.cjs");
+const require_truncator = require("./truncator-DmCu9M4m.cjs");
+const require_vm = require("./vm-X0uTj5Aa.cjs");
 let _bpinternal_zui = require("@bpinternal/zui");
 let ulid = require("ulid");
 let _botpress_client = require("@botpress/client");
@@ -126,19 +126,21 @@ const STREAM_INACTIVITY_TIMEOUT = 18e4;
 /** A syntactically valid prefix must not execute when generation did not finish successfully. */
 const assertSuccessfulGeneration = (metadata) => {
 	if (metadata.provider === "unknown") throw new require_errors.CognitiveError("LLM generation failed: received error metadata with unknown provider");
-	if (metadata.stopReason === "max_tokens" || metadata.stopReason === "content_filter") throw new require_errors.CognitiveError(`LLM generation did not complete: stopReason=${metadata.stopReason}`);
+	if (metadata.stopReason === "max_tokens" || metadata.stopReason === "content_filter" || metadata.stopReason === "other") throw new require_errors.CognitiveError(`LLM generation did not complete: stopReason=${metadata.stopReason}`);
 };
 const getModelOutputLimit = (inputLength) => clamp(RESPONSE_LENGTH_BUFFER.PERCENTAGE * inputLength, RESPONSE_LENGTH_BUFFER.MIN_TOKENS, RESPONSE_LENGTH_BUFFER.MAX_TOKENS);
 const generateCode = async ({ iteration, ctx, cognitive, controller, metadata, onSend, onSendDelta, onRunStart }) => {
 	const startedAt = Date.now();
 	const traces = iteration.traces;
-	const modelRef = Array.isArray(iteration.model) ? iteration.model[0] : iteration.model;
-	const model = await cognitive.getModelDetails(modelRef).catch((thrown) => {
+	controller.signal.throwIfAborted();
+	const modelRefs = Array.isArray(iteration.model) ? iteration.model : [iteration.model];
+	const models = await Promise.all(modelRefs.map((modelRef) => cognitive.getModelDetails(modelRef).catch((thrown) => {
 		throw new require_errors.CognitiveError(`Failed to fetch model details for model "${modelRef}": ${require_utils.getErrorMessage(thrown)}`);
-	});
-	let modelLimit = Math.max(model.input.maxTokens, 8e3);
-	if (ctx.maxTokens) modelLimit = Math.min(ctx.maxTokens, modelLimit);
-	const responseLengthBuffer = getModelOutputLimit(modelLimit);
+	})));
+	controller.signal.throwIfAborted();
+	const model = models[0];
+	const modelLimit = Math.min(...models.map((model) => model.input.maxTokens), ctx.maxTokens ?? Infinity);
+	const responseLengthBuffer = Math.min(getModelOutputLimit(modelLimit), Math.floor(modelLimit / 2));
 	if (iteration.tokens) iteration.tokens.limit = modelLimit;
 	let messages;
 	try {
@@ -358,6 +360,7 @@ const generateCode = async ({ iteration, ctx, cognitive, controller, metadata, o
 		const response = await cognitive.generateText(input, { signal: controller.signal }).catch((thrown) => {
 			throw new require_errors.CognitiveError(`LLM generation failed: ${require_utils.getErrorMessage(thrown)}`);
 		});
+		controller.signal.throwIfAborted();
 		if (response.error) throw new require_errors.CognitiveError(`LLM generation failed: ${response.error}`);
 		if (!response.output) throw new require_errors.CognitiveError("LLM did not return any text output");
 		responseMetadata = response.metadata;
@@ -568,8 +571,10 @@ const finalizeIteration = async ({ iteration, status, controller, onIterationEnd
 //#region src/runtime/tool-wrapper.ts
 const SLOW_TOOL_WARNING = (0, ms.default)("15s");
 function wrapTool({ chat, tool, traces, object, iteration, beforeHook, afterHook, controller }) {
-	const getToolInput = (input) => tool.zInput.safeParse(input).data ?? input;
+	const traceSchema = tool.input ? require_utils.fromJSONSchemaCompat(tool.input) : void 0;
+	const getToolInput = (input) => (traceSchema === null || traceSchema === void 0 ? void 0 : traceSchema.safeParse(input).data) ?? input;
 	return async function(input) {
+		controller.signal.throwIfAborted();
 		const toolCallId = `tcall_${(0, ulid.ulid)()}`;
 		const originalInput = input;
 		let effectiveInput = input;
@@ -645,6 +650,7 @@ function wrapTool({ chat, tool, traces, object, iteration, beforeHook, afterHook
 				object,
 				toolCallId
 			}));
+			controller.signal.throwIfAborted();
 			if (typeof (beforeRes === null || beforeRes === void 0 ? void 0 : beforeRes.input) !== "undefined") effectiveInput = beforeRes.input;
 			output = await tool.execute(effectiveInput, {
 				callId: toolCallId,
@@ -661,9 +667,16 @@ function wrapTool({ chat, tool, traces, object, iteration, beforeHook, afterHook
 			}));
 			if (typeof (afterRes === null || afterRes === void 0 ? void 0 : afterRes.output) !== "undefined") output = afterRes.output;
 		} catch (err) {
-			if (!await handleSignals(err)) {
+			try {
+				if (!await handleSignals(err)) {
+					success = false;
+					error = err;
+					output = void 0;
+				}
+			} catch (hookError) {
 				success = false;
-				error = err;
+				error = hookError;
+				output = void 0;
 			}
 		} finally {
 			clearTimeout(alertSlowTool);