@assistant-ui/react 0.4.1 → 0.4.3

package/dist/index.js

@@ -907,21 +907,30 @@ var ActionBarPrimitiveRoot = (0, import_react25.forwardRef)(({ hideWhenRunning,
 ActionBarPrimitiveRoot.displayName = "ActionBarPrimitive.Root";
 
 // src/utils/createActionButton.tsx
-var import_primitive = require("@radix-ui/primitive");
-var import_react_primitive2 = require("@radix-ui/react-primitive");
 var import_react26 = require("react");
+var import_react_primitive2 = require("@radix-ui/react-primitive");
+var import_primitive = require("@radix-ui/primitive");
 var import_jsx_runtime5 = require("react/jsx-runtime");
-var createActionButton = (displayName, useActionButton) => {
+var createActionButton = (displayName, useActionButton, forwardProps = []) => {
   const ActionButton = (0, import_react26.forwardRef)((props, forwardedRef) => {
-    const callback = useActionButton(props);
+    const forwardedProps = {};
+    const primitiveProps = {};
+    Object.keys(props).forEach((key) => {
+      if (forwardProps.includes(key)) {
+        forwardedProps[key] = props[key];
+      } else {
+        primitiveProps[key] = props[key];
+      }
+    });
+    const callback = useActionButton(forwardedProps);
     return /* @__PURE__ */ (0, import_jsx_runtime5.jsx)(
       import_react_primitive2.Primitive.button,
       {
         type: "button",
         disabled: !callback,
-        ...props,
+        ...primitiveProps,
         ref: forwardedRef,
-        onClick: (0, import_primitive.composeEventHandlers)(props.onClick, () => {
+        onClick: (0, import_primitive.composeEventHandlers)(primitiveProps.onClick, () => {
           callback?.();
         })
       }
@@ -934,7 +943,8 @@ var createActionButton = (displayName, useActionButton) => {
 // src/primitives/actionBar/ActionBarCopy.tsx
 var ActionBarPrimitiveCopy = createActionButton(
   "ActionBarPrimitive.Copy",
-  useActionBarCopy
+  useActionBarCopy,
+  ["copiedDuration"]
 );
 
 // src/primitives/actionBar/ActionBarReload.tsx
@@ -1288,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)
@@ -1297,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;
@@ -1306,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();
@@ -1708,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 });
   };
@@ -1726,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
@@ -1736,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);
@@ -1750,7 +1762,7 @@ var useThreadViewportAutoScroll = ({
   });
   const autoScrollRef = (0, import_react_compose_refs3.useComposedRefs)(resizeRef, scrollRef, divRef);
   useOnScrollToBottom(() => {
-    scrollToBottom();
+    scrollToBottom("auto");
   });
   return autoScrollRef;
 };
@@ -1951,7 +1963,8 @@ var ThreadPrimitiveScrollToBottom = createActionButton(
 // src/primitives/thread/ThreadSuggestion.tsx
 var ThreadPrimitiveSuggestion = createActionButton(
   "ThreadPrimitive.Suggestion",
-  useThreadSuggestion
+  useThreadSuggestion,
+  ["prompt", "autoSend", "method"]
 );
 
 // src/runtimes/local/useLocalRuntime.tsx
@@ -2344,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({
@@ -2403,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() {