@spinabot/brigade 0.1.2 → 1.0.1

package/dist/agents/smart-compaction.js

@@ -0,0 +1,355 @@
+// Smart compaction support for Brigade.
+//
+// THREE responsibilities, all centred on keeping the context window healthy:
+//
+// 1. Tool-result truncation — bound any single tool's output to a safe share
+//    of the model's context window. A 10 MB grep result that lands in the
+//    transcript verbatim will OOM a small-context model on the next turn;
+//    head+tail truncation with a clear notice keeps the transcript usable
+//    while preserving the output's beginning and (if it looks important) end.
+//    [resolveToolResultMaxChars / truncateToolResultText — Brigade-native,
+//    primitive #1 era, has its own tests in smart-compaction.test.ts]
+//
+// 2. Compaction-window math — given the active model's context budget plus
+//    the running token usage, decide whether the next turn should compact
+//    before issuing the prompt. Pi 0.70.x manages compaction internally
+//    when a session is configured for it; this module provides the
+//    threshold + safe-floor helpers that the wrapper uses to decide
+//    whether to *recommend* a compaction up front.
+//    [evaluateCompactionDecision — Brigade-native, primitive #1 era]
+//
+// 3. Two-tier message-history compaction — walk the full message history,
+//    shrink oversized tool results (Pass 1), then if the aggregate sum still
+//    exceeds budget, shrink newest→oldest until under the cap (Pass 2).
+//    Used by the lifted v0.1.3 agent loop's transformContext hook.
+//    [smartCompactToolResults — folded in from src/core/smart-compaction.ts
+//    on 2026-05-08; the previous parallel implementation lived alongside the
+//    lifted v0.1.3 bundle. This file is now the single source of truth for
+//    all compaction concerns.]
+//
+// All limits are configurable via brigade.json
+// (`agents.defaults.contextLimits`); the defaults below are tuned to the
+// observed sweet spot across Anthropic / OpenAI / Google / Ollama models.
+// ─────────────────────────────────────────────────────────────────────────────
+// Tool-result truncation constants.
+//
+// A tool result is bounded by the smaller of:
+//   • a hard cap (`DEFAULT_MAX_LIVE_TOOL_RESULT_CHARS`)
+//   • a context-share cap (`MAX_TOOL_RESULT_CONTEXT_SHARE` × context window)
+// with a floor of `MIN_KEEP_CHARS` so 8k-context models still see something
+// useful from a tool that returned less than 2 KiB.
+// ─────────────────────────────────────────────────────────────────────────────
+export const MAX_TOOL_RESULT_CONTEXT_SHARE = 0.30;
+export const DEFAULT_MAX_LIVE_TOOL_RESULT_CHARS = 16_000;
+export const MIN_KEEP_CHARS = 2_000;
+// Approx. chars per token across modern tokenizers (BPE-flavoured) — close
+// enough for budgeting; the exact rate doesn't matter for "should we
+// truncate" decisions, only for sizing the bucket.
+const APPROX_CHARS_PER_TOKEN = 4;
+const HEAD_TAIL_SPLIT_TAIL_RATIO = 0.30; // up to 30% of budget for tail
+const HEAD_TAIL_TAIL_BUDGET_CAP = 4_000; // never spend more than 4k chars on the tail
+const HEAD_TAIL_OMISSION_MARKER = "\n\n[… middle content omitted — head and tail preserved …]\n\n";
+// Matchers for "this tool result has an important tail" — error blocks,
+// summary lines, JSON-end braces, etc. When matched we use the head+tail
+// strategy; otherwise we keep the head only and discard the rest.
+const IMPORTANT_TAIL_PATTERNS = [
+    /\berror\b/i,
+    /\bfailed\b/i,
+    /\bexception\b/i,
+    /summary\s*:/i,
+    /\}\s*$/, // JSON object close near the end
+    /\]\s*$/, // JSON array close near the end
+];
+export function resolveToolResultMaxChars(args) {
+    const sharedCap = Math.floor(args.contextWindowTokens * APPROX_CHARS_PER_TOKEN * MAX_TOOL_RESULT_CONTEXT_SHARE);
+    const hardCap = args.hardCharOverride ?? DEFAULT_MAX_LIVE_TOOL_RESULT_CHARS;
+    // Floor — even at 8k context (~32k chars), a 30% share is ~9.6k. We want
+    // tools to surface at least 2k of useful output even on a tiny window.
+    return Math.max(MIN_KEEP_CHARS, Math.min(hardCap, sharedCap));
+}
+export function truncateToolResultText(args) {
+    const { text, maxChars } = args;
+    if (typeof text !== "string" || text.length <= maxChars) {
+        return { text, truncated: false, droppedChars: 0 };
+    }
+    // Reserve room for the truncation suffix so the suffix itself doesn't push
+    // us back over the limit.
+    const suffixSample = formatTruncationSuffix(text.length); // worst-case length
+    const budget = Math.max(MIN_KEEP_CHARS, maxChars - suffixSample.length);
+    if (hasImportantTail(text)) {
+        const tailBudget = Math.min(Math.floor(budget * HEAD_TAIL_SPLIT_TAIL_RATIO), HEAD_TAIL_TAIL_BUDGET_CAP);
+        const headBudget = Math.max(0, budget - tailBudget - HEAD_TAIL_OMISSION_MARKER.length);
+        const head = text.slice(0, headBudget);
+        const tail = text.slice(text.length - tailBudget);
+        const droppedChars = text.length - head.length - tail.length;
+        const out = head + HEAD_TAIL_OMISSION_MARKER + tail + formatTruncationSuffix(droppedChars);
+        return { text: out, truncated: true, droppedChars };
+    }
+    // No important tail signal: keep the head only.
+    const head = text.slice(0, budget);
+    const droppedChars = text.length - head.length;
+    return {
+        text: head + formatTruncationSuffix(droppedChars),
+        truncated: true,
+        droppedChars,
+    };
+}
+export function formatTruncationSuffix(droppedChars) {
+    const n = Math.max(1, Math.floor(droppedChars));
+    return `\n\n[… ${n} more characters truncated …]`;
+}
+function hasImportantTail(text) {
+    // Cheap test: only inspect the trailing 1 KiB. A "summary" or "error" block
+    // an entire context window away from the end isn't an important *tail*.
+    const sample = text.slice(Math.max(0, text.length - 1024));
+    for (const p of IMPORTANT_TAIL_PATTERNS)
+        if (p.test(sample))
+            return true;
+    return false;
+}
+// ─────────────────────────────────────────────────────────────────────────────
+// Compaction-window math. Pi triggers compaction internally when context
+// usage breaches its own threshold; this helper lets the wrapper *anticipate*
+// the trigger so a "we're going to compact next" log/heartbeat fires before
+// the user sees a multi-second pause.
+// ─────────────────────────────────────────────────────────────────────────────
+// Below this prompt-budget floor we refuse to compact and warn instead — at
+// 8k tokens of headroom there isn't enough room for the system prompt + a
+// summarisation instruction + a useful summary, so the right move is to
+// surface the overflow rather than blunder a compaction.
+export const MIN_PROMPT_BUDGET_TOKENS = 8_000;
+// What fraction of context must be free post-compaction to be worth doing?
+// Below this the compaction "succeeds" but the next turn immediately tips
+// over again.
+export const MIN_PROMPT_BUDGET_RATIO = 0.5;
+// Compaction is recommended once usage crosses this share of the window.
+export const COMPACTION_TRIGGER_RATIO = 0.85;
+export function evaluateCompactionDecision(args) {
+    const triggerThresholdTokens = Math.floor(args.contextWindowTokens * COMPACTION_TRIGGER_RATIO);
+    const promptBudgetTokens = Math.max(0, args.contextWindowTokens - args.estimatedUsageTokens);
+    if (args.estimatedUsageTokens < triggerThresholdTokens) {
+        return {
+            shouldRecommendCompaction: false,
+            triggerThresholdTokens,
+            promptBudgetTokens,
+            reason: "below-threshold",
+        };
+    }
+    if (promptBudgetTokens < MIN_PROMPT_BUDGET_TOKENS) {
+        return {
+            shouldRecommendCompaction: false,
+            triggerThresholdTokens,
+            promptBudgetTokens,
+            reason: "headroom-too-tight",
+        };
+    }
+    const targetFreeTokens = Math.floor(args.contextWindowTokens * MIN_PROMPT_BUDGET_RATIO);
+    if (promptBudgetTokens < targetFreeTokens) {
+        return {
+            shouldRecommendCompaction: true,
+            triggerThresholdTokens,
+            promptBudgetTokens,
+            reason: "headroom-tight",
+        };
+    }
+    return {
+        shouldRecommendCompaction: true,
+        triggerThresholdTokens,
+        promptBudgetTokens,
+        reason: "ready",
+    };
+}
+/**
+ * Resolve the active compaction provider via the slot config and call its
+ * `summarize` if pinned + registered; otherwise return `{fallback: true}`
+ * so the caller can fall back to the built-in head+tail truncation.
+ *
+ * No behaviour change today — Brigade ships no compaction-provider plugin
+ * and the in-tree compactor (`smartCompactToolResults` below) remains the
+ * single source of truth. This shim is the seam a future plugin slots into.
+ */
+export async function compactWithSlotResolution(args) {
+    const resolved = args.registry?.resolveSlot("compaction", args.config, args.registry.compactionProviders);
+    if (!resolved)
+        return { fallback: true };
+    return resolved.summarize({
+        messages: args.messages,
+        compressionRatio: args.compressionRatio,
+        signal: args.signal,
+    });
+}
+const DEFAULT_OVERSIZED_CAP = 16_000;
+const DEFAULT_AGGREGATE_CAP = 64_000;
+const CHARS_PER_TOKEN_ROUGH = 4;
+/**
+ * Sane FLOORS so the per-result and aggregate budgets never collapse to
+ * effectively-zero on tiny-context models (Cerebras 8K, Groq Llama-3.1-8B
+ * 8K). Without these, a 4K context window would derive a 1.2KB per-result
+ * cap and a 2KB aggregate budget — every tool result would shrink to almost
+ * nothing on every transformContext pass, destroying the model's working
+ * memory.
+ */
+const MIN_OVERSIZED_CAP = 2_000;
+const MIN_AGGREGATE_CAP = 4_000;
+/**
+ * Default context window assumption when the caller didn't pass one. We use
+ * a CONSERVATIVE 32K instead of "Anthropic Sonnet's 200K" so a missing
+ * value doesn't grant pathological budgets on a small model.
+ */
+const SAFE_DEFAULT_CONTEXT_TOKENS = 32_000;
+// Wider error-tail pattern set than `IMPORTANT_TAIL_PATTERNS` above —
+// includes Python tracebacks, segfault, panic, pytest summary lines.
+// Both pattern sets coexist so behaviour of the older
+// truncateToolResultText path is unchanged.
+const ERROR_TAIL_PATTERNS = [
+    /error\b/i,
+    /exception\b/i,
+    /traceback/i,
+    /\bfail(?:ed|ure)?\b/i,
+    /stack\s*trace/i,
+    /\bsegfault/i,
+    /panic\b/i,
+    /^.*(\d+\s+passed.*\d+\s+failed)/im, // pytest-style summary
+];
+/**
+ * Walk message history, shrinking tool-result text blocks per the two-tier
+ * algorithm. Returns a new array; never mutates the input.
+ *
+ * Image / non-text content blocks pass through untouched (truncating base64
+ * would corrupt them).
+ *
+ * `transformContext` callers should run this BEFORE sanitizeMessages so
+ * truncation markers added here don't get a surrogate-strip pass over them.
+ */
+export function smartCompactToolResults(messages, options = {}) {
+    if (!Array.isArray(messages) || messages.length === 0) {
+        return { messages, stats: { oversizedCount: 0, aggregateReducibleChars: 0, totalSavedChars: 0 } };
+    }
+    // Defensive: a missing or non-positive contextWindow falls to a SAFE
+    // 32K default, not the previous 200K. Otherwise an 8K-context Groq /
+    // Cerebras model would inherit Anthropic-Sonnet-sized budgets and never
+    // compact. Negative / zero / NaN all collapse to the safe default.
+    const requestedCtx = options.contextWindowTokens;
+    const ctxTokens = typeof requestedCtx === "number" && Number.isFinite(requestedCtx) && requestedCtx > 0
+        ? requestedCtx
+        : SAFE_DEFAULT_CONTEXT_TOKENS;
+    const ctxChars = ctxTokens * CHARS_PER_TOKEN_ROUGH;
+    // Per-result cap: 30% of context, capped at the oversized ceiling AND
+    // floored at MIN_OVERSIZED_CAP so a 4K model still gets ~2KB per result
+    // (enough to keep one bash output + one read result legible).
+    const maxCharsPerResult = options.maxCharsPerResult ??
+        Math.max(MIN_OVERSIZED_CAP, Math.min(Math.floor(ctxChars * 0.3), DEFAULT_OVERSIZED_CAP));
+    // Aggregate: 50% of context, capped + floored.
+    const aggregateBudget = options.aggregateBudgetChars ??
+        Math.max(MIN_AGGREGATE_CAP, Math.min(Math.floor(ctxChars * 0.5), DEFAULT_AGGREGATE_CAP));
+    // minKeep clamped so it can never EXCEED maxCharsPerResult — that would
+    // cause Pass 2 to try to shrink a result BELOW its already-applied cap,
+    // hitting an infinite "no progress" loop or, worse, growing it back.
+    const requestedMinKeep = options.minKeepChars ?? 2_000;
+    const minKeep = Math.min(maxCharsPerResult, Math.max(200, requestedMinKeep));
+    const preserveTail = options.preserveImportantTail ?? true;
+    const slots = [];
+    for (let mi = 0; mi < messages.length; mi++) {
+        const m = messages[mi];
+        if (m?.role !== "toolResult" || !Array.isArray(m.content))
+            continue;
+        for (let bi = 0; bi < m.content.length; bi++) {
+            const block = m.content[bi];
+            if (block?.type === "text" && typeof block.text === "string") {
+                slots.push({ mi, bi, len: block.text.length });
+            }
+        }
+    }
+    if (slots.length === 0) {
+        return { messages, stats: { oversizedCount: 0, aggregateReducibleChars: 0, totalSavedChars: 0 } };
+    }
+    // Plan reductions per slot (slot key → target length).
+    const targetLength = new Map();
+    const slotKey = (s) => `${s.mi}:${s.bi}`;
+    // PASS 1 — oversized singles get capped to maxCharsPerResult.
+    let oversizedCount = 0;
+    for (const s of slots) {
+        if (s.len > maxCharsPerResult) {
+            oversizedCount++;
+            targetLength.set(slotKey(s), maxCharsPerResult);
+        }
+    }
+    // PASS 2 — aggregate. If the SUM (using post-pass-1 lengths) still
+    // exceeds the aggregate budget, shrink starting from NEWEST to OLDEST
+    // (newer = higher mi) by the amount needed, with a minKeep floor.
+    const lengthAfterPass1 = (s) => targetLength.get(slotKey(s)) ?? s.len;
+    let aggregate = slots.reduce((sum, s) => sum + lengthAfterPass1(s), 0);
+    const aggregateReducibleChars = Math.max(0, aggregate - aggregateBudget);
+    if (aggregate > aggregateBudget) {
+        // Sort newest-first (higher message index first).
+        const ordered = [...slots].sort((a, b) => b.mi - a.mi || b.bi - a.bi);
+        for (const s of ordered) {
+            if (aggregate <= aggregateBudget)
+                break;
+            const current = lengthAfterPass1(s);
+            const reducible = Math.max(0, current - minKeep);
+            if (reducible === 0)
+                continue;
+            const need = aggregate - aggregateBudget;
+            const cut = Math.min(reducible, need);
+            targetLength.set(slotKey(s), current - cut);
+            aggregate -= cut;
+        }
+    }
+    // Apply the plan. If no slot was changed, return the input unchanged.
+    if (targetLength.size === 0) {
+        return {
+            messages,
+            stats: { oversizedCount: 0, aggregateReducibleChars, totalSavedChars: 0 },
+        };
+    }
+    let totalSaved = 0;
+    const changedMsgIndices = new Set([...targetLength.keys()].map((k) => Number(k.split(":")[0])));
+    const out = messages.map((msg, mi) => {
+        if (!changedMsgIndices.has(mi))
+            return msg;
+        const m = msg;
+        const newContent = m.content.map((block, bi) => {
+            const target = targetLength.get(`${mi}:${bi}`);
+            if (target === undefined)
+                return block;
+            if (block?.type !== "text" || typeof block.text !== "string")
+                return block;
+            if (block.text.length <= target)
+                return block;
+            const original = block.text;
+            const truncated = preserveTail && hasErrorPattern(original)
+                ? headAndTail(original, target)
+                : headOnly(original, target);
+            totalSaved += original.length - truncated.length;
+            return { ...block, text: truncated };
+        });
+        return { ...m, content: newContent };
+    });
+    return {
+        messages: out,
+        stats: { oversizedCount, aggregateReducibleChars, totalSavedChars: totalSaved },
+    };
+}
+/* ─────────── helpers for two-tier compaction (smartCompactToolResults) ─────────── */
+function hasErrorPattern(text) {
+    // Cheap check first — only scan the LAST 4KB where errors typically live.
+    const slice = text.length > 4_000 ? text.slice(-4_000) : text;
+    return ERROR_TAIL_PATTERNS.some((p) => p.test(slice));
+}
+function headOnly(text, targetLen) {
+    const marker = "\n\n⚠️ [...truncated...]\n";
+    const head = Math.max(0, targetLen - marker.length);
+    const cut = text.length - head;
+    return `${text.slice(0, head)}${marker} (${cut} chars removed)`;
+}
+function headAndTail(text, targetLen) {
+    const marker = "\n\n⚠️ [...middle truncated, tail preserved...]\n\n";
+    // 60% head, 40% tail (after marker overhead).
+    const usable = Math.max(0, targetLen - marker.length);
+    const headLen = Math.floor(usable * 0.6);
+    const tailLen = Math.max(0, usable - headLen);
+    const cut = text.length - headLen - tailLen;
+    return `${text.slice(0, headLen)}${marker}(${cut} chars removed)\n\n${text.slice(text.length - tailLen)}`;
+}
+//# sourceMappingURL=smart-compaction.js.map

