@abdwhb-png/pi-test-harness 0.7.0

package/dist/mock-tools.js

@@ -0,0 +1,192 @@
+/**
+ * Tool execution interceptor — wraps tool.execute() for tools in mockTools.
+ *
+ * For mocked tools, the mock replaces tool.execute() and returns controlled
+ * values. Extension hooks (tool_call / tool_result) are handled by
+ * AgentSession 0.83's internal beforeToolCall/afterToolCall — the mock
+ * must NOT re-emit them.
+ *
+ * For non-mocked tools, the real execute() is called and results are
+ * collected for event queries.
+ */
+import { formatToolError } from "./diagnostics.js";
+/**
+ * Thrown when an extension hook blocks a tool call.
+ * Exported for test assertions — no longer thrown by the mock itself since
+ * AgentSession 0.83's beforeToolCall handles blocking before execute().
+ */
+export class ToolBlockedError extends Error {
+    toolBlocked = true;
+    constructor(reason) {
+        super(reason);
+        this.name = "ToolBlockedError";
+    }
+}
+/**
+ * Returns true if `err` represents a hook-based tool block.
+ *
+ * Kept for consumers that catch errors from tool execution flows, though
+ * AgentSession 0.83's beforeToolCall blocks before execute() is reached.
+ */
+export function isBlockedError(err) {
+    if (err instanceof ToolBlockedError)
+        return true;
+    if (err instanceof Error) {
+        const msg = err.message;
+        return (msg.includes("blocked") ||
+            msg.includes("Plan mode") ||
+            msg.includes("WRITE operation"));
+    }
+    return false;
+}
+function normalizeMockResult(handler, params) {
+    let raw;
+    if (typeof handler === "string") {
+        raw = handler;
+    }
+    else if (typeof handler === "function") {
+        raw = handler(params);
+    }
+    else {
+        raw = handler;
+    }
+    if (typeof raw === "string") {
+        return {
+            content: [{ type: "text", text: raw }],
+            details: {},
+        };
+    }
+    return raw;
+}
+/**
+ * Intercept tool execution for mocked tools.
+ *
+ * Unlike the old approach, this does NOT emit tool_call/tool_result hooks
+ * manually — AgentSession 0.83's beforeToolCall/afterToolCall handles that.
+ * The mock only replaces execute() to return controlled values. Result
+ * recording is handled by the session subscriber from tool_execution_end
+ * events (which carry the final afterToolCall-modified result).
+ *
+ * Returns a Set of mocked tool names for the session subscriber to set the
+ * mocked flag on recorded results, plus a Set of toolCallIds whose mock
+ * returned a ToolResult with isError:true. Pi 0.84's agent loop hardcodes
+ * successful execute() as non-error (isError:false), so the subscriber must
+ * consult this set to preserve the mock's error intent in collected records.
+ */
+export function interceptToolExecution(tools, mockTools, playbookState, propagateErrors) {
+    const mockedNames = new Set(Object.keys(mockTools));
+    const mockedErrorToolCallIds = new Set();
+    const wrapped = tools.map((tool) => {
+        const mockHandler = mockTools[tool.name];
+        if (!mockHandler) {
+            return wrapForCollection(tool, playbookState, propagateErrors);
+        }
+        return {
+            ...tool,
+            execute: async (toolCallId, params, _signal, _onUpdate) => {
+                const result = normalizeMockResult(mockHandler, params);
+                const text = result.content
+                    .filter((c) => c.type === "text")
+                    .map((c) => c.text)
+                    .join("\n");
+                if (result.isError) {
+                    // Pi 0.84 hardcodes successful execute() as non-error; remember the
+                    // toolCallId so the session subscriber can flag the record.
+                    mockedErrorToolCallIds.add(toolCallId);
+                }
+                // fireThenCallback fires synchronously so .then() sees real data
+                fireThenCallback(playbookState, toolCallId, {
+                    step: playbookState.consumed,
+                    toolName: tool.name,
+                    toolCallId,
+                    text,
+                    content: result.content,
+                    isError: result.isError ?? false,
+                    details: result.details,
+                    mocked: true,
+                });
+                return {
+                    content: result.content,
+                    details: result.details ?? {},
+                };
+            },
+        };
+    });
+    return { tools: wrapped, mockedNames, mockedErrorToolCallIds };
+}
+/**
+ * Wrap a real tool for event collection (non-mocked tools).
+ * Does not push to toolResults — the session subscriber handles recording
+ * from tool_execution_end events.
+ */
+function wrapForCollection(tool, playbookState, propagateErrors) {
+    const originalExecute = tool.execute;
+    return {
+        ...tool,
+        execute: async (toolCallId, params, signal, onUpdate) => {
+            const step = playbookState.consumed;
+            try {
+                const result = await originalExecute.call(tool, toolCallId, params, signal, onUpdate);
+                const text = (result.content ?? [])
+                    .filter((c) => c.type === "text")
+                    .map((c) => c.text)
+                    .join("\n");
+                fireThenCallback(playbookState, toolCallId, {
+                    step,
+                    toolName: tool.name,
+                    toolCallId,
+                    text,
+                    content: result.content ?? [],
+                    isError: !!result.isError,
+                    details: result.details,
+                    mocked: false,
+                });
+                return result;
+            }
+            catch (err) {
+                const errMsg = err instanceof Error ? err.message : String(err);
+                try {
+                    fireThenCallback(playbookState, toolCallId, {
+                        step,
+                        toolName: tool.name,
+                        toolCallId,
+                        text: errMsg,
+                        content: [{ type: "text", text: errMsg }],
+                        isError: true,
+                        details: undefined,
+                        mocked: false,
+                    });
+                }
+                catch {
+                    /* best-effort callback */
+                }
+                if (propagateErrors) {
+                    const diagnostic = formatToolError(step, tool.name, err);
+                    throw new Error(diagnostic, { cause: err });
+                }
+                return {
+                    content: [{ type: "text", text: errMsg }],
+                    details: {},
+                    isError: true,
+                };
+            }
+        },
+    };
+}
+function fireThenCallback(state, toolCallId, record) {
+    const callback = state.pendingCallbacks.get(toolCallId) ??
+        state.pendingCallbacks.get(record.toolName);
+    const key = state.pendingCallbacks.has(toolCallId)
+        ? toolCallId
+        : record.toolName;
+    if (callback) {
+        state.pendingCallbacks.delete(key);
+        try {
+            callback(record);
+        }
+        catch (err) {
+            console.warn(`[pi-test-harness] .then() callback error for ${record.toolName}: ${err}`);
+        }
+    }
+}
+//# sourceMappingURL=mock-tools.js.map

