pulseed 0.1.5 → 0.2.0

package/dist/base/llm/codex-llm-client.js

@@ -0,0 +1,190 @@
+import * as fsp from "node:fs/promises";
+import * as os from "node:os";
+import * as path from "node:path";
+import { spawn } from "node:child_process";
+import { BaseLLMClient, MAX_RETRY_ATTEMPTS, RETRY_DELAYS_MS } from "./base-llm-client.js";
+import { sleep } from "../utils/sleep.js";
+import { LLMError } from "../utils/errors.js";
+// ─── Constants ───
+const DEFAULT_TIMEOUT_MS = 120_000; // 2 minutes per call
+/**
+ * Build a single prompt string from messages and system prompt.
+ * Format:
+ *   System instruction: <system>
+ *
+ *   user: <content>
+ *   assistant: <content>
+ *   ...
+ */
+function buildPrompt(messages, system) {
+    const parts = [];
+    if (system) {
+        parts.push(`System instruction: ${system}`);
+        parts.push("");
+    }
+    for (const msg of messages) {
+        parts.push(`${msg.role}: ${msg.content}`);
+    }
+    return parts.join("\n");
+}
+/**
+ * ILLMClient implementation that uses the `codex exec` CLI for LLM calls.
+ * Routes all PulSeed internal LLM calls through the Codex CLI, which uses
+ * the ChatGPT subscription (no separate API key needed).
+ *
+ * Uses `codex exec -s danger-full-access -o <tmpfile> "PROMPT"` per call.
+ * The -o flag writes the final response to a temp file for clean output.
+ * Usage stats are not available from the CLI and will always be 0.
+ *
+ * Set PULSEED_LLM_PROVIDER=codex to activate via CLIRunner / provider-factory.
+ */
+export class CodexLLMClient extends BaseLLMClient {
+    cliPath;
+    model;
+    repoPath;
+    timeoutMs;
+    constructor(config = {}) {
+        super();
+        this.cliPath = config.cliPath ?? "codex";
+        this.model = config.model;
+        this.lightModel = config.lightModel;
+        this.repoPath = config.repoPath?.trim() || ".";
+        this.timeoutMs = config.timeoutMs ?? DEFAULT_TIMEOUT_MS;
+    }
+    /**
+     * Send a message to the Codex CLI with retry logic.
+     * Retries up to MAX_RETRY_ATTEMPTS times with exponential backoff on spawn failures.
+     */
+    async sendMessage(messages, options) {
+        const model = this.resolveEffectiveModel(options?.model ?? this.model ?? "", options?.model_tier) || undefined;
+        const system = options?.system;
+        const prompt = buildPrompt(messages, system);
+        let lastError;
+        for (let attempt = 0; attempt < MAX_RETRY_ATTEMPTS; attempt++) {
+            try {
+                const content = await this._spawnCodex(prompt, model);
+                return {
+                    content,
+                    usage: {
+                        input_tokens: 0,
+                        output_tokens: 0,
+                    },
+                    stop_reason: "end_turn",
+                };
+            }
+            catch (err) {
+                lastError = err;
+                if (attempt < MAX_RETRY_ATTEMPTS - 1) {
+                    await sleep(RETRY_DELAYS_MS[attempt] ?? 1000);
+                }
+            }
+        }
+        throw lastError;
+    }
+    /** CodexLLMClient does not support function/tool calling. */
+    supportsToolCalling() { return false; }
+    /**
+     * Spawn `codex exec -s danger-full-access [-o <tmpfile>] [--model <model>] "PROMPT"`
+     * and return the response content read from the temp output file.
+     */
+    async _spawnCodex(prompt, model) {
+        // Create a temporary directory asynchronously to avoid blocking the event loop
+        const tmpDir = await fsp.mkdtemp(path.join(os.tmpdir(), "pulseed-codex-"));
+        const tmpFile = path.join(tmpDir, "response.txt");
+        return new Promise((resolve, reject) => {
+            // Build spawn args: exec -s danger-full-access -o <tmpfile> [--model <model>] -
+            // Prompt is sent via stdin (using "-" as positional arg) to avoid arg length limits.
+            // --path is not supported by codex-cli 0.114.0+; use cwd instead (see src/adapters/openai-codex.ts)
+            const spawnArgs = ["exec", "-s", "danger-full-access", "-o", tmpFile];
+            if (model) {
+                spawnArgs.push("--model", model);
+            }
+            spawnArgs.push("-");
+            const child = spawn(this.cliPath, spawnArgs, {
+                stdio: ["pipe", "pipe", "pipe"],
+                env: { ...process.env, TERM: "dumb" },
+                cwd: this.repoPath,
+            });
+            let timedOut = false;
+            let stderrData = "";
+            child.stderr.on("data", (chunk) => { stderrData += chunk.toString(); });
+            // Timeout: send SIGTERM, then SIGKILL after 5s
+            const timeoutHandle = setTimeout(() => {
+                timedOut = true;
+                child.kill("SIGTERM");
+                setTimeout(() => {
+                    try {
+                        child.kill("SIGKILL");
+                    }
+                    catch {
+                        // process already exited
+                    }
+                }, 5000);
+            }, this.timeoutMs);
+            // Suppress EPIPE errors on stdin
+            child.stdin.on("error", (err) => {
+                if (err.code !== "EPIPE")
+                    throw err;
+            });
+            // Write prompt via stdin and close
+            child.stdin.write(prompt);
+            child.stdin.end();
+            child.on("error", (err) => {
+                clearTimeout(timeoutHandle);
+                _cleanupTmp(tmpDir, tmpFile).catch((cleanupErr) => {
+                    console.debug("CodexLLMClient: _cleanupTmp failed (non-critical)", String(cleanupErr));
+                });
+                reject(new LLMError(`CodexLLMClient: spawn error — ${err.message}`));
+            });
+            child.on("close", (code) => {
+                clearTimeout(timeoutHandle);
+                if (timedOut) {
+                    _cleanupTmp(tmpDir, tmpFile).catch((cleanupErr) => {
+                        console.debug("CodexLLMClient: _cleanupTmp failed (non-critical)", String(cleanupErr));
+                    });
+                    reject(new LLMError(`CodexLLMClient: timed out after ${this.timeoutMs}ms`));
+                    return;
+                }
+                if (code !== 0) {
+                    _cleanupTmp(tmpDir, tmpFile).catch((cleanupErr) => {
+                        console.debug("CodexLLMClient: _cleanupTmp failed (non-critical)", String(cleanupErr));
+                    });
+                    const detail = stderrData.trim() ? ` — ${stderrData.trim().slice(0, 500)}` : "";
+                    reject(new LLMError(`CodexLLMClient: process exited with code ${code}${detail}`));
+                    return;
+                }
+                // Read response from temp file
+                fsp.readFile(tmpFile, "utf-8")
+                    .then((raw) => {
+                    _cleanupTmp(tmpDir, tmpFile).catch((cleanupErr) => {
+                        console.debug("CodexLLMClient: _cleanupTmp failed (non-critical)", String(cleanupErr));
+                    });
+                    resolve(raw.trim());
+                })
+                    .catch((readErr) => {
+                    _cleanupTmp(tmpDir, tmpFile).catch((cleanupErr) => {
+                        console.debug("CodexLLMClient: _cleanupTmp failed (non-critical)", String(cleanupErr));
+                    });
+                    reject(new LLMError(`CodexLLMClient: failed to read output file — ${String(readErr)}`));
+                });
+            });
+        });
+    }
+}
+// ─── Helpers ───
+async function _cleanupTmp(tmpDir, tmpFile) {
+    try {
+        await fsp.access(tmpFile);
+        await fsp.unlink(tmpFile);
+    }
+    catch {
+        // file may not exist — ignore
+    }
+    try {
+        await fsp.rmdir(tmpDir);
+    }
+    catch {
+        // best-effort cleanup
+    }
+}
+//# sourceMappingURL=codex-llm-client.js.map

