llmz 0.0.2 → 0.0.4

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

@@ -1,6 +1,3 @@
-import {
-  createJsxComponent
-} from "./chunk-7FIPXMZR.js";
 import {
   AssignmentError,
   CodeExecutionError,
@@ -8,19 +5,20 @@ import {
   Signals,
   SnapshotSignal,
   VMSignal
-} from "./chunk-RJHDQL4C.js";
+} from "./chunk-JKVVQN2P.js";
 import {
   cleanStackTrace
-} from "./chunk-MSTBEKOQ.js";
+} from "./chunk-JQBT7UWN.js";
+import {
+  createJsxComponent
+} from "./chunk-ORQP26SZ.js";
 import {
+  __commonJS,
+  __toESM,
   isFunction_default,
   mapValues_default,
   maxBy_default
-} from "./chunk-7S3UY52U.js";
-import {
-  __commonJS,
-  __toESM
-} from "./chunk-YSQDPG26.js";
+} from "./chunk-7WRN4E42.js";
 
 // ../../node_modules/.pnpm/source-map-js@1.2.1_patch_hash=h25dep36e76b3zca3v6s2554fi/node_modules/source-map-js/lib/base64.js
 var require_base64 = __commonJS({
@@ -2148,7 +2146,7 @@ var replaceCommentBabelPlugin = function({
                 return;
               }
               comments.sort((a, b) => a.start > b.start ? -1 : 1).forEach((comment) => {
-                var _a;
+                var _a3;
                 if (processed.has(comment.loc)) {
                   return;
                 }
@@ -2156,7 +2154,7 @@ var replaceCommentBabelPlugin = function({
                 const commentCall = t.expressionStatement(
                   t.callExpression(t.identifier(CommentFnIdentifier), [
                     t.stringLiteral(comment.value.trim()),
-                    t.numericLiteral(((_a = comment.loc) == null ? void 0 : _a.start.line) ?? 0)
+                    t.numericLiteral(((_a3 = comment.loc) == null ? void 0 : _a3.start.line) ?? 0)
                   ])
                 );
                 const isInsideObjectProperty = t.isObjectProperty(node) || path2.findParent((path3) => t.isObjectProperty(path3.node));
@@ -2183,8 +2181,8 @@ var instrumentLastLinePlugin = function({
   return {
     visitor: {
       FunctionDeclaration(path) {
-        var _a, _b;
-        if (((_a = path.node.id) == null ? void 0 : _a.name) !== "__fn__") {
+        var _a3, _b;
+        if (((_a3 = path.node.id) == null ? void 0 : _a3.name) !== "__fn__") {
           return;
         }
         const statements = path.node.body.body;
@@ -2270,7 +2268,7 @@ var toolCallTrackingPlugin = (calls = /* @__PURE__ */ new Map()) => function({})
         skip.clear();
       },
       CallExpression(path) {
-        var _a;
+        var _a3;
         if (skip.has(path.node) || path.findParent((p) => skip.has(p.node))) {
           return;
         }
@@ -2285,7 +2283,7 @@ var toolCallTrackingPlugin = (calls = /* @__PURE__ */ new Map()) => function({})
           (p) => p.isAssignmentExpression()
         );
         if (declaration) {
-          lval = (_a = declaration.get("declarations")[0]) == null ? void 0 : _a.get("id");
+          lval = (_a3 = declaration.get("declarations")[0]) == null ? void 0 : _a3.get("id");
         }
         if (assignment) {
           const left = assignment.get("left");
@@ -2614,8 +2612,10 @@ function compile(code) {
 
 // src/vm.ts
 var IS_NODE = typeof process !== "undefined" && process.versions != null && process.versions.node != null;
-var IS_CI = !!process.env.CI;
-var VM_DRIVER = process.env.VM_DRIVER ?? (IS_CI ? "node" : "isolated-vm");
+var _a;
+var IS_CI = typeof process !== "undefined" && !!((_a = process == null ? void 0 : process.env) == null ? void 0 : _a.CI);
+var _a2;
+var VM_DRIVER = (typeof process !== "undefined" && ((_a2 = process == null ? void 0 : process.env) == null ? void 0 : _a2.VM_DRIVER)) ?? (IS_CI ? "node" : "isolated-vm");
 var USE_ISOLATED_VM = IS_NODE && VM_DRIVER === "isolated-vm";
 var LINE_OFFSET = USE_ISOLATED_VM ? 3 : 1;
 var MAX_VM_EXECUTION_TIME = 6e4;
@@ -2648,7 +2648,7 @@ function getCompiledCode(code, traces = []) {
     throw new InvalidCodeError(err.message, code);
   }
 }
-async function runAsyncFunction(context, code, traces = [], signal = null) {
+async function runAsyncFunction(context, code, traces = [], signal = null, timeout = MAX_VM_EXECUTION_TIME) {
   const transformed = getCompiledCode(code, traces);
   const lines_executed = /* @__PURE__ */ new Map();
   const variables = {};
@@ -2692,12 +2692,12 @@ async function runAsyncFunction(context, code, traces = [], signal = null) {
   };
   let currentToolCall;
   context[Identifiers.ToolCallTrackerFnIdentifier] = (callId, type, outputOrError) => {
-    var _a;
+    var _a3;
     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: (_a = transformed.toolCalls.get(callId)) == null ? void 0 : _a.assignment
+        assignment: (_a3 = transformed.toolCalls.get(callId)) == null ? void 0 : _a3.assignment
       };
     }
   };
@@ -2947,7 +2947,7 @@ do {
     // TODO: fix getRemainingTimeInMillis ...
     // TODO: probably expose a "timeout" option instead
     // timeout: clamp(Runtime.getRemainingTimeInMillis() - 10_000, 5_000, MAX_VM_EXECUTION_TIME),
-    timeout: MAX_VM_EXECUTION_TIME,
+    timeout,
     copy: true,
     promise: true
   }).then((res) => {
@@ -2990,7 +2990,7 @@ do {
   return final;
 }
 var handleError = (err, code, consumer, traces, variables, lines_executed, currentToolCall) => {
-  var _a, _b, _c;
+  var _a3, _b, _c;
   err = Signals.maybeDeserializeError(err);
   const lines = code.split("\n");
   const stackTrace = err.stack || "";
@@ -3012,7 +3012,7 @@ var handleError = (err, code, consumer, traces, variables, lines_executed, curre
       column: Math.min(minColumn, Number(x[2]))
     };
   });
-  const lastLine = ((_a = maxBy_default(matches, (x) => x.line ?? 0)) == null ? void 0 : _a.line) ?? 0;
+  const lastLine = ((_a3 = maxBy_default(matches, (x) => x.line ?? 0)) == null ? void 0 : _a3.line) ?? 0;
   let debugUserCode = "";
   let truncatedCode = "";
   let truncated = false;

package/dist/{chunk-UG2GCPHF.js → chunk-4KB5WXHR.js}

@@ -1,7 +1,7 @@
 import {
   extractType,
   inspect
-} from "./chunk-Q4DEJGXD.js";
+} from "./chunk-EFGXTO64.js";
 
 // src/snapshots.ts
 import { ulid } from "ulid";
@@ -46,7 +46,8 @@ var Snapshot = class _Snapshot {
       reason: this.reason,
       stack: this.stack,
       variables: this.variables,
-      toolCall: this.toolCall
+      toolCall: this.toolCall,
+      status: this.#status
     };
   }
   static fromJSON(json) {

package/dist/{chunk-YPOOKE3A.cjs → chunk-4QTK5ALH.cjs}

@@ -1,15 +1,15 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } var _class;
 
-var _chunkYUDUY7XWcjs = require('./chunk-YUDUY7XW.cjs');
+var _chunkHYVKY3W6cjs = require('./chunk-HYVKY3W6.cjs');
 
 
 
 
-var _chunkENFB45AIcjs = require('./chunk-ENFB45AI.cjs');
+var _chunkP7J2WCBBcjs = require('./chunk-P7J2WCBB.cjs');
 
 
 
-var _chunk3CGLDS5Tcjs = require('./chunk-3CGLDS5T.cjs');
+var _chunkUQOBUJIQcjs = require('./chunk-UQOBUJIQ.cjs');
 
 // src/tool.ts
 var _zui = require('@bpinternal/zui');
@@ -21,6 +21,8 @@ var Tool = (_class = class _Tool {
   
   
   
+  
+  __init2() {this.MAX_RETRIES = 1e3}
   setStaticInputValues(values) {
     if (values === null || values === void 0) {
       this._staticInputValues = void 0;
@@ -42,8 +44,8 @@ var Tool = (_class = class _Tool {
   }
   get zInput() {
     let input = this.input ? _zui.z.fromJsonSchema(this.input) : _zui.z.any();
-    if (!_chunk3CGLDS5Tcjs.isEmpty_default.call(void 0, this._staticInputValues)) {
-      const inputExtensions = _chunkENFB45AIcjs.convertObjectToZuiLiterals.call(void 0, this._staticInputValues);
+    if (!_chunkUQOBUJIQcjs.isEmpty_default.call(void 0, this._staticInputValues)) {
+      const inputExtensions = _chunkP7J2WCBBcjs.convertObjectToZuiLiterals.call(void 0, this._staticInputValues);
       if (input instanceof _zui.z.ZodObject) {
         input = input.extend(inputExtensions);
       } else if (input instanceof _zui.z.ZodArray) {
@@ -59,33 +61,37 @@ var Tool = (_class = class _Tool {
   }
   rename(name) {
     const before = this.name;
-    if (!_chunkENFB45AIcjs.isValidIdentifier.call(void 0, name)) {
+    if (!_chunkP7J2WCBBcjs.isValidIdentifier.call(void 0, name)) {
       throw new Error(
         `Invalid name for tool ${name}. A tool name must start with a letter and contain only letters, numbers, and underscores. It must be 1-50 characters long.`
       );
     }
     this.name = name;
-    this.aliases = _chunk3CGLDS5Tcjs.uniq_default.call(void 0, [name, ...this.aliases.map((alias) => alias === before ? name : alias)]);
+    this.aliases = _chunkUQOBUJIQcjs.uniq_default.call(void 0, [name, ...this.aliases.map((alias) => alias === before ? name : alias)]);
     return this;
   }
-  clone() {
+  clone(props = {}) {
+    var _a, _b;
     try {
+      const zInput = this.input ? _zui.z.fromJsonSchema(this.input) : void 0;
+      const zOutput = this.output ? _zui.z.fromJsonSchema(this.output) : void 0;
       return new _Tool({
-        name: this.name,
-        aliases: [...this.aliases],
-        description: this.description,
-        metadata: JSON.parse(JSON.stringify(this.metadata)),
-        input: this.input ? _zui.z.fromJsonSchema(this.input) : void 0,
-        output: this.output ? _zui.z.fromJsonSchema(this.output) : void 0,
-        handler: this._handler
-      }).setStaticInputValues(this._staticInputValues);
+        name: _nullishCoalesce(props.name, () => ( this.name)),
+        aliases: _nullishCoalesce(props.aliases, () => ( [...this.aliases])),
+        description: _nullishCoalesce(props.description, () => ( this.description)),
+        metadata: JSON.parse(JSON.stringify(_nullishCoalesce(props.metadata, () => ( this.metadata)))),
+        input: typeof props.input === "function" ? (_a = props.input) == null ? void 0 : _a.call(props, zInput) : props.input instanceof _zui.ZodType ? props.input : zInput,
+        output: typeof props.output === "function" ? (_b = props.output) == null ? void 0 : _b.call(props, zOutput) : props.output instanceof _zui.ZodType ? props.output : zOutput,
+        handler: _nullishCoalesce(props.handler, () => ( this._handler)),
+        retry: _nullishCoalesce(props.retry, () => ( this.retry))
+      }).setStaticInputValues(_nullishCoalesce(props.staticInputValues, () => ( this._staticInputValues)));
     } catch (e) {
       throw new Error(`Failed to clone tool "${this.name}": ${e}`);
     }
   }
   
-  constructor(props) {;_class.prototype.__init.call(this);
-    if (!_chunkENFB45AIcjs.isValidIdentifier.call(void 0, props.name)) {
+  constructor(props) {;_class.prototype.__init.call(this);_class.prototype.__init2.call(this);
+    if (!_chunkP7J2WCBBcjs.isValidIdentifier.call(void 0, props.name)) {
       throw new Error(
         `Invalid name for tool ${props.name}. A tool name must start with a letter and contain only letters, numbers, and underscores. It must be 1-50 characters long.`
       );
@@ -110,13 +116,13 @@ var Tool = (_class = class _Tool {
         `Invalid aliases for tool ${props.name}. Expected an array, but got type "${typeof props.aliases}"`
       );
     }
-    if (props.aliases && props.aliases.some((alias) => !_chunkENFB45AIcjs.isValidIdentifier.call(void 0, alias))) {
+    if (props.aliases && props.aliases.some((alias) => !_chunkP7J2WCBBcjs.isValidIdentifier.call(void 0, alias))) {
       throw new Error(`Invalid aliases for tool ${props.name}. Expected an array of valid identifiers.`);
     }
     if (typeof props.input !== "undefined") {
       if (props.input && "toJsonSchema" in props.input && typeof props.input.toJsonSchema === "function") {
         this.input = props.input.toJsonSchema();
-      } else if (_chunkENFB45AIcjs.isJsonSchema.call(void 0, props.input)) {
+      } else if (_chunkP7J2WCBBcjs.isJsonSchema.call(void 0, props.input)) {
         this.input = props.input;
       } else {
         throw new Error(
@@ -127,7 +133,7 @@ var Tool = (_class = class _Tool {
     if (typeof props.output !== "undefined") {
       if (props.output && "toJsonSchema" in props.output && typeof props.output.toJsonSchema === "function") {
         this.output = props.output.toJsonSchema();
-      } else if (_chunkENFB45AIcjs.isJsonSchema.call(void 0, props.output)) {
+      } else if (_chunkP7J2WCBBcjs.isJsonSchema.call(void 0, props.output)) {
         this.output = props.output;
       } else {
         throw new Error(
@@ -136,26 +142,51 @@ var Tool = (_class = class _Tool {
       }
     }
     this.name = props.name;
-    this.aliases = _chunk3CGLDS5Tcjs.uniq_default.call(void 0, [props.name, ..._nullishCoalesce(props.aliases, () => ( []))]);
+    this.aliases = _chunkUQOBUJIQcjs.uniq_default.call(void 0, [props.name, ..._nullishCoalesce(props.aliases, () => ( []))]);
     this.description = props.description;
     this.metadata = _nullishCoalesce(props.metadata, () => ( {}));
     this._handler = props.handler;
     this.setStaticInputValues(props.staticInputValues);
+    this.retry = props.retry;
   }
-  async execute(input) {
+  async execute(input, ctx) {
+    var _a;
     const pInput = this.zInput.safeParse(input);
     if (!pInput.success) {
       throw new Error(`Tool "${this.name}" received invalid input: ${pInput.error.message}`);
     }
-    const result = await this._handler(pInput.data);
-    const pOutput = this.zOutput.safeParse(result);
-    return pOutput.success ? pOutput.data : result;
+    let attempt = 0;
+    while (attempt < this.MAX_RETRIES) {
+      try {
+        const result = await this._handler(pInput.data, ctx);
+        const pOutput = this.zOutput.safeParse(result);
+        return pOutput.success ? pOutput.data : result;
+      } catch (err) {
+        const shouldRetry = await ((_a = this.retry) == null ? void 0 : _a.call(this, {
+          input: pInput.data,
+          attempt: ++attempt,
+          error: err
+        }));
+        if (!shouldRetry) {
+          throw err;
+        }
+      }
+    }
+    throw new Error(
+      `Tool "${this.name}" failed after ${this.MAX_RETRIES} attempts. Last error: ${JSON.stringify(input)}`
+    );
   }
   async getTypings() {
-    const input = this.input ? _zui.z.fromJsonSchema(this.input) : void 0;
+    let input = this.input ? _zui.z.fromJsonSchema(this.input) : void 0;
     const output = this.output ? _zui.z.fromJsonSchema(this.output) : _zui.z.void();
+    if ((input == null ? void 0 : input.naked()) instanceof _zui.ZodObject && typeof this._staticInputValues === "object" && !_chunkUQOBUJIQcjs.isEmpty_default.call(void 0, this._staticInputValues)) {
+      const inputExtensions = _chunkP7J2WCBBcjs.convertObjectToZuiLiterals.call(void 0, this._staticInputValues);
+      input = input.extend(inputExtensions);
+    } else if (this._staticInputValues !== void 0) {
+      input = _chunkP7J2WCBBcjs.convertObjectToZuiLiterals.call(void 0, this._staticInputValues);
+    }
     const fnType = _zui.z.function(input, _zui.z.promise(output)).title(this.name).describe(_nullishCoalesce(this.description, () => ( "")));
-    return _chunkYUDUY7XWcjs.getTypings.call(void 0, fnType, {
+    return _chunkHYVKY3W6cjs.getTypings.call(void 0, fnType, {
       declaration: true
     });
   }