llmz 0.0.5 → 0.0.6

package/README.md

@@ -4,6 +4,10 @@
 
 ## Getting Started
 
+```
+npm install llmz
+```
+
 ## LLMz vs Tool Calling
 
 ## LLMz vs MCP

package/dist/chat.d.ts

@@ -1,8 +1,8 @@
 import { Component, RenderedComponent } from './component.js';
 import { Context } from './context.js';
 import { ValueOrGetter } from './getter.js';
+import { ExecutionResult } from './result.js';
 import { TranscriptArray, TranscriptMessage } from './transcript.js';
-import { ExecutionResult } from './types.js';
 export type MessageHandler = (input: RenderedComponent) => Promise<void> | void;
 export declare class Chat {
     readonly handler: MessageHandler;

package/dist/{chunk-5NVNEEYW.cjs → chunk-BEPRLBPK.cjs}

@@ -2649,6 +2649,7 @@ function getCompiledCode(code, traces = []) {
   }
 }
 async function runAsyncFunction(context, code, traces = [], signal = null, timeout = MAX_VM_EXECUTION_TIME) {
+  var _a3;
   const transformed = getCompiledCode(code, traces);
   const lines_executed = /* @__PURE__ */ new Map();
   const variables = {};
@@ -2692,12 +2693,12 @@ async function runAsyncFunction(context, code, traces = [], signal = null, timeo
   };
   let currentToolCall;
   context[Identifiers.ToolCallTrackerFnIdentifier] = (callId, type, outputOrError) => {
-    var _a3;
+    var _a4;
     const temp = _chunkJDABP4SDcjs.Signals.maybeDeserializeError(outputOrError == null ? void 0 : outputOrError.message);
     if (type === "end" && temp instanceof _chunkJDABP4SDcjs.SnapshotSignal && (temp == null ? void 0 : temp.toolCall)) {
       currentToolCall = {
         ...temp.toolCall,
-        assignment: (_a3 = transformed.toolCalls.get(callId)) == null ? void 0 : _a3.assignment
+        assignment: (_a4 = transformed.toolCalls.get(callId)) == null ? void 0 : _a4.assignment
       };
     }
   };
@@ -2726,15 +2727,19 @@ async function runAsyncFunction(context, code, traces = [], signal = null, timeo
     const AsyncFunction = async function* () {
     }.constructor;
     return await (async () => {
-      const topLevelProperties = Object.keys(context).filter(
-        (x) => !NO_TRACKING.includes(x) && typeof context[x] !== "function" && typeof context[x] !== "object"
+      const descriptors = Object.getOwnPropertyDescriptors(context);
+      const topLevelProperties = Object.keys(descriptors).filter(
+        (x) => !NO_TRACKING.includes(x) && descriptors[x] && typeof descriptors[x].value !== "function" && typeof descriptors[x].value !== "object"
       );
       const __report = (name, value) => {
-        context[name] = value;
+        if (context[name] !== void 0 && context[name] !== value) {
+          context[name] = value;
+        }
       };
       context.__report = __report;
       const reportAll = topLevelProperties.map((x) => `__report("${x}", ${x})`).join(";");
-      const wrapper = `"use strict"; try { ${transformed.code} } finally { ${reportAll} };`;
+      const assigner = `let __${Identifiers.LineTrackingFnIdentifier} = ${Identifiers.LineTrackingFnIdentifier}; ${Identifiers.LineTrackingFnIdentifier} = function(line) { ${reportAll}; __${Identifiers.LineTrackingFnIdentifier}(line);}`;
+      const wrapper = `"use strict"; try { ${assigner};${transformed.code} } finally { ${reportAll} };`;
       const fn = AsyncFunction(...Object.keys(context), wrapper);
       const res = fn(...Object.values(context));
       do {
@@ -2760,6 +2765,7 @@ async function runAsyncFunction(context, code, traces = [], signal = null, timeo
   const isolatedContext = await isolate.createContext();
   const jail = isolatedContext.global;
   const trackedProperties = /* @__PURE__ */ new Set();
+  const referenceProperties = /* @__PURE__ */ new Set();
   const abort = () => {
     if (USE_ISOLATED_VM) {
       isolate.dispose();
@@ -2779,7 +2785,7 @@ async function runAsyncFunction(context, code, traces = [], signal = null, timeo
           arguments: { reference: true }
         }
       );
-    } else if (typeof context[key] === "object") {
+    } else if (typeof context[key] === "object" && !((_a3 = Object.getOwnPropertyDescriptor(context, key)) == null ? void 0 : _a3.get)) {
       try {
         trackedProperties.add(key);
         const initial = Array.isArray(context[key]) ? new Array(context[key].length) : {};
@@ -2796,9 +2802,28 @@ async function runAsyncFunction(context, code, traces = [], signal = null, timeo
                 }
               );
             } else {
-              await isolatedContext.evalClosure(`global['${key}']['${prop}'] = $0;`, [context[key][prop]], {
-                arguments: { copy: true }
-              });
+              const descriptor = Object.getOwnPropertyDescriptor(context[key], prop);
+              if (descriptor && (descriptor.get || descriptor.set)) {
+                referenceProperties.add(`${key}.${prop}`);
+                await isolatedContext.evalClosure(
+                  `Object.defineProperty(global['${key}'], '${prop}', {
+                    get: () => $0.applySync(null, [], {arguments: {copy: true}}),
+                    set: (value) => $1.applySync(null, [value], {arguments: {copy: true}})
+                  });`,
+                  [
+                    () => context[key][prop],
+                    (value) => {
+                      context[key][prop] = value;
+                      return value;
+                    }
+                  ],
+                  { arguments: { reference: true } }
+                );
+              } else {
+                await isolatedContext.evalClosure(`global['${key}']['${prop}'] = $0;`, [context[key][prop]], {
+                  arguments: { copy: true }
+                });
+              }
             }
           } catch (err) {
             console.error(`Could not copy "${key}.${prop}" (typeof = ${typeof context[key][prop]}) to the sandbox`, err);
@@ -2809,7 +2834,26 @@ async function runAsyncFunction(context, code, traces = [], signal = null, timeo
       }
     } else {
       try {
-        await jail.set(key, context[key], { copy: true });
+        const descriptor = Object.getOwnPropertyDescriptor(context, key);
+        if (descriptor && (descriptor.get || descriptor.set)) {
+          referenceProperties.add(key);
+          await isolatedContext.evalClosure(
+            `Object.defineProperty(global, '${key}', {
+              get: () => $0.applySync(null, [], {arguments: {copy: true}}),
+              set: (value) => $1.applySync(null, [value], {arguments: {copy: true}})
+            });`,
+            [
+              () => context[key],
+              (value) => {
+                context[key] = value;
+                return value;
+              }
+            ],
+            { arguments: { reference: true } }
+          );
+        } else {
+          await jail.set(key, context[key], { copy: true });
+        }
         trackedProperties.add(key);
       } catch (err) {
         console.error(`Could not copy "${key}" to the sandbox (typeof = ${typeof context[key]})`, err);
@@ -2875,7 +2919,7 @@ do {
     }
     copied = true;
     for (const key of trackedProperties) {
-      if (typeof context[key] === "object") {
+      if (typeof context[key] === "object" && !referenceProperties.has(key)) {
         try {
           let properties = [];
           try {
@@ -2890,7 +2934,7 @@ do {
             delete context[key][prop];
           }
           for (const prop of properties) {
-            if (typeof context[key][prop] === "function") {
+            if (typeof context[key][prop] === "function" || referenceProperties.has(`${key}.${prop}`)) {
               continue;
             }
             try {
@@ -2921,6 +2965,9 @@ do {
         }
       } else {
         try {
+          if (referenceProperties.has(key)) {
+            continue;
+          }
           const value = jail.getSync(key, { copy: true });
           try {
             Object.assign(context, { [key]: value });
@@ -2944,9 +2991,6 @@ do {
     }
   };
   const final = await script.run(isolatedContext, {
-    // TODO: fix getRemainingTimeInMillis ...
-    // TODO: probably expose a "timeout" option instead
-    // timeout: clamp(Runtime.getRemainingTimeInMillis() - 10_000, 5_000, MAX_VM_EXECUTION_TIME),
     timeout,
     copy: true,
     promise: true

package/dist/{chunk-MU4LFVY7.cjs → chunk-CH5T6SPC.cjs}

@@ -107,6 +107,72 @@ var Snapshot = class _Snapshot {
   }
 };
 
+// src/result.ts
+var ExecutionResult = class {
+  
+  
+  constructor(status, context) {
+    this.status = status;
+    this.context = context;
+  }
+  isSuccess() {
+    return this.status === "success" && this instanceof SuccessExecutionResult;
+  }
+  isError() {
+    return this.status === "error" && this instanceof ErrorExecutionResult;
+  }
+  isInterrupted() {
+    return this.status === "interrupted" && this instanceof PartialExecutionResult;
+  }
+  is(exit) {
+    return this.status === "success" && this instanceof SuccessExecutionResult && this.result.exit === exit;
+  }
+  get output() {
+    return this.isSuccess() ? this.result.result : null;
+  }
+  get iteration() {
+    return this.context.iterations.at(-1) || null;
+  }
+  get iterations() {
+    return _nullishCoalesce(this.context.iterations, () => ( []));
+  }
+};
+var SuccessExecutionResult = class extends ExecutionResult {
+  
+  constructor(context, result) {
+    super("success", context);
+    this.result = result;
+  }
+  get output() {
+    return this.result.result;
+  }
+  get iteration() {
+    return this.context.iterations.at(-1);
+  }
+};
+var ErrorExecutionResult = class extends ExecutionResult {
+  
+  constructor(context, error) {
+    super("error", context);
+    this.error = error;
+  }
+  get output() {
+    return null;
+  }
+};
+var PartialExecutionResult = class extends ExecutionResult {
+  
+  
+  constructor(context, signal, snapshot) {
+    super("interrupted", context);
+    this.signal = signal;
+    this.snapshot = snapshot;
+  }
+  get output() {
+    return null;
+  }
+};
+
 // src/getter.ts
 var getValue = async (valueOrGetter, ctx) => {
   if (typeof valueOrGetter === "function") {
@@ -121,6 +187,7 @@ var getValue = async (valueOrGetter, ctx) => {
 };
 
 // src/context.ts
+var _zui = require('@bpinternal/zui');
 
 
 // src/handlers.ts
@@ -206,6 +273,20 @@ var ListenExit = new (0, _chunk4I7UPBLNcjs.Exit)({
   name: "listen",
   description: "Listen to the user and provide a response"
 });
+var DefaultExit = new (0, _chunk4I7UPBLNcjs.Exit)({
+  name: "done",
+  description: "When the execution is sucessfully completed or when error recovery is not possible",
+  schema: _zui.z.discriminatedUnion("success", [
+    _zui.z.object({
+      success: _zui.z.literal(true),
+      result: _zui.z.any().describe("The result of the execution")
+    }),
+    _zui.z.object({
+      success: _zui.z.literal(false),
+      error: _zui.z.string().describe("The error message if the execution failed")
+    })
+  ])
+});
 var Iteration = class {
   
   
@@ -565,7 +646,7 @@ var Context = (_class2 = class {
       throw new Error("Too many transcript messages. Expected at most 250 messages.");
     }
     if (!components.length && !exits.length) {
-      throw new Error("When no components are provided, at least one exit is required.");
+      exits.push(DefaultExit);
     }
     return {
       transcript,
@@ -605,4 +686,9 @@ var Context = (_class2 = class {
 
 
 
-exports.Snapshot = Snapshot; exports.getValue = getValue; exports.ThinkExit = ThinkExit; exports.ListenExit = ListenExit; exports.Context = Context;
+
+
+
+
+
+exports.Snapshot = Snapshot; exports.ExecutionResult = ExecutionResult; exports.SuccessExecutionResult = SuccessExecutionResult; exports.ErrorExecutionResult = ErrorExecutionResult; exports.PartialExecutionResult = PartialExecutionResult; exports.getValue = getValue; exports.ThinkExit = ThinkExit; exports.ListenExit = ListenExit; exports.DefaultExit = DefaultExit; exports.Context = Context;

package/dist/{chunk-4EWY34YA.js → chunk-D3ESDRLH.js}

@@ -2649,6 +2649,7 @@ function getCompiledCode(code, traces = []) {
   }
 }
 async function runAsyncFunction(context, code, traces = [], signal = null, timeout = MAX_VM_EXECUTION_TIME) {
+  var _a3;
   const transformed = getCompiledCode(code, traces);
   const lines_executed = /* @__PURE__ */ new Map();
   const variables = {};
@@ -2692,12 +2693,12 @@ async function runAsyncFunction(context, code, traces = [], signal = null, timeo
   };
   let currentToolCall;
   context[Identifiers.ToolCallTrackerFnIdentifier] = (callId, type, outputOrError) => {
-    var _a3;
+    var _a4;
     const temp = Signals.maybeDeserializeError(outputOrError == null ? void 0 : outputOrError.message);
     if (type === "end" && temp instanceof SnapshotSignal && (temp == null ? void 0 : temp.toolCall)) {
       currentToolCall = {
         ...temp.toolCall,
-        assignment: (_a3 = transformed.toolCalls.get(callId)) == null ? void 0 : _a3.assignment
+        assignment: (_a4 = transformed.toolCalls.get(callId)) == null ? void 0 : _a4.assignment
       };
     }
   };
@@ -2726,15 +2727,19 @@ async function runAsyncFunction(context, code, traces = [], signal = null, timeo
     const AsyncFunction = async function* () {
     }.constructor;
     return await (async () => {
-      const topLevelProperties = Object.keys(context).filter(
-        (x) => !NO_TRACKING.includes(x) && typeof context[x] !== "function" && typeof context[x] !== "object"
+      const descriptors = Object.getOwnPropertyDescriptors(context);
+      const topLevelProperties = Object.keys(descriptors).filter(
+        (x) => !NO_TRACKING.includes(x) && descriptors[x] && typeof descriptors[x].value !== "function" && typeof descriptors[x].value !== "object"
       );
       const __report = (name, value) => {
-        context[name] = value;
+        if (context[name] !== void 0 && context[name] !== value) {
+          context[name] = value;
+        }
       };
       context.__report = __report;
       const reportAll = topLevelProperties.map((x) => `__report("${x}", ${x})`).join(";");
-      const wrapper = `"use strict"; try { ${transformed.code} } finally { ${reportAll} };`;
+      const assigner = `let __${Identifiers.LineTrackingFnIdentifier} = ${Identifiers.LineTrackingFnIdentifier}; ${Identifiers.LineTrackingFnIdentifier} = function(line) { ${reportAll}; __${Identifiers.LineTrackingFnIdentifier}(line);}`;
+      const wrapper = `"use strict"; try { ${assigner};${transformed.code} } finally { ${reportAll} };`;
       const fn = AsyncFunction(...Object.keys(context), wrapper);
       const res = fn(...Object.values(context));
       do {
@@ -2760,6 +2765,7 @@ async function runAsyncFunction(context, code, traces = [], signal = null, timeo
   const isolatedContext = await isolate.createContext();
   const jail = isolatedContext.global;
   const trackedProperties = /* @__PURE__ */ new Set();
+  const referenceProperties = /* @__PURE__ */ new Set();
   const abort = () => {
     if (USE_ISOLATED_VM) {
       isolate.dispose();
@@ -2779,7 +2785,7 @@ async function runAsyncFunction(context, code, traces = [], signal = null, timeo
           arguments: { reference: true }
         }
       );
-    } else if (typeof context[key] === "object") {
+    } else if (typeof context[key] === "object" && !((_a3 = Object.getOwnPropertyDescriptor(context, key)) == null ? void 0 : _a3.get)) {
       try {
         trackedProperties.add(key);
         const initial = Array.isArray(context[key]) ? new Array(context[key].length) : {};
@@ -2796,9 +2802,28 @@ async function runAsyncFunction(context, code, traces = [], signal = null, timeo
                 }
               );
             } else {
-              await isolatedContext.evalClosure(`global['${key}']['${prop}'] = $0;`, [context[key][prop]], {
-                arguments: { copy: true }
-              });
+              const descriptor = Object.getOwnPropertyDescriptor(context[key], prop);
+              if (descriptor && (descriptor.get || descriptor.set)) {
+                referenceProperties.add(`${key}.${prop}`);
+                await isolatedContext.evalClosure(
+                  `Object.defineProperty(global['${key}'], '${prop}', {
+                    get: () => $0.applySync(null, [], {arguments: {copy: true}}),
+                    set: (value) => $1.applySync(null, [value], {arguments: {copy: true}})
+                  });`,
+                  [
+                    () => context[key][prop],
+                    (value) => {
+                      context[key][prop] = value;
+                      return value;
+                    }
+                  ],
+                  { arguments: { reference: true } }
+                );
+              } else {
+                await isolatedContext.evalClosure(`global['${key}']['${prop}'] = $0;`, [context[key][prop]], {
+                  arguments: { copy: true }
+                });
+              }
             }
           } catch (err) {
             console.error(`Could not copy "${key}.${prop}" (typeof = ${typeof context[key][prop]}) to the sandbox`, err);
@@ -2809,7 +2834,26 @@ async function runAsyncFunction(context, code, traces = [], signal = null, timeo
       }
     } else {
       try {
-        await jail.set(key, context[key], { copy: true });
+        const descriptor = Object.getOwnPropertyDescriptor(context, key);
+        if (descriptor && (descriptor.get || descriptor.set)) {
+          referenceProperties.add(key);
+          await isolatedContext.evalClosure(
+            `Object.defineProperty(global, '${key}', {
+              get: () => $0.applySync(null, [], {arguments: {copy: true}}),
+              set: (value) => $1.applySync(null, [value], {arguments: {copy: true}})
+            });`,
+            [
+              () => context[key],
+              (value) => {
+                context[key] = value;
+                return value;
+              }
+            ],
+            { arguments: { reference: true } }
+          );
+        } else {
+          await jail.set(key, context[key], { copy: true });
+        }
         trackedProperties.add(key);
       } catch (err) {
         console.error(`Could not copy "${key}" to the sandbox (typeof = ${typeof context[key]})`, err);
@@ -2875,7 +2919,7 @@ do {
     }
     copied = true;
     for (const key of trackedProperties) {
-      if (typeof context[key] === "object") {
+      if (typeof context[key] === "object" && !referenceProperties.has(key)) {
         try {
           let properties = [];
           try {
@@ -2890,7 +2934,7 @@ do {
             delete context[key][prop];
           }
           for (const prop of properties) {
-            if (typeof context[key][prop] === "function") {
+            if (typeof context[key][prop] === "function" || referenceProperties.has(`${key}.${prop}`)) {
               continue;
             }
             try {
@@ -2921,6 +2965,9 @@ do {
         }
       } else {
         try {
+          if (referenceProperties.has(key)) {
+            continue;
+          }
           const value = jail.getSync(key, { copy: true });
           try {
             Object.assign(context, { [key]: value });
@@ -2944,9 +2991,6 @@ do {
     }
   };
   const final = await script.run(isolatedContext, {
-    // TODO: fix getRemainingTimeInMillis ...
-    // TODO: probably expose a "timeout" option instead
-    // timeout: clamp(Runtime.getRemainingTimeInMillis() - 10_000, 5_000, MAX_VM_EXECUTION_TIME),
     timeout,
     copy: true,
     promise: true

package/dist/{chunk-3U57H7GT.js → chunk-XJNRORSK.js}

@@ -107,6 +107,72 @@ var Snapshot = class _Snapshot {
   }
 };
 
+// src/result.ts
+var ExecutionResult = class {
+  status;
+  context;
+  constructor(status, context) {
+    this.status = status;
+    this.context = context;
+  }
+  isSuccess() {
+    return this.status === "success" && this instanceof SuccessExecutionResult;
+  }
+  isError() {
+    return this.status === "error" && this instanceof ErrorExecutionResult;
+  }
+  isInterrupted() {
+    return this.status === "interrupted" && this instanceof PartialExecutionResult;
+  }
+  is(exit) {
+    return this.status === "success" && this instanceof SuccessExecutionResult && this.result.exit === exit;
+  }
+  get output() {
+    return this.isSuccess() ? this.result.result : null;
+  }
+  get iteration() {
+    return this.context.iterations.at(-1) || null;
+  }
+  get iterations() {
+    return this.context.iterations ?? [];
+  }
+};
+var SuccessExecutionResult = class extends ExecutionResult {
+  result;
+  constructor(context, result) {
+    super("success", context);
+    this.result = result;
+  }
+  get output() {
+    return this.result.result;
+  }
+  get iteration() {
+    return this.context.iterations.at(-1);
+  }
+};
+var ErrorExecutionResult = class extends ExecutionResult {
+  error;
+  constructor(context, error) {
+    super("error", context);
+    this.error = error;
+  }
+  get output() {
+    return null;
+  }
+};
+var PartialExecutionResult = class extends ExecutionResult {
+  signal;
+  snapshot;
+  constructor(context, signal, snapshot) {
+    super("interrupted", context);
+    this.signal = signal;
+    this.snapshot = snapshot;
+  }
+  get output() {
+    return null;
+  }
+};
+
 // src/getter.ts
 var getValue = async (valueOrGetter, ctx) => {
   if (typeof valueOrGetter === "function") {
@@ -121,6 +187,7 @@ var getValue = async (valueOrGetter, ctx) => {
 };
 
 // src/context.ts
+import { z } from "@bpinternal/zui";
 import { ulid as ulid2 } from "ulid";
 
 // src/handlers.ts
@@ -206,6 +273,20 @@ var ListenExit = new Exit({
   name: "listen",
   description: "Listen to the user and provide a response"
 });
+var DefaultExit = new Exit({
+  name: "done",
+  description: "When the execution is sucessfully completed or when error recovery is not possible",
+  schema: z.discriminatedUnion("success", [
+    z.object({
+      success: z.literal(true),
+      result: z.any().describe("The result of the execution")
+    }),
+    z.object({
+      success: z.literal(false),
+      error: z.string().describe("The error message if the execution failed")
+    })
+  ])
+});
 var Iteration = class {
   id;
   messages;
@@ -565,7 +646,7 @@ var Context = class {
       throw new Error("Too many transcript messages. Expected at most 250 messages.");
     }
     if (!components.length && !exits.length) {
-      throw new Error("When no components are provided, at least one exit is required.");
+      exits.push(DefaultExit);
     }
     return {
       transcript,
@@ -601,8 +682,13 @@ var Context = class {
 
 export {
   Snapshot,
+  ExecutionResult,
+  SuccessExecutionResult,
+  ErrorExecutionResult,
+  PartialExecutionResult,
   getValue,
   ThinkExit,
   ListenExit,
+  DefaultExit,
   Context
 };

package/dist/context.d.ts

@@ -82,6 +82,13 @@ export declare namespace IterationStatuses {
 }
 export declare const ThinkExit: Exit<unknown>;
 export declare const ListenExit: Exit<unknown>;
+export declare const DefaultExit: Exit<{
+    success: true;
+    result?: any;
+} | {
+    error: string;
+    success: false;
+}>;
 export declare class Iteration {
     id: string;
     messages: OAI.Message[];

package/dist/index.cjs

@@ -3,7 +3,12 @@
 
 
 
-var _chunkMU4LFVY7cjs = require('./chunk-MU4LFVY7.cjs');
+
+
+
+
+
+var _chunkCH5T6SPCcjs = require('./chunk-CH5T6SPC.cjs');
 require('./chunk-6WT5VZBG.cjs');
 
 
@@ -16,6 +21,7 @@ var _chunkHYVKY3W6cjs = require('./chunk-HYVKY3W6.cjs');
 
 
 
+
 var _chunkJDABP4SDcjs = require('./chunk-JDABP4SD.cjs');
 require('./chunk-IKSIOIIP.cjs');
 
@@ -870,16 +876,16 @@ var Chat = class {
 
 // src/index.ts
 var execute = async (props) => {
-  const { executeContext } = await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-VDA4M42R.cjs")));
+  const { executeContext } = await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-CPMVYIOJ.cjs")));
   return executeContext(props);
 };
 var init = async () => {
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-VDA4M42R.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./llmz-CPMVYIOJ.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./component-TSNW3SC7.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./tool-SWJYOR2Z.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./exit-KJ4COC5N.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./jsx-AJAXBWFE.cjs")));
-  await Promise.resolve().then(() => _interopRequireWildcard(require("./vm-GJ5R72AP.cjs")));
+  await Promise.resolve().then(() => _interopRequireWildcard(require("./vm-2DLG7V4G.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./utils-QC4I2L6R.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./truncator-NYN7BGKJ.cjs")));
   await Promise.resolve().then(() => _interopRequireWildcard(require("./typings-XR6CYHW4.cjs")));
@@ -904,4 +910,10 @@ var init = async () => {
 
 
 
-exports.Chat = Chat; exports.CitationsManager = CitationsManager; exports.Component = _chunkEJRCDWBAcjs.Component; exports.DefaultComponents = DefaultComponents; exports.Exit = _chunk4I7UPBLNcjs.Exit; exports.ListenExit = _chunkMU4LFVY7cjs.ListenExit; exports.ObjectInstance = ObjectInstance; exports.Snapshot = _chunkMU4LFVY7cjs.Snapshot; exports.SnapshotSignal = _chunkJDABP4SDcjs.SnapshotSignal; exports.ThinkExit = _chunkMU4LFVY7cjs.ThinkExit; exports.ThinkSignal = _chunkJDABP4SDcjs.ThinkSignal; exports.Tool = _chunk4QTK5ALHcjs.Tool; exports.assertValidComponent = _chunkEJRCDWBAcjs.assertValidComponent; exports.execute = execute; exports.getValue = _chunkMU4LFVY7cjs.getValue; exports.init = init; exports.isAnyComponent = _chunkEJRCDWBAcjs.isAnyComponent; exports.isComponent = _chunkEJRCDWBAcjs.isComponent;
+
+
+
+
+
+
+exports.Chat = Chat; exports.CitationsManager = CitationsManager; exports.Component = _chunkEJRCDWBAcjs.Component; exports.DefaultComponents = DefaultComponents; exports.DefaultExit = _chunkCH5T6SPCcjs.DefaultExit; exports.ErrorExecutionResult = _chunkCH5T6SPCcjs.ErrorExecutionResult; exports.ExecutionResult = _chunkCH5T6SPCcjs.ExecutionResult; exports.Exit = _chunk4I7UPBLNcjs.Exit; exports.ListenExit = _chunkCH5T6SPCcjs.ListenExit; exports.LoopExceededError = _chunkJDABP4SDcjs.LoopExceededError; exports.ObjectInstance = ObjectInstance; exports.PartialExecutionResult = _chunkCH5T6SPCcjs.PartialExecutionResult; exports.Snapshot = _chunkCH5T6SPCcjs.Snapshot; exports.SnapshotSignal = _chunkJDABP4SDcjs.SnapshotSignal; exports.SuccessExecutionResult = _chunkCH5T6SPCcjs.SuccessExecutionResult; exports.ThinkExit = _chunkCH5T6SPCcjs.ThinkExit; exports.ThinkSignal = _chunkJDABP4SDcjs.ThinkSignal; exports.Tool = _chunk4QTK5ALHcjs.Tool; exports.assertValidComponent = _chunkEJRCDWBAcjs.assertValidComponent; exports.execute = execute; exports.getValue = _chunkCH5T6SPCcjs.getValue; exports.init = init; exports.isAnyComponent = _chunkEJRCDWBAcjs.isAnyComponent; exports.isComponent = _chunkEJRCDWBAcjs.isComponent;

package/dist/index.d.ts

@@ -1,16 +1,17 @@
 export { Tool } from './tool.js';
 export { Exit, ExitResult } from './exit.js';
 export { ObjectInstance } from './objects.js';
-export { SnapshotSignal, ThinkSignal } from './errors.js';
+export { SnapshotSignal, ThinkSignal, LoopExceededError } from './errors.js';
 export { Component, RenderedComponent, LeafComponentDefinition, ContainerComponentDefinition, DefaultComponentDefinition, ComponentDefinition, assertValidComponent, isComponent, isAnyComponent, } from './component.js';
 export { Citation, CitationsManager } from './citations.js';
 export { DefaultComponents } from './component.default.js';
 export { Snapshot } from './snapshots.js';
 export { Chat, type MessageHandler } from './chat.js';
 import { type ExecutionProps } from './llmz.js';
-import { type ExecutionResult } from './types.js';
-export { type ExecutionResult, Trace } from './types.js';
-export { type Iteration, ListenExit, ThinkExit, IterationStatuses, IterationStatus } from './context.js';
+import { ExecutionResult } from './result.js';
+export { ErrorExecutionResult, ExecutionResult, PartialExecutionResult, SuccessExecutionResult } from './result.js';
+export { Trace } from './types.js';
+export { type Iteration, ListenExit, ThinkExit, DefaultExit, IterationStatuses, IterationStatus } from './context.js';
 export { type ValueOrGetter, getValue } from './getter.js';
 export declare const execute: (props: ExecutionProps) => Promise<ExecutionResult>;
 /**

package/dist/index.js

@@ -1,9 +1,14 @@
 import {
+  DefaultExit,
+  ErrorExecutionResult,
+  ExecutionResult,
   ListenExit,
+  PartialExecutionResult,
   Snapshot,
+  SuccessExecutionResult,
   ThinkExit,
   getValue
-} from "./chunk-3U57H7GT.js";
+} from "./chunk-XJNRORSK.js";
 import "./chunk-WLUVY5QU.js";
 import {
   Tool
@@ -14,6 +19,7 @@ import {
 } from "./chunk-UNMKB2DA.js";
 import {
   CodeFormattingError,
+  LoopExceededError,
   SnapshotSignal,
   ThinkSignal
 } from "./chunk-JKVVQN2P.js";
@@ -870,16 +876,16 @@ var Chat = class {
 
 // src/index.ts
 var execute = async (props) => {
-  const { executeContext } = await import("./llmz-YE5N54IU.js");
+  const { executeContext } = await import("./llmz-LAN4WDFW.js");
   return executeContext(props);
 };
 var init = async () => {
-  await import("./llmz-YE5N54IU.js");
+  await import("./llmz-LAN4WDFW.js");
   await import("./component-R2Y74VUI.js");
   await import("./tool-LY77IWV2.js");
   await import("./exit-OIYZLBVJ.js");
   await import("./jsx-AEHVFB3L.js");
-  await import("./vm-ZUQOSRRM.js");
+  await import("./vm-FLBMZUA2.js");
   await import("./utils-L5GLCS3C.js");
   await import("./truncator-WAL2GCUY.js");
   await import("./typings-AIV2OASX.js");
@@ -890,11 +896,17 @@ export {
   CitationsManager,
   Component,
   DefaultComponents,
+  DefaultExit,
+  ErrorExecutionResult,
+  ExecutionResult,
   Exit,
   ListenExit,
+  LoopExceededError,
   ObjectInstance,
+  PartialExecutionResult,
   Snapshot,
   SnapshotSignal,
+  SuccessExecutionResult,
   ThinkExit,
   ThinkSignal,
   Tool,

package/dist/{llmz-VDA4M42R.cjs → llmz-CPMVYIOJ.cjs}

@@ -1,10 +1,13 @@
 "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 _chunk5NVNEEYWcjs = require('./chunk-5NVNEEYW.cjs');
+var _chunkBEPRLBPKcjs = require('./chunk-BEPRLBPK.cjs');
 
 
 
-var _chunkMU4LFVY7cjs = require('./chunk-MU4LFVY7.cjs');
+
+
+
+var _chunkCH5T6SPCcjs = require('./chunk-CH5T6SPC.cjs');
 require('./chunk-6WT5VZBG.cjs');
 require('./chunk-4QTK5ALH.cjs');
 require('./chunk-HYVKY3W6.cjs');
@@ -16,6 +19,7 @@ require('./chunk-HYVKY3W6.cjs');
 
 
 
+
 var _chunkJDABP4SDcjs = require('./chunk-JDABP4SD.cjs');
 
 
@@ -69,7 +73,7 @@ var _executeContext = async (props) => {
   const { signal, onIterationEnd, onTrace, onExit, onBeforeExecution } = props;
   const cognitive = props.client instanceof _cognitive.Cognitive ? props.client : new (0, _cognitive.Cognitive)({ client: props.client });
   const cleanups = [];
-  const ctx = new (0, _chunkMU4LFVY7cjs.Context)({
+  const ctx = new (0, _chunkCH5T6SPCcjs.Context)({
     chat: props.chat,
     instructions: props.instructions,
     objects: props.objects,
@@ -84,12 +88,7 @@ var _executeContext = async (props) => {
   try {
     while (true) {
       if (ctx.iterations.length >= ctx.loop) {
-        return {
-          status: "error",
-          context: ctx,
-          error: `Loop limit exceeded. Maximum allowed loops: ${ctx.loop}`,
-          iterations: ctx.iterations
-        };
+        return new (0, _chunkCH5T6SPCcjs.ErrorExecutionResult)(ctx, new (0, _chunkJDABP4SDcjs.LoopExceededError)());
       }
       const iteration = await ctx.nextIteration();
       if (signal == null ? void 0 : signal.aborted) {
@@ -99,12 +98,7 @@ var _executeContext = async (props) => {
             reason: _nullishCoalesce(signal.reason, () => ( "The operation was aborted"))
           }
         });
-        return {
-          status: "error",
-          error: _nullishCoalesce(signal.reason, () => ( "The operation was aborted")),
-          context: ctx,
-          iterations: ctx.iterations
-        };
+        return new (0, _chunkCH5T6SPCcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(signal.reason, () => ( "The operation was aborted")));
       }
       cleanups.push(
         iteration.traces.onPush((traces) => {
@@ -138,42 +132,25 @@ var _executeContext = async (props) => {
       }
       if (iteration.status.type === "exit_success") {
         const exitName = iteration.status.exit_success.exit_name;
-        return {
-          status: "success",
-          context: ctx,
-          iterations: ctx.iterations,
-          result: {
-            exit: iteration.exits.find((x) => x.name === exitName),
-            result: iteration.status.exit_success.return_value
-          }
-        };
+        return new (0, _chunkCH5T6SPCcjs.SuccessExecutionResult)(ctx, {
+          exit: iteration.exits.find((x) => x.name === exitName),
+          result: iteration.status.exit_success.return_value
+        });
       }
       if (iteration.status.type === "callback_requested") {
-        return {
-          status: "interrupted",
-          context: ctx,
-          iterations: ctx.iterations,
-          signal: iteration.status.callback_requested.signal,
-          snapshot: _chunkMU4LFVY7cjs.Snapshot.fromSignal(iteration.status.callback_requested.signal)
-        };
+        return new (0, _chunkCH5T6SPCcjs.PartialExecutionResult)(
+          ctx,
+          iteration.status.callback_requested.signal,
+          _chunkCH5T6SPCcjs.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 {
-        context: ctx,
-        error: _nullishCoalesce(iteration.error, () => ( `Unknown error. Status: ${iteration.status.type}`)),
-        status: "error",
-        iterations: ctx.iterations
-      };
+      return new (0, _chunkCH5T6SPCcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(iteration.error, () => ( `Unknown error. Status: ${iteration.status.type}`)));
     }
   } catch (error) {
-    return {
-      status: "error",
-      iterations: ctx.iterations,
-      context: ctx,
-      error: error instanceof Error ? error.message : _nullishCoalesce((error == null ? void 0 : error.toString()), () => ( "Unknown error"))
-    };
+    return new (0, _chunkCH5T6SPCcjs.ErrorExecutionResult)(ctx, _nullishCoalesce(error, () => ( "Unknown error")));
   } finally {
     for (const cleanup of cleanups) {
       try {
@@ -342,7 +319,7 @@ var executeIteration = async ({
     });
   }
   startedAt = Date.now();
-  const result = await _chunk5NVNEEYWcjs.runAsyncFunction.call(void 0, vmContext, iteration.code, traces, abortSignal, ctx.timeout).catch(
+  const result = await _chunkBEPRLBPKcjs.runAsyncFunction.call(void 0, vmContext, iteration.code, traces, abortSignal, ctx.timeout).catch(
     (err) => {
       return {
         success: false,

package/dist/{llmz-YE5N54IU.js → llmz-LAN4WDFW.js}

@@ -1,10 +1,13 @@
 import {
   runAsyncFunction
-} from "./chunk-4EWY34YA.js";
+} from "./chunk-D3ESDRLH.js";
 import {
   Context,
-  Snapshot
-} from "./chunk-3U57H7GT.js";
+  ErrorExecutionResult,
+  PartialExecutionResult,
+  Snapshot,
+  SuccessExecutionResult
+} from "./chunk-XJNRORSK.js";
 import "./chunk-WLUVY5QU.js";
 import "./chunk-QBXIE6EY.js";
 import "./chunk-UNMKB2DA.js";
@@ -12,6 +15,7 @@ import {
   AssignmentError,
   CodeExecutionError,
   InvalidCodeError,
+  LoopExceededError,
   Signals,
   SnapshotSignal,
   ThinkSignal,
@@ -84,12 +88,7 @@ var _executeContext = async (props) => {
   try {
     while (true) {
       if (ctx.iterations.length >= ctx.loop) {
-        return {
-          status: "error",
-          context: ctx,
-          error: `Loop limit exceeded. Maximum allowed loops: ${ctx.loop}`,
-          iterations: ctx.iterations
-        };
+        return new ErrorExecutionResult(ctx, new LoopExceededError());
       }
       const iteration = await ctx.nextIteration();
       if (signal == null ? void 0 : signal.aborted) {
@@ -99,12 +98,7 @@ var _executeContext = async (props) => {
             reason: signal.reason ?? "The operation was aborted"
           }
         });
-        return {
-          status: "error",
-          error: signal.reason ?? "The operation was aborted",
-          context: ctx,
-          iterations: ctx.iterations
-        };
+        return new ErrorExecutionResult(ctx, signal.reason ?? "The operation was aborted");
       }
       cleanups.push(
         iteration.traces.onPush((traces) => {
@@ -138,42 +132,25 @@ var _executeContext = async (props) => {
       }
       if (iteration.status.type === "exit_success") {
         const exitName = iteration.status.exit_success.exit_name;
-        return {
-          status: "success",
-          context: ctx,
-          iterations: ctx.iterations,
-          result: {
-            exit: iteration.exits.find((x) => x.name === exitName),
-            result: iteration.status.exit_success.return_value
-          }
-        };
+        return new SuccessExecutionResult(ctx, {
+          exit: iteration.exits.find((x) => x.name === exitName),
+          result: iteration.status.exit_success.return_value
+        });
       }
       if (iteration.status.type === "callback_requested") {
-        return {
-          status: "interrupted",
-          context: ctx,
-          iterations: ctx.iterations,
-          signal: iteration.status.callback_requested.signal,
-          snapshot: Snapshot.fromSignal(iteration.status.callback_requested.signal)
-        };
+        return new PartialExecutionResult(
+          ctx,
+          iteration.status.callback_requested.signal,
+          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 {
-        context: ctx,
-        error: iteration.error ?? `Unknown error. Status: ${iteration.status.type}`,
-        status: "error",
-        iterations: ctx.iterations
-      };
+      return new ErrorExecutionResult(ctx, iteration.error ?? `Unknown error. Status: ${iteration.status.type}`);
     }
   } catch (error) {
-    return {
-      status: "error",
-      iterations: ctx.iterations,
-      context: ctx,
-      error: error instanceof Error ? error.message : (error == null ? void 0 : error.toString()) ?? "Unknown error"
-    };
+    return new ErrorExecutionResult(ctx, error ?? "Unknown error");
   } finally {
     for (const cleanup of cleanups) {
       try {

package/dist/llmz.d.ts

@@ -6,7 +6,8 @@ import { ValueOrGetter } from './getter.js';
 import { type ObjectInstance } from './objects.js';
 import { Snapshot } from './snapshots.js';
 import { type Tool } from './tool.js';
-import { ExecutionResult, Trace } from './types.js';
+import { Trace } from './types.js';
+import { ExecutionResult } from './result.js';
 export type ExecutionHooks = {
     /**
      * Called after each iteration ends

package/dist/result.d.ts

@@ -0,0 +1,33 @@
+import { Context, Iteration } from './context.js';
+import { SnapshotSignal } from './errors.js';
+import { Exit, ExitResult } from './exit.js';
+import { Snapshot } from './snapshots.js';
+export declare abstract class ExecutionResult {
+    readonly status: 'success' | 'error' | 'interrupted';
+    readonly context: Context;
+    protected constructor(status: 'success' | 'error' | 'interrupted', context: Context);
+    isSuccess(): this is SuccessExecutionResult;
+    isError(): this is ErrorExecutionResult;
+    isInterrupted(): this is PartialExecutionResult;
+    is<T>(exit: Exit<T>): this is SuccessExecutionResult<T>;
+    get output(): unknown | null;
+    get iteration(): Iteration | null;
+    get iterations(): Iteration[];
+}
+export declare class SuccessExecutionResult<TOutput = unknown> extends ExecutionResult {
+    readonly result: ExitResult<TOutput>;
+    constructor(context: Context, result: ExitResult<TOutput>);
+    get output(): TOutput;
+    get iteration(): Iteration;
+}
+export declare class ErrorExecutionResult extends ExecutionResult {
+    readonly error: unknown;
+    constructor(context: Context, error: unknown);
+    get output(): null;
+}
+export declare class PartialExecutionResult extends ExecutionResult {
+    readonly signal: SnapshotSignal;
+    readonly snapshot: Snapshot;
+    constructor(context: Context, signal: SnapshotSignal, snapshot: Snapshot);
+    get output(): null;
+}

package/dist/types.d.ts

@@ -1,7 +1,4 @@
-import { Iteration, type Context } from './context.js';
-import { type SnapshotSignal, type VMSignal } from './errors.js';
-import { ExitResult } from './exit.js';
-import { type Snapshot } from './snapshots.js';
+import { type VMSignal } from './errors.js';
 export declare namespace Traces {
     type Comment = TraceTemplate<'comment', {
         comment: string;
@@ -101,37 +98,6 @@ export type ObjectMutation = {
     before: any;
     after: any;
 };
-export type SuccessExecutionResult = {
-    status: 'success';
-    iterations: Iteration[];
-    context: Context;
-    result: ExitResult;
-};
-export type PartialExecutionResult = {
-    status: 'interrupted';
-    iterations: Iteration[];
-    context: Context;
-    signal: SnapshotSignal;
-    snapshot: Snapshot;
-};
-export type ErrorExecutionResult = {
-    status: 'error';
-    iterations: Iteration[];
-    context: Context;
-    error: string;
-};
-export type ExecutionResult = SuccessExecutionResult | PartialExecutionResult | ErrorExecutionResult;
-export declare function expectStatus<T extends ExecutionResult['status']>(execution: ExecutionResult, status: T): asserts execution is Extract<ExecutionResult, {
-    status: T;
-}>;
-export type Tool = {
-    name: string;
-    aliases?: string[];
-    description?: string;
-    input?: unknown;
-    output?: unknown;
-    metadata?: Record<string, any>;
-};
 export type ZuiType<Output = any, Input = Output> = {
     readonly __type__: 'ZuiType';
     readonly _output: Output;

package/dist/{vm-GJ5R72AP.cjs → vm-2DLG7V4G.cjs}

@@ -1,7 +1,7 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
 
-var _chunk5NVNEEYWcjs = require('./chunk-5NVNEEYW.cjs');
+var _chunkBEPRLBPKcjs = require('./chunk-BEPRLBPK.cjs');
 require('./chunk-JDABP4SD.cjs');
 require('./chunk-IKSIOIIP.cjs');
 require('./chunk-KMZDFWYZ.cjs');
@@ -9,4 +9,4 @@ require('./chunk-UQOBUJIQ.cjs');
 
 
 
-exports.USE_ISOLATED_VM = _chunk5NVNEEYWcjs.USE_ISOLATED_VM; exports.runAsyncFunction = _chunk5NVNEEYWcjs.runAsyncFunction;
+exports.USE_ISOLATED_VM = _chunkBEPRLBPKcjs.USE_ISOLATED_VM; exports.runAsyncFunction = _chunkBEPRLBPKcjs.runAsyncFunction;

package/dist/{vm-ZUQOSRRM.js → vm-FLBMZUA2.js}

@@ -1,7 +1,7 @@
 import {
   USE_ISOLATED_VM,
   runAsyncFunction
-} from "./chunk-4EWY34YA.js";
+} from "./chunk-D3ESDRLH.js";
 import "./chunk-JKVVQN2P.js";
 import "./chunk-JQBT7UWN.js";
 import "./chunk-ORQP26SZ.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.5",
+  "version": "0.0.6",
   "types": "./dist/index.d.ts",
   "exports": {
     ".": {