@sema-agent/core 1.279.0

package/dist/brain/stream-engine.js

@@ -0,0 +1,295 @@
+import { createAssistantMessageEventStream, } from "../internal/llm.js";
+import { BrainError, classifyHttp } from "./errors.js";
+import { retryBackoffMs } from "./retry.js";
+import { emitBrainStatus, emitBrainTelemetry } from "./status-sink.js";
+import { makeConnectController } from "./timeout.js";
+const EMPTY_USAGE = {
+    input: 0,
+    output: 0,
+    cacheRead: 0,
+    cacheWrite: 0,
+    totalTokens: 0,
+    cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+};
+function emptyAssistant(model) {
+    return {
+        role: "assistant",
+        content: [],
+        api: model.api,
+        provider: model.provider,
+        model: model.id,
+        stopReason: "stop",
+        usage: { ...EMPTY_USAGE, cost: { ...EMPTY_USAGE.cost } },
+        timestamp: Date.now(),
+    };
+}
+function isAbortError(err) {
+    return err instanceof Error && (err.name === "AbortError" || err.name === "TimeoutError");
+}
+function sleep(ms, signal) {
+    return new Promise((resolve) => {
+        if (signal?.aborted)
+            return resolve();
+        let t;
+        const onAbort = () => {
+            clearTimeout(t);
+            resolve();
+        };
+        t = setTimeout(() => {
+            signal?.removeEventListener("abort", onAbort);
+            resolve();
+        }, ms);
+        signal?.addEventListener("abort", onAbort, { once: true });
+    });
+}
+export function runStreamingBrain(args) {
+    const { model, doFetch, signal, config, httpLabel, buildRequest, makeParser } = args;
+    const out = createAssistantMessageEventStream();
+    let cleanup;
+    void run()
+        .catch((err) => {
+        const aborted = signal?.aborted === true || isAbortError(err);
+        const errorMsg = emptyAssistant(model);
+        errorMsg.stopReason = aborted ? "aborted" : "error";
+        errorMsg.errorMessage = err instanceof Error ? err.message : String(err);
+        errorMsg.usageMissing = true;
+        out.push({ type: "error", reason: aborted ? "aborted" : "error", error: errorMsg });
+    })
+        .finally(() => cleanup?.());
+    return out;
+    async function run() {
+        const req = buildRequest();
+        const maxRetries = config.maxRetries ?? 2;
+        const baseDelay = config.retryDelayMs ?? 400;
+        let attempt = 0;
+        let thinkingRetries = 0;
+        let startEmitted = false;
+        for (;;) {
+            let res;
+            let connect;
+            for (;; attempt++) {
+                if (signal?.aborted) {
+                    const e = new Error("aborted");
+                    e.name = "AbortError";
+                    throw e;
+                }
+                const cc = makeConnectController(config.connectTimeoutMs, signal);
+                let r;
+                let netErr;
+                try {
+                    r = await doFetch(req.url, { method: "POST", headers: req.headers, body: req.body, signal: cc.signal });
+                }
+                catch (e) {
+                    if (cc.timedOut()) {
+                        netErr = new Error(`connect timeout after ${config.connectTimeoutMs}ms`);
+                    }
+                    else if (isAbortError(e) || signal?.aborted) {
+                        cc.dispose();
+                        throw e;
+                    }
+                    else {
+                        netErr = e;
+                    }
+                }
+                if (r && r.ok && r.body) {
+                    cc.clearTimer();
+                    connect = cc;
+                    cleanup = () => connect?.dispose();
+                    res = r;
+                    break;
+                }
+                cc.dispose();
+                const retryable = netErr !== undefined ||
+                    (r !== undefined && (r.status >= 500 || r.status === 429 || r.status === 408 || r.status === 409));
+                if (retryable && attempt < maxRetries) {
+                    const delayMs = retryBackoffMs(baseDelay, attempt, r);
+                    const statusPhase = r?.status === 429 ? "rate_limited" : netErr !== undefined ? "reconnecting" : "retrying";
+                    emitBrainTelemetry({ kind: "retry", attempt: attempt + 1, phase: "connect" });
+                    emitBrainStatus({
+                        phase: statusPhase,
+                        detail: statusPhase === "rate_limited"
+                            ? "rate limited, backing off"
+                            : statusPhase === "reconnecting"
+                                ? "connection lost, reconnecting"
+                                : "transient error, retrying",
+                        retryInSec: Math.ceil(delayMs / 1000),
+                    });
+                    await sleep(delayMs, signal);
+                    continue;
+                }
+                if (netErr)
+                    throw new BrainError("network", netErr instanceof Error ? netErr.message : String(netErr));
+                const detail = r ? await r.text().catch(() => "") : "";
+                const status = r?.status ?? 0;
+                throw new BrainError(classifyHttp(status), `${httpLabel} HTTP ${status || "ERR"}: ${detail.slice(0, 500)}`, status);
+            }
+            const partial = emptyAssistant(model);
+            if (!startEmitted) {
+                startEmitted = true;
+                out.push({ type: "start", partial: { ...partial } });
+            }
+            const reader = res.body.getReader();
+            const decoder = new TextDecoder();
+            let buffer = "";
+            let firstTokenSeen = false;
+            let firstTokenTimedOut = false;
+            let ftTimer;
+            if (config.firstTokenTimeoutMs && config.firstTokenTimeoutMs > 0) {
+                ftTimer = setTimeout(() => {
+                    firstTokenTimedOut = true;
+                    void reader.cancel().catch(() => undefined);
+                }, config.firstTokenTimeoutMs);
+            }
+            let idleTimedOut = false;
+            let idleTimer;
+            const bumpIdle = () => {
+                if (!config.idleTimeoutMs || config.idleTimeoutMs <= 0)
+                    return;
+                idleTimedOut = false;
+                if (idleTimer)
+                    clearTimeout(idleTimer);
+                idleTimer = setTimeout(() => {
+                    idleTimedOut = true;
+                    void reader.cancel().catch(() => undefined);
+                }, config.idleTimeoutMs);
+            };
+            const sawContentToken = () => {
+                bumpIdle();
+                if (firstTokenSeen)
+                    return;
+                firstTokenSeen = true;
+                firstTokenTimedOut = false;
+                if (ftTimer) {
+                    clearTimeout(ftTimer);
+                    ftTimer = undefined;
+                }
+            };
+            cleanup = () => {
+                if (ftTimer) {
+                    clearTimeout(ftTimer);
+                    ftTimer = undefined;
+                }
+                if (idleTimer) {
+                    clearTimeout(idleTimer);
+                    idleTimer = undefined;
+                }
+                connect?.dispose();
+                void reader.cancel().catch(() => undefined);
+            };
+            let midstreamWarning;
+            const parserOut = Object.create(out);
+            parserOut.push = (ev) => {
+                if (midstreamWarning !== undefined && (ev.type === "done" || ev.type === "error")) {
+                    const msg = ev.type === "done" ? ev.message : ev.error;
+                    msg.partialFinalized = true;
+                    msg.errorMessage = msg.errorMessage ? `${msg.errorMessage}\n${midstreamWarning}` : midstreamWarning;
+                }
+                out.push(ev);
+            };
+            const parser = makeParser({
+                out: parserOut,
+                partial,
+                sawContentToken,
+                cancel: () => void reader.cancel().catch(() => undefined),
+            });
+            const INBAND_OVERLOAD_WARNING = "API overloaded mid-response. The response above may be incomplete.";
+            const onLineOverloadTierA = (line) => {
+                try {
+                    parser.onLine(line);
+                    return false;
+                }
+                catch (e) {
+                    if (!(e instanceof BrainError) || e.code !== "server" || !parser.snapshot().hasSubstantiveText)
+                        throw e;
+                    return true;
+                }
+            };
+            let failure;
+            readLoop: for (;;) {
+                let readRes;
+                try {
+                    readRes = await reader.read();
+                }
+                catch (e) {
+                    if (isAbortError(e) || signal?.aborted)
+                        throw e;
+                    failure = {
+                        kind: "connection",
+                        err: new BrainError("network", `mid-stream read failed: ${e instanceof Error ? e.message : String(e)}`),
+                    };
+                    break readLoop;
+                }
+                const { done, value } = readRes;
+                if (done)
+                    break;
+                buffer += decoder.decode(value, { stream: true });
+                let nl;
+                let start = 0;
+                while ((nl = buffer.indexOf("\n", start)) !== -1) {
+                    const line = buffer.slice(start, nl);
+                    start = nl + 1;
+                    if (onLineOverloadTierA(line)) {
+                        midstreamWarning = INBAND_OVERLOAD_WARNING;
+                        parser.finalize();
+                        return;
+                    }
+                }
+                if (start > 0)
+                    buffer = buffer.slice(start);
+            }
+            if (ftTimer) {
+                clearTimeout(ftTimer);
+                ftTimer = undefined;
+            }
+            if (idleTimer) {
+                clearTimeout(idleTimer);
+                idleTimer = undefined;
+            }
+            if (!failure && firstTokenTimedOut) {
+                failure = { kind: "stalled", err: new BrainError("network", `first-token timeout after ${config.firstTokenTimeoutMs}ms`) };
+            }
+            if (!failure && idleTimedOut) {
+                failure = { kind: "stalled", err: new BrainError("network", `stream idle timeout after ${config.idleTimeoutMs}ms`) };
+            }
+            if (!failure) {
+                buffer += decoder.decode();
+                if (buffer.trim() && onLineOverloadTierA(buffer))
+                    midstreamWarning = INBAND_OVERLOAD_WARNING;
+                parser.finalize();
+                return;
+            }
+            const snap = parser.snapshot();
+            if (snap.hasSubstantiveText || snap.hasCompletedToolCall) {
+                midstreamWarning =
+                    failure.kind === "stalled"
+                        ? "Response stalled mid-stream. The response above may be incomplete."
+                        : "Connection closed mid-response. The response above may be incomplete.";
+                parser.finalize();
+                return;
+            }
+            if (snap.hasCompletedToolCall)
+                throw failure.err;
+            cleanup?.();
+            cleanup = undefined;
+            if (snap.hasOnlyThinking) {
+                if (thinkingRetries >= 2)
+                    throw failure.err;
+                thinkingRetries++;
+                parser.sealForRetry();
+                const delayMs = 100 * thinkingRetries;
+                emitBrainTelemetry({ kind: "retry", attempt: thinkingRetries, phase: "midstream" });
+                emitBrainStatus({ phase: "reconnecting", detail: "connection lost, reconnecting", retryInSec: Math.ceil(delayMs / 1000) });
+                await sleep(delayMs, signal);
+                continue;
+            }
+            if (attempt >= maxRetries)
+                throw failure.err;
+            const delayMs = retryBackoffMs(baseDelay, attempt);
+            attempt++;
+            emitBrainTelemetry({ kind: "retry", attempt, phase: "midstream" });
+            emitBrainStatus({ phase: "reconnecting", detail: "connection lost, reconnecting", retryInSec: Math.ceil(delayMs / 1000) });
+            await sleep(delayMs, signal);
+        }
+    }
+}
+//# sourceMappingURL=stream-engine.js.map

