diff --git a/ernie_chat.py b/ernie_chat.py
index e2df888..80db8ad 100644
--- a/ernie_chat.py
+++ b/ernie_chat.py
@@ -36,6 +36,7 @@ class ErnieChat(PyOperator):
         self.kwargs = kwargs
 
     def __call__(self, messages: List[dict]):
+        messages = self.parse_inputs(messages)
         response = erniebot.ChatCompletion.create(
             model=self._model,
             messages=messages,