package/dist/base/llm/codex-llm-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codex-llm-client.js","sourceRoot":"","sources":["../../../src/base/llm/codex-llm-client.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,GAAG,MAAM,kBAAkB,CAAC;AACxC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3C,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAO1F,OAAO,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAC1C,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAE9C,oBAAoB;AAEpB,MAAM,kBAAkB,GAAG,OAAO,CAAC,CAAC,qBAAqB;AAEzD;;;;;;;;GAQG;AACH,SAAS,WAAW,CAAC,QAAsB,EAAE,MAAe;IAC1D,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,IAAI,MAAM,EAAE,CAAC;QACX,KAAK,CAAC,IAAI,CAAC,uBAAuB,MAAM,EAAE,CAAC,CAAC;QAC5C,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACjB,CAAC;IAED,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,KAAK,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,IAAI,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;IAC5C,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAiBD;;;;;;;;;;GAUG;AACH,MAAM,OAAO,cAAe,SAAQ,aAAa;IAC9B,OAAO,CAAS;IAChB,KAAK,CAAqB;IAC1B,QAAQ,CAAS;IACjB,SAAS,CAAS;IAEnC,YAAY,SAA+B,EAAE;QAC3C,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,OAAO,CAAC;QACzC,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC1B,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;QACpC,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,GAAG,CAAC;QAC/C,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,kBAAkB,CAAC;IAC1D,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,WAAW,CACf,QAAsB,EACtB,OAA2B;QAE3B,MAAM,KAAK,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,EAAE,KAAK,IAAI,IAAI,CAAC,KAAK,IAAI,EAAE,EAAE,OAAO,EAAE,UAAU,CAAC,IAAI,SAAS,CAAC;QAC/G,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,CAAC;QAE/B,MAAM,MAAM,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAE7C,IAAI,SAAkB,CAAC;QAEvB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,kBAAkB,EAAE,OAAO,EAAE,EAAE,CAAC;YAC9D,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;gBACtD,OAAO;oBACL,OAAO;oBACP,KAAK,EAAE;wBACL,YAAY,EAAE,CAAC;wBACf,aAAa,EAAE,CAAC;qBACjB;oBACD,WAAW,EAAE,UAAU;iBACxB,CAAC;YACJ,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,SAAS,GAAG,GAAG,CAAC;gBAChB,IAAI,OAAO,GAAG,kBAAkB,GAAG,CAAC,EAAE,CAAC;oBACrC,MAAM,KAAK,CAAC,eAAe,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,CAAC;gBAChD,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,SAAS,CAAC;IAClB,CAAC;IAED,6DAA6D;IAC7D,mBAAmB,KAAc,OAAO,KAAK,CAAC,CAAC,CAAC;IAEhD;;;OAGG;IACK,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,KAAc;QACtD,+EAA+E;QAC/E,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,gBAAgB,CAAC,CAAC,CAAC;QAC3E,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;QAElD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAErC,gFAAgF;YAChF,qFAAqF;YACrF,oGAAoG;YACpG,MAAM,SAAS,GAAa,CAAC,MAAM,EAAE,IAAI,EAAE,oBAAoB,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;YAEhF,IAAI,KAAK,EAAE,CAAC;gBACV,SAAS,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YACnC,CAAC;YAED,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAEpB,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE;gBAC3C,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;gBAC/B,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE;gBACrC,GAAG,EAAE,IAAI,CAAC,QAAQ;aACnB,CAAC,CAAC;YAEH,IAAI,QAAQ,GAAG,KAAK,CAAC;YACrB,IAAI,UAAU,GAAG,EAAE,CAAC;YACpB,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,GAAG,UAAU,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAEhF,+CAA+C;YAC/C,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,EAAE;gBACpC,QAAQ,GAAG,IAAI,CAAC;gBAChB,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACtB,UAAU,CAAC,GAAG,EAAE;oBACd,IAAI,CAAC;wBACH,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;oBACxB,CAAC;oBAAC,MAAM,CAAC;wBACP,yBAAyB;oBAC3B,CAAC;gBACH,CAAC,EAAE,IAAI,CAAC,CAAC;YACX,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;YAEnB,iCAAiC;YACjC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAA0B,EAAE,EAAE;gBACrD,IAAI,GAAG,CAAC,IAAI,KAAK,OAAO;oBAAE,MAAM,GAAG,CAAC;YACtC,CAAC,CAAC,CAAC;YAEH,mCAAmC;YACnC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC1B,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;YAElB,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAU,EAAE,EAAE;gBAC/B,YAAY,CAAC,aAAa,CAAC,CAAC;gBAC5B,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,EAAE;oBAChD,OAAO,CAAC,KAAK,CAAC,mDAAmD,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC;gBACzF,CAAC,CAAC,CAAC;gBACH,MAAM,CAAC,IAAI,QAAQ,CAAC,iCAAiC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACvE,CAAC,CAAC,CAAC;YAEH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAmB,EAAE,EAAE;gBACxC,YAAY,CAAC,aAAa,CAAC,CAAC;gBAE5B,IAAI,QAAQ,EAAE,CAAC;oBACb,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,EAAE;wBAChD,OAAO,CAAC,KAAK,CAAC,mDAAmD,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC;oBACzF,CAAC,CAAC,CAAC;oBACH,MAAM,CACJ,IAAI,QAAQ,CACV,mCAAmC,IAAI,CAAC,SAAS,IAAI,CACtD,CACF,CAAC;oBACF,OAAO;gBACT,CAAC;gBAED,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;oBACf,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,EAAE;wBAChD,OAAO,CAAC,KAAK,CAAC,mDAAmD,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC;oBACzF,CAAC,CAAC,CAAC;oBACH,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,MAAM,UAAU,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAChF,MAAM,CACJ,IAAI,QAAQ,CACV,4CAA4C,IAAI,GAAG,MAAM,EAAE,CAC5D,CACF,CAAC;oBACF,OAAO;gBACT,CAAC;gBAED,+BAA+B;gBAC/B,GAAG,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;qBAC3B,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE;oBACZ,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,EAAE;wBAChD,OAAO,CAAC,KAAK,CAAC,mDAAmD,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC;oBACzF,CAAC,CAAC,CAAC;oBACH,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;gBACtB,CAAC,CAAC;qBACD,KAAK,CAAC,CAAC,OAAO,EAAE,EAAE;oBACjB,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,EAAE;wBAChD,OAAO,CAAC,KAAK,CAAC,mDAAmD,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC;oBACzF,CAAC,CAAC,CAAC;oBACH,MAAM,CACJ,IAAI,QAAQ,CACV,gDAAgD,MAAM,CAAC,OAAO,CAAC,EAAE,CAClE,CACF,CAAC;gBACJ,CAAC,CAAC,CAAC;YACP,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED,kBAAkB;AAElB,KAAK,UAAU,WAAW,CAAC,MAAc,EAAE,OAAe;IACxD,IAAI,CAAC;QACH,MAAM,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC1B,MAAM,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC5B,CAAC;IAAC,MAAM,CAAC;QACP,8BAA8B;IAChC,CAAC;IACD,IAAI,CAAC;QACH,MAAM,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC1B,CAAC;IAAC,MAAM,CAAC;QACP,sBAAsB;IACxB,CAAC;AACH,CAAC"}

