Update multipurpose_chatbot/demos/chat_interface.py

multipurpose_chatbot/demos/chat_interface.py

@@ -110,8 +110,14 @@ def format_conversation(history, system_prompt=None):
         )
         for h in history
     ])
+    _str = ""
+    for mes, res in history:
+        if mes is not None:
+            _str += f'<<<User>>> {mes}\n'
+        if res is not None:
+            _str += f'<<<Asst>>> {res}\n'
     if system_prompt is not None:
-        _str = f"<<<System>>> {system_prompt}\n" + _str
+        _str = f"<<<Syst>>> {system_prompt}\n" + _str
     return _str
 
     
@@ -153,6 +159,8 @@ def chat_response_stream_multiturn_engine(
         else:
             response, num_tokens = outputs, -1
         yield response, num_tokens
+
+    print(format_conversation(history + [[message, response]]))
     
     if response is not None:
         yield response, num_tokens