llmz 0.2.1 → 0.2.3

package/dist/compiler/ast.d.ts

@@ -0,0 +1,26 @@
+import { type Comment, type Node, type Options, type Program } from 'acorn';
+import type MagicString from 'magic-string';
+/** An acorn node with free access to its type-specific properties */
+export type AnyNode = Node & Record<string, any>;
+export type Ctx = {
+    code: string;
+    ms: MagicString;
+    ast: Program;
+    comments: Comment[];
+};
+export declare function parseScript(code: string, opts?: Partial<Options> & {
+    comments?: Comment[];
+}): Program;
+/**
+ * Depth-first pre-order walk. `ancestors` is ordered outermost → innermost and
+ * excludes `node` itself; `parent` is `ancestors[ancestors.length - 1]`.
+ */
+export declare function walk(root: Node, enter: (node: AnyNode, parent: AnyNode | null, ancestors: AnyNode[]) => void): void;
+export declare const isStatementContainer: (node: AnyNode | null) => boolean;
+/**
+ * The statement that contains `node`: the outermost ancestor (or the node
+ * itself) whose parent is a statement container.
+ */
+export declare function enclosingStatement(node: AnyNode, ancestors: AnyNode[]): AnyNode | null;
+/** The innermost node whose range contains [start, end] */
+export declare function innermostContaining(ast: Program, start: number, end: number): AnyNode | null;

package/dist/compiler/compiler.d.ts

@@ -1,6 +1,4 @@
-import { type TransformOptions } from '@babel/core';
 import { ToolCallEntry } from './plugins/track-tool-calls.js';
