WIP: colorizing the reasoning content

Author: bandoti

common/console.cpp

@@ -155,6 +155,10 @@ namespace console {
                     break;
                 case error:
                     fprintf(out, ANSI_BOLD ANSI_COLOR_RED);
+                    break;
+                case reasoning:
+                    fprintf(out, ANSI_COLOR_BLUE);
+                    break;
             }
             current_display = display;
             fflush(out);

common/console.h

@@ -9,7 +9,8 @@ namespace console {
         reset = 0,
         prompt,
         user_input,
-        error
+        error,
+        reasoning
     };
 
     void init(bool use_simple_io, bool use_advanced_display);

tools/main/main.cpp

@@ -828,8 +828,15 @@ int main(int argc, char ** argv) {
                 if (auto * formatter = chat_add_and_format.get_partial_formatter()) {
                     auto outputs = (*formatter)(assistant_ss.str());
                     for (const auto & out : outputs) {
-                        LOG("%s", out.formatted.c_str());
+                        if (out.type == partial_formatter::REASONING) {
+                            console::set_display(console::reasoning);
+                        } else {
+                            console::set_display(console::reset);
+                        }
+                        fprintf(stdout, "%s", out.formatted.c_str());
+                        fflush(stdout);
                     }
+                    console::set_display(console::reset);
                 }
             }