@lukeashford/aurelius 3.8.0 → 3.8.1

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];
@@ -6788,19 +6786,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 +6849,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,