@assistant-ui/react 0.14.7 → 0.14.9

package/dist/utils/smooth/SmoothContext.js

@@ -1,35 +1,41 @@
 "use client";
-import { jsx as _jsx } from "react/jsx-runtime";
-import { createContext, forwardRef, useContext, useState, } from "react";
-import { create } from "zustand";
-import { useAui } from "@assistant-ui/store";
 import { createContextStoreHook } from "../../context/react/utils/createContextStoreHook.js";
+import { useAui } from "@assistant-ui/store";
+import { createContext, forwardRef, useContext, useState } from "react";
+import { create } from "zustand";
+import { jsx } from "react/jsx-runtime";
+//#region src/utils/smooth/SmoothContext.tsx
 const SmoothContext = createContext(null);
 const makeSmoothContext = (initialState) => {
-    const useSmoothStatus = create(() => initialState);
-    return { useSmoothStatus };
+	return { useSmoothStatus: create(() => initialState) };
 };
-export const SmoothContextProvider = ({ children }) => {
-    const outer = useSmoothContext({ optional: true });
-    const aui = useAui();
-    const [context] = useState(() => makeSmoothContext(aui.part().getState().status));
-    // do not wrap if there is an outer SmoothContextProvider
-    if (outer)
-        return children;
-    return (_jsx(SmoothContext.Provider, { value: context, children: children }));
+const SmoothContextProvider = ({ children }) => {
+	const outer = useSmoothContext({ optional: true });
+	const aui = useAui();
+	const [context] = useState(() => makeSmoothContext(aui.part().getState().status));
+	if (outer) return children;
+	return /* @__PURE__ */ jsx(SmoothContext.Provider, {
+		value: context,
+		children
+	});
 };
-export const withSmoothContextProvider = (Component) => {
-    const Wrapped = forwardRef((props, ref) => {
-        return (_jsx(SmoothContextProvider, { children: _jsx(Component, { ...props, ref: ref }) }));
-    });
-    Wrapped.displayName = Component.displayName;
-    return Wrapped;
+const withSmoothContextProvider = (Component) => {
+	const Wrapped = forwardRef((props, ref) => {
+		return /* @__PURE__ */ jsx(SmoothContextProvider, { children: /* @__PURE__ */ jsx(Component, {
+			...props,
+			ref
+		}) });
+	});
+	Wrapped.displayName = Component.displayName;
+	return Wrapped;
 };
 function useSmoothContext(options) {
-    const context = useContext(SmoothContext);
-    if (!options?.optional && !context)
-        throw new Error("This component must be used within a SmoothContextProvider.");
-    return context;
+	const context = useContext(SmoothContext);
+	if (!options?.optional && !context) throw new Error("This component must be used within a SmoothContextProvider.");
+	return context;
 }
-export const { useSmoothStatus, useSmoothStatusStore } = createContextStoreHook(useSmoothContext, "useSmoothStatus");
+const { useSmoothStatus, useSmoothStatusStore } = createContextStoreHook(useSmoothContext, "useSmoothStatus");
+//#endregion
+export { SmoothContextProvider, useSmoothStatus, useSmoothStatusStore, withSmoothContextProvider };
+
 //# sourceMappingURL=SmoothContext.js.map

package/dist/utils/smooth/SmoothContext.js.map

@@ -1 +1 @@
-{"version":3,"file":"SmoothContext.js","sourceRoot":"","sources":["../../../src/utils/smooth/SmoothContext.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AAEb,OAAO,EAEL,aAAa,EAEb,UAAU,EAEV,UAAU,EACV,QAAQ,GACT,MAAM,OAAO,CAAC;AAEf,OAAO,EAAE,MAAM,EAAsB,MAAM,SAAS,CAAC;AAKrD,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,sBAAsB,EAAE,4DAAyD;AAQ1F,MAAM,aAAa,GAAG,aAAa,CAA4B,IAAI,CAAC,CAAC;AAErE,MAAM,iBAAiB,GAAG,CACxB,YAA2D,EAC3D,EAAE;IACF,MAAM,eAAe,GAAG,MAAM,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,CAAC;IACnD,OAAO,EAAE,eAAe,EAAE,CAAC;AAC7B,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,qBAAqB,GAA0B,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;IAC3E,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;IAErB,MAAM,CAAC,OAAO,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE,CAC9B,iBAAiB,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAChD,CAAC;IAEF,yDAAyD;IACzD,IAAI,KAAK;QAAE,OAAO,QAAQ,CAAC;IAE3B,OAAO,CACL,KAAC,aAAa,CAAC,QAAQ,IAAC,KAAK,EAAE,OAAO,YAAG,QAAQ,GAA0B,CAC5E,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,yBAAyB,GAAG,CACvC,SAAY,EACT,EAAE;IACL,MAAM,OAAO,GAAG,UAAU,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QACxC,OAAO,CACL,KAAC,qBAAqB,cACpB,KAAC,SAAS,OAAM,KAAa,EAAE,GAAG,EAAE,GAAG,GAAI,GACrB,CACzB,CAAC;IACJ,CAAC,CAAC,CAAC;IACH,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC,WAAW,CAAC;IAC5C,OAAO,OAAc,CAAC;AACxB,CAAC,CAAC;AAQF,SAAS,gBAAgB,CAAC,OAA4C;IACpE,MAAM,OAAO,GAAG,UAAU,CAAC,aAAa,CAAC,CAAC;IAC1C,IAAI,CAAC,OAAO,EAAE,QAAQ,IAAI,CAAC,OAAO;QAChC,MAAM,IAAI,KAAK,CACb,6DAA6D,CAC9D,CAAC;IACJ,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,MAAM,EAAE,eAAe,EAAE,oBAAoB,EAAE,GAAG,sBAAsB,CAC7E,gBAAgB,EAChB,iBAAiB,CAClB,CAAC"}
+{"version":3,"file":"SmoothContext.js","names":[],"sources":["../../../src/utils/smooth/SmoothContext.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  type ComponentType,\n  createContext,\n  type FC,\n  forwardRef,\n  type PropsWithChildren,\n  useContext,\n  useState,\n} from \"react\";\nimport type { ReadonlyStore } from \"../../context/ReadonlyStore\";\nimport { create, type UseBoundStore } from \"zustand\";\nimport type {\n  MessagePartStatus,\n  ToolCallMessagePartStatus,\n} from \"@assistant-ui/core\";\nimport { useAui } from \"@assistant-ui/store\";\nimport { createContextStoreHook } from \"../../context/react/utils/createContextStoreHook\";\n\ntype SmoothContextValue = {\n  useSmoothStatus: UseBoundStore<\n    ReadonlyStore<MessagePartStatus | ToolCallMessagePartStatus>\n  >;\n};\n\nconst SmoothContext = createContext<SmoothContextValue | null>(null);\n\nconst makeSmoothContext = (\n  initialState: MessagePartStatus | ToolCallMessagePartStatus,\n) => {\n  const useSmoothStatus = create(() => initialState);\n  return { useSmoothStatus };\n};\n\nexport const SmoothContextProvider: FC<PropsWithChildren> = ({ children }) => {\n  const outer = useSmoothContext({ optional: true });\n  const aui = useAui();\n\n  const [context] = useState(() =>\n    makeSmoothContext(aui.part().getState().status),\n  );\n\n  // do not wrap if there is an outer SmoothContextProvider\n  if (outer) return children;\n\n  return (\n    <SmoothContext.Provider value={context}>{children}</SmoothContext.Provider>\n  );\n};\n\nexport const withSmoothContextProvider = <C extends ComponentType<any>>(\n  Component: C,\n): C => {\n  const Wrapped = forwardRef((props, ref) => {\n    return (\n      <SmoothContextProvider>\n        <Component {...(props as any)} ref={ref} />\n      </SmoothContextProvider>\n    );\n  });\n  Wrapped.displayName = Component.displayName;\n  return Wrapped as any;\n};\n\nfunction useSmoothContext(options?: {\n  optional?: false | undefined;\n}): SmoothContextValue;\nfunction useSmoothContext(options?: {\n  optional?: boolean | undefined;\n}): SmoothContextValue | null;\nfunction useSmoothContext(options?: { optional?: boolean | undefined }) {\n  const context = useContext(SmoothContext);\n  if (!options?.optional && !context)\n    throw new Error(\n      \"This component must be used within a SmoothContextProvider.\",\n    );\n  return context;\n}\n\nexport const { useSmoothStatus, useSmoothStatusStore } = createContextStoreHook(\n  useSmoothContext,\n  \"useSmoothStatus\",\n);\n"],"mappings":";;;;;;;AA0BA,MAAM,gBAAgB,cAAyC,IAAI;AAEnE,MAAM,qBACJ,iBACG;CAEH,OAAO,EAAE,iBADe,aAAa,YACd,EAAE;AAC3B;AAEA,MAAa,yBAAgD,EAAE,eAAe;CAC5E,MAAM,QAAQ,iBAAiB,EAAE,UAAU,KAAK,CAAC;CACjD,MAAM,MAAM,OAAO;CAEnB,MAAM,CAAC,WAAW,eAChB,kBAAkB,IAAI,KAAK,EAAE,SAAS,EAAE,MAAM,CAChD;CAGA,IAAI,OAAO,OAAO;CAElB,OACE,oBAAC,cAAc,UAAf;EAAwB,OAAO;EAAU;CAAiC,CAAA;AAE9E;AAEA,MAAa,6BACX,cACM;CACN,MAAM,UAAU,YAAY,OAAO,QAAQ;EACzC,OACE,oBAAC,uBAAD,EAAA,UACE,oBAAC,WAAD;GAAW,GAAK;GAAoB;EAAM,CAAA,EACrB,CAAA;CAE3B,CAAC;CACD,QAAQ,cAAc,UAAU;CAChC,OAAO;AACT;AAQA,SAAS,iBAAiB,SAA8C;CACtE,MAAM,UAAU,WAAW,aAAa;CACxC,IAAI,CAAC,SAAS,YAAY,CAAC,SACzB,MAAM,IAAI,MACR,6DACF;CACF,OAAO;AACT;AAEA,MAAa,EAAE,iBAAiB,yBAAyB,uBACvD,kBACA,iBACF"}

package/dist/utils/smooth/useSmooth.d.ts

@@ -1,3 +1,7 @@
-import type { ReasoningMessagePart, TextMessagePart, MessagePartState } from "@assistant-ui/core";
-export declare const useSmooth: (state: MessagePartState & (TextMessagePart | ReasoningMessagePart), smooth?: boolean) => MessagePartState & (TextMessagePart | ReasoningMessagePart);
+import { MessagePartState, ReasoningMessagePart, TextMessagePart } from "@assistant-ui/core";
+
+//#region src/utils/smooth/useSmooth.d.ts
+declare const useSmooth: (state: MessagePartState & (TextMessagePart | ReasoningMessagePart), smooth?: boolean) => MessagePartState & (TextMessagePart | ReasoningMessagePart);
+//#endregion
+export { useSmooth };
 //# sourceMappingURL=useSmooth.d.ts.map

package/dist/utils/smooth/useSmooth.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useSmooth.d.ts","sourceRoot":"","sources":["../../../src/utils/smooth/useSmooth.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAEV,oBAAoB,EACpB,eAAe,EACf,gBAAgB,EACjB,MAAM,oBAAoB,CAAC;AA+D5B,eAAO,MAAM,SAAS,GACpB,OAAO,gBAAgB,GAAG,CAAC,eAAe,GAAG,oBAAoB,CAAC,EAClE,SAAQ,OAAe,KACtB,gBAAgB,GAAG,CAAC,eAAe,GAAG,oBAAoB,CAmG5D,CAAC"}
+{"version":3,"file":"useSmooth.d.ts","names":[],"sources":["../../../src/utils/smooth/useSmooth.ts"],"mappings":";;;cAwEa,SAAA,GACX,KAAA,EAAO,gBAAA,IAAoB,eAAA,GAAkB,oBAAA,GAC7C,MAAA,eACC,gBAAA,IAAoB,eAAA,GAAkB,oBAAA"}

package/dist/utils/smooth/useSmooth.js

@@ -1,137 +1,127 @@
 "use client";
-import { useEffect, useMemo, useRef, useState } from "react";
+import { writableStore } from "../../context/ReadonlyStore.js";
+import { useSmoothStatusStore } from "./SmoothContext.js";
 import { useAui, useAuiState } from "@assistant-ui/store";
+import { useEffect, useMemo, useRef, useState } from "react";
 import { useCallbackRef } from "@radix-ui/react-use-callback-ref";
-import { useSmoothStatusStore } from "./SmoothContext.js";
-import { writableStore } from "../../context/ReadonlyStore.js";
-class TextStreamAnimator {
-    currentText;
-    setText;
-    animationFrameId = null;
-    lastUpdateTime = Date.now();
-    targetText = "";
-    constructor(currentText, setText) {
-        this.currentText = currentText;
-        this.setText = setText;
-    }
-    start() {
-        if (this.animationFrameId !== null)
-            return;
-        this.lastUpdateTime = Date.now();
-        this.animate();
-    }
-    stop() {
-        if (this.animationFrameId !== null) {
-            cancelAnimationFrame(this.animationFrameId);
-            this.animationFrameId = null;
-        }
-    }
-    animate = () => {
-        const currentTime = Date.now();
-        const deltaTime = currentTime - this.lastUpdateTime;
-        let timeToConsume = deltaTime;
-        const remainingChars = this.targetText.length - this.currentText.length;
-        const baseTimePerChar = Math.min(5, 250 / remainingChars);
-        let charsToAdd = 0;
-        while (timeToConsume >= baseTimePerChar && charsToAdd < remainingChars) {
-            charsToAdd++;
-            timeToConsume -= baseTimePerChar;
-        }
-        if (charsToAdd !== remainingChars) {
-            this.animationFrameId = requestAnimationFrame(this.animate);
-        }
-        else {
-            this.animationFrameId = null;
-        }
-        if (charsToAdd === 0)
-            return;
-        this.currentText = this.targetText.slice(0, this.currentText.length + charsToAdd);
-        this.lastUpdateTime = currentTime - timeToConsume;
-        this.setText(this.currentText);
-    };
-}
-const SMOOTH_STATUS = Object.freeze({
-    type: "running",
-});
-export const useSmooth = (state, smooth = false) => {
-    const { text } = state;
-    const [displayedText, setDisplayedText] = useState(state.status.type === "running" ? "" : text);
-    // Render-phase resync on part flip or text discontinuity, so the
-    // first paint after a thread switch never shows the previous
-    // part's text (#4051). `displayedText` is already a prefix of
-    // `text` during normal streaming, so use it as the previous-text
-    // reference instead of carrying separate state — avoids the
-    // double render per streaming token. Read part identity through
-    // `useAuiState` so we actually subscribe to its changes instead
-    // of relying on a render-time proxy reference that may be stable
-    // across thread swaps.
-    const aui = useAui();
-    const part = useAuiState(() => aui.part());
-    const [prevPart, setPrevPart] = useState(part);
-    if (part !== prevPart || !text.startsWith(displayedText)) {
-        setPrevPart(part);
-        setDisplayedText(state.status.type === "running" ? "" : text);
-    }
-    const smoothStatusStore = useSmoothStatusStore({ optional: true });
-    const setText = useCallbackRef((text) => {
-        setDisplayedText(text);
-        if (smoothStatusStore) {
-            const target = displayedText !== text || state.status.type === "running"
-                ? SMOOTH_STATUS
-                : state.status;
-            writableStore(smoothStatusStore).setState(target, true);
-        }
-    });
-    // TODO this is hacky
-    useEffect(() => {
-        if (smoothStatusStore) {
-            const target = smooth && (displayedText !== text || state.status.type === "running")
-                ? SMOOTH_STATUS
-                : state.status;
-            writableStore(smoothStatusStore).setState(target, true);
-        }
-    }, [smoothStatusStore, smooth, text, displayedText, state.status]);
-    const [animatorRef] = useState(new TextStreamAnimator(displayedText, setText));
-    const animatorPartRef = useRef(part);
-    useEffect(() => {
-        if (!smooth) {
-            animatorRef.stop();
-            return;
-        }
-        // Discontinuity: part flipped, or new text breaks continuation
-        // of the animator's current target. Either case requires
-        // resetting the cursor — without the part check, a new part
-        // whose text happens to share a prefix with the previous target
-        // would keep the stale cursor and flicker.
-        const partChanged = animatorPartRef.current !== part;
-        animatorPartRef.current = part;
-        if (partChanged || !text.startsWith(animatorRef.targetText)) {
-            if (state.status.type === "running") {
-                animatorRef.currentText = "";
-                animatorRef.targetText = text;
-                animatorRef.start();
-            }
-            else {
-                animatorRef.currentText = text;
-                animatorRef.targetText = text;
-                animatorRef.stop();
-            }
-            return;
-        }
-        animatorRef.targetText = text;
-        animatorRef.start();
-    }, [animatorRef, smooth, text, state.status.type, part]);
-    useEffect(() => {
-        return () => {
-            animatorRef.stop();
-        };
-    }, [animatorRef]);
-    return useMemo(() => smooth
-        ? {
-            type: "text",
-            text: displayedText,
-            status: text === displayedText ? state.status : SMOOTH_STATUS,
-        }
-        : state, [smooth, displayedText, state, text]);
+//#region src/utils/smooth/useSmooth.ts
+var TextStreamAnimator = class {
+	currentText;
+	setText;
+	animationFrameId = null;
+	lastUpdateTime = Date.now();
+	targetText = "";
+	constructor(currentText, setText) {
+		this.currentText = currentText;
+		this.setText = setText;
+	}
+	start() {
+		if (this.animationFrameId !== null) return;
+		this.lastUpdateTime = Date.now();
+		this.animate();
+	}
+	stop() {
+		if (this.animationFrameId !== null) {
+			cancelAnimationFrame(this.animationFrameId);
+			this.animationFrameId = null;
+		}
+	}
+	animate = () => {
+		const currentTime = Date.now();
+		let timeToConsume = currentTime - this.lastUpdateTime;
+		const remainingChars = this.targetText.length - this.currentText.length;
+		const baseTimePerChar = Math.min(5, 250 / remainingChars);
+		let charsToAdd = 0;
+		while (timeToConsume >= baseTimePerChar && charsToAdd < remainingChars) {
+			charsToAdd++;
+			timeToConsume -= baseTimePerChar;
+		}
+		if (charsToAdd !== remainingChars) this.animationFrameId = requestAnimationFrame(this.animate);
+		else this.animationFrameId = null;
+		if (charsToAdd === 0) return;
+		this.currentText = this.targetText.slice(0, this.currentText.length + charsToAdd);
+		this.lastUpdateTime = currentTime - timeToConsume;
+		this.setText(this.currentText);
+	};
+};
+const SMOOTH_STATUS = Object.freeze({ type: "running" });
+const useSmooth = (state, smooth = false) => {
+	const { text } = state;
+	const [displayedText, setDisplayedText] = useState(state.status.type === "running" ? "" : text);
+	const aui = useAui();
+	const part = useAuiState(() => aui.part());
+	const [prevPart, setPrevPart] = useState(part);
+	if (part !== prevPart || !text.startsWith(displayedText)) {
+		setPrevPart(part);
+		setDisplayedText(state.status.type === "running" ? "" : text);
+	}
+	const smoothStatusStore = useSmoothStatusStore({ optional: true });
+	const setText = useCallbackRef((text) => {
+		setDisplayedText(text);
+		if (smoothStatusStore) {
+			const target = displayedText !== text || state.status.type === "running" ? SMOOTH_STATUS : state.status;
+			writableStore(smoothStatusStore).setState(target, true);
+		}
+	});
+	useEffect(() => {
+		if (smoothStatusStore) {
+			const target = smooth && (displayedText !== text || state.status.type === "running") ? SMOOTH_STATUS : state.status;
+			writableStore(smoothStatusStore).setState(target, true);
+		}
+	}, [
+		smoothStatusStore,
+		smooth,
+		text,
+		displayedText,
+		state.status
+	]);
+	const [animatorRef] = useState(new TextStreamAnimator(displayedText, setText));
+	const animatorPartRef = useRef(part);
+	useEffect(() => {
+		if (!smooth) {
+			animatorRef.stop();
+			return;
+		}
+		const partChanged = animatorPartRef.current !== part;
+		animatorPartRef.current = part;
+		if (partChanged || !text.startsWith(animatorRef.targetText)) {
+			if (state.status.type === "running") {
+				animatorRef.currentText = "";
+				animatorRef.targetText = text;
+				animatorRef.start();
+			} else {
+				animatorRef.currentText = text;
+				animatorRef.targetText = text;
+				animatorRef.stop();
+			}
+			return;
+		}
+		animatorRef.targetText = text;
+		animatorRef.start();
+	}, [
+		animatorRef,
+		smooth,
+		text,
+		state.status.type,
+		part
+	]);
+	useEffect(() => {
+		return () => {
+			animatorRef.stop();
+		};
+	}, [animatorRef]);
+	return useMemo(() => smooth ? {
+		type: "text",
+		text: displayedText,
+		status: text === displayedText ? state.status : SMOOTH_STATUS
+	} : state, [
+		smooth,
+		displayedText,
+		state,
+		text
+	]);
 };
+//#endregion
+export { useSmooth };
+
 //# sourceMappingURL=useSmooth.js.map

package/dist/utils/smooth/useSmooth.js.map

@@ -1 +1 @@
-{"version":3,"file":"useSmooth.js","sourceRoot":"","sources":["../../../src/utils/smooth/useSmooth.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC7D,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAO1D,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAClE,OAAO,EAAE,oBAAoB,EAAE,2BAAwB;AACvD,OAAO,EAAE,aAAa,EAAE,uCAAoC;AAE5D,MAAM,kBAAkB;IAOb;IACC;IAPF,gBAAgB,GAAkB,IAAI,CAAC;IACvC,cAAc,GAAW,IAAI,CAAC,GAAG,EAAE,CAAC;IAErC,UAAU,GAAW,EAAE,CAAC;IAE/B,YACS,WAAmB,EAClB,OAAkC;QADnC,gBAAW,GAAX,WAAW,CAAQ;QAClB,YAAO,GAAP,OAAO,CAA2B;IACzC,CAAC;IAEJ,KAAK;QACH,IAAI,IAAI,CAAC,gBAAgB,KAAK,IAAI;YAAE,OAAO;QAC3C,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACjC,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;IAED,IAAI;QACF,IAAI,IAAI,CAAC,gBAAgB,KAAK,IAAI,EAAE,CAAC;YACnC,oBAAoB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAC5C,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC/B,CAAC;IACH,CAAC;IAEO,OAAO,GAAG,GAAG,EAAE;QACrB,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAG,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC;QACpD,IAAI,aAAa,GAAG,SAAS,CAAC;QAE9B,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;QACxE,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,GAAG,cAAc,CAAC,CAAC;QAE1D,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,OAAO,aAAa,IAAI,eAAe,IAAI,UAAU,GAAG,cAAc,EAAE,CAAC;YACvE,UAAU,EAAE,CAAC;YACb,aAAa,IAAI,eAAe,CAAC;QACnC,CAAC;QAED,IAAI,UAAU,KAAK,cAAc,EAAE,CAAC;YAClC,IAAI,CAAC,gBAAgB,GAAG,qBAAqB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9D,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC/B,CAAC;QACD,IAAI,UAAU,KAAK,CAAC;YAAE,OAAO;QAE7B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CACtC,CAAC,EACD,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,UAAU,CACrC,CAAC;QACF,IAAI,CAAC,cAAc,GAAG,WAAW,GAAG,aAAa,CAAC;QAClD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACjC,CAAC,CAAC;CACH;AAED,MAAM,aAAa,GAAsB,MAAM,CAAC,MAAM,CAAC;IACrD,IAAI,EAAE,SAAS;CAChB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,SAAS,GAAG,CACvB,KAAkE,EAClE,SAAkB,KAAK,EACsC,EAAE;IAC/D,MAAM,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC;IAEvB,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAChD,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAC5C,CAAC;IAEF,iEAAiE;IACjE,6DAA6D;IAC7D,8DAA8D;IAC9D,iEAAiE;IACjE,4DAA4D;IAC5D,gEAAgE;IAChE,gEAAgE;IAChE,iEAAiE;IACjE,uBAAuB;IACvB,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;IACrB,MAAM,IAAI,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;IAC3C,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC/C,IAAI,IAAI,KAAK,QAAQ,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;QACzD,WAAW,CAAC,IAAI,CAAC,CAAC;QAClB,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAChE,CAAC;IAED,MAAM,iBAAiB,GAAG,oBAAoB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IACnE,MAAM,OAAO,GAAG,cAAc,CAAC,CAAC,IAAY,EAAE,EAAE;QAC9C,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACvB,IAAI,iBAAiB,EAAE,CAAC;YACtB,MAAM,MAAM,GACV,aAAa,KAAK,IAAI,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS;gBACvD,CAAC,CAAC,aAAa;gBACf,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;YACnB,aAAa,CAAC,iBAAiB,CAAC,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,qBAAqB;IACrB,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,iBAAiB,EAAE,CAAC;YACtB,MAAM,MAAM,GACV,MAAM,IAAI,CAAC,aAAa,KAAK,IAAI,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS,CAAC;gBACnE,CAAC,CAAC,aAAa;gBACf,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;YACnB,aAAa,CAAC,iBAAiB,CAAC,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC,EAAE,CAAC,iBAAiB,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAEnE,MAAM,CAAC,WAAW,CAAC,GAAG,QAAQ,CAC5B,IAAI,kBAAkB,CAAC,aAAa,EAAE,OAAO,CAAC,CAC/C,CAAC;IAEF,MAAM,eAAe,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;IACrC,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,WAAW,CAAC,IAAI,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QAED,+DAA+D;QAC/D,yDAAyD;QACzD,4DAA4D;QAC5D,gEAAgE;QAChE,2CAA2C;QAC3C,MAAM,WAAW,GAAG,eAAe,CAAC,OAAO,KAAK,IAAI,CAAC;QACrD,eAAe,CAAC,OAAO,GAAG,IAAI,CAAC;QAC/B,IAAI,WAAW,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE,CAAC;YAC5D,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBACpC,WAAW,CAAC,WAAW,GAAG,EAAE,CAAC;gBAC7B,WAAW,CAAC,UAAU,GAAG,IAAI,CAAC;gBAC9B,WAAW,CAAC,KAAK,EAAE,CAAC;YACtB,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,WAAW,GAAG,IAAI,CAAC;gBAC/B,WAAW,CAAC,UAAU,GAAG,IAAI,CAAC;gBAC9B,WAAW,CAAC,IAAI,EAAE,CAAC;YACrB,CAAC;YACD,OAAO;QACT,CAAC;QAED,WAAW,CAAC,UAAU,GAAG,IAAI,CAAC;QAC9B,WAAW,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC,EAAE,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;IAEzD,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,GAAG,EAAE;YACV,WAAW,CAAC,IAAI,EAAE,CAAC;QACrB,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC;IAElB,OAAO,OAAO,CACZ,GAAG,EAAE,CACH,MAAM;QACJ,CAAC,CAAC;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,aAAa;YACnB,MAAM,EAAE,IAAI,KAAK,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa;SAC9D;QACH,CAAC,CAAC,KAAK,EACX,CAAC,MAAM,EAAE,aAAa,EAAE,KAAK,EAAE,IAAI,CAAC,CACrC,CAAC;AACJ,CAAC,CAAC"}
+{"version":3,"file":"useSmooth.js","names":[],"sources":["../../../src/utils/smooth/useSmooth.ts"],"sourcesContent":["\"use client\";\n\nimport { useEffect, useMemo, useRef, useState } from \"react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport type {\n  MessagePartStatus,\n  ReasoningMessagePart,\n  TextMessagePart,\n  MessagePartState,\n} from \"@assistant-ui/core\";\nimport { useCallbackRef } from \"@radix-ui/react-use-callback-ref\";\nimport { useSmoothStatusStore } from \"./SmoothContext\";\nimport { writableStore } from \"../../context/ReadonlyStore\";\n\nclass TextStreamAnimator {\n  private animationFrameId: number | null = null;\n  private lastUpdateTime: number = Date.now();\n\n  public targetText: string = \"\";\n\n  constructor(\n    public currentText: string,\n    private setText: (newText: string) => void,\n  ) {}\n\n  start() {\n    if (this.animationFrameId !== null) return;\n    this.lastUpdateTime = Date.now();\n    this.animate();\n  }\n\n  stop() {\n    if (this.animationFrameId !== null) {\n      cancelAnimationFrame(this.animationFrameId);\n      this.animationFrameId = null;\n    }\n  }\n\n  private animate = () => {\n    const currentTime = Date.now();\n    const deltaTime = currentTime - this.lastUpdateTime;\n    let timeToConsume = deltaTime;\n\n    const remainingChars = this.targetText.length - this.currentText.length;\n    const baseTimePerChar = Math.min(5, 250 / remainingChars);\n\n    let charsToAdd = 0;\n    while (timeToConsume >= baseTimePerChar && charsToAdd < remainingChars) {\n      charsToAdd++;\n      timeToConsume -= baseTimePerChar;\n    }\n\n    if (charsToAdd !== remainingChars) {\n      this.animationFrameId = requestAnimationFrame(this.animate);\n    } else {\n      this.animationFrameId = null;\n    }\n    if (charsToAdd === 0) return;\n\n    this.currentText = this.targetText.slice(\n      0,\n      this.currentText.length + charsToAdd,\n    );\n    this.lastUpdateTime = currentTime - timeToConsume;\n    this.setText(this.currentText);\n  };\n}\n\nconst SMOOTH_STATUS: MessagePartStatus = Object.freeze({\n  type: \"running\",\n});\n\nexport const useSmooth = (\n  state: MessagePartState & (TextMessagePart | ReasoningMessagePart),\n  smooth: boolean = false,\n): MessagePartState & (TextMessagePart | ReasoningMessagePart) => {\n  const { text } = state;\n\n  const [displayedText, setDisplayedText] = useState(\n    state.status.type === \"running\" ? \"\" : text,\n  );\n\n  // Render-phase resync on part flip or text discontinuity, so the\n  // first paint after a thread switch never shows the previous\n  // part's text (#4051). `displayedText` is already a prefix of\n  // `text` during normal streaming, so use it as the previous-text\n  // reference instead of carrying separate state — avoids the\n  // double render per streaming token. Read part identity through\n  // `useAuiState` so we actually subscribe to its changes instead\n  // of relying on a render-time proxy reference that may be stable\n  // across thread swaps.\n  const aui = useAui();\n  const part = useAuiState(() => aui.part());\n  const [prevPart, setPrevPart] = useState(part);\n  if (part !== prevPart || !text.startsWith(displayedText)) {\n    setPrevPart(part);\n    setDisplayedText(state.status.type === \"running\" ? \"\" : text);\n  }\n\n  const smoothStatusStore = useSmoothStatusStore({ optional: true });\n  const setText = useCallbackRef((text: string) => {\n    setDisplayedText(text);\n    if (smoothStatusStore) {\n      const target =\n        displayedText !== text || state.status.type === \"running\"\n          ? SMOOTH_STATUS\n          : state.status;\n      writableStore(smoothStatusStore).setState(target, true);\n    }\n  });\n\n  // TODO this is hacky\n  useEffect(() => {\n    if (smoothStatusStore) {\n      const target =\n        smooth && (displayedText !== text || state.status.type === \"running\")\n          ? SMOOTH_STATUS\n          : state.status;\n      writableStore(smoothStatusStore).setState(target, true);\n    }\n  }, [smoothStatusStore, smooth, text, displayedText, state.status]);\n\n  const [animatorRef] = useState<TextStreamAnimator>(\n    new TextStreamAnimator(displayedText, setText),\n  );\n\n  const animatorPartRef = useRef(part);\n  useEffect(() => {\n    if (!smooth) {\n      animatorRef.stop();\n      return;\n    }\n\n    // Discontinuity: part flipped, or new text breaks continuation\n    // of the animator's current target. Either case requires\n    // resetting the cursor — without the part check, a new part\n    // whose text happens to share a prefix with the previous target\n    // would keep the stale cursor and flicker.\n    const partChanged = animatorPartRef.current !== part;\n    animatorPartRef.current = part;\n    if (partChanged || !text.startsWith(animatorRef.targetText)) {\n      if (state.status.type === \"running\") {\n        animatorRef.currentText = \"\";\n        animatorRef.targetText = text;\n        animatorRef.start();\n      } else {\n        animatorRef.currentText = text;\n        animatorRef.targetText = text;\n        animatorRef.stop();\n      }\n      return;\n    }\n\n    animatorRef.targetText = text;\n    animatorRef.start();\n  }, [animatorRef, smooth, text, state.status.type, part]);\n\n  useEffect(() => {\n    return () => {\n      animatorRef.stop();\n    };\n  }, [animatorRef]);\n\n  return useMemo(\n    () =>\n      smooth\n        ? {\n            type: \"text\",\n            text: displayedText,\n            status: text === displayedText ? state.status : SMOOTH_STATUS,\n          }\n        : state,\n    [smooth, displayedText, state, text],\n  );\n};\n"],"mappings":";;;;;;;AAcA,IAAM,qBAAN,MAAyB;CAOd;CACC;CAPV,mBAA0C;CAC1C,iBAAiC,KAAK,IAAI;CAE1C,aAA4B;CAE5B,YACE,aACA,SACA;EAFO,KAAA,cAAA;EACC,KAAA,UAAA;CACP;CAEH,QAAQ;EACN,IAAI,KAAK,qBAAqB,MAAM;EACpC,KAAK,iBAAiB,KAAK,IAAI;EAC/B,KAAK,QAAQ;CACf;CAEA,OAAO;EACL,IAAI,KAAK,qBAAqB,MAAM;GAClC,qBAAqB,KAAK,gBAAgB;GAC1C,KAAK,mBAAmB;EAC1B;CACF;CAEA,gBAAwB;EACtB,MAAM,cAAc,KAAK,IAAI;EAE7B,IAAI,gBADc,cAAc,KAAK;EAGrC,MAAM,iBAAiB,KAAK,WAAW,SAAS,KAAK,YAAY;EACjE,MAAM,kBAAkB,KAAK,IAAI,GAAG,MAAM,cAAc;EAExD,IAAI,aAAa;EACjB,OAAO,iBAAiB,mBAAmB,aAAa,gBAAgB;GACtE;GACA,iBAAiB;EACnB;EAEA,IAAI,eAAe,gBACjB,KAAK,mBAAmB,sBAAsB,KAAK,OAAO;OAE1D,KAAK,mBAAmB;EAE1B,IAAI,eAAe,GAAG;EAEtB,KAAK,cAAc,KAAK,WAAW,MACjC,GACA,KAAK,YAAY,SAAS,UAC5B;EACA,KAAK,iBAAiB,cAAc;EACpC,KAAK,QAAQ,KAAK,WAAW;CAC/B;AACF;AAEA,MAAM,gBAAmC,OAAO,OAAO,EACrD,MAAM,UACR,CAAC;AAED,MAAa,aACX,OACA,SAAkB,UAC8C;CAChE,MAAM,EAAE,SAAS;CAEjB,MAAM,CAAC,eAAe,oBAAoB,SACxC,MAAM,OAAO,SAAS,YAAY,KAAK,IACzC;CAWA,MAAM,MAAM,OAAO;CACnB,MAAM,OAAO,kBAAkB,IAAI,KAAK,CAAC;CACzC,MAAM,CAAC,UAAU,eAAe,SAAS,IAAI;CAC7C,IAAI,SAAS,YAAY,CAAC,KAAK,WAAW,aAAa,GAAG;EACxD,YAAY,IAAI;EAChB,iBAAiB,MAAM,OAAO,SAAS,YAAY,KAAK,IAAI;CAC9D;CAEA,MAAM,oBAAoB,qBAAqB,EAAE,UAAU,KAAK,CAAC;CACjE,MAAM,UAAU,gBAAgB,SAAiB;EAC/C,iBAAiB,IAAI;EACrB,IAAI,mBAAmB;GACrB,MAAM,SACJ,kBAAkB,QAAQ,MAAM,OAAO,SAAS,YAC5C,gBACA,MAAM;GACZ,cAAc,iBAAiB,EAAE,SAAS,QAAQ,IAAI;EACxD;CACF,CAAC;CAGD,gBAAgB;EACd,IAAI,mBAAmB;GACrB,MAAM,SACJ,WAAW,kBAAkB,QAAQ,MAAM,OAAO,SAAS,aACvD,gBACA,MAAM;GACZ,cAAc,iBAAiB,EAAE,SAAS,QAAQ,IAAI;EACxD;CACF,GAAG;EAAC;EAAmB;EAAQ;EAAM;EAAe,MAAM;CAAM,CAAC;CAEjE,MAAM,CAAC,eAAe,SACpB,IAAI,mBAAmB,eAAe,OAAO,CAC/C;CAEA,MAAM,kBAAkB,OAAO,IAAI;CACnC,gBAAgB;EACd,IAAI,CAAC,QAAQ;GACX,YAAY,KAAK;GACjB;EACF;EAOA,MAAM,cAAc,gBAAgB,YAAY;EAChD,gBAAgB,UAAU;EAC1B,IAAI,eAAe,CAAC,KAAK,WAAW,YAAY,UAAU,GAAG;GAC3D,IAAI,MAAM,OAAO,SAAS,WAAW;IACnC,YAAY,cAAc;IAC1B,YAAY,aAAa;IACzB,YAAY,MAAM;GACpB,OAAO;IACL,YAAY,cAAc;IAC1B,YAAY,aAAa;IACzB,YAAY,KAAK;GACnB;GACA;EACF;EAEA,YAAY,aAAa;EACzB,YAAY,MAAM;CACpB,GAAG;EAAC;EAAa;EAAQ;EAAM,MAAM,OAAO;EAAM;CAAI,CAAC;CAEvD,gBAAgB;EACd,aAAa;GACX,YAAY,KAAK;EACnB;CACF,GAAG,CAAC,WAAW,CAAC;CAEhB,OAAO,cAEH,SACI;EACE,MAAM;EACN,MAAM;EACN,QAAQ,SAAS,gBAAgB,MAAM,SAAS;CAClD,IACA,OACN;EAAC;EAAQ;EAAe;EAAO;CAAI,CACrC;AACF"}

package/dist/utils/useToolArgsFieldStatus.d.ts

@@ -1,12 +1,15 @@
-export declare const useToolArgsFieldStatus: (fieldPath: (string | number)[]) => {
-    readonly type: "running";
+//#region src/utils/useToolArgsFieldStatus.d.ts
+declare const useToolArgsFieldStatus: (fieldPath: (string | number)[]) => {
+  type: string;
 } | {
-    readonly type: "complete";
+  readonly type: "running";
 } | {
-    readonly type: "incomplete";
-    readonly reason: "cancelled" | "length" | "content-filter" | "other" | "error";
-    readonly error?: unknown;
+  readonly type: "complete";
 } | {
-    type: string;
+  readonly type: "incomplete";
+  readonly reason: "cancelled" | "length" | "content-filter" | "other" | "error";
+  readonly error?: unknown;
 };
+//#endregion
+export { useToolArgsFieldStatus };
 //# sourceMappingURL=useToolArgsFieldStatus.d.ts.map

package/dist/utils/useToolArgsFieldStatus.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useToolArgsFieldStatus.d.ts","sourceRoot":"","sources":["../../src/utils/useToolArgsFieldStatus.ts"],"names":[],"mappings":"AAKA,eAAO,MAAM,sBAAsB,GAAI,WAAW,CAAC,MAAM,GAAG,MAAM,CAAC,EAAE;;;;;;;;;;CAYpE,CAAC"}
+{"version":3,"file":"useToolArgsFieldStatus.d.ts","names":[],"sources":["../../src/utils/useToolArgsFieldStatus.ts"],"mappings":";cAKa,sBAAA,GAA0B,SAAA"}

package/dist/utils/useToolArgsFieldStatus.js

@@ -1,14 +1,15 @@
-import { getPartialJsonObjectFieldState } from "assistant-stream/utils";
 import { useAuiState } from "@assistant-ui/store";
+import { getPartialJsonObjectFieldState } from "assistant-stream/utils";
+//#region src/utils/useToolArgsFieldStatus.ts
 const COMPLETE_STATUS = { type: "complete" };
-export const useToolArgsFieldStatus = (fieldPath) => {
-    return useAuiState((s) => {
-        if (s.part.type !== "tool-call")
-            throw new Error("useToolArgsFieldStatus can only be used inside tool-call message parts");
-        const state = getPartialJsonObjectFieldState(s.part.args, fieldPath);
-        if (state === "complete" || s.part.status?.type === "requires-action")
-            return COMPLETE_STATUS;
-        return s.part.status;
-    });
+const useToolArgsFieldStatus = (fieldPath) => {
+	return useAuiState((s) => {
+		if (s.part.type !== "tool-call") throw new Error("useToolArgsFieldStatus can only be used inside tool-call message parts");
+		if (getPartialJsonObjectFieldState(s.part.args, fieldPath) === "complete" || s.part.status?.type === "requires-action") return COMPLETE_STATUS;
+		return s.part.status;
+	});
 };
+//#endregion
+export { useToolArgsFieldStatus };
+
 //# sourceMappingURL=useToolArgsFieldStatus.js.map

package/dist/utils/useToolArgsFieldStatus.js.map

@@ -1 +1 @@
-{"version":3,"file":"useToolArgsFieldStatus.js","sourceRoot":"","sources":["../../src/utils/useToolArgsFieldStatus.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,8BAA8B,EAAE,MAAM,wBAAwB,CAAC;AACxE,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAElD,MAAM,eAAe,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;AAE7C,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,SAA8B,EAAE,EAAE;IACvE,OAAO,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE;QACvB,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW;YAC7B,MAAM,IAAI,KAAK,CACb,wEAAwE,CACzE,CAAC;QAEJ,MAAM,KAAK,GAAG,8BAA8B,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QACrE,IAAI,KAAK,KAAK,UAAU,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,KAAK,iBAAiB;YACnE,OAAO,eAAe,CAAC;QACzB,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;IACvB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC"}
+{"version":3,"file":"useToolArgsFieldStatus.js","names":[],"sources":["../../src/utils/useToolArgsFieldStatus.ts"],"sourcesContent":["import { getPartialJsonObjectFieldState } from \"assistant-stream/utils\";\nimport { useAuiState } from \"@assistant-ui/store\";\n\nconst COMPLETE_STATUS = { type: \"complete\" };\n\nexport const useToolArgsFieldStatus = (fieldPath: (string | number)[]) => {\n  return useAuiState((s) => {\n    if (s.part.type !== \"tool-call\")\n      throw new Error(\n        \"useToolArgsFieldStatus can only be used inside tool-call message parts\",\n      );\n\n    const state = getPartialJsonObjectFieldState(s.part.args, fieldPath);\n    if (state === \"complete\" || s.part.status?.type === \"requires-action\")\n      return COMPLETE_STATUS;\n    return s.part.status;\n  });\n};\n"],"mappings":";;;AAGA,MAAM,kBAAkB,EAAE,MAAM,WAAW;AAE3C,MAAa,0BAA0B,cAAmC;CACxE,OAAO,aAAa,MAAM;EACxB,IAAI,EAAE,KAAK,SAAS,aAClB,MAAM,IAAI,MACR,wEACF;EAGF,IADc,+BAA+B,EAAE,KAAK,MAAM,SAClD,MAAM,cAAc,EAAE,KAAK,QAAQ,SAAS,mBAClD,OAAO;EACT,OAAO,EAAE,KAAK;CAChB,CAAC;AACH"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react",
-  "version": "0.14.7",
+  "version": "0.14.9",
   "description": "Open-source TypeScript/React library for building production-grade AI chat experiences",
   "keywords": [
     "radix-ui",
@@ -48,21 +48,21 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.2.4",
-    "@assistant-ui/store": "^0.2.11",
-    "@assistant-ui/tap": "^0.5.11",
+    "@assistant-ui/core": "^0.2.6",
+    "@assistant-ui/store": "^0.2.12",
+    "@assistant-ui/tap": "^0.5.12",
     "@radix-ui/primitive": "^1.1.3",
     "@radix-ui/react-compose-refs": "^1.1.2",
     "@radix-ui/react-context": "^1.1.3",
     "@radix-ui/react-primitive": "^2.1.4",
     "@radix-ui/react-use-callback-ref": "^1.1.1",
     "@radix-ui/react-use-escape-keydown": "^1.1.1",
-    "assistant-cloud": "^0.1.28",
-    "assistant-stream": "^0.3.15",
+    "assistant-cloud": "^0.1.29",
+    "assistant-stream": "^0.3.17",
     "nanoid": "^5.1.11",
     "radix-ui": "^1.4.3",
     "react-textarea-autosize": "^8.5.9",
-    "safe-content-frame": "^0.0.19",
+    "safe-content-frame": "^0.0.20",
     "zod": "^4.4.3",
     "zustand": "^5.0.13"
   },
@@ -83,14 +83,14 @@
   "devDependencies": {
     "@testing-library/react": "^16.3.2",
     "@types/json-schema": "^7.0.15",
-    "@types/node": "^25.6.0",
-    "@types/react": "^19.2.14",
+    "@types/node": "^25.9.1",
+    "@types/react": "^19.2.15",
     "@types/react-dom": "^19.2.3",
     "jsdom": "^29.1.1",
-    "react": "^19.2.5",
-    "react-dom": "^19.2.5",
-    "vitest": "^4.1.5",
-    "@assistant-ui/x-buildutils": "0.0.8"
+    "react": "^19.2.6",
+    "react-dom": "^19.2.6",
+    "vitest": "^4.1.7",
+    "@assistant-ui/x-buildutils": "0.0.9"
   },
   "publishConfig": {
     "access": "public",

package/src/client/ExternalThread.ts

@@ -220,6 +220,7 @@ const PartResource = resource(
       getState: () => state,
       addToolResult: () => {},
       resumeToolCall: () => {},
+      respondToToolApproval: () => {},
     };
   },
 );

package/src/index.ts

@@ -254,6 +254,7 @@ export * as ThreadListItemPrimitive from "./primitives/threadListItem";
 export * as ThreadListItemMorePrimitive from "./primitives/threadListItemMore";
 export * as SelectionToolbarPrimitive from "./primitives/selectionToolbar";
 
+export { groupPartByType } from "@assistant-ui/core/react";
 export { useMessagePartText } from "./primitives/messagePart/useMessagePartText";
 export { useMessagePartReasoning } from "./primitives/messagePart/useMessagePartReasoning";
 export { useMessagePartSource } from "./primitives/messagePart/useMessagePartSource";
@@ -279,6 +280,9 @@ export type {
   ImageMessagePart,
   FileMessagePart,
   DataMessagePart,
+  GenerativeUIMessagePart,
+  GenerativeUINode,
+  GenerativeUISpec,
   Unstable_AudioMessagePart,
   ToolCallMessagePart,
   ToolModelContentPart,
@@ -321,10 +325,20 @@ export type {
   ReasoningGroupComponent,
   QuoteMessagePartComponent,
   QuoteMessagePartProps,
+  GenerativeUIComponentRegistry,
+  GenerativeUIMessagePartComponent,
+  GenerativeUIMessagePartProps,
+  GenerativeUIRenderProps,
   EnrichedPartState,
   PartState,
 } from "@assistant-ui/core/react";
 
+// Generative UI runtime error + headless renderer (re-exported from core)
+export {
+  GenerativeUIRender,
+  GenerativeUIRenderError,
+} from "@assistant-ui/core/react";
+
 // Thread list item types
 export type { ThreadListItemStatus } from "@assistant-ui/core";
 

package/src/legacy-runtime/cloud/auiV0.ts

@@ -163,7 +163,7 @@ export function auiV0Encode(message: ThreadMessage): AuiV0Message {
           };
 
         default: {
-          const unhandledType: "audio" | "data" = type;
+          const unhandledType: "audio" | "data" | "generative-ui" = type;
           throw new Error(
             `Message part type not supported by aui/v0: ${unhandledType}`,
           );