Browse Source
fix message parsing
Signed-off-by: ChengZi <chen.zhang@zilliz.com>
main
ChengZi
1 year ago
1 changed files with
1 additions and
0 deletions
-
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, |
|
|