package/dist/mock-tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock-tools.js","sourceRoot":"","sources":["../src/mock-tools.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAKH,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAEnD;;;;GAIG;AACH,MAAM,OAAO,gBAAiB,SAAQ,KAAK;IACjC,WAAW,GAAG,IAAa,CAAC;IAErC,YAAY,MAAc;QACzB,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,IAAI,GAAG,kBAAkB,CAAC;IAChC,CAAC;CACD;AAED;;;;;GAKG;AACH,MAAM,UAAU,cAAc,CAAC,GAAY;IAC1C,IAAI,GAAG,YAAY,gBAAgB;QAAE,OAAO,IAAI,CAAC;IACjD,IAAI,GAAG,YAAY,KAAK,EAAE,CAAC;QAC1B,MAAM,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC;QACxB,OAAO,CACN,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC;YACvB,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC;YACzB,GAAG,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAC/B,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC;AAED,SAAS,mBAAmB,CAC3B,OAAwB,EACxB,MAA+B;IAE/B,IAAI,GAAwB,CAAC;IAE7B,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QACjC,GAAG,GAAG,OAAO,CAAC;IACf,CAAC;SAAM,IAAI,OAAO,OAAO,KAAK,UAAU,EAAE,CAAC;QAC1C,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IACvB,CAAC;SAAM,CAAC;QACP,GAAG,GAAG,OAAO,CAAC;IACf,CAAC;IAED,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;QAC7B,OAAO;YACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;YACtC,OAAO,EAAE,EAAE;SACX,CAAC;IACH,CAAC;IAED,OAAO,GAAG,CAAC;AACZ,CAAC;AAED;;;;;;;;;;;;;;GAcG;AACH,MAAM,UAAU,sBAAsB,CACrC,KAAkB,EAClB,SAA0C,EAC1C,aAA4B,EAC5B,eAAwB;IAMxB,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;IACpD,MAAM,sBAAsB,GAAG,IAAI,GAAG,EAAU,CAAC;IAEjD,MAAM,OAAO,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QAClC,MAAM,WAAW,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,CAAC,WAAW,EAAE,CAAC;YAClB,OAAO,iBAAiB,CAAC,IAAI,EAAE,aAAa,EAAE,eAAe,CAAC,CAAC;QAChE,CAAC;QAED,OAAO;YACN,GAAG,IAAI;YACP,OAAO,EAAE,KAAK,EACb,UAAkB,EAClB,MAA+B,EAC/B,OAAqB,EACrB,SAAe,EACd,EAAE;gBACH,MAAM,MAAM,GAAG,mBAAmB,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;gBACxD,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO;qBACzB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;qBAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;qBAClB,IAAI,CAAC,IAAI,CAAC,CAAC;gBACb,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;oBACpB,oEAAoE;oBACpE,4DAA4D;oBAC5D,sBAAsB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBACxC,CAAC;gBACD,iEAAiE;gBACjE,gBAAgB,CAAC,aAAa,EAAE,UAAU,EAAE;oBAC3C,IAAI,EAAE,aAAa,CAAC,QAAQ;oBAC5B,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,UAAU;oBACV,IAAI;oBACJ,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,KAAK;oBAChC,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,MAAM,EAAE,IAAI;iBACZ,CAAC,CAAC;gBAEH,OAAO;oBACN,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,EAAE;iBAC7B,CAAC;YACH,CAAC;SACY,CAAC;IAChB,CAAC,CAAC,CAAC;IAEH,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,sBAAsB,EAAE,CAAC;AAChE,CAAC;AAED;;;;GAIG;AACH,SAAS,iBAAiB,CACzB,IAAe,EACf,aAA4B,EAC5B,eAAwB;IAExB,MAAM,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC;IAErC,OAAO;QACN,GAAG,IAAI;QACP,OAAO,EAAE,KAAK,EACb,UAAkB,EAClB,MAA+B,EAC/B,MAAoB,EACpB,QAAc,EACb,EAAE;YACH,MAAM,IAAI,GAAG,aAAa,CAAC,QAAQ,CAAC;YAEpC,IAAI,CAAC;gBACJ,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,IAAI,CACxC,IAAI,EACJ,UAAU,EACV,MAAM,EACN,MAAM,EACN,QAAQ,CACR,CAAC;gBAEF,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;qBACjC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;qBACrC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;qBACvB,IAAI,CAAC,IAAI,CAAC,CAAC;gBAEb,gBAAgB,CAAC,aAAa,EAAE,UAAU,EAAE;oBAC3C,IAAI;oBACJ,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,UAAU;oBACV,IAAI;oBACJ,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,EAAE;oBAC7B,OAAO,EAAE,CAAC,CAAE,MAAc,CAAC,OAAO;oBAClC,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,MAAM,EAAE,KAAK;iBACb,CAAC,CAAC;gBAEH,OAAO,MAAM,CAAC;YACf,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACd,MAAM,MAAM,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAEhE,IAAI,CAAC;oBACJ,gBAAgB,CAAC,aAAa,EAAE,UAAU,EAAE;wBAC3C,IAAI;wBACJ,QAAQ,EAAE,IAAI,CAAC,IAAI;wBACnB,UAAU;wBACV,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;wBACzC,OAAO,EAAE,IAAI;wBACb,OAAO,EAAE,SAAS;wBAClB,MAAM,EAAE,KAAK;qBACb,CAAC,CAAC;gBACJ,CAAC;gBAAC,MAAM,CAAC;oBACR,0BAA0B;gBAC3B,CAAC;gBAED,IAAI,eAAe,EAAE,CAAC;oBACrB,MAAM,UAAU,GAAG,eAAe,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;oBACzD,MAAM,IAAI,KAAK,CAAC,UAAU,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;gBAC7C,CAAC;gBAED,OAAO;oBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;oBACzC,OAAO,EAAE,EAAE;oBACX,OAAO,EAAE,IAAI;iBACb,CAAC;YACH,CAAC;QACF,CAAC;KACY,CAAC;AAChB,CAAC;AAED,SAAS,gBAAgB,CACxB,KAAoB,EACpB,UAAkB,EAClB,MAAwB;IAExB,MAAM,QAAQ,GACb,KAAK,CAAC,gBAAgB,CAAC,GAAG,CAAC,UAAU,CAAC;QACtC,KAAK,CAAC,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC7C,MAAM,GAAG,GAAG,KAAK,CAAC,gBAAgB,CAAC,GAAG,CAAC,UAAU,CAAC;QACjD,CAAC,CAAC,UAAU;QACZ,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC;IACnB,IAAI,QAAQ,EAAE,CAAC;QACd,KAAK,CAAC,gBAAgB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACnC,IAAI,CAAC;YACJ,QAAQ,CAAC,MAAM,CAAC,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACd,OAAO,CAAC,IAAI,CACX,gDAAgD,MAAM,CAAC,QAAQ,KAAK,GAAG,EAAE,CACzE,CAAC;QACH,CAAC;IACF,CAAC;AACF,CAAC"}

