From dbf6d348028565e99b244ba29956da858492bf66 Mon Sep 17 00:00:00 2001
From: Justine Tunney <jtunney@mozilla.com>
Date: Tue, 8 Oct 2024 19:30:18 -0700
Subject: [PATCH] Improve markdown flush method

---
 llamafile/highlight_markdown.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/llamafile/highlight_markdown.cpp b/llamafile/highlight_markdown.cpp
index 9fb6d321e4..cde006f01c 100644
--- a/llamafile/highlight_markdown.cpp
+++ b/llamafile/highlight_markdown.cpp
@@ -163,6 +163,13 @@ void HighlightMarkdown::flush(std::string *r) {
     case LANG:
         lang_.clear();
         break;
+    case STAR:
+        *r += '*';
+        break;
+    case STRONG:
+    case STRONG_STAR:
+        *r += RESET;
+        break;
     case CODE:
     case CODE_TICK:
     case CODE_TICK_TICK: