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)