llmz 0.0.34 → 0.0.36

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

@@ -5,7 +5,7 @@ import {
 } from "./chunk-DOIKBDLR.js";
 import {
   Tool
-} from "./chunk-GXLMCQYF.js";
+} from "./chunk-PUFOVD74.js";
 import {
   LoopExceededError
 } from "./chunk-WSVDMGMR.js";

package/dist/{chunk-GXLMCQYF.js → chunk-PUFOVD74.js}

@@ -415,8 +415,10 @@ var Tool = class _Tool {
    *
    * @internal This method is primarily used internally by the LLMz execution engine
    */
-  async execute(input, ctx) {
+  async execute(rawInput, ctx) {
     var _a;
+    const isZodObject = this.zInput._def.typeName === "ZodObject";
+    const input = isZodObject ? rawInput ?? {} : rawInput;
     const pInput = this.zInput.safeParse(input);
     if (!pInput.success) {
       throw new Error(`Tool "${this.name}" received invalid input: ${pInput.error.message}`);

package/dist/{chunk-B742DUFY.cjs → chunk-RD3LPQOW.cjs}

@@ -5,7 +5,7 @@
 var _chunkQRFGEFROcjs = require('./chunk-QRFGEFRO.cjs');
 
 
-var _chunkQR54TZ36cjs = require('./chunk-QR54TZ36.cjs');
+var _chunkWP7SDPJ6cjs = require('./chunk-WP7SDPJ6.cjs');
 
 
 var _chunkPK72FAKDcjs = require('./chunk-PK72FAKD.cjs');
@@ -1274,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 = _chunkQR54TZ36cjs.Tool.withUniqueNames(await _asyncNullishCoalesce(await getValue(this.tools, this), async () => ( [])));
+    const tools = _chunkWP7SDPJ6cjs.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);
@@ -1306,7 +1306,7 @@ var Context = (_class2 = class {
       "boolean",
       "array"
     ];
-    const MessageTool = this.chat && components.length ? new (0, _chunkQR54TZ36cjs.Tool)({
+    const MessageTool = this.chat && components.length ? new (0, _chunkWP7SDPJ6cjs.Tool)({
       name: "Message",
       description: "Send a message to the user",
       aliases: Array.from(

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

@@ -415,8 +415,10 @@ var Tool = (_class = class _Tool {
    *
    * @internal This method is primarily used internally by the LLMz execution engine
    */
-  async execute(input, ctx) {
+  async execute(rawInput, ctx) {
     var _a;
+    const isZodObject = this.zInput._def.typeName === "ZodObject";
+    const input = isZodObject ? _nullishCoalesce(rawInput, () => ( {})) : rawInput;
     const pInput = this.zInput.safeParse(input);
     if (!pInput.success) {
       throw new Error(`Tool "${this.name}" received invalid input: ${pInput.error.message}`);

package/dist/index.cjs

@@ -9,11 +9,11 @@
 
 
 
-var _chunkB742DUFYcjs = require('./chunk-B742DUFY.cjs');
+var _chunkRD3LPQOWcjs = require('./chunk-RD3LPQOW.cjs');
 require('./chunk-QRFGEFRO.cjs');
 
 
-var _chunkQR54TZ36cjs = require('./chunk-QR54TZ36.cjs');
+var _chunkWP7SDPJ6cjs = require('./chunk-WP7SDPJ6.cjs');
 
 
 
@@ -428,7 +428,7 @@ var ObjectInstance = class {
     this.description = props.description;
     this.metadata = _nullishCoalesce(props.metadata, () => ( {}));
     this.properties = props.properties;
-    this.tools = _chunkQR54TZ36cjs.Tool.withUniqueNames(_nullishCoalesce(props.tools, () => ( [])));
+    this.tools = _chunkWP7SDPJ6cjs.Tool.withUniqueNames(_nullishCoalesce(props.tools, () => ( [])));
   }
   /**
    * Generates TypeScript namespace declarations for this object.
@@ -1112,13 +1112,13 @@ var utils = {
   truncateWrappedContent: _chunkGZPN7RGHcjs.truncateWrappedContent
 };
 var execute = async (props) => {
-  const { executeContext } = await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-YU4TWA2R.cjs")));
+  const { executeContext } = await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-I7RVUI6Q.cjs")));
   return executeContext(props);
 };
 var init = async () => {
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-YU4TWA2R.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-I7RVUI6Q.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("./tool-OC3QHJWX.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")));
@@ -1155,4 +1155,4 @@ var init = async () => {
 
 
 
-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;
+exports.Chat = Chat; exports.CitationsManager = CitationsManager; exports.Component = _chunkQY2TRWGKcjs.Component; exports.DefaultComponents = DefaultComponents; exports.DefaultExit = _chunkRD3LPQOWcjs.DefaultExit; exports.ErrorExecutionResult = _chunkRD3LPQOWcjs.ErrorExecutionResult; exports.ExecutionResult = _chunkRD3LPQOWcjs.ExecutionResult; exports.Exit = _chunk3G3BS5IAcjs.Exit; exports.ListenExit = _chunkRD3LPQOWcjs.ListenExit; exports.LoopExceededError = _chunkPK72FAKDcjs.LoopExceededError; exports.ObjectInstance = ObjectInstance; exports.PartialExecutionResult = _chunkRD3LPQOWcjs.PartialExecutionResult; exports.Snapshot = _chunkRD3LPQOWcjs.Snapshot; exports.SnapshotSignal = _chunkPK72FAKDcjs.SnapshotSignal; exports.SuccessExecutionResult = _chunkRD3LPQOWcjs.SuccessExecutionResult; exports.ThinkExit = _chunkRD3LPQOWcjs.ThinkExit; exports.ThinkSignal = _chunkPK72FAKDcjs.ThinkSignal; exports.Tool = _chunkWP7SDPJ6cjs.Tool; exports.assertValidComponent = _chunkQY2TRWGKcjs.assertValidComponent; exports.execute = execute; exports.getValue = _chunkRD3LPQOWcjs.getValue; exports.init = init; exports.isAnyComponent = _chunkQY2TRWGKcjs.isAnyComponent; exports.isComponent = _chunkQY2TRWGKcjs.isComponent; exports.parseExit = _chunkRD3LPQOWcjs.parseExit; exports.renderToTsx = _chunkQY2TRWGKcjs.renderToTsx; exports.utils = utils;

package/dist/index.js

@@ -9,11 +9,11 @@ import {
   ThinkExit,
   getValue,
   parseExit
-} from "./chunk-RANA7NBF.js";
+} from "./chunk-4KXGNHLO.js";
 import "./chunk-DOIKBDLR.js";
 import {
   Tool
-} from "./chunk-GXLMCQYF.js";
+} from "./chunk-PUFOVD74.js";
 import {
   formatTypings,
   getTypings
@@ -1112,13 +1112,13 @@ var utils = {
   truncateWrappedContent
 };
 var execute = async (props) => {
-  const { executeContext } = await import("./llmz-HAHKYXAJ.js");
+  const { executeContext } = await import("./llmz-7KVW3LXE.js");
   return executeContext(props);
 };
 var init = async () => {
-  await import("./llmz-HAHKYXAJ.js");
+  await import("./llmz-7KVW3LXE.js");
   await import("./component-EOMTLA64.js");
-  await import("./tool-ADXBCB4R.js");
+  await import("./tool-3YL3EY2R.js");
   await import("./exit-YLO7BY7Z.js");
   await import("./jsx-AEHVFB3L.js");
   await import("./vm-Y3WY2627.js");

package/dist/{llmz-HAHKYXAJ.js → llmz-7KVW3LXE.js}

@@ -8,9 +8,9 @@ import {
   Snapshot,
   SuccessExecutionResult,
   parseExit
-} from "./chunk-RANA7NBF.js";
+} from "./chunk-4KXGNHLO.js";
 import "./chunk-DOIKBDLR.js";
-import "./chunk-GXLMCQYF.js";
+import "./chunk-PUFOVD74.js";
 import "./chunk-B42FGPCX.js";
 import {
   AssignmentError,
@@ -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);
@@ -252,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");
   }
@@ -428,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-YU4TWA2R.cjs → llmz-I7RVUI6Q.cjs}

@@ -8,9 +8,9 @@ var _chunk5FWOHMO4cjs = require('./chunk-5FWOHMO4.cjs');
 
 
 
-var _chunkB742DUFYcjs = require('./chunk-B742DUFY.cjs');
+var _chunkRD3LPQOWcjs = require('./chunk-RD3LPQOW.cjs');
 require('./chunk-QRFGEFRO.cjs');
-require('./chunk-QR54TZ36.cjs');
+require('./chunk-WP7SDPJ6.cjs');
 require('./chunk-KED57HR5.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, _chunkB742DUFYcjs.Context)({
+  const ctx = new (0, _chunkRD3LPQOWcjs.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, _chunkB742DUFYcjs.ErrorExecutionResult)(ctx, new (0, _chunkPK72FAKDcjs.LoopExceededError)());
+        return new (0, _chunkRD3LPQOWcjs.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, _chunkB742DUFYcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(controller.signal.reason, () => ( "The operation was aborted")));
+        return new (0, _chunkRD3LPQOWcjs.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, _chunkB742DUFYcjs.SuccessExecutionResult)(ctx, {
+        return new (0, _chunkRD3LPQOWcjs.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, _chunkB742DUFYcjs.PartialExecutionResult)(
+        return new (0, _chunkRD3LPQOWcjs.PartialExecutionResult)(
           ctx,
           iteration.status.callback_requested.signal,
-          _chunkB742DUFYcjs.Snapshot.fromSignal(iteration.status.callback_requested.signal)
+          _chunkRD3LPQOWcjs.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, _chunkB742DUFYcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(iteration.error, () => ( `Unknown error. Status: ${iteration.status.type}`)));
+      return new (0, _chunkRD3LPQOWcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(iteration.error, () => ( `Unknown error. Status: ${iteration.status.type}`)));
     }
   } catch (error) {
-    return new (0, _chunkB742DUFYcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(error, () => ( "Unknown error")));
+    return new (0, _chunkRD3LPQOWcjs.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);
@@ -252,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");
   }
@@ -428,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")))
       }
     });
@@ -472,7 +474,7 @@ var executeIteration = async ({
       }
     });
   }
-  const parsedExit = _chunkB742DUFYcjs.parseExit.call(void 0, returnValue, iteration.exits);
+  const parsedExit = _chunkRD3LPQOWcjs.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/{tool-ADXBCB4R.js → tool-3YL3EY2R.js}

@@ -1,6 +1,6 @@
 import {
   Tool
-} from "./chunk-GXLMCQYF.js";
+} from "./chunk-PUFOVD74.js";
 import "./chunk-B42FGPCX.js";
 import "./chunk-WSVDMGMR.js";
 import "./chunk-YEAWWJSJ.js";

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

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

package/dist/tool.d.ts

@@ -464,7 +464,7 @@ export declare class Tool<I extends ZuiType = ZuiType, O extends ZuiType = ZuiTy
      *
      * @internal This method is primarily used internally by the LLMz execution engine
      */
-    execute(input: TypeOf<I>, ctx: ToolCallContext): Promise<TypeOf<O>>;
+    execute(rawInput: TypeOf<I>, ctx: ToolCallContext): Promise<TypeOf<O>>;
     /**
      * Generates TypeScript type definitions for this tool.
      *

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.34",
+  "version": "0.0.36",
   "types": "./dist/index.d.ts",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",
@@ -71,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.3"
+    "@bpinternal/zui": "1.3.1"
   },
   "dependenciesMeta": {
     "@bpinternal/zui": {