package/dist/brain/stream-engine.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-engine.js","sourceRoot":"","sources":["../../src/brain/stream-engine.ts"],"names":[],"mappings":"AAAA,OAAO,EAKL,iCAAiC,GAClC,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,EAAE,eAAe,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACvE,OAAO,EAA2B,qBAAqB,EAAE,MAAM,cAAc,CAAC;AAqE9E,MAAM,WAAW,GAAU;IACzB,KAAK,EAAE,CAAC;IACR,MAAM,EAAE,CAAC;IACT,SAAS,EAAE,CAAC;IACZ,UAAU,EAAE,CAAC;IACb,WAAW,EAAE,CAAC;IACd,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;CACrE,CAAC;AAEF,SAAS,cAAc,CAAC,KAAY;IAClC,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,EAAE;QACX,GAAG,EAAE,KAAK,CAAC,GAAG;QACd,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,KAAK,EAAE,KAAK,CAAC,EAAE;QACf,UAAU,EAAE,MAAM;QAClB,KAAK,EAAE,EAAE,GAAG,WAAW,EAAE,IAAI,EAAE,EAAE,GAAG,WAAW,CAAC,IAAI,EAAE,EAAE;QACxD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACtB,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,GAAY;IAChC,OAAO,GAAG,YAAY,KAAK,IAAI,CAAC,GAAG,CAAC,IAAI,KAAK,YAAY,IAAI,GAAG,CAAC,IAAI,KAAK,cAAc,CAAC,CAAC;AAC5F,CAAC;AACD,SAAS,KAAK,CAAC,EAAU,EAAE,MAAoB;IAC7C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,IAAI,MAAM,EAAE,OAAO;YAAE,OAAO,OAAO,EAAE,CAAC;QAOtC,IAAI,CAAgC,CAAC;QACrC,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,YAAY,CAAC,CAAC,CAAC,CAAC;YAChB,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC;QACF,CAAC,GAAG,UAAU,CAAC,GAAG,EAAE;YAClB,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC9C,OAAO,EAAE,CAAC;QACZ,CAAC,EAAE,EAAE,CAAC,CAAC;QACP,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;AACL,CAAC;AAMD,MAAM,UAAU,iBAAiB,CAAC,IAWjC;IACC,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC;IACrF,MAAM,GAAG,GAAG,iCAAiC,EAAE,CAAC;IAShD,IAAI,OAAiC,CAAC;IACtC,KAAK,GAAG,EAAE;SACP,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;QACtB,MAAM,OAAO,GAAG,MAAM,EAAE,OAAO,KAAK,IAAI,IAAI,YAAY,CAAC,GAAG,CAAC,CAAC;QAC9D,MAAM,QAAQ,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;QACvC,QAAQ,CAAC,UAAU,GAAG,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;QACpD,QAAQ,CAAC,YAAY,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAKzE,QAAQ,CAAC,YAAY,GAAG,IAAI,CAAC;QAC7B,GAAG,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;IACtF,CAAC,CAAC;SACD,OAAO,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC9B,OAAO,GAAG,CAAC;IAEX,KAAK,UAAU,GAAG;QAChB,MAAM,GAAG,GAAG,YAAY,EAAE,CAAC;QAC3B,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,CAAC,CAAC;QAC1C,MAAM,SAAS,GAAG,MAAM,CAAC,YAAY,IAAI,GAAG,CAAC;QAU7C,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,IAAI,eAAe,GAAG,CAAC,CAAC;QACxB,IAAI,YAAY,GAAG,KAAK,CAAC;QAEzB,SAAS,CAAC;YAER,IAAI,GAAc,CAAC;YACnB,IAAI,OAA6D,CAAC;YAClE,QAAS,OAAO,EAAE,EAAE,CAAC;gBACnB,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;oBACpB,MAAM,CAAC,GAAG,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;oBAC/B,CAAC,CAAC,IAAI,GAAG,YAAY,CAAC;oBACtB,MAAM,CAAC,CAAC;gBACV,CAAC;gBAGD,MAAM,EAAE,GAAG,qBAAqB,CAAC,MAAM,CAAC,gBAAgB,EAAE,MAAM,CAAC,CAAC;gBAClE,IAAI,CAAuB,CAAC;gBAC5B,IAAI,MAAe,CAAC;gBACpB,IAAI,CAAC;oBACH,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC1G,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,IAAI,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC;wBAClB,MAAM,GAAG,IAAI,KAAK,CAAC,yBAAyB,MAAM,CAAC,gBAAgB,IAAI,CAAC,CAAC;oBAC3E,CAAC;yBAAM,IAAI,YAAY,CAAC,CAAC,CAAC,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;wBAC9C,EAAE,CAAC,OAAO,EAAE,CAAC;wBACb,MAAM,CAAC,CAAC;oBACV,CAAC;yBAAM,CAAC;wBACN,MAAM,GAAG,CAAC,CAAC;oBACb,CAAC;gBACH,CAAC;gBACD,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;oBACxB,EAAE,CAAC,UAAU,EAAE,CAAC;oBAChB,OAAO,GAAG,EAAE,CAAC;oBAGb,OAAO,GAAG,GAAG,EAAE,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;oBACnC,GAAG,GAAG,CAAC,CAAC;oBACR,MAAM;gBACR,CAAC;gBACD,EAAE,CAAC,OAAO,EAAE,CAAC;gBAGb,MAAM,SAAS,GACb,MAAM,KAAK,SAAS;oBACpB,CAAC,CAAC,KAAK,SAAS,IAAI,CAAC,CAAC,CAAC,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,MAAM,KAAK,GAAG,CAAC,CAAC,CAAC;gBACrG,IAAI,SAAS,IAAI,OAAO,GAAG,UAAU,EAAE,CAAC;oBACtC,MAAM,OAAO,GAAG,cAAc,CAAC,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;oBAItD,MAAM,WAAW,GAAG,CAAC,EAAE,MAAM,KAAK,GAAG,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,UAAU,CAAC;oBAG5G,kBAAkB,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,GAAG,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;oBAC9E,eAAe,CAAC;wBACd,KAAK,EAAE,WAAW;wBAElB,MAAM,EACJ,WAAW,KAAK,cAAc;4BAC5B,CAAC,CAAC,2BAA2B;4BAC7B,CAAC,CAAC,WAAW,KAAK,cAAc;gCAC9B,CAAC,CAAC,+BAA+B;gCACjC,CAAC,CAAC,2BAA2B;wBACnC,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;qBACtC,CAAC,CAAC;oBACH,MAAM,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;oBAC7B,SAAS;gBACX,CAAC;gBACD,IAAI,MAAM;oBAAE,MAAM,IAAI,UAAU,CAAC,SAAS,EAAE,MAAM,YAAY,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;gBACvG,MAAM,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvD,MAAM,MAAM,GAAG,CAAC,EAAE,MAAM,IAAI,CAAC,CAAC;gBAC9B,MAAM,IAAI,UAAU,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,GAAG,SAAS,SAAS,MAAM,IAAI,KAAK,KAAK,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;YACtH,CAAC;YAGD,MAAM,OAAO,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;YACtC,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,YAAY,GAAG,IAAI,CAAC;gBACpB,GAAG,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,EAAE,CAAC,CAAC;YACvD,CAAC;YAED,MAAM,MAAM,GAAG,GAAG,CAAC,IAAK,CAAC,SAAS,EAAE,CAAC;YACrC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;YAClC,IAAI,MAAM,GAAG,EAAE,CAAC;YAEhB,IAAI,cAAc,GAAG,KAAK,CAAC;YAC3B,IAAI,kBAAkB,GAAG,KAAK,CAAC;YAC/B,IAAI,OAAkD,CAAC;YACvD,IAAI,MAAM,CAAC,mBAAmB,IAAI,MAAM,CAAC,mBAAmB,GAAG,CAAC,EAAE,CAAC;gBACjE,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;oBACxB,kBAAkB,GAAG,IAAI,CAAC;oBAC1B,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;gBAC9C,CAAC,EAAE,MAAM,CAAC,mBAAmB,CAAC,CAAC;YACjC,CAAC;YACD,IAAI,YAAY,GAAG,KAAK,CAAC;YACzB,IAAI,SAAoD,CAAC;YACzD,MAAM,QAAQ,GAAG,GAAS,EAAE;gBAC1B,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,aAAa,IAAI,CAAC;oBAAE,OAAO;gBAG/D,YAAY,GAAG,KAAK,CAAC;gBACrB,IAAI,SAAS;oBAAE,YAAY,CAAC,SAAS,CAAC,CAAC;gBACvC,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;oBAC1B,YAAY,GAAG,IAAI,CAAC;oBACpB,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;gBAC9C,CAAC,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC;YAC3B,CAAC,CAAC;YACF,MAAM,eAAe,GAAG,GAAS,EAAE;gBACjC,QAAQ,EAAE,CAAC;gBACX,IAAI,cAAc;oBAAE,OAAO;gBAC3B,cAAc,GAAG,IAAI,CAAC;gBAGtB,kBAAkB,GAAG,KAAK,CAAC;gBAC3B,IAAI,OAAO,EAAE,CAAC;oBACZ,YAAY,CAAC,OAAO,CAAC,CAAC;oBACtB,OAAO,GAAG,SAAS,CAAC;gBACtB,CAAC;YACH,CAAC,CAAC;YACF,OAAO,GAAG,GAAG,EAAE;gBACb,IAAI,OAAO,EAAE,CAAC;oBACZ,YAAY,CAAC,OAAO,CAAC,CAAC;oBACtB,OAAO,GAAG,SAAS,CAAC;gBACtB,CAAC;gBACD,IAAI,SAAS,EAAE,CAAC;oBACd,YAAY,CAAC,SAAS,CAAC,CAAC;oBACxB,SAAS,GAAG,SAAS,CAAC;gBACxB,CAAC;gBACD,OAAO,EAAE,OAAO,EAAE,CAAC;gBAInB,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YAC9C,CAAC,CAAC;YAMF,IAAI,gBAAoC,CAAC;YACzC,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAA0B,CAAC;YAC9D,SAAS,CAAC,IAAI,GAAG,CAAC,EAAyB,EAAQ,EAAE;gBACnD,IAAI,gBAAgB,KAAK,SAAS,IAAI,CAAC,EAAE,CAAC,IAAI,KAAK,MAAM,IAAI,EAAE,CAAC,IAAI,KAAK,OAAO,CAAC,EAAE,CAAC;oBAClF,MAAM,GAAG,GAAG,EAAE,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC;oBACvD,GAAG,CAAC,gBAAgB,GAAG,IAAI,CAAC;oBAK5B,GAAG,CAAC,YAAY,GAAG,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,YAAY,KAAK,gBAAgB,EAAE,CAAC,CAAC,CAAC,gBAAgB,CAAC;gBACtG,CAAC;gBACD,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACf,CAAC,CAAC;YAEF,MAAM,MAAM,GAAG,UAAU,CAAC;gBACxB,GAAG,EAAE,SAAS;gBACd,OAAO;gBACP,eAAe;gBAEf,MAAM,EAAE,GAAG,EAAE,CAAC,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;aAC1D,CAAC,CAAC;YAcH,MAAM,uBAAuB,GAAG,oEAAoE,CAAC;YAErG,MAAM,mBAAmB,GAAG,CAAC,IAAY,EAAW,EAAE;gBACpD,IAAI,CAAC;oBACH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;oBACpB,OAAO,KAAK,CAAC;gBACf,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,IAAI,CAAC,CAAC,CAAC,YAAY,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,kBAAkB;wBAAE,MAAM,CAAC,CAAC;oBACxG,OAAO,IAAI,CAAC;gBACd,CAAC;YACH,CAAC,CAAC;YACF,IAAI,OAAwE,CAAC;YAC7E,QAAQ,EAAE,SAAS,CAAC;gBAClB,IAAI,OAA8C,CAAC;gBACnD,IAAI,CAAC;oBACH,OAAO,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAChC,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBAGX,IAAI,YAAY,CAAC,CAAC,CAAC,IAAI,MAAM,EAAE,OAAO;wBAAE,MAAM,CAAC,CAAC;oBAChD,OAAO,GAAG;wBACR,IAAI,EAAE,YAAY;wBAClB,GAAG,EAAE,IAAI,UAAU,CAAC,SAAS,EAAE,2BAA2B,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;qBACxG,CAAC;oBACF,MAAM,QAAQ,CAAC;gBACjB,CAAC;gBACD,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,OAAgD,CAAC;gBACzE,IAAI,IAAI;oBAAE,MAAM;gBAChB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBAElD,IAAI,EAAU,CAAC;gBACf,IAAI,KAAK,GAAG,CAAC,CAAC;gBACd,OAAO,CAAC,EAAE,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;oBACjD,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;oBACrC,KAAK,GAAG,EAAE,GAAG,CAAC,CAAC;oBACf,IAAI,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;wBAG9B,gBAAgB,GAAG,uBAAuB,CAAC;wBAC3C,MAAM,CAAC,QAAQ,EAAE,CAAC;wBAClB,OAAO;oBACT,CAAC;gBACH,CAAC;gBACD,IAAI,KAAK,GAAG,CAAC;oBAAE,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC9C,CAAC;YACD,IAAI,OAAO,EAAE,CAAC;gBACZ,YAAY,CAAC,OAAO,CAAC,CAAC;gBACtB,OAAO,GAAG,SAAS,CAAC;YACtB,CAAC;YACD,IAAI,SAAS,EAAE,CAAC;gBACd,YAAY,CAAC,SAAS,CAAC,CAAC;gBACxB,SAAS,GAAG,SAAS,CAAC;YACxB,CAAC;YACD,IAAI,CAAC,OAAO,IAAI,kBAAkB,EAAE,CAAC;gBACnC,OAAO,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,EAAE,IAAI,UAAU,CAAC,SAAS,EAAE,6BAA6B,MAAM,CAAC,mBAAmB,IAAI,CAAC,EAAE,CAAC;YAC7H,CAAC;YACD,IAAI,CAAC,OAAO,IAAI,YAAY,EAAE,CAAC;gBAC7B,OAAO,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,EAAE,IAAI,UAAU,CAAC,SAAS,EAAE,6BAA6B,MAAM,CAAC,aAAa,IAAI,CAAC,EAAE,CAAC;YACvH,CAAC;YAED,IAAI,CAAC,OAAO,EAAE,CAAC;gBAGb,MAAM,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBAC3B,IAAI,MAAM,CAAC,IAAI,EAAE,IAAI,mBAAmB,CAAC,MAAM,CAAC;oBAAE,gBAAgB,GAAG,uBAAuB,CAAC;gBAE7F,MAAM,CAAC,QAAQ,EAAE,CAAC;gBAClB,OAAO;YACT,CAAC;YAGD,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;YAE/B,IAAI,IAAI,CAAC,kBAAkB,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;gBAMzD,gBAAgB;oBACd,OAAO,CAAC,IAAI,KAAK,SAAS;wBACxB,CAAC,CAAC,oEAAoE;wBACtE,CAAC,CAAC,uEAAuE,CAAC;gBAC9E,MAAM,CAAC,QAAQ,EAAE,CAAC;gBAClB,OAAO;YACT,CAAC;YAMD,IAAI,IAAI,CAAC,oBAAoB;gBAAE,MAAM,OAAO,CAAC,GAAG,CAAC;YAIjD,OAAO,EAAE,EAAE,CAAC;YACZ,OAAO,GAAG,SAAS,CAAC;YAEpB,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;gBAMzB,IAAI,eAAe,IAAI,CAAC;oBAAE,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC5C,eAAe,EAAE,CAAC;gBAClB,MAAM,CAAC,YAAY,EAAE,CAAC;gBACtB,MAAM,OAAO,GAAG,GAAG,GAAG,eAAe,CAAC;gBACtC,kBAAkB,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;gBACpF,eAAe,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,MAAM,EAAE,+BAA+B,EAAE,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC3H,MAAM,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;gBAC7B,SAAS;YACX,CAAC;YAMD,IAAI,OAAO,IAAI,UAAU;gBAAE,MAAM,OAAO,CAAC,GAAG,CAAC;YAC7C,MAAM,OAAO,GAAG,cAAc,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACnD,OAAO,EAAE,CAAC;YACV,kBAAkB,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;YACnE,eAAe,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,MAAM,EAAE,+BAA+B,EAAE,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC;YAC3H,MAAM,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAE/B,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/brain/timeout.d.ts