package/dist/base/llm/json-sanitizer.d.ts

@@ -0,0 +1,34 @@
+/**
+ * json-sanitizer.ts — Shared LLM JSON response sanitization utilities.
+ *
+ * Centralizes all sanitization logic so it can be applied consistently
+ * before JSON.parse() and Zod validation.
+ */
+/**
+ * Sanitizes LLM-returned threshold_type strings to valid enum values.
+ * Handles the union of all known non-standard values from both
+ * GoalRefiner (leaf test) and GoalTreeManager (subgoal decomposition).
+ *
+ * Uses regex replacement so it works on raw JSON strings before parsing.
+ */
+export declare function sanitizeThresholdTypes(raw: string): string;
+/**
+ * Sanitizes LLM-returned threshold_value when threshold_type is "present".
+ * When the LLM returns an object (e.g. `{"type":"present"}`) as the value for
+ * a present threshold, replace it with null so downstream Zod schemas accept it.
+ *
+ * Operates on the raw JSON string before parsing to avoid any type-safety issues
+ * with the un-parsed LLM output.
+ */
+export declare function sanitizeThresholdValues(raw: string): string;
+/**
+ * Applies all LLM JSON sanitizers in sequence.
+ *
+ * Order:
+ *   1. Remove trailing commas (syntactic fix — must come before JSON.parse attempts)
+ *   2. Replace NaN/Infinity with null (syntactic fix)
+ *   3. Sanitize threshold_type enum values (domain-specific string replacement)
+ *   4. Sanitize threshold_value for "present" thresholds (requires parse+reserialize)
+ */
+export declare function sanitizeLLMJson(raw: string): string;
+//# sourceMappingURL=json-sanitizer.d.ts.map

