diff --git a/ernie_chat.py b/ernie_chat.py
index b8fcbbd..a3f9870 100644
--- a/ernie_chat.py
+++ b/ernie_chat.py
@@ -60,14 +60,16 @@ class ErnieChat(PyOperator):
                 for k, v in m.items():
                     if k == 'question':
                         new_m = {'role': 'user', 'content': v}
+                        new_messages.append(new_m)
                     elif k == 'answer':
                         new_m = {'role': 'assistant', 'content': v}
+                        new_messages.append(new_m)
                     elif k == 'system':
-                        new_m = {'role': 'system', 'content': v}
+                        pass
+                    #     new_m = {'role': 'system', 'content': v}
                     else:
                         raise KeyError(
-                            'Invalid message key: only accept key value from ["system", "question", "answer"].')
-                    new_messages.append(new_m)
+                            'Invalid message key: only accept key value from ["question", "answer"].')
         return new_messages
 
     def stream_output(self, response):