@@ -0,0 +1,12 @@
+export interface BrainTimeoutConfig {
+    connectTimeoutMs?: number;
+    firstTokenTimeoutMs?: number;
+    idleTimeoutMs?: number;
+}
+export declare function makeConnectController(connectTimeoutMs: number | undefined, outerSignal: AbortSignal | undefined): {
+    signal: AbortSignal;
+    timedOut: () => boolean;
+    clearTimer: () => void;
+    dispose: () => void;
+};
+//# sourceMappingURL=timeout.d.ts.map

package/dist/brain/timeout.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"timeout.d.ts","sourceRoot":"","sources":["../../src/brain/timeout.ts"],"names":[],"mappings":"AAWA,MAAM,WAAW,kBAAkB;IAEjC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAO1B,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAQ7B,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AASD,wBAAgB,qBAAqB,CACnC,gBAAgB,EAAE,MAAM,GAAG,SAAS,EACpC,WAAW,EAAE,WAAW,GAAG,SAAS,GACnC;IACD,MAAM,EAAE,WAAW,CAAC;IACpB,QAAQ,EAAE,MAAM,OAAO,CAAC;IACxB,UAAU,EAAE,MAAM,IAAI,CAAC;IACvB,OAAO,EAAE,MAAM,IAAI,CAAC;CACrB,CA4BA"}

