From 400645d81d58e80192ec302322389ab283b3a591 Mon Sep 17 00:00:00 2001 From: ChengZi Date: Mon, 30 Oct 2023 11:02:10 +0800 Subject: [PATCH] remove ernie system message because it dont support it now. Signed-off-by: ChengZi --- ernie_chat.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/ernie_chat.py b/ernie_chat.py index b8fcbbd..a3f9870 100644 --- a/ernie_chat.py +++ b/ernie_chat.py @@ -60,14 +60,16 @@ class ErnieChat(PyOperator): for k, v in m.items(): if k == 'question': new_m = {'role': 'user', 'content': v} + new_messages.append(new_m) elif k == 'answer': new_m = {'role': 'assistant', 'content': v} + new_messages.append(new_m) elif k == 'system': - new_m = {'role': 'system', 'content': v} + pass + # new_m = {'role': 'system', 'content': v} else: raise KeyError( - 'Invalid message key: only accept key value from ["system", "question", "answer"].') - new_messages.append(new_m) + 'Invalid message key: only accept key value from ["question", "answer"].') return new_messages def stream_output(self, response):