@assistant-ui/react 0.4.2 → 0.4.3

package/dist/index.js

@@ -1298,6 +1298,9 @@ var TextStreamAnimator = class {
   };
 };
 var useSmooth = (text, smooth = false) => {
+  const { useMessage } = useMessageContext();
+  const id = useMessage((m) => m.message.id);
+  const idRef = (0, import_react35.useRef)(id);
   const [displayedText, setDisplayedText] = (0, import_react35.useState)(text);
   const [animatorRef] = (0, import_react35.useState)(
     new TextStreamAnimator(text, setDisplayedText)
@@ -1307,7 +1310,8 @@ var useSmooth = (text, smooth = false) => {
       animatorRef.stop();
       return;
     }
-    if (!text.startsWith(animatorRef.targetText)) {
+    if (idRef.current !== id || !text.startsWith(animatorRef.targetText)) {
+      idRef.current = id;
       setDisplayedText(text);
       animatorRef.currentText = text;
       animatorRef.targetText = text;
@@ -1316,7 +1320,7 @@ var useSmooth = (text, smooth = false) => {
     }
     animatorRef.targetText = text;
     animatorRef.start();
-  }, [animatorRef, smooth, text]);
+  }, [animatorRef, id, smooth, text]);
   (0, import_react35.useEffect)(() => {
     return () => {
       animatorRef.stop();
@@ -1718,14 +1722,11 @@ var useThreadViewportAutoScroll = ({
 }) => {
   const divRef = (0, import_react46.useRef)(null);
   const { useViewport } = useThreadContext();
-  const firstRenderRef = (0, import_react46.useRef)(true);
   const lastScrollTop = (0, import_react46.useRef)(0);
   const isScrollingToBottomRef = (0, import_react46.useRef)(false);
-  const scrollToBottom = () => {
+  const scrollToBottom = (behavior) => {
     const div = divRef.current;
     if (!div || !autoScroll) return;
-    const behavior = firstRenderRef.current ? "instant" : "auto";
-    firstRenderRef.current = false;
     isScrollingToBottomRef.current = true;
     div.scrollTo({ top: div.scrollHeight, behavior });
   };
@@ -1736,7 +1737,9 @@ var useThreadViewportAutoScroll = ({
     const newIsAtBottom = div.scrollHeight - div.scrollTop <= div.clientHeight;
     if (!newIsAtBottom && lastScrollTop.current < div.scrollTop) {
     } else {
-      isScrollingToBottomRef.current = newIsAtBottom;
+      if (newIsAtBottom) {
+        isScrollingToBottomRef.current = false;
+      }
       if (newIsAtBottom !== isAtBottom) {
         useViewport.setState({
           isAtBottom: newIsAtBottom
@@ -1746,11 +1749,10 @@ var useThreadViewportAutoScroll = ({
     lastScrollTop.current = div.scrollTop;
   };
   const resizeRef = useOnResizeContent(() => {
-    if (!isScrollingToBottomRef.current && !useViewport.getState().isAtBottom && !firstRenderRef.current) {
-      handleScroll();
-    } else {
-      scrollToBottom();
+    if (isScrollingToBottomRef.current || useViewport.getState().isAtBottom) {
+      scrollToBottom("instant");
     }
+    handleScroll();
   });
   const scrollRef = useManagedRef((el) => {
     el.addEventListener("scroll", handleScroll);
@@ -1760,7 +1762,7 @@ var useThreadViewportAutoScroll = ({
   });
   const autoScrollRef = (0, import_react_compose_refs3.useComposedRefs)(resizeRef, scrollRef, divRef);
   useOnScrollToBottom(() => {
-    scrollToBottom();
+    scrollToBottom("auto");
   });
   return autoScrollRef;
 };
@@ -2355,7 +2357,7 @@ function assistantDecoderStream() {
   let currentToolCall;
   return new TransformStream({
     transform(chunk, controller) {
-      const [code, valueJson] = chunk.split(":");
+      const [code, valueJson] = parseStreamPart(chunk);
       const value = JSON.parse(valueJson);
       if (currentToolCall && code !== "2" /* ToolCallArgsTextDelta */ && code !== "E" /* Error */) {
         controller.enqueue({
@@ -2414,6 +2416,14 @@ function assistantDecoderStream() {
     }
   });
 }
+var parseStreamPart = (part) => {
+  const index = part.indexOf(":");
+  if (index === -1) throw new Error("Invalid stream part");
+  return [
+    part.slice(0, index),
+    part.slice(index + 1)
+  ];
+};
 
 // src/runtimes/edge/streams/chunkByLineStream.ts
 function chunkByLineStream() {