package/dist/base/llm/json-sanitizer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"json-sanitizer.d.ts","sourceRoot":"","sources":["../../../src/base/llm/json-sanitizer.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAeH;;;;;;GAMG;AACH,wBAAgB,sBAAsB,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAS1D;AAED;;;;;;;GAOG;AACH,wBAAgB,uBAAuB,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAY3D;AAgDD;;;;;;;;GAQG;AACH,wBAAgB,eAAe,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAOnD"}

package/dist/base/llm/json-sanitizer.js

@@ -0,0 +1,110 @@
+/**
+ * json-sanitizer.ts — Shared LLM JSON response sanitization utilities.
+ *
+ * Centralizes all sanitization logic so it can be applied consistently
+ * before JSON.parse() and Zod validation.
+ */
+// ─── Threshold type sanitizer ───
+const THRESHOLD_TYPE_MAP = {
+    exact: "match",
+    scale: "min",
+    qualitative: "min",
+    boolean: "present",
+    percentage: "min",
+    count: "min",
+};
+const VALID_THRESHOLD_TYPES = new Set(["min", "max", "range", "present", "match"]);
+/**
+ * Sanitizes LLM-returned threshold_type strings to valid enum values.
+ * Handles the union of all known non-standard values from both
+ * GoalRefiner (leaf test) and GoalTreeManager (subgoal decomposition).
+ *
+ * Uses regex replacement so it works on raw JSON strings before parsing.
+ */
+export function sanitizeThresholdTypes(raw) {
+    return raw.replace(/"threshold_type"\s*:\s*"([^"]+)"/g, (_match, val) => {
+        if (VALID_THRESHOLD_TYPES.has(val))
+            return `"threshold_type": "${val}"`;
+        const mapped = THRESHOLD_TYPE_MAP[val] ?? "min";
+        return `"threshold_type": "${mapped}"`;
+    });
+}
+/**
+ * Sanitizes LLM-returned threshold_value when threshold_type is "present".
+ * When the LLM returns an object (e.g. `{"type":"present"}`) as the value for
+ * a present threshold, replace it with null so downstream Zod schemas accept it.
+ *
+ * Operates on the raw JSON string before parsing to avoid any type-safety issues
+ * with the un-parsed LLM output.
+ */
+export function sanitizeThresholdValues(raw) {
+    try {
+        const parsed = JSON.parse(raw);
+        const sanitized = sanitizePresentThresholdValues(parsed);
+        return JSON.stringify(sanitized);
+    }
+    catch (err) {
+        // JSON is not yet valid — syntactic fixups haven't fully resolved. This is expected
+        // when sanitizeThresholdValues runs as part of the sanitizeLLMJson pipeline.
+        console.debug("[sanitizeThresholdValues] JSON not yet parseable, skipping threshold cleanup");
+        void err;
+        return raw;
+    }
+}
+function sanitizePresentThresholdValues(value) {
+    if (Array.isArray(value)) {
+        return value.map(sanitizePresentThresholdValues);
+    }
+    if (value !== null && typeof value === "object") {
+        const record = value;
+        const result = {};
+        for (const [k, v] of Object.entries(record)) {
+            result[k] = sanitizePresentThresholdValues(v);
+        }
+        // If this object has threshold_type === "present" and threshold_value is an object, null it out.
+        if (result["threshold_type"] === "present" &&
+            typeof result["threshold_value"] === "object" &&
+            result["threshold_value"] !== null) {
+            result["threshold_value"] = null;
+        }
+        return result;
+    }
+    return value;
+}
+// ─── Common LLM drift pattern sanitizers ───
+/**
+ * Removes trailing commas before `}` or `]` — a common LLM formatting mistake.
+ * e.g. `{"a": 1,}` → `{"a": 1}`
+ */
+function removeTrailingCommas(raw) {
+    return raw.replace(/,\s*([}\]])/g, "$1");
+}
+/**
+ * Replaces JSON-invalid numeric literals produced by LLMs with null.
+ * Handles: NaN, Infinity, -Infinity (all unquoted).
+ */
+function replaceInvalidNumbers(raw) {
+    return raw
+        .replace(/\bNaN\b/g, "null")
+        .replace(/-Infinity\b/g, "null")
+        .replace(/\bInfinity\b/g, "null");
+}
+// ─── Main entry point ───
+/**
+ * Applies all LLM JSON sanitizers in sequence.
+ *
+ * Order:
+ *   1. Remove trailing commas (syntactic fix — must come before JSON.parse attempts)
+ *   2. Replace NaN/Infinity with null (syntactic fix)
+ *   3. Sanitize threshold_type enum values (domain-specific string replacement)
+ *   4. Sanitize threshold_value for "present" thresholds (requires parse+reserialize)
+ */
+export function sanitizeLLMJson(raw) {
+    let result = raw;
+    result = removeTrailingCommas(result);
+    result = replaceInvalidNumbers(result);
+    result = sanitizeThresholdTypes(result);
+    result = sanitizeThresholdValues(result);
+    return result;
+}
+//# sourceMappingURL=json-sanitizer.js.map

