mirror of
https://github.com/tiennm99/goclaw.git
synced 2026-10-14 16:17:25 +00:00
viettranx
31ba182b40
Zalo API does not support any markup rendering. LLM output containing markdown artifacts (**bold**, `code`, headers, etc.) was sent as-is, making messages hard to read. Add StripMarkdown() to convert markdown to clean plain text, and inject a system prompt hint instructing the model to output plain text on Zalo channels.
64 lines
2.0 KiB
Go
64 lines
2.0 KiB
Go
package zalo
|
|
|
|
import "testing"
|
|
|
|
func TestStripMarkdown(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
in string
|
|
want string
|
|
}{
|
|
{"empty", "", ""},
|
|
{"plain text", "hello world", "hello world"},
|
|
|
|
// Bold & italic
|
|
{"bold stars", "this is **bold** text", "this is bold text"},
|
|
{"bold underscores", "this is __bold__ text", "this is bold text"},
|
|
{"bold+italic stars", "***important***", "important"},
|
|
{"strikethrough", "this is ~~deleted~~ text", "this is deleted text"},
|
|
|
|
// Code
|
|
{"inline code", "use `fmt.Println` here", "use fmt.Println here"},
|
|
{"fenced code block", "before\n```go\nfmt.Println(\"hi\")\n```\nafter", "before\nfmt.Println(\"hi\")\n\nafter"},
|
|
{"fenced code block no lang", "```\ncode here\n```", "code here"},
|
|
|
|
// Links & images
|
|
{"link", "click [here](https://example.com) now", "click here (https://example.com) now"},
|
|
{"image", "see  below", "see below"},
|
|
|
|
// Headers
|
|
{"h1", "# Title", "Title"},
|
|
{"h3", "### Section", "Section"},
|
|
{"h6", "###### Deep", "Deep"},
|
|
|
|
// Horizontal rules
|
|
{"hr dashes", "above\n---\nbelow", "above\n\nbelow"},
|
|
{"hr stars", "above\n***\nbelow", "above\n\nbelow"},
|
|
|
|
// Blockquotes
|
|
{"blockquote", "> this is quoted\n> second line", "this is quoted\nsecond line"},
|
|
{"nested blockquote", "> > deep", "> deep"},
|
|
|
|
// Bullets
|
|
{"dash bullet", "- item one\n- item two", "• item one\n• item two"},
|
|
{"star bullet", "* item one\n* item two", "• item one\n• item two"},
|
|
{"plus bullet", "+ item one", "• item one"},
|
|
{"indented bullet", "list:\n - nested item", "list:\n • nested item"},
|
|
|
|
// Excessive newlines
|
|
{"excessive newlines", "a\n\n\n\nb", "a\n\nb"},
|
|
|
|
// Mixed
|
|
{"mixed markdown", "## Hello\n\nThis is **bold** and `code`.\n\n- item\n- [link](url)\n\n> quote", "Hello\n\nThis is bold and code.\n\n• item\n• link (url)\n\nquote"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got := StripMarkdown(tt.in)
|
|
if got != tt.want {
|
|
t.Errorf("StripMarkdown(%q)\n got: %q\nwant: %q", tt.in, got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|