llmz 0.0.25 → 0.0.27

package/dist/{dual-modes-7FI4T35O.js → dual-modes-DW3KRXT2.js}

@@ -1,7 +1,7 @@
 import {
   DualModePrompt
-} from "./chunk-2D2DE7CD.js";
-import "./chunk-JQBT7UWN.js";
+} from "./chunk-HCC76DDO.js";
+import "./chunk-YEAWWJSJ.js";
 import "./chunk-GGWM6X2K.js";
 import "./chunk-ORQP26SZ.js";
 import "./chunk-3JYCCI4S.js";

package/dist/{dual-modes-OFHV2C3X.cjs → dual-modes-F4UV5VAZ.cjs}

@@ -1,7 +1,7 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunkPIDLNYIPcjs = require('./chunk-PIDLNYIP.cjs');
-require('./chunk-IKSIOIIP.cjs');
+var _chunkWP4F6KMWcjs = require('./chunk-WP4F6KMW.cjs');
+require('./chunk-KIN7Y247.cjs');
 require('./chunk-ZRCU35UV.cjs');
 require('./chunk-KMZDFWYZ.cjs');
 require('./chunk-GZPN7RGH.cjs');
@@ -9,4 +9,4 @@ require('./chunk-WHNOR4ZU.cjs');
 require('./chunk-UQOBUJIQ.cjs');
 
 
-exports.DualModePrompt = _chunkPIDLNYIPcjs.DualModePrompt;
+exports.DualModePrompt = _chunkWP4F6KMWcjs.DualModePrompt;

package/dist/errors.d.ts

@@ -62,3 +62,6 @@ export declare class CodeFormattingError extends Error {
 export declare class AssignmentError extends Error {
     constructor(message: string);
 }
+export declare class CognitiveError extends Error {
+    constructor(message: string);
+}

package/dist/index.cjs

@@ -8,22 +8,22 @@
 
 
 
-var _chunkUGXCQIHBcjs = require('./chunk-UGXCQIHB.cjs');
-require('./chunk-PIDLNYIP.cjs');
+var _chunkVADA6DMRcjs = require('./chunk-VADA6DMR.cjs');
+require('./chunk-WP4F6KMW.cjs');
 
 
-var _chunkXGJOEQMWcjs = require('./chunk-XGJOEQMW.cjs');
+var _chunk273DEMEUcjs = require('./chunk-273DEMEU.cjs');
 
 
 
-var _chunkFZJHYLM2cjs = require('./chunk-FZJHYLM2.cjs');
+var _chunkRLOPQZTQcjs = require('./chunk-RLOPQZTQ.cjs');
 
 
 
 
 
-var _chunkJDABP4SDcjs = require('./chunk-JDABP4SD.cjs');
-require('./chunk-IKSIOIIP.cjs');
+var _chunkT63Y6GTWcjs = require('./chunk-T63Y6GTW.cjs');
+require('./chunk-KIN7Y247.cjs');
 
 
 var _chunk3G3BS5IAcjs = require('./chunk-3G3BS5IA.cjs');
@@ -296,12 +296,12 @@ async function hoistTypings(code, formatOptions) {
     } catch (err) {
       console.error(err);
       if (formatOptions.throwOnError) {
-        throw new (0, _chunkJDABP4SDcjs.CodeFormattingError)(err instanceof Error ? err.message : String(_nullishCoalesce(err, () => ( "Unknown Error"))), code);
+        throw new (0, _chunkT63Y6GTWcjs.CodeFormattingError)(err instanceof Error ? err.message : String(_nullishCoalesce(err, () => ( "Unknown Error"))), code);
       }
       break;
     }
   }
-  return _chunkFZJHYLM2cjs.formatTypings.call(void 0, code, formatOptions);
+  return _chunkRLOPQZTQcjs.formatTypings.call(void 0, code, formatOptions);
 }
 
 // src/objects.ts