package/dist/brain/timeout.js

@@ -0,0 +1,34 @@
+export function makeConnectController(connectTimeoutMs, outerSignal) {
+    const ac = new AbortController();
+    let timedOut = false;
+    const onAbort = () => ac.abort(outerSignal?.reason);
+    if (outerSignal) {
+        if (outerSignal.aborted)
+            ac.abort(outerSignal.reason);
+        else
+            outerSignal.addEventListener("abort", onAbort, { once: true });
+    }
+    let timer;
+    if (connectTimeoutMs && connectTimeoutMs > 0) {
+        timer = setTimeout(() => {
+            timedOut = true;
+            ac.abort();
+        }, connectTimeoutMs);
+    }
+    const clearTimer = () => {
+        if (timer)
+            clearTimeout(timer);
+        timer = undefined;
+    };
+    return {
+        signal: ac.signal,
+        timedOut: () => timedOut,
+        clearTimer,
+        dispose: () => {
+            clearTimer();
+            if (outerSignal)
+                outerSignal.removeEventListener("abort", onAbort);
+        },
+    };
+}
+//# sourceMappingURL=timeout.js.map

package/dist/brain/timeout.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"timeout.js","sourceRoot":"","sources":["../../src/brain/timeout.ts"],"names":[],"mappings":"AAsCA,MAAM,UAAU,qBAAqB,CACnC,gBAAoC,EACpC,WAAoC;IAOpC,MAAM,EAAE,GAAG,IAAI,eAAe,EAAE,CAAC;IACjC,IAAI,QAAQ,GAAG,KAAK,CAAC;IACrB,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,CAAE,WAAuC,EAAE,MAAM,CAAC,CAAC;IACjF,IAAI,WAAW,EAAE,CAAC;QAChB,IAAI,WAAW,CAAC,OAAO;YAAE,EAAE,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;;YACjD,WAAW,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IACtE,CAAC;IACD,IAAI,KAAgD,CAAC;IACrD,IAAI,gBAAgB,IAAI,gBAAgB,GAAG,CAAC,EAAE,CAAC;QAC7C,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YACtB,QAAQ,GAAG,IAAI,CAAC;YAChB,EAAE,CAAC,KAAK,EAAE,CAAC;QACb,CAAC,EAAE,gBAAgB,CAAC,CAAC;IACvB,CAAC;IACD,MAAM,UAAU,GAAG,GAAG,EAAE;QACtB,IAAI,KAAK;YAAE,YAAY,CAAC,KAAK,CAAC,CAAC;QAC/B,KAAK,GAAG,SAAS,CAAC;IACpB,CAAC,CAAC;IACF,OAAO;QACL,MAAM,EAAE,EAAE,CAAC,MAAM;QACjB,QAAQ,EAAE,GAAG,EAAE,CAAC,QAAQ;QACxB,UAAU;QACV,OAAO,EAAE,GAAG,EAAE;YACZ,UAAU,EAAE,CAAC;YACb,IAAI,WAAW;gBAAE,WAAW,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACrE,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/brain/tool-call-repair.d.ts

