llmz 0.1.5 → 0.2.0

package/dist/{chunk-L74UZONX.js → chunk-HM5H556L.js}

@@ -142,6 +142,7 @@ var _MessageContentParser = class {
 };
 var _createRegex = () => new RegExp(REGEXP, "g");
 var _renderRemainingWrappers = (content) => content.replace(_createRegex(), "$2");
+var stripTruncationTags = (content) => _renderRemainingWrappers(content);
 var _countTotalTokens = (parts) => parts.reduce((acc, x) => acc + x.reduce((acc2, y) => acc2 + y.tokens, 0), 0);
 var _getTwoBiggestTruncables = (parts) => {
   var _a, _b;
@@ -167,5 +168,6 @@ var _getTwoBiggestTruncables = (parts) => {
 
 export {
   wrapContent,
-  truncateWrappedContent
+  truncateWrappedContent,
+  stripTruncationTags
 };

package/dist/chunk-I56MPOVG.cjs

@@ -0,0 +1,62 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+
+
+var _chunk5ZF6NTYIcjs = require('./chunk-5ZF6NTYI.cjs');
+
+// src/component.ts
+function assertValidComponent(component) {
+  if (!component.name) {
+    throw new Error("Component must have a name");
+  }
+  const nameRegex = /^[A-Z]{3,50}$/i;
+  if (!nameRegex.test(component.name)) {
+    throw new Error(
+      `Component name "${component.name}" must be 3-50 characters long and start with an uppercase letter`
+    );
+  }
+  if (component.aliases) {
+    for (const alias of component.aliases) {
+      if (!nameRegex.test(alias)) {
+        throw new Error(`Alias "${alias}" must be 3-50 characters long and start with an uppercase letter`);
+      }
+    }
+  }
+  if (!component.description) {
+    throw new Error("Component must have a description");
+  }
+  if (component.type === "default" && !component.default) {
+    throw new Error("Default component must have default props and children");
+  }
+  if (component.type === "leaf" && !component.leaf) {
+    throw new Error("Leaf component must have leaf props");
+  }
+  if (component.type === "container" && !component.container) {
+    throw new Error("Container component must have container props and children");
+  }
+}
+var Component = class {
+  
+  
+  // phantom type for inference
+  constructor(definition) {
+    assertValidComponent(definition);
+    this.definition = definition;
+  }
+  render(props, children = []) {
+    return _chunk5ZF6NTYIcjs.createJsxComponent.call(void 0, { type: this.definition.name, props, children });
+  }
+};
+function isComponent(rendered, component) {
+  return _chunk5ZF6NTYIcjs.isJsxComponent.call(void 0, component.definition.name, rendered);
+}
+function isAnyComponent(message) {
+  return _chunk5ZF6NTYIcjs.isAnyJsxComponent.call(void 0, message);
+}
+
+
+
+
+
+
+exports.assertValidComponent = assertValidComponent; exports.Component = Component; exports.isComponent = isComponent; exports.isAnyComponent = isAnyComponent;

package/dist/{chunk-C7T6OJB4.cjs → chunk-IKLFMVSW.cjs}

@@ -2,7 +2,7 @@
 
 
 
-var _chunkBBX2LDZEcjs = require('./chunk-BBX2LDZE.cjs');
+var _chunkYGKQEVQQcjs = require('./chunk-YGKQEVQQ.cjs');
 
 
 var _chunkWODK7Y4Jcjs = require('./chunk-WODK7Y4J.cjs');
@@ -14,10 +14,14 @@ var _chunkPK72FAKDcjs = require('./chunk-PK72FAKD.cjs');
 var _chunkD6PU5YAScjs = require('./chunk-D6PU5YAS.cjs');
 
 
-var _chunkCQ3UJW76cjs = require('./chunk-CQ3UJW76.cjs');
+var _chunkI56MPOVGcjs = require('./chunk-I56MPOVG.cjs');
 
 
-var _chunkZPHRTTQXcjs = require('./chunk-ZPHRTTQX.cjs');
+
+var _chunkYQT5G77Wcjs = require('./chunk-YQT5G77W.cjs');
+
+
+var _chunkFPLSJF4Scjs = require('./chunk-FPLSJF4S.cjs');
 
 
 
@@ -493,10 +497,10 @@ var Snapshot = class _Snapshot {
 };
 function parseVariables(variableMap) {
   return Object.entries(variableMap).map(([name, value]) => {
-    const type = _chunkBBX2LDZEcjs.extractType.call(void 0, value);
+    const type = _chunkYGKQEVQQcjs.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(_chunkBBX2LDZEcjs.inspect.call(void 0, value, name), () => ( "N/A")) } : { name, type, bytes, truncated: false, value };
+    return truncated ? { name, type, bytes, truncated: true, preview: _nullishCoalesce(_chunkYGKQEVQQcjs.inspect.call(void 0, value, name), () => ( "N/A")) } : { name, type, bytes, truncated: false, value };
   });
 }
 
@@ -507,9 +511,9 @@ function getMessagePreview(message) {
     return message.content;
   }
   if (message.role === "event") {
-    return _chunkBBX2LDZEcjs.inspect.call(void 0, message.payload, message.name, { tokens: 1e3 });
+    return _chunkYGKQEVQQcjs.inspect.call(void 0, message.payload, message.name, { tokens: 1e3 });
   }
-  return _chunkBBX2LDZEcjs.inspect.call(void 0, message, void 0, { tokens: 1e3 });
+  return _chunkYGKQEVQQcjs.inspect.call(void 0, message, void 0, { tokens: 1e3 });
 }
 function getMessageType(message) {
   if (message.role === "assistant" || message.role === "user" || message.role === "summary") {
@@ -777,6 +781,33 @@ var ExecutionResult = class {
   get iterations() {
     return _nullishCoalesce(this.context.iterations, () => ( []));
   }
+  /**
+   * Gets the total token usage of the execution, aggregated across all iterations.
+   *
+   * `input`/`output` are the provider-reported token counts of every LLM call made
+   * during the execution. Per-iteration detail (including the measured context size
+   * breakdown by prompt part) is available on each iteration's `tokens` property.
+   *
+   * @example
+   * ```typescript
+   * const result = await execute({ ... })
+   * console.log(`Execution used ${result.tokens.total} tokens`)
+   *
+   * for (const iteration of result.iterations) {
+   *   console.log(iteration.id, iteration.tokens?.context)
+   * }
+   * ```
+   */
+  get tokens() {
+    var _a, _b;
+    let input = 0;
+    let output = 0;
+    for (const iteration of this.iterations) {
+      input += _nullishCoalesce(((_a = iteration.tokens) == null ? void 0 : _a.input), () => ( 0));
+      output += _nullishCoalesce(((_b = iteration.tokens) == null ? void 0 : _b.output), () => ( 0));
+    }
+    return { input, output, total: input + output };
+  }
 };
 var SuccessExecutionResult = class extends ExecutionResult {
   
@@ -959,7 +990,17 @@ var Iteration = class {
   
   
   
+  /** Messages (`■send` blocks) parsed from the assistant response, in order. */
   
+  /** The `■next` exit parsed from the assistant response, if any. */
+  
+  
+  
+  /**
+   * Token usage of this iteration's LLM call. The `context` breakdown is measured
+   * when the prompt is assembled; `input`/`output` are filled in once the LLM call
+   * completes, from the provider-reported usage.
+   */
   
   
   
@@ -994,7 +1035,7 @@ var Iteration = class {
     return this._parameters.reasoningEffort;
   }
   get exits() {
-    const exits = [...this._parameters.exits, ThinkExit];
+    const exits = [...this._parameters.exits];
     if (this.isChatEnabled) {
       exits.push(ListenExit);
     }
@@ -1048,7 +1089,7 @@ var Iteration = class {
     return null;
   }
   get isChatEnabled() {
-    return this._parameters.tools.find((x) => x.name.toLowerCase() === "message") !== void 0;
+    return this._parameters.components.length > 0;
   }
   constructor(props) {
     this.id = props.id;
@@ -1082,6 +1123,7 @@ var Iteration = class {
       status: this.status,
       mutations: [...this._mutations.values()],
       llm: this.llm,
+      tokens: this.tokens,
       transcript: [...this._parameters.transcript],
       tools: this._parameters.tools.map((tool) => tool.toJSON()),
       objects: this._parameters.objects.map((obj) => obj.toJSON()),
@@ -1103,10 +1145,20 @@ var Context = (_class2 = class {
   
   
   
-  __init3() {this.version = _chunkBBX2LDZEcjs.DualModePrompt}
+  __init3() {this.version = _chunkYGKQEVQQcjs.DualModePrompt}
   __init4() {this.timeout = 6e4}
   // Default timeout of 60 seconds
   
+  /**
+   * Optional cap on the model's context window. The effective limit is
+   * `min(maxTokens, model's max input tokens)`.
+   */
+  
+  /**
+   * Maximum time to wait for the first streamed token, in milliseconds,
+   * before the cognitive service falls back to the next model/provider.
+   */
+  
   
   
   __init5() {this.iteration = 0}
@@ -1121,13 +1173,15 @@ var Context = (_class2 = class {
       );
     }
     const parameters = await this._refreshIterationParameters();
-    const messages = await this._getIterationMessages(parameters);
+    const { messages, parts } = await this._getIterationMessages(parameters);
+    const contextTokens = this._measureContextTokens(messages, parts);
     const iteration = new Iteration({
       id: `${this.id}_${this.iterations.length + 1}`,
       variables: this._getIterationVariables(),
       parameters,
       messages
     });
+    iteration.tokens = { input: 0, output: 0, total: 0, context: contextTokens };
     this.iterations.push(iteration);
     this.iteration = this.iterations.length;
     this.snapshot = void 0;
@@ -1155,121 +1209,129 @@ var Context = (_class2 = class {
     }
     return variables;
   }
+  /**
+   * Measures the token size of each part of the prompt (pre-truncation).
+   * The named parts come from the system prompt; everything else in the system
+   * message is `framework`. Non-system messages count towards `iterations`,
+   * except the very first user message of a fresh execution (task recap),
+   * which is prompt scaffolding.
+   */
+  _measureContextTokens(messages, parts) {
+    const tokenizer = _chunkFPLSJF4Scjs.getTokenizer.call(void 0, );
+    const countText = (text) => (text == null ? void 0 : text.length) ? tokenizer.count(_chunkYQT5G77Wcjs.stripTruncationTags.call(void 0, text)) : 0;
+    const countMessage = (message) => {
+      if (typeof message.content === "string") {
+        return countText(message.content);
+      }
+      if (Array.isArray(message.content)) {
+        return message.content.reduce((acc, part) => acc + (part.type === "text" ? countText(part.text) : 0), 0);
+      }
+      return 0;
+    };
+    const instructions = countText(parts.instructions);
+    const tools = countText(parts.tools);
+    const transcript = countText(parts.transcript);
+    const protocol = countText(parts.protocol);
+    const systemTokens = messages.filter((x) => x.role === "system").reduce((acc, x) => acc + countMessage(x), 0);
+    const otherTokens = messages.filter((x) => x.role !== "system").reduce((acc, x) => acc + countMessage(x), 0);
+    const isFirstIteration = this.iterations.length === 0 && !this.snapshot;
+    const framework = Math.max(0, systemTokens - (instructions + tools + transcript + protocol));
+    const iterations = isFirstIteration ? 0 : otherTokens;
+    return {
+      total: framework + instructions + tools + transcript + protocol + iterations + (isFirstIteration ? otherTokens : 0),
+      framework: framework + (isFirstIteration ? otherTokens : 0),
+      instructions,
+      tools,
+      transcript,
+      protocol,
+      iterations
+    };
+  }
   async _getIterationMessages(parameters) {
     var _a, _b, _c, _d, _e, _f;
     const lastIteration = this.iterations.at(-1);
+    const promptProps = {
+      globalTools: parameters.tools,
+      objects: parameters.objects,
+      instructions: parameters.instructions,
+      transcript: parameters.transcript,
+      // ListenExit is protocol-level in chat mode: it must be documented alongside user-defined exits
+      exits: parameters.components.length ? [...parameters.exits, ListenExit] : parameters.exits,
+      components: parameters.components
+    };
+    const { message: systemMessage, parts } = await this.version.getSystemMessage(promptProps);
+    const withParts = (messages) => ({ messages, parts });
     if (((_a = this.snapshot) == null ? void 0 : _a.status.type) === "resolved") {
-      return [
-        await this.version.getSystemMessage({
-          globalTools: parameters.tools,
-          objects: parameters.objects,
-          instructions: parameters.instructions,
-          transcript: parameters.transcript,
-          exits: parameters.exits,
-          components: parameters.components
-        }),
+      return withParts([
+        systemMessage,
         this.version.getSnapshotResolvedMessage({
           snapshot: this.snapshot
         })
-      ];
+      ]);
     }
     if (((_b = this.snapshot) == null ? void 0 : _b.status.type) === "rejected") {
-      return [
-        await this.version.getSystemMessage({
-          globalTools: parameters.tools,
-          objects: parameters.objects,
-          instructions: parameters.instructions,
-          transcript: parameters.transcript,
-          exits: parameters.exits,
-          components: parameters.components
-        }),
+      return withParts([
+        systemMessage,
         this.version.getSnapshotRejectedMessage({
           snapshot: this.snapshot
         })
-      ];
+      ]);
     }
     if (!lastIteration) {
-      return [
-        await this.version.getSystemMessage({
-          globalTools: parameters.tools,
-          objects: parameters.objects,
-          instructions: parameters.instructions,
-          transcript: parameters.transcript,
-          exits: parameters.exits,
-          components: parameters.components
-        }),
-        await this.version.getInitialUserMessage({
-          globalTools: parameters.tools,
-          objects: parameters.objects,
-          instructions: parameters.instructions,
-          transcript: parameters.transcript,
-          exits: parameters.exits,
-          components: parameters.components
-        })
-      ];
+      return withParts([systemMessage, await this.version.getInitialUserMessage(promptProps)]);
     }
-    const lastIterationMessages = [
-      await this.version.getSystemMessage({
-        globalTools: parameters.tools,
-        objects: parameters.objects,
-        instructions: parameters.instructions,
-        transcript: parameters.transcript,
-        exits: parameters.exits,
-        components: parameters.components
-      }),
-      ...lastIteration.messages.filter((x) => x.role !== "system")
-    ];
+    const lastIterationMessages = [systemMessage, ...lastIteration.messages.filter((x) => x.role !== "system")];
     if ((lastIteration == null ? void 0 : lastIteration.status.type) === "thinking_requested") {
-      return [
+      return withParts([
         ...lastIterationMessages,
         {
           role: "assistant",
-          content: _chunkZPHRTTQXcjs.wrapContent.call(void 0, _nullishCoalesce(((_c = lastIteration.llm) == null ? void 0 : _c.output), () => ( "")), { preserve: "top", flex: 4, minTokens: 25 })
+          content: _chunkYQT5G77Wcjs.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,
           variables: lastIteration.status.thinking_requested.variables
         })
-      ];
+      ]);
     }
     if ((lastIteration == null ? void 0 : lastIteration.status.type) === "exit_error") {
-      return [
+      return withParts([
         ...lastIterationMessages,
         {
           role: "assistant",
-          content: _chunkZPHRTTQXcjs.wrapContent.call(void 0, _nullishCoalesce(((_d = lastIteration.llm) == null ? void 0 : _d.output), () => ( "")), { preserve: "top", flex: 4, minTokens: 25 })
+          content: _chunkYQT5G77Wcjs.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")),
           message: `Invalid return statement (action: ${lastIteration.status.exit_error.exit}): ${lastIteration.status.exit_error.message}`
         })
-      ];
+      ]);
     }
     if ((lastIteration == null ? void 0 : lastIteration.status.type) === "invalid_code_error") {
-      return [
+      return withParts([
         ...lastIterationMessages,
         {
           role: "assistant",
-          content: _chunkZPHRTTQXcjs.wrapContent.call(void 0, _nullishCoalesce(((_e = lastIteration.llm) == null ? void 0 : _e.output), () => ( "")), { preserve: "top", flex: 4, minTokens: 25 })
+          content: _chunkYQT5G77Wcjs.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")),
           message: lastIteration.status.invalid_code_error.message
         })
-      ];
+      ]);
     }
     if ((lastIteration == null ? void 0 : lastIteration.status.type) === "execution_error") {
-      return [
+      return withParts([
         ...lastIterationMessages,
         {
           role: "assistant",
-          content: _chunkZPHRTTQXcjs.wrapContent.call(void 0, _nullishCoalesce(((_f = lastIteration.llm) == null ? void 0 : _f.output), () => ( "")), { preserve: "top", flex: 4, minTokens: 25 })
+          content: _chunkYQT5G77Wcjs.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,
           stacktrace: lastIteration.status.execution_error.stack
         })
-      ];
+      ]);
     }
     throw new Error(
       `Unexpected iteration status: ${lastIteration == null ? void 0 : lastIteration.status.type}. This is likely a bug, please report it.`
@@ -1293,7 +1355,7 @@ var Context = (_class2 = class {
       throw new Error("Too many tools. Expected at most 100 tools.");
     }
     for (const component of components) {
-      _chunkCQ3UJW76cjs.assertValidComponent.call(void 0, component.definition);
+      _chunkI56MPOVGcjs.assertValidComponent.call(void 0, component.definition);
     }
     const ReservedToolNames = [
       "think",
@@ -1312,18 +1374,6 @@ var Context = (_class2 = class {
       "boolean",
       "array"
     ];
-    const MessageTool = this.chat && components.length ? new (0, _chunkWODK7Y4Jcjs.Tool)({
-      name: "Message",
-      description: "Send a message to the user",
-      aliases: Array.from(
-        /* @__PURE__ */ new Set(["message", ...components.flatMap((x) => [x.definition.name, ..._nullishCoalesce(x.definition.aliases, () => ( []))])])
-      ),
-      handler: async (message) => {
-        var _a2, _b2;
-        return await ((_b2 = (_a2 = this.chat) == null ? void 0 : _a2.handler) == null ? void 0 : _b2.call(_a2, message));
-      }
-    }) : null;
-    const allTools = MessageTool ? [MessageTool, ...tools] : tools;
     for (const tool of tools) {
       for (let name of [...tool.aliases, tool.name]) {
         name = name.toLowerCase();
@@ -1380,7 +1430,7 @@ var Context = (_class2 = class {
     }
     return {
       transcript,
-      tools: allTools,
+      tools,
       objects,
       exits,
       instructions,
@@ -1405,9 +1455,17 @@ var Context = (_class2 = class {
     this.iterations = [];
     this.metadata = _nullishCoalesce(props.metadata, () => ( {}));
     this.snapshot = props.snapshot;
+    this.maxTokens = props.maxTokens;
+    this.maxTimeToFirstToken = props.maxTimeToFirstToken;
     if (this.loop < 1 || this.loop > 100) {
       throw new Error("Invalid loop. Expected a number between 1 and 100.");
     }
+    if (this.maxTokens !== void 0 && (!Number.isFinite(this.maxTokens) || this.maxTokens < 1)) {
+      throw new Error("Invalid maxTokens. Expected a positive number.");
+    }
+    if (this.maxTimeToFirstToken !== void 0 && (!Number.isFinite(this.maxTimeToFirstToken) || this.maxTimeToFirstToken < 1)) {
+      throw new Error("Invalid maxTimeToFirstToken. Expected a positive number of milliseconds.");
+    }
   }
   toJSON() {
     var _a;

package/dist/chunk-KWQTYXAT.js

@@ -0,0 +1,62 @@
+import {
+  createJsxComponent,
+  isAnyJsxComponent,
+  isJsxComponent
+} from "./chunk-XTEHT2TK.js";
+
+// src/component.ts
+function assertValidComponent(component) {
+  if (!component.name) {
+    throw new Error("Component must have a name");
+  }
+  const nameRegex = /^[A-Z]{3,50}$/i;
+  if (!nameRegex.test(component.name)) {
+    throw new Error(
+      `Component name "${component.name}" must be 3-50 characters long and start with an uppercase letter`
+    );
+  }
+  if (component.aliases) {
+    for (const alias of component.aliases) {
+      if (!nameRegex.test(alias)) {
+        throw new Error(`Alias "${alias}" must be 3-50 characters long and start with an uppercase letter`);
+      }
+    }
+  }
+  if (!component.description) {
+    throw new Error("Component must have a description");
+  }
+  if (component.type === "default" && !component.default) {
+    throw new Error("Default component must have default props and children");
+  }
+  if (component.type === "leaf" && !component.leaf) {
+    throw new Error("Leaf component must have leaf props");
+  }
+  if (component.type === "container" && !component.container) {
+    throw new Error("Container component must have container props and children");
+  }
+}
+var Component = class {
+  definition;
+  propsType;
+  // phantom type for inference
+  constructor(definition) {
+    assertValidComponent(definition);
+    this.definition = definition;
+  }
+  render(props, children = []) {
+    return createJsxComponent({ type: this.definition.name, props, children });
+  }
+};
+function isComponent(rendered, component) {
+  return isJsxComponent(component.definition.name, rendered);
+}
+function isAnyComponent(message) {
+  return isAnyJsxComponent(message);
+}
+
+export {
+  assertValidComponent,
+  Component,
+  isComponent,
+  isAnyComponent
+};