@@ -427,7 +427,7 @@ var ObjectInstance = class {
     this.description = props.description;
     this.metadata = _nullishCoalesce(props.metadata, () => ( {}));
     this.properties = props.properties;
-    this.tools = _chunkXGJOEQMWcjs.Tool.withUniqueNames(_nullishCoalesce(props.tools, () => ( [])));
+    this.tools = _chunk273DEMEUcjs.Tool.withUniqueNames(_nullishCoalesce(props.tools, () => ( [])));
   }
   /**
    * Generates TypeScript namespace declarations for this object.
@@ -508,7 +508,7 @@ function getObjectTypings(obj) {
         }
         let type = "unknown";
         if (prop.type) {
-          type = await _chunkFZJHYLM2cjs.getTypings.call(void 0, prop.type, {});
+          type = await _chunkRLOPQZTQcjs.getTypings.call(void 0, prop.type, {});
         } else if (prop.value !== void 0) {
           type = typeof prop.value;
         }
@@ -528,7 +528,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 _chunkFZJHYLM2cjs.getTypings.call(void 0, fnType, {
+        let temp = await _chunkRLOPQZTQcjs.getTypings.call(void 0, fnType, {
           declaration: true
         });
         temp = temp.replace("declare function ", "function ");
@@ -551,7 +551,7 @@ function getObjectTypings(obj) {
     if ((_a = obj.description) == null ? void 0 : _a.trim().length) {
       header = _chunkWHNOR4ZUcjs.getMultilineComment.call(void 0, obj.description);
     }
-    return _chunkFZJHYLM2cjs.formatTypings.call(void 0, 
+    return _chunkRLOPQZTQcjs.formatTypings.call(void 0, 
       `${header}
       export namespace ${obj.name} {
       ${body}
@@ -1103,20 +1103,20 @@ var utils = {
   truncateWrappedContent: _chunkGZPN7RGHcjs.truncateWrappedContent
 };
 var execute = async (props) => {
-  const { executeContext } = await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-HRL52DC6.cjs")));
+  const { executeContext } = await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-M7XYQZT7.cjs")));
   return executeContext(props);
 };
 var init = async () => {
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-HRL52DC6.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-M7XYQZT7.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./component-R4WTW6DZ.cjs")));
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./tool-O4SFRIE4.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./tool-GMYMVXUK.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-PRZ5GEM2.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./vm-YJX6M4NJ.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-Y45GMPZT.cjs")));
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./dual-modes-OFHV2C3X.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./typings-2RAAZ2YP.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./dual-modes-F4UV5VAZ.cjs")));
 };
 
 
@@ -1145,4 +1145,4 @@ var init = async () => {
 
 
 
-exports.Chat = Chat; exports.CitationsManager = CitationsManager; exports.Component = _chunkZRCU35UVcjs.Component; exports.DefaultComponents = DefaultComponents; exports.DefaultExit = _chunkUGXCQIHBcjs.DefaultExit; exports.ErrorExecutionResult = _chunkUGXCQIHBcjs.ErrorExecutionResult; exports.ExecutionResult = _chunkUGXCQIHBcjs.ExecutionResult; exports.Exit = _chunk3G3BS5IAcjs.Exit; exports.ListenExit = _chunkUGXCQIHBcjs.ListenExit; exports.LoopExceededError = _chunkJDABP4SDcjs.LoopExceededError; exports.ObjectInstance = ObjectInstance; exports.PartialExecutionResult = _chunkUGXCQIHBcjs.PartialExecutionResult; exports.Snapshot = _chunkUGXCQIHBcjs.Snapshot; exports.SnapshotSignal = _chunkJDABP4SDcjs.SnapshotSignal; exports.SuccessExecutionResult = _chunkUGXCQIHBcjs.SuccessExecutionResult; exports.ThinkExit = _chunkUGXCQIHBcjs.ThinkExit; exports.ThinkSignal = _chunkJDABP4SDcjs.ThinkSignal; exports.Tool = _chunkXGJOEQMWcjs.Tool; exports.assertValidComponent = _chunkZRCU35UVcjs.assertValidComponent; exports.execute = execute; exports.getValue = _chunkUGXCQIHBcjs.getValue; exports.init = init; exports.isAnyComponent = _chunkZRCU35UVcjs.isAnyComponent; exports.isComponent = _chunkZRCU35UVcjs.isComponent; exports.renderToTsx = _chunkZRCU35UVcjs.renderToTsx; exports.utils = utils;
+exports.Chat = Chat; exports.CitationsManager = CitationsManager; exports.Component = _chunkZRCU35UVcjs.Component; exports.DefaultComponents = DefaultComponents; exports.DefaultExit = _chunkVADA6DMRcjs.DefaultExit; exports.ErrorExecutionResult = _chunkVADA6DMRcjs.ErrorExecutionResult; exports.ExecutionResult = _chunkVADA6DMRcjs.ExecutionResult; exports.Exit = _chunk3G3BS5IAcjs.Exit; exports.ListenExit = _chunkVADA6DMRcjs.ListenExit; exports.LoopExceededError = _chunkT63Y6GTWcjs.LoopExceededError; exports.ObjectInstance = ObjectInstance; exports.PartialExecutionResult = _chunkVADA6DMRcjs.PartialExecutionResult; exports.Snapshot = _chunkVADA6DMRcjs.Snapshot; exports.SnapshotSignal = _chunkT63Y6GTWcjs.SnapshotSignal; exports.SuccessExecutionResult = _chunkVADA6DMRcjs.SuccessExecutionResult; exports.ThinkExit = _chunkVADA6DMRcjs.ThinkExit; exports.ThinkSignal = _chunkT63Y6GTWcjs.ThinkSignal; exports.Tool = _chunk273DEMEUcjs.Tool; exports.assertValidComponent = _chunkZRCU35UVcjs.assertValidComponent; exports.execute = execute; exports.getValue = _chunkVADA6DMRcjs.getValue; exports.init = init; exports.isAnyComponent = _chunkZRCU35UVcjs.isAnyComponent; exports.isComponent = _chunkZRCU35UVcjs.isComponent; exports.renderToTsx = _chunkZRCU35UVcjs.renderToTsx; exports.utils = utils;

package/dist/index.d.ts

@@ -39,7 +39,7 @@ export declare const utils: {
  * @param props.exits - Array of Exit definitions for structured completion (static or dynamic)
  * @param props.snapshot - Optional Snapshot to resume paused execution
  * @param props.signal - Optional AbortSignal to cancel execution
- * @param props.model - Optional model name (static or dynamic function)
+ * @param props.model - Optional model name (or array or models to use as fallback) (static or dynamic function)
  * @param props.temperature - Optional temperature value (static or dynamic function)
  * @param props.options - Optional execution options (loop limit, timeout)
  * @param props.onTrace - Optional non-blocking hook for monitoring traces during execution

package/dist/index.js

@@ -8,22 +8,22 @@ import {
   SuccessExecutionResult,
   ThinkExit,
   getValue
-} from "./chunk-DBRWIHIH.js";
-import "./chunk-2D2DE7CD.js";
+} from "./chunk-KQPGB6GB.js";
+import "./chunk-HCC76DDO.js";
 import {
   Tool
-} from "./chunk-A7QHWVD7.js";
+} from "./chunk-7POUFE5M.js";
 import {
   formatTypings,
   getTypings
-} from "./chunk-EE6NVDID.js";
+} from "./chunk-IUE5BW56.js";
 import {
   CodeFormattingError,
   LoopExceededError,
   SnapshotSignal,
   ThinkSignal
-} from "./chunk-JKVVQN2P.js";
-import "./chunk-JQBT7UWN.js";
+} from "./chunk-MYLTD5WT.js";
+import "./chunk-YEAWWJSJ.js";
 import {
   Exit
 } from "./chunk-VPTFUOIK.js";
@@ -1103,20 +1103,20 @@ var utils = {
   truncateWrappedContent
 };
 var execute = async (props) => {
-  const { executeContext } = await import("./llmz-M5AWNDMV.js");
+  const { executeContext } = await import("./llmz-WPO4HYQY.js");
   return executeContext(props);
 };
 var init = async () => {
-  await import("./llmz-M5AWNDMV.js");
+  await import("./llmz-WPO4HYQY.js");
   await import("./component-WFVDVSDK.js");
-  await import("./tool-PCOYOCRH.js");
+  await import("./tool-GEBXW6AQ.js");
   await import("./exit-YLO7BY7Z.js");
   await import("./jsx-AEHVFB3L.js");
-  await import("./vm-CYC5RLKU.js");
+  await import("./vm-I4G4IH7T.js");
   await import("./utils-RQHQ2KOG.js");
   await import("./truncator-BSP6PQPC.js");
-  await import("./typings-WYHEFCYB.js");
-  await import("./dual-modes-7FI4T35O.js");
+  await import("./typings-3VYUEACY.js");
+  await import("./dual-modes-DW3KRXT2.js");
 };
 export {
   Chat,

package/dist/{llmz-HRL52DC6.cjs → llmz-M7XYQZT7.cjs}

@@ -1,16 +1,16 @@
 "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 _chunkBNDY7RGXcjs = require('./chunk-BNDY7RGX.cjs');
+var _chunkM3LCKL55cjs = require('./chunk-M3LCKL55.cjs');
 
 
 
 
 
 
-var _chunkUGXCQIHBcjs = require('./chunk-UGXCQIHB.cjs');
-require('./chunk-PIDLNYIP.cjs');
-require('./chunk-XGJOEQMW.cjs');
-require('./chunk-FZJHYLM2.cjs');
+var _chunkVADA6DMRcjs = require('./chunk-VADA6DMR.cjs');
+require('./chunk-WP4F6KMW.cjs');
+require('./chunk-273DEMEU.cjs');
+require('./chunk-RLOPQZTQ.cjs');
 
 
 
@@ -20,10 +20,11 @@ require('./chunk-FZJHYLM2.cjs');
 
 
 
-var _chunkJDABP4SDcjs = require('./chunk-JDABP4SD.cjs');
 
+var _chunkT63Y6GTWcjs = require('./chunk-T63Y6GTW.cjs');
 
-var _chunkIKSIOIIPcjs = require('./chunk-IKSIOIIP.cjs');
+
+var _chunkKIN7Y247cjs = require('./chunk-KIN7Y247.cjs');
 require('./chunk-3G3BS5IA.cjs');
 require('./chunk-ZRCU35UV.cjs');
 require('./chunk-KMZDFWYZ.cjs');
@@ -105,7 +106,7 @@ var _executeContext = async (props) => {
   const { 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, _chunkUGXCQIHBcjs.Context)({
+  const ctx = new (0, _chunkVADA6DMRcjs.Context)({
     chat: props.chat,
     instructions: props.instructions,
     objects: props.objects,
@@ -120,7 +121,7 @@ var _executeContext = async (props) => {
   try {
     while (true) {
       if (ctx.iterations.length >= ctx.loop) {
-        return new (0, _chunkUGXCQIHBcjs.ErrorExecutionResult)(ctx, new (0, _chunkJDABP4SDcjs.LoopExceededError)());
+        return new (0, _chunkVADA6DMRcjs.ErrorExecutionResult)(ctx, new (0, _chunkT63Y6GTWcjs.LoopExceededError)());
       }
       const iteration = await ctx.nextIteration();
       if (controller.signal.aborted) {
@@ -130,7 +131,7 @@ var _executeContext = async (props) => {
             reason: _nullishCoalesce(controller.signal.reason, () => ( "The operation was aborted"))
           }
         });
-        return new (0, _chunkUGXCQIHBcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(controller.signal.reason, () => ( "The operation was aborted")));
+        return new (0, _chunkVADA6DMRcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(controller.signal.reason, () => ( "The operation was aborted")));
       }
       cleanups.push(
         iteration.traces.onPush((traces) => {
@@ -155,7 +156,7 @@ var _executeContext = async (props) => {
           type: "execution_error",
           execution_error: {
             message: "An unexpected error occurred: " + getErrorMessage(err),
-            stack: _chunkIKSIOIIPcjs.cleanStackTrace.call(void 0, _nullishCoalesce(err.stack, () => ( "No stack trace available")))
+            stack: _chunkKIN7Y247cjs.cleanStackTrace.call(void 0, _nullishCoalesce(err.stack, () => ( "No stack trace available")))
           }
         });
       }
@@ -166,25 +167,25 @@ var _executeContext = async (props) => {
       }
       if (iteration.status.type === "exit_success") {
         const exitName = iteration.status.exit_success.exit_name;
-        return new (0, _chunkUGXCQIHBcjs.SuccessExecutionResult)(ctx, {
+        return new (0, _chunkVADA6DMRcjs.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, _chunkUGXCQIHBcjs.PartialExecutionResult)(
+        return new (0, _chunkVADA6DMRcjs.PartialExecutionResult)(
           ctx,
           iteration.status.callback_requested.signal,
-          _chunkUGXCQIHBcjs.Snapshot.fromSignal(iteration.status.callback_requested.signal)
+          _chunkVADA6DMRcjs.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, _chunkUGXCQIHBcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(iteration.error, () => ( `Unknown error. Status: ${iteration.status.type}`)));
+      return new (0, _chunkVADA6DMRcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(iteration.error, () => ( `Unknown error. Status: ${iteration.status.type}`)));
     }
   } catch (error) {
-    return new (0, _chunkUGXCQIHBcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(error, () => ( "Unknown error")));
+    return new (0, _chunkVADA6DMRcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(error, () => ( "Unknown error")));
   } finally {
     for (const cleanup of cleanups) {
       try {
@@ -207,8 +208,8 @@ var executeIteration = async ({
   var _a, _b, _c, _d, _e;
   let startedAt = Date.now();
   const traces = iteration.traces;
-  const model = await cognitive.getModelDetails(_nullishCoalesce(iteration.model, () => ( "best")));
-  const modelLimit = model.input.maxTokens;
+  const model = await cognitive.getModelDetails(Array.isArray(iteration.model) ? iteration.model[0] : iteration.model);
+  const modelLimit = Math.max(model.input.maxTokens, 8e3);
   const responseLengthBuffer = getModelOutputLimit(modelLimit);
   const messages = _chunkGZPN7RGHcjs.truncateWrappedContent.call(void 0, {
     messages: iteration.messages,
@@ -238,7 +239,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 Error("No output from LLM");
+    throw new (0, _chunkT63Y6GTWcjs.CognitiveError)("LLM did not return any text output");
   }
   const assistantResponse = ctx.version.parseAssistantResponse(out);
   iteration.code = assistantResponse.code.trim();
@@ -249,7 +250,7 @@ var executeIteration = async ({
         iteration.code = hookRes.code.trim();
       }
     } catch (err) {
-      if (err instanceof _chunkJDABP4SDcjs.ThinkSignal) {
+      if (err instanceof _chunkT63Y6GTWcjs.ThinkSignal) {
         return iteration.end({
           type: "thinking_requested",
           thinking_requested: {
@@ -262,7 +263,7 @@ var executeIteration = async ({
         type: "execution_error",
         execution_error: {
           message: `Error in onBeforeExecution hook: ${getErrorMessage(err)}`,
-          stack: _chunkIKSIOIIPcjs.cleanStackTrace.call(void 0, _nullishCoalesce(err.stack, () => ( "No stack trace available")))
+          stack: _chunkKIN7Y247cjs.cleanStackTrace.call(void 0, _nullishCoalesce(err.stack, () => ( "No stack trace available")))
         }
       });
     }
@@ -303,14 +304,14 @@ var executeIteration = async ({
             return;
           }
           if (!writable) {
-            throw new (0, _chunkJDABP4SDcjs.AssignmentError)(`Property ${obj.name}.${name} is read-only and cannot be modified`);
+            throw new (0, _chunkT63Y6GTWcjs.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, _chunkJDABP4SDcjs.AssignmentError)(
+            throw new (0, _chunkT63Y6GTWcjs.AssignmentError)(
               `Invalid value for Object property ${obj.name}.${name}: ${getErrorMessage(parsed.error)}`
             );
           }
@@ -361,7 +362,7 @@ var executeIteration = async ({
     });
   }
   startedAt = Date.now();
-  const result = await _chunkBNDY7RGXcjs.runAsyncFunction.call(void 0, 
+  const result = await _chunkM3LCKL55cjs.runAsyncFunction.call(void 0, 
     vmContext,
     iteration.code,
     traces,
@@ -376,7 +377,7 @@ var executeIteration = async ({
       variables: {}
     };
   });
-  if (result.error && result.error instanceof _chunkJDABP4SDcjs.InvalidCodeError) {
+  if (result.error && result.error instanceof _chunkT63Y6GTWcjs.InvalidCodeError) {
     return iteration.end({
       type: "invalid_code_error",
       invalid_code_error: {
@@ -390,12 +391,12 @@ var executeIteration = async ({
     started_at: startedAt,
     ended_at: Date.now()
   });
-  if (result.error && result.error instanceof _chunkJDABP4SDcjs.CodeExecutionError) {
+  if (result.error && result.error instanceof _chunkT63Y6GTWcjs.CodeExecutionError) {
     return iteration.end({
       type: "execution_error",
       execution_error: {
         message: result.error.message,
-        stack: _chunkIKSIOIIPcjs.cleanStackTrace.call(void 0, _nullishCoalesce(_nullishCoalesce(result.error.stacktrace, () => ( result.error.stack)), () => ( "No stack trace available")))
+        stack: _chunkKIN7Y247cjs.cleanStackTrace.call(void 0, _nullishCoalesce(_nullishCoalesce(result.error.stacktrace, () => ( result.error.stack)), () => ( "No stack trace available")))
       }
     });
   }
@@ -412,11 +413,11 @@ var executeIteration = async ({
       type: "execution_error",
       execution_error: {
         message: _nullishCoalesce(((_e = result == null ? void 0 : result.error) == null ? void 0 : _e.message), () => ( "Unknown error occurred")),
-        stack: _chunkIKSIOIIPcjs.cleanStackTrace.call(void 0, _nullishCoalesce(result.error.stack, () => ( "No stack trace available")))
+        stack: _chunkKIN7Y247cjs.cleanStackTrace.call(void 0, _nullishCoalesce(result.error.stack, () => ( "No stack trace available")))
       }
     });
   }
-  if (result.signal instanceof _chunkJDABP4SDcjs.ThinkSignal) {
+  if (result.signal instanceof _chunkT63Y6GTWcjs.ThinkSignal) {
     return iteration.end({
       type: "thinking_requested",
       thinking_requested: {
@@ -425,7 +426,7 @@ var executeIteration = async ({
       }
     });
   }
-  if (result.signal instanceof _chunkJDABP4SDcjs.SnapshotSignal) {
+  if (result.signal instanceof _chunkT63Y6GTWcjs.SnapshotSignal) {
     return iteration.end({
       type: "callback_requested",
       callback_requested: {
@@ -538,16 +539,16 @@ function wrapTool({ tool, traces, object, iteration, beforeHook, afterHook, cont
       if (output === error2) {
         return true;
       }
-      if (error2 instanceof _chunkJDABP4SDcjs.SnapshotSignal) {
+      if (error2 instanceof _chunkT63Y6GTWcjs.SnapshotSignal) {
         error2.toolCall = {
           name: tool.name,
           inputSchema: tool.input,
           outputSchema: tool.output,
           input
         };
-        error2.message = _chunkJDABP4SDcjs.Signals.serializeError(error2);
+        error2.message = _chunkT63Y6GTWcjs.Signals.serializeError(error2);
       }
-      if (error2 instanceof _chunkJDABP4SDcjs.ThinkSignal) {
+      if (error2 instanceof _chunkT63Y6GTWcjs.ThinkSignal) {
         traces.push({
           type: "think_signal",
           started_at: Date.now(),
@@ -638,7 +639,7 @@ function wrapTool({ tool, traces, object, iteration, beforeHook, afterHook, cont
     if (!success) {
       throw error;
     }
-    if (output instanceof _chunkJDABP4SDcjs.VMSignal) {
+    if (output instanceof _chunkT63Y6GTWcjs.VMSignal) {
       throw output;
     }
     return output;

package/dist/{llmz-M5AWNDMV.js → llmz-WPO4HYQY.js}

@@ -1,29 +1,30 @@
 import {
   runAsyncFunction
-} from "./chunk-LUX4X7XO.js";
+} from "./chunk-7G5QAIUX.js";
 import {
   Context,
   ErrorExecutionResult,
   PartialExecutionResult,
   Snapshot,
   SuccessExecutionResult
-} from "./chunk-DBRWIHIH.js";
-import "./chunk-2D2DE7CD.js";
-import "./chunk-A7QHWVD7.js";
-import "./chunk-EE6NVDID.js";
+} from "./chunk-KQPGB6GB.js";
+import "./chunk-HCC76DDO.js";
+import "./chunk-7POUFE5M.js";
+import "./chunk-IUE5BW56.js";
 import {
   AssignmentError,
   CodeExecutionError,
+  CognitiveError,
   InvalidCodeError,
   LoopExceededError,
   Signals,
   SnapshotSignal,
   ThinkSignal,
   VMSignal
-} from "./chunk-JKVVQN2P.js";
+} from "./chunk-MYLTD5WT.js";
 import {
   cleanStackTrace
-} from "./chunk-JQBT7UWN.js";
+} from "./chunk-YEAWWJSJ.js";
 import "./chunk-VPTFUOIK.js";
 import "./chunk-GGWM6X2K.js";
 import "./chunk-ORQP26SZ.js";
@@ -207,8 +208,8 @@ var executeIteration = async ({
   var _a, _b, _c, _d, _e;
   let startedAt = Date.now();
   const traces = iteration.traces;
-  const model = await cognitive.getModelDetails(iteration.model ?? "best");
-  const modelLimit = model.input.maxTokens;
+  const model = await cognitive.getModelDetails(Array.isArray(iteration.model) ? iteration.model[0] : iteration.model);
+  const modelLimit = Math.max(model.input.maxTokens, 8e3);
   const responseLengthBuffer = getModelOutputLimit(modelLimit);
   const messages = truncateWrappedContent({
     messages: iteration.messages,
@@ -238,7 +239,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 Error("No output from LLM");
+    throw new CognitiveError("LLM did not return any text output");
   }
   const assistantResponse = ctx.version.parseAssistantResponse(out);
   iteration.code = assistantResponse.code.trim();

package/dist/llmz.d.ts

@@ -1,6 +1,6 @@
-import { Cognitive, type BotpressClientLike } from '@botpress/cognitive';
+import { Cognitive, Models, type BotpressClientLike } from '@botpress/cognitive';
 import { Chat } from './chat.js';
-import { Context, Iteration, Model } from './context.js';
+import { Context, Iteration } from './context.js';
 import { Exit, ExitResult } from './exit.js';
 import { ValueOrGetter } from './getter.js';
 import { type ObjectInstance } from './objects.js';
@@ -175,7 +175,7 @@ export type ExecutionProps = {
      * The model to use for the LLM.
      * This can be a static model name or a function that returns a model name based on the current context.
      */
