From 6481841e6601798769fa0774005be7bd2e0b9480 Mon Sep 17 00:00:00 2001
From: Matt Low <matt@mlow.ca>
Date: Mon, 3 Feb 2025 06:13:47 +0000
Subject: [PATCH] Fix thinking formatting

---
 pkg/tui/views/chat/view.go | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/pkg/tui/views/chat/view.go b/pkg/tui/views/chat/view.go
index 333bd75..b52c650 100644
--- a/pkg/tui/views/chat/view.go
+++ b/pkg/tui/views/chat/view.go
@@ -123,18 +123,24 @@ func (m *Model) renderMessage(i int) string {
 	hasReasoning := msg.ReasoningContent != ""
 
 	if hasReasoning {
-		_ = m.App.Ctx.Chroma.Highlight(sb, "<thinking>\n")
+		_ = m.App.Ctx.Chroma.Highlight(sb, "<thinking>")
+		sb.WriteString("\n")
+		var reasoningContent string
 		if m.showDetails {
 			//_ = m.App.Ctx.Chroma.Highlight(sb, msg.ReasoningContent)
-			sb.WriteString(faintStyle.Render(msg.ReasoningContent))
+			reasoningContent = msg.ReasoningContent
 		} else {
-			sb.WriteString(faintStyle.Render("..."))
+			reasoningContent = "..."
 		}
 		if m.state == pendingResponse && isLast && isAssistant && msg.Content == "" {
 			// Show the assistant's cursor
-			sb.WriteString(m.replyCursor.View())
+			reasoningContent = reasoningContent + m.replyCursor.View()
 		}
-		_ = m.App.Ctx.Chroma.Highlight(sb, "\n</thinking>")
+		sb.WriteString(faintStyle.Render(reasoningContent))
+		if reasoningContent[len(reasoningContent)-1] != '\n' {
+			sb.WriteString("\n")
+		}
+		_ = m.App.Ctx.Chroma.Highlight(sb, "</thinking>")
 	}
 
 	if msg.Content != "" {