Disable syntax highlighting in raw completion cli

Mozilla-Ocho · Nov 23, 2024 · a494bd3 · a494bd3
1 parent fb59488
commit a494bd3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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);