cli: Trim new lines before/after assistant message

Author: jahooma

cli/src/components/blocks/agent-branch-wrapper.tsx

@@ -9,7 +9,7 @@ import React, {
 
 import { AgentBlockGrid } from './agent-block-grid'
 import { AgentBranchItem } from './agent-branch-item'
-import { trimTrailingNewlines, sanitizePreview } from './block-helpers'
+import { trimNewlines, sanitizePreview } from './block-helpers'
 import { ContentWithMarkdown } from './content-with-markdown'
 import { ImplementorGroup } from './implementor-row'
 import { ThinkingBlock } from './thinking-block'
@@ -248,7 +248,7 @@ const AgentBody = memo(
             const isNestedStreamingText =
               p.parentIsStreaming || nestedStatus === 'running'
             const filteredNestedContent = isNestedStreamingText
-              ? trimTrailingNewlines(textBlock.content)
+              ? trimNewlines(textBlock.content)
               : textBlock.content.trim()
             const markdownOptionsForLevel = p.getAgentMarkdownOptions(0)
             const marginTop = textBlock.marginTop ?? 0

cli/src/components/blocks/block-helpers.ts

@@ -1,6 +1,6 @@
 
-export function trimTrailingNewlines(str: string): string {
-  return str.replace(/\n+$/, '')
+export function trimNewlines(str: string): string {
+  return str.replace(/^\n+|\n+$/g, '')
 }
 
 export function sanitizePreview(text: string): string {

cli/src/components/blocks/single-block.tsx

@@ -4,7 +4,7 @@ import React, { memo, type ReactNode } from 'react'
 import { AgentBranchWrapper } from './agent-branch-wrapper'
 import { AgentListBranch } from './agent-list-branch'
 import { AskUserBranch } from './ask-user-branch'
-import { trimTrailingNewlines, isReasoningTextBlock } from './block-helpers'
+import { trimNewlines, isReasoningTextBlock } from './block-helpers'
 import { ContentWithMarkdown } from './content-with-markdown'
 import { ImageBlock } from './image-block'
 import { UserBlockTextWithInlineCopy } from './user-content-copy'
@@ -68,7 +68,7 @@ export const SingleBlock = memo(
         const textBlock = block as TextContentBlock
         const isStreamingText = isLoading || !isComplete
         const filteredContent = isStreamingText
-          ? trimTrailingNewlines(textBlock.content)
+          ? trimNewlines(textBlock.content)
           : textBlock.content.trim()
         const renderKey = `${messageId}-text-${idx}`
         const prevBlock = idx > 0 && blocks ? blocks[idx - 1] : null

cli/src/components/blocks/user-content-copy.tsx

@@ -2,7 +2,7 @@ import { TextAttributes } from '@opentui/core'
 import React, { memo } from 'react'
 
 import { CopyButton } from '../copy-button'
-import { trimTrailingNewlines } from './block-helpers'
+import { trimNewlines } from './block-helpers'
 import { ContentWithMarkdown } from './content-with-markdown'
 
 import type { MarkdownPalette } from '../../utils/markdown-renderer'
@@ -33,7 +33,7 @@ export const UserContentWithCopyButton = memo(
   }: UserContentWithCopyButtonProps) => {
     const isStreamingMessage = isLoading || !isComplete
     const normalizedContent = isStreamingMessage
-      ? trimTrailingNewlines(content)
+      ? trimNewlines(content)
       : content.trim()
 
     const hasContent = normalizedContent.length > 0