@tetral-ai/sdk 0.110.0-alpha.1

package/lib/tools/SessionToolRunner.mjs

@@ -0,0 +1,470 @@
+var _SessionToolRunner_instances, _SessionToolRunner_consumed, _SessionToolRunner_controller, _SessionToolRunner_detachExternal, _SessionToolRunner_requestOpts, _SessionToolRunner_toolByName, _SessionToolRunner_logger, _SessionToolRunner_seen, _SessionToolRunner_answered, _SessionToolRunner_results, _SessionToolRunner_inFlightCount, _SessionToolRunner_onIdle, _SessionToolRunner_idleTimer, _SessionToolRunner_requestOptions, _SessionToolRunner_streamLoop, _SessionToolRunner_reconcile, _SessionToolRunner_ingestHistory, _SessionToolRunner_handleStreamEvent, _SessionToolRunner_armIdleTimer, _SessionToolRunner_disarmIdleTimer, _SessionToolRunner_execute, _SessionToolRunner_sendResult, _SessionToolRunner_drain;
+import { __classPrivateFieldGet, __classPrivateFieldSet } from "../../internal/tslib.mjs";
+import { AnthropicError } from "../../core/error.mjs";
+import { loggerFor } from "../../internal/utils/log.mjs";
+import { sleep } from "../../internal/utils/sleep.mjs";
+import { isFatal4xx } from "../../internal/utils/backoff.mjs";
+import { linkAbort } from "../../internal/utils/abort.mjs";
+import { AsyncQueue } from "../../internal/utils/async-queue.mjs";
+import { buildHeaders } from "../../internal/headers.mjs";
+import { helperHeader } from "../../internal/stainless-helper-header.mjs";
+import { runRunnableTool, toolName } from "./BetaRunnableTool.mjs";
+/** Beta header for the managed-agents API. */
+export const MANAGED_AGENTS_BETA = 'managed-agents-2026-04-01';
+const STREAM_BACKOFF_START_MS = 500;
+const STREAM_BACKOFF_CAP_MS = 10000;
+const TOOL_TIMEOUT_MS = 120000;
+const DRAIN_TIMEOUT_MS = 30000;
+const SEND_RETRIES = 3;
+/** Default {@link SessionToolRunnerOptions.maxIdleMs}: 60 seconds. */
+export const DEFAULT_MAX_IDLE_MS = 60000;
+/** Returns true if `ev` is a `session.status_idle` with `stop_reason` `end_turn`. */
+function isEndTurnIdle(ev) {
+    return ev.type === 'session.status_idle' && ev.stop_reason?.type === 'end_turn';
+}
+/**
+ * The sessions-side counterpart to `client.beta.messages.toolRunner`: an
+ * async-iterable that attaches to a managed-agents session, executes every
+ * incoming `agent.tool_use` and `agent.custom_tool_use` event against a local
+ * tool registry, posts the matching result back (`user.tool_result` for the
+ * former, `user.custom_tool_result` for the latter), and yields one
+ * {@link DispatchedToolCall} per completed call. Server-side `agent.mcp_tool_use`
+ * calls are not dispatched. Internally drives event-stream reconnect and result
+ * posting.
+ *
+ * Iteration ends when the session terminates (`session.status_terminated` /
+ * `session.deleted`), when the consumer `break`s out of the loop or aborts the
+ * supplied signal, or — once the session has gone idle with
+ * `stop_reason.type === "end_turn"` — when `maxIdleMs` elapses with no new
+ * event (any new event resets that countdown; it re-arms on the next `end_turn`
+ * idle; `maxIdleMs <= 0` disables it). The `finally` branch drains any in-flight
+ * tool calls and runs each tool's `close()` cleanup hook. It does *not* touch
+ * the work-item lease — wrap it in an `EnvironmentWorker` if you need
+ * heartbeating / force-stop.
+ *
+ * Tetral Cloud-hosted runtime rejects client-supplied tool-result event
+ * variants in this stage. When such a post is rejected, yielded calls have
+ * `posted: false`; consumers must not treat that as hosted-runtime success.
+ *
+ * @example
+ * ```ts
+ * import { betaAgentToolset20260401 } from '@tetral-ai/sdk/tools/agent-toolset/node';
+ *
+ * for await (const call of client.beta.sessions.events.toolRunner(work.data.id, {
+ *   tools: [...betaAgentToolset20260401({ workdir }), myTool],
+ * })) {
+ *   console.log(`${call.name} -> ${call.isError ? 'error' : 'ok'}`);
+ * }
+ * ```
+ */
+export class SessionToolRunner {
+    constructor(sessionId, opts) {
+        _SessionToolRunner_instances.add(this);
+        _SessionToolRunner_consumed.set(this, false);
+        _SessionToolRunner_controller.set(this, void 0);
+        _SessionToolRunner_detachExternal.set(this, void 0);
+        _SessionToolRunner_requestOpts.set(this, void 0);
+        _SessionToolRunner_toolByName.set(this, void 0);
+        _SessionToolRunner_logger.set(this, void 0);
+        _SessionToolRunner_seen.set(this, new Set());
+        _SessionToolRunner_answered.set(this, new Set());
+        _SessionToolRunner_results.set(this, new AsyncQueue());
+        _SessionToolRunner_inFlightCount.set(this, 0);
+        _SessionToolRunner_onIdle.set(this, null);
+        // When the session is idle past an `end_turn`, the pending stop timer; cleared
+        // by any new event. Event-driven — there is no polling watchdog.
+        _SessionToolRunner_idleTimer.set(this, void 0);
+        this.client = opts.client;
+        this.sessionId = sessionId;
+        this.tools = opts.tools;
+        this.maxIdleMs = opts.maxIdleMs ?? DEFAULT_MAX_IDLE_MS;
+        __classPrivateFieldSet(this, _SessionToolRunner_logger, loggerFor(opts.client), "f");
+        __classPrivateFieldSet(this, _SessionToolRunner_toolByName, new Map(opts.tools.map((t) => [toolName(t), t])), "f");
+        __classPrivateFieldSet(this, _SessionToolRunner_controller, new AbortController(), "f");
+        __classPrivateFieldSet(this, _SessionToolRunner_detachExternal, linkAbort(opts.signal, __classPrivateFieldGet(this, _SessionToolRunner_controller, "f")), "f");
+        __classPrivateFieldSet(this, _SessionToolRunner_requestOpts, opts.requestOptions, "f");
+    }
+    /** Read-only view of this runner's abort signal. */
+    get signal() {
+        return __classPrivateFieldGet(this, _SessionToolRunner_controller, "f").signal;
+    }
+    /** Abort the runner. Background tasks will wind down and `for await` will exit cleanly. */
+    abort() {
+        __classPrivateFieldGet(this, _SessionToolRunner_controller, "f").abort();
+    }
+    async *[(_SessionToolRunner_consumed = new WeakMap(), _SessionToolRunner_controller = new WeakMap(), _SessionToolRunner_detachExternal = new WeakMap(), _SessionToolRunner_requestOpts = new WeakMap(), _SessionToolRunner_toolByName = new WeakMap(), _SessionToolRunner_logger = new WeakMap(), _SessionToolRunner_seen = new WeakMap(), _SessionToolRunner_answered = new WeakMap(), _SessionToolRunner_results = new WeakMap(), _SessionToolRunner_inFlightCount = new WeakMap(), _SessionToolRunner_onIdle = new WeakMap(), _SessionToolRunner_idleTimer = new WeakMap(), _SessionToolRunner_instances = new WeakSet(), Symbol.asyncIterator)]() {
+        if (__classPrivateFieldGet(this, _SessionToolRunner_consumed, "f")) {
+            throw new AnthropicError('Cannot iterate over a consumed SessionToolRunner');
+        }
+        __classPrivateFieldSet(this, _SessionToolRunner_consumed, true, "f");
+        __classPrivateFieldGet(this, _SessionToolRunner_logger, "f").info('session tool runner starting', {
+            component: 'session-tool-runner',
+            session_id: this.sessionId,
+        });
+        // The one background promise: drives the event stream and dispatches tools.
+        // Its `.catch` aborts the controller so the main loop unwinds.
+        const streamPromise = __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_streamLoop).call(this).catch((e) => {
+            if (!__classPrivateFieldGet(this, _SessionToolRunner_controller, "f").signal.aborted) {
+                __classPrivateFieldGet(this, _SessionToolRunner_logger, "f").error('stream loop failed', { error: String(e) });
+            }
+            __classPrivateFieldGet(this, _SessionToolRunner_controller, "f").abort();
+        });
+        try {
+            // Phase 1: yield results as they arrive. `next(signal)` resolves
+            // `done: true` when the controller aborts — cancellation is handled in
+            // the queue read, no outer `Promise.race` needed.
+            while (true) {
+                const next = await __classPrivateFieldGet(this, _SessionToolRunner_results, "f").next(__classPrivateFieldGet(this, _SessionToolRunner_controller, "f").signal);
+                if (next.done)
+                    break;
+                yield next.value;
+            }
+            // Phase 2: let the stream loop settle (and push any final results), then
+            // drain whatever is still queued before closing.
+            await streamPromise;
+            let pending;
+            while ((pending = __classPrivateFieldGet(this, _SessionToolRunner_results, "f").tryShift()) !== undefined) {
+                yield pending;
+            }
+        }
+        finally {
+            __classPrivateFieldGet(this, _SessionToolRunner_controller, "f").abort();
+            __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_disarmIdleTimer).call(this);
+            // Re-await defensively in case the consumer broke out of phase 1 before
+            // phase 2 ran — a no-op if it already settled.
+            await streamPromise;
+            try {
+                await __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_drain).call(this);
+            }
+            catch (e) {
+                __classPrivateFieldGet(this, _SessionToolRunner_logger, "f").warn('drain failed', { error: String(e) });
+            }
+            __classPrivateFieldGet(this, _SessionToolRunner_results, "f").close();
+            for (const t of this.tools) {
+                try {
+                    // `close` is typed `() => Promisable<void>`, so a single `await`
+                    // covers both the sync and async return.
+                    await t.close?.();
+                }
+                catch (e) {
+                    __classPrivateFieldGet(this, _SessionToolRunner_logger, "f").warn('tool.close failed', { tool: toolName(t), error: String(e) });
+                }
+            }
+            // Detach from the external signal so the consumer can drop their signal
+            // reference without leaking this iterator instance.
+            __classPrivateFieldGet(this, _SessionToolRunner_detachExternal, "f").call(this);
+        }
+    }
+}
+_SessionToolRunner_requestOptions = function _SessionToolRunner_requestOptions() {
+    return {
+        ...__classPrivateFieldGet(this, _SessionToolRunner_requestOpts, "f"),
+        headers: buildHeaders([helperHeader('session-tool-runner'), __classPrivateFieldGet(this, _SessionToolRunner_requestOpts, "f")?.headers]),
+        signal: __classPrivateFieldGet(this, _SessionToolRunner_controller, "f").signal,
+    };
+}, _SessionToolRunner_streamLoop = 
+// ===== event stream =====
+async function _SessionToolRunner_streamLoop() {
+    const ctrl = __classPrivateFieldGet(this, _SessionToolRunner_controller, "f");
+    let backoff = STREAM_BACKOFF_START_MS;
+    while (!ctrl.signal.aborted) {
+        try {
+            // Establish the event stream *before* reconciling history, so an event
+            // emitted in the gap between listing and attaching is buffered on the
+            // stream rather than lost. `seen`/`answered` dedup any event that shows
+            // up both in the reconcile pass and on the live stream.
+            const stream = await this.client.beta.sessions.events.stream(this.sessionId, {}, __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_requestOptions).call(this));
+            await __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_reconcile).call(this);
+            for await (const ev of stream) {
+                backoff = STREAM_BACKOFF_START_MS;
+                if (await __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_handleStreamEvent).call(this, ev))
+                    return;
+            }
+        }
+        catch (e) {
+            // An abort throws to unwind the caller (the iterator's `streamPromise`
+            // `.catch`) rather than returning early and letting it carry on.
+            ctrl.signal.throwIfAborted();
+            if (isFatal4xx(e)) {
+                __classPrivateFieldGet(this, _SessionToolRunner_logger, "f").error('permanent stream failure, shutting down', { error: String(e) });
+                ctrl.abort();
+                throw e;
+            }
+            __classPrivateFieldGet(this, _SessionToolRunner_logger, "f").warn('stream disconnected, reconnecting', {
+                error: String(e),
+                backoff_ms: backoff,
+            });
+        }
+        ctrl.signal.throwIfAborted();
+        await sleep(backoff, ctrl.signal);
+        backoff = Math.min(backoff * 2, STREAM_BACKOFF_CAP_MS);
+    }
+}, _SessionToolRunner_reconcile = 
+/**
+ * Read full history before dispatching so a `tool_use` whose result appears
+ * later in the same history is not re-executed. Runs after the live stream is
+ * already attached (see {@link SessionToolRunner.#streamLoop}).
+ */
+async function _SessionToolRunner_reconcile() {
+    const ctrl = __classPrivateFieldGet(this, _SessionToolRunner_controller, "f");
+    const pending = [];
+    let lastWasEndTurn = false;
+    try {
+        for await (const ev of this.client.beta.sessions.events.list(this.sessionId, { limit: 1000 }, __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_requestOptions).call(this))) {
+            __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_ingestHistory).call(this, ev, pending);
+            lastWasEndTurn = isEndTurnIdle(ev);
+        }
+    }
+    catch (e) {
+        // An abort throws to unwind the caller; a real list failure is
+        // non-fatal — undo the speculative `seen` entries and let `#streamLoop`
+        // carry on with the live stream.
+        ctrl.signal.throwIfAborted();
+        __classPrivateFieldGet(this, _SessionToolRunner_logger, "f").warn('reconcile list failed', { error: String(e) });
+        // If list itself failed, undo the speculative `seen` entries so the next
+        // reconcile pass (or the live stream) can pick them up. Leave the idle
+        // timer untouched — the history we read may be incomplete.
+        for (const ev of pending)
+            __classPrivateFieldGet(this, _SessionToolRunner_seen, "f").delete(ev.id);
+        return;
+    }
+    const unanswered = pending.filter((ev) => !__classPrivateFieldGet(this, _SessionToolRunner_answered, "f").has(ev.id));
+    // If the most recent event in history is an `end_turn` idle and there's no
+    // outstanding tool work, the session is done — arm the idle timer so the
+    // runner stops even if that `end_turn` arrived during a disconnect.
+    if (lastWasEndTurn && unanswered.length === 0)
+        __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_armIdleTimer).call(this);
+    else
+        __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_disarmIdleTimer).call(this);
+    for (const ev of unanswered)
+        await __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_execute).call(this, ev);
+}, _SessionToolRunner_ingestHistory = function _SessionToolRunner_ingestHistory(ev, pending) {
+    if (ev.type === 'agent.tool_use' || ev.type === 'agent.custom_tool_use') {
+        // Mark the event seen so a replay on the live stream is not dispatched
+        // twice, but decide whether it still needs executing from `answered`, not
+        // `seen`: a call whose result post failed is seen-but-unanswered, and must
+        // be retried on the next reconcile pass rather than silently dropped.
+        __classPrivateFieldGet(this, _SessionToolRunner_seen, "f").add(ev.id);
+        if (!__classPrivateFieldGet(this, _SessionToolRunner_answered, "f").has(ev.id))
+            pending.push(ev);
+    }
+    else if (ev.type === 'user.tool_result') {
+        __classPrivateFieldGet(this, _SessionToolRunner_answered, "f").add(ev.tool_use_id);
+    }
+    else if (ev.type === 'user.custom_tool_result') {
+        __classPrivateFieldGet(this, _SessionToolRunner_answered, "f").add(ev.custom_tool_use_id);
+    }
+}, _SessionToolRunner_handleStreamEvent = 
+/** Returns true when the runner should exit. */
+async function _SessionToolRunner_handleStreamEvent(ev) {
+    // Arm/disarm the idle timer: an `end_turn` idle starts the grace countdown;
+    // any other event cancels it.
+    if (isEndTurnIdle(ev))
+        __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_armIdleTimer).call(this);
+    else
+        __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_disarmIdleTimer).call(this);
+    switch (ev.type) {
+        case 'agent.tool_use':
+        case 'agent.custom_tool_use':
+            if (!__classPrivateFieldGet(this, _SessionToolRunner_seen, "f").has(ev.id)) {
+                __classPrivateFieldGet(this, _SessionToolRunner_seen, "f").add(ev.id);
+                await __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_execute).call(this, ev);
+            }
+            return false;
+        case 'user.tool_result':
+            __classPrivateFieldGet(this, _SessionToolRunner_answered, "f").add(ev.tool_use_id);
+            return false;
+        case 'user.custom_tool_result':
+            __classPrivateFieldGet(this, _SessionToolRunner_answered, "f").add(ev.custom_tool_use_id);
+            return false;
+        case 'session.status_terminated':
+        case 'session.deleted':
+            __classPrivateFieldGet(this, _SessionToolRunner_logger, "f").info('session terminated', {
+                component: 'session-tool-runner',
+                session_id: this.sessionId,
+            });
+            __classPrivateFieldGet(this, _SessionToolRunner_controller, "f").abort();
+            return true;
+        default:
+            return false;
+    }
+}, _SessionToolRunner_armIdleTimer = function _SessionToolRunner_armIdleTimer() {
+    __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_disarmIdleTimer).call(this);
+    if (this.maxIdleMs <= 0)
+        return;
+    __classPrivateFieldSet(this, _SessionToolRunner_idleTimer, setTimeout(() => {
+        __classPrivateFieldGet(this, _SessionToolRunner_logger, "f").info('session idle after end_turn; stopping', {
+            component: 'session-tool-runner',
+            session_id: this.sessionId,
+            max_idle_ms: this.maxIdleMs,
+        });
+        __classPrivateFieldGet(this, _SessionToolRunner_controller, "f").abort();
+    }, this.maxIdleMs), "f");
+}, _SessionToolRunner_disarmIdleTimer = function _SessionToolRunner_disarmIdleTimer() {
+    if (__classPrivateFieldGet(this, _SessionToolRunner_idleTimer, "f") !== undefined) {
+        clearTimeout(__classPrivateFieldGet(this, _SessionToolRunner_idleTimer, "f"));
+        __classPrivateFieldSet(this, _SessionToolRunner_idleTimer, undefined, "f");
+    }
+}, _SessionToolRunner_execute = 
+// ===== tool execution =====
+async function _SessionToolRunner_execute(ev) {
+    var _a, _b;
+    if (__classPrivateFieldGet(this, _SessionToolRunner_answered, "f").has(ev.id))
+        return;
+    __classPrivateFieldGet(this, _SessionToolRunner_logger, "f").info('executing tool', {
+        component: 'session-tool-runner',
+        session_id: this.sessionId,
+        tool: ev.name,
+        tool_use_id: ev.id,
+    });
+    __classPrivateFieldSet(this, _SessionToolRunner_inFlightCount, (_a = __classPrivateFieldGet(this, _SessionToolRunner_inFlightCount, "f"), _a++, _a), "f");
+    try {
+        const tool = __classPrivateFieldGet(this, _SessionToolRunner_toolByName, "f").get(ev.name);
+        if (!tool) {
+            // Skip (split-client partial fulfilment): a name this runner
+            // is not registered for belongs to the other client servicing this
+            // session (typically the customer's app backend handling custom tools).
+            // Post NO result, do not mark it answered, and leave the tool_use_id
+            // pending for its owner — claiming it would corrupt the conversation.
+            // Still yield the call so the consumer can observe the unowned
+            // dispatch; nothing was sent, so `posted`/`isError` stay false and no
+            // `result` event is populated. The id stays unanswered, so reconcile
+            // keeps it out of the idle/end-turn accounting and re-surfaces it after
+            // a reconnect until its owner answers it.
+            __classPrivateFieldGet(this, _SessionToolRunner_logger, "f").info('tool not owned by this runner; leaving the tool_use_id pending for its owner', {
+                component: 'session-tool-runner',
+                session_id: this.sessionId,
+                tool: ev.name,
+                tool_use_id: ev.id,
+            });
+            __classPrivateFieldGet(this, _SessionToolRunner_results, "f").push({ event: ev, toolUseId: ev.id, name: ev.name, isError: false, posted: false });
+            return;
+        }
+        let content;
+        let isError;
+        // Per-tool controller: aborts on the runner's own signal *or* the
+        // per-tool timeout, so an in-flight tool stops promptly when the runner
+        // is aborted instead of running until the timeout.
+        const toolCtrl = new AbortController();
+        const detachTool = linkAbort(__classPrivateFieldGet(this, _SessionToolRunner_controller, "f").signal, toolCtrl);
+        const timer = setTimeout(() => toolCtrl.abort(), TOOL_TIMEOUT_MS);
+        try {
+            // Pass the source `agent.tool_use` / `agent.custom_tool_use` event
+            // straight through as the run context's `toolUse` — it is a union
+            // member of `BetaToolUse`, no Messages-block adapter needed.
+            const outcome = await runRunnableTool(tool, ev.input, {
+                toolUse: ev,
+                toolUseBlock: ev,
+                signal: toolCtrl.signal,
+            });
+            content = outcome.content;
+            isError = outcome.isError;
+        }
+        finally {
+            clearTimeout(timer);
+            detachTool();
+        }
+        // Answer with the result event that matches the call kind: a
+        // `user.tool_result` for an `agent.tool_use`, a `user.custom_tool_result`
+        // for an `agent.custom_tool_use`. Posting the wrong one leaves the call
+        // unanswered and the session stuck.
+        const result = buildResultEvent(ev, isError, toSessionContent(content));
+        const posted = await __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_sendResult).call(this, result, ev.id);
+        __classPrivateFieldGet(this, _SessionToolRunner_results, "f").push({
+            event: ev,
+            result,
+            toolUseId: ev.id,
+            name: ev.name,
+            isError,
+            posted,
+        });
+    }
+    finally {
+        __classPrivateFieldSet(this, _SessionToolRunner_inFlightCount, (_b = __classPrivateFieldGet(this, _SessionToolRunner_inFlightCount, "f"), _b--, _b), "f");
+        if (__classPrivateFieldGet(this, _SessionToolRunner_inFlightCount, "f") === 0)
+            __classPrivateFieldGet(this, _SessionToolRunner_onIdle, "f")?.call(this);
+    }
+}, _SessionToolRunner_sendResult = async function _SessionToolRunner_sendResult(result, toolUseId) {
+    const ctrl = __classPrivateFieldGet(this, _SessionToolRunner_controller, "f");
+    let lastErr;
+    for (let i = 0; i < SEND_RETRIES; i++) {
+        // An abort throws to unwind the caller rather than returning a
+        // `posted: false` result the iterator would carry on past.
+        ctrl.signal.throwIfAborted();
+        try {
+            await this.client.beta.sessions.events.send(this.sessionId, { events: [result] }, __classPrivateFieldGet(this, _SessionToolRunner_instances, "m", _SessionToolRunner_requestOptions).call(this));
+            __classPrivateFieldGet(this, _SessionToolRunner_answered, "f").add(toolUseId);
+            return true;
+        }
+        catch (e) {
+            lastErr = e;
+            // Only short-circuit on a permanent 4xx; 408/409/429 deserve the
+            // remaining retries (aligned with the core client's retry policy).
+            if (isFatal4xx(e))
+                break;
+            // Back off only *between* attempts — never after the final one, since
+            // there is no further try left to wait for.
+            if (i < SEND_RETRIES - 1)
+                await sleep((i + 1) * 1000, ctrl.signal);
+        }
+    }
+    __classPrivateFieldGet(this, _SessionToolRunner_logger, "f").error('failed to send tool result', {
+        tool_use_id: toolUseId,
+        error: String(lastErr),
+    });
+    return false;
+}, _SessionToolRunner_drain = 
+/** Wait (bounded) for in-flight tool executions to finish during teardown. */
+async function _SessionToolRunner_drain() {
+    if (__classPrivateFieldGet(this, _SessionToolRunner_inFlightCount, "f") === 0)
+        return;
+    await Promise.race([new Promise((r) => (__classPrivateFieldSet(this, _SessionToolRunner_onIdle, r, "f"))), sleep(DRAIN_TIMEOUT_MS)]);
+    __classPrivateFieldSet(this, _SessionToolRunner_onIdle, null, "f");
+    if (__classPrivateFieldGet(this, _SessionToolRunner_inFlightCount, "f") > 0) {
+        __classPrivateFieldGet(this, _SessionToolRunner_logger, "f").warn('drain timeout exceeded');
+    }
+};
+/**
+ * Build the result event that answers `ev`: a `user.tool_result` for a builtin
+ * `agent.tool_use`, a `user.custom_tool_result` for a custom
+ * `agent.custom_tool_use`. The two `(use, result)` pairs are distinct API event
+ * types and must be matched exactly — a `user.tool_result` does not answer a
+ * custom tool call.
+ */
+function buildResultEvent(ev, isError, content) {
+    if (ev.type === 'agent.custom_tool_use') {
+        return { type: 'user.custom_tool_result', custom_tool_use_id: ev.id, is_error: isError, content };
+    }
+    return { type: 'user.tool_result', tool_use_id: ev.id, is_error: isError, content };
+}
+// The Messages-API tool-result block union is wider than the Sessions-API
+// tool_result content union; pass through text/image/document and stringify
+// anything else so a BetaRunnableTool authored for toolRunner still works here.
+function toSessionContent(content) {
+    if (typeof content === 'string')
+        return [{ type: 'text', text: content || '(no output)' }];
+    const out = content.map((b) => {
+        if (b.type === 'text')
+            return { type: 'text', text: b.text || '(no output)' };
+        if (b.type === 'image' || b.type === 'document')
+            return b;
+        if (b.type === 'search_result') {
+            // The Messages `search_result` block param maps field-for-field onto the
+            // Sessions `BetaManagedAgentsSearchResultBlock`; map it explicitly rather
+            // than letting it fall through to the JSON.stringify branch (which would
+            // bury a structured result inside a text block). `citations` is required
+            // on the Sessions side and optional on the Messages side — default the
+            // flag to `false` when the producer left it unset.
+            return {
+                type: 'search_result',
+                source: b.source,
+                title: b.title,
+                content: b.content.map((c) => ({ type: 'text', text: c.text })),
+                citations: { enabled: b.citations?.enabled ?? false },
+            };
+        }
+        return { type: 'text', text: JSON.stringify(b) };
+    });
+    return out.length > 0 ? out : [{ type: 'text', text: '(no output)' }];
+}
+//# sourceMappingURL=SessionToolRunner.mjs.map

