logo
Ernie
repo-copy-icon

copied

Browse Source

fix message parsing

Signed-off-by: ChengZi <chen.zhang@zilliz.com>
main
ChengZi 1 year ago
parent
commit
9f473bafb7
  1. 1
      ernie_chat.py

1
ernie_chat.py

@ -36,6 +36,7 @@ class ErnieChat(PyOperator):
self.kwargs = kwargs self.kwargs = kwargs
def __call__(self, messages: List[dict]): def __call__(self, messages: List[dict]):
messages = self.parse_inputs(messages)
response = erniebot.ChatCompletion.create( response = erniebot.ChatCompletion.create(
model=self._model, model=self._model,
messages=messages, messages=messages,

Loading…
Cancel
Save