package/dist/agents/smart-compaction.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"smart-compaction.js","sourceRoot":"","sources":["../../src/agents/smart-compaction.ts"],"names":[],"mappings":"AAAA,wCAAwC;AACxC,EAAE;AACF,6EAA6E;AAC7E,EAAE;AACF,6EAA6E;AAC7E,0EAA0E;AAC1E,0EAA0E;AAC1E,0EAA0E;AAC1E,8EAA8E;AAC9E,2EAA2E;AAC3E,sEAAsE;AACtE,EAAE;AACF,2EAA2E;AAC3E,0EAA0E;AAC1E,wEAAwE;AACxE,mEAAmE;AACnE,oEAAoE;AACpE,mDAAmD;AACnD,qEAAqE;AACrE,EAAE;AACF,0EAA0E;AAC1E,6EAA6E;AAC7E,wEAAwE;AACxE,mEAAmE;AACnE,4EAA4E;AAC5E,6EAA6E;AAC7E,2EAA2E;AAC3E,+BAA+B;AAC/B,EAAE;AACF,+CAA+C;AAC/C,yEAAyE;AACzE,0EAA0E;AAE1E,gFAAgF;AAChF,oCAAoC;AACpC,EAAE;AACF,8CAA8C;AAC9C,wDAAwD;AACxD,6EAA6E;AAC7E,4EAA4E;AAC5E,oDAAoD;AACpD,gFAAgF;AAEhF,MAAM,CAAC,MAAM,6BAA6B,GAAG,IAAI,CAAC;AAClD,MAAM,CAAC,MAAM,kCAAkC,GAAG,MAAM,CAAC;AACzD,MAAM,CAAC,MAAM,cAAc,GAAG,KAAK,CAAC;AAEpC,2EAA2E;AAC3E,qEAAqE;AACrE,mDAAmD;AACnD,MAAM,sBAAsB,GAAG,CAAC,CAAC;AAEjC,MAAM,0BAA0B,GAAG,IAAI,CAAC,CAAU,+BAA+B;AACjF,MAAM,yBAAyB,GAAG,KAAK,CAAC,CAAU,6CAA6C;AAC/F,MAAM,yBAAyB,GAC7B,gEAAgE,CAAC;AAEnE,wEAAwE;AACxE,yEAAyE;AACzE,kEAAkE;AAClE,MAAM,uBAAuB,GAAa;IACxC,YAAY;IACZ,aAAa;IACb,gBAAgB;IAChB,cAAc;IACd,QAAQ,EAAE,iCAAiC;IAC3C,QAAQ,EAAE,gCAAgC;CAC3C,CAAC;AAWF,MAAM,UAAU,yBAAyB,CAAC,IAAgC;IACxE,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAC1B,IAAI,CAAC,mBAAmB,GAAG,sBAAsB,GAAG,6BAA6B,CAClF,CAAC;IACF,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,IAAI,kCAAkC,CAAC;IAC5E,yEAAyE;IACzE,uEAAuE;IACvE,OAAO,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC;AAChE,CAAC;AAaD,MAAM,UAAU,sBAAsB,CAAC,IAA4B;IACjE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;IAChC,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,IAAI,QAAQ,EAAE,CAAC;QACxD,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;IACrD,CAAC;IAED,2EAA2E;IAC3E,0BAA0B;IAC1B,MAAM,YAAY,GAAG,sBAAsB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,oBAAoB;IAC9E,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,QAAQ,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;IAExE,IAAI,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;QAC3B,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CACzB,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,0BAA0B,CAAC,EAC/C,yBAAyB,CAC1B,CAAC;QACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,GAAG,UAAU,GAAG,yBAAyB,CAAC,MAAM,CAAC,CAAC;QACvF,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;QACvC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,UAAU,CAAC,CAAC;QAClD,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC7D,MAAM,GAAG,GAAG,IAAI,GAAG,yBAAyB,GAAG,IAAI,GAAG,sBAAsB,CAAC,YAAY,CAAC,CAAC;QAC3F,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;IACtD,CAAC;IAED,gDAAgD;IAChD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;IACnC,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;IAC/C,OAAO;QACL,IAAI,EAAE,IAAI,GAAG,sBAAsB,CAAC,YAAY,CAAC;QACjD,SAAS,EAAE,IAAI;QACf,YAAY;KACb,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,YAAoB;IACzD,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;IAChD,OAAO,UAAU,CAAC,+BAA+B,CAAC;AACpD,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAY;IACpC,4EAA4E;IAC5E,wEAAwE;IACxE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC;IAC3D,KAAK,MAAM,CAAC,IAAI,uBAAuB;QAAE,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;YAAE,OAAO,IAAI,CAAC;IACzE,OAAO,KAAK,CAAC;AACf,CAAC;AAED,gFAAgF;AAChF,yEAAyE;AACzE,8EAA8E;AAC9E,4EAA4E;AAC5E,sCAAsC;AACtC,gFAAgF;AAEhF,4EAA4E;AAC5E,0EAA0E;AAC1E,wEAAwE;AACxE,yDAAyD;AACzD,MAAM,CAAC,MAAM,wBAAwB,GAAG,KAAK,CAAC;AAC9C,2EAA2E;AAC3E,0EAA0E;AAC1E,cAAc;AACd,MAAM,CAAC,MAAM,uBAAuB,GAAG,GAAG,CAAC;AAC3C,yEAAyE;AACzE,MAAM,CAAC,MAAM,wBAAwB,GAAG,IAAI,CAAC;AAc7C,MAAM,UAAU,0BAA0B,CAAC,IAA4B;IACrE,MAAM,sBAAsB,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,mBAAmB,GAAG,wBAAwB,CAAC,CAAC;IAC/F,MAAM,kBAAkB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC,oBAAoB,CAAC,CAAC;IAE7F,IAAI,IAAI,CAAC,oBAAoB,GAAG,sBAAsB,EAAE,CAAC;QACvD,OAAO;YACL,yBAAyB,EAAE,KAAK;YAChC,sBAAsB;YACtB,kBAAkB;YAClB,MAAM,EAAE,iBAAiB;SAC1B,CAAC;IACJ,CAAC;IACD,IAAI,kBAAkB,GAAG,wBAAwB,EAAE,CAAC;QAClD,OAAO;YACL,yBAAyB,EAAE,KAAK;YAChC,sBAAsB;YACtB,kBAAkB;YAClB,MAAM,EAAE,oBAAoB;SAC7B,CAAC;IACJ,CAAC;IACD,MAAM,gBAAgB,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,mBAAmB,GAAG,uBAAuB,CAAC,CAAC;IACxF,IAAI,kBAAkB,GAAG,gBAAgB,EAAE,CAAC;QAC1C,OAAO;YACL,yBAAyB,EAAE,IAAI;YAC/B,sBAAsB;YACtB,kBAAkB;YAClB,MAAM,EAAE,gBAAgB;SACzB,CAAC;IACJ,CAAC;IACD,OAAO;QACL,yBAAyB,EAAE,IAAI;QAC/B,sBAAsB;QACtB,kBAAkB;QAClB,MAAM,EAAE,OAAO;KAChB,CAAC;AACJ,CAAC;AAoCD;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,IAAmC;IAEnC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE,WAAW,CACzC,YAAY,EACZ,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAClC,CAAC;IACF,IAAI,CAAC,QAAQ;QAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IACzC,OAAO,QAAQ,CAAC,SAAS,CAAC;QACxB,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;QACvC,MAAM,EAAE,IAAI,CAAC,MAAM;KACpB,CAAC,CAAC;AACL,CAAC;AAuDD,MAAM,qBAAqB,GAAG,MAAM,CAAC;AACrC,MAAM,qBAAqB,GAAG,MAAM,CAAC;AACrC,MAAM,qBAAqB,GAAG,CAAC,CAAC;AAChC;;;;;;;GAOG;AACH,MAAM,iBAAiB,GAAG,KAAK,CAAC;AAChC,MAAM,iBAAiB,GAAG,KAAK,CAAC;AAChC;;;;GAIG;AACH,MAAM,2BAA2B,GAAG,MAAM,CAAC;AAE3C,sEAAsE;AACtE,qEAAqE;AACrE,sDAAsD;AACtD,4CAA4C;AAC5C,MAAM,mBAAmB,GAAG;IAC1B,UAAU;IACV,cAAc;IACd,YAAY;IACZ,sBAAsB;IACtB,gBAAgB;IAChB,aAAa;IACb,UAAU;IACV,mCAAmC,EAAE,uBAAuB;CAC7D,CAAC;AAEF;;;;;;;;;GASG;AACH,MAAM,UAAU,uBAAuB,CACrC,QAAwB,EACxB,UAAkC,EAAE;IAEpC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACtD,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE,uBAAuB,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;IACpG,CAAC;IAED,qEAAqE;IACrE,qEAAqE;IACrE,wEAAwE;IACxE,mEAAmE;IACnE,MAAM,YAAY,GAAG,OAAO,CAAC,mBAAmB,CAAC;IACjD,MAAM,SAAS,GACb,OAAO,YAAY,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,YAAY,GAAG,CAAC;QACnF,CAAC,CAAC,YAAY;QACd,CAAC,CAAC,2BAA2B,CAAC;IAClC,MAAM,QAAQ,GAAG,SAAS,GAAG,qBAAqB,CAAC;IACnD,sEAAsE;IACtE,wEAAwE;IACxE,8DAA8D;IAC9D,MAAM,iBAAiB,GACrB,OAAO,CAAC,iBAAiB;QACzB,IAAI,CAAC,GAAG,CAAC,iBAAiB,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,GAAG,CAAC,EAAE,qBAAqB,CAAC,CAAC,CAAC;IAC3F,+CAA+C;IAC/C,MAAM,eAAe,GACnB,OAAO,CAAC,oBAAoB;QAC5B,IAAI,CAAC,GAAG,CAAC,iBAAiB,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,GAAG,CAAC,EAAE,qBAAqB,CAAC,CAAC,CAAC;IAC3F,wEAAwE;IACxE,wEAAwE;IACxE,qEAAqE;IACrE,MAAM,gBAAgB,GAAG,OAAO,CAAC,YAAY,IAAI,KAAK,CAAC;IACvD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,iBAAiB,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,CAAC,CAAC;IAC7E,MAAM,YAAY,GAAG,OAAO,CAAC,qBAAqB,IAAI,IAAI,CAAC;IAI3D,MAAM,KAAK,GAAW,EAAE,CAAC;IACzB,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,QAAQ,CAAC,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC;QAC5C,MAAM,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAQ,CAAC;QAC9B,IAAI,CAAC,EAAE,IAAI,KAAK,YAAY,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;YAAE,SAAS;QACpE,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC;YAC7C,MAAM,KAAK,GAAG,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC5B,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC7D,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;YACjD,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE,uBAAuB,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;IACpG,CAAC;IAED,uDAAuD;IACvD,MAAM,YAAY,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC/C,MAAM,OAAO,GAAG,CAAC,CAAO,EAAU,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC;IAEvD,8DAA8D;IAC9D,IAAI,cAAc,GAAG,CAAC,CAAC;IACvB,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;QACtB,IAAI,CAAC,CAAC,GAAG,GAAG,iBAAiB,EAAE,CAAC;YAC9B,cAAc,EAAE,CAAC;YACjB,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,iBAAiB,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAED,mEAAmE;IACnE,sEAAsE;IACtE,kEAAkE;IAClE,MAAM,gBAAgB,GAAG,CAAC,CAAO,EAAU,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC;IACpF,IAAI,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACvE,MAAM,uBAAuB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,GAAG,eAAe,CAAC,CAAC;IAEzE,IAAI,SAAS,GAAG,eAAe,EAAE,CAAC;QAChC,kDAAkD;QAClD,MAAM,OAAO,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC;QACtE,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACxB,IAAI,SAAS,IAAI,eAAe;gBAAE,MAAM;YACxC,MAAM,OAAO,GAAG,gBAAgB,CAAC,CAAC,CAAC,CAAC;YACpC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,CAAC;YACjD,IAAI,SAAS,KAAK,CAAC;gBAAE,SAAS;YAC9B,MAAM,IAAI,GAAG,SAAS,GAAG,eAAe,CAAC;YACzC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YACtC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,GAAG,GAAG,CAAC,CAAC;YAC5C,SAAS,IAAI,GAAG,CAAC;QACnB,CAAC;IACH,CAAC;IAED,sEAAsE;IACtE,IAAI,YAAY,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QAC5B,OAAO;YACL,QAAQ;YACR,KAAK,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE,uBAAuB,EAAE,eAAe,EAAE,CAAC,EAAE;SAC1E,CAAC;IACJ,CAAC;IAED,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAChG,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,EAAE,EAAE;QACnC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,EAAE,CAAC;YAAE,OAAO,GAAG,CAAC;QAC3C,MAAM,CAAC,GAAG,GAAU,CAAC;QACrB,MAAM,UAAU,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,KAAU,EAAE,EAAU,EAAE,EAAE;YAC1D,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;YAC/C,IAAI,MAAM,KAAK,SAAS;gBAAE,OAAO,KAAK,CAAC;YACvC,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ;gBAAE,OAAO,KAAK,CAAC;YAC3E,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,IAAI,MAAM;gBAAE,OAAO,KAAK,CAAC;YAC9C,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC;YAC5B,MAAM,SAAS,GAAG,YAAY,IAAI,eAAe,CAAC,QAAQ,CAAC;gBACzD,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC;gBAC/B,CAAC,CAAC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAC/B,UAAU,IAAI,QAAQ,CAAC,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC;YACjD,OAAO,EAAE,GAAG,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACvC,CAAC,CAAC,CAAC;QACH,OAAO,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,OAAO;QACL,QAAQ,EAAE,GAAG;QACb,KAAK,EAAE,EAAE,cAAc,EAAE,uBAAuB,EAAE,eAAe,EAAE,UAAU,EAAE;KAChF,CAAC;AACJ,CAAC;AAED,uFAAuF;AAEvF,SAAS,eAAe,CAAC,IAAY;IACnC,0EAA0E;IAC1E,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAC9D,OAAO,mBAAmB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;AACxD,CAAC;AAED,SAAS,QAAQ,CAAC,IAAY,EAAE,SAAiB;IAC/C,MAAM,MAAM,GAAG,4BAA4B,CAAC;IAC5C,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;IACpD,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IAC/B,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,MAAM,KAAK,GAAG,iBAAiB,CAAC;AAClE,CAAC;AAED,SAAS,WAAW,CAAC,IAAY,EAAE,SAAiB;IAClD,MAAM,MAAM,GAAG,qDAAqD,CAAC;IACrE,8CAA8C;IAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;IACtD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC;IACzC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC;IAC9C,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,GAAG,OAAO,GAAG,OAAO,CAAC;IAC5C,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,GAAG,MAAM,IAAI,GAAG,sBAAsB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,EAAE,CAAC;AAC5G,CAAC"}

package/dist/agents/stream-wrappers.d.ts

@@ -0,0 +1,20 @@
+export type BrigadeStreamFn = (...args: unknown[]) => unknown;
+export interface IdleTimeoutOptions {
+    timeoutMs: number;
+    onIdleTimeout?: (elapsedMs: number) => void;
+}
+export declare class BrigadeIdleTimeoutError extends Error {
+    readonly timeoutMs: number;
+    constructor(timeoutMs: number);
+}
+export declare function wrapStreamFnWithIdleTimeout<F extends BrigadeStreamFn>(base: F, options: IdleTimeoutOptions): F;
+export interface StopReasonSignal {
+    type: "stop_reason_signal";
+    reason: string;
+    source: "pi" | "wrapper";
+}
+export declare function wrapStreamFnWithStopReasonRecovery<F extends BrigadeStreamFn>(base: F): F;
+export declare function wrapStreamFnWithToolCallRepair<F extends BrigadeStreamFn>(base: F): F;
+export declare function repairArgumentJson(raw: string): string | null;
+export declare function extractBalancedJsonPrefix(text: string): string | null;
+//# sourceMappingURL=stream-wrappers.d.ts.map

package/dist/agents/stream-wrappers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-wrappers.d.ts","sourceRoot":"","sources":["../../src/agents/stream-wrappers.ts"],"names":[],"mappings":"AAqCA,MAAM,MAAM,eAAe,GAAG,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC;AA0C9D,MAAM,WAAW,kBAAkB;IACjC,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;CAC7C;AAED,qBAAa,uBAAwB,SAAQ,KAAK;IAChD,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;gBACf,SAAS,EAAE,MAAM;CAK9B;AAED,wBAAgB,2BAA2B,CAAC,CAAC,SAAS,eAAe,EACnE,IAAI,EAAE,CAAC,EACP,OAAO,EAAE,kBAAkB,GAC1B,CAAC,CAWH;AAmED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,oBAAoB,CAAC;IAC3B,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,IAAI,GAAG,SAAS,CAAC;CAC1B;AAED,wBAAgB,kCAAkC,CAAC,CAAC,SAAS,eAAe,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,CAOxF;AAgED,wBAAgB,8BAA8B,CAAC,CAAC,SAAS,eAAe,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,CAOpF;AA2BD,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAqB7D;AAED,wBAAgB,yBAAyB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAgCrE"}