llmz 0.8.0 → 0.9.1

package/dist/runtime/fixtures/protocol-languages.d.ts

@@ -0,0 +1,74 @@
+/** Synthetic multilingual inputs shared by parser/execution and live model tests. */
+export declare const protocolLanguages: readonly [{
+    readonly language: 'English';
+    readonly hello: 'Hello!';
+    readonly question: 'Help me choose a plan.';
+    readonly reply: 'How many conversations do you expect per month?';
+    readonly checking: 'Checking.';
+}, {
+    readonly language: 'French';
+    readonly hello: 'Bonjour !';
+    readonly question: 'Aidez-moi à choisir un forfait.';
+    readonly reply: 'Combien de conversations prévoyez-vous par mois ?';
+    readonly checking: 'Je vérifie.';
+}, {
+    readonly language: 'Spanish';
+    readonly hello: '¡Hola!';
+    readonly question: 'Ayúdame a elegir un plan.';
+    readonly reply: '¿Cuántas conversaciones esperas al mes?';
+    readonly checking: 'Estoy comprobando.';
+}, {
+    readonly language: 'German';
+    readonly hello: 'Hallo!';
+    readonly question: 'Hilf mir, einen Tarif auszuwählen.';
+    readonly reply: 'Wie viele Gespräche erwartest du pro Monat?';
+    readonly checking: 'Ich prüfe das.';
+}, {
+    readonly language: 'Portuguese';
+    readonly hello: 'Olá!';
+    readonly question: 'Ajude-me a escolher um plano.';
+    readonly reply: 'Quantas conversas você espera por mês?';
+    readonly checking: 'Estou verificando.';
+}, {
+    readonly language: 'Italian';
+    readonly hello: 'Ciao!';
+    readonly question: 'Aiutami a scegliere un piano.';
+    readonly reply: 'Quante conversazioni prevedi al mese?';
+    readonly checking: 'Sto controllando.';
+}, {
+    readonly language: 'Arabic';
+    readonly hello: 'مرحباً!';
+    readonly question: 'ساعدني في اختيار خطة.';
+    readonly reply: 'كم محادثة تتوقع في الشهر؟';
+    readonly checking: 'أتحقق الآن.';
+}, {
+    readonly language: 'Chinese';
+    readonly hello: '你好！';
+    readonly question: '帮我选择一个套餐。';
+    readonly reply: '你预计每月有多少次对话？';
+    readonly checking: '正在查询。';
+}, {
+    readonly language: 'Japanese';
+    readonly hello: 'こんにちは！';
+    readonly question: 'プラン選びを手伝ってください。';
+    readonly reply: '月に何件の会話を想定していますか？';
+    readonly checking: '確認しています。';
+}, {
+    readonly language: 'Korean';
+    readonly hello: '안녕하세요!';
+    readonly question: '요금제 선택을 도와주세요.';
+    readonly reply: '월별 대화 수는 얼마나 예상하시나요?';
+    readonly checking: '확인 중입니다.';
+}, {
+    readonly language: 'Hindi';
+    readonly hello: 'नमस्ते!';
+    readonly question: 'एक योजना चुनने में मेरी मदद करें।';
+    readonly reply: 'आपको हर महीने कितनी बातचीत की उम्मीद है?';
+    readonly checking: 'मैं जाँच कर रहा हूँ।';
+}, {
+    readonly language: 'Russian';
+    readonly hello: 'Здравствуйте!';
+    readonly question: 'Помогите мне выбрать тариф.';
+    readonly reply: 'Сколько разговоров вы ожидаете в месяц?';
+    readonly checking: 'Проверяю.';
+}];

package/dist/runtime/fixtures/qwen-protocol-failures.d.ts