package/dist/mock-ui.d.ts

@@ -0,0 +1,13 @@
+/**
+ * Mock UI context — intercepts ctx.ui.* calls from extensions.
+ * All calls are collected for assertions. Interactive methods return
+ * configured mock responses.
+ */
+import type { ExtensionUIContext } from "@earendil-works/pi-coding-agent";
+import type { MockUIConfig, UICallRecord } from "./types.js";
+/**
+ * Create a mock ExtensionUIContext that records all calls and returns
+ * configured responses.
+ */
+export declare function createMockUIContext(config: MockUIConfig | undefined, uiLog: UICallRecord[]): ExtensionUIContext;
+//# sourceMappingURL=mock-ui.d.ts.map

package/dist/mock-ui.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock-ui.d.ts","sourceRoot":"","sources":["../src/mock-ui.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,iCAAiC,CAAC;AAC1E,OAAO,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE7D;;;GAGG;AACH,wBAAgB,mBAAmB,CAClC,MAAM,EAAE,YAAY,YAAK,EACzB,KAAK,EAAE,YAAY,EAAE,GACnB,kBAAkB,CAmLpB"}

package/dist/mock-ui.js

@@ -0,0 +1,159 @@
+/**
+ * Mock UI context — intercepts ctx.ui.* calls from extensions.
+ * All calls are collected for assertions. Interactive methods return
+ * configured mock responses.
+ */
+/**
+ * Create a mock ExtensionUIContext that records all calls and returns
+ * configured responses.
+ */
+export function createMockUIContext(config = {}, uiLog) {
+    function record(method, args, returnValue) {
+        uiLog.push({ method, args, returnValue });
+    }
+    const mockUI = {
+        async select(title, options, _opts) {
+            let result;
+            const handler = config.select;
+            if (handler === undefined || handler === null) {
+                result = options[0]; // default: first item
+            }
+            else if (typeof handler === "number") {
+                result = options[handler];
+            }
+            else if (typeof handler === "string") {
+                result = options.find((o) => o === handler) ?? options[0];
+            }
+            else if (typeof handler === "function") {
+                result = handler(title, options);
+            }
+            record("select", [title, options], result);
+            return result;
+        },
+        async confirm(title, message, _opts) {
+            let result;
+            const handler = config.confirm;
+            if (handler === undefined || handler === null) {
+                result = true; // default: approve
+            }
+            else if (typeof handler === "boolean") {
+                result = handler;
+            }
+            else if (typeof handler === "function") {
+                result = handler(title, message);
+            }
+            else {
+                result = true;
+            }
+            record("confirm", [title, message], result);
+            return result;
+        },
+        async input(title, placeholder, _opts) {
+            let result;
+            const handler = config.input;
+            if (handler === undefined || handler === null) {
+                result = "";
+            }
+            else if (typeof handler === "string") {
+                result = handler;
+            }
+            else if (typeof handler === "function") {
+                result = handler(title, placeholder);
+            }
+            record("input", [title, placeholder], result);
+            return result;
+        },
+        async editor(title, prefill) {
+            let result;
+            const handler = config.editor;
+            if (handler === undefined || handler === null) {
+                result = "";
+            }
+            else if (typeof handler === "string") {
+                result = handler;
+            }
+            else if (typeof handler === "function") {
+                result = handler(title, prefill);
+            }
+            record("editor", [title, prefill], result);
+            return result;
+        },
+        notify(message, type) {
+            record("notify", [message, type]);
+        },
+        onTerminalInput() {
+            return () => { };
+        },
+        setStatus(key, text) {
+            record("setStatus", [key, text]);
+        },
+        setWorkingMessage(message) {
+            record("setWorkingMessage", [message]);
+        },
+        setWorkingVisible(visible) {
+            record("setWorkingVisible", [visible]);
+        },
+        setWorkingIndicator(options) {
+            record("setWorkingIndicator", [options]);
+        },
+        setHiddenThinkingLabel(label) {
+            record("setHiddenThinkingLabel", [label]);
+        },
+        setWidget(key, content, _options) {
+            record("setWidget", [key, content]);
+        },
+        setFooter(...args) {
+            record("setFooter", args);
+        },
+        setHeader(...args) {
+            record("setHeader", args);
+        },
+        setTitle(title) {
+            record("setTitle", [title]);
+        },
+        async custom() {
+            return undefined;
+        },
+        pasteToEditor(text) {
+            record("pasteToEditor", [text]);
+        },
+        setEditorText(text) {
+            record("setEditorText", [text]);
+        },
+        getEditorText() {
+            return "";
+        },
+        setEditorComponent(factory) {
+            record("setEditorComponent", [factory]);
+        },
+        addAutocompleteProvider(factory) {
+            record("addAutocompleteProvider", [factory]);
+        },
+        getEditorComponent() {
+            return undefined;
+        },
+        get theme() {
+            return {
+                fg: (_color, text) => text,
+                bold: (text) => text,
+                italic: (text) => text,
+                strikethrough: (text) => text,
+            };
+        },
+        getAllThemes() {
+            return [];
+        },
+        getTheme() {
+            return undefined;
+        },
+        setTheme() {
+            return { success: false, error: "Test mode" };
+        },
+        getToolsExpanded() {
+            return false;
+        },
+        setToolsExpanded() { },
+    };
+    return mockUI;
+}
+//# sourceMappingURL=mock-ui.js.map

