llmz 0.0.6 → 0.0.8

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

@@ -2,22 +2,22 @@
 
 
 
-var _chunk6WT5VZBGcjs = require('./chunk-6WT5VZBG.cjs');
+var _chunkSHJDRZF5cjs = require('./chunk-SHJDRZF5.cjs');
 
 
-var _chunk4QTK5ALHcjs = require('./chunk-4QTK5ALH.cjs');
+var _chunkC6WNNTEVcjs = require('./chunk-C6WNNTEV.cjs');
 
 
 var _chunkJDABP4SDcjs = require('./chunk-JDABP4SD.cjs');
 
 
-var _chunk4I7UPBLNcjs = require('./chunk-4I7UPBLN.cjs');
+var _chunkGWFYZDURcjs = require('./chunk-GWFYZDUR.cjs');
 
 
-var _chunkEJRCDWBAcjs = require('./chunk-EJRCDWBA.cjs');
+var _chunkZRCU35UVcjs = require('./chunk-ZRCU35UV.cjs');
 
 
-var _chunkFIVFS4HGcjs = require('./chunk-FIVFS4HG.cjs');
+var _chunkJGVAZO4Xcjs = require('./chunk-JGVAZO4X.cjs');
 
 
 
@@ -46,10 +46,10 @@ var Snapshot = class _Snapshot {
   }
   static fromSignal(signal) {
     const variables = Object.entries(signal.variables).map(([name, value]) => {
-      const type = _chunk6WT5VZBGcjs.extractType.call(void 0, value);
+      const type = _chunkSHJDRZF5cjs.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(_chunk6WT5VZBGcjs.inspect.call(void 0, value, name), () => ( "N/A")) } : { name, type, bytes, truncated: false, value };
+      return truncated ? { name, type, bytes, truncated: true, preview: _nullishCoalesce(_chunkSHJDRZF5cjs.inspect.call(void 0, value, name), () => ( "N/A")) } : { name, type, bytes, truncated: false, value };
     });
     return new _Snapshot({
       id: "snapshot_" + _ulid.ulid.call(void 0, ),
@@ -107,6 +107,77 @@ var Snapshot = class _Snapshot {
   }
 };
 
+// src/transcript.ts
+var MAX_MESSAGE_LENGTH = 5e3;
+function getMessagePreview(message) {
+  if (message.role === "assistant" || message.role === "user" || message.role === "summary") {
+    return message.content;
+  }
+  if (message.role === "event") {
+    return _chunkSHJDRZF5cjs.inspect.call(void 0, message.payload, message.name, { tokens: 1e3 });
+  }
+  return _chunkSHJDRZF5cjs.inspect.call(void 0, message, void 0, { tokens: 1e3 });
+}
+function getMessageType(message) {
+  if (message.role === "assistant" || message.role === "user" || message.role === "summary") {
+    return message.role;
+  }
+  if (message.role === "event") {
+    return `event:${message.name}`;
+  }
+  return "unknown";
+}
+var TranscriptArray = class extends Array {
+  constructor(items = []) {
+    items = Array.isArray(items) ? items : [];
+    super(...items);
+    items.forEach((item) => {
+      if (!["user", "assistant"].includes(item.role)) {
+        throw new Error(`Invalid role "${item.role}" in transcript message`);
+      }
+      if ("name" in item && item.name && typeof item.name !== "string") {
+        throw new Error(`Invalid name for transcript message. Expected a string, but got type "${typeof item.name}"`);
+      }
+      if ("content" in item && typeof item.content !== "string") {
+        throw new Error(
+          `Invalid content for transcript message. Expected a string, but got type "${typeof item.content}"`
+        );
+      }
+    });
+    Object.setPrototypeOf(this, new.target.prototype);
+  }
+  toString() {
+    if (!this.length) {
+      return "";
+    }
+    return this.map((item, idx) => {
+      var _a, _b;
+      const msgIdx = getMessageType(item) + "-" + String(idx + 1).padStart(3, "0");
+      let preview = getMessagePreview(item);
+      if (preview.length > MAX_MESSAGE_LENGTH) {
+        preview = preview.slice(0, MAX_MESSAGE_LENGTH) + "\n... (truncated)";
+      }
+      const alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
+      if ((item.role === "event" || item.role === "user") && ((_a = item.attachments) == null ? void 0 : _a.length)) {
+        for (let i = 0; i < item.attachments.length; i++) {
+          const attachmentIdx = alphabet[i % alphabet.length];
+          preview += `
+[Attachment ${msgIdx}-${attachmentIdx}]`;
+        }
+      }
+      const tags = [];
+      tags.push({ key: "role", value: item.role });
+      if ("name" in item && ((_b = item.name) == null ? void 0 : _b.length)) {
+        tags.push({ key: "name", value: item.name });
+      }
+      const tagsString = tags.map(({ key, value }) => `${key}="${value}"`).join(" ");
+      return `<${msgIdx} ${tagsString}>
+${preview.trim()}
+</${msgIdx}>`;
+    }).join("\n");
+  }
+};
+
 // src/result.ts
 var ExecutionResult = class {
   
@@ -221,59 +292,16 @@ var HookedArray = class extends Array {
   }
 };
 
-// src/transcript.ts
-var MAX_MESSAGE_LENGTH = 5e3;
-var TranscriptArray = class extends Array {
-  constructor(items = []) {
-    items = Array.isArray(items) ? items : [];
-    super(...items);
-    items.forEach((item) => {
-      if (!["user", "assistant"].includes(item.role)) {
-        throw new Error(`Invalid role "${item.role}" in transcript message`);
-      }
-      if (item.name && typeof item.name !== "string") {
-        throw new Error(`Invalid name for transcript message. Expected a string, but got type "${typeof item.name}"`);
-      }
-      if (typeof item.content !== "string") {
-        throw new Error(
-          `Invalid content for transcript message. Expected a string, but got type "${typeof item.content}"`
-        );
-      }
-    });
-    Object.setPrototypeOf(this, new.target.prototype);
-  }
-  toString() {
-    if (!this.length) {
-      return "";
-    }
-    return this.map((item, idx) => {
-      var _a;
-      const msgIdx = "message-" + String(idx + 1).padStart(3, "0");
-      const shouldTruncate = item.content.length > MAX_MESSAGE_LENGTH;
-      const preview = shouldTruncate ? item.content.slice(0, MAX_MESSAGE_LENGTH) + "\n... (truncated)" : item.content;
-      const tags = [];
-      tags.push({ key: "role", value: item.role });
-      if ((_a = item.name) == null ? void 0 : _a.length) {
-        tags.push({ key: "name", value: item.name });
-      }
-      const tagsString = tags.map(({ key, value }) => `${key}="${value}"`).join(" ");
-      return `<${msgIdx} ${tagsString}>
-${preview.trim()}
-</${msgIdx}>`;
-    }).join("\n");
-  }
-};
-
 // src/context.ts
-var ThinkExit = new (0, _chunk4I7UPBLNcjs.Exit)({
+var ThinkExit = new (0, _chunkGWFYZDURcjs.Exit)({
   name: "think",
   description: "Think about the current situation and provide a response"
 });
-var ListenExit = new (0, _chunk4I7UPBLNcjs.Exit)({
+var ListenExit = new (0, _chunkGWFYZDURcjs.Exit)({
   name: "listen",
   description: "Listen to the user and provide a response"
 });
-var DefaultExit = new (0, _chunk4I7UPBLNcjs.Exit)({
+var DefaultExit = new (0, _chunkGWFYZDURcjs.Exit)({
   name: "done",
   description: "When the execution is sucessfully completed or when error recovery is not possible",
   schema: _zui.z.discriminatedUnion("success", [
@@ -395,7 +423,7 @@ var Context = (_class2 = class {
   
   
   
-  __init3() {this.version = _chunk6WT5VZBGcjs.DualModePrompt}
+  __init3() {this.version = _chunkSHJDRZF5cjs.DualModePrompt}
   __init4() {this.timeout = 6e4}
   // Default timeout of 60 seconds
   
@@ -510,7 +538,7 @@ var Context = (_class2 = class {
         ...lastIterationMessages,
         {
           role: "assistant",
-          content: _chunkFIVFS4HGcjs.wrapContent.call(void 0, _nullishCoalesce(((_c = lastIteration.llm) == null ? void 0 : _c.output), () => ( "")), { preserve: "top", flex: 4, minTokens: 25 })
+          content: _chunkJGVAZO4Xcjs.wrapContent.call(void 0, _nullishCoalesce(((_c = lastIteration.llm) == null ? void 0 : _c.output), () => ( "")), { preserve: "top", flex: 4, minTokens: 25 })
         },
         await this.version.getThinkingMessage({
           reason: lastIteration.status.thinking_requested.reason,
@@ -523,7 +551,7 @@ var Context = (_class2 = class {
         ...lastIterationMessages,
         {
           role: "assistant",
-          content: _chunkFIVFS4HGcjs.wrapContent.call(void 0, _nullishCoalesce(((_d = lastIteration.llm) == null ? void 0 : _d.output), () => ( "")), { preserve: "top", flex: 4, minTokens: 25 })
+          content: _chunkJGVAZO4Xcjs.wrapContent.call(void 0, _nullishCoalesce(((_d = lastIteration.llm) == null ? void 0 : _d.output), () => ( "")), { preserve: "top", flex: 4, minTokens: 25 })
         },
         await this.version.getInvalidCodeMessage({
           code: _nullishCoalesce(lastIteration.code, () => ( "// No code generated")),
@@ -536,7 +564,7 @@ var Context = (_class2 = class {
         ...lastIterationMessages,
         {
           role: "assistant",
-          content: _chunkFIVFS4HGcjs.wrapContent.call(void 0, _nullishCoalesce(((_e = lastIteration.llm) == null ? void 0 : _e.output), () => ( "")), { preserve: "top", flex: 4, minTokens: 25 })
+          content: _chunkJGVAZO4Xcjs.wrapContent.call(void 0, _nullishCoalesce(((_e = lastIteration.llm) == null ? void 0 : _e.output), () => ( "")), { preserve: "top", flex: 4, minTokens: 25 })
         },
         await this.version.getInvalidCodeMessage({
           code: _nullishCoalesce(lastIteration.code, () => ( "// No code generated")),
@@ -549,7 +577,7 @@ var Context = (_class2 = class {
         ...lastIterationMessages,
         {
           role: "assistant",
-          content: _chunkFIVFS4HGcjs.wrapContent.call(void 0, _nullishCoalesce(((_f = lastIteration.llm) == null ? void 0 : _f.output), () => ( "")), { preserve: "top", flex: 4, minTokens: 25 })
+          content: _chunkJGVAZO4Xcjs.wrapContent.call(void 0, _nullishCoalesce(((_f = lastIteration.llm) == null ? void 0 : _f.output), () => ( "")), { preserve: "top", flex: 4, minTokens: 25 })
         },
         await this.version.getCodeExecutionErrorMessage({
           message: lastIteration.status.execution_error.message,
@@ -565,7 +593,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 = _chunk4QTK5ALHcjs.Tool.withUniqueNames(await _asyncNullishCoalesce(await getValue(this.tools, this), async () => ( [])));
+    const tools = _chunkC6WNNTEVcjs.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);
@@ -576,7 +604,7 @@ var Context = (_class2 = class {
       throw new Error("Too many tools. Expected at most 100 tools.");
     }
     for (const component of components) {
-      _chunkEJRCDWBAcjs.assertValidComponent.call(void 0, component.definition);
+      _chunkZRCU35UVcjs.assertValidComponent.call(void 0, component.definition);
     }
     const ReservedToolNames = [
       "think",
@@ -595,7 +623,7 @@ var Context = (_class2 = class {
       "boolean",
       "array"
     ];
-    const MessageTool = this.chat && components.length ? new (0, _chunk4QTK5ALHcjs.Tool)({
+    const MessageTool = this.chat && components.length ? new (0, _chunkC6WNNTEVcjs.Tool)({
       name: "Message",
       description: "Send a message to the user",
       aliases: Array.from(

package/dist/{chunk-UNMKB2DA.js → chunk-IH2WQFO5.js}

@@ -5,7 +5,7 @@ import {
   escapeString,
   getMultilineComment,
   toPropertyKey
-} from "./chunk-5TRUJES5.js";
+} from "./chunk-4MNIJGK6.js";
 
 // src/typings.ts
 import { z } from "@bpinternal/zui";
@@ -268,7 +268,7 @@ ${typeof schema.value === "string" ? escapeString(schema.value) : schema.value}`
     return `${description} { [key: (${keyType})]: (${valueType}) }`;
   }
   try {
-    let typings = await (schema == null ? void 0 : schema.toTypescriptAsync());
+    let typings = schema == null ? void 0 : schema.toTypescriptType();
     typings ??= "unknown";
     return stripSpaces(typings);
   } catch (error) {

package/dist/{chunk-QBXIE6EY.js → chunk-JAGB2AOU.js}

@@ -1,18 +1,19 @@
 import {
   getTypings
-} from "./chunk-UNMKB2DA.js";
+} from "./chunk-IH2WQFO5.js";
 import {
   convertObjectToZuiLiterals,
   isJsonSchema,
-  isValidIdentifier
-} from "./chunk-5TRUJES5.js";
+  isValidIdentifier,
+  isZuiSchema
+} from "./chunk-4MNIJGK6.js";
 import {
   isEmpty_default,
   uniq_default
 } from "./chunk-7WRN4E42.js";
 
 // src/tool.ts
-import { z, ZodObject, ZodType } from "@bpinternal/zui";
+import { z, transforms, ZodObject, ZodType } from "@bpinternal/zui";
 var Tool = class _Tool {
   _staticInputValues;
   name;
@@ -28,7 +29,7 @@ var Tool = class _Tool {
       this._staticInputValues = void 0;
       return this;
     }
-    const input = this.input ? z.fromJsonSchema(this.input) : z.any();
+    const input = this.input ? transforms.fromJSONSchemaLegacy(this.input) : z.any();
     if (input instanceof z.ZodObject && typeof values !== "object") {
       throw new Error(
         `Invalid static input values for tool ${this.name}. Expected an object, but got type "${typeof values}"`
@@ -43,7 +44,7 @@ var Tool = class _Tool {
     return this;
   }
   get zInput() {
-    let input = this.input ? z.fromJsonSchema(this.input) : z.any();
+    let input = this.input ? transforms.fromJSONSchemaLegacy(this.input) : z.any();
     if (!isEmpty_default(this._staticInputValues)) {
       const inputExtensions = convertObjectToZuiLiterals(this._staticInputValues);
       if (input instanceof z.ZodObject) {
@@ -57,7 +58,7 @@ var Tool = class _Tool {
     return input;
   }
   get zOutput() {
-    return this.output ? z.fromJsonSchema(this.output) : z.void();
+    return this.output ? transforms.fromJSONSchemaLegacy(this.output) : z.void();
   }
   rename(name) {
     const before = this.name;
@@ -73,8 +74,8 @@ var Tool = class _Tool {
   clone(props = {}) {
     var _a, _b;
     try {
-      const zInput = this.input ? z.fromJsonSchema(this.input) : void 0;
-      const zOutput = this.output ? z.fromJsonSchema(this.output) : void 0;
+      const zInput = this.input ? transforms.fromJSONSchemaLegacy(this.input) : void 0;
+      const zOutput = this.output ? transforms.fromJSONSchemaLegacy(this.output) : void 0;
       return new _Tool({
         name: props.name ?? this.name,
         aliases: props.aliases ?? [...this.aliases],
@@ -120,8 +121,8 @@ var Tool = class _Tool {
       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();
+      if (isZuiSchema(props.input)) {
+        this.input = transforms.toJSONSchemaLegacy(props.input);
       } else if (isJsonSchema(props.input)) {
         this.input = props.input;
       } else {
@@ -131,8 +132,8 @@ var Tool = class _Tool {
       }
     }
     if (typeof props.output !== "undefined") {
-      if (props.output && "toJsonSchema" in props.output && typeof props.output.toJsonSchema === "function") {
-        this.output = props.output.toJsonSchema();
+      if (isZuiSchema(props.output)) {
+        this.output = transforms.toJSONSchemaLegacy(props.output);
       } else if (isJsonSchema(props.output)) {
         this.output = props.output;
       } else {
@@ -177,8 +178,8 @@ var Tool = class _Tool {
     );
   }
   async getTypings() {
-    let input = this.input ? z.fromJsonSchema(this.input) : void 0;
-    const output = this.output ? z.fromJsonSchema(this.output) : z.void();
+    let input = this.input ? transforms.fromJSONSchemaLegacy(this.input) : void 0;
+    const output = this.output ? transforms.fromJSONSchemaLegacy(this.output) : z.void();
     if ((input == null ? void 0 : input.naked()) instanceof ZodObject && typeof this._staticInputValues === "object" && !isEmpty_default(this._staticInputValues)) {
       const inputExtensions = convertObjectToZuiLiterals(this._staticInputValues);
       input = input.extend(inputExtensions);

package/dist/{chunk-FIVFS4HG.cjs → chunk-JGVAZO4X.cjs}

@@ -1,6 +1,6 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } }
 
-var _chunkP7J2WCBBcjs = require('./chunk-P7J2WCBB.cjs');
+var _chunk276Q6EWPcjs = require('./chunk-276Q6EWP.cjs');
 
 // src/truncator.ts
 var DEFAULT_REMOVE_CHUNK = 250;
@@ -25,11 +25,11 @@ function truncateWrappedContent({
   throwOnFailure = true
 }) {
   var _a, _b, _c;
-  const tokenizer = _chunkP7J2WCBBcjs.getTokenizer.call(void 0, );
+  const tokenizer = _chunk276Q6EWPcjs.getTokenizer.call(void 0, );
   const parts = [];
   for (const msg of messages) {
     const current = [];
-    const content = msg.content;
+    const content = typeof msg.content === "string" ? msg.content : "";
     let match;
     const regex = getRegex();
     let lastIndex = 0;
@@ -134,14 +134,14 @@ function truncateWrappedContent({
     const p = parts[i];
     return {
       ...msg,
-      content: removeRedundantWrappers(
+      content: typeof msg.content === "string" ? removeRedundantWrappers(
         p.map((part) => {
           if (part.truncatable) {
             return part.content;
           }
           return part.content;
         }).join("")
-      )
+      ) : msg.content
     };
   });
 }

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

@@ -1,13 +1,14 @@
 import {
   isJsonSchema,
-  isValidIdentifier
-} from "./chunk-5TRUJES5.js";
+  isValidIdentifier,
+  isZuiSchema
+} from "./chunk-4MNIJGK6.js";
 import {
   uniq_default
 } from "./chunk-7WRN4E42.js";
 
 // src/exit.ts
-import { z } from "@bpinternal/zui";
+import { transforms } from "@bpinternal/zui";
 var Exit = class _Exit {
   name;
   aliases = [];
@@ -15,7 +16,7 @@ var Exit = class _Exit {
   metadata;
   schema;
   get zSchema() {
-    return this.schema ? z.fromJsonSchema(this.schema) : void 0;
+    return this.schema ? transforms.fromJSONSchemaLegacy(this.schema) : void 0;
   }
   rename(name) {
     const before = this.name;
@@ -68,8 +69,8 @@ var Exit = class _Exit {
       throw new Error(`Invalid aliases for exit ${props.name}. Expected an array of valid identifiers.`);
     }
     if (typeof props.schema !== "undefined") {
-      if (props.schema && "toJsonSchema" in props.schema && typeof props.schema.toJsonSchema === "function") {
-        this.schema = props.schema.toJsonSchema();
+      if (isZuiSchema(props.schema)) {
+        this.schema = transforms.toJSONSchemaLegacy(props.schema);
       } else if (isJsonSchema(props.schema)) {
         this.schema = props.schema;
       } else {