logo
Ernie
repo-copy-icon

copied

Browse Source

Fix bug

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

2
ernie_chat.py

@ -68,7 +68,7 @@ class ErnieChat(PyOperator):
elif k == 'answer':
new_m = {'role': 'assistant', 'content': v}
else:
'Invalid message key: only accept key value from ["question", "answer"].'
raise KeyError('Invalid message key: only accept key value from ["question", "answer"].')
new_messages.append(new_m)
return new_messages

Loading…
Cancel
Save