llmz 0.0.78 → 0.0.80

package/dist/{chunk-6EAURSDP.js → chunk-2KRDL5AW.js}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-YEAWWJSJ.js";
 import {
   getComponentReference
-} from "./chunk-AMQFJRRX.js";
+} from "./chunk-YTDNCBQW.js";
 import {
   wrapContent
 } from "./chunk-M4WTTARE.js";

package/dist/{chunk-3SZK4YSR.cjs → chunk-BSKF4J2E.cjs}

@@ -3,7 +3,7 @@
 var _chunkKIN7Y247cjs = require('./chunk-KIN7Y247.cjs');
 
 
-var _chunkK2B5YEGJcjs = require('./chunk-K2B5YEGJ.cjs');
+var _chunkWDYTCDIKcjs = require('./chunk-WDYTCDIK.cjs');
 
 
 var _chunkUCXQ4U7Pcjs = require('./chunk-UCXQ4U7P.cjs');
@@ -526,7 +526,7 @@ ${variables_example}
       variables_example,
       exits,
       components: _chunkUCXQ4U7Pcjs.wrapContent.call(void 0, 
-        props.components.map((component) => _chunkK2B5YEGJcjs.getComponentReference.call(void 0, component.definition)).join("\n\n"),
+        props.components.map((component) => _chunkWDYTCDIKcjs.getComponentReference.call(void 0, component.definition)).join("\n\n"),
         {
           preserve: "top",
           minTokens: 500

package/dist/{chunk-EXSETPV7.js → chunk-ESMRQVFC.js}

@@ -2,10 +2,10 @@ import {
   DualModePrompt,
   extractType,
   inspect
-} from "./chunk-6EAURSDP.js";
+} from "./chunk-2KRDL5AW.js";
 import {
   Tool
-} from "./chunk-G5INQ5LJ.js";
+} from "./chunk-UI6J4WKQ.js";
 import {
   LoopExceededError
 } from "./chunk-WSVDMGMR.js";
@@ -14,7 +14,7 @@ import {
 } from "./chunk-E5YW5YQN.js";
 import {
   assertValidComponent
-} from "./chunk-AMQFJRRX.js";
+} from "./chunk-YTDNCBQW.js";
 import {
   wrapContent
 } from "./chunk-M4WTTARE.js";

package/dist/{chunk-2JTZOTO5.cjs → chunk-QOM7BK7U.cjs}