package/lib/tools/SessionToolRunner.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"SessionToolRunner.mjs","sourceRoot":"","sources":["../../src/lib/tools/SessionToolRunner.ts"],"names":[],"mappings":";;AAAA,OAAO,EAAE,cAAc,EAAE,6BAAyB;AAWlD,OAAO,EAAE,SAAS,EAAe,qCAAiC;AAClE,OAAO,EAAE,KAAK,EAAE,uCAAmC;AACnD,OAAO,EAAE,UAAU,EAAE,yCAAqC;AAC1D,OAAO,EAAE,SAAS,EAAE,uCAAmC;AACvD,OAAO,EAAE,UAAU,EAAE,6CAAyC;AAC9D,OAAO,EAAE,YAAY,EAAE,mCAA+B;AACtD,OAAO,EAAE,YAAY,EAAE,mDAA+C;AAEtE,OAAO,EAAE,eAAe,EAAE,QAAQ,EAAyB,+BAA2B;AAGtF,8CAA8C;AAC9C,MAAM,CAAC,MAAM,mBAAmB,GAAG,2BAA2B,CAAC;AAE/D,MAAM,uBAAuB,GAAG,GAAG,CAAC;AACpC,MAAM,qBAAqB,GAAG,KAAM,CAAC;AACrC,MAAM,eAAe,GAAG,MAAO,CAAC;AAChC,MAAM,gBAAgB,GAAG,KAAM,CAAC;AAChC,MAAM,YAAY,GAAG,CAAC,CAAC;AAuDvB,sEAAsE;AACtE,MAAM,CAAC,MAAM,mBAAmB,GAAG,KAAM,CAAC;AAsD1C,qFAAqF;AACrF,SAAS,aAAa,CAAC,EAAsD;IAC3E,OAAO,EAAE,CAAC,IAAI,KAAK,qBAAqB,IAAI,EAAE,CAAC,WAAW,EAAE,IAAI,KAAK,UAAU,CAAC;AAClF,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACH,MAAM,OAAO,iBAAiB;IAqB5B,YAAY,SAAiB,EAAE,IAA8B;;QAf7D,sCAAY,KAAK,EAAC;QACT,gDAA6B;QAC7B,oDAA4B;QAC5B,iDAAuD;QACvD,gDAA2C;QAC3C,4CAAgB;QAChB,kCAAQ,IAAI,GAAG,EAAU,EAAC;QAC1B,sCAAY,IAAI,GAAG,EAAU,EAAC;QAC9B,qCAAW,IAAI,UAAU,EAAsB,EAAC;QACzD,2CAAiB,CAAC,EAAC;QACnB,oCAA+B,IAAI,EAAC;QACpC,+EAA+E;QAC/E,iEAAiE;QACjE,+CAAsD;QAGpD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACxB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,mBAAmB,CAAC;QACvD,uBAAA,IAAI,6BAAW,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,MAAA,CAAC;QACtC,uBAAA,IAAI,iCAAe,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAA,CAAC;QACpE,uBAAA,IAAI,iCAAe,IAAI,eAAe,EAAE,MAAA,CAAC;QACzC,uBAAA,IAAI,qCAAmB,SAAS,CAAC,IAAI,CAAC,MAAM,EAAE,uBAAA,IAAI,qCAAY,CAAC,MAAA,CAAC;QAChE,uBAAA,IAAI,kCAAgB,IAAI,CAAC,cAAc,MAAA,CAAC;IAC1C,CAAC;IAED,oDAAoD;IACpD,IAAI,MAAM;QACR,OAAO,uBAAA,IAAI,qCAAY,CAAC,MAAM,CAAC;IACjC,CAAC;IAED,2FAA2F;IAC3F,KAAK;QACH,uBAAA,IAAI,qCAAY,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,CAAC,slBAAC,MAAM,CAAC,aAAa,EAAC;QAC3B,IAAI,uBAAA,IAAI,mCAAU,EAAE,CAAC;YACnB,MAAM,IAAI,cAAc,CAAC,kDAAkD,CAAC,CAAC;QAC/E,CAAC;QACD,uBAAA,IAAI,+BAAa,IAAI,MAAA,CAAC;QACtB,uBAAA,IAAI,iCAAQ,CAAC,IAAI,CAAC,8BAA8B,EAAE;YAChD,SAAS,EAAE,qBAAqB;YAChC,UAAU,EAAE,IAAI,CAAC,SAAS;SAC3B,CAAC,CAAC;QAEH,4EAA4E;QAC5E,+DAA+D;QAC/D,MAAM,aAAa,GAAG,uBAAA,IAAI,mEAAY,MAAhB,IAAI,CAAc,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;YACnD,IAAI,CAAC,uBAAA,IAAI,qCAAY,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACrC,uBAAA,IAAI,iCAAQ,CAAC,KAAK,CAAC,oBAAoB,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACjE,CAAC;YACD,uBAAA,IAAI,qCAAY,CAAC,KAAK,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,iEAAiE;YACjE,uEAAuE;YACvE,kDAAkD;YAClD,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,IAAI,GAAG,MAAM,uBAAA,IAAI,kCAAS,CAAC,IAAI,CAAC,uBAAA,IAAI,qCAAY,CAAC,MAAM,CAAC,CAAC;gBAC/D,IAAI,IAAI,CAAC,IAAI;oBAAE,MAAM;gBACrB,MAAM,IAAI,CAAC,KAAK,CAAC;YACnB,CAAC;YAED,yEAAyE;YACzE,iDAAiD;YACjD,MAAM,aAAa,CAAC;YACpB,IAAI,OAAuC,CAAC;YAC5C,OAAO,CAAC,OAAO,GAAG,uBAAA,IAAI,kCAAS,CAAC,QAAQ,EAAE,CAAC,KAAK,SAAS,EAAE,CAAC;gBAC1D,MAAM,OAAO,CAAC;YAChB,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,uBAAA,IAAI,qCAAY,CAAC,KAAK,EAAE,CAAC;YACzB,uBAAA,IAAI,wEAAiB,MAArB,IAAI,CAAmB,CAAC;YACxB,wEAAwE;YACxE,+CAA+C;YAC/C,MAAM,aAAa,CAAC;YACpB,IAAI,CAAC;gBACH,MAAM,uBAAA,IAAI,8DAAO,MAAX,IAAI,CAAS,CAAC;YACtB,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,uBAAA,IAAI,iCAAQ,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAC1D,CAAC;YACD,uBAAA,IAAI,kCAAS,CAAC,KAAK,EAAE,CAAC;YACtB,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBAC3B,IAAI,CAAC;oBACH,iEAAiE;oBACjE,yCAAyC;oBACzC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC;gBACpB,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,uBAAA,IAAI,iCAAQ,CAAC,IAAI,CAAC,mBAAmB,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBAClF,CAAC;YACH,CAAC;YACD,wEAAwE;YACxE,oDAAoD;YACpD,uBAAA,IAAI,yCAAgB,MAApB,IAAI,CAAkB,CAAC;QACzB,CAAC;IACH,CAAC;CA8RF;;IApRG,OAAO;QACL,GAAG,uBAAA,IAAI,sCAAa;QACpB,OAAO,EAAE,YAAY,CAAC,CAAC,YAAY,CAAC,qBAAqB,CAAC,EAAE,uBAAA,IAAI,sCAAa,EAAE,OAAO,CAAC,CAAC;QACxF,MAAM,EAAE,uBAAA,IAAI,qCAAY,CAAC,MAAM;KAChC,CAAC;AACJ,CAAC;AAED,2BAA2B;AAE3B,KAAK;IACH,MAAM,IAAI,GAAG,uBAAA,IAAI,qCAAY,CAAC;IAC9B,IAAI,OAAO,GAAG,uBAAuB,CAAC;IACtC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QAC5B,IAAI,CAAC;YACH,uEAAuE;YACvE,sEAAsE;YACtE,wEAAwE;YACxE,wDAAwD;YACxD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAC1D,IAAI,CAAC,SAAS,EACd,EAAE,EACF,uBAAA,IAAI,uEAAgB,MAApB,IAAI,CAAkB,CACvB,CAAC;YACF,MAAM,uBAAA,IAAI,kEAAW,MAAf,IAAI,CAAa,CAAC;YACxB,IAAI,KAAK,EAAE,MAAM,EAAE,IAAI,MAAM,EAAE,CAAC;gBAC9B,OAAO,GAAG,uBAAuB,CAAC;gBAClC,IAAI,MAAM,uBAAA,IAAI,0EAAmB,MAAvB,IAAI,EAAoB,EAAE,CAAC;oBAAE,OAAO;YAChD,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,uEAAuE;YACvE,iEAAiE;YACjE,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YAC7B,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;gBAClB,uBAAA,IAAI,iCAAQ,CAAC,KAAK,CAAC,yCAAyC,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACpF,IAAI,CAAC,KAAK,EAAE,CAAC;gBACb,MAAM,CAAC,CAAC;YACV,CAAC;YACD,uBAAA,IAAI,iCAAQ,CAAC,IAAI,CAAC,mCAAmC,EAAE;gBACrD,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;gBAChB,UAAU,EAAE,OAAO;aACpB,CAAC,CAAC;QACL,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;QAC7B,MAAM,KAAK,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAClC,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,GAAG,CAAC,EAAE,qBAAqB,CAAC,CAAC;IACzD,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,KAAK;IACH,MAAM,IAAI,GAAG,uBAAA,IAAI,qCAAY,CAAC;IAC9B,MAAM,OAAO,GAA6B,EAAE,CAAC;IAC7C,IAAI,cAAc,GAAG,KAAK,CAAC;IAC3B,IAAI,CAAC;QACH,IAAI,KAAK,EAAE,MAAM,EAAE,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAC1D,IAAI,CAAC,SAAS,EACd,EAAE,KAAK,EAAE,IAAI,EAAE,EACf,uBAAA,IAAI,uEAAgB,MAApB,IAAI,CAAkB,CACvB,EAAE,CAAC;YACF,uBAAA,IAAI,sEAAe,MAAnB,IAAI,EAAgB,EAAE,EAAE,OAAO,CAAC,CAAC;YACjC,cAAc,GAAG,aAAa,CAAC,EAAE,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,+DAA+D;QAC/D,wEAAwE;QACxE,iCAAiC;QACjC,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;QAC7B,uBAAA,IAAI,iCAAQ,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACjE,yEAAyE;QACzE,uEAAuE;QACvE,2DAA2D;QAC3D,KAAK,MAAM,EAAE,IAAI,OAAO;YAAE,uBAAA,IAAI,+BAAM,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QACnD,OAAO;IACT,CAAC;IACD,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,uBAAA,IAAI,mCAAU,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACtE,2EAA2E;IAC3E,yEAAyE;IACzE,oEAAoE;IACpE,IAAI,cAAc,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;QAAE,uBAAA,IAAI,qEAAc,MAAlB,IAAI,CAAgB,CAAC;;QAC/D,uBAAA,IAAI,wEAAiB,MAArB,IAAI,CAAmB,CAAC;IAC7B,KAAK,MAAM,EAAE,IAAI,UAAU;QAAE,MAAM,uBAAA,IAAI,gEAAS,MAAb,IAAI,EAAU,EAAE,CAAC,CAAC;AACvD,CAAC,+EAEc,EAAiC,EAAE,OAAiC;IACjF,IAAI,EAAE,CAAC,IAAI,KAAK,gBAAgB,IAAI,EAAE,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;QACxE,uEAAuE;QACvE,0EAA0E;QAC1E,2EAA2E;QAC3E,sEAAsE;QACtE,uBAAA,IAAI,+BAAM,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QACtB,IAAI,CAAC,uBAAA,IAAI,mCAAU,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;YAAE,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACnD,CAAC;SAAM,IAAI,EAAE,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;QAC1C,uBAAA,IAAI,mCAAU,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC;IACrC,CAAC;SAAM,IAAI,EAAE,CAAC,IAAI,KAAK,yBAAyB,EAAE,CAAC;QACjD,uBAAA,IAAI,mCAAU,CAAC,GAAG,CAAC,EAAE,CAAC,kBAAkB,CAAC,CAAC;IAC5C,CAAC;AACH,CAAC;AAED,gDAAgD;AAChD,KAAK,+CAAoB,EAAwC;IAC/D,4EAA4E;IAC5E,8BAA8B;IAC9B,IAAI,aAAa,CAAC,EAAE,CAAC;QAAE,uBAAA,IAAI,qEAAc,MAAlB,IAAI,CAAgB,CAAC;;QACvC,uBAAA,IAAI,wEAAiB,MAArB,IAAI,CAAmB,CAAC;IAC7B,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC;QAChB,KAAK,gBAAgB,CAAC;QACtB,KAAK,uBAAuB;YAC1B,IAAI,CAAC,uBAAA,IAAI,+BAAM,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;gBAC3B,uBAAA,IAAI,+BAAM,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;gBACtB,MAAM,uBAAA,IAAI,gEAAS,MAAb,IAAI,EAAU,EAAE,CAAC,CAAC;YAC1B,CAAC;YACD,OAAO,KAAK,CAAC;QACf,KAAK,kBAAkB;YACrB,uBAAA,IAAI,mCAAU,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC;YACnC,OAAO,KAAK,CAAC;QACf,KAAK,yBAAyB;YAC5B,uBAAA,IAAI,mCAAU,CAAC,GAAG,CAAC,EAAE,CAAC,kBAAkB,CAAC,CAAC;YAC1C,OAAO,KAAK,CAAC;QACf,KAAK,2BAA2B,CAAC;QACjC,KAAK,iBAAiB;YACpB,uBAAA,IAAI,iCAAQ,CAAC,IAAI,CAAC,oBAAoB,EAAE;gBACtC,SAAS,EAAE,qBAAqB;gBAChC,UAAU,EAAE,IAAI,CAAC,SAAS;aAC3B,CAAC,CAAC;YACH,uBAAA,IAAI,qCAAY,CAAC,KAAK,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC;QACd;YACE,OAAO,KAAK,CAAC;IACjB,CAAC;AACH,CAAC;IAMC,uBAAA,IAAI,wEAAiB,MAArB,IAAI,CAAmB,CAAC;IACxB,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC;QAAE,OAAO;IAChC,uBAAA,IAAI,gCAAc,UAAU,CAAC,GAAG,EAAE;QAChC,uBAAA,IAAI,iCAAQ,CAAC,IAAI,CAAC,uCAAuC,EAAE;YACzD,SAAS,EAAE,qBAAqB;YAChC,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,WAAW,EAAE,IAAI,CAAC,SAAS;SAC5B,CAAC,CAAC;QACH,uBAAA,IAAI,qCAAY,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAA,CAAC;AACrB,CAAC;IAIC,IAAI,uBAAA,IAAI,oCAAW,KAAK,SAAS,EAAE,CAAC;QAClC,YAAY,CAAC,uBAAA,IAAI,oCAAW,CAAC,CAAC;QAC9B,uBAAA,IAAI,gCAAc,SAAS,MAAA,CAAC;IAC9B,CAAC;AACH,CAAC;AAED,6BAA6B;AAE7B,KAAK,qCAAU,EAA0B;;IACvC,IAAI,uBAAA,IAAI,mCAAU,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;QAAE,OAAO;IACtC,uBAAA,IAAI,iCAAQ,CAAC,IAAI,CAAC,gBAAgB,EAAE;QAClC,SAAS,EAAE,qBAAqB;QAChC,UAAU,EAAE,IAAI,CAAC,SAAS;QAC1B,IAAI,EAAE,EAAE,CAAC,IAAI;QACb,WAAW,EAAE,EAAE,CAAC,EAAE;KACnB,CAAC,CAAC;IACH,+DAAA,CAAA,wEAAmB,EAAnB,IAAqB,IAAA,CAAA,MAAA,CAAC;IACtB,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,uBAAA,IAAI,qCAAY,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;QAC3C,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,6DAA6D;YAC7D,mEAAmE;YACnE,wEAAwE;YACxE,qEAAqE;YACrE,sEAAsE;YACtE,+DAA+D;YAC/D,sEAAsE;YACtE,qEAAqE;YACrE,wEAAwE;YACxE,0CAA0C;YAC1C,uBAAA,IAAI,iCAAQ,CAAC,IAAI,CAAC,8EAA8E,EAAE;gBAChG,SAAS,EAAE,qBAAqB;gBAChC,UAAU,EAAE,IAAI,CAAC,SAAS;gBAC1B,IAAI,EAAE,EAAE,CAAC,IAAI;gBACb,WAAW,EAAE,EAAE,CAAC,EAAE;aACnB,CAAC,CAAC;YACH,uBAAA,IAAI,kCAAS,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;YAClG,OAAO;QACT,CAAC;QACD,IAAI,OAAwD,CAAC;QAC7D,IAAI,OAAgB,CAAC;QACrB,kEAAkE;QAClE,wEAAwE;QACxE,mDAAmD;QACnD,MAAM,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAC;QACvC,MAAM,UAAU,GAAG,SAAS,CAAC,uBAAA,IAAI,qCAAY,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAChE,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,KAAK,EAAE,EAAE,eAAe,CAAC,CAAC;QAClE,IAAI,CAAC;YACH,mEAAmE;YACnE,kEAAkE;YAClE,6DAA6D;YAC7D,MAAM,OAAO,GAAG,MAAM,eAAe,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,EAAE;gBACpD,OAAO,EAAE,EAAE;gBACX,YAAY,EAAE,EAAE;gBAChB,MAAM,EAAE,QAAQ,CAAC,MAAM;aACxB,CAAC,CAAC;YACH,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;YAC1B,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAC5B,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,UAAU,EAAE,CAAC;QACf,CAAC;QACD,6DAA6D;QAC7D,0EAA0E;QAC1E,wEAAwE;QACxE,oCAAoC;QACpC,MAAM,MAAM,GAAG,gBAAgB,CAAC,EAAE,EAAE,OAAO,EAAE,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;QACxE,MAAM,MAAM,GAAG,MAAM,uBAAA,IAAI,mEAAY,MAAhB,IAAI,EAAa,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;QACrD,uBAAA,IAAI,kCAAS,CAAC,IAAI,CAAC;YACjB,KAAK,EAAE,EAAE;YACT,MAAM;YACN,SAAS,EAAE,EAAE,CAAC,EAAE;YAChB,IAAI,EAAE,EAAE,CAAC,IAAI;YACb,OAAO;YACP,MAAM;SACP,CAAC,CAAC;IACL,CAAC;YAAS,CAAC;QACT,+DAAA,CAAA,wEAAmB,EAAnB,IAAqB,IAAA,CAAA,MAAA,CAAC;QACtB,IAAI,uBAAA,IAAI,wCAAe,KAAK,CAAC;YAAE,uBAAA,IAAI,iCAAQ,EAAE,KAAd,IAAI,CAAY,CAAC;IAClD,CAAC;AACH,CAAC,kCAED,KAAK,wCAAa,MAAkC,EAAE,SAAiB;IACrE,MAAM,IAAI,GAAG,uBAAA,IAAI,qCAAY,CAAC;IAC9B,IAAI,OAAgB,CAAC;IACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,+DAA+D;QAC/D,2DAA2D;QAC3D,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CACzC,IAAI,CAAC,SAAS,EACd,EAAE,MAAM,EAAE,CAAC,MAAM,CAAC,EAAE,EACpB,uBAAA,IAAI,uEAAgB,MAApB,IAAI,CAAkB,CACvB,CAAC;YACF,uBAAA,IAAI,mCAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAC9B,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,GAAG,CAAC,CAAC;YACZ,iEAAiE;YACjE,mEAAmE;YACnE,IAAI,UAAU,CAAC,CAAC,CAAC;gBAAE,MAAM;YACzB,sEAAsE;YACtE,4CAA4C;YAC5C,IAAI,CAAC,GAAG,YAAY,GAAG,CAAC;gBAAE,MAAM,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;IACD,uBAAA,IAAI,iCAAQ,CAAC,KAAK,CAAC,4BAA4B,EAAE;QAC/C,WAAW,EAAE,SAAS;QACtB,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC;KACvB,CAAC,CAAC;IACH,OAAO,KAAK,CAAC;AACf,CAAC;AAED,8EAA8E;AAC9E,KAAK;IACH,IAAI,uBAAA,IAAI,wCAAe,KAAK,CAAC;QAAE,OAAO;IACtC,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,uBAAA,IAAI,6BAAW,CAAC,MAAA,CAAC,CAAC,EAAE,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;IAC5F,uBAAA,IAAI,6BAAW,IAAI,MAAA,CAAC;IACpB,IAAI,uBAAA,IAAI,wCAAe,GAAG,CAAC,EAAE,CAAC;QAC5B,uBAAA,IAAI,iCAAQ,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAC9C,CAAC;AACH,CAAC;AAGH;;;;;;GAMG;AACH,SAAS,gBAAgB,CACvB,EAA0B,EAC1B,OAAgB,EAChB,OAA8B;IAE9B,IAAI,EAAE,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;QACxC,OAAO,EAAE,IAAI,EAAE,yBAAyB,EAAE,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;IACpG,CAAC;IACD,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;AACtF,CAAC;AAED,0EAA0E;AAC1E,4EAA4E;AAC5E,gFAAgF;AAChF,SAAS,gBAAgB,CAAC,OAAwD;IAChF,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,IAAI,aAAa,EAAE,CAAC,CAAC;IAC3F,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAuB,EAAE;QACjD,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM;YAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,aAAa,EAAE,CAAC;QAC9E,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,IAAI,CAAC,CAAC,IAAI,KAAK,UAAU;YAAE,OAAO,CAAwB,CAAC;QACjF,IAAI,CAAC,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;YAC/B,yEAAyE;YACzE,0EAA0E;YAC1E,yEAAyE;YACzE,yEAAyE;YACzE,uEAAuE;YACvE,mDAAmD;YACnD,OAAO;gBACL,IAAI,EAAE,eAAe;gBACrB,MAAM,EAAE,CAAC,CAAC,MAAM;gBAChB,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC/D,SAAS,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,SAAS,EAAE,OAAO,IAAI,KAAK,EAAE;aACtD,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;IACnD,CAAC,CAAC,CAAC;IACH,OAAO,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC;AACxE,CAAC"}

