Improve markdown flush method

Mozilla-Ocho · Oct 9, 2024 · dbf6d34 · dbf6d34
1 parent 63205ee
commit dbf6d34
Showing 1 changed file with 7 additions and 0 deletions.
diff --git 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: