@assistant-ui/core 0.2.4 → 0.2.6

package/dist/react/runtimes/useToolInvocations.js

@@ -1,490 +1,380 @@
-import { useEffect, useRef, useState } from "react";
-import { createAssistantStreamController, ToolResponse, unstable_toolResultStream, } from "assistant-stream";
-import { AssistantMetaTransformStream, } from "assistant-stream/utils";
 import { isJSONValueEqual } from "../../utils/json/is-json-equal.js";
+import { ToolResponse, createAssistantStreamController, unstable_toolResultStream } from "assistant-stream";
+import { AssistantMetaTransformStream } from "assistant-stream/utils";
+import { useEffect, useRef, useState } from "react";
+//#region src/react/runtimes/useToolInvocations.ts
 const isArgsTextComplete = (argsText) => {
-    try {
-        JSON.parse(argsText);
-        return true;
-    }
-    catch {
-        return false;
-    }
+	try {
+		JSON.parse(argsText);
+		return true;
+	} catch {
+		return false;
+	}
 };
 const parseArgsText = (argsText) => {
-    try {
-        return JSON.parse(argsText);
-    }
-    catch {
-        return undefined;
-    }
+	try {
+		return JSON.parse(argsText);
+	} catch {
+		return;
+	}
 };
 const isEquivalentCompleteArgsText = (previous, next) => {
-    const previousValue = parseArgsText(previous);
-    const nextValue = parseArgsText(next);
-    if (previousValue === undefined || nextValue === undefined)
-        return false;
-    return isJSONValueEqual(previousValue, nextValue);
+	const previousValue = parseArgsText(previous);
+	const nextValue = parseArgsText(next);
+	if (previousValue === void 0 || nextValue === void 0) return false;
+	return isJSONValueEqual(previousValue, nextValue);
 };