package/lib/tools/ToolError.d.mts

@@ -0,0 +1,33 @@
+import { BetaToolResultContentBlockParam } from "../../resources/beta.mjs";
+/**
+ * An error that can be thrown from a tool's `run` method to return structured
+ * content blocks as the error result, rather than just a string message.
+ *
+ * When the ToolRunner catches this error, it will use the `content` property
+ * as the tool result with `is_error: true`.
+ *
+ * @example
+ * ```ts
+ * const tool = {
+ *   name: 'my_tool',
+ *   run: async (input) => {
+ *     if (somethingWentWrong) {
+ *       throw new ToolError([
+ *         { type: 'text', text: 'Error details here' },
+ *         { type: 'image', source: { type: 'base64', data: '...', media_type: 'image/png' } },
+ *       ]);
+ *     }
+ *     return 'success';
+ *   },
+ * };
+ * ```
+ */
+export declare class ToolError extends Error {
+    /**
+     * The content to return as the tool result. This will be sent back to the model
+     * with `is_error: true`.
+     */
+    readonly content: string | Array<BetaToolResultContentBlockParam>;
+    constructor(content: string | Array<BetaToolResultContentBlockParam>);
+}
+//# sourceMappingURL=ToolError.d.mts.map

package/lib/tools/ToolError.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolError.d.mts","sourceRoot":"","sources":["../../src/lib/tools/ToolError.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,+BAA+B,EAAE,iCAA6B;AAEvE;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,qBAAa,SAAU,SAAQ,KAAK;IAClC;;;OAGG;IACH,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,KAAK,CAAC,+BAA+B,CAAC,CAAC;gBAEtD,OAAO,EAAE,MAAM,GAAG,KAAK,CAAC,+BAA+B,CAAC;CAcrE"}

