fix(cli): clean up code redundancies and improve markdown rendering

- Simplify marginTop ternary logic
- Don't trim markdown content to preserve paragraph spacing
- Only strip trailing newlines from markdown output, not all whitespace

Author: brandonkachen

cli/src/components/message-block.tsx

@@ -437,22 +437,20 @@ export const MessageBlock = ({
           {blocks.map((block, idx) => {
             if (block.type === 'text') {
               const isStreamingText = isLoading || !isComplete
+              const hasMarkdownContent = hasMarkdown(block.content)
               const rawContent = isStreamingText
                 ? trimTrailingNewlines(block.content)
-                : block.content.trim()
+                : hasMarkdownContent
+                  ? block.content
+                  : block.content.trim()
               const renderKey = `${messageId}-text-${idx}`
-              const renderedContent = hasMarkdown(rawContent)
+              const renderedContent = hasMarkdownContent
                 ? isStreamingText
                   ? renderStreamingMarkdown(rawContent, markdownOptions)
                   : renderMarkdown(rawContent, markdownOptions)
                 : rawContent
               const prevBlock = idx > 0 ? blocks[idx - 1] : null
-              const marginTop =
-                prevBlock && prevBlock.type === 'text'
-                  ? 1
-                  : prevBlock && (prevBlock.type === 'tool' || prevBlock.type === 'agent')
-                    ? 0
-                    : 0
+              const marginTop = prevBlock && prevBlock.type === 'text' ? 1 : 0
               const blockTextColor = block.color ?? textColor
             return (
               <text key={renderKey} style={{ fg: blockTextColor, marginTop }}>

cli/src/utils/markdown-renderer.tsx

@@ -152,7 +152,7 @@ export function renderMarkdown(
     void palette // Keep signature compatibility for future color styling
 
     const ast = processor.parse(markdown)
-    const text = nodeToPlainText(ast).replace(/\s+$/g, '')
+    const text = nodeToPlainText(ast).replace(/\n+$/g, '')
     return text
   } catch (error) {
     logger.error(error, 'Failed to parse markdown')