@@ -391,6 +391,26 @@ var Tool = (_class = class _Tool {
     this.setStaticInputValues(props.staticInputValues);
     this.retry = props.retry;
   }
+  async _executeHandler(input, ctx, chat, yieldedCount, setYieldedCount) {
+    var _a;
+    const handler = this._handler(input, ctx);
+    const isGen = typeof handler === "object" && handler !== null && "next" in handler;
+    if (!isGen) {
+      return handler;
+    }
+    let yieldIndex = 0;
+    while (true) {
+      const { value, done } = await handler.next();
+      if (done) {
+        return value;
+      }
+      if (yieldIndex >= yieldedCount) {
+        setYieldedCount(yieldIndex + 1);
+        await ((_a = chat == null ? void 0 : chat.handler) == null ? void 0 : _a.call(chat, value));
+      }
+      yieldIndex++;
+    }
+  }
   /**
    * Executes the tool with the given input and context.
    *
@@ -415,7 +435,7 @@ var Tool = (_class = class _Tool {
    *
    * @internal This method is primarily used internally by the LLMz execution engine
    */
-  async execute(rawInput, ctx) {
+  async execute(rawInput, ctx, chat) {
     var _a;
     const isZodObject = this.zInput._def.typeName === "ZodObject";
     const input = isZodObject ? _nullishCoalesce(rawInput, () => ( {})) : rawInput;
@@ -424,9 +444,12 @@ var Tool = (_class = class _Tool {
       throw new Error(`Tool "${this.name}" received invalid input: ${pInput.error.message}`);
     }
     let attempt = 0;
+    let yieldedCount = 0;
     while (attempt < this.MAX_RETRIES) {
       try {
-        const result = await this._handler(pInput.data, ctx);
+        const result = await this._executeHandler(pInput.data, ctx, chat, yieldedCount, (n) => {
+          yieldedCount = n;
+        });
         const pOutput = this.zOutput.safeParse(result);
         return pOutput.success ? pOutput.data : result;
       } catch (err) {

package/dist/{chunk-G5INQ5LJ.js → chunk-UI6J4WKQ.js}

@@ -391,6 +391,26 @@ var Tool = class _Tool {
     this.setStaticInputValues(props.staticInputValues);
     this.retry = props.retry;
   }
+  async _executeHandler(input, ctx, chat, yieldedCount, setYieldedCount) {
+    var _a;
+    const handler = this._handler(input, ctx);
+    const isGen = typeof handler === "object" && handler !== null && "next" in handler;
+    if (!isGen) {
+      return handler;
+    }
+    let yieldIndex = 0;
+    while (true) {
+      const { value, done } = await handler.next();
+      if (done) {
+        return value;
+      }
+      if (yieldIndex >= yieldedCount) {
+        setYieldedCount(yieldIndex + 1);
+        await ((_a = chat == null ? void 0 : chat.handler) == null ? void 0 : _a.call(chat, value));
+      }
+      yieldIndex++;
+    }
+  }
   /**
    * Executes the tool with the given input and context.
    *
@@ -415,7 +435,7 @@ var Tool = class _Tool {
    *
    * @internal This method is primarily used internally by the LLMz execution engine
    */
-  async execute(rawInput, ctx) {
+  async execute(rawInput, ctx, chat) {
     var _a;
     const isZodObject = this.zInput._def.typeName === "ZodObject";
     const input = isZodObject ? rawInput ?? {} : rawInput;
@@ -424,9 +444,12 @@ var Tool = class _Tool {
       throw new Error(`Tool "${this.name}" received invalid input: ${pInput.error.message}`);
     }
     let attempt = 0;
+    let yieldedCount = 0;
     while (attempt < this.MAX_RETRIES) {
       try {
-        const result = await this._handler(pInput.data, ctx);
+        const result = await this._executeHandler(pInput.data, ctx, chat, yieldedCount, (n) => {
+          yieldedCount = n;
+        });
         const pOutput = this.zOutput.safeParse(result);
         return pOutput.success ? pOutput.data : result;
       } catch (err) {

package/dist/{chunk-K2B5YEGJ.cjs → chunk-WDYTCDIK.cjs}

@@ -1,6 +1,7 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
 
+
 var _chunk54G63644cjs = require('./chunk-54G63644.cjs');
 
 // src/component.ts
@@ -134,6 +135,9 @@ var Component = class {
     assertValidComponent(definition);
     this.definition = definition;
   }
+  render(props, children = []) {
+    return _chunk54G63644cjs.createJsxComponent.call(void 0, { type: this.definition.name, props, children });
+  }
 };
 function isComponent(rendered, component) {
   return _chunk54G63644cjs.isJsxComponent.call(void 0, component.definition.name, rendered);

package/dist/{chunk-BFZJ4UV2.cjs → chunk-XLUYPCCY.cjs}

@@ -2,10 +2,10 @@
 
 
 
-var _chunk3SZK4YSRcjs = require('./chunk-3SZK4YSR.cjs');
+var _chunkBSKF4J2Ecjs = require('./chunk-BSKF4J2E.cjs');
 
 
-var _chunk2JTZOTO5cjs = require('./chunk-2JTZOTO5.cjs');
+var _chunkQOM7BK7Ucjs = require('./chunk-QOM7BK7U.cjs');
 
 
 var _chunkPK72FAKDcjs = require('./chunk-PK72FAKD.cjs');
@@ -14,7 +14,7 @@ var _chunkPK72FAKDcjs = require('./chunk-PK72FAKD.cjs');
 var _chunkF4OZHUMScjs = require('./chunk-F4OZHUMS.cjs');
 
 
-var _chunkK2B5YEGJcjs = require('./chunk-K2B5YEGJ.cjs');
+var _chunkWDYTCDIKcjs = require('./chunk-WDYTCDIK.cjs');
 
 
 var _chunkUCXQ4U7Pcjs = require('./chunk-UCXQ4U7P.cjs');
@@ -493,10 +493,10 @@ var Snapshot = class _Snapshot {
 };
 function parseVariables(variableMap) {
   return Object.entries(variableMap).map(([name, value]) => {
-    const type = _chunk3SZK4YSRcjs.extractType.call(void 0, value);
+    const type = _chunkBSKF4J2Ecjs.extractType.call(void 0, value);
     const bytes = JSON.stringify(value || "").length;
     const truncated = bytes > MAX_SNAPSHOT_SIZE_BYTES;
-    return truncated ? { name, type, bytes, truncated: true, preview: _nullishCoalesce(_chunk3SZK4YSRcjs.inspect.call(void 0, value, name), () => ( "N/A")) } : { name, type, bytes, truncated: false, value };
+    return truncated ? { name, type, bytes, truncated: true, preview: _nullishCoalesce(_chunkBSKF4J2Ecjs.inspect.call(void 0, value, name), () => ( "N/A")) } : { name, type, bytes, truncated: false, value };
   });
 }
 
@@ -507,9 +507,9 @@ function getMessagePreview(message) {
     return message.content;
   }
   if (message.role === "event") {
-    return _chunk3SZK4YSRcjs.inspect.call(void 0, message.payload, message.name, { tokens: 1e3 });
+    return _chunkBSKF4J2Ecjs.inspect.call(void 0, message.payload, message.name, { tokens: 1e3 });
   }
-  return _chunk3SZK4YSRcjs.inspect.call(void 0, message, void 0, { tokens: 1e3 });
+  return _chunkBSKF4J2Ecjs.inspect.call(void 0, message, void 0, { tokens: 1e3 });
 }
 function getMessageType(message) {
   if (message.role === "assistant" || message.role === "user" || message.role === "summary") {
@@ -1103,7 +1103,7 @@ var Context = (_class2 = class {
   
   
   
-  __init3() {this.version = _chunk3SZK4YSRcjs.DualModePrompt}
+  __init3() {this.version = _chunkBSKF4J2Ecjs.DualModePrompt}
   __init4() {this.timeout = 6e4}
   // Default timeout of 60 seconds
   
@@ -1279,7 +1279,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 = _chunk2JTZOTO5cjs.Tool.withUniqueNames(await _asyncNullishCoalesce(await getValue(this.tools, this), async () => ( [])));
+    const tools = _chunkQOM7BK7Ucjs.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);
@@ -1293,7 +1293,7 @@ var Context = (_class2 = class {
       throw new Error("Too many tools. Expected at most 100 tools.");
     }
     for (const component of components) {
-      _chunkK2B5YEGJcjs.assertValidComponent.call(void 0, component.definition);
+      _chunkWDYTCDIKcjs.assertValidComponent.call(void 0, component.definition);
     }
     const ReservedToolNames = [
       "think",
@@ -1312,7 +1312,7 @@ var Context = (_class2 = class {
       "boolean",
       "array"
     ];
-    const MessageTool = this.chat && components.length ? new (0, _chunk2JTZOTO5cjs.Tool)({
+    const MessageTool = this.chat && components.length ? new (0, _chunkQOM7BK7Ucjs.Tool)({
       name: "Message",
       description: "Send a message to the user",
       aliases: Array.from(

package/dist/{chunk-AMQFJRRX.js → chunk-YTDNCBQW.js}

@@ -1,4 +1,5 @@
 import {
+  createJsxComponent,
   isAnyJsxComponent,
   isJsxComponent
 } from "./chunk-UOPTKXLE.js";
@@ -134,6 +135,9 @@ var Component = class {
     assertValidComponent(definition);
     this.definition = definition;
   }
+  render(props, children = []) {
+    return createJsxComponent({ type: this.definition.name, props, children });
+  }
 };
 function isComponent(rendered, component) {
   return isJsxComponent(component.definition.name, rendered);

package/dist/{component-EZAROCAF.js → component-DWFDDH5N.js}

@@ -5,7 +5,7 @@ import {
   isAnyComponent,
   isComponent,
   renderToTsx
-} from "./chunk-AMQFJRRX.js";
+} from "./chunk-YTDNCBQW.js";
 import "./chunk-UOPTKXLE.js";
 import "./chunk-5XC6ZMS5.js";
 export {

package/dist/component-RVYDIBUM.cjs

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

package/dist/component.d.ts

@@ -1,4 +1,5 @@
 import { z } from '@bpinternal/zui';
+import { JsxComponent } from './jsx.js';
 export type ExampleUsage = {
     name: string;
     description: string;
@@ -44,16 +45,12 @@ export type ContainerComponentDefinition<T extends z.ZodObject<any> = z.ZodObjec
 };
 export type ComponentDefinition = DefaultComponentDefinition | LeafComponentDefinition | ContainerComponentDefinition;
 type ExtractComponentProps<T extends ComponentDefinition> = T extends LeafComponentDefinition<infer P> ? z.infer<P> : T extends ContainerComponentDefinition<infer P> ? z.infer<P> : T extends DefaultComponentDefinition<infer P> ? z.infer<P> : never;
-export type RenderedComponent<TProps = Record<string, any>> = {
-    __jsx: true;
-    type: string;
-    children: any[];
-    props: TProps;
-};
+export type RenderedComponent<TProps extends {} = {}> = JsxComponent<string, TProps>;
 export declare class Component<T extends ComponentDefinition = ComponentDefinition> {
     readonly definition: T;
     readonly propsType: ExtractComponentProps<T>;
     constructor(definition: T);
+    render<TChildren extends any = any>(props: Component['propsType'], children?: Array<TChildren>): RenderedComponent;
 }
 export declare function isComponent<T extends ComponentDefinition>(rendered: RenderedComponent<any>, component: Component<T>): rendered is RenderedComponent<ExtractComponentProps<T>>;
 export declare function isAnyComponent(message: unknown): message is RenderedComponent;

package/dist/{dual-modes-EFWC4CAY.cjs → dual-modes-A4SC4APN.cjs}

@@ -1,12 +1,12 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunk3SZK4YSRcjs = require('./chunk-3SZK4YSR.cjs');
+var _chunkBSKF4J2Ecjs = require('./chunk-BSKF4J2E.cjs');
 require('./chunk-KIN7Y247.cjs');
-require('./chunk-K2B5YEGJ.cjs');
+require('./chunk-WDYTCDIK.cjs');
 require('./chunk-54G63644.cjs');
 require('./chunk-UCXQ4U7P.cjs');
 require('./chunk-4VYDQ3VF.cjs');
 require('./chunk-RKKRSBX3.cjs');
 
 
-exports.DualModePrompt = _chunk3SZK4YSRcjs.DualModePrompt;
+exports.DualModePrompt = _chunkBSKF4J2Ecjs.DualModePrompt;

package/dist/{dual-modes-ILL7OVJG.js → dual-modes-FNC3BGK3.js}

@@ -1,8 +1,8 @@
 import {
   DualModePrompt
-} from "./chunk-6EAURSDP.js";
+} from "./chunk-2KRDL5AW.js";
 import "./chunk-YEAWWJSJ.js";
-import "./chunk-AMQFJRRX.js";
+import "./chunk-YTDNCBQW.js";
 import "./chunk-UOPTKXLE.js";
 import "./chunk-M4WTTARE.js";
 import "./chunk-EKXZXQJB.js";

package/dist/index.cjs

@@ -9,11 +9,11 @@
 
 
 
-var _chunkBFZJ4UV2cjs = require('./chunk-BFZJ4UV2.cjs');
-require('./chunk-3SZK4YSR.cjs');
+var _chunkXLUYPCCYcjs = require('./chunk-XLUYPCCY.cjs');
+require('./chunk-BSKF4J2E.cjs');
 
 
-var _chunk2JTZOTO5cjs = require('./chunk-2JTZOTO5.cjs');
+var _chunkQOM7BK7Ucjs = require('./chunk-QOM7BK7U.cjs');
 
 
 
@@ -34,7 +34,7 @@ var _chunkF4OZHUMScjs = require('./chunk-F4OZHUMS.cjs');
 
 
 
-var _chunkK2B5YEGJcjs = require('./chunk-K2B5YEGJ.cjs');
+var _chunkWDYTCDIKcjs = require('./chunk-WDYTCDIK.cjs');
 require('./chunk-54G63644.cjs');
 
 
@@ -53,7 +53,7 @@ var _chunk4VYDQ3VFcjs = require('./chunk-4VYDQ3VF.cjs');
 var _chunkRKKRSBX3cjs = require('./chunk-RKKRSBX3.cjs');
 
 // package.json
-var version = "0.0.78";
+var version = "0.0.80";
 
 // src/objects.ts
 var _zui = require('@bpinternal/zui');
@@ -431,7 +431,7 @@ var ObjectInstance = class {
     this.description = props.description;
     this.metadata = _nullishCoalesce(props.metadata, () => ( {}));
     this.properties = props.properties;
-    this.tools = _chunk2JTZOTO5cjs.Tool.withUniqueNames(_nullishCoalesce(props.tools, () => ( [])));
+    this.tools = _chunkQOM7BK7Ucjs.Tool.withUniqueNames(_nullishCoalesce(props.tools, () => ( [])));
   }
   /**
    * Generates TypeScript namespace declarations for this object.
@@ -769,7 +769,7 @@ var CitationsManager = (_class = class {constructor() { _class.prototype.__init.
 
 // src/component.default.ts
 
-var Button = new (0, _chunkK2B5YEGJcjs.Component)({
+var Button = new (0, _chunkWDYTCDIKcjs.Component)({
   type: "leaf",
   description: "A button component that can perform actions when clicked",
   name: "Button",
@@ -801,7 +801,7 @@ var Button = new (0, _chunkK2B5YEGJcjs.Component)({
     })
   }
 });
-var Image = new (0, _chunkK2B5YEGJcjs.Component)({
+var Image = new (0, _chunkWDYTCDIKcjs.Component)({
   type: "leaf",
   name: "Image",
   description: "Displays an image from a URL.",
@@ -823,7 +823,7 @@ var Image = new (0, _chunkK2B5YEGJcjs.Component)({
     })
   }
 });
-var File = new (0, _chunkK2B5YEGJcjs.Component)({
+var File = new (0, _chunkWDYTCDIKcjs.Component)({
   type: "leaf",
   name: "File",
   description: "Sends a downloadable file to the user.",
@@ -845,7 +845,7 @@ var File = new (0, _chunkK2B5YEGJcjs.Component)({
     })
   }
 });
-var Video = new (0, _chunkK2B5YEGJcjs.Component)({
+var Video = new (0, _chunkWDYTCDIKcjs.Component)({
   type: "leaf",
   name: "Video",
   description: "Embeds a video from a URL.",
@@ -867,7 +867,7 @@ var Video = new (0, _chunkK2B5YEGJcjs.Component)({
     })
   }
 });
-var Audio = new (0, _chunkK2B5YEGJcjs.Component)({
+var Audio = new (0, _chunkWDYTCDIKcjs.Component)({
   type: "leaf",
   name: "Audio",
   description: "Plays an audio clip from a URL.",
@@ -889,7 +889,7 @@ var Audio = new (0, _chunkK2B5YEGJcjs.Component)({
     })
   }
 });
-var Card = new (0, _chunkK2B5YEGJcjs.Component)({
+var Card = new (0, _chunkWDYTCDIKcjs.Component)({
   type: "container",
   name: "Card",
   description: "A visual card component that can include an image and buttons.",
@@ -925,7 +925,7 @@ var Card = new (0, _chunkK2B5YEGJcjs.Component)({
     ]
   }
 });
-var Carousel = new (0, _chunkK2B5YEGJcjs.Component)({
+var Carousel = new (0, _chunkWDYTCDIKcjs.Component)({
   type: "container",
   name: "Carousel",
   description: "A virtual container for displaying 1 to 10 Card components as a carousel.",
@@ -959,7 +959,7 @@ var Carousel = new (0, _chunkK2B5YEGJcjs.Component)({
     ]
   }
 });
-var Text = new (0, _chunkK2B5YEGJcjs.Component)({
+var Text = new (0, _chunkWDYTCDIKcjs.Component)({
   type: "default",
   name: "Message",
   aliases: ["Text", "Markdown"],
@@ -1115,20 +1115,20 @@ var utils = {
   truncateWrappedContent: _chunkUCXQ4U7Pcjs.truncateWrappedContent
 };
 var execute = async (props) => {
-  const { executeContext } = await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-YPI4WWBW.cjs")));
+  const { executeContext } = await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-XRURR7Z5.cjs")));
   return executeContext(props);
 };
 var init = async () => {
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-YPI4WWBW.cjs")));
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./component-36KG7B5H.cjs")));
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./tool-NCCTC74P.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-XRURR7Z5.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./component-RVYDIBUM.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./tool-A7EOCUQG.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./exit-OOGJ3TAS.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./jsx-ENFGHP6S.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./vm-CR5NE6NS.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./utils-BZUEJT4W.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./truncator-6M2EP2NZ.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./typings-54UIN7ZR.cjs")));
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./dual-modes-EFWC4CAY.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./dual-modes-A4SC4APN.cjs")));
 };
 
 
@@ -1159,4 +1159,4 @@ var init = async () => {
 
 
 
-exports.Chat = Chat; exports.CitationsManager = CitationsManager; exports.Component = _chunkK2B5YEGJcjs.Component; exports.DefaultComponents = DefaultComponents; exports.DefaultExit = _chunkBFZJ4UV2cjs.DefaultExit; exports.ErrorExecutionResult = _chunkBFZJ4UV2cjs.ErrorExecutionResult; exports.ExecutionResult = _chunkBFZJ4UV2cjs.ExecutionResult; exports.Exit = _chunkF4OZHUMScjs.Exit; exports.ListenExit = _chunkBFZJ4UV2cjs.ListenExit; exports.LoopExceededError = _chunkPK72FAKDcjs.LoopExceededError; exports.ObjectInstance = ObjectInstance; exports.PartialExecutionResult = _chunkBFZJ4UV2cjs.PartialExecutionResult; exports.Snapshot = _chunkBFZJ4UV2cjs.Snapshot; exports.SnapshotSignal = _chunkPK72FAKDcjs.SnapshotSignal; exports.SuccessExecutionResult = _chunkBFZJ4UV2cjs.SuccessExecutionResult; exports.ThinkExit = _chunkBFZJ4UV2cjs.ThinkExit; exports.ThinkSignal = _chunkPK72FAKDcjs.ThinkSignal; exports.Tool = _chunk2JTZOTO5cjs.Tool; exports.assertValidComponent = _chunkK2B5YEGJcjs.assertValidComponent; exports.execute = execute; exports.getValue = _chunkBFZJ4UV2cjs.getValue; exports.init = init; exports.isAnyComponent = _chunkK2B5YEGJcjs.isAnyComponent; exports.isComponent = _chunkK2B5YEGJcjs.isComponent; exports.parseExit = _chunkBFZJ4UV2cjs.parseExit; exports.renderToTsx = _chunkK2B5YEGJcjs.renderToTsx; exports.utils = utils; exports.version = version;
+exports.Chat = Chat; exports.CitationsManager = CitationsManager; exports.Component = _chunkWDYTCDIKcjs.Component; exports.DefaultComponents = DefaultComponents; exports.DefaultExit = _chunkXLUYPCCYcjs.DefaultExit; exports.ErrorExecutionResult = _chunkXLUYPCCYcjs.ErrorExecutionResult; exports.ExecutionResult = _chunkXLUYPCCYcjs.ExecutionResult; exports.Exit = _chunkF4OZHUMScjs.Exit; exports.ListenExit = _chunkXLUYPCCYcjs.ListenExit; exports.LoopExceededError = _chunkPK72FAKDcjs.LoopExceededError; exports.ObjectInstance = ObjectInstance; exports.PartialExecutionResult = _chunkXLUYPCCYcjs.PartialExecutionResult; exports.Snapshot = _chunkXLUYPCCYcjs.Snapshot; exports.SnapshotSignal = _chunkPK72FAKDcjs.SnapshotSignal; exports.SuccessExecutionResult = _chunkXLUYPCCYcjs.SuccessExecutionResult; exports.ThinkExit = _chunkXLUYPCCYcjs.ThinkExit; exports.ThinkSignal = _chunkPK72FAKDcjs.ThinkSignal; exports.Tool = _chunkQOM7BK7Ucjs.Tool; exports.assertValidComponent = _chunkWDYTCDIKcjs.assertValidComponent; exports.execute = execute; exports.getValue = _chunkXLUYPCCYcjs.getValue; exports.init = init; exports.isAnyComponent = _chunkWDYTCDIKcjs.isAnyComponent; exports.isComponent = _chunkWDYTCDIKcjs.isComponent; exports.parseExit = _chunkXLUYPCCYcjs.parseExit; exports.renderToTsx = _chunkWDYTCDIKcjs.renderToTsx; exports.utils = utils; exports.version = version;

package/dist/index.js

@@ -9,11 +9,11 @@ import {
   ThinkExit,
   getValue,
   parseExit
-} from "./chunk-EXSETPV7.js";
-import "./chunk-6EAURSDP.js";
+} from "./chunk-ESMRQVFC.js";
+import "./chunk-2KRDL5AW.js";
 import {
   Tool
-} from "./chunk-G5INQ5LJ.js";
+} from "./chunk-UI6J4WKQ.js";
 import {
   formatTypings,
   getTypings
@@ -34,7 +34,7 @@ import {
   isAnyComponent,
   isComponent,
   renderToTsx
-} from "./chunk-AMQFJRRX.js";
+} from "./chunk-YTDNCBQW.js";
 import "./chunk-UOPTKXLE.js";
 import {
   truncateWrappedContent,
@@ -53,7 +53,7 @@ import {
 } from "./chunk-5XC6ZMS5.js";
 
 // package.json
-var version = "0.0.78";
+var version = "0.0.80";
 
 // src/objects.ts
 import { z } from "@bpinternal/zui";
@@ -1115,20 +1115,20 @@ var utils = {
   truncateWrappedContent
 };
 var execute = async (props) => {
-  const { executeContext } = await import("./llmz-D4MQK65F.js");
+  const { executeContext } = await import("./llmz-F227HVGB.js");
   return executeContext(props);
 };
 var init = async () => {
-  await import("./llmz-D4MQK65F.js");
-  await import("./component-EZAROCAF.js");
-  await import("./tool-ZP6BC3P2.js");
+  await import("./llmz-F227HVGB.js");
+  await import("./component-DWFDDH5N.js");
+  await import("./tool-FNWHZGDU.js");
   await import("./exit-56DISPLS.js");
   await import("./jsx-PCXQVMLU.js");
   await import("./vm-GDB4GW4S.js");
   await import("./utils-D3J443FA.js");
   await import("./truncator-2FSONJM2.js");
   await import("./typings-DVMU3KVB.js");
-  await import("./dual-modes-ILL7OVJG.js");
+  await import("./dual-modes-FNC3BGK3.js");
 };
 export {
   Chat,

package/dist/{llmz-D4MQK65F.js → llmz-F227HVGB.js}

@@ -8,9 +8,9 @@ import {
   Snapshot,
   SuccessExecutionResult,
   parseExit
-} from "./chunk-EXSETPV7.js";
-import "./chunk-6EAURSDP.js";
-import "./chunk-G5INQ5LJ.js";
+} from "./chunk-ESMRQVFC.js";
+import "./chunk-2KRDL5AW.js";
+import "./chunk-UI6J4WKQ.js";
 import "./chunk-LR6KTFHO.js";
 import {
   AssignmentError,
@@ -27,7 +27,7 @@ import {
   cleanStackTrace
 } from "./chunk-YEAWWJSJ.js";
 import "./chunk-E5YW5YQN.js";
-import "./chunk-AMQFJRRX.js";
+import "./chunk-YTDNCBQW.js";
 import "./chunk-UOPTKXLE.js";
 import {
   truncateWrappedContent
@@ -359,6 +359,7 @@ var executeIteration = async ({
     }
     for (const tool of obj.tools ?? []) {
       instance[tool.name] = wrapTool({
+        chat: ctx.chat,
         tool,
         traces,
         object: obj.name,
@@ -373,7 +374,15 @@ var executeIteration = async ({
     vmContext[obj.name] = instance;
   }
   for (const tool of iteration.tools) {
-    const wrapped = wrapTool({ tool, traces, iteration, beforeHook: onBeforeTool, afterHook: onAfterTool, controller });
+    const wrapped = wrapTool({
+      chat: ctx.chat,
+      tool,
+      traces,
+      iteration,
+      beforeHook: onBeforeTool,
+      afterHook: onAfterTool,
+      controller
+    });
     for (const key of [tool.name, ...tool.aliases ?? []]) {
       vmContext[key] = wrapped;
     }
@@ -522,7 +531,7 @@ var executeIteration = async ({
     }
   });
 };
-function wrapTool({ tool, traces, object, iteration, beforeHook, afterHook, controller }) {
+function wrapTool({ chat, tool, traces, object, iteration, beforeHook, afterHook, controller }) {
   const getToolInput = (input) => tool.zInput.safeParse(input).data ?? input;
   return function(input) {
     const toolCallId = `tcall_${ulid()}`;
@@ -582,9 +591,13 @@ function wrapTool({ tool, traces, object, iteration, beforeHook, afterHook, cont
         if (typeof (beforeRes == null ? void 0 : beforeRes.input) !== "undefined") {
           input2 = beforeRes.input;
         }
-        let output2 = await tool.execute(input2, {
-          callId: toolCallId
-        });
+        let output2 = await tool.execute(
+          input2,
+          {
+            callId: toolCallId
+          },
+          chat
+        );
         const afterRes = await (afterHook == null ? void 0 : afterHook({
           iteration,
           tool,

package/dist/{llmz-YPI4WWBW.cjs → llmz-XRURR7Z5.cjs}

@@ -8,9 +8,9 @@ var _chunk5CXOW4DCcjs = require('./chunk-5CXOW4DC.cjs');
 
 
 
-var _chunkBFZJ4UV2cjs = require('./chunk-BFZJ4UV2.cjs');
-require('./chunk-3SZK4YSR.cjs');
-require('./chunk-2JTZOTO5.cjs');
+var _chunkXLUYPCCYcjs = require('./chunk-XLUYPCCY.cjs');
+require('./chunk-BSKF4J2E.cjs');
+require('./chunk-QOM7BK7U.cjs');
 require('./chunk-BLB44ZDU.cjs');
 
 
@@ -27,7 +27,7 @@ var _chunkPK72FAKDcjs = require('./chunk-PK72FAKD.cjs');
 
 var _chunkKIN7Y247cjs = require('./chunk-KIN7Y247.cjs');
 require('./chunk-F4OZHUMS.cjs');
-require('./chunk-K2B5YEGJ.cjs');
+require('./chunk-WDYTCDIK.cjs');
 require('./chunk-54G63644.cjs');
 
 
@@ -109,7 +109,7 @@ var _executeContext = async (props) => {
   const client = _nullishCoalesce(props.client, () => ( new (0, _client.Client)()));
   const cognitive = _cognitive.Cognitive.isCognitiveClient(client) ? client : new (0, _cognitive.Cognitive)({ client, __experimental_beta: true });
   const cleanups = [];
-  const ctx = new (0, _chunkBFZJ4UV2cjs.Context)({
+  const ctx = new (0, _chunkXLUYPCCYcjs.Context)({
     chat: props.chat,
     instructions: props.instructions,
     objects: props.objects,
@@ -125,7 +125,7 @@ var _executeContext = async (props) => {
   try {
     while (true) {
       if (ctx.iterations.length >= ctx.loop) {
-        return new (0, _chunkBFZJ4UV2cjs.ErrorExecutionResult)(ctx, new (0, _chunkPK72FAKDcjs.LoopExceededError)());
+        return new (0, _chunkXLUYPCCYcjs.ErrorExecutionResult)(ctx, new (0, _chunkPK72FAKDcjs.LoopExceededError)());
       }
       const iteration = await ctx.nextIteration();
       try {
@@ -148,7 +148,7 @@ var _executeContext = async (props) => {
             reason: _nullishCoalesce(controller.signal.reason, () => ( "The operation was aborted"))
           }
         });
-        return new (0, _chunkBFZJ4UV2cjs.ErrorExecutionResult)(ctx, _nullishCoalesce(controller.signal.reason, () => ( "The operation was aborted")));
+        return new (0, _chunkXLUYPCCYcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(controller.signal.reason, () => ( "The operation was aborted")));
       }
       cleanups.push(
         iteration.traces.onPush((traces) => {
@@ -170,7 +170,7 @@ var _executeContext = async (props) => {
         });
       } catch (err) {
         if (err instanceof _chunkPK72FAKDcjs.CognitiveError) {
-          return new (0, _chunkBFZJ4UV2cjs.ErrorExecutionResult)(ctx, err);
+          return new (0, _chunkXLUYPCCYcjs.ErrorExecutionResult)(ctx, err);
         }
         iteration.end({
           type: "execution_error",
@@ -187,25 +187,25 @@ var _executeContext = async (props) => {
       }
       if (iteration.status.type === "exit_success") {
         const exitName = iteration.status.exit_success.exit_name;
-        return new (0, _chunkBFZJ4UV2cjs.SuccessExecutionResult)(ctx, {
+        return new (0, _chunkXLUYPCCYcjs.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, _chunkBFZJ4UV2cjs.PartialExecutionResult)(
+        return new (0, _chunkXLUYPCCYcjs.PartialExecutionResult)(
           ctx,
           iteration.status.callback_requested.signal,
-          _chunkBFZJ4UV2cjs.Snapshot.fromSignal(iteration.status.callback_requested.signal)
+          _chunkXLUYPCCYcjs.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, _chunkBFZJ4UV2cjs.ErrorExecutionResult)(ctx, _nullishCoalesce(iteration.error, () => ( `Unknown error. Status: ${iteration.status.type}`)));
+      return new (0, _chunkXLUYPCCYcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(iteration.error, () => ( `Unknown error. Status: ${iteration.status.type}`)));
     }
   } catch (error) {
-    return new (0, _chunkBFZJ4UV2cjs.ErrorExecutionResult)(ctx, _nullishCoalesce(error, () => ( "Unknown error")));
+    return new (0, _chunkXLUYPCCYcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(error, () => ( "Unknown error")));
   } finally {
     for (const cleanup of cleanups) {
       try {
@@ -359,6 +359,7 @@ var executeIteration = async ({
     }
     for (const tool of _nullishCoalesce(obj.tools, () => ( []))) {
       instance[tool.name] = wrapTool({
+        chat: ctx.chat,
         tool,
         traces,
         object: obj.name,
@@ -373,7 +374,15 @@ var executeIteration = async ({
     vmContext[obj.name] = instance;
   }
   for (const tool of iteration.tools) {
-    const wrapped = wrapTool({ tool, traces, iteration, beforeHook: onBeforeTool, afterHook: onAfterTool, controller });
+    const wrapped = wrapTool({
+      chat: ctx.chat,
+      tool,
+      traces,
+      iteration,
+      beforeHook: onBeforeTool,
+      afterHook: onAfterTool,
+      controller
+    });
     for (const key of [tool.name, ..._nullishCoalesce(tool.aliases, () => ( []))]) {
       vmContext[key] = wrapped;
     }
@@ -486,7 +495,7 @@ var executeIteration = async ({
       }
     });
   }
-  const parsedExit = _chunkBFZJ4UV2cjs.parseExit.call(void 0, returnValue, iteration.exits);
+  const parsedExit = _chunkXLUYPCCYcjs.parseExit.call(void 0, returnValue, iteration.exits);
   if (!parsedExit.success) {
     return iteration.end({
       type: "exit_error",
@@ -522,7 +531,7 @@ var executeIteration = async ({
     }
   });
 };
-function wrapTool({ tool, traces, object, iteration, beforeHook, afterHook, controller }) {
+function wrapTool({ chat, tool, traces, object, iteration, beforeHook, afterHook, controller }) {
   const getToolInput = (input) => _nullishCoalesce(tool.zInput.safeParse(input).data, () => ( input));
   return function(input) {
     const toolCallId = `tcall_${_ulid.ulid.call(void 0, )}`;
@@ -582,9 +591,13 @@ function wrapTool({ tool, traces, object, iteration, beforeHook, afterHook, cont
         if (typeof (beforeRes == null ? void 0 : beforeRes.input) !== "undefined") {
           input2 = beforeRes.input;
         }
-        let output2 = await tool.execute(input2, {
-          callId: toolCallId
-        });
+        let output2 = await tool.execute(
+          input2,
+          {
+            callId: toolCallId
+          },
+          chat
+        );
         const afterRes = await (afterHook == null ? void 0 : afterHook({
           iteration,
           tool,

package/dist/{tool-NCCTC74P.cjs → tool-A7EOCUQG.cjs}

@@ -1,6 +1,6 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunk2JTZOTO5cjs = require('./chunk-2JTZOTO5.cjs');
+var _chunkQOM7BK7Ucjs = require('./chunk-QOM7BK7U.cjs');
 require('./chunk-BLB44ZDU.cjs');
 require('./chunk-PK72FAKD.cjs');
 require('./chunk-KIN7Y247.cjs');
@@ -8,4 +8,4 @@ require('./chunk-4VYDQ3VF.cjs');
 require('./chunk-RKKRSBX3.cjs');
 
 
-exports.Tool = _chunk2JTZOTO5cjs.Tool;
+exports.Tool = _chunkQOM7BK7Ucjs.Tool;

package/dist/{tool-ZP6BC3P2.js → tool-FNWHZGDU.js}

@@ -1,6 +1,6 @@
 import {
   Tool
-} from "./chunk-G5INQ5LJ.js";
+} from "./chunk-UI6J4WKQ.js";
 import "./chunk-LR6KTFHO.js";
 import "./chunk-WSVDMGMR.js";
 import "./chunk-YEAWWJSJ.js";

package/dist/tool.d.ts

@@ -1,5 +1,7 @@
 import { TypeOf, z } from '@bpinternal/zui';
 import { JSONSchema7 } from 'json-schema';
+import { Chat } from './chat.js';
+import { RenderedComponent } from './component.js';
 import { Serializable } from './types.js';
 /**
  * Input parameters passed to tool retry functions.
@@ -349,7 +351,7 @@ export declare class Tool<I extends z.ZodType = z.ZodType, O extends z.ZodType =
         input: IX | ((original: I | undefined) => IX);
         output: OX | ((original: O | undefined) => OX);
         staticInputValues?: SmartPartial<TypeOf<IX>>;
-        handler: (args: TypeOf<IX>, ctx: ToolCallContext) => Promise<TypeOf<OX>>;
+        handler: (args: TypeOf<IX>, ctx: ToolCallContext) => AsyncGenerator<RenderedComponent, TypeOf<OX>> | Promise<TypeOf<OX>>;
         retry: ToolRetryFn<TypeOf<IX>>;
     }>): Tool<IX, OX>;
     private _handler;
@@ -437,9 +439,10 @@ export declare class Tool<I extends z.ZodType = z.ZodType, O extends z.ZodType =
         input?: I;
         output?: O;
         staticInputValues?: Partial<TypeOf<I>>;
-        handler: (args: TypeOf<I>, ctx: ToolCallContext) => Promise<TypeOf<O>>;
+        handler: (args: TypeOf<I>, ctx: ToolCallContext) => AsyncGenerator<RenderedComponent, TypeOf<O>> | Promise<TypeOf<O>>;
         retry?: ToolRetryFn<TypeOf<I>>;
     });
+    private _executeHandler;
     /**
      * Executes the tool with the given input and context.
      *
@@ -464,7 +467,7 @@ export declare class Tool<I extends z.ZodType = z.ZodType, O extends z.ZodType =
      *
      * @internal This method is primarily used internally by the LLMz execution engine
      */
-    execute(rawInput: TypeOf<I>, ctx: ToolCallContext): Promise<TypeOf<O>>;
+    execute(rawInput: TypeOf<I>, ctx: ToolCallContext, chat?: Chat): 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.78",
+  "version": "0.0.80",
   "types": "./dist/index.d.ts",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",
@@ -72,9 +72,9 @@
   },
   "peerDependencies": {
     "@botpress/client": "1.46.0",
-    "@botpress/cognitive": "0.5.4",
+    "@botpress/cognitive": "0.6.0",
     "@bpinternal/thicktoken": "^2.0.0",
-    "@bpinternal/zui": "^2.1.1"
+    "@bpinternal/zui": "^2.3.0"
   },
   "dependenciesMeta": {
     "@bpinternal/zui": {

package/dist/component-36KG7B5H.cjs

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