@@ -0,0 +1,29 @@
+export declare const qwenProtocolFailures: readonly [{
+    readonly name: 'unmarked-estimation-answer';
+    readonly output: 'To estimate monthly conversations, look at your recent conversation totals rather than counting individual messages or visits. Check how many complete conversations your team handled in the last month (or average across a few months) and use that number as your expected monthly volume.\n■next=listen\n';
+    readonly reply: 'Please tell me your expected monthly conversation volume.';
+}, {
+    readonly name: 'thinking-close-before-message';
+    readonly output: 'The user wants help choosing a plan. According to the instructions, when asked to choose a plan, I should first ask for expected monthly conversations. I need to ask one question at a time and never recommend a plan before knowing that volume.\n</think>\n\n■send=message\nSure! To point you to the right plan, how many conversations do you expect to handle each month?\n■next=listen';
+    readonly reply: 'Sure! To point you to the right plan, how many conversations do you expect to handle each month?';
+}, {
+    readonly name: 'wrong-bullet-instead-of-send';
+    readonly output: '● Hello! It looks like you just sent a greeting. How can I help you today? I can assist with choosing a plan or answering questions about Cedar Desk.\n■next=listen';
+    readonly reply: 'Please tell me your expected monthly conversation volume.';
+}, {
+    readonly name: 'xml-tool-call-instead-of-protocol';
+    readonly output: '<tool_call>\n<function=■send=message>\nI can help with that. To start, how many conversations do you expect to have each month?\n</function>\n</tool_call>';
+    readonly reply: 'Please tell me your expected monthly conversation volume.';
+}, {
+    readonly name: 'repeated-messages-after-listen';
+    readonly output: "A conversation is a back-and-forth between a visitor and your team, not a single message or page visit. To estimate monthly conversations, look at your recent conversation totals (for example, last month's total) and use that as your baseline, adjusting if your traffic is trending up or down.\n■next=listen\n■send=message\nA conversation is a back-and-forth between a visitor and your team, not a single message or page visit. To estimate monthly conversations, look at your recent conversation totals (for example, last month's total) and use that as your baseline, adjusting if your traffic is trending up or down.\n■next=listen\n■next=listen\n■send=message\nA conversation is a back-and-forth between a visitor and your team, not a single message or page visit. To estimate monthly conversations, look at your recent conversation totals (for example, last month's total) and use that as your baseline, adjusting if your traffic is trending up or down.\n■next=listen\n■next=listen\n■send=message\nA conversation is a back-and-forth between a visitor and your team, not a single message or page visit. To estimate monthly conversations, look at your recent conversation totals (for example, last month's total) and use that as your baseline, adjusting if your traffic is trending up or down.\n■next=listen\n■next=listen";
+    readonly reply: 'Please tell me your expected monthly conversation volume.';
+}, {
+    readonly name: 'premature-listen-before-thinking-close';
+    readonly output: 'I need to understand your expected monthly conversations before I can recommend a plan.\n\nHow many conversations do you expect to handle each month?\n■next=listen\n</think>\n\n■send=message\nI can help with that. To get started, how many conversations do you expect to handle each month?\n■next=listen';
+    readonly reply: 'Please tell me your expected monthly conversation volume.';
+}, {
+    readonly name: 'draft-answer-before-explicit-send';
+    readonly output: 'The user wants help choosing a plan. According to the instructions, I should first ask for expected monthly conversations. Messages, visits, and conversations are different units.\nI should ask one question at a time.\n\n"Great! To point you to the right plan, could you tell me how many conversations you expect to have each month? (Just a rough estimate is fine.)"\n\nThen wait for the user.\n\n■send=message\nGreat! To point you to the right plan, could you tell me roughly how many conversations you expect to have each month?\n■next=listen';
+    readonly reply: 'Great! To point you to the right plan, could you tell me roughly how many conversations you expect to have each month?';
+}];

package/dist/runtime/generate.d.ts

@@ -9,8 +9,7 @@ type GenerateCodeProps = {
     controller: AbortController;
     metadata?: Record<string, string>;
     /**
-     * Called for each completed `■send` block. On streaming clients this fires
-     * while the model is still generating — messages are delivered progressively.
+     * Called for each send after a complete, valid response and successful transport.
      */
     onSend?: (send: ParsedSend, metadata: MessageMetadata) => Promise<void>;
     /**
@@ -24,12 +23,6 @@ type GenerateCodeProps = {
      * being generated (streaming clients only). Used to pre-warm the VM.
      */
     onRunStart?: () => void;
-    /**
-     * Called with the code of the first `■run` block the moment it is fully
-     * parsed — the rest of the response (e.g. `■next`) may still be streaming
-     * (streaming clients only). Used to start executing the code early.
-     */
-    onRunComplete?: (code: string) => void;
 };
