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,