package/lib/tools/ToolError.d.ts

@@ -0,0 +1,33 @@
+import { BetaToolResultContentBlockParam } from "../../resources/beta.js";
+/**
+ * An error that can be thrown from a tool's `run` method to return structured
+ * content blocks as the error result, rather than just a string message.
+ *
+ * When the ToolRunner catches this error, it will use the `content` property
+ * as the tool result with `is_error: true`.
+ *
+ * @example
+ * ```ts
+ * const tool = {
+ *   name: 'my_tool',
+ *   run: async (input) => {
+ *     if (somethingWentWrong) {
+ *       throw new ToolError([
+ *         { type: 'text', text: 'Error details here' },
+ *         { type: 'image', source: { type: 'base64', data: '...', media_type: 'image/png' } },
+ *       ]);
+ *     }
+ *     return 'success';
+ *   },
+ * };
+ * ```
+ */
+export declare class ToolError extends Error {
+    /**
+     * The content to return as the tool result. This will be sent back to the model
+     * with `is_error: true`.
+     */
+    readonly content: string | Array<BetaToolResultContentBlockParam>;
+    constructor(content: string | Array<BetaToolResultContentBlockParam>);
+}
+//# sourceMappingURL=ToolError.d.ts.map

package/lib/tools/ToolError.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolError.d.ts","sourceRoot":"","sources":["../../src/lib/tools/ToolError.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,+BAA+B,EAAE,gCAA6B;AAEvE;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,qBAAa,SAAU,SAAQ,KAAK;IAClC;;;OAGG;IACH,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,KAAK,CAAC,+BAA+B,CAAC,CAAC;gBAEtD,OAAO,EAAE,MAAM,GAAG,KAAK,CAAC,+BAA+B,CAAC;CAcrE"}

