Browse Source
Fix system message bug
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
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) |
|
|
|