package/dist/mock-ui.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock-ui.js","sourceRoot":"","sources":["../src/mock-ui.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH;;;GAGG;AACH,MAAM,UAAU,mBAAmB,CAClC,SAAuB,EAAE,EACzB,KAAqB;IAErB,SAAS,MAAM,CACd,MAAc,EACd,IAAe,EACf,WAAqB;QAErB,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,MAAM,GAAuB;QAClC,KAAK,CAAC,MAAM,CACX,KAAa,EACb,OAAiB,EACjB,KAAW;YAEX,IAAI,MAA0B,CAAC;YAC/B,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC;YAC9B,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;gBAC/C,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,sBAAsB;YAC5C,CAAC;iBAAM,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACxC,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;YAC3B,CAAC;iBAAM,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACxC,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,OAAO,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC;YAC3D,CAAC;iBAAM,IAAI,OAAO,OAAO,KAAK,UAAU,EAAE,CAAC;gBAC1C,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAClC,CAAC;YACD,MAAM,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,CAAC;YAC3C,OAAO,MAAM,CAAC;QACf,CAAC;QAED,KAAK,CAAC,OAAO,CACZ,KAAa,EACb,OAAe,EACf,KAAW;YAEX,IAAI,MAAe,CAAC;YACpB,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;YAC/B,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;gBAC/C,MAAM,GAAG,IAAI,CAAC,CAAC,mBAAmB;YACnC,CAAC;iBAAM,IAAI,OAAO,OAAO,KAAK,SAAS,EAAE,CAAC;gBACzC,MAAM,GAAG,OAAO,CAAC;YAClB,CAAC;iBAAM,IAAI,OAAO,OAAO,KAAK,UAAU,EAAE,CAAC;gBAC1C,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAClC,CAAC;iBAAM,CAAC;gBACP,MAAM,GAAG,IAAI,CAAC;YACf,CAAC;YACD,MAAM,CAAC,SAAS,EAAE,CAAC,KAAK,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,CAAC;YAC5C,OAAO,MAAM,CAAC;QACf,CAAC;QAED,KAAK,CAAC,KAAK,CACV,KAAa,EACb,WAAoB,EACpB,KAAW;YAEX,IAAI,MAA0B,CAAC;YAC/B,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC;YAC7B,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;gBAC/C,MAAM,GAAG,EAAE,CAAC;YACb,CAAC;iBAAM,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACxC,MAAM,GAAG,OAAO,CAAC;YAClB,CAAC;iBAAM,IAAI,OAAO,OAAO,KAAK,UAAU,EAAE,CAAC;gBAC1C,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;YACtC,CAAC;YACD,MAAM,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,WAAW,CAAC,EAAE,MAAM,CAAC,CAAC;YAC9C,OAAO,MAAM,CAAC;QACf,CAAC;QAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,OAAgB;YAC3C,IAAI,MAA0B,CAAC;YAC/B,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC;YAC9B,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;gBAC/C,MAAM,GAAG,EAAE,CAAC;YACb,CAAC;iBAAM,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACxC,MAAM,GAAG,OAAO,CAAC;YAClB,CAAC;iBAAM,IAAI,OAAO,OAAO,KAAK,UAAU,EAAE,CAAC;gBAC1C,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAClC,CAAC;YACD,MAAM,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,CAAC;YAC3C,OAAO,MAAM,CAAC;QACf,CAAC;QAED,MAAM,CAAC,OAAe,EAAE,IAAmC;YAC1D,MAAM,CAAC,QAAQ,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;QACnC,CAAC;QAED,eAAe;YACd,OAAO,GAAG,EAAE,GAAE,CAAC,CAAC;QACjB,CAAC;QAED,SAAS,CAAC,GAAW,EAAE,IAAwB;YAC9C,MAAM,CAAC,WAAW,EAAE,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC;QAClC,CAAC;QAED,iBAAiB,CAAC,OAAgB;YACjC,MAAM,CAAC,mBAAmB,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;QACxC,CAAC;QAED,iBAAiB,CAAC,OAAgB;YACjC,MAAM,CAAC,mBAAmB,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;QACxC,CAAC;QAED,mBAAmB,CAAC,OAGnB;YACA,MAAM,CAAC,qBAAqB,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;QAC1C,CAAC;QAED,sBAAsB,CAAC,KAAc;YACpC,MAAM,CAAC,wBAAwB,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3C,CAAC;QAED,SAAS,CAAC,GAAW,EAAE,OAAY,EAAE,QAAc;YAClD,MAAM,CAAC,WAAW,EAAE,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC,CAAC;QACrC,CAAC;QAED,SAAS,CAAC,GAAG,IAAe;YAC3B,MAAM,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;QAC3B,CAAC;QACD,SAAS,CAAC,GAAG,IAAe;YAC3B,MAAM,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;QAC3B,CAAC;QAED,QAAQ,CAAC,KAAa;YACrB,MAAM,CAAC,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;QAC7B,CAAC;QAED,KAAK,CAAC,MAAM;YACX,OAAO,SAAkB,CAAC;QAC3B,CAAC;QAED,aAAa,CAAC,IAAY;YACzB,MAAM,CAAC,eAAe,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;QACjC,CAAC;QACD,aAAa,CAAC,IAAY;YACzB,MAAM,CAAC,eAAe,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;QACjC,CAAC;QACD,aAAa;YACZ,OAAO,EAAE,CAAC;QACX,CAAC;QAED,kBAAkB,CAAC,OAAY;YAC9B,MAAM,CAAC,oBAAoB,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;QACzC,CAAC;QAED,uBAAuB,CAAC,OAAY;YACnC,MAAM,CAAC,yBAAyB,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;QAC9C,CAAC;QAED,kBAAkB;YACjB,OAAO,SAAS,CAAC;QAClB,CAAC;QAED,IAAI,KAAK;YACR,OAAO;gBACN,EAAE,EAAE,CAAC,MAAc,EAAE,IAAY,EAAE,EAAE,CAAC,IAAI;gBAC1C,IAAI,EAAE,CAAC,IAAY,EAAE,EAAE,CAAC,IAAI;gBAC5B,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE,CAAC,IAAI;gBAC9B,aAAa,EAAE,CAAC,IAAY,EAAE,EAAE,CAAC,IAAI;aACrC,CAAC;QACH,CAAC;QAED,YAAY;YACX,OAAO,EAAE,CAAC;QACX,CAAC;QACD,QAAQ;YACP,OAAO,SAAS,CAAC;QAClB,CAAC;QACD,QAAQ;YACP,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;QAC/C,CAAC;QACD,gBAAgB;YACf,OAAO,KAAK,CAAC;QACd,CAAC;QACD,gBAAgB,KAAU,CAAC;KAC3B,CAAC;IAEF,OAAO,MAAM,CAAC;AACf,CAAC"}

