From 9f473bafb781b319290b16cb2a0f9e62f3ae2fa5 Mon Sep 17 00:00:00 2001
From: ChengZi <chen.zhang@zilliz.com>
Date: Fri, 20 Oct 2023 14:22:59 +0800
Subject: [PATCH] fix message parsing

Signed-off-by: ChengZi <chen.zhang@zilliz.com>
---
 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,