-    model?: ValueOrGetter<Model, Context>;
+    model?: ValueOrGetter<Models | Models[], Context>;
     /**
      * The temperature to use for the LLM.
      * This can be a static temperature or a function that returns a temperature based on the current context.

package/dist/result.d.ts

@@ -388,7 +388,7 @@ export declare class SuccessExecutionResult<TOutput = unknown> extends Execution
                 id: string;
                 messages: import("./index.js").LLMzPrompts.Message[];
                 code: string | undefined;
-                model: import("./context.js").Model;
+                model: import("@botpress/cognitive").Models | import("@botpress/cognitive").Models[];
                 temperature: number;
                 traces: import("./types.js").Traces.Trace[];
                 variables: Record<string, any>;
@@ -543,7 +543,7 @@ export declare class ErrorExecutionResult extends ExecutionResult implements Ser
                 id: string;
                 messages: import("./index.js").LLMzPrompts.Message[];
                 code: string | undefined;
-                model: import("./context.js").Model;
+                model: import("@botpress/cognitive").Models | import("@botpress/cognitive").Models[];
                 temperature: number;
                 traces: import("./types.js").Traces.Trace[];
                 variables: Record<string, any>;
@@ -718,7 +718,7 @@ export declare class PartialExecutionResult extends ExecutionResult implements S
                 id: string;
                 messages: import("./index.js").LLMzPrompts.Message[];
                 code: string | undefined;
-                model: import("./context.js").Model;
+                model: import("@botpress/cognitive").Models | import("@botpress/cognitive").Models[];
                 temperature: number;
                 traces: import("./types.js").Traces.Trace[];
                 variables: Record<string, any>;

package/dist/tool-GEBXW6AQ.js

@@ -0,0 +1,11 @@
+import {
+  Tool
+} from "./chunk-7POUFE5M.js";
+import "./chunk-IUE5BW56.js";
+import "./chunk-MYLTD5WT.js";
+import "./chunk-YEAWWJSJ.js";
+import "./chunk-ZORRILUV.js";
+import "./chunk-7WRN4E42.js";
+export {
+  Tool
+};

package/dist/tool-GMYMVXUK.cjs

@@ -0,0 +1,11 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunk273DEMEUcjs = require('./chunk-273DEMEU.cjs');
+require('./chunk-RLOPQZTQ.cjs');
+require('./chunk-T63Y6GTW.cjs');
+require('./chunk-KIN7Y247.cjs');
+require('./chunk-WHNOR4ZU.cjs');
+require('./chunk-UQOBUJIQ.cjs');
+
+
+exports.Tool = _chunk273DEMEUcjs.Tool;

package/dist/typings-2RAAZ2YP.cjs

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

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

@@ -1,8 +1,8 @@
 import {
   getTypings
-} from "./chunk-EE6NVDID.js";
-import "./chunk-JKVVQN2P.js";
-import "./chunk-JQBT7UWN.js";
+} from "./chunk-IUE5BW56.js";
+import "./chunk-MYLTD5WT.js";
+import "./chunk-YEAWWJSJ.js";
 import "./chunk-ZORRILUV.js";
 import "./chunk-7WRN4E42.js";
 export {

package/dist/{vm-CYC5RLKU.js → vm-I4G4IH7T.js}

@@ -1,9 +1,9 @@
 import {
   CAN_USE_ISOLATED_VM,
   runAsyncFunction
-} from "./chunk-LUX4X7XO.js";
-import "./chunk-JKVVQN2P.js";
-import "./chunk-JQBT7UWN.js";
+} from "./chunk-7G5QAIUX.js";
+import "./chunk-MYLTD5WT.js";
+import "./chunk-YEAWWJSJ.js";
 import "./chunk-ORQP26SZ.js";
 import "./chunk-7WRN4E42.js";
 export {

package/dist/vm-YJX6M4NJ.cjs

@@ -0,0 +1,12 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+
+var _chunkM3LCKL55cjs = require('./chunk-M3LCKL55.cjs');
+require('./chunk-T63Y6GTW.cjs');
+require('./chunk-KIN7Y247.cjs');
+require('./chunk-KMZDFWYZ.cjs');
+require('./chunk-UQOBUJIQ.cjs');
+
+
+
+exports.CAN_USE_ISOLATED_VM = _chunkM3LCKL55cjs.CAN_USE_ISOLATED_VM; exports.runAsyncFunction = _chunkM3LCKL55cjs.runAsyncFunction;