Browse Source
Fix bug
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
ernie_chat.py
|
|
@ -68,7 +68,7 @@ class ErnieChat(PyOperator): |
|
|
elif k == 'answer': |
|
|
elif k == 'answer': |
|
|
new_m = {'role': 'assistant', 'content': v} |
|
|
new_m = {'role': 'assistant', 'content': v} |
|
|
else: |
|
|
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) |
|
|
new_messages.append(new_m) |
|
|
return new_messages |
|
|
return new_messages |
|
|
|
|
|
|
|
|
|