package/dist/pi-loader-parity.d.ts

@@ -0,0 +1,36 @@
+/**
+ * Extension-loader parity with Pi's shipped runtimes.
+ */
+/**
+ * Run `fn` with jiti's native import/require fast-path disabled.
+ *
+ * **Why this exists**: Pi loads every extension entrypoint through jiti
+ * (`loadExtensionModule` in `core/extensions/loader.ts`). Which jiti options it
+ * passes depends on how Pi itself is running:
+ *
+ * ```ts
+ * isBunBinary || isNodeSeaBinary || isBundledNode
+ *   ? { virtualModules, tryNative: false } // compiled binary / bundled Node
+ *   : isTypeScriptSourceRuntime
+ *     ? { virtualModules, tsconfigPaths: true }
+ *     : { alias: getAliases() } // unbundled Node build
+ * ```
+ *
+ * The last branch — the one an in-process harness always takes, because it
+ * imports the package's compiled entrypoint — leaves `tryNative` at jiti's
+ * default, and that default is "enabled if Bun is detected". Under a Bun test
+ * runner this combination hands Pi a factory for a module whose body has not
+ * finished evaluating: an entrypoint with a module-level `await import(...)`
+ * reports as loaded before its top-level `const` is initialized, so the hoisted
+ * default export Pi calls throws `Cannot access 'x' before initialization`.
+ *
+ * Pi's shipped runtimes never take that path, so the harness pins the choice
+ * they make (`tryNative: false`) for the duration of the load. This keeps
+ * `extensionFactories`-style and path-loaded extensions in exactly the loader
+ * configuration the real CLI uses, instead of one that varies by test runner.
+ *
+ * Nesting is reference-counted so two sessions loading concurrently in one
+ * process cannot have one restore the variable while the other is mid-load.
+ */
+export declare function withoutJitiNativeImport<T>(fn: () => Promise<T>): Promise<T>;
+//# sourceMappingURL=pi-loader-parity.d.ts.map

