mlow/lmcli
Private
Public Access
1
0
Fork 0
Code Activity

Fix thinking formatting

Browse Source
This commit is contained in:
Matt Low
2025-02-03 06:13:47 +00:00
parent e564a93c37
commit 6481841e66
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
pkg/tui/views/chat/view.go
View File

@@ -123,18 +123,24 @@ func (m *Model) renderMessage(i int) string {
hasReasoning := msg.ReasoningContent != "" hasReasoning := msg.ReasoningContent != ""
if hasReasoning { 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 { if m.showDetails {
//_ = m.App.Ctx.Chroma.Highlight(sb, msg.ReasoningContent) //_ = m.App.Ctx.Chroma.Highlight(sb, msg.ReasoningContent)
sb.WriteString(faintStyle.Render(msg.ReasoningContent)) reasoningContent = msg.ReasoningContent
} else { } else {
sb.WriteString(faintStyle.Render("...")) reasoningContent = "..."
} }
if m.state == pendingResponse && isLast && isAssistant && msg.Content == "" { if m.state == pendingResponse && isLast && isAssistant && msg.Content == "" {
// Show the assistant's cursor // 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 != "" { if msg.Content != "" {