@@ -0,0 +1,6 @@
+import type { ToolCall } from "../internal/llm.js";
+export declare function repairTextToolCalls(text: string, toolNames: Set<string>): {
+    toolCalls: ToolCall[];
+    cleanedText: string;
+};
+//# sourceMappingURL=tool-call-repair.d.ts.map

package/dist/brain/tool-call-repair.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-call-repair.d.ts","sourceRoot":"","sources":["../../src/brain/tool-call-repair.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAiEnD,wBAAgB,mBAAmB,CACjC,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,GAAG,CAAC,MAAM,CAAC,GACrB;IAAE,SAAS,EAAE,QAAQ,EAAE,CAAC;IAAC,WAAW,EAAE,MAAM,CAAA;CAAE,CAShD"}

package/dist/brain/tool-call-repair.js

@@ -0,0 +1,58 @@
+const TAG_RE = /<(tool_call|function_call|tool_calls|function_calls)>\s*([\s\S]*?)\s*<\/\1>/gi;
+const FENCE_RE = /```(?:json|tool_call|tool|function)?\s*([\s\S]*?)```/gi;
+function asRecord(v) {
+    return v && typeof v === "object" && !Array.isArray(v) ? v : undefined;
+}
+function extractCall(obj, toolNames) {
+    const rec = asRecord(obj);
+    if (!rec) {
+        return undefined;
+    }
+    const fn = asRecord(rec.function);
+    const name = rec.name ?? rec.tool ?? rec.tool_name ?? fn?.name;
+    if (typeof name !== "string" || !toolNames.has(name)) {
+        return undefined;
+    }
+    let rawArgs = rec.arguments ?? rec.parameters ?? rec.input ?? rec.args ?? fn?.arguments;
+    if (typeof rawArgs === "string") {
+        try {
+            rawArgs = JSON.parse(rawArgs);
+        }
+        catch {
+            rawArgs = {};
+        }
+    }
+    const args = asRecord(rawArgs) ?? {};
+    return {
+        type: "toolCall",
+        id: `repair_${name}_${Math.random().toString(36).slice(2, 8)}`,
+        name,
+        arguments: args,
+    };
+}
+function parseBlock(raw, toolNames, out) {
+    let parsed;
+    try {
+        parsed = JSON.parse(raw.trim());
+    }
+    catch {
+        return false;
+    }
+    const items = Array.isArray(parsed) ? parsed : [parsed];
+    let any = false;
+    for (const item of items) {
+        const call = extractCall(item, toolNames);
+        if (call) {
+            out.push(call);
+            any = true;
+        }
+    }
+    return any;
+}
+export function repairTextToolCalls(text, toolNames) {
+    const toolCalls = [];
+    let cleaned = text.replace(TAG_RE, (m, _tag, inner) => parseBlock(inner, toolNames, toolCalls) ? "" : m);
+    cleaned = cleaned.replace(FENCE_RE, (m, inner) => parseBlock(inner, toolNames, toolCalls) ? "" : m);
+    return { toolCalls, cleanedText: cleaned.trim() };
+}
+//# sourceMappingURL=tool-call-repair.js.map