package/dist/pi-loader-parity.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pi-loader-parity.d.ts","sourceRoot":"","sources":["../src/pi-loader-parity.ts"],"names":[],"mappings":"AAAA;;GAEG;AAYH;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,wBAAsB,uBAAuB,CAAC,CAAC,EAC9C,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GAClB,OAAO,CAAC,CAAC,CAAC,CAaZ"}

package/dist/pi-loader-parity.js

@@ -0,0 +1,60 @@
+/**
+ * Extension-loader parity with Pi's shipped runtimes.
+ */
+/**
+ * Environment variable jiti reads when it decides whether to try the runtime's
+ * native require/import before applying its own transform.
+ * @internal
+ */
+const JITI_TRY_NATIVE = "JITI_TRY_NATIVE";
+/** Nesting depth of `withoutJitiNativeImport` calls in this process. */
+let suppressionDepth = 0;
+/**
+ * Run `fn` with jiti's native import/require fast-path disabled.
+ *
+ * **Why this exists**: Pi loads every extension entrypoint through jiti
+ * (`loadExtensionModule` in `core/extensions/loader.ts`). Which jiti options it
+ * passes depends on how Pi itself is running:
+ *
+ * ```ts
+ * isBunBinary || isNodeSeaBinary || isBundledNode
+ *   ? { virtualModules, tryNative: false } // compiled binary / bundled Node
+ *   : isTypeScriptSourceRuntime
+ *     ? { virtualModules, tsconfigPaths: true }
+ *     : { alias: getAliases() } // unbundled Node build
+ * ```
+ *
+ * The last branch — the one an in-process harness always takes, because it
+ * imports the package's compiled entrypoint — leaves `tryNative` at jiti's
+ * default, and that default is "enabled if Bun is detected". Under a Bun test
+ * runner this combination hands Pi a factory for a module whose body has not
+ * finished evaluating: an entrypoint with a module-level `await import(...)`
+ * reports as loaded before its top-level `const` is initialized, so the hoisted
+ * default export Pi calls throws `Cannot access 'x' before initialization`.
+ *
+ * Pi's shipped runtimes never take that path, so the harness pins the choice
+ * they make (`tryNative: false`) for the duration of the load. This keeps
+ * `extensionFactories`-style and path-loaded extensions in exactly the loader
+ * configuration the real CLI uses, instead of one that varies by test runner.
+ *
+ * Nesting is reference-counted so two sessions loading concurrently in one
+ * process cannot have one restore the variable while the other is mid-load.
+ */
+export async function withoutJitiNativeImport(fn) {
+    const previous = process.env[JITI_TRY_NATIVE];
+    suppressionDepth += 1;
+    process.env[JITI_TRY_NATIVE] = "0";
+    try {
+        return await fn();
+    }
+    finally {
+        suppressionDepth -= 1;
+        if (suppressionDepth === 0) {
+            if (previous === undefined)
+                delete process.env[JITI_TRY_NATIVE];
+            else
+                process.env[JITI_TRY_NATIVE] = previous;
+        }
+    }
+}
+//# sourceMappingURL=pi-loader-parity.js.map

