@lukeashford/aurelius 3.8.0 → 3.9.0

package/dist/index.mjs

@@ -4504,14 +4504,13 @@ function addMessageToTree(tree, message, parentId = null) {
   } else if (!parentId) {
     branchIndex = newRootIds.length;
   }
-  const newNode = {
+  newNodes[message.id] = {
     ...message,
     parentId,
     children: [],
     branchIndex,
     createdAt: message.createdAt ?? Date.now()
   };
-  newNodes[message.id] = newNode;
   if (parentId && newNodes[parentId]) {
     newNodes[parentId] = {
       ...newNodes[parentId],
@@ -4587,8 +4586,7 @@ function switchBranch(tree, nodeId, direction) {
     return tree;
   }
   const newIndex = direction === "next" ? (currentIndex + 1) % siblings.length : (currentIndex - 1 + siblings.length) % siblings.length;
-  const newNodeId = siblings[newIndex];
-  let leafId = newNodeId;
+  let leafId = siblings[newIndex];
   let currentNode = tree.nodes[leafId];
   while (currentNode && currentNode.children.length > 0) {
     leafId = currentNode.children[0];
@@ -4648,6 +4646,7 @@ var ChatInput = React59.forwardRef(
     onStop,
     attachments: controlledAttachments,
     onAttachmentsChange,
+    onAttachmentRemove,
     showAttachmentButton = true,
     acceptedFileTypes,
     notice,
@@ -4734,15 +4733,19 @@ var ChatInput = React59.forwardRef(
     );
     const handleRemoveAttachment = useCallback13(
       (id) => {
+        const attachment = attachments.find((a) => a.id === id);
+        if (attachment && onAttachmentRemove) {
+          onAttachmentRemove(attachment);
+        }
         setAttachments((prev) => {
-          const attachment = prev.find((a) => a.id === id);
-          if (attachment?.previewUrl) {
-            URL.revokeObjectURL(attachment.previewUrl);
+          const attachmentToRemove = prev.find((a) => a.id === id);
+          if (attachmentToRemove?.previewUrl) {
+            URL.revokeObjectURL(attachmentToRemove.previewUrl);
           }
           return prev.filter((a) => a.id !== id);
         });
       },
-      [setAttachments]
+      [attachments, onAttachmentRemove, setAttachments]
     );
     const handleDragEnter = useCallback13((e) => {
       e.preventDefault();
@@ -4774,7 +4777,8 @@ var ChatInput = React59.forwardRef(
     );
     const isCentered = position === "centered";
     const hasAttachments = attachments.length > 0;
-    const canSubmit = value.trim() && !disabled && !isStreaming;
+    const isUploading = attachments.some((a) => a.status === "uploading");
+    const canSubmit = value.trim() && !disabled && !isStreaming && !isUploading;
     return /* @__PURE__ */ React59.createElement(
       "div",
       {
@@ -6688,6 +6692,7 @@ var ChatInterface = React74.forwardRef(
     enableMessageActions = true,
     attachments: propsAttachments,
     onAttachmentsChange,
+    onAttachmentRemove,
     artifactNodes,
     isArtifactsPanelOpen,
     onArtifactsPanelOpenChange,
@@ -6788,19 +6793,13 @@ var ChatInterface = React74.forwardRef(
     const isTreeMode = !!conversationTree;
     const effectiveMessages = useMemo5(() => {
       if (isTreeMode && conversationTree) {
-        const pathNodes = getActivePathMessages(conversationTree);
-        return pathNodes.map((node) => ({
-          id: node.id,
-          variant: node.role,
-          content: node.content,
-          isStreaming: node.isStreaming
-        }));
+        return getActivePathMessages(conversationTree);
       }
-      return messages;
+      return messages || [];
     }, [isTreeMode, conversationTree, messages]);
     const latestUserMessageIndex = useMemo5(() => {
       for (let i = effectiveMessages.length - 1; i >= 0; i--) {
-        if (effectiveMessages[i].variant === "user") {
+        if (effectiveMessages[i].role === "user") {
           return i;
         }
       }
@@ -6857,10 +6856,23 @@ var ChatInterface = React74.forwardRef(
         }
         const actions = enableMessageActions ? {
           showCopy: true,
-          onEdit: msg.variant === "user" && onEditMessage ? (newContent) => onEditMessage(msg.id, newContent) : void 0,
-          onRetry: msg.variant === "assistant" && onRetryMessage ? () => onRetryMessage(msg.id) : void 0
+          onEdit: msg.role === "user" && onEditMessage ? (newContent) => onEditMessage(msg.id, newContent) : void 0,
+          onRetry: msg.role === "assistant" && onRetryMessage ? () => onRetryMessage(msg.id) : void 0
         } : void 0;
-        return { ...msg, branchInfo, actions };
+        const {
+          role,
+          parentId,
+          children,
+          branchIndex,
+          createdAt,
+          ...rest2
+        } = msg;
+        return {
+          ...rest2,
+          variant: role,
+          branchInfo,
+          actions
+        };
       });
     }, [
       effectiveMessages,
@@ -6990,6 +7002,7 @@ var ChatInterface = React74.forwardRef(
           showAttachmentButton,
           attachments: propsAttachments,
           onAttachmentsChange,
+          onAttachmentRemove,
           notice: inputNotice,
           onInputChange,
           initialInputValue,