package/lib/tools/ToolError.js

@@ -0,0 +1,42 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolError = void 0;
+/**
+ * An error that can be thrown from a tool's `run` method to return structured
+ * content blocks as the error result, rather than just a string message.
+ *
+ * When the ToolRunner catches this error, it will use the `content` property
+ * as the tool result with `is_error: true`.
+ *
+ * @example
+ * ```ts
+ * const tool = {
+ *   name: 'my_tool',
+ *   run: async (input) => {
+ *     if (somethingWentWrong) {
+ *       throw new ToolError([
+ *         { type: 'text', text: 'Error details here' },
+ *         { type: 'image', source: { type: 'base64', data: '...', media_type: 'image/png' } },
+ *       ]);
+ *     }
+ *     return 'success';
+ *   },
+ * };
+ * ```
+ */
+class ToolError extends Error {
+    constructor(content) {
+        const message = typeof content === 'string' ? content : (content
+            .map((block) => {
+            if (block.type === 'text')
+                return block.text;
+            return `[${block.type}]`;
+        })
+            .join(' '));
+        super(message);
+        this.name = 'ToolError';
+        this.content = content;
+    }
+}
+exports.ToolError = ToolError;
+//# sourceMappingURL=ToolError.js.map

package/lib/tools/ToolError.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolError.js","sourceRoot":"","sources":["../../src/lib/tools/ToolError.ts"],"names":[],"mappings":";;;AAEA;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,MAAa,SAAU,SAAQ,KAAK;IAOlC,YAAY,OAAwD;QAClE,MAAM,OAAO,GACX,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CACtC,OAAO;aACJ,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YACb,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM;gBAAE,OAAO,KAAK,CAAC,IAAI,CAAC;YAC7C,OAAO,IAAI,KAAK,CAAC,IAAI,GAAG,CAAC;QAC3B,CAAC,CAAC;aACD,IAAI,CAAC,GAAG,CAAC,CACb,CAAC;QACJ,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,WAAW,CAAC;QACxB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;CACF;AArBD,8BAqBC"}