package/dist/brain/tool-call-repair.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-call-repair.js","sourceRoot":"","sources":["../../src/brain/tool-call-repair.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,GACV,+EAA+E,CAAC;AAClF,MAAM,QAAQ,GAAG,wDAAwD,CAAC;AAE1E,SAAS,QAAQ,CAAC,CAAU;IAC1B,OAAO,CAAC,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAE,CAA6B,CAAC,CAAC,CAAC,SAAS,CAAC;AACtG,CAAC;AAED,SAAS,WAAW,CAAC,GAAY,EAAE,SAAsB;IACvD,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;IAC1B,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,EAAE,GAAG,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAClC,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,SAAS,IAAI,EAAE,EAAE,IAAI,CAAC;IAC/D,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;QACrD,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,OAAO,GAAY,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,IAAI,IAAI,EAAE,EAAE,SAAS,CAAC;IACjG,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,IAAI,CAAC;YACH,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAChC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,GAAG,EAAE,CAAC;QACf,CAAC;IACH,CAAC;IACD,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;IACrC,OAAO;QACL,IAAI,EAAE,UAAU;QAChB,EAAE,EAAE,UAAU,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;QAC9D,IAAI;QACJ,SAAS,EAAE,IAAI;KAChB,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CAAC,GAAW,EAAE,SAAsB,EAAE,GAAe;IACtE,IAAI,MAAe,CAAC;IACpB,IAAI,CAAC;QACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;IAClC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;IACD,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;IACxD,IAAI,GAAG,GAAG,KAAK,CAAC;IAChB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAC1C,IAAI,IAAI,EAAE,CAAC;YACT,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACf,GAAG,GAAG,IAAI,CAAC;QACb,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAWD,MAAM,UAAU,mBAAmB,CACjC,IAAY,EACZ,SAAsB;IAEtB,MAAM,SAAS,GAAe,EAAE,CAAC;IACjC,IAAI,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,KAAa,EAAE,EAAE,CAC5D,UAAU,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CACjD,CAAC;IACF,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,KAAa,EAAE,EAAE,CACvD,UAAU,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CACjD,CAAC;IACF,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;AACpD,CAAC"}