package/dist/pi-loader-parity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pi-loader-parity.js","sourceRoot":"","sources":["../src/pi-loader-parity.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH;;;;GAIG;AACH,MAAM,eAAe,GAAG,iBAAiB,CAAC;AAE1C,wEAAwE;AACxE,IAAI,gBAAgB,GAAG,CAAC,CAAC;AAEzB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC5C,EAAoB;IAEpB,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;IAC9C,gBAAgB,IAAI,CAAC,CAAC;IACtB,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,GAAG,GAAG,CAAC;IACnC,IAAI,CAAC;QACJ,OAAO,MAAM,EAAE,EAAE,CAAC;IACnB,CAAC;YAAS,CAAC;QACV,gBAAgB,IAAI,CAAC,CAAC;QACtB,IAAI,gBAAgB,KAAK,CAAC,EAAE,CAAC;YAC5B,IAAI,QAAQ,KAAK,SAAS;gBAAE,OAAO,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;;gBAC3D,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,GAAG,QAAQ,CAAC;QAC9C,CAAC;IACF,CAAC;AACF,CAAC"}

package/dist/playbook.d.ts

@@ -0,0 +1,44 @@
+/**
+ * PlaybookStreamFn — replaces the model with scripted responses.
+ *
+ * The playbook is a queue of actions. Each streamFn call dequeues the next
+ * action and returns it as an AssistantMessageEventStream.
+ */
+import { type AssistantMessageEventStream, type Context, type Model, type SimpleStreamOptions } from "@earendil-works/pi-ai";
+import type { PlaybookAction, Turn, ToolResultRecord } from "./types.js";
+/** Chainable call action builder */
+declare class CallAction {
+    readonly action: PlaybookAction;
+    constructor(toolName: string, params: Record<string, unknown> | (() => Record<string, unknown>));
+    then(callback: (result: ToolResultRecord) => void): CallAction;
+}
+/**
+ * The model calls a tool.
+ * @param toolName Tool to call
+ * @param params Static params or function for late binding
+ */
+export declare function calls(toolName: string, params?: Record<string, unknown> | (() => Record<string, unknown>)): CallAction;
+/**
+ * The model emits text. Agent loop ends for this turn.
+ */
+export declare function says(text: string): PlaybookAction;
+/**
+ * Define one user→model turn.
+ * @param prompt The actual user prompt text
+ * @param actions What the model does in response (call/say sequence)
+ */
+export declare function when(prompt: string, actions: Array<CallAction | PlaybookAction>): Turn;
+export interface PlaybookState {
+    consumed: number;
+    remaining: number;
+    /** The action objects for each consumed step (for diagnostics) */
+    consumedActions: PlaybookAction[];
+    /** Callbacks pending for completed tool calls */
+    pendingCallbacks: Map<string, (result: ToolResultRecord) => void>;
+}
+export declare function createPlaybookStreamFn(turns: Turn[]): {
+    streamFn: (model: Model<any>, context: Context, options?: SimpleStreamOptions) => AssistantMessageEventStream;
+    state: PlaybookState;
+};
+export {};
+//# sourceMappingURL=playbook.d.ts.map