-export declare const generateCode: ({ iteration, ctx, cognitive, controller, metadata, onSend, onSendDelta, onRunStart, onRunComplete, }: GenerateCodeProps) => Promise<void>;
+export declare const generateCode: ({ iteration, ctx, cognitive, controller, metadata, onSend, onSendDelta, onRunStart, }: GenerateCodeProps) => Promise<void>;
 export {};

package/dist/runtime/generate.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/runtime/protocol-adherence.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/runtime/types.d.ts

@@ -124,13 +124,10 @@ type Options = Partial<Pick<Context, 'loop' | 'timeout'>> & {
      */
     maxTimeToFirstToken?: number;
     /**
-     * Allow Cognitive to restart failed streams on another model. No buffered
-     * delivery and no final delivery queue: delta text and completed Chat.handler
-     * sends stream immediately during generation. A reset-only delta (restart:
-     * true) on Chat.onMessageDelta invalidates ALL current-iteration messages,
-     * including completed sends — consumers must retract/replace them (not safe for
-     * irreversible transports). Only TOOL/CODE execution waits for a successful
-     * stream. Streaming-only; defaults to false, preserving progressive sending.
+     * Allow Cognitive to restart a failed stream on another model. Previews remain
+     * live and are retracted with a restart delta before replacement output.
+     * Completed sends and code always wait for a valid response and successful
+     * transport. Streaming-only; defaults to false.
      */
     midStreamFallback?: boolean;
     /**

package/dist/{tool-Bv8N4IPj.js → tool-B6Pmuk6K.js}

@@ -164,13 +164,15 @@ var tool_exports = /* @__PURE__ */ __exportAll({ Tool: () => Tool });
 *
 * Tools automatically handle:
 * - **Input validation**: Invalid inputs throw descriptive errors
-* - **Output validation**: Outputs are validated but invalid outputs are still returned
+* - **Output descriptions**: Output schemas describe likely results; handler output is returned unchanged
 * - **Handler errors**: Exceptions in handlers are caught and can trigger retries
 * - **Type coercion**: Basic type coercion where possible
 *
 */
 var Tool = class Tool {
 	_staticInputValues;
+	_inputSchema;
+	_outputSchema;
 	name;
 	aliases = [];
 	description;
@@ -241,23 +243,14 @@ var Tool = class Tool {
 	* @internal
 	*/
 	get zInput() {
-		let input;
-		if (this.input) try {
-			input = transforms.fromJSONSchema(this.input);
-		} catch {
-			input = transforms.fromJSONSchemaLegacy(this.input);
-		}
-		else input = z.any();
-		if (!isEmpty(this._staticInputValues)) {
-			if (z.is.zuiObject(input)) {
-				const inputExtensions = convertObjectToZuiLiterals(this._staticInputValues);
-				input = input.extend(inputExtensions);
-			} else if (z.is.zuiArray(input)) {
-				const inputExtensions = convertObjectToZuiLiterals(this._staticInputValues);
-				input = z.array(input.element.extend(inputExtensions));
-			} else input = convertObjectToZuiLiterals(this._staticInputValues);
-		}
-		return input;
+		const input = this._inputSchema ?? (this.input ? fromJSONSchemaCompat(this.input) : z.any());
+		if (isEmpty(this._staticInputValues)) return input;
+		const shape = this.input ? fromJSONSchemaCompat(this.input) : z.any();
+		let overrides;
+		if (z.is.zuiObject(shape)) overrides = z.object(convertObjectToZuiLiterals(this._staticInputValues)).passthrough();
+		else if (z.is.zuiArray(shape)) overrides = z.array(z.object(convertObjectToZuiLiterals(this._staticInputValues)).passthrough());
+		else overrides = convertObjectToZuiLiterals(this._staticInputValues);
+		return overrides.pipe(input);
 	}
 	/**
 	* Gets the output schema for validation.
@@ -354,18 +347,8 @@ var Tool = class Tool {
 	clone(props = {}) {
 		try {
 			var _props$input, _props$output;
-			let zInput;
-			if (this.input) try {
-				zInput = transforms.fromJSONSchema(this.input);
-			} catch {
-				zInput = transforms.fromJSONSchemaLegacy(this.input);
-			}
-			let zOutput;
-			if (this.output) try {
-				zOutput = transforms.fromJSONSchema(this.output);
-			} catch {
-				zOutput = transforms.fromJSONSchemaLegacy(this.output);
-			}
+			const zInput = this._inputSchema ?? (this.input ? fromJSONSchemaCompat(this.input) : void 0);
+			const zOutput = this._outputSchema ?? (this.output ? fromJSONSchemaCompat(this.output) : void 0);
 			return new Tool({
 				name: props.name ?? this.name,
 				aliases: props.aliases ?? [...this.aliases],
@@ -388,7 +371,7 @@ var Tool = class Tool {
 	* @param props.name - Unique tool name (must be valid TypeScript identifier)
 	* @param props.description - Human-readable description for the LLM
 	* @param props.input - Zui/Zod schema for input validation (optional)
-	* @param props.output - Zui/Zod schema for output validation (optional)
+	* @param props.output - Zui/Zod schema describing output (optional)
 	* @param props.handler - Async function that implements the tool logic
 	* @param props.aliases - Alternative names for the tool (optional)
 	* @param props.metadata - Additional metadata for the tool (optional)
@@ -465,21 +448,25 @@ var Tool = class Tool {
 		if (props.aliases !== void 0 && !Array.isArray(props.aliases)) throw new Error(`Invalid aliases for tool ${props.name}. Expected an array, but got type "${typeof props.aliases}"`);
 		if (props.aliases && props.aliases.some((alias) => !isValidIdentifier(alias))) throw new Error(`Invalid aliases for tool ${props.name}. Expected an array of valid identifiers.`);
 		if (typeof props.input !== "undefined") {
-			if (isZuiSchema(props.input)) try {
-				this.input = transforms.toJSONSchema(props.input);
-			} catch {
-				this.input = transforms.toJSONSchemaLegacy(props.input);
-			}
-			else if (isJsonSchema(props.input)) this.input = props.input;
+			if (isZuiSchema(props.input)) {
+				this._inputSchema = props.input;
+				try {
+					this.input = transforms.toJSONSchema(props.input);
+				} catch {
+					this.input = transforms.toJSONSchemaLegacy(props.input);
+				}
+			} else if (isJsonSchema(props.input)) this.input = props.input;
 			else throw new Error(`Invalid input schema for tool ${props.name}. Expected a ZodType or JSONSchema, but got type "${typeof props.input}"`);
 		}
 		if (typeof props.output !== "undefined") {
-			if (isZuiSchema(props.output)) try {
-				this.output = transforms.toJSONSchema(props.output);
-			} catch {
-				this.output = transforms.toJSONSchemaLegacy(props.output);
-			}
-			else if (isJsonSchema(props.output)) this.output = props.output;
+			if (isZuiSchema(props.output)) {
+				this._outputSchema = props.output;
+				try {
+					this.output = transforms.toJSONSchema(props.output);
+				} catch {
+					this.output = transforms.toJSONSchemaLegacy(props.output);
+				}
+			} else if (isJsonSchema(props.output)) this.output = props.output;
 			else throw new Error(`Invalid output schema for tool ${props.name}. Expected a ZodType or JSONSchema, but got type "${typeof props.output}"`);
 		}
 		this.name = props.name;
@@ -511,7 +498,7 @@ var Tool = class Tool {
 	/**
 	* Executes the tool with the given input and context.
 	*
-	* This method handles input validation, retry logic, output validation, and error handling.
+	* This method handles input validation, retry logic, and error handling.
 	* It's called internally by the LLMz execution engine when generated code calls the tool.
 	*
 	* @param input - Input data to pass to the tool handler
@@ -533,17 +520,16 @@ var Tool = class Tool {
 	* @internal This method is primarily used internally by the LLMz execution engine
 	*/
 	async execute(rawInput, ctx, chat) {
-		const input = this.zInput._def.typeName === "ZodObject" ? rawInput ?? {} : rawInput;
-		const pInput = this.zInput.safeParse(input);
+		const shape = this.input ? fromJSONSchemaCompat(this.input) : z.any();
+		const input = z.is.zuiObject(shape) ? rawInput ?? {} : rawInput;
+		const pInput = await this.zInput.safeParseAsync(input);
 		if (!pInput.success) throw new Error(`Tool "${this.name}" received invalid input: ${pInput.error.message}`);
 		let attempt = 0;
 		let yieldedCount = 0;
 		while (attempt < this.MAX_RETRIES) try {
-			const result = await this._executeHandler(pInput.data, ctx, chat, yieldedCount, (n) => {
+			return await this._executeHandler(pInput.data, ctx, chat, yieldedCount, (n) => {
 				yieldedCount = n;
 			});
-			const pOutput = this.zOutput.safeParse(result);
-			return pOutput.success ? pOutput.data : result;
 		} catch (err) {
 			var _this$retry;
 			if (!await ((_this$retry = this.retry) === null || _this$retry === void 0 ? void 0 : _this$retry.call(this, {

package/dist/{tool-Dy6AJpJH.cjs → tool-BLTrA-sL.cjs}

@@ -1,8 +1,8 @@
-const require_compiler = require("./compiler-C-OI9sim.cjs");
+const require_compiler = require("./compiler-BJJUmNS9.cjs");
 const require_uniq = require("./uniq-G872RWmE.cjs");
-const require_utils = require("./utils-3ESwP3yg.cjs");
+const require_utils = require("./utils-DGQ4sRke.cjs");
 const require_isEmpty = require("./isEmpty-LY533Wmh.cjs");
-const require_typings = require("./typings-3lsqCbkn.cjs");
+const require_typings = require("./typings-DXJMu59O.cjs");
 let _bpinternal_zui = require("@bpinternal/zui");
 //#region src/convert.ts
 function convertObjectToZuiLiterals(obj, nested = false) {
@@ -164,13 +164,15 @@ var tool_exports = /* @__PURE__ */ require_compiler.__exportAll({ Tool: () => To
 *
 * Tools automatically handle:
 * - **Input validation**: Invalid inputs throw descriptive errors
-* - **Output validation**: Outputs are validated but invalid outputs are still returned
+* - **Output descriptions**: Output schemas describe likely results; handler output is returned unchanged
 * - **Handler errors**: Exceptions in handlers are caught and can trigger retries
 * - **Type coercion**: Basic type coercion where possible
 *
 */
 var Tool = class Tool {
 	_staticInputValues;
+	_inputSchema;
+	_outputSchema;
 	name;
 	aliases = [];
 	description;
@@ -241,23 +243,14 @@ var Tool = class Tool {
 	* @internal
 	*/
 	get zInput() {
-		let input;
-		if (this.input) try {
-			input = _bpinternal_zui.transforms.fromJSONSchema(this.input);
-		} catch {
-			input = _bpinternal_zui.transforms.fromJSONSchemaLegacy(this.input);
-		}
-		else input = _bpinternal_zui.z.any();
-		if (!require_isEmpty.isEmpty(this._staticInputValues)) {
-			if (_bpinternal_zui.z.is.zuiObject(input)) {
-				const inputExtensions = convertObjectToZuiLiterals(this._staticInputValues);
-				input = input.extend(inputExtensions);
-			} else if (_bpinternal_zui.z.is.zuiArray(input)) {
-				const inputExtensions = convertObjectToZuiLiterals(this._staticInputValues);
-				input = _bpinternal_zui.z.array(input.element.extend(inputExtensions));
-			} else input = convertObjectToZuiLiterals(this._staticInputValues);
-		}
-		return input;
+		const input = this._inputSchema ?? (this.input ? require_utils.fromJSONSchemaCompat(this.input) : _bpinternal_zui.z.any());
+		if (require_isEmpty.isEmpty(this._staticInputValues)) return input;
+		const shape = this.input ? require_utils.fromJSONSchemaCompat(this.input) : _bpinternal_zui.z.any();
+		let overrides;
+		if (_bpinternal_zui.z.is.zuiObject(shape)) overrides = _bpinternal_zui.z.object(convertObjectToZuiLiterals(this._staticInputValues)).passthrough();
+		else if (_bpinternal_zui.z.is.zuiArray(shape)) overrides = _bpinternal_zui.z.array(_bpinternal_zui.z.object(convertObjectToZuiLiterals(this._staticInputValues)).passthrough());
+		else overrides = convertObjectToZuiLiterals(this._staticInputValues);
+		return overrides.pipe(input);
 	}
 	/**
 	* Gets the output schema for validation.
@@ -354,18 +347,8 @@ var Tool = class Tool {
 	clone(props = {}) {
 		try {
 			var _props$input, _props$output;
-			let zInput;
-			if (this.input) try {
-				zInput = _bpinternal_zui.transforms.fromJSONSchema(this.input);
-			} catch {
-				zInput = _bpinternal_zui.transforms.fromJSONSchemaLegacy(this.input);
-			}
-			let zOutput;
-			if (this.output) try {
-				zOutput = _bpinternal_zui.transforms.fromJSONSchema(this.output);
-			} catch {
-				zOutput = _bpinternal_zui.transforms.fromJSONSchemaLegacy(this.output);
-			}
+			const zInput = this._inputSchema ?? (this.input ? require_utils.fromJSONSchemaCompat(this.input) : void 0);
+			const zOutput = this._outputSchema ?? (this.output ? require_utils.fromJSONSchemaCompat(this.output) : void 0);
 			return new Tool({
 				name: props.name ?? this.name,
 				aliases: props.aliases ?? [...this.aliases],
@@ -388,7 +371,7 @@ var Tool = class Tool {
 	* @param props.name - Unique tool name (must be valid TypeScript identifier)
 	* @param props.description - Human-readable description for the LLM
 	* @param props.input - Zui/Zod schema for input validation (optional)
-	* @param props.output - Zui/Zod schema for output validation (optional)
+	* @param props.output - Zui/Zod schema describing output (optional)
 	* @param props.handler - Async function that implements the tool logic
 	* @param props.aliases - Alternative names for the tool (optional)
 	* @param props.metadata - Additional metadata for the tool (optional)
@@ -465,21 +448,25 @@ var Tool = class Tool {
 		if (props.aliases !== void 0 && !Array.isArray(props.aliases)) throw new Error(`Invalid aliases for tool ${props.name}. Expected an array, but got type "${typeof props.aliases}"`);
 		if (props.aliases && props.aliases.some((alias) => !require_utils.isValidIdentifier(alias))) throw new Error(`Invalid aliases for tool ${props.name}. Expected an array of valid identifiers.`);
 		if (typeof props.input !== "undefined") {
-			if (require_utils.isZuiSchema(props.input)) try {
-				this.input = _bpinternal_zui.transforms.toJSONSchema(props.input);
-			} catch {
-				this.input = _bpinternal_zui.transforms.toJSONSchemaLegacy(props.input);
-			}
-			else if (require_utils.isJsonSchema(props.input)) this.input = props.input;
+			if (require_utils.isZuiSchema(props.input)) {
+				this._inputSchema = props.input;
+				try {
+					this.input = _bpinternal_zui.transforms.toJSONSchema(props.input);
+				} catch {
+					this.input = _bpinternal_zui.transforms.toJSONSchemaLegacy(props.input);
+				}
+			} else if (require_utils.isJsonSchema(props.input)) this.input = props.input;
 			else throw new Error(`Invalid input schema for tool ${props.name}. Expected a ZodType or JSONSchema, but got type "${typeof props.input}"`);
 		}
 		if (typeof props.output !== "undefined") {
-			if (require_utils.isZuiSchema(props.output)) try {
-				this.output = _bpinternal_zui.transforms.toJSONSchema(props.output);
-			} catch {
-				this.output = _bpinternal_zui.transforms.toJSONSchemaLegacy(props.output);
-			}
-			else if (require_utils.isJsonSchema(props.output)) this.output = props.output;
+			if (require_utils.isZuiSchema(props.output)) {
+				this._outputSchema = props.output;
+				try {
+					this.output = _bpinternal_zui.transforms.toJSONSchema(props.output);
+				} catch {
+					this.output = _bpinternal_zui.transforms.toJSONSchemaLegacy(props.output);
+				}
+			} else if (require_utils.isJsonSchema(props.output)) this.output = props.output;
 			else throw new Error(`Invalid output schema for tool ${props.name}. Expected a ZodType or JSONSchema, but got type "${typeof props.output}"`);
 		}
 		this.name = props.name;
@@ -511,7 +498,7 @@ var Tool = class Tool {
 	/**
 	* Executes the tool with the given input and context.
 	*
-	* This method handles input validation, retry logic, output validation, and error handling.
+	* This method handles input validation, retry logic, and error handling.
 	* It's called internally by the LLMz execution engine when generated code calls the tool.
 	*
 	* @param input - Input data to pass to the tool handler
@@ -533,17 +520,16 @@ var Tool = class Tool {
 	* @internal This method is primarily used internally by the LLMz execution engine
 	*/
 	async execute(rawInput, ctx, chat) {
-		const input = this.zInput._def.typeName === "ZodObject" ? rawInput ?? {} : rawInput;
-		const pInput = this.zInput.safeParse(input);
+		const shape = this.input ? require_utils.fromJSONSchemaCompat(this.input) : _bpinternal_zui.z.any();
+		const input = _bpinternal_zui.z.is.zuiObject(shape) ? rawInput ?? {} : rawInput;
+		const pInput = await this.zInput.safeParseAsync(input);
 		if (!pInput.success) throw new Error(`Tool "${this.name}" received invalid input: ${pInput.error.message}`);
 		let attempt = 0;
 		let yieldedCount = 0;
 		while (attempt < this.MAX_RETRIES) try {
-			const result = await this._executeHandler(pInput.data, ctx, chat, yieldedCount, (n) => {
+			return await this._executeHandler(pInput.data, ctx, chat, yieldedCount, (n) => {
 				yieldedCount = n;
 			});
-			const pOutput = this.zOutput.safeParse(result);
-			return pOutput.success ? pOutput.data : result;
 		} catch (err) {
 			var _this$retry;
 			if (!await ((_this$retry = this.retry) === null || _this$retry === void 0 ? void 0 : _this$retry.call(this, {

package/dist/tool.d.ts

@@ -200,13 +200,15 @@ export declare namespace Tool {
  *
  * Tools automatically handle:
  * - **Input validation**: Invalid inputs throw descriptive errors
- * - **Output validation**: Outputs are validated but invalid outputs are still returned
+ * - **Output descriptions**: Output schemas describe likely results; handler output is returned unchanged
  * - **Handler errors**: Exceptions in handlers are caught and can trigger retries
  * - **Type coercion**: Basic type coercion where possible
  *
  */
 export declare class Tool<I extends z.ZodType = z.ZodType, O extends z.ZodType = z.ZodType> implements Serializable<Tool.JSON> {
     private _staticInputValues?;
+    private _inputSchema?;
+    private _outputSchema?;
     name: string;
     aliases: string[];
     description?: string;
@@ -364,7 +366,7 @@ export declare class Tool<I extends z.ZodType = z.ZodType, O extends z.ZodType =
      * @param props.name - Unique tool name (must be valid TypeScript identifier)
      * @param props.description - Human-readable description for the LLM
      * @param props.input - Zui/Zod schema for input validation (optional)
-     * @param props.output - Zui/Zod schema for output validation (optional)
+     * @param props.output - Zui/Zod schema describing output (optional)
      * @param props.handler - Async function that implements the tool logic
      * @param props.aliases - Alternative names for the tool (optional)
      * @param props.metadata - Additional metadata for the tool (optional)
@@ -448,7 +450,7 @@ export declare class Tool<I extends z.ZodType = z.ZodType, O extends z.ZodType =
     /**
      * Executes the tool with the given input and context.
      *
-     * This method handles input validation, retry logic, output validation, and error handling.
+     * This method handles input validation, retry logic, and error handling.
      * It's called internally by the LLMz execution engine when generated code calls the tool.
      *
      * @param input - Input data to pass to the tool handler

package/dist/{truncator-DZqEqnpD.js → truncator-CD7QtE84.js}

@@ -126,9 +126,9 @@ function truncateWrappedContent({ messages, tokenLimit, throwOnFailure = true })
 	* After                       { content: 'content', tokens: 10, truncatable: false }
 	*/
 	const parts = [];
-	for (const msg of messages) {
+	const contents = messages.flatMap((message) => typeof message.content === "string" ? [message.content] : (message.content ?? []).flatMap((part) => part.type === "text" ? [part.text ?? ""] : []));
+	for (const content of contents) {
 		const current = [];
-		const content = typeof msg.content === "string" ? msg.content : "";
 		let match;
 		const parser = new _MessageContentParser();
 		while ((match = parser.parse(content)) !== null) {
@@ -190,11 +190,16 @@ function truncateWrappedContent({ messages, tokenLimit, throwOnFailure = true })
 		biggest.tokens -= toRemove;
 		currentCount -= toRemove;
 	}
-	return messages.map((msg, i) => {
-		const p = parts[i];
+	let textIndex = 0;
+	const renderText = () => _renderRemainingWrappers(parts[textIndex++].map((part) => part.content).join(""));
+	return messages.map((message) => {
+		var _message$content;
 		return {
-			...msg,
-			content: typeof msg.content === "string" ? _renderRemainingWrappers(p.map((part) => part.content).join("")) : msg.content
+			...message,
+			content: typeof message.content === "string" ? renderText() : ((_message$content = message.content) === null || _message$content === void 0 ? void 0 : _message$content.map((part) => part.type === "text" ? {
+				...part,
+				text: renderText()
+			} : part)) ?? message.content
 		};
 	});
 }

package/dist/{truncator-CLkouTCO.cjs → truncator-DmCu9M4m.cjs}

@@ -1,5 +1,5 @@
-const require_compiler = require("./compiler-C-OI9sim.cjs");
-const require_utils = require("./utils-3ESwP3yg.cjs");
+const require_compiler = require("./compiler-BJJUmNS9.cjs");
+const require_utils = require("./utils-DGQ4sRke.cjs");
 //#region src/truncator.ts
 var truncator_exports = /* @__PURE__ */ require_compiler.__exportAll({
 	stripTruncationTags: () => stripTruncationTags,
@@ -126,9 +126,9 @@ function truncateWrappedContent({ messages, tokenLimit, throwOnFailure = true })
 	* After                       { content: 'content', tokens: 10, truncatable: false }
 	*/
 	const parts = [];
-	for (const msg of messages) {
+	const contents = messages.flatMap((message) => typeof message.content === "string" ? [message.content] : (message.content ?? []).flatMap((part) => part.type === "text" ? [part.text ?? ""] : []));
+	for (const content of contents) {
 		const current = [];
-		const content = typeof msg.content === "string" ? msg.content : "";
 		let match;
 		const parser = new _MessageContentParser();
 		while ((match = parser.parse(content)) !== null) {
@@ -190,11 +190,16 @@ function truncateWrappedContent({ messages, tokenLimit, throwOnFailure = true })
 		biggest.tokens -= toRemove;
 		currentCount -= toRemove;
 	}
-	return messages.map((msg, i) => {
-		const p = parts[i];
+	let textIndex = 0;
+	const renderText = () => _renderRemainingWrappers(parts[textIndex++].map((part) => part.content).join(""));
+	return messages.map((message) => {
+		var _message$content;
 		return {
-			...msg,
-			content: typeof msg.content === "string" ? _renderRemainingWrappers(p.map((part) => part.content).join("")) : msg.content
+			...message,
+			content: typeof message.content === "string" ? renderText() : ((_message$content = message.content) === null || _message$content === void 0 ? void 0 : _message$content.map((part) => part.type === "text" ? {
+				...part,
+				text: renderText()
+			} : part)) ?? message.content
 		};
 	});
 }

package/dist/{typings-3lsqCbkn.cjs → typings-DXJMu59O.cjs}

@@ -1,5 +1,5 @@
-const require_compiler = require("./compiler-C-OI9sim.cjs");
-const require_utils = require("./utils-3ESwP3yg.cjs");
+const require_compiler = require("./compiler-BJJUmNS9.cjs");
+const require_utils = require("./utils-DGQ4sRke.cjs");
 const require_errors = require("./errors-C962uJdS.cjs");
 let _bpinternal_zui = require("@bpinternal/zui");
 //#region src/typings-formatter.ts

package/dist/{utils-3ESwP3yg.cjs → utils-DGQ4sRke.cjs}

@@ -1,4 +1,4 @@
-const require_compiler = require("./compiler-C-OI9sim.cjs");
+const require_compiler = require("./compiler-BJJUmNS9.cjs");
 const require__baseIteratee = require("./_baseIteratee-CF4qP2a8.cjs");
 const require__getPrototype = require("./_getPrototype-DQDlNVVM.cjs");
 let _bpinternal_zui = require("@bpinternal/zui");