llmz 0.0.31 → 0.0.33

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

@@ -1,6 +1,6 @@
 import {
   getTypings
-} from "./chunk-IUE5BW56.js";
+} from "./chunk-U4HWJLF2.js";
 import {
   convertObjectToZuiLiterals,
   isJsonSchema,

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

@@ -5,10 +5,10 @@ import {
 } from "./chunk-AAHUDKBY.js";
 import {
   Tool
-} from "./chunk-RC6YO5UW.js";
+} from "./chunk-23SY6IDW.js";
 import {
   LoopExceededError
-} from "./chunk-MYLTD5WT.js";
+} from "./chunk-WSVDMGMR.js";
 import {
   Exit
 } from "./chunk-VPTFUOIK.js";
@@ -972,6 +972,9 @@ var Iteration = class {
   get model() {
     return this._parameters.model;
   }
+  set model(value) {
+    this._parameters.model = value;
+  }
   get temperature() {
     return this._parameters.temperature;
   }

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

@@ -5,10 +5,10 @@
 var _chunk5BEKU5MZcjs = require('./chunk-5BEKU5MZ.cjs');
 
 
-var _chunkJ57224PDcjs = require('./chunk-J57224PD.cjs');
+var _chunkSOEKWFU2cjs = require('./chunk-SOEKWFU2.cjs');
 
 
-var _chunkT63Y6GTWcjs = require('./chunk-T63Y6GTW.cjs');
+var _chunkPK72FAKDcjs = require('./chunk-PK72FAKD.cjs');
 
 
 var _chunk3G3BS5IAcjs = require('./chunk-3G3BS5IA.cjs');
@@ -972,6 +972,9 @@ var Iteration = class {
   get model() {
     return this._parameters.model;
   }
+  set model(value) {
+    this._parameters.model = value;
+  }
   get temperature() {
     return this._parameters.temperature;
   }
@@ -1093,7 +1096,7 @@ var Context = (_class2 = class {
   
   async nextIteration() {
     if (this.iterations.length >= this.loop) {
-      throw new (0, _chunkT63Y6GTWcjs.LoopExceededError)();
+      throw new (0, _chunkPK72FAKDcjs.LoopExceededError)();
     }
     if (this.snapshot && this.snapshot.status.type === "pending") {
       throw new Error(
@@ -1251,7 +1254,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 = _chunkJ57224PDcjs.Tool.withUniqueNames(await _asyncNullishCoalesce(await getValue(this.tools, this), async () => ( [])));
+    const tools = _chunkSOEKWFU2cjs.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);
@@ -1283,7 +1286,7 @@ var Context = (_class2 = class {
       "boolean",
       "array"
     ];
-    const MessageTool = this.chat && components.length ? new (0, _chunkJ57224PDcjs.Tool)({
+    const MessageTool = this.chat && components.length ? new (0, _chunkSOEKWFU2cjs.Tool)({
       name: "Message",
       description: "Send a message to the user",
       aliases: Array.from(

package/dist/{chunk-CA7FRCZT.cjs → chunk-5FWOHMO4.cjs}

@@ -4,7 +4,7 @@
 
 
 
-var _chunkT63Y6GTWcjs = require('./chunk-T63Y6GTW.cjs');
+var _chunkPK72FAKDcjs = require('./chunk-PK72FAKD.cjs');
 
 
 var _chunkKIN7Y247cjs = require('./chunk-KIN7Y247.cjs');
@@ -2960,7 +2960,7 @@ function getCompiledCode(code, traces = []) {
       code,
       started_at: Date.now()
     });
-    throw new (0, _chunkT63Y6GTWcjs.InvalidCodeError)(err.message, code);
+    throw new (0, _chunkPK72FAKDcjs.InvalidCodeError)(err.message, code);
   }
 }
 async function runAsyncFunction(context, code, traces = [], signal = null, timeout = MAX_VM_EXECUTION_TIME) {
@@ -3036,8 +3036,8 @@ async function runAsyncFunction(context, code, traces = [], signal = null, timeo
       let currentToolCall;
       context[Identifiers.ToolCallTrackerFnIdentifier] = (callId, type, outputOrError) => {
         var _a2;
-        const temp = _chunkT63Y6GTWcjs.Signals.maybeDeserializeError(outputOrError == null ? void 0 : outputOrError.message);
-        if (type === "end" && temp instanceof _chunkT63Y6GTWcjs.SnapshotSignal && (temp == null ? void 0 : temp.toolCall)) {
+        const temp = _chunkPK72FAKDcjs.Signals.maybeDeserializeError(outputOrError == null ? void 0 : outputOrError.message);
+        if (type === "end" && temp instanceof _chunkPK72FAKDcjs.SnapshotSignal && (temp == null ? void 0 : temp.toolCall)) {
           currentToolCall = {
             ...temp.toolCall,
             assignment: (_a2 = transformed.toolCalls.get(callId)) == null ? void 0 : _a2.assignment
@@ -3535,8 +3535,8 @@ ${transformed.code}
           const errorStackResult = vm.evalCode("globalThis.__llmz_error_stack");
           const errorStack = vm.dump(errorStackResult.unwrap()) || "";
           errorStackResult.unwrap().dispose();
-          const deserializedError = _chunkT63Y6GTWcjs.Signals.maybeDeserializeError(errorValue);
-          if (deserializedError instanceof _chunkT63Y6GTWcjs.VMSignal) {
+          const deserializedError = _chunkPK72FAKDcjs.Signals.maybeDeserializeError(errorValue);
+          if (deserializedError instanceof _chunkPK72FAKDcjs.VMSignal) {
             deserializedError.stack = errorStack;
             throw deserializedError;
           }
@@ -3554,11 +3554,11 @@ ${transformed.code}
           returnValue = vm.dump(resultResult.unwrap());
           resultResult.unwrap().dispose();
         }
-        returnValue = _chunkT63Y6GTWcjs.Signals.maybeDeserializeError(returnValue);
+        returnValue = _chunkPK72FAKDcjs.Signals.maybeDeserializeError(returnValue);
         return {
           success: true,
           variables: _chunkUQOBUJIQcjs.mapValues_default.call(void 0, variables, (getter) => _chunkUQOBUJIQcjs.isFunction_default.call(void 0, getter) ? getter() : getter),
-          signal: returnValue instanceof _chunkT63Y6GTWcjs.VMSignal ? returnValue : void 0,
+          signal: returnValue instanceof _chunkPK72FAKDcjs.VMSignal ? returnValue : void 0,
           lines_executed: Array.from(lines_executed),
           return_value: returnValue
         };
@@ -3657,8 +3657,8 @@ ${transformed.code}
     let currentToolCall;
     context[Identifiers.ToolCallTrackerFnIdentifier] = (callId, type, outputOrError) => {
       var _a2;
-      const temp = _chunkT63Y6GTWcjs.Signals.maybeDeserializeError(outputOrError == null ? void 0 : outputOrError.message);
-      if (type === "end" && temp instanceof _chunkT63Y6GTWcjs.SnapshotSignal && (temp == null ? void 0 : temp.toolCall)) {
+      const temp = _chunkPK72FAKDcjs.Signals.maybeDeserializeError(outputOrError == null ? void 0 : outputOrError.message);
+      if (type === "end" && temp instanceof _chunkPK72FAKDcjs.SnapshotSignal && (temp == null ? void 0 : temp.toolCall)) {
         currentToolCall = {
           ...temp.toolCall,
           assignment: (_a2 = transformed.toolCalls.get(callId)) == null ? void 0 : _a2.assignment
@@ -3712,11 +3712,11 @@ ${transformed.code}
         await context[Identifiers.AsyncIterYieldFnIdentifier](value);
       } while (true);
     })().then((res) => {
-      res = _chunkT63Y6GTWcjs.Signals.maybeDeserializeError(res);
+      res = _chunkPK72FAKDcjs.Signals.maybeDeserializeError(res);
       return {
         success: true,
         variables: _chunkUQOBUJIQcjs.mapValues_default.call(void 0, variables, (getter) => _chunkUQOBUJIQcjs.isFunction_default.call(void 0, getter) ? getter() : getter),
-        signal: res instanceof _chunkT63Y6GTWcjs.VMSignal ? res : void 0,
+        signal: res instanceof _chunkPK72FAKDcjs.VMSignal ? res : void 0,
         lines_executed: Array.from(lines_executed),
         return_value: res
       };
@@ -3726,7 +3726,7 @@ ${transformed.code}
 }
 var handleErrorQuickJS = (err, code, _consumer, traces, variables, lines_executed, userCodeStartLine, currentToolCall) => {
   var _a, _b, _c;
-  err = _chunkT63Y6GTWcjs.Signals.maybeDeserializeError(err);
+  err = _chunkPK72FAKDcjs.Signals.maybeDeserializeError(err);
   const lines = code.split("\n");
   const stackTrace = err.stack || "";
   const LINE_OFFSET = 1;
@@ -3785,7 +3785,7 @@ var handleErrorQuickJS = (err, code, _consumer, traces, variables, lines_execute
   }
   debugUserCode = _chunkKIN7Y247cjs.cleanStackTrace.call(void 0, debugUserCode).trim();
   truncatedCode = _chunkKIN7Y247cjs.cleanStackTrace.call(void 0, truncatedCode).trim();
-  if (err instanceof _chunkT63Y6GTWcjs.VMSignal) {
+  if (err instanceof _chunkPK72FAKDcjs.VMSignal) {
     const signalError = err;
     signalError.stack = debugUserCode;
     signalError.truncatedCode = truncatedCode;
@@ -3805,8 +3805,8 @@ var handleErrorQuickJS = (err, code, _consumer, traces, variables, lines_execute
       stackTrace: debugUserCode,
       started_at: Date.now()
     });
-    const codeError = new (0, _chunkT63Y6GTWcjs.CodeExecutionError)(err.message, code, debugUserCode);
-    const deserializedError = _chunkT63Y6GTWcjs.Signals.maybeDeserializeError(codeError);
+    const codeError = new (0, _chunkPK72FAKDcjs.CodeExecutionError)(err.message, code, debugUserCode);
+    const deserializedError = _chunkPK72FAKDcjs.Signals.maybeDeserializeError(codeError);
     return {
       success: false,
       variables: _chunkUQOBUJIQcjs.mapValues_default.call(void 0, variables, (getter) => _chunkUQOBUJIQcjs.isFunction_default.call(void 0, getter) ? getter() : getter),
@@ -3818,7 +3818,7 @@ var handleErrorQuickJS = (err, code, _consumer, traces, variables, lines_execute
 };
 var handleErrorNode = (err, code, consumer, traces, variables, _lines_executed, currentToolCall) => {
   var _a, _b, _c;
-  err = _chunkT63Y6GTWcjs.Signals.maybeDeserializeError(err);
+  err = _chunkPK72FAKDcjs.Signals.maybeDeserializeError(err);
   const lines = code.split("\n");
   const stackTrace = err.stack || "";
   const LINE_OFFSET = 1;
@@ -3869,7 +3869,7 @@ var handleErrorNode = (err, code, consumer, traces, variables, _lines_executed,
   }
   debugUserCode = _chunkKIN7Y247cjs.cleanStackTrace.call(void 0, debugUserCode).trim();
   truncatedCode = _chunkKIN7Y247cjs.cleanStackTrace.call(void 0, truncatedCode).trim();
-  if (err instanceof _chunkT63Y6GTWcjs.VMSignal) {
+  if (err instanceof _chunkPK72FAKDcjs.VMSignal) {
     err.stack = debugUserCode;
     err.truncatedCode = truncatedCode;
     err.variables = _chunkUQOBUJIQcjs.mapValues_default.call(void 0, variables, (getter) => _chunkUQOBUJIQcjs.isFunction_default.call(void 0, getter) ? getter() : getter);
@@ -3883,16 +3883,16 @@ var handleErrorNode = (err, code, consumer, traces, variables, _lines_executed,
       stackTrace: debugUserCode,
       started_at: Date.now()
     });
-    throw new (0, _chunkT63Y6GTWcjs.CodeExecutionError)(err.message, code, debugUserCode);
+    throw new (0, _chunkPK72FAKDcjs.CodeExecutionError)(err.message, code, debugUserCode);
   }
 };
 var handleCatch = (err, traces, variables, lines_executed) => {
-  err = _chunkT63Y6GTWcjs.Signals.maybeDeserializeError(err);
+  err = _chunkPK72FAKDcjs.Signals.maybeDeserializeError(err);
   return {
-    success: err instanceof _chunkT63Y6GTWcjs.VMSignal ? true : false,
+    success: err instanceof _chunkPK72FAKDcjs.VMSignal ? true : false,
     variables: _chunkUQOBUJIQcjs.mapValues_default.call(void 0, variables, (getter) => _chunkUQOBUJIQcjs.isFunction_default.call(void 0, getter) ? getter() : getter),
     error: err,
-    signal: err instanceof _chunkT63Y6GTWcjs.VMSignal ? err : void 0,
+    signal: err instanceof _chunkPK72FAKDcjs.VMSignal ? err : void 0,
     traces,
     lines_executed: Array.from(lines_executed)
   };

package/dist/{chunk-IAYPKHFV.js → chunk-G3LSBWJT.js}

@@ -4,7 +4,7 @@ import {
   Signals,
   SnapshotSignal,
   VMSignal
-} from "./chunk-MYLTD5WT.js";
+} from "./chunk-WSVDMGMR.js";
 import {
   cleanStackTrace
 } from "./chunk-YEAWWJSJ.js";

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

@@ -1,6 +1,6 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } }
 
-var _chunkT63Y6GTWcjs = require('./chunk-T63Y6GTW.cjs');
+var _chunkPK72FAKDcjs = require('./chunk-PK72FAKD.cjs');
 
 
 
@@ -38,7 +38,7 @@ async function formatTypings(typings, options) {
     return result;
   } catch (err) {
     if (options == null ? void 0 : options.throwOnError) {
-      throw new (0, _chunkT63Y6GTWcjs.CodeFormattingError)(err instanceof Error ? err.message : _nullishCoalesce((err == null ? void 0 : err.toString()), () => ( "Unknown Error")), typings);
+      throw new (0, _chunkPK72FAKDcjs.CodeFormattingError)(err instanceof Error ? err.message : _nullishCoalesce((err == null ? void 0 : err.toString()), () => ( "Unknown Error")), typings);
     }
     return typings;
   }

package/dist/{chunk-T63Y6GTW.cjs → chunk-PK72FAKD.cjs}

@@ -96,10 +96,11 @@ var VMLoopSignal = class extends VMSignal {
   }
 };
 var ThinkSignal = class extends VMLoopSignal {
-  constructor(reason, context) {
+  constructor(reason, context, metadata) {
     super("Think signal received: " + reason);
     this.reason = reason;
     this.context = context;
+    this.metadata = metadata;
     this.message = Signals.serializeError(this);
     this.stack = "";
   }

package/dist/{chunk-J57224PD.cjs → chunk-SOEKWFU2.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 _chunkRLOPQZTQcjs = require('./chunk-RLOPQZTQ.cjs');
+var _chunkINDOGCAQcjs = require('./chunk-INDOGCAQ.cjs');
 
 
 
@@ -476,7 +476,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 _chunkRLOPQZTQcjs.getTypings.call(void 0, fnType, {
+    return _chunkINDOGCAQcjs.getTypings.call(void 0, fnType, {
       declaration: true
     });
   }

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

@@ -1,6 +1,6 @@
 import {
   CodeFormattingError
-} from "./chunk-MYLTD5WT.js";
+} from "./chunk-WSVDMGMR.js";
 import {
   escapeString,
   getMultilineComment,

package/dist/{chunk-MYLTD5WT.js → chunk-WSVDMGMR.js}

@@ -96,10 +96,11 @@ var VMLoopSignal = class extends VMSignal {
   }
 };
 var ThinkSignal = class extends VMLoopSignal {
-  constructor(reason, context) {
+  constructor(reason, context, metadata) {
     super("Think signal received: " + reason);
     this.reason = reason;
     this.context = context;
+    this.metadata = metadata;
     this.message = Signals.serializeError(this);
     this.stack = "";
   }

package/dist/context.d.ts

@@ -50,6 +50,7 @@ export declare namespace IterationStatuses {
         thinking_requested: {
             reason?: string;
             variables: Record<string, unknown>;
+            metadata?: Record<string, unknown>;
         };
     };
     type Callback = {
@@ -351,6 +352,7 @@ export declare class Iteration implements Serializable<Iteration.JSON> {
     get tools(): Tool<import("./types.js").ZuiType, import("./types.js").ZuiType>[];
     get objects(): ObjectInstance[];
     get model(): Models | Models[];
+    set model(value: Models | Models[]);
     get temperature(): number;
     get exits(): Exit<unknown>[];
     get instructions(): string | undefined;
@@ -363,6 +365,12 @@ export declare class Iteration implements Serializable<Iteration.JSON> {
         spend: number;
         output: string;
         model: string;
+        usage: {
+            inputCost: number;
+            outputCost: number;
+            inputTokens: number;
+            outputTokens: number;
+        };
     };
     hasExited(this: this): this is this & {
         status: IterationStatuses.ExitSuccess;
@@ -407,6 +415,12 @@ export declare class Iteration implements Serializable<Iteration.JSON> {
             spend: number;
             output: string;
             model: string;
+            usage: {
+                inputCost: number;
+                outputCost: number;
+                inputTokens: number;
+                outputTokens: number;
+            };
         } | undefined;
         transcript: Transcript.Message[];
         tools: {
@@ -517,6 +531,12 @@ export declare class Context implements Serializable<Context.JSON> {
                 spend: number;
                 output: string;
                 model: string;
+                usage: {
+                    inputCost: number;
+                    outputCost: number;
+                    inputTokens: number;
+                    outputTokens: number;
+                };
             } | undefined;
             transcript: Transcript.Message[];
             tools: {

package/dist/errors.d.ts

@@ -40,7 +40,8 @@ export declare class VMLoopSignal extends VMSignal {
 export declare class ThinkSignal extends VMLoopSignal {
     reason: string;
     context?: any;
-    constructor(reason: string, context?: any);
+    metadata?: Record<string, unknown> | undefined;
+    constructor(reason: string, context?: any, metadata?: Record<string, unknown> | undefined);
     toString(): string;
 }
 export declare class CodeExecutionError extends Error {

package/dist/index.cjs

@@ -9,21 +9,21 @@
 
 
 
-var _chunk66NCLCNTcjs = require('./chunk-66NCLCNT.cjs');
+var _chunk4XIOQWVZcjs = require('./chunk-4XIOQWVZ.cjs');
 require('./chunk-5BEKU5MZ.cjs');
 
 
-var _chunkJ57224PDcjs = require('./chunk-J57224PD.cjs');
+var _chunkSOEKWFU2cjs = require('./chunk-SOEKWFU2.cjs');
 
 
 
-var _chunkRLOPQZTQcjs = require('./chunk-RLOPQZTQ.cjs');
+var _chunkINDOGCAQcjs = require('./chunk-INDOGCAQ.cjs');
 
 
 
 
 
-var _chunkT63Y6GTWcjs = require('./chunk-T63Y6GTW.cjs');
+var _chunkPK72FAKDcjs = require('./chunk-PK72FAKD.cjs');
 require('./chunk-KIN7Y247.cjs');
 
 
@@ -297,12 +297,12 @@ async function hoistTypings(code, formatOptions) {
     } catch (err) {
       console.error(err);
       if (formatOptions.throwOnError) {
-        throw new (0, _chunkT63Y6GTWcjs.CodeFormattingError)(err instanceof Error ? err.message : String(_nullishCoalesce(err, () => ( "Unknown Error"))), code);
+        throw new (0, _chunkPK72FAKDcjs.CodeFormattingError)(err instanceof Error ? err.message : String(_nullishCoalesce(err, () => ( "Unknown Error"))), code);
       }
       break;
     }
   }
-  return _chunkRLOPQZTQcjs.formatTypings.call(void 0, code, formatOptions);
+  return _chunkINDOGCAQcjs.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 = _chunkJ57224PDcjs.Tool.withUniqueNames(_nullishCoalesce(props.tools, () => ( [])));
+    this.tools = _chunkSOEKWFU2cjs.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 _chunkRLOPQZTQcjs.getTypings.call(void 0, prop.type, {});
+          type = await _chunkINDOGCAQcjs.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 _chunkRLOPQZTQcjs.getTypings.call(void 0, fnType, {
+        let temp = await _chunkINDOGCAQcjs.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 _chunkRLOPQZTQcjs.formatTypings.call(void 0, 
+    return _chunkINDOGCAQcjs.formatTypings.call(void 0, 
       `${header}
       export namespace ${obj.name} {
       ${body}
@@ -1104,19 +1104,19 @@ var utils = {
   truncateWrappedContent: _chunkGZPN7RGHcjs.truncateWrappedContent
 };
 var execute = async (props) => {
-  const { executeContext } = await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-NB4CQ5PW.cjs")));
+  const { executeContext } = await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-HGUVAYIN.cjs")));
   return executeContext(props);
 };
 var init = async () => {
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-NB4CQ5PW.cjs")));
+  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-YCYYKKB3.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./tool-JVLOALQN.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-2LG42J6U.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-2RAAZ2YP.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./typings-OLI56LGT.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./dual-modes-UBHAMQW4.cjs")));
 };
 
@@ -1147,4 +1147,4 @@ var init = async () => {
 
 
 
-exports.Chat = Chat; exports.CitationsManager = CitationsManager; exports.Component = _chunkZRCU35UVcjs.Component; exports.DefaultComponents = DefaultComponents; exports.DefaultExit = _chunk66NCLCNTcjs.DefaultExit; exports.ErrorExecutionResult = _chunk66NCLCNTcjs.ErrorExecutionResult; exports.ExecutionResult = _chunk66NCLCNTcjs.ExecutionResult; exports.Exit = _chunk3G3BS5IAcjs.Exit; exports.ListenExit = _chunk66NCLCNTcjs.ListenExit; exports.LoopExceededError = _chunkT63Y6GTWcjs.LoopExceededError; exports.ObjectInstance = ObjectInstance; exports.PartialExecutionResult = _chunk66NCLCNTcjs.PartialExecutionResult; exports.Snapshot = _chunk66NCLCNTcjs.Snapshot; exports.SnapshotSignal = _chunkT63Y6GTWcjs.SnapshotSignal; exports.SuccessExecutionResult = _chunk66NCLCNTcjs.SuccessExecutionResult; exports.ThinkExit = _chunk66NCLCNTcjs.ThinkExit; exports.ThinkSignal = _chunkT63Y6GTWcjs.ThinkSignal; exports.Tool = _chunkJ57224PDcjs.Tool; exports.assertValidComponent = _chunkZRCU35UVcjs.assertValidComponent; exports.execute = execute; exports.getValue = _chunk66NCLCNTcjs.getValue; exports.init = init; exports.isAnyComponent = _chunkZRCU35UVcjs.isAnyComponent; exports.isComponent = _chunkZRCU35UVcjs.isComponent; exports.parseExit = _chunk66NCLCNTcjs.parseExit; exports.renderToTsx = _chunkZRCU35UVcjs.renderToTsx; exports.utils = utils;
+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;

package/dist/index.js

@@ -9,21 +9,21 @@ import {
   ThinkExit,
   getValue,
   parseExit
-} from "./chunk-WYFTNO2Y.js";
+} from "./chunk-4JN4PPGP.js";
 import "./chunk-AAHUDKBY.js";
 import {
   Tool
-} from "./chunk-RC6YO5UW.js";
+} from "./chunk-23SY6IDW.js";
 import {
   formatTypings,
   getTypings
-} from "./chunk-IUE5BW56.js";
+} from "./chunk-U4HWJLF2.js";
 import {
   CodeFormattingError,
   LoopExceededError,
   SnapshotSignal,
   ThinkSignal
-} from "./chunk-MYLTD5WT.js";
+} from "./chunk-WSVDMGMR.js";
 import "./chunk-YEAWWJSJ.js";
 import {
   Exit
@@ -1104,19 +1104,19 @@ var utils = {
   truncateWrappedContent
 };
 var execute = async (props) => {
-  const { executeContext } = await import("./llmz-VOXE65UW.js");
+  const { executeContext } = await import("./llmz-3E2JM3HM.js");
   return executeContext(props);
 };
 var init = async () => {
-  await import("./llmz-VOXE65UW.js");
+  await import("./llmz-3E2JM3HM.js");
   await import("./component-WFVDVSDK.js");
-  await import("./tool-U6SV6BZ6.js");
+  await import("./tool-7QXH6A24.js");
   await import("./exit-YLO7BY7Z.js");
   await import("./jsx-AEHVFB3L.js");
-  await import("./vm-FVQBX2XV.js");
+  await import("./vm-Y3WY2627.js");
   await import("./utils-RQHQ2KOG.js");
   await import("./truncator-BSP6PQPC.js");
-  await import("./typings-3VYUEACY.js");
+  await import("./typings-ALZEKGV6.js");
   await import("./dual-modes-LEAHGCOF.js");
 };
 export {

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

@@ -1,6 +1,6 @@
 import {
   runAsyncFunction
-} from "./chunk-IAYPKHFV.js";
+} from "./chunk-G3LSBWJT.js";
 import {
   Context,
   ErrorExecutionResult,
@@ -8,10 +8,10 @@ import {
   Snapshot,
   SuccessExecutionResult,
   parseExit
-} from "./chunk-WYFTNO2Y.js";
+} from "./chunk-4JN4PPGP.js";
 import "./chunk-AAHUDKBY.js";
-import "./chunk-RC6YO5UW.js";
-import "./chunk-IUE5BW56.js";
+import "./chunk-23SY6IDW.js";
+import "./chunk-U4HWJLF2.js";
 import {
   AssignmentError,
   CodeExecutionError,
@@ -22,7 +22,7 @@ import {
   SnapshotSignal,
   ThinkSignal,
   VMSignal
-} from "./chunk-MYLTD5WT.js";
+} from "./chunk-WSVDMGMR.js";
 import {
   cleanStackTrace
 } from "./chunk-YEAWWJSJ.js";
@@ -104,7 +104,7 @@ var executeContext = async (props) => {
 var _executeContext = async (props) => {
   var _a, _b;
   const controller = createJoinedAbortController([props.signal]);
-  const { onIterationEnd, onTrace, onExit, onBeforeExecution, onAfterTool, onBeforeTool } = props;
+  const { onIterationStart, onIterationEnd, onTrace, onExit, onBeforeExecution, onAfterTool, onBeforeTool } = props;
   const cognitive = Cognitive.isCognitiveClient(props.client) ? props.client : new Cognitive({ client: props.client });
   const cleanups = [];
   const ctx = new Context({
@@ -125,6 +125,19 @@ var _executeContext = async (props) => {
         return new ErrorExecutionResult(ctx, new LoopExceededError());
       }
       const iteration = await ctx.nextIteration();
+      try {
+        await executeOnIterationStartHook({
+          iteration,
+          ctx,
+          onIterationStart,
+          controller,
+          onIterationEnd
+        });
+      } catch (err) {
+        if (err instanceof ThinkSignal) {
+          continue;
+        }
+      }
       if (controller.signal.aborted) {
         iteration.end({
           type: "aborted",
@@ -277,7 +290,8 @@ var executeIteration = async ({
     tokens: output.meta.tokens.input + output.meta.tokens.output,
     spend: output.meta.cost.input + output.meta.cost.output,
     output: assistantResponse.raw,
-    model: `${output.meta.model.integration}:${output.meta.model.model}`
+    model: `${output.meta.model.integration}:${output.meta.model.model}`,
+    usage: output.output.usage
   };
   traces.push({
     type: "llm_call_success",
@@ -423,7 +437,8 @@ var executeIteration = async ({
       type: "thinking_requested",
       thinking_requested: {
         variables: result.signal.context,
-        reason: result.signal.reason
+        reason: result.signal.reason,
+        metadata: result.signal.metadata
       }
     });
   }
@@ -435,7 +450,6 @@ var executeIteration = async ({
       }
     });
   }
-  const _validActions = [...iteration.exits.map((x) => x.name.toLowerCase()), "think"];
   let returnValue = result.success && result.return_value ? result.return_value : null;
   const returnAction = returnValue == null ? void 0 : returnValue.action;
   if (returnAction === "think") {
@@ -546,7 +560,9 @@ function wrapTool({ tool, traces, object, iteration, beforeHook, afterHook, cont
           iteration,
           tool,
           input: input2,
-          controller
+          controller,
+          object,
+          toolCallId
         }));
         if (typeof (beforeRes == null ? void 0 : beforeRes.input) !== "undefined") {
           input2 = beforeRes.input;
@@ -559,7 +575,9 @@ function wrapTool({ tool, traces, object, iteration, beforeHook, afterHook, cont
           tool,
           input: input2,
           output: output2,
-          controller
+          controller,
+          object,
+          toolCallId
         }));
         if (typeof (afterRes == null ? void 0 : afterRes.output) !== "undefined") {
           output2 = afterRes.output;
@@ -572,10 +590,23 @@ function wrapTool({ tool, traces, object, iteration, beforeHook, afterHook, cont
           output = res;
           success = true;
           return res;
-        }).catch((err) => {
+        }).catch(async (err) => {
           if (!handleSignals(err)) {
             success = false;
             error = err;
+          } else {
+            const afterRes = await (afterHook == null ? void 0 : afterHook({
+              iteration,
+              tool,
+              input,
+              output,
+              controller,
+              object,
+              toolCallId
+            }));
+            if (typeof (afterRes == null ? void 0 : afterRes.output) !== "undefined") {
+              output = afterRes.output;
+            }
           }
           throw err;
         }).finally(() => {
@@ -624,6 +655,39 @@ function wrapTool({ tool, traces, object, iteration, beforeHook, afterHook, cont
     return output;
   };
 }
+var executeOnIterationStartHook = async (props) => {
+  const { iteration, ctx, onIterationStart, controller, onIterationEnd } = props;
+  try {
+    const hookRes = await (onIterationStart == null ? void 0 : onIterationStart(iteration, controller, ctx));
+    if (hookRes) {
+      Object.assign(iteration, hookRes);
+    }
+  } catch (err) {
+    if (err instanceof ThinkSignal) {
+      iteration.end({
+        type: "thinking_requested",
+        thinking_requested: {
+          variables: err.context,
+          reason: err.reason
+        }
+      });
+      try {
+        await (onIterationEnd == null ? void 0 : onIterationEnd(iteration, controller));
+      } catch (err2) {
+        console.error(err2);
+      }
+    } else {
+      iteration.end({
+        type: "execution_error",
+        execution_error: {
+          message: `Error in onIterationStart hook: ${getErrorMessage(err)}`,
+          stack: cleanStackTrace(err.stack ?? "No stack trace available")
+        }
+      });
+    }
+    throw err;
+  }
+};
 export {
   _executeContext,
   executeContext

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

@@ -1,6 +1,6 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } }
 
-var _chunkCA7FRCZTcjs = require('./chunk-CA7FRCZT.cjs');
+var _chunk5FWOHMO4cjs = require('./chunk-5FWOHMO4.cjs');
 
 
 
@@ -8,10 +8,10 @@ var _chunkCA7FRCZTcjs = require('./chunk-CA7FRCZT.cjs');
 
 
 
-var _chunk66NCLCNTcjs = require('./chunk-66NCLCNT.cjs');
+var _chunk4XIOQWVZcjs = require('./chunk-4XIOQWVZ.cjs');
 require('./chunk-5BEKU5MZ.cjs');
-require('./chunk-J57224PD.cjs');
-require('./chunk-RLOPQZTQ.cjs');
+require('./chunk-SOEKWFU2.cjs');
+require('./chunk-INDOGCAQ.cjs');
 
 
 
@@ -22,7 +22,7 @@ require('./chunk-RLOPQZTQ.cjs');
 
 
 
-var _chunkT63Y6GTWcjs = require('./chunk-T63Y6GTW.cjs');
+var _chunkPK72FAKDcjs = require('./chunk-PK72FAKD.cjs');
 
 
 var _chunkKIN7Y247cjs = require('./chunk-KIN7Y247.cjs');
@@ -104,10 +104,10 @@ var executeContext = async (props) => {
 var _executeContext = async (props) => {
   var _a, _b;
   const controller = createJoinedAbortController([props.signal]);
-  const { onIterationEnd, onTrace, onExit, onBeforeExecution, onAfterTool, onBeforeTool } = props;
+  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 cleanups = [];
-  const ctx = new (0, _chunk66NCLCNTcjs.Context)({
+  const ctx = new (0, _chunk4XIOQWVZcjs.Context)({
     chat: props.chat,
     instructions: props.instructions,
     objects: props.objects,
@@ -122,9 +122,22 @@ var _executeContext = async (props) => {
   try {
     while (true) {
       if (ctx.iterations.length >= ctx.loop) {
-        return new (0, _chunk66NCLCNTcjs.ErrorExecutionResult)(ctx, new (0, _chunkT63Y6GTWcjs.LoopExceededError)());
+        return new (0, _chunk4XIOQWVZcjs.ErrorExecutionResult)(ctx, new (0, _chunkPK72FAKDcjs.LoopExceededError)());
       }
       const iteration = await ctx.nextIteration();
+      try {
+        await executeOnIterationStartHook({
+          iteration,
+          ctx,
+          onIterationStart,
+          controller,
+          onIterationEnd
+        });
+      } catch (err) {
+        if (err instanceof _chunkPK72FAKDcjs.ThinkSignal) {
+          continue;
+        }
+      }
       if (controller.signal.aborted) {
         iteration.end({
           type: "aborted",
@@ -132,7 +145,7 @@ var _executeContext = async (props) => {
             reason: _nullishCoalesce(controller.signal.reason, () => ( "The operation was aborted"))
           }
         });
-        return new (0, _chunk66NCLCNTcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(controller.signal.reason, () => ( "The operation was aborted")));
+        return new (0, _chunk4XIOQWVZcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(controller.signal.reason, () => ( "The operation was aborted")));
       }
       cleanups.push(
         iteration.traces.onPush((traces) => {
@@ -168,25 +181,25 @@ var _executeContext = async (props) => {
       }
       if (iteration.status.type === "exit_success") {
         const exitName = iteration.status.exit_success.exit_name;
-        return new (0, _chunk66NCLCNTcjs.SuccessExecutionResult)(ctx, {
+        return new (0, _chunk4XIOQWVZcjs.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, _chunk66NCLCNTcjs.PartialExecutionResult)(
+        return new (0, _chunk4XIOQWVZcjs.PartialExecutionResult)(
           ctx,
           iteration.status.callback_requested.signal,
-          _chunk66NCLCNTcjs.Snapshot.fromSignal(iteration.status.callback_requested.signal)
+          _chunk4XIOQWVZcjs.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, _chunk66NCLCNTcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(iteration.error, () => ( `Unknown error. Status: ${iteration.status.type}`)));
+      return new (0, _chunk4XIOQWVZcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(iteration.error, () => ( `Unknown error. Status: ${iteration.status.type}`)));
     }
   } catch (error) {
-    return new (0, _chunk66NCLCNTcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(error, () => ( "Unknown error")));
+    return new (0, _chunk4XIOQWVZcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(error, () => ( "Unknown error")));
   } finally {
     for (const cleanup of cleanups) {
       try {
@@ -240,7 +253,7 @@ var executeIteration = async ({
   });
   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;
   if (!out) {
-    throw new (0, _chunkT63Y6GTWcjs.CognitiveError)("LLM did not return any text output");
+    throw new (0, _chunkPK72FAKDcjs.CognitiveError)("LLM did not return any text output");
   }
   const assistantResponse = ctx.version.parseAssistantResponse(out);
   iteration.code = assistantResponse.code.trim();
@@ -251,7 +264,7 @@ var executeIteration = async ({
         iteration.code = hookRes.code.trim();
       }
     } catch (err) {
-      if (err instanceof _chunkT63Y6GTWcjs.ThinkSignal) {
+      if (err instanceof _chunkPK72FAKDcjs.ThinkSignal) {
         return iteration.end({
           type: "thinking_requested",
           thinking_requested: {
@@ -277,7 +290,8 @@ var executeIteration = async ({
     tokens: output.meta.tokens.input + output.meta.tokens.output,
     spend: output.meta.cost.input + output.meta.cost.output,
     output: assistantResponse.raw,
-    model: `${output.meta.model.integration}:${output.meta.model.model}`
+    model: `${output.meta.model.integration}:${output.meta.model.model}`,
+    usage: output.output.usage
   };
   traces.push({
     type: "llm_call_success",
@@ -305,14 +319,14 @@ var executeIteration = async ({
             return;
           }
           if (!writable) {
-            throw new (0, _chunkT63Y6GTWcjs.AssignmentError)(`Property ${obj.name}.${name} is read-only and cannot be modified`);
+            throw new (0, _chunkPK72FAKDcjs.AssignmentError)(`Property ${obj.name}.${name} is read-only and cannot be modified`);
           }
           if (value2 === internalValues[name]) {
             return;
           }
           const parsed = schema.safeParse(value2);
           if (!parsed.success) {
-            throw new (0, _chunkT63Y6GTWcjs.AssignmentError)(
+            throw new (0, _chunkPK72FAKDcjs.AssignmentError)(
               `Invalid value for Object property ${obj.name}.${name}: ${getErrorMessage(parsed.error)}`
             );
           }
@@ -363,7 +377,7 @@ var executeIteration = async ({
     });
   }
   startedAt = Date.now();
-  const result = await _chunkCA7FRCZTcjs.runAsyncFunction.call(void 0, 
+  const result = await _chunk5FWOHMO4cjs.runAsyncFunction.call(void 0, 
     vmContext,
     iteration.code,
     traces,
@@ -378,7 +392,7 @@ var executeIteration = async ({
       variables: {}
     };
   });
-  if (result.error && result.error instanceof _chunkT63Y6GTWcjs.InvalidCodeError) {
+  if (result.error && result.error instanceof _chunkPK72FAKDcjs.InvalidCodeError) {
     return iteration.end({
       type: "invalid_code_error",
       invalid_code_error: {
@@ -400,7 +414,7 @@ var executeIteration = async ({
       }
     });
   }
-  if (result.error && result.error instanceof _chunkT63Y6GTWcjs.CodeExecutionError) {
+  if (result.error && result.error instanceof _chunkPK72FAKDcjs.CodeExecutionError) {
     return iteration.end({
       type: "execution_error",
       execution_error: {
@@ -418,16 +432,17 @@ var executeIteration = async ({
       }
     });
   }
-  if (result.signal instanceof _chunkT63Y6GTWcjs.ThinkSignal) {
+  if (result.signal instanceof _chunkPK72FAKDcjs.ThinkSignal) {
     return iteration.end({
       type: "thinking_requested",
       thinking_requested: {
         variables: result.signal.context,
-        reason: result.signal.reason
+        reason: result.signal.reason,
+        metadata: result.signal.metadata
       }
     });
   }
-  if (result.signal instanceof _chunkT63Y6GTWcjs.SnapshotSignal) {
+  if (result.signal instanceof _chunkPK72FAKDcjs.SnapshotSignal) {
     return iteration.end({
       type: "callback_requested",
       callback_requested: {
@@ -435,7 +450,6 @@ var executeIteration = async ({
       }
     });
   }
-  const _validActions = [...iteration.exits.map((x) => x.name.toLowerCase()), "think"];
   let returnValue = result.success && result.return_value ? result.return_value : null;
   const returnAction = returnValue == null ? void 0 : returnValue.action;
   if (returnAction === "think") {
@@ -457,7 +471,7 @@ var executeIteration = async ({
       }
     });
   }
-  const parsedExit = _chunk66NCLCNTcjs.parseExit.call(void 0, returnValue, iteration.exits);
+  const parsedExit = _chunk4XIOQWVZcjs.parseExit.call(void 0, returnValue, iteration.exits);
   if (!parsedExit.success) {
     return iteration.end({
       type: "exit_error",
@@ -518,16 +532,16 @@ function wrapTool({ tool, traces, object, iteration, beforeHook, afterHook, cont
       if (output === error2) {
         return true;
       }
-      if (error2 instanceof _chunkT63Y6GTWcjs.SnapshotSignal) {
+      if (error2 instanceof _chunkPK72FAKDcjs.SnapshotSignal) {
         error2.toolCall = {
           name: tool.name,
           inputSchema: tool.input,
           outputSchema: tool.output,
           input
         };
-        error2.message = _chunkT63Y6GTWcjs.Signals.serializeError(error2);
+        error2.message = _chunkPK72FAKDcjs.Signals.serializeError(error2);
       }
-      if (error2 instanceof _chunkT63Y6GTWcjs.ThinkSignal) {
+      if (error2 instanceof _chunkPK72FAKDcjs.ThinkSignal) {
         traces.push({
           type: "think_signal",
           started_at: Date.now(),
@@ -546,7 +560,9 @@ function wrapTool({ tool, traces, object, iteration, beforeHook, afterHook, cont
           iteration,
           tool,
           input: input2,
-          controller
+          controller,
+          object,
+          toolCallId
         }));
         if (typeof (beforeRes == null ? void 0 : beforeRes.input) !== "undefined") {
           input2 = beforeRes.input;
@@ -559,7 +575,9 @@ function wrapTool({ tool, traces, object, iteration, beforeHook, afterHook, cont
           tool,
           input: input2,
           output: output2,
-          controller
+          controller,
+          object,
+          toolCallId
         }));
         if (typeof (afterRes == null ? void 0 : afterRes.output) !== "undefined") {
           output2 = afterRes.output;
@@ -572,10 +590,23 @@ function wrapTool({ tool, traces, object, iteration, beforeHook, afterHook, cont
           output = res;
           success = true;
           return res;
-        }).catch((err) => {
+        }).catch(async (err) => {
           if (!handleSignals(err)) {
             success = false;
             error = err;
+          } else {
+            const afterRes = await (afterHook == null ? void 0 : afterHook({
+              iteration,
+              tool,
+              input,
+              output,
+              controller,
+              object,
+              toolCallId
+            }));
+            if (typeof (afterRes == null ? void 0 : afterRes.output) !== "undefined") {
+              output = afterRes.output;
+            }
           }
           throw err;
         }).finally(() => {
@@ -618,12 +649,45 @@ function wrapTool({ tool, traces, object, iteration, beforeHook, afterHook, cont
     if (!success) {
       throw error;
     }
-    if (output instanceof _chunkT63Y6GTWcjs.VMSignal) {
+    if (output instanceof _chunkPK72FAKDcjs.VMSignal) {
       throw output;
     }
     return output;
   };
 }
+var executeOnIterationStartHook = async (props) => {
+  const { iteration, ctx, onIterationStart, controller, onIterationEnd } = props;
+  try {
+    const hookRes = await (onIterationStart == null ? void 0 : onIterationStart(iteration, controller, ctx));
+    if (hookRes) {
+      Object.assign(iteration, hookRes);
+    }
+  } catch (err) {
+    if (err instanceof _chunkPK72FAKDcjs.ThinkSignal) {
+      iteration.end({
+        type: "thinking_requested",
+        thinking_requested: {
+          variables: err.context,
+          reason: err.reason
+        }
+      });
+      try {
+        await (onIterationEnd == null ? void 0 : onIterationEnd(iteration, controller));
+      } catch (err2) {
+        console.error(err2);
+      }
+    } else {
+      iteration.end({
+        type: "execution_error",
+        execution_error: {
+          message: `Error in onIterationStart hook: ${getErrorMessage(err)}`,
+          stack: _chunkKIN7Y247cjs.cleanStackTrace.call(void 0, _nullishCoalesce(err.stack, () => ( "No stack trace available")))
+        }
+      });
+    }
+    throw err;
+  }
+};
 
 
 

package/dist/llmz.d.ts

@@ -22,6 +22,14 @@ export type ExecutionHooks = {
         trace: Trace;
         iteration: number;
     }) => void;
+    /**
+     * BLOCKING HOOK
+     *   This hook will block the execution of the iteration until it resolves.
+     *
+     * This hook will be called before each iteration starts, regardless of the status.
+     * This is useful for logging or dynamically change model arguments
+     */
+    onIterationStart?: (iteration: Iteration, controller: AbortController, context: Context) => Promise<void | Partial<Iteration>> | void | Partial<Iteration>;
     /**
      * BLOCKING HOOK
      *   This hook will block the execution of the iteration until it resolves.
@@ -70,6 +78,8 @@ export type ExecutionHooks = {
         tool: Tool;
         input: any;
         controller: AbortController;
+        toolCallId: string;
+        object?: string;
     }) => Promise<{
         input?: any;
     } | void>;
@@ -88,6 +98,8 @@ export type ExecutionHooks = {
         input: any;
         output: any;
         controller: AbortController;
+        toolCallId: string;
+        object?: string;
     }) => Promise<{
         output?: any;
     } | void>;

package/dist/objects.d.ts

@@ -1,6 +1,5 @@
-import { z } from '@bpinternal/zui';
 import { Tool } from './tool.js';
-import { Serializable } from './types.js';
+import { Serializable, ZuiType } from './types.js';
 /**
  * Defines a property within an ObjectInstance.
  *
@@ -25,7 +24,7 @@ export type ObjectProperty = {
     /** The current value of the property */
     value: any;
     /** Optional Zod schema for validation when the property is modified */
-    type?: z.Schema;
+    type?: ZuiType;
     /** Optional human-readable description of the property */
     description?: string;
     /** Whether the LLM can modify this property (default: false) */

package/dist/result.d.ts

@@ -405,6 +405,12 @@ export declare class SuccessExecutionResult<TOutput = unknown> extends Execution
                     spend: number;
                     output: string;
                     model: string;
+                    usage: {
+                        inputCost: number;
+                        outputCost: number;
+                        inputTokens: number;
+                        outputTokens: number;
+                    };
                 } | undefined;
                 transcript: import("./transcript.js").Transcript.Message[];
                 tools: {
@@ -560,6 +566,12 @@ export declare class ErrorExecutionResult extends ExecutionResult implements Ser
                     spend: number;
                     output: string;
                     model: string;
+                    usage: {
+                        inputCost: number;
+                        outputCost: number;
+                        inputTokens: number;
+                        outputTokens: number;
+                    };
                 } | undefined;
                 transcript: import("./transcript.js").Transcript.Message[];
                 tools: {
@@ -735,6 +747,12 @@ export declare class PartialExecutionResult extends ExecutionResult implements S
                     spend: number;
                     output: string;
                     model: string;
+                    usage: {
+                        inputCost: number;
+                        outputCost: number;
+                        inputTokens: number;
+                        outputTokens: number;
+                    };
                 } | undefined;
                 transcript: import("./transcript.js").Transcript.Message[];
                 tools: {

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

@@ -1,8 +1,8 @@
 import {
   Tool
-} from "./chunk-RC6YO5UW.js";
-import "./chunk-IUE5BW56.js";
-import "./chunk-MYLTD5WT.js";
+} from "./chunk-23SY6IDW.js";
+import "./chunk-U4HWJLF2.js";
+import "./chunk-WSVDMGMR.js";
 import "./chunk-YEAWWJSJ.js";
 import "./chunk-ZORRILUV.js";
 import "./chunk-7WRN4E42.js";

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

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

package/dist/{typings-3VYUEACY.js → typings-ALZEKGV6.js}

@@ -1,7 +1,7 @@
 import {
   getTypings
-} from "./chunk-IUE5BW56.js";
-import "./chunk-MYLTD5WT.js";
+} from "./chunk-U4HWJLF2.js";
+import "./chunk-WSVDMGMR.js";
 import "./chunk-YEAWWJSJ.js";
 import "./chunk-ZORRILUV.js";
 import "./chunk-7WRN4E42.js";

package/dist/{typings-2RAAZ2YP.cjs → typings-OLI56LGT.cjs}

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

package/dist/{vm-2LG42J6U.cjs → vm-VFORKC54.cjs}

@@ -1,10 +1,10 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunkCA7FRCZTcjs = require('./chunk-CA7FRCZT.cjs');
-require('./chunk-T63Y6GTW.cjs');
+var _chunk5FWOHMO4cjs = require('./chunk-5FWOHMO4.cjs');
+require('./chunk-PK72FAKD.cjs');
 require('./chunk-KIN7Y247.cjs');
 require('./chunk-KMZDFWYZ.cjs');
 require('./chunk-UQOBUJIQ.cjs');
 
 
-exports.runAsyncFunction = _chunkCA7FRCZTcjs.runAsyncFunction;
+exports.runAsyncFunction = _chunk5FWOHMO4cjs.runAsyncFunction;

package/dist/{vm-FVQBX2XV.js → vm-Y3WY2627.js}

@@ -1,7 +1,7 @@
 import {
   runAsyncFunction
-} from "./chunk-IAYPKHFV.js";
-import "./chunk-MYLTD5WT.js";
+} from "./chunk-G3LSBWJT.js";
+import "./chunk-WSVDMGMR.js";
 import "./chunk-YEAWWJSJ.js";
 import "./chunk-ORQP26SZ.js";
 import "./chunk-7WRN4E42.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.31",
+  "version": "0.0.33",
   "types": "./dist/index.d.ts",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",
@@ -70,8 +70,8 @@
     "tsx": "^4.19.2"
   },
   "peerDependencies": {
-    "@botpress/client": "1.27.1",
-    "@botpress/cognitive": "0.2.1",
+    "@botpress/client": "1.27.2",
+    "@botpress/cognitive": "0.3.1",
     "@bpinternal/thicktoken": "^1.0.5",
     "@bpinternal/zui": "1.2.1"
   },