-export declare const DEFAULT_TRANSFORM_OPTIONS: TransformOptions;
 export declare const Identifiers: {
     ConsoleObjIdentifier: string;
     LineTrackingFnIdentifier: string;
@@ -17,18 +15,22 @@ export type CompiledCode = ReturnType<typeof compile>;
  * Falls back to a word-boundary match if the code cannot be parsed.
  */
 export declare function hasTopLevelReturn(code: string): boolean;
+/**
+ * Compiles the agent's generated JavaScript for VM execution:
+ *
+ * 1. wraps it in an async `__fn__` so top-level `await`/`return` parse
+ * 2. instruments it in a single parse + text-edit pass:
+ *    line tracking, last-line return, tool-call tracking, comment tracing
+ *    and variable tracking
+ * 3. unwraps it — the VM drivers re-wrap the bare statements themselves
+ *
+ * All edits preserve line numbers 1:1 with the wrapped source, so positions
+ * reported at runtime map straight back to the user code.
+ */
 export declare function compile(code: string): {
     code: string;
     codeWithMarkers: string;
-    map: {
-        version: number;
-        sources: string[];
-        names: string[];
-        sourceRoot?: string | undefined;
-        sourcesContent?: string[] | undefined;
-        mappings: string;
-        file: string;
-    } | null | undefined;
+    map: import("magic-string").SourceMap;
     variables: Set<string>;
     toolCalls: Map<number, ToolCallEntry>;
 };

package/dist/compiler/plugins/line-tracking.d.ts

@@ -1,5 +1,7 @@
-import type * as BabelCore from '@babel/core';
+import { type Ctx } from '../ast.js';
 export declare const LineTrackingFnIdentifier = "__track__";
-export declare const lineTrackingBabelPlugin: ({ types: t }: {
-    types: typeof BabelCore.types;
-}) => BabelCore.PluginObj;
+/**
+ * Inserts a `__track__(<line>)` call before the statement containing each
+ * tracked node — at most one per source line.
+ */
+export declare function applyLineTracking(ctx: Ctx): void;

package/dist/compiler/plugins/replace-comment.d.ts

@@ -1,5 +1,9 @@
-import type * as BabelCore from '@babel/core';
+import { type Ctx } from '../ast.js';
 export declare const CommentFnIdentifier = "__comment__";
-export declare const replaceCommentBabelPlugin: ({ types: t, }: {
-    types: typeof BabelCore.types;
-}) => BabelCore.PluginObj;
+/**
+ * Replaces comments at statement position with `__comment__("<text>", <line>)`
+ * calls so the agent's inline "thinking" comments show up as traces at runtime.
+ * Comments inside expressions and the user-code markers are blanked; comments
+ * inside instrumented tool-call ranges are left verbatim.
+ */
+export declare function applyCommentReplacement(ctx: Ctx, skipRanges: Array<[number, number]>): void;

package/dist/compiler/plugins/return-async.d.ts

@@ -1,4 +1,18 @@
-import type * as BabelCore from '@babel/core';
-export declare const instrumentLastLinePlugin: ({ types: t, }: {
-    types: typeof BabelCore.types;
-}) => BabelCore.PluginObj;
+import { type Program } from 'acorn';
+export type LastLineEdit = {
+    prefixPos: number;
+    prefix: string;
+    suffixPos: number;
+    suffix: string;
+};
+/**
+ * Plans the instrumentation of the last statement of the `__fn__` wrapper so
+ * the generated code always returns an awaited value (the isolated VM cannot
+ * return a pending Promise):
+ * - `return expr` → `return await (expr)`
+ * - a trailing `call()` / `await call()` statement → `return await (…)`
+ *
+ * The prefix must be applied before, and the suffix after, all other edits so
+ * they wrap around the tool-call instrumentation.
+ */
+export declare function planLastLineInstrumentation(ast: Program): LastLineEdit | null;

package/dist/compiler/plugins/track-tool-calls.d.ts

@@ -1,4 +1,4 @@
-import type * as BabelCore from '@babel/core';
+import { type Ctx } from '../ast.js';
 export declare const ToolCallTrackerFnIdentifier = "__toolc__";
 export declare const ToolTrackerRetIdentifier = "__ret__";
 export type Assignment = {
@@ -11,6 +11,23 @@ export type ToolCallEntry = {
     tool: string;
     assignment: Assignment;
 };
-export declare const toolCallTrackingPlugin: (calls?: Map<number, ToolCallEntry>) => ({}: {
-    types: typeof BabelCore.types;
-}) => BabelCore.PluginObj;
+/**
+ * Wraps every outermost call expression in an IIFE that reports
+ * `__toolc__(<id>, "start" | "end", …)` events around the call:
+ *
+ *     const x = (() => {
+ *       try {
+ *         __toolc__(0, "start");
+ *         const __ret__ = tool();
+ *         __toolc__(0, "end", __ret__);
+ *         return __ret__;
+ *       } catch (err) { __toolc__(1, "end", err); throw new Error(err.message); }
+ *     })()
+ *
+ * Calls assigned to a variable are registered in `calls` (keyed by the id used
+ * in the catch clause — the id the runtime sees when a tool throws a
+ * SnapshotSignal). Awaited calls get an async IIFE that awaits the call.
+ *
+ * Returns the wrapped source ranges so later passes can avoid editing inside.
+ */
+export declare function applyToolCallTracking(ctx: Ctx, calls: Map<number, ToolCallEntry>): Array<[number, number]>;

package/dist/compiler/plugins/variable-extraction.d.ts

@@ -1,5 +1,10 @@
-import type * as BabelCore from '@babel/core';
+import { type Ctx } from '../ast.js';
 export declare const VariableTrackingFnIdentifier = "__var__";
-export declare const variableTrackingPlugin: (variables?: Set<string>) => ({ types: t }: {
-    types: typeof BabelCore.types;
-}) => BabelCore.PluginObj;
+/**
+ * Tracks user variables so the VM can read their values between iterations:
+ * - after each variable declaration: `__var__("a", () => eval("a"));`
+ * - at the top of function bodies, for their parameters
+ *
+ * The `eval` indirection captures the variable lazily in its own scope.
+ */
+export declare function applyVariableTracking(ctx: Ctx, variables: Set<string>): void;

package/dist/component.default.d.ts

@@ -17,17 +17,7 @@ export declare const DefaultComponents: {
                 label: z.ZodString;
                 value: z.ZodOptional<z.ZodString>;
                 url: z.ZodOptional<z.ZodString>;
-            }, "strip", {
-                action: "say" | "url" | "postback";
-                label: string;
-                value?: string | undefined;
-                url?: string | undefined;
-            }, {
-                label: string;
-                value?: string | undefined;
-                url?: string | undefined;
-                action?: "say" | "url" | "postback" | undefined;
-            }>;
+            }, "strip">;
         };
     }>;
     Image: Component<{
@@ -44,13 +34,7 @@ export declare const DefaultComponents: {
             props: z.ZodObject<{
                 url: z.ZodString;
                 alt: z.ZodOptional<z.ZodString>;
-            }, "strip", {
-                url: string;
-                alt?: string | undefined;
-            }, {
-                url: string;
-                alt?: string | undefined;
-            }>;
+            }, "strip">;
         };
     }>;
     File: Component<{
@@ -67,13 +51,7 @@ export declare const DefaultComponents: {
             props: z.ZodObject<{
                 url: z.ZodString;
                 name: z.ZodOptional<z.ZodString>;
-            }, "strip", {
-                url: string;
-                name?: string | undefined;
-            }, {
-                url: string;
-                name?: string | undefined;
-            }>;
+            }, "strip">;
         };
     }>;
     Video: Component<{
@@ -90,13 +68,7 @@ export declare const DefaultComponents: {
             props: z.ZodObject<{
                 url: z.ZodString;
                 title: z.ZodOptional<z.ZodString>;
-            }, "strip", {
-                url: string;
-                title?: string | undefined;
-            }, {
-                url: string;
-                title?: string | undefined;
-            }>;
+            }, "strip">;
         };
     }>;
     Audio: Component<{
@@ -113,13 +85,7 @@ export declare const DefaultComponents: {
             props: z.ZodObject<{
                 url: z.ZodString;
                 title: z.ZodOptional<z.ZodString>;
-            }, "strip", {
-                url: string;
-                title?: string | undefined;
-            }, {
-                url: string;
-                title?: string | undefined;
-            }>;
+            }, "strip">;
         };
     }>;
     Card: Component<{
@@ -136,13 +102,7 @@ export declare const DefaultComponents: {
             props: z.ZodObject<{
                 title: z.ZodString;
                 subtitle: z.ZodOptional<z.ZodString>;
-            }, "strip", {
-                title: string;
-                subtitle?: string | undefined;
-            }, {
-                title: string;
-                subtitle?: string | undefined;
-            }>;
+            }, "strip">;
             children: ({
                 description: string;
                 component: {
@@ -159,13 +119,7 @@ export declare const DefaultComponents: {
                         props: z.ZodObject<{
                             url: z.ZodString;
                             alt: z.ZodOptional<z.ZodString>;
-                        }, "strip", {
-                            url: string;
-                            alt?: string | undefined;
-                        }, {
-                            url: string;
-                            alt?: string | undefined;
-                        }>;
+                        }, "strip">;
                     };
                 };
             } | {
@@ -186,17 +140,7 @@ export declare const DefaultComponents: {
                             label: z.ZodString;
                             value: z.ZodOptional<z.ZodString>;
                             url: z.ZodOptional<z.ZodString>;
-                        }, "strip", {
-                            action: "say" | "url" | "postback";
-                            label: string;
-                            value?: string | undefined;
-                            url?: string | undefined;
-                        }, {
-                            label: string;
-                            value?: string | undefined;
-                            url?: string | undefined;
-                            action?: "say" | "url" | "postback" | undefined;
-                        }>;
+                        }, "strip">;
                     };
                 };
             })[];
