From c24b34e82af97377ba3055902fbc9ac05790af1c Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Fri, 30 Jun 2023 17:26:15 +0800 Subject: [PATCH] Fix system message bug Signed-off-by: Jael Gu --- ernie_chat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ernie_chat.py b/ernie_chat.py index 61b3363..951a967 100644 --- a/ernie_chat.py +++ b/ernie_chat.py @@ -65,7 +65,7 @@ class ErnieChat(PyOperator): for k, v in m.items(): if ('role' and 'content' in m) and (m['role'] in ['system', 'assistant', 'user']): if m['role'] == 'system': - new_messages.append(m) + new_messages.append({'role': 'user', 'content': m['content']}) new_messages.append({'role': 'assistant', 'content': 'OK.'}) else: new_messages.append(m)