package/dist/base/llm/json-sanitizer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"json-sanitizer.js","sourceRoot":"","sources":["../../../src/base/llm/json-sanitizer.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,mCAAmC;AAEnC,MAAM,kBAAkB,GAA2B;IACjD,KAAK,EAAE,OAAO;IACd,KAAK,EAAE,KAAK;IACZ,WAAW,EAAE,KAAK;IAClB,OAAO,EAAE,SAAS;IAClB,UAAU,EAAE,KAAK;IACjB,KAAK,EAAE,KAAK;CACb,CAAC;AAEF,MAAM,qBAAqB,GAAG,IAAI,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC;AAEnF;;;;;;GAMG;AACH,MAAM,UAAU,sBAAsB,CAAC,GAAW;IAChD,OAAO,GAAG,CAAC,OAAO,CAChB,mCAAmC,EACnC,CAAC,MAAc,EAAE,GAAW,EAAE,EAAE;QAC9B,IAAI,qBAAqB,CAAC,GAAG,CAAC,GAAG,CAAC;YAAE,OAAO,sBAAsB,GAAG,GAAG,CAAC;QACxE,MAAM,MAAM,GAAG,kBAAkB,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC;QAChD,OAAO,sBAAsB,MAAM,GAAG,CAAC;IACzC,CAAC,CACF,CAAC;AACJ,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,uBAAuB,CAAC,GAAW;IACjD,IAAI,CAAC;QACH,MAAM,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxC,MAAM,SAAS,GAAG,8BAA8B,CAAC,MAAM,CAAC,CAAC;QACzD,OAAO,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IACnC,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,oFAAoF;QACpF,6EAA6E;QAC7E,OAAO,CAAC,KAAK,CAAC,8EAA8E,CAAC,CAAC;QAC9F,KAAK,GAAG,CAAC;QACT,OAAO,GAAG,CAAC;IACb,CAAC;AACH,CAAC;AAED,SAAS,8BAA8B,CAAC,KAAc;IACpD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IACnD,CAAC;IACD,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAChD,MAAM,MAAM,GAAG,KAAgC,CAAC;QAChD,MAAM,MAAM,GAA4B,EAAE,CAAC;QAC3C,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YAC5C,MAAM,CAAC,CAAC,CAAC,GAAG,8BAA8B,CAAC,CAAC,CAAC,CAAC;QAChD,CAAC;QACD,iGAAiG;QACjG,IACE,MAAM,CAAC,gBAAgB,CAAC,KAAK,SAAS;YACtC,OAAO,MAAM,CAAC,iBAAiB,CAAC,KAAK,QAAQ;YAC7C,MAAM,CAAC,iBAAiB,CAAC,KAAK,IAAI,EAClC,CAAC;YACD,MAAM,CAAC,iBAAiB,CAAC,GAAG,IAAI,CAAC;QACnC,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,8CAA8C;AAE9C;;;GAGG;AACH,SAAS,oBAAoB,CAAC,GAAW;IACvC,OAAO,GAAG,CAAC,OAAO,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;AAC3C,CAAC;AAED;;;GAGG;AACH,SAAS,qBAAqB,CAAC,GAAW;IACxC,OAAO,GAAG;SACP,OAAO,CAAC,UAAU,EAAE,MAAM,CAAC;SAC3B,OAAO,CAAC,cAAc,EAAE,MAAM,CAAC;SAC/B,OAAO,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;AACtC,CAAC;AAED,2BAA2B;AAE3B;;;;;;;;GAQG;AACH,MAAM,UAAU,eAAe,CAAC,GAAW;IACzC,IAAI,MAAM,GAAG,GAAG,CAAC;IACjB,MAAM,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;IACtC,MAAM,GAAG,qBAAqB,CAAC,MAAM,CAAC,CAAC;IACvC,MAAM,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;IACxC,MAAM,GAAG,uBAAuB,CAAC,MAAM,CAAC,CAAC;IACzC,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/base/llm/llm-client.d.ts

@@ -0,0 +1,93 @@
+import type { ZodSchema } from "zod";
+import { BaseLLMClient, DEFAULT_MAX_TOKENS, extractJSON } from "./base-llm-client.js";
+import type { ModelTier, ParseJSONMessage, ParseJSONOptions } from "./base-llm-client.js";
+import { GuardrailRunner } from "../../platform/traits/guardrail-runner.js";
+export interface LLMMessage {
+    role: "user" | "assistant";
+    content: string;
+}
+export type { ModelTier };
+export interface ToolDefinition {
+    type: "function";
+    function: {
+        name: string;
+        description: string;
+        parameters: Record<string, unknown>;
+    };
+}
+export interface ToolCallResult {
+    id: string;
+    function: {
+        name: string;
+        arguments: string;
+    };
+}
+export interface LLMRequestOptions {
+    model?: string;
+    max_tokens?: number;
+    system?: string;
+    temperature?: number;
+    /** Route to light model when configured. Defaults to 'main' for backward compat. */
+    model_tier?: ModelTier;
+    /** Tool definitions for function calling (tool use). */
+    tools?: ToolDefinition[];
+}
+export interface LLMResponse {
+    content: string;
+    usage: {
+        input_tokens: number;
+        output_tokens: number;
+    };
+    stop_reason: string;
+    /** Tool call results when the model invokes tools. */
+    tool_calls?: ToolCallResult[];
+}
+export interface ILLMClient {
+    sendMessage(messages: LLMMessage[], options?: LLMRequestOptions): Promise<LLMResponse>;
+    parseJSON<T>(content: string, schema: ZodSchema<T>): T;
+    parseJSON<T>(content: string, schema: ZodSchema<T>, options: ParseJSONOptions): Promise<T>;
+    /**
+     * Whether this client supports function/tool calling in sendMessage().
+     * CLI-wrapping clients (e.g., CodexLLMClient) that cannot handle tool
+     * definitions should override this to return false.
+     * When absent or returning true, the chat runner routes through executeWithTools.
+     */
+    supportsToolCalling?(): boolean;
+}
+export { extractJSON, DEFAULT_MAX_TOKENS };
+/**
+ * Thin wrapper around the Anthropic SDK.
+ * Provides retry logic and JSON extraction/validation.
+ *
+ * Constructor throws if no API key is provided.
+ */
+export declare class LLMClient extends BaseLLMClient implements ILLMClient {
+    private readonly client;
+    private guardrailRunner?;
+    private readonly defaultModel;
+    constructor(apiKey: string, guardrailRunner?: GuardrailRunner, lightModel?: string, model?: string);
+    /**
+     * Send a message to the Anthropic API with retry logic.
+     * Retries up to MAX_RETRY_ATTEMPTS times with exponential backoff.
+     * Retries up to RATE_LIMIT_RETRY_DELAYS_MS.length times on HTTP 429 with extended backoff.
+     */
+    sendMessage(messages: LLMMessage[], options?: LLMRequestOptions): Promise<LLMResponse>;
+    /**
+     * Low-level LLM call used by parseJSON retry logic.
+     * Sends messages and returns the raw response text.
+     */
+    protected callLLMRaw(messages: ParseJSONMessage[], systemPrompt?: string): Promise<string>;
+}
+/**
+ * Mock implementation for testing.
+ * Returns provided responses in order, tracking call count.
+ */
+export declare class MockLLMClient extends BaseLLMClient implements ILLMClient {
+    private readonly responses;
+    private _callCount;
+    constructor(responses: (string | LLMResponse)[]);
+    get callCount(): number;
+    sendMessage(_messages: LLMMessage[], _options?: LLMRequestOptions): Promise<LLMResponse>;
+    protected callLLMRaw(_messages: ParseJSONMessage[], _systemPrompt?: string): Promise<string>;
+}
+//# sourceMappingURL=llm-client.d.ts.map

package/dist/base/llm/llm-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm-client.d.ts","sourceRoot":"","sources":["../../../src/base/llm/llm-client.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,KAAK,CAAC;AAErC,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAqI,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACzN,OAAO,KAAK,EAAE,SAAS,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAE1F,OAAO,EAAE,eAAe,EAAE,MAAM,2CAA2C,CAAC;AAI5E,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;IAC3B,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,YAAY,EAAE,SAAS,EAAE,CAAC;AAE1B,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACrC,CAAC;CACH;AAED,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;CACH;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,oFAAoF;IACpF,UAAU,CAAC,EAAE,SAAS,CAAC;IACvB,wDAAwD;IACxD,KAAK,CAAC,EAAE,cAAc,EAAE,CAAC;CAC1B;AAED,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE;QACL,YAAY,EAAE,MAAM,CAAC;QACrB,aAAa,EAAE,MAAM,CAAC;KACvB,CAAC;IACF,WAAW,EAAE,MAAM,CAAC;IACpB,sDAAsD;IACtD,UAAU,CAAC,EAAE,cAAc,EAAE,CAAC;CAC/B;AAID,MAAM,WAAW,UAAU;IACzB,WAAW,CACT,QAAQ,EAAE,UAAU,EAAE,EACtB,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,WAAW,CAAC,CAAC;IACxB,SAAS,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IACvD,SAAS,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IAC3F;;;;;OAKG;IACH,mBAAmB,CAAC,IAAI,OAAO,CAAC;CACjC;AAQD,OAAO,EAAE,WAAW,EAAE,kBAAkB,EAAE,CAAC;AAI3C;;;;;GAKG;AACH,qBAAa,SAAU,SAAQ,aAAc,YAAW,UAAU;IAChE,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAY;IACnC,OAAO,CAAC,eAAe,CAAC,CAAkB;IAC1C,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAS;gBAE1B,MAAM,EAAE,MAAM,EAAE,eAAe,CAAC,EAAE,eAAe,EAAE,UAAU,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM;IAalG;;;;OAIG;IACG,WAAW,CACf,QAAQ,EAAE,UAAU,EAAE,EACtB,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,WAAW,CAAC;IAwIvB;;;OAGG;cACa,UAAU,CAAC,QAAQ,EAAE,gBAAgB,EAAE,EAAE,YAAY,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CAOjG;AAID;;;GAGG;AACH,qBAAa,aAAc,SAAQ,aAAc,YAAW,UAAU;IACpE,OAAO,CAAC,QAAQ,CAAC,SAAS,CAA2B;IACrD,OAAO,CAAC,UAAU,CAAa;gBAEnB,SAAS,EAAE,CAAC,MAAM,GAAG,WAAW,CAAC,EAAE;IAK/C,IAAI,SAAS,IAAI,MAAM,CAEtB;IAEK,WAAW,CACf,SAAS,EAAE,UAAU,EAAE,EACvB,QAAQ,CAAC,EAAE,iBAAiB,GAC3B,OAAO,CAAC,WAAW,CAAC;cA2BP,UAAU,CAAC,SAAS,EAAE,gBAAgB,EAAE,EAAE,aAAa,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CAInG"}

package/dist/base/llm/llm-client.js

@@ -0,0 +1,205 @@
+import Anthropic from "@anthropic-ai/sdk";
+import { sleep } from "../utils/sleep.js";
+import { BaseLLMClient, DEFAULT_MAX_TOKENS, DEFAULT_LLM_TIMEOUT_MS, MAX_RETRY_ATTEMPTS, RETRY_DELAYS_MS, RATE_LIMIT_RETRY_DELAYS_MS, isRateLimitError, getRateLimitRetryDelay, extractJSON } from "./base-llm-client.js";
+import { LLMError } from "../utils/errors.js";
+// ─── Constants ───
+const DEFAULT_MODEL = "claude-sonnet-4-20250514";
+const DEFAULT_TEMPERATURE = 0;
+// Re-export shared utilities for consumers that import from this module
+export { extractJSON, DEFAULT_MAX_TOKENS };
+// ─── LLMClient ───
+/**
+ * Thin wrapper around the Anthropic SDK.
+ * Provides retry logic and JSON extraction/validation.
+ *
+ * Constructor throws if no API key is provided.
+ */
+export class LLMClient extends BaseLLMClient {
+    client;
+    guardrailRunner;
+    defaultModel;
+    constructor(apiKey, guardrailRunner, lightModel, model) {
+        super();
+        if (!apiKey) {
+            throw new LLMError("LLMClient: no API key provided. Pass apiKey to constructor.");
+        }
+        this.client = new Anthropic({ apiKey });
+        this.guardrailRunner = guardrailRunner;
+        this.lightModel = lightModel;
+        this.defaultModel = model ?? DEFAULT_MODEL;
+    }
+    /**
+     * Send a message to the Anthropic API with retry logic.
+     * Retries up to MAX_RETRY_ATTEMPTS times with exponential backoff.
+     * Retries up to RATE_LIMIT_RETRY_DELAYS_MS.length times on HTTP 429 with extended backoff.
+     */
+    async sendMessage(messages, options) {
+        const model = this.resolveEffectiveModel(options?.model ?? this.defaultModel, options?.model_tier);
+        const max_tokens = options?.max_tokens ?? DEFAULT_MAX_TOKENS;
+        const temperature = options?.temperature ?? DEFAULT_TEMPERATURE;
+        let system = options?.system;
+        // before_model guardrail
+        if (this.guardrailRunner) {
+            const beforeResult = await this.guardrailRunner.run("before_model", {
+                checkpoint: "before_model",
+                input: { messages, options },
+                metadata: {},
+            });
+            if (!beforeResult.allowed) {
+                throw new Error(`Guardrail rejected: ${beforeResult.results.map((r) => r.reason).filter(Boolean).join("; ")}`);
+            }
+            if (beforeResult.modified_input) {
+                const modified = beforeResult.modified_input;
+                if (modified.messages)
+                    messages = modified.messages;
+                if (modified.system)
+                    system = modified.system;
+                if (modified.options)
+                    options = { ...options, ...modified.options };
+            }
+        }
+        let lastError;
+        let result;
+        let normalAttempts = 0;
+        let rateLimitAttempts = 0;
+        while (normalAttempts < MAX_RETRY_ATTEMPTS) {
+            try {
+                // Convert ToolDefinition[] to Anthropic SDK tool format
+                const anthropicTools = options?.tools?.map((t) => ({
+                    name: t.function.name,
+                    description: t.function.description,
+                    input_schema: {
+                        type: "object",
+                        ...t.function.parameters,
+                    },
+                }));
+                const response = await this.client.messages.create({
+                    model,
+                    max_tokens,
+                    temperature,
+                    ...(system ? { system } : {}),
+                    ...(anthropicTools?.length ? { tools: anthropicTools } : {}),
+                    messages: messages.map((m) => ({
+                        role: m.role,
+                        content: m.content,
+                    })),
+                }, { timeout: DEFAULT_LLM_TIMEOUT_MS });
+                // Extract text content from response
+                const textBlock = response.content.find((b) => b.type === "text");
+                const content = textBlock && "text" in textBlock ? textBlock.text : "";
+                // Extract tool_use blocks into ToolCallResult[]
+                const toolUseBlocks = response.content.filter((b) => b.type === "tool_use" && "id" in b && "name" in b && "input" in b);
+                const tool_calls = toolUseBlocks.length > 0
+                    ? toolUseBlocks.map((b) => ({
+                        id: b.id,
+                        function: {
+                            name: b.name,
+                            arguments: JSON.stringify(b.input),
+                        },
+                    }))
+                    : undefined;
+                result = {
+                    content,
+                    usage: {
+                        input_tokens: response.usage.input_tokens,
+                        output_tokens: response.usage.output_tokens,
+                    },
+                    stop_reason: response.stop_reason ?? "unknown",
+                    ...(tool_calls ? { tool_calls } : {}),
+                };
+                break;
+            }
+            catch (err) {
+                lastError = err;
+                // Rate limit: retry with extended backoff (does not count against normalAttempts)
+                if (isRateLimitError(err) && rateLimitAttempts < RATE_LIMIT_RETRY_DELAYS_MS.length) {
+                    await sleep(getRateLimitRetryDelay(err, rateLimitAttempts));
+                    rateLimitAttempts++;
+                    continue;
+                }
+                // Other 4xx client errors: throw immediately
+                if (err instanceof Error &&
+                    "status" in err &&
+                    typeof err.status === "number" &&
+                    err.status >= 400 &&
+                    err.status < 500) {
+                    throw err;
+                }
+                normalAttempts++;
+                if (normalAttempts < MAX_RETRY_ATTEMPTS) {
+                    await sleep(RETRY_DELAYS_MS[normalAttempts - 1] ?? 1000);
+                }
+            }
+        }
+        if (result === undefined) {
+            throw lastError;
+        }
+        // after_model guardrail
+        if (this.guardrailRunner) {
+            const afterResult = await this.guardrailRunner.run("after_model", {
+                checkpoint: "after_model",
+                input: { response: result, messages, options },
+                metadata: {},
+            });
+            if (!afterResult.allowed) {
+                throw new Error(`Guardrail rejected response: ${afterResult.results.map((r) => r.reason).filter(Boolean).join("; ")}`);
+            }
+            if (afterResult.modified_input !== undefined) {
+                result = afterResult.modified_input;
+            }
+        }
+        return result;
+    }
+    /**
+     * Low-level LLM call used by parseJSON retry logic.
+     * Sends messages and returns the raw response text.
+     */
+    async callLLMRaw(messages, systemPrompt) {
+        const llmMessages = messages
+            .filter((m) => m.role === "user" || m.role === "assistant")
+            .map((m) => ({ role: m.role, content: m.content }));
+        const response = await this.sendMessage(llmMessages, systemPrompt ? { system: systemPrompt } : undefined);
+        return response.content;
+    }
+}
+// ─── MockLLMClient ───
+/**
+ * Mock implementation for testing.
+ * Returns provided responses in order, tracking call count.
+ */
+export class MockLLMClient extends BaseLLMClient {
+    responses;
+    _callCount = 0;
+    constructor(responses) {
+        super();
+        this.responses = responses;
+    }
+    get callCount() {
+        return this._callCount;
+    }
+    async sendMessage(_messages, _options) {
+        const index = this._callCount;
+        this._callCount++;
+        if (index >= this.responses.length) {
+            throw new Error(`MockLLMClient: no response at index ${index} (only ${this.responses.length} responses configured)`);
+        }
+        const entry = this.responses[index];
+        // If the entry is already an LLMResponse, return it directly
+        if (typeof entry === "object") {
+            return entry;
+        }
+        return {
+            content: entry,
+            usage: {
+                input_tokens: 10,
+                output_tokens: entry.length,
+            },
+            stop_reason: "end_turn",
+        };
+    }
+    async callLLMRaw(_messages, _systemPrompt) {
+        const response = await this.sendMessage([]);
+        return response.content;
+    }
+}
+//# sourceMappingURL=llm-client.js.map