fix message parsing

Signed-off-by: ChengZi <chen.zhang@zilliz.com>
1 year ago · 9f473bafb7
1 changed files with 1 additions and 0 deletions
--- 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,