logo
Ernie
repo-copy-icon

copied

Browse Source

Fix system message bug

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 2 years ago
parent
commit
c24b34e82a
  1. 2
      ernie_chat.py

2
ernie_chat.py

@ -65,7 +65,7 @@ class ErnieChat(PyOperator):
for k, v in m.items(): for k, v in m.items():
if ('role' and 'content' in m) and (m['role'] in ['system', 'assistant', 'user']): if ('role' and 'content' in m) and (m['role'] in ['system', 'assistant', 'user']):
if m['role'] == 'system': if m['role'] == 'system':
new_messages.append(m)
new_messages.append({'role': 'user', 'content': m['content']})
new_messages.append({'role': 'assistant', 'content': 'OK.'}) new_messages.append({'role': 'assistant', 'content': 'OK.'})
else: else:
new_messages.append(m) new_messages.append(m)

Loading…
Cancel
Save