package/dist/brain/walltime.d.ts

@@ -0,0 +1,2 @@
+export declare const WALLTIME_CUTOFF_MESSAGE = "walltime cutoff: the response crossed the task's soft wall-clock deadline and was cut off to preserve a write-out window";
+//# sourceMappingURL=walltime.d.ts.map

package/dist/brain/walltime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"walltime.d.ts","sourceRoot":"","sources":["../../src/brain/walltime.ts"],"names":[],"mappings":"AAaA,eAAO,MAAM,uBAAuB,6HACwF,CAAC"}

package/dist/brain/walltime.js

@@ -0,0 +1,2 @@
+export const WALLTIME_CUTOFF_MESSAGE = "walltime cutoff: the response crossed the task's soft wall-clock deadline and was cut off to preserve a write-out window";
+//# sourceMappingURL=walltime.js.map

package/dist/brain/walltime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"walltime.js","sourceRoot":"","sources":["../../src/brain/walltime.ts"],"names":[],"mappings":"AAaA,MAAM,CAAC,MAAM,uBAAuB,GAClC,0HAA0H,CAAC"}

package/dist/core/arg-summary.d.ts

@@ -0,0 +1,3 @@
+export declare function scrubSecrets(s: string): string;
+export declare function primaryActivityArg(args: unknown): string | undefined;
+//# sourceMappingURL=arg-summary.d.ts.map

package/dist/core/arg-summary.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"arg-summary.d.ts","sourceRoot":"","sources":["../../src/core/arg-summary.ts"],"names":[],"mappings":"AAuBA,wBAAgB,YAAY,CAAC,CAAC,EAAE,MAAM,GAAG,MAAM,CAI9C;AAiBD,wBAAgB,kBAAkB,CAAC,IAAI,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CA2CpE"}

package/dist/core/arg-summary.js

