llmz 0.0.33 → 0.0.35

package/dist/{chunk-U4HWJLF2.js → chunk-B42FGPCX.js}

@@ -124,7 +124,7 @@ declare function ${schema.identifier}${typings};${closingTag}`);
 declare const ${schema.identifier}: ${typings};${closingTag}`);
   }
   if (schema instanceof KeyValue) {
-    if (schema.value instanceof z.ZodOptional) {
+    if (schema.value instanceof z.ZodOptional || schema.value instanceof z.ZodDefault) {
       let innerType = schema.value._def.innerType;
       if (innerType instanceof z.Schema && !innerType.description && schema.value.description) {
         innerType = innerType == null ? void 0 : innerType.describe(schema.value.description);

package/dist/{chunk-4XIOQWVZ.cjs → chunk-B742DUFY.cjs}

@@ -2,10 +2,10 @@
 
 
 
-var _chunk5BEKU5MZcjs = require('./chunk-5BEKU5MZ.cjs');
+var _chunkQRFGEFROcjs = require('./chunk-QRFGEFRO.cjs');
 
 
-var _chunkSOEKWFU2cjs = require('./chunk-SOEKWFU2.cjs');
+var _chunkQR54TZ36cjs = require('./chunk-QR54TZ36.cjs');
 
 
 var _chunkPK72FAKDcjs = require('./chunk-PK72FAKD.cjs');
@@ -14,7 +14,7 @@ var _chunkPK72FAKDcjs = require('./chunk-PK72FAKD.cjs');
 var _chunk3G3BS5IAcjs = require('./chunk-3G3BS5IA.cjs');
 
 
-var _chunkZRCU35UVcjs = require('./chunk-ZRCU35UV.cjs');
+var _chunkQY2TRWGKcjs = require('./chunk-QY2TRWGK.cjs');
 
 
 var _chunkGZPN7RGHcjs = require('./chunk-GZPN7RGH.cjs');
@@ -346,17 +346,11 @@ var Snapshot = class _Snapshot {
    * @internal
    */
   static fromSignal(signal) {
-    const variables = Object.entries(signal.variables).map(([name, value]) => {
-      const type = _chunk5BEKU5MZcjs.extractType.call(void 0, value);
-      const bytes = JSON.stringify(value || "").length;
-      const truncated = bytes > MAX_SNAPSHOT_SIZE_BYTES;
-      return truncated ? { name, type, bytes, truncated: true, preview: _nullishCoalesce(_chunk5BEKU5MZcjs.inspect.call(void 0, value, name), () => ( "N/A")) } : { name, type, bytes, truncated: false, value };
-    });
     return new _Snapshot({
       id: "snapshot_" + _ulid.ulid.call(void 0, ),
       reason: signal.message,
       stack: signal.truncatedCode,
-      variables,
+      variables: parseVariables(signal.variables),
       toolCall: signal.toolCall,
       status: { type: "pending" }
     });
@@ -453,9 +447,19 @@ var Snapshot = class _Snapshot {
    * ```
    */
   resolve(value) {
+    var _a;
     if (this.#status.type !== "pending") {
       throw new Error(`Cannot resolve snapshot because it is already settled: ${this.#status.type}`);
     }
+    const assignment = (_a = this.toolCall) == null ? void 0 : _a.assignment;
+    if (assignment) {
+      try {
+        const fn = new Function(assignment.evalFn);
+        const assignmentValue = fn(value);
+        this.variables = [...this.variables, ...parseVariables(assignmentValue)];
+      } catch (e2) {
+      }
+    }
     this.#status = { type: "resolved", value };
   }
   /**
@@ -487,6 +491,14 @@ var Snapshot = class _Snapshot {
     this.#status = { type: "rejected", error };
   }
 };
+function parseVariables(variableMap) {
+  return Object.entries(variableMap).map(([name, value]) => {
+    const type = _chunkQRFGEFROcjs.extractType.call(void 0, value);
+    const bytes = JSON.stringify(value || "").length;
+    const truncated = bytes > MAX_SNAPSHOT_SIZE_BYTES;
+    return truncated ? { name, type, bytes, truncated: true, preview: _nullishCoalesce(_chunkQRFGEFROcjs.inspect.call(void 0, value, name), () => ( "N/A")) } : { name, type, bytes, truncated: false, value };
+  });
+}
 
 // src/transcript.ts
 var MAX_MESSAGE_LENGTH = 5e3;
@@ -495,9 +507,9 @@ function getMessagePreview(message) {
     return message.content;
   }
   if (message.role === "event") {
-    return _chunk5BEKU5MZcjs.inspect.call(void 0, message.payload, message.name, { tokens: 1e3 });
+    return _chunkQRFGEFROcjs.inspect.call(void 0, message.payload, message.name, { tokens: 1e3 });
   }
-  return _chunk5BEKU5MZcjs.inspect.call(void 0, message, void 0, { tokens: 1e3 });
+  return _chunkQRFGEFROcjs.inspect.call(void 0, message, void 0, { tokens: 1e3 });
 }
 function getMessageType(message) {
   if (message.role === "assistant" || message.role === "user" || message.role === "summary") {
@@ -1086,7 +1098,7 @@ var Context = (_class2 = class {
   
   
   
-  __init3() {this.version = _chunk5BEKU5MZcjs.DualModePrompt}
+  __init3() {this.version = _chunkQRFGEFROcjs.DualModePrompt}
   __init4() {this.timeout = 6e4}
   // Default timeout of 60 seconds
   
@@ -1117,6 +1129,7 @@ var Context = (_class2 = class {
     return iteration;
   }
   _getIterationVariables() {
+    var _a;
     const lastIteration = this.iterations.at(-1);
     const variables = {};
     if ((lastIteration == null ? void 0 : lastIteration.status.type) === "thinking_requested") {
@@ -1128,6 +1141,13 @@ var Context = (_class2 = class {
     if (_chunkUQOBUJIQcjs.isPlainObject_default.call(void 0, lastIteration == null ? void 0 : lastIteration.variables)) {
       Object.assign(variables, _chunkUQOBUJIQcjs.cloneDeep_default.call(void 0, _nullishCoalesce((lastIteration == null ? void 0 : lastIteration.variables), () => ( {}))));
     }
+    if (((_a = this.snapshot) == null ? void 0 : _a.status.type) === "resolved") {
+      for (const v of this.snapshot.variables) {
+        if (!v.truncated && v.value !== void 0) {
+          variables[v.name] = v.value;
+        }
+      }
+    }
     return variables;
   }
   async _getIterationMessages(parameters) {
@@ -1254,7 +1274,7 @@ var Context = (_class2 = class {
     var _a, _b;
     const instructions = await getValue(this.instructions, this);
     const transcript = new TranscriptArray(await getValue(_nullishCoalesce(((_a = this.chat) == null ? void 0 : _a.transcript), () => ( [])), this));
-    const tools = _chunkSOEKWFU2cjs.Tool.withUniqueNames(await _asyncNullishCoalesce(await getValue(this.tools, this), async () => ( [])));
+    const tools = _chunkQR54TZ36cjs.Tool.withUniqueNames(await _asyncNullishCoalesce(await getValue(this.tools, this), async () => ( [])));
     const objects = await _asyncNullishCoalesce(await getValue(this.objects, this), async () => ( []));
     const exits = await _asyncNullishCoalesce(await getValue(this.exits, this), async () => ( []));
     const components = await getValue(_nullishCoalesce(((_b = this.chat) == null ? void 0 : _b.components), () => ( [])), this);
@@ -1267,7 +1287,7 @@ var Context = (_class2 = class {
       throw new Error("Too many tools. Expected at most 100 tools.");
     }
     for (const component of components) {
-      _chunkZRCU35UVcjs.assertValidComponent.call(void 0, component.definition);
+      _chunkQY2TRWGKcjs.assertValidComponent.call(void 0, component.definition);
     }
     const ReservedToolNames = [
       "think",
@@ -1286,7 +1306,7 @@ var Context = (_class2 = class {
       "boolean",
       "array"
     ];
-    const MessageTool = this.chat && components.length ? new (0, _chunkSOEKWFU2cjs.Tool)({
+    const MessageTool = this.chat && components.length ? new (0, _chunkQR54TZ36cjs.Tool)({
       name: "Message",
       description: "Send a message to the user",
       aliases: Array.from(

package/dist/{chunk-AAHUDKBY.js → chunk-DOIKBDLR.js}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-YEAWWJSJ.js";
 import {
   getComponentReference
-} from "./chunk-GGWM6X2K.js";
+} from "./chunk-HA52ORZL.js";
 import {
   wrapContent
 } from "./chunk-3JYCCI4S.js";
@@ -449,7 +449,7 @@ var getSystemMessage = async (props) => {
 
 `;
   }
-  for (const tool of props.globalTools) {
+  for (const tool of props.globalTools.filter((t) => t.name.toLowerCase() !== "message")) {
     dts += await tool.getTypings() + "\n";
     tool_names.push(tool.name);
   }

package/dist/{chunk-23SY6IDW.js → chunk-GXLMCQYF.js}

@@ -1,6 +1,6 @@
 import {
   getTypings
-} from "./chunk-U4HWJLF2.js";
+} from "./chunk-B42FGPCX.js";
 import {
   convertObjectToZuiLiterals,
   isJsonSchema,
@@ -123,14 +123,15 @@ var Tool = class _Tool {
    * @internal
    */
   get zOutput() {
-    if (!this.output) {
-      return z.void();
-    }
-    try {
-      return transforms.fromJSONSchema(this.output);
-    } catch {
-      return transforms.fromJSONSchemaLegacy(this.output);
+    let output = z.void();
+    if (this.output) {
+      try {
+        output = transforms.fromJSONSchema(this.output);
+      } catch {
+        output = transforms.fromJSONSchemaLegacy(this.output);
+      }
     }
+    return z.promise(output);
   }
   /**
    * Renames the tool and updates its aliases.

package/dist/{chunk-GGWM6X2K.js → chunk-HA52ORZL.js}

@@ -121,8 +121,6 @@ ${example.code.trim()}
     default:
       doc += "**Props:**\n\n";
       doc += getPropsDoc(component.default.props);
-      doc += "**Children:**\n\n";
-      doc += getChildrenDoc(component.default.children);
       doc += getExamplesDoc(component.examples);
       break;
   }

package/dist/{chunk-INDOGCAQ.cjs → chunk-KED57HR5.cjs}

@@ -124,7 +124,7 @@ declare function ${schema.identifier}${typings};${closingTag}`);
 declare const ${schema.identifier}: ${typings};${closingTag}`);
   }
   if (schema instanceof KeyValue) {
-    if (schema.value instanceof _zui.z.ZodOptional) {
+    if (schema.value instanceof _zui.z.ZodOptional || schema.value instanceof _zui.z.ZodDefault) {
       let innerType = schema.value._def.innerType;
       if (innerType instanceof _zui.z.Schema && !innerType.description && schema.value.description) {
         innerType = innerType == null ? void 0 : innerType.describe(schema.value.description);

package/dist/{chunk-SOEKWFU2.cjs → chunk-QR54TZ36.cjs}

@@ -1,6 +1,6 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } var _class;
 
-var _chunkINDOGCAQcjs = require('./chunk-INDOGCAQ.cjs');
+var _chunkKED57HR5cjs = require('./chunk-KED57HR5.cjs');
 
 
 
@@ -123,14 +123,15 @@ var Tool = (_class = class _Tool {
    * @internal
    */
   get zOutput() {
-    if (!this.output) {
-      return _zui.z.void();
-    }
-    try {
-      return _zui.transforms.fromJSONSchema(this.output);
-    } catch (e3) {
-      return _zui.transforms.fromJSONSchemaLegacy(this.output);
+    let output = _zui.z.void();
+    if (this.output) {
+      try {
+        output = _zui.transforms.fromJSONSchema(this.output);
+      } catch (e3) {
+        output = _zui.transforms.fromJSONSchemaLegacy(this.output);
+      }
     }
+    return _zui.z.promise(output);
   }
   /**
    * Renames the tool and updates its aliases.
@@ -476,7 +477,7 @@ var Tool = (_class = class _Tool {
       input = _chunkWHNOR4ZUcjs.convertObjectToZuiLiterals.call(void 0, this._staticInputValues);
     }
     const fnType = _zui.z.function(input, _zui.z.promise(output)).title(this.name).describe(_nullishCoalesce(this.description, () => ( "")));
-    return _chunkINDOGCAQcjs.getTypings.call(void 0, fnType, {
+    return _chunkKED57HR5cjs.getTypings.call(void 0, fnType, {
       declaration: true
     });
   }

package/dist/{chunk-5BEKU5MZ.cjs → chunk-QRFGEFRO.cjs}

@@ -3,7 +3,7 @@
 var _chunkKIN7Y247cjs = require('./chunk-KIN7Y247.cjs');
 
 
-var _chunkZRCU35UVcjs = require('./chunk-ZRCU35UV.cjs');
+var _chunkQY2TRWGKcjs = require('./chunk-QY2TRWGK.cjs');
 
 
 var _chunkGZPN7RGHcjs = require('./chunk-GZPN7RGH.cjs');
@@ -449,7 +449,7 @@ var getSystemMessage = async (props) => {
 
 `;
   }
-  for (const tool of props.globalTools) {
+  for (const tool of props.globalTools.filter((t) => t.name.toLowerCase() !== "message")) {
     dts += await tool.getTypings() + "\n";
     tool_names.push(tool.name);
   }
@@ -509,7 +509,7 @@ ${variables_example}
       variables_example,
       exits,
       components: _chunkGZPN7RGHcjs.wrapContent.call(void 0, 
-        props.components.map((component) => _chunkZRCU35UVcjs.getComponentReference.call(void 0, component.definition)).join("\n\n"),
+        props.components.map((component) => _chunkQY2TRWGKcjs.getComponentReference.call(void 0, component.definition)).join("\n\n"),
         {
           preserve: "top",
           minTokens: 500

package/dist/{chunk-ZRCU35UV.cjs → chunk-QY2TRWGK.cjs}

@@ -121,8 +121,6 @@ ${example.code.trim()}
     default:
       doc += "**Props:**\n\n";
       doc += getPropsDoc(component.default.props);
-      doc += "**Children:**\n\n";
-      doc += getChildrenDoc(component.default.children);
       doc += getExamplesDoc(component.examples);
       break;
   }

package/dist/{chunk-4JN4PPGP.js → chunk-RANA7NBF.js}

@@ -2,10 +2,10 @@ import {
   DualModePrompt,
   extractType,
   inspect
-} from "./chunk-AAHUDKBY.js";
+} from "./chunk-DOIKBDLR.js";
 import {
   Tool
-} from "./chunk-23SY6IDW.js";
+} from "./chunk-GXLMCQYF.js";
 import {
   LoopExceededError
 } from "./chunk-WSVDMGMR.js";
@@ -14,7 +14,7 @@ import {
 } from "./chunk-VPTFUOIK.js";
 import {
   assertValidComponent
-} from "./chunk-GGWM6X2K.js";
+} from "./chunk-HA52ORZL.js";
 import {
   wrapContent
 } from "./chunk-3JYCCI4S.js";
@@ -315,8 +315,8 @@ var Snapshot = class _Snapshot {
   id;
   reason;
   stack;
-  variables;
   toolCall;
+  variables;
   #status;
   /**
    * Gets the current status of the snapshot.
@@ -346,17 +346,11 @@ var Snapshot = class _Snapshot {
    * @internal
    */
   static fromSignal(signal) {
-    const variables = Object.entries(signal.variables).map(([name, value]) => {
-      const type = extractType(value);
-      const bytes = JSON.stringify(value || "").length;
-      const truncated = bytes > MAX_SNAPSHOT_SIZE_BYTES;
-      return truncated ? { name, type, bytes, truncated: true, preview: inspect(value, name) ?? "N/A" } : { name, type, bytes, truncated: false, value };
-    });
     return new _Snapshot({
       id: "snapshot_" + ulid(),
       reason: signal.message,
       stack: signal.truncatedCode,
-      variables,
+      variables: parseVariables(signal.variables),
       toolCall: signal.toolCall,
       status: { type: "pending" }
     });
@@ -453,9 +447,19 @@ var Snapshot = class _Snapshot {
    * ```
    */
   resolve(value) {
+    var _a;
     if (this.#status.type !== "pending") {
       throw new Error(`Cannot resolve snapshot because it is already settled: ${this.#status.type}`);
     }
+    const assignment = (_a = this.toolCall) == null ? void 0 : _a.assignment;
+    if (assignment) {
+      try {
+        const fn = new Function(assignment.evalFn);
+        const assignmentValue = fn(value);
+        this.variables = [...this.variables, ...parseVariables(assignmentValue)];
+      } catch {
+      }
+    }
     this.#status = { type: "resolved", value };
   }
   /**
@@ -487,6 +491,14 @@ var Snapshot = class _Snapshot {
     this.#status = { type: "rejected", error };
   }
 };
+function parseVariables(variableMap) {
+  return Object.entries(variableMap).map(([name, value]) => {
+    const type = extractType(value);
+    const bytes = JSON.stringify(value || "").length;
+    const truncated = bytes > MAX_SNAPSHOT_SIZE_BYTES;
+    return truncated ? { name, type, bytes, truncated: true, preview: inspect(value, name) ?? "N/A" } : { name, type, bytes, truncated: false, value };
+  });
+}
 
 // src/transcript.ts
 var MAX_MESSAGE_LENGTH = 5e3;
@@ -1117,6 +1129,7 @@ var Context = class {
     return iteration;
   }
   _getIterationVariables() {
+    var _a;
     const lastIteration = this.iterations.at(-1);
     const variables = {};
     if ((lastIteration == null ? void 0 : lastIteration.status.type) === "thinking_requested") {
@@ -1128,6 +1141,13 @@ var Context = class {
     if (isPlainObject_default(lastIteration == null ? void 0 : lastIteration.variables)) {
       Object.assign(variables, cloneDeep_default((lastIteration == null ? void 0 : lastIteration.variables) ?? {}));
     }
+    if (((_a = this.snapshot) == null ? void 0 : _a.status.type) === "resolved") {
+      for (const v of this.snapshot.variables) {
+        if (!v.truncated && v.value !== void 0) {
+          variables[v.name] = v.value;
+        }
+      }
+    }
     return variables;
   }
   async _getIterationMessages(parameters) {

package/dist/component-AAEMNCHB.cjs

@@ -0,0 +1,18 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+
+
+
+
+
+var _chunkQY2TRWGKcjs = require('./chunk-QY2TRWGK.cjs');
+require('./chunk-KMZDFWYZ.cjs');
+require('./chunk-UQOBUJIQ.cjs');
+
+
+
+
+
+
+
+exports.Component = _chunkQY2TRWGKcjs.Component; exports.assertValidComponent = _chunkQY2TRWGKcjs.assertValidComponent; exports.getComponentReference = _chunkQY2TRWGKcjs.getComponentReference; exports.isAnyComponent = _chunkQY2TRWGKcjs.isAnyComponent; exports.isComponent = _chunkQY2TRWGKcjs.isComponent; exports.renderToTsx = _chunkQY2TRWGKcjs.renderToTsx;

package/dist/{component-WFVDVSDK.js → component-EOMTLA64.js}

@@ -5,7 +5,7 @@ import {
   isAnyComponent,
   isComponent,
   renderToTsx
-} from "./chunk-GGWM6X2K.js";
+} from "./chunk-HA52ORZL.js";
 import "./chunk-ORQP26SZ.js";
 import "./chunk-7WRN4E42.js";
 export {

package/dist/{dual-modes-UBHAMQW4.cjs → dual-modes-MHUM2LEI.cjs}

@@ -1,12 +1,12 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunk5BEKU5MZcjs = require('./chunk-5BEKU5MZ.cjs');
+var _chunkQRFGEFROcjs = require('./chunk-QRFGEFRO.cjs');
 require('./chunk-KIN7Y247.cjs');
-require('./chunk-ZRCU35UV.cjs');
+require('./chunk-QY2TRWGK.cjs');
 require('./chunk-KMZDFWYZ.cjs');
 require('./chunk-GZPN7RGH.cjs');
 require('./chunk-WHNOR4ZU.cjs');
 require('./chunk-UQOBUJIQ.cjs');
 
 
-exports.DualModePrompt = _chunk5BEKU5MZcjs.DualModePrompt;
+exports.DualModePrompt = _chunkQRFGEFROcjs.DualModePrompt;

package/dist/{dual-modes-LEAHGCOF.js → dual-modes-NJGSJEE4.js}

@@ -1,8 +1,8 @@
 import {
   DualModePrompt
-} from "./chunk-AAHUDKBY.js";
+} from "./chunk-DOIKBDLR.js";
 import "./chunk-YEAWWJSJ.js";
-import "./chunk-GGWM6X2K.js";
+import "./chunk-HA52ORZL.js";
 import "./chunk-ORQP26SZ.js";
 import "./chunk-3JYCCI4S.js";
 import "./chunk-ZORRILUV.js";

package/dist/index.cjs

@@ -9,15 +9,15 @@
 
 
 
-var _chunk4XIOQWVZcjs = require('./chunk-4XIOQWVZ.cjs');
-require('./chunk-5BEKU5MZ.cjs');
+var _chunkB742DUFYcjs = require('./chunk-B742DUFY.cjs');
+require('./chunk-QRFGEFRO.cjs');
 
 
-var _chunkSOEKWFU2cjs = require('./chunk-SOEKWFU2.cjs');
+var _chunkQR54TZ36cjs = require('./chunk-QR54TZ36.cjs');
 
 
 
-var _chunkINDOGCAQcjs = require('./chunk-INDOGCAQ.cjs');
+var _chunkKED57HR5cjs = require('./chunk-KED57HR5.cjs');
 
 
 
@@ -34,7 +34,7 @@ var _chunk3G3BS5IAcjs = require('./chunk-3G3BS5IA.cjs');
 
 
 
-var _chunkZRCU35UVcjs = require('./chunk-ZRCU35UV.cjs');
+var _chunkQY2TRWGKcjs = require('./chunk-QY2TRWGK.cjs');
 require('./chunk-KMZDFWYZ.cjs');
 
 
@@ -302,7 +302,7 @@ async function hoistTypings(code, formatOptions) {
       break;
     }
   }
-  return _chunkINDOGCAQcjs.formatTypings.call(void 0, code, formatOptions);
+  return _chunkKED57HR5cjs.formatTypings.call(void 0, code, formatOptions);
 }
 
 // src/objects.ts
@@ -428,7 +428,7 @@ var ObjectInstance = class {
     this.description = props.description;
     this.metadata = _nullishCoalesce(props.metadata, () => ( {}));
     this.properties = props.properties;
-    this.tools = _chunkSOEKWFU2cjs.Tool.withUniqueNames(_nullishCoalesce(props.tools, () => ( [])));
+    this.tools = _chunkQR54TZ36cjs.Tool.withUniqueNames(_nullishCoalesce(props.tools, () => ( [])));
   }
   /**
    * Generates TypeScript namespace declarations for this object.
@@ -509,7 +509,7 @@ function getObjectTypings(obj) {
         }
         let type = "unknown";
         if (prop.type) {
-          type = await _chunkINDOGCAQcjs.getTypings.call(void 0, prop.type, {});
+          type = await _chunkKED57HR5cjs.getTypings.call(void 0, prop.type, {});
         } else if (prop.value !== void 0) {
           type = typeof prop.value;
         }
@@ -529,7 +529,7 @@ function getObjectTypings(obj) {
       typings.push("");
       for (const tool of obj.tools) {
         const fnType = _zui.z.function(tool.zInput, tool.zOutput).title(tool.name).describe(_nullishCoalesce(tool.description, () => ( "")));
-        let temp = await _chunkINDOGCAQcjs.getTypings.call(void 0, fnType, {
+        let temp = await _chunkKED57HR5cjs.getTypings.call(void 0, fnType, {
           declaration: true
         });
         temp = temp.replace("declare function ", "function ");
@@ -552,7 +552,7 @@ function getObjectTypings(obj) {
     if ((_a = obj.description) == null ? void 0 : _a.trim().length) {
       header = _chunkWHNOR4ZUcjs.getMultilineComment.call(void 0, obj.description);
     }
-    return _chunkINDOGCAQcjs.formatTypings.call(void 0, 
+    return _chunkKED57HR5cjs.formatTypings.call(void 0, 
       `${header}
       export namespace ${obj.name} {
       ${body}
@@ -766,7 +766,7 @@ var CitationsManager = (_class = class {constructor() { _class.prototype.__init.
 
 // src/component.default.ts
 
-var Button = new (0, _chunkZRCU35UVcjs.Component)({
+var Button = new (0, _chunkQY2TRWGKcjs.Component)({
   type: "leaf",
   description: "A button component that can perform actions when clicked",
   name: "Button",
@@ -775,28 +775,30 @@ var Button = new (0, _chunkZRCU35UVcjs.Component)({
     {
       name: "Say action",
       description: "A button that triggers a say action",
-      code: `<Message>
+      code: `yield <Message>
   <Button action="say" label="Hello" />
 </Message>`
     },
     {
       name: "Postback action",
       description: "A button that sends a postback value",
-      code: `<Message>
+      code: `yield <Message>
+  Choose an option:
   <Button action="postback" label="Buy" value="buy_product" />
+  <Button action="postback" label="Buy" value="buy_product2" />
 </Message>`
     }
   ],
   leaf: {
     props: _zui.z.object({
       action: _zui.z.enum(["say", "url", "postback"]).default("say").describe('The action to perform when the button is clicked. Can be "say", "url", or "postback"'),
-      label: _zui.z.string().describe("The text displayed on the button"),
+      label: _zui.z.string().describe("The text displayed on the button (min 1 character, max 250 characters)"),
       value: _zui.z.string().optional().describe('The postback value to send when the button is clicked. Required if action is "postback"'),
       url: _zui.z.string().optional().describe('The URL to open when the button is clicked. Required if action is "url"')
     })
   }
 });
-var Image = new (0, _chunkZRCU35UVcjs.Component)({
+var Image = new (0, _chunkQY2TRWGKcjs.Component)({
   type: "leaf",
   name: "Image",
   description: "Displays an image from a URL.",
@@ -805,7 +807,8 @@ var Image = new (0, _chunkZRCU35UVcjs.Component)({
     {
       name: "Basic image",
       description: "A simple image with alt text",
-      code: `<Message>
+      code: `yield <Message>
+  An example image:
   <Image url="https://example.com/photo.jpg" alt="Example image" />
 </Message>`
     }
@@ -817,7 +820,7 @@ var Image = new (0, _chunkZRCU35UVcjs.Component)({
     })
   }
 });
-var File = new (0, _chunkZRCU35UVcjs.Component)({
+var File = new (0, _chunkQY2TRWGKcjs.Component)({
   type: "leaf",
   name: "File",
   description: "Sends a downloadable file to the user.",
@@ -826,7 +829,8 @@ var File = new (0, _chunkZRCU35UVcjs.Component)({
     {
       name: "PDF download",
       description: "Send a PDF file with a name",
-      code: `<Message>
+      code: `yield <Message>
+  Here is your report:
   <File url="https://example.com/report.pdf" name="Report.pdf" />
 </Message>`
     }
@@ -838,7 +842,7 @@ var File = new (0, _chunkZRCU35UVcjs.Component)({
     })
   }
 });
-var Video = new (0, _chunkZRCU35UVcjs.Component)({
+var Video = new (0, _chunkQY2TRWGKcjs.Component)({
   type: "leaf",
   name: "Video",
   description: "Embeds a video from a URL.",
@@ -847,7 +851,8 @@ var Video = new (0, _chunkZRCU35UVcjs.Component)({
     {
       name: "Intro video",
       description: "A video with a title",
-      code: `<Message>
+      code: `yield <Message>
+  Watch this video:
   <Video url="https://example.com/intro.mp4" title="Welcome" />
 </Message>`
     }
@@ -859,7 +864,7 @@ var Video = new (0, _chunkZRCU35UVcjs.Component)({
     })
   }
 });
-var Audio = new (0, _chunkZRCU35UVcjs.Component)({
+var Audio = new (0, _chunkQY2TRWGKcjs.Component)({
   type: "leaf",
   name: "Audio",
   description: "Plays an audio clip from a URL.",
@@ -868,7 +873,8 @@ var Audio = new (0, _chunkZRCU35UVcjs.Component)({
     {
       name: "Sample audio",
       description: "Play a short audio clip with a title",
-      code: `<Message>
+      code: `yield <Message>
+  Listen to this audio:
   <Audio url="https://example.com/audio.mp3" title="Sample" />
 </Message>`
     }
@@ -880,7 +886,7 @@ var Audio = new (0, _chunkZRCU35UVcjs.Component)({
     })
   }
 });
-var Card = new (0, _chunkZRCU35UVcjs.Component)({
+var Card = new (0, _chunkQY2TRWGKcjs.Component)({
   type: "container",
   name: "Card",
   description: "A visual card component that can include an image and buttons.",
@@ -889,7 +895,8 @@ var Card = new (0, _chunkZRCU35UVcjs.Component)({
     {
       name: "Product card",
       description: "A card with an image and two buttons",
-      code: `<Message>
+      code: `yield <Message>
+  Featured product:
   <Card title="Product Name" subtitle="Limited offer">
     <Image url="https://example.com/product.jpg" alt="Product image" />
     <Button action="postback" label="Buy" value="buy_product" />
@@ -915,7 +922,7 @@ var Card = new (0, _chunkZRCU35UVcjs.Component)({
     ]
   }
 });
-var Carousel = new (0, _chunkZRCU35UVcjs.Component)({
+var Carousel = new (0, _chunkQY2TRWGKcjs.Component)({
   type: "container",
   name: "Carousel",
   description: "A virtual container for displaying 1 to 10 Card components as a carousel.",
@@ -924,7 +931,8 @@ var Carousel = new (0, _chunkZRCU35UVcjs.Component)({
     {
       name: "Product carousel",
       description: "A carousel with multiple cards",
-      code: `<Message>
+      code: `yield <Message>
+  Here are some products you might like:
   <Carousel>
     <Card title="Item 1" subtitle="First product">
       <Image url="https://example.com/item1.jpg" alt="Item 1" />
@@ -948,7 +956,7 @@ var Carousel = new (0, _chunkZRCU35UVcjs.Component)({
     ]
   }
 });
-var Text = new (0, _chunkZRCU35UVcjs.Component)({
+var Text = new (0, _chunkQY2TRWGKcjs.Component)({
   type: "default",
   name: "Message",
   aliases: ["Text", "Markdown"],
@@ -957,7 +965,7 @@ var Text = new (0, _chunkZRCU35UVcjs.Component)({
     {
       name: "Basic Markdown",
       description: "Simple markdown content inside a message",
-      code: `<Message>
+      code: `yield <Message>
   **Hello**, welcome to our service!
 </Message>`
     }
@@ -1104,20 +1112,20 @@ var utils = {
   truncateWrappedContent: _chunkGZPN7RGHcjs.truncateWrappedContent
 };
 var execute = async (props) => {
-  const { executeContext } = await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-HGUVAYIN.cjs")));
+  const { executeContext } = await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-XRJIDXIZ.cjs")));
   return executeContext(props);
 };
 var init = async () => {
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-HGUVAYIN.cjs")));
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./component-R4WTW6DZ.cjs")));
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./tool-JVLOALQN.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-XRJIDXIZ.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./component-AAEMNCHB.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./tool-JYXLFJGQ.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./exit-XAYKJ6TR.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./jsx-AJAXBWFE.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./vm-VFORKC54.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./utils-L5QAQXV2.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./truncator-W3NXBLYJ.cjs")));
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./typings-OLI56LGT.cjs")));
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./dual-modes-UBHAMQW4.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./typings-7C2RXO75.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./dual-modes-MHUM2LEI.cjs")));
 };
 
 
@@ -1147,4 +1155,4 @@ var init = async () => {
 
 
 
-exports.Chat = Chat; exports.CitationsManager = CitationsManager; exports.Component = _chunkZRCU35UVcjs.Component; exports.DefaultComponents = DefaultComponents; exports.DefaultExit = _chunk4XIOQWVZcjs.DefaultExit; exports.ErrorExecutionResult = _chunk4XIOQWVZcjs.ErrorExecutionResult; exports.ExecutionResult = _chunk4XIOQWVZcjs.ExecutionResult; exports.Exit = _chunk3G3BS5IAcjs.Exit; exports.ListenExit = _chunk4XIOQWVZcjs.ListenExit; exports.LoopExceededError = _chunkPK72FAKDcjs.LoopExceededError; exports.ObjectInstance = ObjectInstance; exports.PartialExecutionResult = _chunk4XIOQWVZcjs.PartialExecutionResult; exports.Snapshot = _chunk4XIOQWVZcjs.Snapshot; exports.SnapshotSignal = _chunkPK72FAKDcjs.SnapshotSignal; exports.SuccessExecutionResult = _chunk4XIOQWVZcjs.SuccessExecutionResult; exports.ThinkExit = _chunk4XIOQWVZcjs.ThinkExit; exports.ThinkSignal = _chunkPK72FAKDcjs.ThinkSignal; exports.Tool = _chunkSOEKWFU2cjs.Tool; exports.assertValidComponent = _chunkZRCU35UVcjs.assertValidComponent; exports.execute = execute; exports.getValue = _chunk4XIOQWVZcjs.getValue; exports.init = init; exports.isAnyComponent = _chunkZRCU35UVcjs.isAnyComponent; exports.isComponent = _chunkZRCU35UVcjs.isComponent; exports.parseExit = _chunk4XIOQWVZcjs.parseExit; exports.renderToTsx = _chunkZRCU35UVcjs.renderToTsx; exports.utils = utils;
+exports.Chat = Chat; exports.CitationsManager = CitationsManager; exports.Component = _chunkQY2TRWGKcjs.Component; exports.DefaultComponents = DefaultComponents; exports.DefaultExit = _chunkB742DUFYcjs.DefaultExit; exports.ErrorExecutionResult = _chunkB742DUFYcjs.ErrorExecutionResult; exports.ExecutionResult = _chunkB742DUFYcjs.ExecutionResult; exports.Exit = _chunk3G3BS5IAcjs.Exit; exports.ListenExit = _chunkB742DUFYcjs.ListenExit; exports.LoopExceededError = _chunkPK72FAKDcjs.LoopExceededError; exports.ObjectInstance = ObjectInstance; exports.PartialExecutionResult = _chunkB742DUFYcjs.PartialExecutionResult; exports.Snapshot = _chunkB742DUFYcjs.Snapshot; exports.SnapshotSignal = _chunkPK72FAKDcjs.SnapshotSignal; exports.SuccessExecutionResult = _chunkB742DUFYcjs.SuccessExecutionResult; exports.ThinkExit = _chunkB742DUFYcjs.ThinkExit; exports.ThinkSignal = _chunkPK72FAKDcjs.ThinkSignal; exports.Tool = _chunkQR54TZ36cjs.Tool; exports.assertValidComponent = _chunkQY2TRWGKcjs.assertValidComponent; exports.execute = execute; exports.getValue = _chunkB742DUFYcjs.getValue; exports.init = init; exports.isAnyComponent = _chunkQY2TRWGKcjs.isAnyComponent; exports.isComponent = _chunkQY2TRWGKcjs.isComponent; exports.parseExit = _chunkB742DUFYcjs.parseExit; exports.renderToTsx = _chunkQY2TRWGKcjs.renderToTsx; exports.utils = utils;

package/dist/index.js

@@ -9,15 +9,15 @@ import {
   ThinkExit,
   getValue,
   parseExit
-} from "./chunk-4JN4PPGP.js";
-import "./chunk-AAHUDKBY.js";
+} from "./chunk-RANA7NBF.js";
+import "./chunk-DOIKBDLR.js";
 import {
   Tool
-} from "./chunk-23SY6IDW.js";
+} from "./chunk-GXLMCQYF.js";
 import {
   formatTypings,
   getTypings
-} from "./chunk-U4HWJLF2.js";
+} from "./chunk-B42FGPCX.js";
 import {
   CodeFormattingError,
   LoopExceededError,
@@ -34,7 +34,7 @@ import {
   isAnyComponent,
   isComponent,
   renderToTsx
-} from "./chunk-GGWM6X2K.js";
+} from "./chunk-HA52ORZL.js";
 import "./chunk-ORQP26SZ.js";
 import {
   truncateWrappedContent,
@@ -775,22 +775,24 @@ var Button = new Component({
     {
       name: "Say action",
       description: "A button that triggers a say action",
-      code: `<Message>
+      code: `yield <Message>
   <Button action="say" label="Hello" />
 </Message>`
     },
     {
       name: "Postback action",
       description: "A button that sends a postback value",
-      code: `<Message>
+      code: `yield <Message>
+  Choose an option:
   <Button action="postback" label="Buy" value="buy_product" />
+  <Button action="postback" label="Buy" value="buy_product2" />
 </Message>`
     }
   ],
   leaf: {
     props: z2.object({
       action: z2.enum(["say", "url", "postback"]).default("say").describe('The action to perform when the button is clicked. Can be "say", "url", or "postback"'),
-      label: z2.string().describe("The text displayed on the button"),
+      label: z2.string().describe("The text displayed on the button (min 1 character, max 250 characters)"),
       value: z2.string().optional().describe('The postback value to send when the button is clicked. Required if action is "postback"'),
       url: z2.string().optional().describe('The URL to open when the button is clicked. Required if action is "url"')
     })
@@ -805,7 +807,8 @@ var Image = new Component({
     {
       name: "Basic image",
       description: "A simple image with alt text",
-      code: `<Message>
+      code: `yield <Message>
+  An example image:
   <Image url="https://example.com/photo.jpg" alt="Example image" />
 </Message>`
     }
@@ -826,7 +829,8 @@ var File = new Component({
     {
       name: "PDF download",
       description: "Send a PDF file with a name",
-      code: `<Message>
+      code: `yield <Message>
+  Here is your report:
   <File url="https://example.com/report.pdf" name="Report.pdf" />
 </Message>`
     }
@@ -847,7 +851,8 @@ var Video = new Component({
     {
       name: "Intro video",
       description: "A video with a title",
-      code: `<Message>
+      code: `yield <Message>
+  Watch this video:
   <Video url="https://example.com/intro.mp4" title="Welcome" />
 </Message>`
     }
@@ -868,7 +873,8 @@ var Audio = new Component({
     {
       name: "Sample audio",
       description: "Play a short audio clip with a title",
-      code: `<Message>
+      code: `yield <Message>
+  Listen to this audio:
   <Audio url="https://example.com/audio.mp3" title="Sample" />
 </Message>`
     }
@@ -889,7 +895,8 @@ var Card = new Component({
     {
       name: "Product card",
       description: "A card with an image and two buttons",
-      code: `<Message>
+      code: `yield <Message>
+  Featured product:
   <Card title="Product Name" subtitle="Limited offer">
     <Image url="https://example.com/product.jpg" alt="Product image" />
     <Button action="postback" label="Buy" value="buy_product" />
@@ -924,7 +931,8 @@ var Carousel = new Component({
     {
       name: "Product carousel",
       description: "A carousel with multiple cards",
-      code: `<Message>
+      code: `yield <Message>
+  Here are some products you might like:
   <Carousel>
     <Card title="Item 1" subtitle="First product">
       <Image url="https://example.com/item1.jpg" alt="Item 1" />
@@ -957,7 +965,7 @@ var Text = new Component({
     {
       name: "Basic Markdown",
       description: "Simple markdown content inside a message",
-      code: `<Message>
+      code: `yield <Message>
   **Hello**, welcome to our service!
 </Message>`
     }
@@ -1104,20 +1112,20 @@ var utils = {
   truncateWrappedContent
 };
 var execute = async (props) => {
-  const { executeContext } = await import("./llmz-3E2JM3HM.js");
+  const { executeContext } = await import("./llmz-NX2764TL.js");
   return executeContext(props);
 };
 var init = async () => {
-  await import("./llmz-3E2JM3HM.js");
-  await import("./component-WFVDVSDK.js");
-  await import("./tool-7QXH6A24.js");
+  await import("./llmz-NX2764TL.js");
+  await import("./component-EOMTLA64.js");
+  await import("./tool-ADXBCB4R.js");
   await import("./exit-YLO7BY7Z.js");
   await import("./jsx-AEHVFB3L.js");
   await import("./vm-Y3WY2627.js");
   await import("./utils-RQHQ2KOG.js");
   await import("./truncator-BSP6PQPC.js");
-  await import("./typings-ALZEKGV6.js");
-  await import("./dual-modes-LEAHGCOF.js");
+  await import("./typings-AQGJAMVH.js");
+  await import("./dual-modes-NJGSJEE4.js");
 };
 export {
   Chat,

package/dist/{llmz-3E2JM3HM.js → llmz-NX2764TL.js}

@@ -8,10 +8,10 @@ import {
   Snapshot,
   SuccessExecutionResult,
   parseExit
-} from "./chunk-4JN4PPGP.js";
-import "./chunk-AAHUDKBY.js";
-import "./chunk-23SY6IDW.js";
-import "./chunk-U4HWJLF2.js";
+} from "./chunk-RANA7NBF.js";
+import "./chunk-DOIKBDLR.js";
+import "./chunk-GXLMCQYF.js";
+import "./chunk-B42FGPCX.js";
 import {
   AssignmentError,
   CodeExecutionError,
@@ -27,7 +27,7 @@ import {
   cleanStackTrace
 } from "./chunk-YEAWWJSJ.js";
 import "./chunk-VPTFUOIK.js";
-import "./chunk-GGWM6X2K.js";
+import "./chunk-HA52ORZL.js";
 import "./chunk-ORQP26SZ.js";
 import {
   truncateWrappedContent
@@ -44,6 +44,7 @@ import {
 } from "./chunk-7WRN4E42.js";
 
 // src/llmz.ts
+import { Client } from "@botpress/client";
 import { Cognitive } from "@botpress/cognitive";
 import { z } from "@bpinternal/zui";
 import ms from "ms";
@@ -105,7 +106,8 @@ var _executeContext = async (props) => {
   var _a, _b;
   const controller = createJoinedAbortController([props.signal]);
   const { onIterationStart, onIterationEnd, onTrace, onExit, onBeforeExecution, onAfterTool, onBeforeTool } = props;
-  const cognitive = Cognitive.isCognitiveClient(props.client) ? props.client : new Cognitive({ client: props.client });
+  const client = props.client ?? new Client();
+  const cognitive = Cognitive.isCognitiveClient(client) ? client : new Cognitive({ client });
   const cleanups = [];
   const ctx = new Context({
     chat: props.chat,
@@ -219,7 +221,7 @@ var executeIteration = async ({
   onBeforeTool,
   onAfterTool
 }) => {
-  var _a, _b, _c, _d, _e;
+  var _a, _b, _c, _d;
   let startedAt = Date.now();
   const traces = iteration.traces;
   const model = await cognitive.getModelDetails(Array.isArray(iteration.model) ? iteration.model[0] : iteration.model);
@@ -236,6 +238,7 @@ var executeIteration = async ({
       typeof x.content !== "string" || x.content.trim().length > 0
     )
   );
+  iteration.messages = messages;
   traces.push({
     type: "llm_call_started",
     started_at: startedAt,
@@ -251,7 +254,7 @@ var executeIteration = async ({
     messages: messages.filter((x) => x.role !== "system"),
     stopSequences: ctx.version.getStopTokens()
   });
-  const out = ((_c = (_b = output.output.choices) == null ? void 0 : _b[0]) == null ? void 0 : _c.type) === "text" && typeof ((_d = output.output.choices) == null ? void 0 : _d[0].content) === "string" ? output.output.choices[0].content : null;
+  const out = typeof ((_c = (_b = output.output.choices) == null ? void 0 : _b[0]) == null ? void 0 : _c.content) === "string" ? output.output.choices[0].content : null;
   if (!out) {
     throw new CognitiveError("LLM did not return any text output");
   }
@@ -427,7 +430,7 @@ var executeIteration = async ({
     return iteration.end({
       type: "execution_error",
       execution_error: {
-        message: ((_e = result == null ? void 0 : result.error) == null ? void 0 : _e.message) ?? "Unknown error occurred",
+        message: ((_d = result == null ? void 0 : result.error) == null ? void 0 : _d.message) ?? "Unknown error occurred",
         stack: cleanStackTrace(result.error.stack ?? "No stack trace available")
       }
     });

package/dist/{llmz-HGUVAYIN.cjs → llmz-XRJIDXIZ.cjs}

@@ -8,10 +8,10 @@ var _chunk5FWOHMO4cjs = require('./chunk-5FWOHMO4.cjs');
 
 
 
-var _chunk4XIOQWVZcjs = require('./chunk-4XIOQWVZ.cjs');
-require('./chunk-5BEKU5MZ.cjs');
-require('./chunk-SOEKWFU2.cjs');
-require('./chunk-INDOGCAQ.cjs');
+var _chunkB742DUFYcjs = require('./chunk-B742DUFY.cjs');
+require('./chunk-QRFGEFRO.cjs');
+require('./chunk-QR54TZ36.cjs');
+require('./chunk-KED57HR5.cjs');
 
 
 
@@ -27,7 +27,7 @@ var _chunkPK72FAKDcjs = require('./chunk-PK72FAKD.cjs');
 
 var _chunkKIN7Y247cjs = require('./chunk-KIN7Y247.cjs');
 require('./chunk-3G3BS5IA.cjs');
-require('./chunk-ZRCU35UV.cjs');
+require('./chunk-QY2TRWGK.cjs');
 require('./chunk-KMZDFWYZ.cjs');
 
 
@@ -44,6 +44,7 @@ var _chunkWHNOR4ZUcjs = require('./chunk-WHNOR4ZU.cjs');
 var _chunkUQOBUJIQcjs = require('./chunk-UQOBUJIQ.cjs');
 
 // src/llmz.ts
+var _client = require('@botpress/client');
 var _cognitive = require('@botpress/cognitive');
 var _zui = require('@bpinternal/zui');
 var _ms = require('ms'); var _ms2 = _interopRequireDefault(_ms);
@@ -105,9 +106,10 @@ var _executeContext = async (props) => {
   var _a, _b;
   const controller = createJoinedAbortController([props.signal]);
   const { onIterationStart, onIterationEnd, onTrace, onExit, onBeforeExecution, onAfterTool, onBeforeTool } = props;
-  const cognitive = _cognitive.Cognitive.isCognitiveClient(props.client) ? props.client : new (0, _cognitive.Cognitive)({ client: props.client });
+  const client = _nullishCoalesce(props.client, () => ( new (0, _client.Client)()));
+  const cognitive = _cognitive.Cognitive.isCognitiveClient(client) ? client : new (0, _cognitive.Cognitive)({ client });
   const cleanups = [];
-  const ctx = new (0, _chunk4XIOQWVZcjs.Context)({
+  const ctx = new (0, _chunkB742DUFYcjs.Context)({
     chat: props.chat,
     instructions: props.instructions,
     objects: props.objects,
@@ -122,7 +124,7 @@ var _executeContext = async (props) => {
   try {
     while (true) {
       if (ctx.iterations.length >= ctx.loop) {
-        return new (0, _chunk4XIOQWVZcjs.ErrorExecutionResult)(ctx, new (0, _chunkPK72FAKDcjs.LoopExceededError)());
+        return new (0, _chunkB742DUFYcjs.ErrorExecutionResult)(ctx, new (0, _chunkPK72FAKDcjs.LoopExceededError)());
       }
       const iteration = await ctx.nextIteration();
       try {
@@ -145,7 +147,7 @@ var _executeContext = async (props) => {
             reason: _nullishCoalesce(controller.signal.reason, () => ( "The operation was aborted"))
           }
         });
-        return new (0, _chunk4XIOQWVZcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(controller.signal.reason, () => ( "The operation was aborted")));
+        return new (0, _chunkB742DUFYcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(controller.signal.reason, () => ( "The operation was aborted")));
       }
       cleanups.push(
         iteration.traces.onPush((traces) => {
@@ -181,25 +183,25 @@ var _executeContext = async (props) => {
       }
       if (iteration.status.type === "exit_success") {
         const exitName = iteration.status.exit_success.exit_name;
-        return new (0, _chunk4XIOQWVZcjs.SuccessExecutionResult)(ctx, {
+        return new (0, _chunkB742DUFYcjs.SuccessExecutionResult)(ctx, {
           exit: iteration.exits.find((x) => x.name === exitName),
           result: iteration.status.exit_success.return_value
         });
       }
       if (iteration.status.type === "callback_requested") {
-        return new (0, _chunk4XIOQWVZcjs.PartialExecutionResult)(
+        return new (0, _chunkB742DUFYcjs.PartialExecutionResult)(
           ctx,
           iteration.status.callback_requested.signal,
-          _chunk4XIOQWVZcjs.Snapshot.fromSignal(iteration.status.callback_requested.signal)
+          _chunkB742DUFYcjs.Snapshot.fromSignal(iteration.status.callback_requested.signal)
         );
       }
       if (iteration.status.type === "thinking_requested" || iteration.status.type === "exit_error" || iteration.status.type === "execution_error" || iteration.status.type === "invalid_code_error") {
         continue;
       }
-      return new (0, _chunk4XIOQWVZcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(iteration.error, () => ( `Unknown error. Status: ${iteration.status.type}`)));
+      return new (0, _chunkB742DUFYcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(iteration.error, () => ( `Unknown error. Status: ${iteration.status.type}`)));
     }
   } catch (error) {
-    return new (0, _chunk4XIOQWVZcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(error, () => ( "Unknown error")));
+    return new (0, _chunkB742DUFYcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(error, () => ( "Unknown error")));
   } finally {
     for (const cleanup of cleanups) {
       try {
@@ -219,7 +221,7 @@ var executeIteration = async ({
   onBeforeTool,
   onAfterTool
 }) => {
-  var _a, _b, _c, _d, _e;
+  var _a, _b, _c, _d;
   let startedAt = Date.now();
   const traces = iteration.traces;
   const model = await cognitive.getModelDetails(Array.isArray(iteration.model) ? iteration.model[0] : iteration.model);
@@ -236,6 +238,7 @@ var executeIteration = async ({
       typeof x.content !== "string" || x.content.trim().length > 0
     )
   );
+  iteration.messages = messages;
   traces.push({
     type: "llm_call_started",
     started_at: startedAt,
@@ -251,7 +254,7 @@ var executeIteration = async ({
     messages: messages.filter((x) => x.role !== "system"),
     stopSequences: ctx.version.getStopTokens()
   });
-  const out = ((_c = (_b = output.output.choices) == null ? void 0 : _b[0]) == null ? void 0 : _c.type) === "text" && typeof ((_d = output.output.choices) == null ? void 0 : _d[0].content) === "string" ? output.output.choices[0].content : null;
+  const out = typeof ((_c = (_b = output.output.choices) == null ? void 0 : _b[0]) == null ? void 0 : _c.content) === "string" ? output.output.choices[0].content : null;
   if (!out) {
     throw new (0, _chunkPK72FAKDcjs.CognitiveError)("LLM did not return any text output");
   }
@@ -427,7 +430,7 @@ var executeIteration = async ({
     return iteration.end({
       type: "execution_error",
       execution_error: {
-        message: _nullishCoalesce(((_e = result == null ? void 0 : result.error) == null ? void 0 : _e.message), () => ( "Unknown error occurred")),
+        message: _nullishCoalesce(((_d = result == null ? void 0 : result.error) == null ? void 0 : _d.message), () => ( "Unknown error occurred")),
         stack: _chunkKIN7Y247cjs.cleanStackTrace.call(void 0, _nullishCoalesce(result.error.stack, () => ( "No stack trace available")))
       }
     });
@@ -471,7 +474,7 @@ var executeIteration = async ({
       }
     });
   }
-  const parsedExit = _chunk4XIOQWVZcjs.parseExit.call(void 0, returnValue, iteration.exits);
+  const parsedExit = _chunkB742DUFYcjs.parseExit.call(void 0, returnValue, iteration.exits);
   if (!parsedExit.success) {
     return iteration.end({
       type: "exit_error",

package/dist/llmz.d.ts

@@ -167,8 +167,9 @@ export type ExecutionProps = {
     /**
      * An instance of a Botpress Client, or an instance of Cognitive Client (@botpress/cognitive).
      * This is used to generate content using the LLM and to access the Botpress API.
+     * If not provided, a default client will be created using environment variables.
      */
-    client: Cognitive | BotpressClientLike;
+    client?: Cognitive | BotpressClientLike;
     /**
      * When provided, the execution will immediately stop when the signal is aborted.
      * This will stop the LLM generation, as well as kill the VM sandbox execution.

package/dist/snapshots.d.ts

@@ -116,8 +116,8 @@ export declare class Snapshot implements Serializable<Snapshot.JSON> {
     readonly id: string;
     readonly reason?: string;
     readonly stack: string;
-    readonly variables: Variable[];
     readonly toolCall?: ToolCall;
+    variables: Variable[];
     /**
      * Gets the current status of the snapshot.
      *

package/dist/{tool-7QXH6A24.js → tool-ADXBCB4R.js}

@@ -1,7 +1,7 @@
 import {
   Tool
-} from "./chunk-23SY6IDW.js";
-import "./chunk-U4HWJLF2.js";
+} from "./chunk-GXLMCQYF.js";
+import "./chunk-B42FGPCX.js";
 import "./chunk-WSVDMGMR.js";
 import "./chunk-YEAWWJSJ.js";
 import "./chunk-ZORRILUV.js";

package/dist/{tool-JVLOALQN.cjs → tool-JYXLFJGQ.cjs}

@@ -1,11 +1,11 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunkSOEKWFU2cjs = require('./chunk-SOEKWFU2.cjs');
-require('./chunk-INDOGCAQ.cjs');
+var _chunkQR54TZ36cjs = require('./chunk-QR54TZ36.cjs');
+require('./chunk-KED57HR5.cjs');
 require('./chunk-PK72FAKD.cjs');
 require('./chunk-KIN7Y247.cjs');
 require('./chunk-WHNOR4ZU.cjs');
 require('./chunk-UQOBUJIQ.cjs');
 
 
-exports.Tool = _chunkSOEKWFU2cjs.Tool;
+exports.Tool = _chunkQR54TZ36cjs.Tool;

package/dist/tool.d.ts

@@ -1,4 +1,4 @@
-import { TypeOf } from '@bpinternal/zui';
+import { TypeOf, z } from '@bpinternal/zui';
 import { JSONSchema7 } from 'json-schema';
 import { Serializable, ZuiType } from './types.js';
 /**
@@ -269,7 +269,7 @@ export declare class Tool<I extends ZuiType = ZuiType, O extends ZuiType = ZuiTy
      * @returns The Zui schema for output validation
      * @internal
      */
-    get zOutput(): import("@bpinternal/zui").ZodTypeAny | import("@bpinternal/zui").ZodVoid;
+    get zOutput(): import("@bpinternal/zui").ZodPromise<z.Schema<any, import("@bpinternal/zui").ZodTypeDef, any>>;
     /**
      * Renames the tool and updates its aliases.
      *

package/dist/{typings-OLI56LGT.cjs → typings-7C2RXO75.cjs}

@@ -1,10 +1,10 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunkINDOGCAQcjs = require('./chunk-INDOGCAQ.cjs');
+var _chunkKED57HR5cjs = require('./chunk-KED57HR5.cjs');
 require('./chunk-PK72FAKD.cjs');
 require('./chunk-KIN7Y247.cjs');
 require('./chunk-WHNOR4ZU.cjs');
 require('./chunk-UQOBUJIQ.cjs');
 
 
-exports.getTypings = _chunkINDOGCAQcjs.getTypings;
+exports.getTypings = _chunkKED57HR5cjs.getTypings;

package/dist/{typings-ALZEKGV6.js → typings-AQGJAMVH.js}

@@ -1,6 +1,6 @@
 import {
   getTypings
-} from "./chunk-U4HWJLF2.js";
+} from "./chunk-B42FGPCX.js";
 import "./chunk-WSVDMGMR.js";
 import "./chunk-YEAWWJSJ.js";
 import "./chunk-ZORRILUV.js";

package/package.json

@@ -2,7 +2,7 @@
   "name": "llmz",
   "type": "module",
   "description": "LLMz – An LLM-native Typescript VM built on top of Zui",
-  "version": "0.0.33",
+  "version": "0.0.35",
   "types": "./dist/index.d.ts",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",
@@ -49,6 +49,7 @@
     "ulid": "^2.3.0"
   },
   "devDependencies": {
+    "@botpress/cli": "workspace:*",
     "@microsoft/api-extractor": "^7.49.0",
     "@types/babel__core": "^7.20.5",
     "@types/babel__generator": "^7.6.8",
@@ -70,10 +71,10 @@
     "tsx": "^4.19.2"
   },
   "peerDependencies": {
-    "@botpress/client": "1.27.2",
-    "@botpress/cognitive": "0.3.1",
+    "@botpress/client": "1.28.0",
+    "@botpress/cognitive": "0.3.3",
     "@bpinternal/thicktoken": "^1.0.5",
-    "@bpinternal/zui": "1.2.1"
+    "@bpinternal/zui": "1.3.1"
   },
   "dependenciesMeta": {
     "@bpinternal/zui": {

package/dist/component-R4WTW6DZ.cjs

@@ -1,18 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});
-
-
-
-
-
-
-var _chunkZRCU35UVcjs = require('./chunk-ZRCU35UV.cjs');
-require('./chunk-KMZDFWYZ.cjs');
-require('./chunk-UQOBUJIQ.cjs');
-
-
-
-
-
-
-
-exports.Component = _chunkZRCU35UVcjs.Component; exports.assertValidComponent = _chunkZRCU35UVcjs.assertValidComponent; exports.getComponentReference = _chunkZRCU35UVcjs.getComponentReference; exports.isAnyComponent = _chunkZRCU35UVcjs.isAnyComponent; exports.isComponent = _chunkZRCU35UVcjs.isComponent; exports.renderToTsx = _chunkZRCU35UVcjs.renderToTsx;