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 != ""
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 != "" {