From a494bd3f998c4002177b50d31c943cdfcde2396b Mon Sep 17 00:00:00 2001
From: Justine Tunney <jtunney@mozilla.com>
Date: Sat, 23 Nov 2024 11:19:03 -0800
Subject: [PATCH] Disable syntax highlighting in raw completion cli

---
 llamafile/chatbot_repl.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/llamafile/chatbot_repl.cpp b/llamafile/chatbot_repl.cpp
index a9aad32b51..0e8ccbdf64 100644
--- a/llamafile/chatbot_repl.cpp
+++ b/llamafile/chatbot_repl.cpp
@@ -129,8 +129,9 @@ void repl() {
     }
 
     // perform important setup
-    HighlightMarkdown highlighter;
-    ColorBleeder bleeder(&highlighter);
+    HighlightTxt txt;
+    HighlightMarkdown markdown;
+    ColorBleeder bleeder(is_base_model() ? (Highlight *)&txt : (Highlight *)&markdown);
     llama_sampling_context *sampler = llama_sampling_init(g_params.sparams);
     signal(SIGINT, on_sigint);