@@ -0,0 +1,76 @@
+const ACTIVITY_ARG_MAX = 80;
+const SENSITIVE_KEY = /token|secret|key|password|passwd|credential|auth/i;
+const SECRET_PATTERNS = [
+    /(?:sk|pk|rk|gh[opsur])[-_][A-Za-z0-9_-]{8,}/gi,
+    /AIza[A-Za-z0-9_-]{20,}/g,
+    /xox[baprs]-[A-Za-z0-9-]{10,}/gi,
+    /AKIA[0-9A-Z]{12,}/g,
+    /eyJ[A-Za-z0-9._-]{20,}/g,
+    /-----BEGIN[ A-Z]*PRIVATE KEY-----/g,
+    /Bearer\s+[A-Za-z0-9._~+/-]{8,}=*/gi,
+    /(?:token|secret|key|password|passwd|credential|authorization)["']?\s*[=:]\s*["']?[^\s"';|&]{4,}/gi,
+];
+export function scrubSecrets(s) {
+    let out = s;
+    for (const p of SECRET_PATTERNS)
+        out = out.replace(p, "[redacted]");
+    return out;
+}
+function truncCodePoints(s) {
+    const cp = Array.from(s);
+    return cp.length > ACTIVITY_ARG_MAX ? cp.slice(0, ACTIVITY_ARG_MAX).join("") + "…" : s;
+}
+const basename = (p) => p.split(/[/\\]/).filter((x) => x.length > 0).pop() ?? p;
+export function primaryActivityArg(args) {
+    if (!args || typeof args !== "object")
+        return undefined;
+    const a = args;
+    const str = (k) => typeof a[k] === "string" && a[k].length > 0 ? a[k] : undefined;
+    let val;
+    const command = str("command");
+    const path = str("file_path") ?? str("filePath") ?? str("path") ?? str("notebook_path");
+    const url = str("url");
+    if (command !== undefined) {
+        const tokens = command.trim().split(/\s+/);
+        let i = 0;
+        while (i < tokens.length && /^[A-Za-z_][A-Za-z0-9_]*=/.test(tokens[i])) {
+            if (/["'\\]/.test(tokens[i]))
+                return undefined;
+            i++;
+        }
+        if (i >= tokens.length)
+            return undefined;
+        const name = basename(tokens[i]);
+        val = tokens.length > i + 1 ? `${name} …` : name;
+    }
+    else if (path !== undefined) {
+        val = basename(path);
+    }
+    else if (url !== undefined) {
+        try {
+            const u = new URL(url);
+            val = `${u.origin}${u.pathname}`;
+        }
+        catch {
+            val = url.split(/[?#]/)[0].replace(/^([a-z][a-z0-9+.-]*:\/\/)[^/@]*@/i, "$1");
+        }
+    }
+    else {
+        val = str("pattern") ?? str("query") ?? str("description");
+        if (val === undefined) {
+            for (const k of Object.keys(a)) {
+                if (SENSITIVE_KEY.test(k))
+                    continue;
+                const v = a[k];
+                if (typeof v === "string" && v.length > 0) {
+                    val = v;
+                    break;
+                }
+            }
+        }
+    }
+    if (val === undefined)
+        return undefined;
+    return truncCodePoints(scrubSecrets(val));
+}
+//# sourceMappingURL=arg-summary.js.map

package/dist/core/arg-summary.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"arg-summary.js","sourceRoot":"","sources":["../../src/core/arg-summary.ts"],"names":[],"mappings":"AAQA,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAE5B,MAAM,aAAa,GAAG,mDAAmD,CAAC;AAG1E,MAAM,eAAe,GAAa;IAChC,+CAA+C;IAC/C,yBAAyB;IACzB,gCAAgC;IAChC,oBAAoB;IACpB,yBAAyB;IACzB,oCAAoC;IACpC,oCAAoC;IACpC,mGAAmG;CACpG,CAAC;AACF,MAAM,UAAU,YAAY,CAAC,CAAS;IACpC,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,KAAK,MAAM,CAAC,IAAI,eAAe;QAAE,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;IACpE,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,eAAe,CAAC,CAAS;IAChC,MAAM,EAAE,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACzB,OAAO,EAAE,CAAC,MAAM,GAAG,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;AACzF,CAAC;AACD,MAAM,QAAQ,GAAG,CAAC,CAAS,EAAU,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;AAWhG,MAAM,UAAU,kBAAkB,CAAC,IAAa;IAC9C,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IACxD,MAAM,CAAC,GAAG,IAA+B,CAAC;IAC1C,MAAM,GAAG,GAAG,CAAC,CAAS,EAAsB,EAAE,CAC5C,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAK,CAAC,CAAC,CAAC,CAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,CAAY,CAAC,CAAC,CAAC,SAAS,CAAC;IACzF,IAAI,GAAuB,CAAC;IAC5B,MAAM,OAAO,GAAG,GAAG,CAAC,SAAS,CAAC,CAAC;IAC/B,MAAM,IAAI,GAAG,GAAG,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,GAAG,CAAC,eAAe,CAAC,CAAC;IACxF,MAAM,GAAG,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC;IACvB,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,GAAG,CAAC,CAAC;QACV,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,IAAI,0BAA0B,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACvE,IAAI,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAAE,OAAO,SAAS,CAAC;YAC/C,CAAC,EAAE,CAAC;QACN,CAAC;QACD,IAAI,CAAC,IAAI,MAAM,CAAC,MAAM;YAAE,OAAO,SAAS,CAAC;QACzC,MAAM,IAAI,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QACjC,GAAG,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;IACnD,CAAC;SAAM,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QAC9B,GAAG,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC;SAAM,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,MAAM,CAAC,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;YACvB,GAAG,GAAG,GAAG,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;QACnC,CAAC;QAAC,MAAM,CAAC;YACP,GAAG,GAAG,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,mCAAmC,EAAE,IAAI,CAAC,CAAC;QAChF,CAAC;IACH,CAAC;SAAM,CAAC;QACN,GAAG,GAAG,GAAG,CAAC,SAAS,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,aAAa,CAAC,CAAC;QAC3D,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YACtB,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC/B,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;oBAAE,SAAS;gBACpC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBACf,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC1C,GAAG,GAAG,CAAC,CAAC;oBACR,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IACD,IAAI,GAAG,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IACxC,OAAO,eAAe,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC;AAC5C,CAAC"}

package/dist/core/ask-question.d.ts

@@ -0,0 +1,36 @@
+import type { AgentTool } from "../internal/harness.js";
+import type { ToolPolicy } from "./tool-policy.js";
+export declare const ASK_USER_QUESTION_TOOL_NAME = "AskUserQuestion";
+export interface AskQuestionOption {
+    label: string;
+    description: string;
+    preview?: string;
+}
+export interface AskQuestion {
+    question: string;
+    header: string;
+    options: AskQuestionOption[];
+    multiSelect?: boolean;
+}
+export interface AskQuestionRequest {
+    toolCallId: string;
+    questions: AskQuestion[];
+    readonly principal?: string;
+    readonly sourceTaskId?: string;
+}
+export interface QuestionAnswerItem {
+    header: string;
+    selected: string[];
+    note?: string;
+}
+export interface QuestionAnswer {
+    answers: QuestionAnswerItem[];
+}
+export type OnQuestion = (req: AskQuestionRequest, signal?: AbortSignal) => Promise<QuestionAnswer>;
+export declare function createDurableQuestionPolicy(): ToolPolicy;
+export declare const QUESTION_AWAITS_RESUME: OnQuestion;
+export declare function createAskUserQuestionTool(onQuestion?: OnQuestion, source?: {
+    principal?: string;
+    sourceTaskId?: string;
+}): AgentTool;
+//# sourceMappingURL=ask-question.d.ts.map

package/dist/core/ask-question.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ask-question.d.ts","sourceRoot":"","sources":["../../src/core/ask-question.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAInD,eAAO,MAAM,2BAA2B,oBAAoB,CAAC;AAgB7D,MAAM,WAAW,iBAAiB;IAEhC,KAAK,EAAE,MAAM,CAAC;IAEd,WAAW,EAAE,MAAM,CAAC;IAIpB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,WAAW;IAE1B,QAAQ,EAAE,MAAM,CAAC;IAEjB,MAAM,EAAE,MAAM,CAAC;IAEf,OAAO,EAAE,iBAAiB,EAAE,CAAC;IAE7B,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB;AAED,MAAM,WAAW,kBAAkB;IACjC,UAAU,EAAE,MAAM,CAAC;IAEnB,SAAS,EAAE,WAAW,EAAE,CAAC;IAOzB,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,CAAC;IAM5B,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC;CAChC;AAGD,MAAM,WAAW,kBAAkB;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,cAAc;IAC7B,OAAO,EAAE,kBAAkB,EAAE,CAAC;CAC/B;AAQD,MAAM,MAAM,UAAU,GAAG,CAAC,GAAG,EAAE,kBAAkB,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,cAAc,CAAC,CAAC;AA6BpG,wBAAgB,2BAA2B,IAAI,UAAU,CASxD;AAQD,eAAO,MAAM,sBAAsB,EAAE,UAWpC,CAAC;AAIF,wBAAgB,yBAAyB,CACvC,UAAU,CAAC,EAAE,UAAU,EAGvB,MAAM,CAAC,EAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAC;IAAC,YAAY,CAAC,EAAE,MAAM,CAAA;CAAE,GACrD,SAAS,CA8LX"}