llmz 0.0.4 → 0.0.6

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-JK2LZW2G.cjs → chunk-EJRCDWBA.cjs}

@@ -1,8 +1,10 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
+
 var _chunkKMZDFWYZcjs = require('./chunk-KMZDFWYZ.cjs');
 
 // src/component.ts
+var _zui = require('@bpinternal/zui');
 function assertValidComponent(component) {
   if (!component.name) {
     throw new Error("Component must have a name");
@@ -36,6 +38,16 @@ function assertValidComponent(component) {
     throw new Error("Container component must have container props and children");
   }
 }
+var getDefaultValue = (schema) => {
+  if (schema._def.defaultValue !== void 0) {
+    if (typeof schema._def.defaultValue === "function") {
+      return String(schema._def.defaultValue()).toString();
+    } else {
+      return String(schema._def.defaultValue);
+    }
+  }
+  return "";
+};
 function getComponentReference(component) {
   let doc = `### <${component.name}>
 
@@ -44,12 +56,35 @@ function getComponentReference(component) {
 
 `;
   const getPropsDoc = (props) => {
-    if (props.length === 0)
+    const shape = props.shape;
+    if (Object.keys(shape).length === 0)
       return "_No props._\n\n";
-    return props.map((prop) => {
-      const required = prop.required ? "**(required)**" : "(optional)";
-      const def = prop.default !== void 0 ? ` _Default: \`${prop.default}\`_` : "";
-      return `- \`${prop.name}: ${prop.type}\` ${required} \u2014 ${prop.description || ""}${def}`;
+    const zodTypeToTsType = {
+      ZodString: "string",
+      ZodNumber: "number",
+      ZodBoolean: "boolean",
+      ZodEnum: "enum",
+      ZodArray: "array",
+      ZodObject: "object",
+      ZodDate: "date",
+      ZodBigInt: "bigint",
+      ZodSymbol: "symbol",
+      ZodUndefined: "undefined",
+      ZodNull: "null",
+      ZodVoid: "void",
+      ZodNever: "never",
+      ZodUnknown: "unknown",
+      ZodAny: "any"
+    };
+    return Object.entries(shape).map(([name, schema]) => {
+      const naked = schema.naked();
+      const zodType = naked._def.typeName;
+      const defValue = getDefaultValue(schema);
+      const typings = naked instanceof _zui.z.ZodEnum ? naked._def.values.map((x) => `"${x}"`).join(" | ") : zodTypeToTsType[zodType] || zodType;
+      const required = !schema.isOptional() ? "**(required)**" : "(optional)";
+      const def = defValue ? ` _Default: \`${defValue}\`_` : "";
+      const description = schema.description || schema.naked().description || (schema == null ? void 0 : schema._def.description) || "";
+      return `- \`${name}: ${typings}\` ${required} \u2014 ${description}${def}`;
     }).join("\n") + "\n\n";
   };
   const getChildrenDoc = (children) => {
@@ -105,10 +140,14 @@ var Component = class {
 function isComponent(rendered, component) {
   return _chunkKMZDFWYZcjs.isJsxComponent.call(void 0, component.definition.name, rendered);
 }
+function isAnyComponent(message) {
+  return _chunkKMZDFWYZcjs.isAnyJsxComponent.call(void 0, message);
+}
+
 
 
 
 
 
 
-exports.assertValidComponent = assertValidComponent; exports.getComponentReference = getComponentReference; exports.Component = Component; exports.isComponent = isComponent;
+exports.assertValidComponent = assertValidComponent; exports.getComponentReference = getComponentReference; exports.Component = Component; exports.isComponent = isComponent; exports.isAnyComponent = isAnyComponent;

package/dist/{chunk-ZCPQ3QOW.js → chunk-HP7RKM25.js}

@@ -1,8 +1,10 @@
 import {
+  isAnyJsxComponent,
   isJsxComponent
 } from "./chunk-ORQP26SZ.js";
 
 // src/component.ts
+import { z } from "@bpinternal/zui";
 function assertValidComponent(component) {
   if (!component.name) {
     throw new Error("Component must have a name");
@@ -36,6 +38,16 @@ function assertValidComponent(component) {
     throw new Error("Container component must have container props and children");
   }
 }
+var getDefaultValue = (schema) => {
+  if (schema._def.defaultValue !== void 0) {
+    if (typeof schema._def.defaultValue === "function") {
+      return String(schema._def.defaultValue()).toString();
+    } else {
+      return String(schema._def.defaultValue);
+    }
+  }
+  return "";
+};
 function getComponentReference(component) {
   let doc = `### <${component.name}>
 
@@ -44,12 +56,35 @@ function getComponentReference(component) {
 
 `;
   const getPropsDoc = (props) => {
-    if (props.length === 0)
+    const shape = props.shape;
+    if (Object.keys(shape).length === 0)
       return "_No props._\n\n";
-    return props.map((prop) => {
-      const required = prop.required ? "**(required)**" : "(optional)";
-      const def = prop.default !== void 0 ? ` _Default: \`${prop.default}\`_` : "";
-      return `- \`${prop.name}: ${prop.type}\` ${required} \u2014 ${prop.description || ""}${def}`;
+    const zodTypeToTsType = {
+      ZodString: "string",
+      ZodNumber: "number",
+      ZodBoolean: "boolean",
+      ZodEnum: "enum",
+      ZodArray: "array",
+      ZodObject: "object",
+      ZodDate: "date",
+      ZodBigInt: "bigint",
+      ZodSymbol: "symbol",
+      ZodUndefined: "undefined",
+      ZodNull: "null",
+      ZodVoid: "void",
+      ZodNever: "never",
+      ZodUnknown: "unknown",
+      ZodAny: "any"
+    };
+    return Object.entries(shape).map(([name, schema]) => {
+      const naked = schema.naked();
+      const zodType = naked._def.typeName;
+      const defValue = getDefaultValue(schema);
+      const typings = naked instanceof z.ZodEnum ? naked._def.values.map((x) => `"${x}"`).join(" | ") : zodTypeToTsType[zodType] || zodType;
+      const required = !schema.isOptional() ? "**(required)**" : "(optional)";
+      const def = defValue ? ` _Default: \`${defValue}\`_` : "";
+      const description = schema.description || schema.naked().description || (schema == null ? void 0 : schema._def.description) || "";
+      return `- \`${name}: ${typings}\` ${required} \u2014 ${description}${def}`;
     }).join("\n") + "\n\n";
   };
   const getChildrenDoc = (children) => {
@@ -105,10 +140,14 @@ var Component = class {
 function isComponent(rendered, component) {
   return isJsxComponent(component.definition.name, rendered);
 }
+function isAnyComponent(message) {
+  return isAnyJsxComponent(message);
+}
 
 export {
   assertValidComponent,
   getComponentReference,
   Component,
-  isComponent
+  isComponent,
+  isAnyComponent
 };

package/dist/{chunk-S6WICIDW.js → chunk-S6FOL2HY.js}

@@ -40,8 +40,8 @@ var Exit = class _Exit {
   is(exit) {
     return this.name === exit.name;
   }
-  parse(exit, value) {
-    return !!value && exit.name === this.name;
+  match(result) {
+    return result.exit instanceof _Exit && this.name === result.exit.name;
   }
   constructor(props) {
     if (!isValidIdentifier(props.name)) {