-export function useToolInvocations({ state, getTools, onResult, setToolStatuses, }) {
-    /**
-     * Single source of truth for per-tool-call lifecycle. Keyed by *logical*
-     * toolCallId (the id the host knows). Restored entries have no controller;
-     * active entries carry their stream id and rewrite/execution bookkeeping.
-     */
-    const entriesRef = useRef(new Map());
-    /**
-     * Reverse alias map populated only when a rewrite assigns a synthetic stream
-     * id to an entry. Identity mappings are implicit via the fallback in
-     * `getLogicalToolCallId`.
-     */
-    const streamToLogicalRef = useRef(new Map());
-    /**
-     * Stream ids whose `result` chunks must be dropped before reaching `onResult`.
-     * Populated when:
-     *   - an argsText rewrite supersedes a stream (the old stream's result, if
-     *     any, is no longer authoritative)
-     *   - `reset()` is called while a pre-resolved tool call has a never-settling
-     *     Promise pending in the executor — the eventual cancellation chunk
-     *     would otherwise be forwarded to a host that has already moved on.
-     */
-    const abandonedStreamIdsRef = useRef(new Set());
-    /**
-     * Stream ids whose `execute` should be short-circuited in the tool wrapper.
-     * Tracked by physical stream id (not logical id) so cleanup is keyed off
-     * the same id the wrapper sees in its context.
-     */
-    const skipExecuteStreamIdsRef = useRef(new Set());
-    const humanInputRef = useRef(new Map());
-    /**
-     * In-flight `execute` invocations keyed by physical stream id. Lives outside
-     * `entriesRef` so `reset()` can drop tool-call state without orphaning the
-     * cleanup the cancellation `onExecutionEnd` still needs.
-     */
-    const executingRef = useRef(new Map());
-    const acRef = useRef(new AbortController());
-    const executingCountRef = useRef(0);
-    const settledResolversRef = useRef([]);
-    const rewriteCounterRef = useRef(0);
-    /**
-     * `true` until the first snapshot has been processed; `reset()` flips it
-     * back to `true`. Snapshots observed while this is `true` are treated as
-     * historical: their tool calls are recorded in `entriesRef` as restored
-     * but no streamCall/execute fires. The next snapshot is processed as live.
-     */
-    const pendingRestoreRef = useRef(true);
-    const getLogicalToolCallId = (streamId) => streamToLogicalRef.current.get(streamId) ?? streamId;
-    const getWrappedTools = () => {
-        const tools = getTools();
-        if (!tools)
-            return undefined;
-        return Object.fromEntries(Object.entries(tools).map(([name, tool]) => {
-            const execute = tool.execute;
-            const streamCall = tool.streamCall;
-            const toModelOutput = tool.toModelOutput;
-            const wrappedTool = {
-                ...tool,
-                ...(execute !== undefined && {
-                    execute: (...[args, context]) => {
-                        if (skipExecuteStreamIdsRef.current.has(context.toolCallId)) {
-                            // Pre-resolved on first live observation: never invoke the
-                            // host's execute fn. Returning a never-settling Promise keeps
-                            // the executor's pending entry alive but enqueues nothing.
-                            // The membership in skipExecuteStreamIdsRef must outlive the
-                            // wrapper call so `reset()`'s seeding loop (which reads this
-                            // Set to identify pre-resolved entries needing cancellation
-                            // suppression) sees the entry. Growth is bounded by the
-                            // number of pre-resolved tool calls observed in the session.
-                            return new Promise(() => { });
-                        }
-                        return execute(args, {
-                            ...context,
-                            toolCallId: getLogicalToolCallId(context.toolCallId),
-                        });
-                    },
-                }),
-                ...(streamCall !== undefined && {
-                    streamCall: (...[reader, context]) => streamCall(reader, {
-                        ...context,
-                        toolCallId: getLogicalToolCallId(context.toolCallId),
-                    }),
-                }),
-                ...(toModelOutput !== undefined && {
-                    toModelOutput: (options) => toModelOutput({
-                        ...options,
-                        toolCallId: getLogicalToolCallId(options.toolCallId),
-                    }),
-                }),
-            };
-            return [name, wrappedTool];
-        }));
-    };
-    const resolveAllSettledResolvers = () => {
-        const resolvers = settledResolversRef.current;
-        settledResolversRef.current = [];
-        // biome-ignore lint/suspicious/useIterableCallbackReturn: forEach callback intentionally has no return
-        resolvers.forEach((resolve) => resolve());
-    };
-    const [controller] = useState(() => {
-        const [stream, controller] = createAssistantStreamController();
-        const transform = unstable_toolResultStream(getWrappedTools, () => acRef.current?.signal ?? new AbortController().signal, (toolCallId, payload) => {
-            const logicalToolCallId = getLogicalToolCallId(toolCallId);
-            return new Promise((resolve, reject) => {
-                const previous = humanInputRef.current.get(logicalToolCallId);
-                if (previous) {
-                    previous.reject(new Error("Human input request was superseded by a new request"));
-                }
-                humanInputRef.current.set(logicalToolCallId, { resolve, reject });
-                setToolStatuses((prev) => ({
-                    ...prev,
-                    [logicalToolCallId]: {
-                        type: "interrupt",
-                        payload: { type: "human", payload },
-                    },
-                }));
-            });
-        }, {
-            onExecutionStart: (streamId) => {
-                if (skipExecuteStreamIdsRef.current.has(streamId))
-                    return;
-                const logicalToolCallId = getLogicalToolCallId(streamId);
-                const abandoned = abandonedStreamIdsRef.current.has(streamId);
-                executingRef.current.set(streamId, {
-                    logicalToolCallId,
-                    abandoned,
-                });
-                executingCountRef.current++;
-                if (!abandoned) {
-                    setToolStatuses((prev) => ({
-                        ...prev,
-                        [logicalToolCallId]: { type: "executing" },
-                    }));
-                }
-            },
-            onExecutionEnd: (streamId) => {
-                const info = executingRef.current.get(streamId);
-                if (!info)
-                    return;
-                executingRef.current.delete(streamId);
-                executingCountRef.current--;
-                if (!info.abandoned) {
-                    setToolStatuses((prev) => {
-                        const next = { ...prev };
-                        delete next[info.logicalToolCallId];
-                        return next;
-                    });
-                }
-                if (executingCountRef.current === 0) {
-                    resolveAllSettledResolvers();
-                }
-            },
-        });
-        stream
-            .pipeThrough(transform)
-            .pipeThrough(new AssistantMetaTransformStream())
-            .pipeTo(new WritableStream({
-            write(chunk) {
-                if (chunk.type !== "result")
-                    return;
-                const streamId = chunk.meta.toolCallId;
-                const logicalToolCallId = getLogicalToolCallId(streamId);
-                const entry = entriesRef.current.get(logicalToolCallId);
-                // Result chunk from a rewrite-superseded stream: drop and clean
-                // up the alias.
-                if (abandonedStreamIdsRef.current.delete(streamId)) {
-                    streamToLogicalRef.current.delete(streamId);
-                    return;
-                }
-                // Pre-resolved tool call whose entry has been cleared by
-                // `reset()`. Both the real result chunk and the post-abort
-                // cancellation chunk can land here in either order; suppress
-                // both via the long-lived `skipExecuteStreamIdsRef` marker.
-                if (!entry && skipExecuteStreamIdsRef.current.has(streamId)) {
-                    return;
-                }
-                // The host already set the result (via the live snapshot's
-                // `setResponse` path). Suppress the executor's redundant emit.
-                if (entry?.hasResult)
-                    return;
-                if (streamId !== logicalToolCallId) {
-                    streamToLogicalRef.current.delete(streamId);
-                }
-                onResult({
-                    type: "add-tool-result",
-                    toolCallId: logicalToolCallId,
-                    toolName: chunk.meta.toolName,
-                    result: chunk.result,
-                    isError: chunk.isError,
-                    ...(chunk.artifact !== undefined && {
-                        artifact: chunk.artifact,
-                    }),
-                    ...(chunk.modelContent !== undefined && {
-                        modelContent: chunk.modelContent,
-                    }),
-                });
-            },
-        }));
-        return controller;
-    });
-    useEffect(() => {
-        const hasExecutableTool = (toolName) => {
-            const tool = getTools()?.[toolName];
-            return tool?.execute !== undefined || tool?.streamCall !== undefined;
-        };
-        const shouldCloseArgsStream = ({ toolName, argsText, hasResult, }) => {
-            if (hasResult)
-                return true;
-            if (!hasExecutableTool(toolName)) {
-                // Non-executable tools can emit parseable JSON mid-stream; wait for
-                // the run to settle before closing.
-                return !state.isRunning && isArgsTextComplete(argsText);
-            }
-            return isArgsTextComplete(argsText);
-        };
-        const startActiveEntry = (toolCallId, toolName, skipExecute) => {
-            const toolCallController = controller.addToolCallPart({
-                toolName,
-                toolCallId,
-            });
-            if (skipExecute) {
-                skipExecuteStreamIdsRef.current.add(toolCallId);
-            }
-            const entry = {
-                toolName,
-                controller: toolCallController,
-                streamId: toolCallId,
-                argsText: "",
-                hasResult: false,
-                argsComplete: false,
-            };
-            entriesRef.current.set(toolCallId, entry);
-            return entry;
-        };
-        const restartArgsStream = (entry, toolCallId) => {
-            if (!entry.controller)
-                return;
-            abandonedStreamIdsRef.current.add(entry.streamId);
-            // The wrapper's execute short-circuit follows the current stream id;
-            // the abandoned id stays in `skipExecuteStreamIdsRef` if it was there,
-            // which is harmless and keeps in-flight chunks consistent.
-            const wasSkipExecute = skipExecuteStreamIdsRef.current.has(entry.streamId);
-            entry.controller.argsText.close();
-            const newStreamId = `${toolCallId}:rewrite:${rewriteCounterRef.current++}`;
-            streamToLogicalRef.current.set(newStreamId, toolCallId);
-            const newController = controller.addToolCallPart({
-                toolName: entry.toolName,
-                toolCallId: newStreamId,
-            });
-            if (wasSkipExecute) {
-                skipExecuteStreamIdsRef.current.add(newStreamId);
-            }
-            if (process.env.NODE_ENV !== "production") {
-                console.warn("started replacement stream tool call", {
-                    toolCallId,
-                    streamToolCallId: newStreamId,
-                });
-            }
-            entry.controller = newController;
-            entry.streamId = newStreamId;
-            entry.argsText = "";
-            entry.argsComplete = false;
-        };
-        const processArgsText = (entry, content) => {
-            if (!entry.controller)
-                return;
-            const hasResult = content.result !== undefined;
-            if (content.argsText !== entry.argsText) {
-                let shouldWriteArgsText = true;
-                if (entry.argsComplete) {
-                    if (isEquivalentCompleteArgsText(entry.argsText, content.argsText)) {
-                        entry.argsText = content.argsText;
-                        shouldWriteArgsText = false;
-                    }
-                    else {
-                        const canRestart = !entry.hasResult && !executingRef.current.has(entry.streamId);
-                        if (process.env.NODE_ENV !== "production") {
-                            console.warn(canRestart
-                                ? "argsText updated after controller was closed, restarting tool args stream:"
-                                : "argsText updated after controller was closed:", { previous: entry.argsText, next: content.argsText });
-                        }
-                        if (!canRestart) {
-                            entry.argsText = content.argsText;
-                            shouldWriteArgsText = false;
-                        }
-                        else {
-                            restartArgsStream(entry, content.toolCallId);
-                        }
-                    }
-                }
-                else if (!content.argsText.startsWith(entry.argsText)) {
-                    // Mid-stream rewrite. If both texts parse to equivalent JSON it's a
-                    // key-reorder snapshot — accept silently. Otherwise restart.
-                    if (isArgsTextComplete(entry.argsText) &&
-                        isArgsTextComplete(content.argsText) &&
-                        isEquivalentCompleteArgsText(entry.argsText, content.argsText)) {
-                        const shouldClose = shouldCloseArgsStream({
-                            toolName: content.toolName,
-                            argsText: content.argsText,
-                            hasResult,
-                        });
-                        if (shouldClose)
-                            entry.controller.argsText.close();
-                        entry.argsText = content.argsText;
-                        entry.argsComplete = shouldClose;
-                        shouldWriteArgsText = false;
-                    }
-                    else {
-                        if (process.env.NODE_ENV !== "production") {
-                            console.warn("argsText rewrote previous snapshot, restarting tool args stream:", {
-                                previous: entry.argsText,
-                                next: content.argsText,
-                                toolCallId: content.toolCallId,
-                            });
-                        }
-                        restartArgsStream(entry, content.toolCallId);
-                    }
-                }
-                if (shouldWriteArgsText) {
-                    const delta = content.argsText.slice(entry.argsText.length);
-                    entry.controller.argsText.append(delta);
-                    const shouldClose = shouldCloseArgsStream({
-                        toolName: content.toolName,
-                        argsText: content.argsText,
-                        hasResult,
-                    });
-                    if (shouldClose)
-                        entry.controller.argsText.close();
-                    entry.argsText = content.argsText;
-                    entry.argsComplete = shouldClose;
-                }
-            }
-            if (!entry.argsComplete) {
-                const shouldClose = shouldCloseArgsStream({
-                    toolName: content.toolName,
-                    argsText: content.argsText,
-                    hasResult,
-                });
-                if (shouldClose) {
-                    entry.controller.argsText.close();
-                    entry.argsText = content.argsText;
-                    entry.argsComplete = true;
-                }
-            }
-        };
-        const processMessages = (messages) => {
-            const isRestore = pendingRestoreRef.current;
-            messages.forEach((message) => {
-                message.content.forEach((content) => {
-                    if (content.type !== "tool-call")
-                        return;
-                    const existing = entriesRef.current.get(content.toolCallId);
-                    if (isRestore) {
-                        // Don't overwrite an already-active entry (e.g. live tool-call
-                        // observed before this restore snapshot landed). Restore can only
-                        // seed entries the runtime has never seen.
-                        if (!existing?.controller) {
-                            entriesRef.current.set(content.toolCallId, {
-                                toolName: content.toolName,
-                                argsText: content.argsText,
-                                hasResult: content.result !== undefined,
-                            });
-                        }
-                        if (content.messages)
-                            processMessages(content.messages);
-                        return;
-                    }
-                    // Live snapshot.
-                    let entry = existing;
-                    if (entry && !entry.controller) {
-                        // Restored entry observed in a live snapshot. Promote if its
-                        // signature has changed; otherwise treat as still-historical.
-                        const signatureChanged = content.argsText !== entry.argsText ||
-                            (content.result !== undefined) !== entry.hasResult;
-                        if (!signatureChanged) {
-                            if (content.messages)
-                                processMessages(content.messages);
-                            return;
-                        }
-                        entriesRef.current.delete(content.toolCallId);
-                        entry = undefined;
-                    }
-                    if (!entry) {
-                        entry = startActiveEntry(content.toolCallId, content.toolName, content.result !== undefined);
-                    }
-                    processArgsText(entry, content);
-                    if (content.result !== undefined && !entry.hasResult) {
-                        // `entry` is in active phase from this point — either it was
-                        // just created by `startActiveEntry` above, or it pre-existed
-                        // and `processArgsText` preserved (or replaced via rewrite) its
-                        // controller. Narrow once instead of asserting at every use.
-                        const { controller: activeController } = entry;
-                        if (!activeController)
-                            return;
-                        entry.hasResult = true;
-                        entry.argsComplete = true;
-                        activeController.setResponse(new ToolResponse({
-                            result: content.result,
-                            artifact: content.artifact,
-                            isError: content.isError,
-                            ...(content.modelContent !== undefined
-                                ? { modelContent: content.modelContent }
-                                : {}),
-                        }));
-                        activeController.close();
-                    }
-                    if (content.messages)
-                        processMessages(content.messages);
-                });
-            });
-        };
-        processMessages(state.messages);
-        pendingRestoreRef.current = false;
-    }, [state, controller, getTools]);
-    const abort = () => {
-        humanInputRef.current.forEach(({ reject }) => {
-            reject(new Error("Tool execution aborted"));
-        });
-        humanInputRef.current.clear();
-        acRef.current.abort();
-        acRef.current = new AbortController();
-        if (executingCountRef.current === 0) {
-            return Promise.resolve();
-        }
-        return new Promise((resolve) => {
-            settledResolversRef.current.push(resolve);
-        });
-    };
-    return {
-        reset: () => {
-            pendingRestoreRef.current = true;
-            entriesRef.current.clear();
-            // `skipExecuteStreamIdsRef` is not cleared: it has to outlive `reset()`
-            // so (a) any wrapper call still inbound through the stream pipeline
-            // continues to short-circuit `execute`, and (b) the consumer can
-            // recognize and drop any post-abort cancellation `result` chunks for
-            // pre-resolved streams whose entries have been cleared. Membership
-            // grows by one per pre-resolved tool call observed in the session.
-            void abort().finally(() => {
-                executingRef.current.clear();
-                streamToLogicalRef.current.clear();
-                rewriteCounterRef.current = 0;
-            });
-        },
-        abort,
-        resume: (toolCallId, payload) => {
-            const handlers = humanInputRef.current.get(toolCallId);
-            if (handlers) {
-                humanInputRef.current.delete(toolCallId);
-                setToolStatuses((prev) => ({
-                    ...prev,
-                    [toolCallId]: { type: "executing" },
-                }));
-                handlers.resolve(payload);
-            }
-            else {
-                throw new Error(`Tool call ${toolCallId} is not waiting for human input`);
-            }
-        },
-    };
+function useToolInvocations({ state, getTools, onResult, setToolStatuses }) {
+	/**
+	* Single source of truth for per-tool-call lifecycle. Keyed by *logical*
+	* toolCallId (the id the host knows). Restored entries have no controller;
+	* active entries carry their stream id and rewrite/execution bookkeeping.
+	*/
+	const entriesRef = useRef(/* @__PURE__ */ new Map());
+	/**
+	* Reverse alias map populated only when a rewrite assigns a synthetic stream
+	* id to an entry. Identity mappings are implicit via the fallback in
+	* `getLogicalToolCallId`.
+	*/
+	const streamToLogicalRef = useRef(/* @__PURE__ */ new Map());
+	/**
+	* Stream ids whose `result` chunks must be dropped before reaching `onResult`.
+	* Populated when:
+	*   - an argsText rewrite supersedes a stream (the old stream's result, if
+	*     any, is no longer authoritative)
+	*   - `reset()` is called while a pre-resolved tool call has a never-settling
+	*     Promise pending in the executor — the eventual cancellation chunk
+	*     would otherwise be forwarded to a host that has already moved on.
+	*/
+	const abandonedStreamIdsRef = useRef(/* @__PURE__ */ new Set());
+	/**
+	* Stream ids whose `execute` should be short-circuited in the tool wrapper.
+	* Tracked by physical stream id (not logical id) so cleanup is keyed off
+	* the same id the wrapper sees in its context.
+	*/
+	const skipExecuteStreamIdsRef = useRef(/* @__PURE__ */ new Set());
+	const humanInputRef = useRef(/* @__PURE__ */ new Map());
+	/**
+	* In-flight `execute` invocations keyed by physical stream id. Lives outside
+	* `entriesRef` so `reset()` can drop tool-call state without orphaning the
+	* cleanup the cancellation `onExecutionEnd` still needs.
+	*/
+	const executingRef = useRef(/* @__PURE__ */ new Map());
+	const acRef = useRef(new AbortController());
+	const executingCountRef = useRef(0);
+	const settledResolversRef = useRef([]);
+	const rewriteCounterRef = useRef(0);
+	/**
+	* `true` until the first snapshot has been processed; `reset()` flips it
+	* back to `true`. Snapshots observed while this is `true` are treated as
+	* historical: their tool calls are recorded in `entriesRef` as restored
+	* but no streamCall/execute fires. The next snapshot is processed as live.
+	*/
+	const pendingRestoreRef = useRef(true);
+	const getLogicalToolCallId = (streamId) => streamToLogicalRef.current.get(streamId) ?? streamId;
+	const getWrappedTools = () => {
+		const tools = getTools();
+		if (!tools) return void 0;
+		return Object.fromEntries(Object.entries(tools).map(([name, tool]) => {
+			const execute = tool.execute;
+			const streamCall = tool.streamCall;
+			const toModelOutput = tool.toModelOutput;
+			return [name, {
+				...tool,
+				...execute !== void 0 && { execute: (...[args, context]) => {
+					if (skipExecuteStreamIdsRef.current.has(context.toolCallId)) return new Promise(() => {});
+					return execute(args, {
+						...context,
+						toolCallId: getLogicalToolCallId(context.toolCallId)
+					});
+				} },
+				...streamCall !== void 0 && { streamCall: (...[reader, context]) => streamCall(reader, {
+					...context,
+					toolCallId: getLogicalToolCallId(context.toolCallId)
+				}) },
+				...toModelOutput !== void 0 && { toModelOutput: (options) => toModelOutput({
+					...options,
+					toolCallId: getLogicalToolCallId(options.toolCallId)
+				}) }
+			}];
+		}));
+	};
+	const resolveAllSettledResolvers = () => {
+		const resolvers = settledResolversRef.current;
+		settledResolversRef.current = [];
+		resolvers.forEach((resolve) => resolve());
+	};
+	const [controller] = useState(() => {
+		const [stream, controller] = createAssistantStreamController();
+		const transform = unstable_toolResultStream(getWrappedTools, () => acRef.current?.signal ?? new AbortController().signal, (toolCallId, payload) => {
+			const logicalToolCallId = getLogicalToolCallId(toolCallId);
+			return new Promise((resolve, reject) => {
+				const previous = humanInputRef.current.get(logicalToolCallId);
+				if (previous) previous.reject(/* @__PURE__ */ new Error("Human input request was superseded by a new request"));
+				humanInputRef.current.set(logicalToolCallId, {
+					resolve,
+					reject
+				});
+				setToolStatuses((prev) => ({
+					...prev,
+					[logicalToolCallId]: {
+						type: "interrupt",
+						payload: {
+							type: "human",
+							payload
+						}
+					}
+				}));
+			});
+		}, {
+			onExecutionStart: (streamId) => {
+				if (skipExecuteStreamIdsRef.current.has(streamId)) return;
+				const logicalToolCallId = getLogicalToolCallId(streamId);
+				const abandoned = abandonedStreamIdsRef.current.has(streamId);
+				executingRef.current.set(streamId, {
+					logicalToolCallId,
+					abandoned
+				});
+				executingCountRef.current++;
+				if (!abandoned) setToolStatuses((prev) => ({
+					...prev,
+					[logicalToolCallId]: { type: "executing" }
+				}));
+			},
+			onExecutionEnd: (streamId) => {
+				const info = executingRef.current.get(streamId);
+				if (!info) return;
+				executingRef.current.delete(streamId);
+				executingCountRef.current--;
+				if (!info.abandoned) setToolStatuses((prev) => {
+					const next = { ...prev };
+					delete next[info.logicalToolCallId];
+					return next;
+				});
+				if (executingCountRef.current === 0) resolveAllSettledResolvers();
+			}
+		});
+		stream.pipeThrough(transform).pipeThrough(new AssistantMetaTransformStream()).pipeTo(new WritableStream({ write(chunk) {
+			if (chunk.type !== "result") return;
+			const streamId = chunk.meta.toolCallId;
+			const logicalToolCallId = getLogicalToolCallId(streamId);
+			const entry = entriesRef.current.get(logicalToolCallId);
+			if (abandonedStreamIdsRef.current.delete(streamId)) {
+				streamToLogicalRef.current.delete(streamId);
+				return;
+			}
+			if (!entry && skipExecuteStreamIdsRef.current.has(streamId)) return;
+			if (entry?.hasResult) return;
+			if (streamId !== logicalToolCallId) streamToLogicalRef.current.delete(streamId);
+			onResult({
+				type: "add-tool-result",
+				toolCallId: logicalToolCallId,
+				toolName: chunk.meta.toolName,
+				result: chunk.result,
+				isError: chunk.isError,
+				...chunk.artifact !== void 0 && { artifact: chunk.artifact },
+				...chunk.modelContent !== void 0 && { modelContent: chunk.modelContent }
+			});
+		} }));
+		return controller;
+	});
+	useEffect(() => {
+		const hasExecutableTool = (toolName) => {
+			const tool = getTools()?.[toolName];
+			return tool?.execute !== void 0 || tool?.streamCall !== void 0;
+		};
+		const shouldCloseArgsStream = ({ toolName, argsText, hasResult }) => {
+			if (hasResult) return true;
+			if (!hasExecutableTool(toolName)) return !state.isRunning && isArgsTextComplete(argsText);
+			return isArgsTextComplete(argsText);
+		};
+		const startActiveEntry = (toolCallId, toolName, skipExecute) => {
+			const toolCallController = controller.addToolCallPart({
+				toolName,
+				toolCallId
+			});
+			if (skipExecute) skipExecuteStreamIdsRef.current.add(toolCallId);
+			const entry = {
+				toolName,
+				controller: toolCallController,
+				streamId: toolCallId,
+				argsText: "",
+				hasResult: false,
+				argsComplete: false
+			};
+			entriesRef.current.set(toolCallId, entry);
+			return entry;
+		};
+		const restartArgsStream = (entry, toolCallId) => {
+			if (!entry.controller) return;
+			abandonedStreamIdsRef.current.add(entry.streamId);
+			const wasSkipExecute = skipExecuteStreamIdsRef.current.has(entry.streamId);
+			entry.controller.argsText.close();
+			const newStreamId = `${toolCallId}:rewrite:${rewriteCounterRef.current++}`;
+			streamToLogicalRef.current.set(newStreamId, toolCallId);
+			const newController = controller.addToolCallPart({
+				toolName: entry.toolName,
+				toolCallId: newStreamId
+			});
+			if (wasSkipExecute) skipExecuteStreamIdsRef.current.add(newStreamId);
+			if (process.env.NODE_ENV !== "production") console.warn("started replacement stream tool call", {
+				toolCallId,
+				streamToolCallId: newStreamId
+			});
+			entry.controller = newController;
+			entry.streamId = newStreamId;
+			entry.argsText = "";
+			entry.argsComplete = false;
+		};
+		const processArgsText = (entry, content) => {
+			if (!entry.controller) return;
+			const hasResult = content.result !== void 0;
+			if (content.argsText !== entry.argsText) {
+				let shouldWriteArgsText = true;
+				if (entry.argsComplete) if (isEquivalentCompleteArgsText(entry.argsText, content.argsText)) {
+					entry.argsText = content.argsText;
+					shouldWriteArgsText = false;
+				} else {
+					const canRestart = !entry.hasResult && !executingRef.current.has(entry.streamId);
+					if (process.env.NODE_ENV !== "production") console.warn(canRestart ? "argsText updated after controller was closed, restarting tool args stream:" : "argsText updated after controller was closed:", {
+						previous: entry.argsText,
+						next: content.argsText
+					});
+					if (!canRestart) {
+						entry.argsText = content.argsText;
+						shouldWriteArgsText = false;
+					} else restartArgsStream(entry, content.toolCallId);
+				}
+				else if (!content.argsText.startsWith(entry.argsText)) if (isArgsTextComplete(entry.argsText) && isArgsTextComplete(content.argsText) && isEquivalentCompleteArgsText(entry.argsText, content.argsText)) {
+					const shouldClose = shouldCloseArgsStream({
+						toolName: content.toolName,
+						argsText: content.argsText,
+						hasResult
+					});
+					if (shouldClose) entry.controller.argsText.close();
+					entry.argsText = content.argsText;
+					entry.argsComplete = shouldClose;
+					shouldWriteArgsText = false;
+				} else {
+					if (process.env.NODE_ENV !== "production") console.warn("argsText rewrote previous snapshot, restarting tool args stream:", {
+						previous: entry.argsText,
+						next: content.argsText,
+						toolCallId: content.toolCallId
+					});
+					restartArgsStream(entry, content.toolCallId);
+				}
+				if (shouldWriteArgsText) {
+					const delta = content.argsText.slice(entry.argsText.length);
+					entry.controller.argsText.append(delta);
+					const shouldClose = shouldCloseArgsStream({
+						toolName: content.toolName,
+						argsText: content.argsText,
+						hasResult
+					});
+					if (shouldClose) entry.controller.argsText.close();
+					entry.argsText = content.argsText;
+					entry.argsComplete = shouldClose;
+				}
+			}
+			if (!entry.argsComplete) {
+				if (shouldCloseArgsStream({
+					toolName: content.toolName,
+					argsText: content.argsText,
+					hasResult
+				})) {
+					entry.controller.argsText.close();
+					entry.argsText = content.argsText;
+					entry.argsComplete = true;
+				}
+			}
+		};
+		const processMessages = (messages) => {
+			const isRestore = pendingRestoreRef.current;
+			messages.forEach((message) => {
+				message.content.forEach((content) => {
+					if (content.type !== "tool-call") return;
+					const existing = entriesRef.current.get(content.toolCallId);
+					if (isRestore) {
+						if (!existing?.controller) entriesRef.current.set(content.toolCallId, {
+							toolName: content.toolName,
+							argsText: content.argsText,
+							hasResult: content.result !== void 0
+						});
+						if (content.messages) processMessages(content.messages);
+						return;
+					}
+					let entry = existing;
+					if (entry && !entry.controller) {
+						if (!(content.argsText !== entry.argsText || content.result !== void 0 !== entry.hasResult)) {
+							if (content.messages) processMessages(content.messages);
+							return;
+						}
+						entriesRef.current.delete(content.toolCallId);
+						entry = void 0;
+					}
+					if (!entry) entry = startActiveEntry(content.toolCallId, content.toolName, content.result !== void 0);
+					processArgsText(entry, content);
+					if (content.result !== void 0 && !entry.hasResult) {
+						const { controller: activeController } = entry;
+						if (!activeController) return;
+						entry.hasResult = true;
+						entry.argsComplete = true;
+						activeController.setResponse(new ToolResponse({
+							result: content.result,
+							artifact: content.artifact,
+							isError: content.isError,
+							...content.modelContent !== void 0 ? { modelContent: content.modelContent } : {}
+						}));
+						activeController.close();
+					}
+					if (content.messages) processMessages(content.messages);
+				});
+			});
+		};
+		processMessages(state.messages);
+		pendingRestoreRef.current = false;
+	}, [
+		state,
+		controller,
+		getTools
+	]);
+	const abort = () => {
+		humanInputRef.current.forEach(({ reject }) => {
+			reject(/* @__PURE__ */ new Error("Tool execution aborted"));
+		});
+		humanInputRef.current.clear();
+		acRef.current.abort();
+		acRef.current = new AbortController();
+		if (executingCountRef.current === 0) return Promise.resolve();
+		return new Promise((resolve) => {
+			settledResolversRef.current.push(resolve);
+		});
+	};
+	return {
+		reset: () => {
+			pendingRestoreRef.current = true;
+			entriesRef.current.clear();
+			abort().finally(() => {
+				executingRef.current.clear();
+				streamToLogicalRef.current.clear();
+				rewriteCounterRef.current = 0;
+			});
+		},
+		abort,
+		resume: (toolCallId, payload) => {
+			const handlers = humanInputRef.current.get(toolCallId);
+			if (handlers) {
+				humanInputRef.current.delete(toolCallId);
+				setToolStatuses((prev) => ({
+					...prev,
+					[toolCallId]: { type: "executing" }
+				}));
+				handlers.resolve(payload);
+			} else throw new Error(`Tool call ${toolCallId} is not waiting for human input`);
+		}
+	};
 }
+//#endregion
+export { useToolInvocations };
+
 //# sourceMappingURL=useToolInvocations.js.map