add new message types from kernel manager:

message_status for 'Kernel is ready' and the like
message_error for error messages/backtraces from execution

Author: dasmy

frontend/src/components/Chat.tsx

@@ -39,7 +39,7 @@ function Message(props: {
           </div>
         )}
 
-        {props.type == "message" &&
+        {["message", "message_status"].indexOf(props.type) !== -1 &&
           (props.showLoader ? (
             <div>
               {text} {props.showLoader ? <div className="loader"></div> : null}
@@ -53,15 +53,15 @@ function Message(props: {
             <div className="cell-output" dangerouslySetInnerHTML={{ __html: text }}></div>
           ))}
 
-        {(props.type == "message_raw") &&
+        {(["message_raw", "message_error"].indexOf(props.type) !== -1) &&
           (props.showLoader ? (
             <div>
               {text} {props.showLoader ? <div className="loader"></div> : null}
             </div>
           ) : (
             <div className="cell-output" dangerouslySetInnerHTML={{ __html: text }}></div>
           ))}
-        
+
         {props.type == "image/png" &&
           <div className="cell-output-image" dangerouslySetInnerHTML={{ __html: `<img src='data:image/png;base64,${text}' />` }}></div>
         }

gpt_code_ui/kernel_program/kernel_manager.py

@@ -149,7 +149,7 @@ def flush_kernel_msgs(kc, tries=1, timeout=0.2):
                 elif msg["msg_type"] == "error":
                     send_message(
                         utils.escape_ansi("\n".join(msg["content"]["traceback"])),
-                        "message_raw",
+                        "message_error",
                     )
             except queue.Empty:
                 hit_empty += 1

gpt_code_ui/kernel_program/main.py

@@ -80,10 +80,10 @@ def on_recv(conn, ident, message):
                 logger.debug("Kernel is ready.")
                 result_queue.put({
                     "value": "Kernel is ready.",
-                    "type": "message"
+                    "type": "message_status"
                 })
 
-        elif message["type"] in ["message", "message_raw", "image/png", "image/jpeg"]:
+        elif message["type"] in ["message", "message_raw", "message_error", "image/png", "image/jpeg"]:
             # TODO: 1:1 kernel <> channel mapping
             logger.debug("%s of type %s" % (message["value"], message["type"]))