From ed0d8784d595d6b84644b50973bd476eeb429d17 Mon Sep 17 00:00:00 2001 From: Matt Low Date: Mon, 27 Jan 2025 06:36:02 +0000 Subject: [PATCH] Don't syntax highlight model thinking process It gets rather long and the syntax highlighting gets rather expensive --- pkg/tui/views/chat/view.go | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/pkg/tui/views/chat/view.go b/pkg/tui/views/chat/view.go index 0b686cb..333bd75 100644 --- a/pkg/tui/views/chat/view.go +++ b/pkg/tui/views/chat/view.go @@ -123,20 +123,18 @@ func (m *Model) renderMessage(i int) string { hasReasoning := msg.ReasoningContent != "" if hasReasoning { - reasoning := strings.Builder{} - reasoning.WriteString("\n") + _ = m.App.Ctx.Chroma.Highlight(sb, "\n") if m.showDetails { //_ = m.App.Ctx.Chroma.Highlight(sb, msg.ReasoningContent) - reasoning.WriteString(msg.ReasoningContent) + sb.WriteString(faintStyle.Render(msg.ReasoningContent)) } else { - reasoning.WriteString("...") + sb.WriteString(faintStyle.Render("...")) } if m.state == pendingResponse && isLast && isAssistant && msg.Content == "" { // Show the assistant's cursor - reasoning.WriteString(m.replyCursor.View()) + sb.WriteString(m.replyCursor.View()) } - reasoning.WriteString("\n") - _ = m.App.Ctx.Chroma.Highlight(sb, reasoning.String()) + _ = m.App.Ctx.Chroma.Highlight(sb, "\n") } if msg.Content != "" {