@lukeashford/aurelius 3.7.0 → 3.8.1

package/dist/index.mjs

@@ -3972,7 +3972,7 @@ var Message = React55.forwardRef(
     const isUser = variant === "user";
     const [copied, setCopied] = useState10(false);
     const [isEditing, setIsEditing] = useState10(false);
-    const [editValue, setEditValue] = useState10(content);
+    const [editValue, setEditValue] = useState10(typeof content === "string" ? content : "");
     const textareaRef = useRef5(null);
     const showBranchNav = branchInfo && branchInfo.total > 1;
     const showActions = actions && !hideActions && !isStreaming;
@@ -3986,6 +3986,9 @@ var Message = React55.forwardRef(
       }
     }, [isEditing]);
     const handleCopy = async () => {
+      if (typeof content !== "string") {
+        return;
+      }
       try {
         await navigator.clipboard.writeText(content);
         setCopied(true);
@@ -4002,16 +4005,20 @@ var Message = React55.forwardRef(
       }
     };
     const handleStartEdit = () => {
-      setEditValue(content);
-      setIsEditing(true);
+      if (typeof content === "string") {
+        setEditValue(content);
+        setIsEditing(true);
+      }
     };
     const handleCancelEdit = () => {
       setIsEditing(false);
-      setEditValue(content);
+      if (typeof content === "string") {
+        setEditValue(content);
+      }
     };
     const handleSubmitEdit = () => {
       const trimmed = editValue.trim();
-      if (trimmed && trimmed !== content) {
+      if (typeof content === "string" && trimmed && trimmed !== content) {
         actions?.onEdit?.(trimmed);
       }
       setIsEditing(false);
@@ -4078,7 +4085,7 @@ var Message = React55.forwardRef(
             variantStyles2[variant]
           )
         },
-        /* @__PURE__ */ React55.createElement(
+        typeof content === "string" ? /* @__PURE__ */ React55.createElement(
           MarkdownContent,
           {
             content,
@@ -4086,7 +4093,7 @@ var Message = React55.forwardRef(
             isStreaming,
             cursorClassName: "ml-0.5"
           }
-        )
+        ) : content
       ),
       showActions && !isEditing && /* @__PURE__ */ React55.createElement("div", { className: cx(
         "flex items-center gap-0.5 mt-1",
@@ -4098,7 +4105,7 @@ var Message = React55.forwardRef(
           label: copied ? "Copied!" : "Copy message"
         },
         copied ? /* @__PURE__ */ React55.createElement(CheckIcon, null) : /* @__PURE__ */ React55.createElement(CopyIcon, null)
-      ), isUser && actions.onEdit && /* @__PURE__ */ React55.createElement(ActionButton, { onClick: handleStartEdit, label: "Edit message" }, /* @__PURE__ */ React55.createElement(PencilIcon, null)), !isUser && actions.onRetry && /* @__PURE__ */ React55.createElement(ActionButton, { onClick: actions.onRetry, label: "Regenerate response" }, /* @__PURE__ */ React55.createElement(RetryIcon, null)), showBranchNav && /* @__PURE__ */ React55.createElement(React55.Fragment, null, /* @__PURE__ */ React55.createElement("div", { className: "w-px h-4 bg-ash/40 mx-1" }), /* @__PURE__ */ React55.createElement("div", { className: "flex items-center gap-0.5 text-silver/70" }, /* @__PURE__ */ React55.createElement(GitBranchIcon, null), /* @__PURE__ */ React55.createElement(
+      ), isUser && actions.onEdit && typeof content === "string" && /* @__PURE__ */ React55.createElement(ActionButton, { onClick: handleStartEdit, label: "Edit message" }, /* @__PURE__ */ React55.createElement(PencilIcon, null)), !isUser && actions.onRetry && /* @__PURE__ */ React55.createElement(ActionButton, { onClick: actions.onRetry, label: "Regenerate response" }, /* @__PURE__ */ React55.createElement(RetryIcon, null)), showBranchNav && /* @__PURE__ */ React55.createElement(React55.Fragment, null, /* @__PURE__ */ React55.createElement("div", { className: "w-px h-4 bg-ash/40 mx-1" }), /* @__PURE__ */ React55.createElement("div", { className: "flex items-center gap-0.5 text-silver/70" }, /* @__PURE__ */ React55.createElement(GitBranchIcon, null), /* @__PURE__ */ React55.createElement(
         "button",
         {
           type: "button",
@@ -4497,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],
@@ -4580,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];
@@ -6781,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;
         }
       }
@@ -6850,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,