llmz 0.0.2 → 0.0.5

package/dist/tool.d.ts

@@ -1,6 +1,17 @@
-import { type JSONSchema } from '@bpinternal/zui';
+import { type JSONSchema, TypeOf } from '@bpinternal/zui';
 import { ZuiType } from './types.js';
-export declare class Tool<I = unknown, O = unknown> {
+type ToolRetryInput<I> = {
+    input: I;
+    attempt: number;
+    error?: unknown;
+};
+export type ToolRetryFn<I> = (args: ToolRetryInput<I>) => boolean | Promise<boolean>;
+type IsObject<T> = T extends object ? (T extends Function ? false : true) : false;
+type SmartPartial<T> = IsObject<T> extends true ? Partial<T> : T;
+type ToolCallContext = {
+    callId: string;
+};
+export declare class Tool<I extends ZuiType = ZuiType, O extends ZuiType = ZuiType> {
     private _staticInputValues?;
     name: string;
     aliases: string[];
@@ -8,23 +19,37 @@ export declare class Tool<I = unknown, O = unknown> {
     metadata: Record<string, unknown>;
     input?: JSONSchema;
     output?: JSONSchema;
-    setStaticInputValues(values: Partial<I>): this;
+    retry?: ToolRetryFn<TypeOf<I>>;
+    MAX_RETRIES: number;
+    setStaticInputValues(values: SmartPartial<TypeOf<I>>): this;
     get zInput(): import("@bpinternal/zui").ZodTypeAny;
     get zOutput(): import("@bpinternal/zui").ZodTypeAny | import("@bpinternal/zui").ZodVoid;
     rename(name: string): this;
-    clone(): Tool<I, O>;
+    clone<IX extends ZuiType = I, OX extends ZuiType = O>(props?: Partial<{
+        name: string;
+        aliases?: string[];
+        description?: string;
+        metadata?: Record<string, unknown>;
+        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>>;
+        retry: ToolRetryFn<TypeOf<IX>>;
+    }>): Tool<IX, OX>;
     private _handler;
     constructor(props: {
         name: string;
         aliases?: string[];
         description?: string;
         metadata?: Record<string, unknown>;
-        input?: ZuiType<I>;
-        output?: ZuiType<O>;
-        staticInputValues?: Partial<I>;
-        handler: (args: I) => Promise<O>;
+        input?: I;
+        output?: O;
+        staticInputValues?: Partial<TypeOf<I>>;
+        handler: (args: TypeOf<I>, ctx: ToolCallContext) => Promise<TypeOf<O>>;
+        retry?: ToolRetryFn<TypeOf<I>>;
     });
-    execute(input: I): Promise<O>;
+    execute(input: TypeOf<I>, ctx: ToolCallContext): Promise<TypeOf<O>>;
     getTypings(): Promise<string>;
     static withUniqueNames: (tools: Tool<any, any>[]) => Tool<any, any>[];
 }
+export {};

package/dist/truncator-NYN7BGKJ.cjs

@@ -0,0 +1,10 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+
+var _chunkFIVFS4HGcjs = require('./chunk-FIVFS4HG.cjs');
+require('./chunk-P7J2WCBB.cjs');
+require('./chunk-UQOBUJIQ.cjs');
+
+
+
+exports.truncateWrappedContent = _chunkFIVFS4HGcjs.truncateWrappedContent; exports.wrapContent = _chunkFIVFS4HGcjs.wrapContent;

package/dist/truncator-WAL2GCUY.js

@@ -0,0 +1,10 @@
+import {
+  truncateWrappedContent,
+  wrapContent
+} from "./chunk-QT4QF3YA.js";
+import "./chunk-5TRUJES5.js";
+import "./chunk-7WRN4E42.js";
+export {
+  truncateWrappedContent,
+  wrapContent
+};

package/dist/types.d.ts

@@ -1,5 +1,6 @@
 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';
 export declare namespace Traces {
     type Comment = TraceTemplate<'comment', {
@@ -13,6 +14,7 @@ export declare namespace Traces {
     type ToolCall = TraceTemplate<'tool_call', {
         success: true;
         tool_name: string;
+        tool_call_id: string;
         object?: string;
         input: any;
         output: any;
@@ -20,6 +22,7 @@ export declare namespace Traces {
     } | {
         success: false;
         tool_name: string;
+        tool_call_id: string;
         object?: string;
         input: any;
         error: any;
@@ -28,6 +31,7 @@ export declare namespace Traces {
     type ToolSlow = TraceTemplate<'tool_slow', {
         tool_name: string;
         object?: string;
+        tool_call_id: string;
         input: any;
         duration: number;
     }>;
@@ -47,6 +51,9 @@ export declare namespace Traces {
         message: string;
         stackTrace: string;
     }>;
+    type LLMCallStart = TraceTemplate<'llm_call_started', {
+        model: string;
+    }>;
     type LLMCallSuccess = TraceTemplate<'llm_call_success', {
         model: string;
         code: string;
@@ -66,7 +73,7 @@ export declare namespace Traces {
         started_at: number;
         ended_at?: number;
     } & Content;
-    type Trace = Comment | Log | ToolCall | ToolSlow | PropertyMutation | YieldTrace | LLMCallSuccess | ThinkSignal | CodeExecution | CodeExecutionException | InvalidCodeExceptionTrace | AbortTrace;
+    type Trace = Comment | Log | ToolCall | ToolSlow | PropertyMutation | YieldTrace | LLMCallStart | LLMCallSuccess | ThinkSignal | CodeExecution | CodeExecutionException | InvalidCodeExceptionTrace | AbortTrace;
 }
 export type Trace = Traces.Trace;
 export type VMExecutionResult = {
@@ -98,6 +105,7 @@ export type SuccessExecutionResult = {
     status: 'success';
     iterations: Iteration[];
     context: Context;
+    result: ExitResult;
 };
 export type PartialExecutionResult = {
     status: 'interrupted';

package/dist/typings-AIV2OASX.js

@@ -0,0 +1,10 @@
+import {
+  getTypings
+} from "./chunk-UNMKB2DA.js";
+import "./chunk-JKVVQN2P.js";
+import "./chunk-JQBT7UWN.js";
+import "./chunk-5TRUJES5.js";
+import "./chunk-7WRN4E42.js";
+export {
+  getTypings
+};

package/dist/typings-XR6CYHW4.cjs

@@ -0,0 +1,10 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunkHYVKY3W6cjs = require('./chunk-HYVKY3W6.cjs');
+require('./chunk-JDABP4SD.cjs');
+require('./chunk-IKSIOIIP.cjs');
+require('./chunk-P7J2WCBB.cjs');
+require('./chunk-UQOBUJIQ.cjs');
+
+
+exports.getTypings = _chunkHYVKY3W6cjs.getTypings;

package/dist/{utils-TS23YJPE.js → utils-L5GLCS3C.js}

@@ -15,9 +15,8 @@ import {
   toPropertyKey,
   toValidFunctionName,
   toValidObjectName
-} from "./chunk-DFZ6GX5C.js";
-import "./chunk-7S3UY52U.js";
-import "./chunk-YSQDPG26.js";
+} from "./chunk-5TRUJES5.js";
+import "./chunk-7WRN4E42.js";
 export {
   Identifier,
   Tokens,

package/dist/utils-QC4I2L6R.cjs

@@ -0,0 +1,37 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+var _chunkP7J2WCBBcjs = require('./chunk-P7J2WCBB.cjs');
+require('./chunk-UQOBUJIQ.cjs');
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+exports.Identifier = _chunkP7J2WCBBcjs.Identifier; exports.Tokens = _chunkP7J2WCBBcjs.Tokens; exports.awaitObject = _chunkP7J2WCBBcjs.awaitObject; exports.convertObjectToZuiLiterals = _chunkP7J2WCBBcjs.convertObjectToZuiLiterals; exports.escapeString = _chunkP7J2WCBBcjs.escapeString; exports.getMultilineComment = _chunkP7J2WCBBcjs.getMultilineComment; exports.getTokenizer = _chunkP7J2WCBBcjs.getTokenizer; exports.init = _chunkP7J2WCBBcjs.init; exports.isJsonSchema = _chunkP7J2WCBBcjs.isJsonSchema; exports.isValidIdentifier = _chunkP7J2WCBBcjs.isValidIdentifier; exports.isValidMessageName = _chunkP7J2WCBBcjs.isValidMessageName; exports.isValidSchema = _chunkP7J2WCBBcjs.isValidSchema; exports.stripInvalidIdentifiers = _chunkP7J2WCBBcjs.stripInvalidIdentifiers; exports.toPropertyKey = _chunkP7J2WCBBcjs.toPropertyKey; exports.toValidFunctionName = _chunkP7J2WCBBcjs.toValidFunctionName; exports.toValidObjectName = _chunkP7J2WCBBcjs.toValidObjectName;

package/dist/vm-GJ5R72AP.cjs

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

package/dist/vm-ZUQOSRRM.js

@@ -0,0 +1,12 @@
+import {
+  USE_ISOLATED_VM,
+  runAsyncFunction
+} from "./chunk-4EWY34YA.js";
+import "./chunk-JKVVQN2P.js";
+import "./chunk-JQBT7UWN.js";
+import "./chunk-ORQP26SZ.js";
+import "./chunk-7WRN4E42.js";
+export {
+  USE_ISOLATED_VM,
+  runAsyncFunction
+};

package/dist/vm.d.ts

@@ -1,3 +1,3 @@
 import { Trace, VMExecutionResult } from './types.js';
 export declare const USE_ISOLATED_VM: boolean;
-export declare function runAsyncFunction(context: any, code: string, traces?: Trace[], signal?: AbortSignal | null): Promise<VMExecutionResult>;
+export declare function runAsyncFunction(context: any, code: string, traces?: Trace[], signal?: AbortSignal | null, timeout?: number): Promise<VMExecutionResult>;

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.2",
+  "version": "0.0.5",
   "types": "./dist/index.d.ts",
   "exports": {
     ".": {
@@ -11,12 +11,13 @@
     }
   },
   "scripts": {
-    "build": "pnpm build:markdown && tsup && pnpm tsc --emitDeclarationOnly",
+    "generate": "pnpm build:markdown",
+    "build": "tsup && pnpm tsc --emitDeclarationOnly",
     "watch": "tsup --watch",
     "test": "vitest run --config vitest.config.ts",
     "test:update": "vitest -u run --config vitest.config.ts",
     "test:watch": "vitest --config vitest.config.ts",
-    "build:markdown": "tsx ./scripts/compile-markdown.mts"
+    "build:markdown": "node --loader ts-node/esm ./scripts/compile-markdown.mts"
   },
   "keywords": [],
   "author": "",
@@ -30,6 +31,7 @@
     "@babel/standalone": "^7.26.4",
     "@babel/traverse": "^7.26.4",
     "@babel/types": "^7.26.3",
+    "@botpress/client": "1.15.1",
     "bytes": "^3.1.2",
     "exponential-backoff": "^3.1.1",
     "handlebars": "^4.7.8",
@@ -41,7 +43,6 @@
     "ulid": "^2.3.0"
   },
   "devDependencies": {
-    "@botpress/client": "1.12.0",
     "@microsoft/api-extractor": "^7.49.0",
     "@types/babel__core": "^7.20.5",
     "@types/babel__generator": "^7.6.8",
@@ -51,6 +52,7 @@
     "@types/lodash-es": "^4.17.12",
     "@types/ms": "^0.7.34",
     "@types/node": "^18.19.67",
+    "chalk": "^4.1.2",
     "diff": "^8.0.1",
     "dotenv": "^16.4.4",
     "esbuild": "^0.16.12",
@@ -61,9 +63,9 @@
     "tsx": "^4.19.2"
   },
   "peerDependencies": {
-    "@botpress/cognitive": "^0.1.20",
+    "@botpress/cognitive": "0.1.23",
     "@bpinternal/thicktoken": "^1.0.5",
-    "@bpinternal/zui": "^0.22.4"
+    "@bpinternal/zui": "^0.22.5"
   },
   "dependenciesMeta": {
     "@bpinternal/zui": {

package/dist/chunk-A7BDFLAE.cjs

@@ -1,30 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var __create = Object.create;
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __getProtoOf = Object.getPrototypeOf;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __commonJS = (cb, mod) => function __require() {
-  return mod || (0, cb[__getOwnPropNames(cb)[0]])((mod = { exports: {} }).exports, mod), mod.exports;
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
-  // If the importer is in node compatibility mode or this is not an ESM
-  // file that has been converted to a CommonJS file using a Babel-
-  // compatible transform (i.e. "__esModule" has not been set), then set
-  // "default" to the CommonJS "module.exports" for node compatibility.
-  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
-  mod
-));
-
-
-
-
-exports.__commonJS = __commonJS; exports.__toESM = __toESM;

package/dist/chunk-C2TSWGYC.cjs

@@ -1,91 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } }
-
-
-var _chunkQZGFCB4Tcjs = require('./chunk-QZGFCB4T.cjs');
-
-// src/snapshots.ts
-var _ulid = require('ulid');
-var MAX_SNAPSHOT_SIZE_BYTES = 4e3;
-var Snapshot = class _Snapshot {
-  
-  
-  
-  
-  
-  #status;
-  get status() {
-    return Object.freeze({ ...this.#status });
-  }
-  constructor(props) {
-    this.id = props.id;
-    this.stack = props.stack;
-    this.reason = props.reason;
-    this.variables = props.variables;
-    this.toolCall = props.toolCall;
-    this.#status = props.status;
-  }
-  static fromSignal(signal) {
-    const variables = Object.entries(signal.variables).map(([name, value]) => {
-      const type = _chunkQZGFCB4Tcjs.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(_chunkQZGFCB4Tcjs.inspect.call(void 0, value, name), () => ( "N/A")) } : { name, type, bytes, truncated: false, value };
-    });
-    return new _Snapshot({
-      id: "snapshot_" + _ulid.ulid.call(void 0, ),
-      reason: signal.message,
-      stack: signal.truncatedCode,
-      variables,
-      toolCall: signal.toolCall,
-      status: { type: "pending" }
-    });
-  }
-  toJSON() {
-    return {
-      id: this.id,
-      reason: this.reason,
-      stack: this.stack,
-      variables: this.variables,
-      toolCall: this.toolCall
-    };
-  }
-  static fromJSON(json) {
-    return new _Snapshot({
-      id: json.id,
-      reason: json.reason,
-      stack: json.stack,
-      variables: json.variables,
-      toolCall: json.toolCall,
-      status: json.status
-    });
-  }
-  clone() {
-    return new _Snapshot({
-      id: this.id,
-      reason: this.reason,
-      stack: this.stack,
-      variables: this.variables,
-      toolCall: this.toolCall,
-      status: this.#status
-    });
-  }
-  reset() {
-    this.#status = { type: "pending" };
-  }
-  resolve(value) {
-    if (this.#status.type !== "pending") {
-      throw new Error(`Cannot resolve snapshot because it is already settled: ${this.#status.type}`);
-    }
-    this.#status = { type: "resolved", value };
-  }
-  reject(error) {
-    if (this.#status.type !== "pending") {
-      throw new Error(`Cannot reject snapshot because it is already settled: ${this.#status.type}`);
-    }
-    this.#status = { type: "rejected", error };
-  }
-};
-
-
-
-exports.Snapshot = Snapshot;