package/dist/playbook.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"playbook.d.ts","sourceRoot":"","sources":["../src/playbook.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAGN,KAAK,2BAA2B,EAChC,KAAK,OAAO,EACZ,KAAK,KAAK,EACV,KAAK,mBAAmB,EACxB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAKzE,oCAAoC;AACpC,cAAM,UAAU;IACf,QAAQ,CAAC,MAAM,EAAE,cAAc,CAAC;gBAEpB,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAI/F,IAAI,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,gBAAgB,KAAK,IAAI,GAAG,UAAU;CAI9D;AAED;;;;GAIG;AACH,wBAAgB,KAAK,CACpB,QAAQ,EAAE,MAAM,EAChB,MAAM,GAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAM,GACpE,UAAU,CAEZ;AAED;;GAEG;AACH,wBAAgB,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc,CAEjD;AAED;;;;GAIG;AACH,wBAAgB,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,UAAU,GAAG,cAAc,CAAC,GAAG,IAAI,CAKtF;AA4CD,MAAM,WAAW,aAAa;IAC7B,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,kEAAkE;IAClE,eAAe,EAAE,cAAc,EAAE,CAAC;IAClC,iDAAiD;IACjD,gBAAgB,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,MAAM,EAAE,gBAAgB,KAAK,IAAI,CAAC,CAAC;CAClE;AAED,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,IAAI,EAAE,GAAG;IACtD,QAAQ,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,mBAAmB,KAAK,2BAA2B,CAAC;IAC9G,KAAK,EAAE,aAAa,CAAC;CACrB,CAqEA"}