From 9f473bafb781b319290b16cb2a0f9e62f3ae2fa5 Mon Sep 17 00:00:00 2001 From: ChengZi Date: Fri, 20 Oct 2023 14:22:59 +0800 Subject: [PATCH] fix message parsing Signed-off-by: ChengZi --- ernie_chat.py | 1 + 1 file changed, 1 insertion(+) 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,