diff options
| -rw-r--r-- | packages/tui/internal/components/chat/message.go | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/packages/tui/internal/components/chat/message.go b/packages/tui/internal/components/chat/message.go index 86ce25bdd..e4b3fbb0e 100644 --- a/packages/tui/internal/components/chat/message.go +++ b/packages/tui/internal/components/chat/message.go @@ -25,8 +25,6 @@ import ( func toMarkdown(content string, width int, backgroundColor compat.AdaptiveColor) string { r := styles.GetMarkdownRenderer(width, backgroundColor) content = strings.ReplaceAll(content, app.RootPath+"/", "") - content = strings.ReplaceAll(content, "<", "\\<") - content = strings.ReplaceAll(content, ">", "\\>") rendered, _ := r.Render(content) lines := strings.Split(rendered, "\n") @@ -230,6 +228,10 @@ func renderText(message client.MessageInfo, text string, author string) string { if message.Role == client.Assistant { markdownWidth = width - padding - 4 - 2 } + if message.Role == client.User { + text = strings.ReplaceAll(text, "<", "\\<") + text = strings.ReplaceAll(text, ">", "\\>") + } content := toMarkdown(text, markdownWidth, t.BackgroundPanel()) content = strings.Join([]string{content, info}, "\n") |