@@ -213,7 +157,7 @@ export declare const DefaultComponents: {
             code: string;
         }[];
         container: {
-            props: z.ZodObject<{}, "strip", {}, {}>;
+            props: z.ZodObject<{}, "strip">;
             children: {
                 description: string;
                 component: {
@@ -230,13 +174,7 @@ export declare const DefaultComponents: {
                         props: z.ZodObject<{
                             title: z.ZodString;
                             subtitle: z.ZodOptional<z.ZodString>;
-                        }, "strip", {
-                            title: string;
-                            subtitle?: string | undefined;
-                        }, {
-                            title: string;
-                            subtitle?: string | undefined;
-                        }>;
+                        }, "strip">;
                         children: ({
                             description: string;
                             component: {
@@ -253,13 +191,7 @@ export declare const DefaultComponents: {
                                     props: z.ZodObject<{
                                         url: z.ZodString;
                                         alt: z.ZodOptional<z.ZodString>;
-                                    }, "strip", {
-                                        url: string;
-                                        alt?: string | undefined;
-                                    }, {
-                                        url: string;
-                                        alt?: string | undefined;
-                                    }>;
+                                    }, "strip">;
                                 };
                             };
                         } | {
@@ -280,17 +212,7 @@ export declare const DefaultComponents: {
                                         label: z.ZodString;
                                         value: z.ZodOptional<z.ZodString>;
                                         url: z.ZodOptional<z.ZodString>;
-                                    }, "strip", {
-                                        action: "say" | "url" | "postback";
-                                        label: string;
-                                        value?: string | undefined;
-                                        url?: string | undefined;
-                                    }, {
-                                        label: string;
-                                        value?: string | undefined;
-                                        url?: string | undefined;
-                                        action?: "say" | "url" | "postback" | undefined;
-                                    }>;
+                                    }, "strip">;
                                 };
                             };
                         })[];
@@ -310,7 +232,7 @@ export declare const DefaultComponents: {
             code: string;
         }[];
         default: {
-            props: z.ZodObject<{}, "strip", {}, {}>;
+            props: z.ZodObject<{}, "strip">;
             children: never[];
         };
     }>;

package/dist/{dual-modes-RMF7V7ZQ.js → dual-modes-BYABHBDI.js}

@@ -1,9 +1,9 @@
 import {
   DualModePrompt
-} from "./chunk-E5NHK3FS.js";
+} from "./chunk-OXEKAOVI.js";
 import "./chunk-YEAWWJSJ.js";
-import "./chunk-HM5H556L.js";
-import "./chunk-YX27XM67.js";
+import "./chunk-72ZWBNZV.js";
+import "./chunk-GLXADNTR.js";
 import "./chunk-34GJUS6U.js";
 export {
   DualModePrompt

package/dist/dual-modes-HO6RRIZ5.cjs

@@ -0,0 +1,10 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunkOVRDJMZJcjs = require('./chunk-OVRDJMZJ.cjs');
+require('./chunk-KIN7Y247.cjs');
+require('./chunk-HEAQAEFZ.cjs');
+require('./chunk-GOWPACSK.cjs');
+require('./chunk-HC2QZTVT.cjs');
+
+
+exports.DualModePrompt = _chunkOVRDJMZJcjs.DualModePrompt;

package/dist/{execute-LNUBLVMZ.js → execute-DSGJ2UNL.js}

@@ -2,7 +2,7 @@ import {
   hasTopLevelReturn,
   runAsyncFunction,
   warmupVM
-} from "./chunk-DH35WO5G.js";
+} from "./chunk-YRS7KRKP.js";
 import {
   Context,
   ErrorExecutionResult,
@@ -11,13 +11,14 @@ import {
   SuccessExecutionResult,
   _CustomModelClient,
   parseExit
-} from "./chunk-ZWUWCRHG.js";
+} from "./chunk-XMFQVARN.js";
 import {
   StreamingMessageParser,
   toParsedAssistantResponse
-} from "./chunk-E5NHK3FS.js";
-import "./chunk-GC5IQDGV.js";
-import "./chunk-665O7CAN.js";
+} from "./chunk-OXEKAOVI.js";
+import "./chunk-UGHFNCPD.js";
+import "./chunk-TDALRNGL.js";
+import "./chunk-GDJTKI7J.js";
 import {
   AssignmentError,
   CodeExecutionError,
@@ -31,19 +32,19 @@ import {
 import {
   cleanStackTrace
 } from "./chunk-YEAWWJSJ.js";
-import "./chunk-XVVYNOTX.js";
+import "./chunk-HB2G32AK.js";
 import "./chunk-KWQTYXAT.js";
 import {
   createJsxComponent
 } from "./chunk-XTEHT2TK.js";
 import {
   truncateWrappedContent
-} from "./chunk-HM5H556L.js";
+} from "./chunk-72ZWBNZV.js";
 import {
   getErrorMessage,
   init,
   stripInvalidIdentifiers
-} from "./chunk-YX27XM67.js";
+} from "./chunk-GLXADNTR.js";
 import {
   clamp_default,
   isEqual_default