From 502268be80b55ca9c68b90401cc5eca3650da679 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Wed, 21 Jun 2023 10:39:22 +0800 Subject: [PATCH] Fix bug Signed-off-by: Jael Gu --- ernie_chat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ernie_chat.py b/ernie_chat.py index 2e6c99b..7a44f00 100644 --- a/ernie_chat.py +++ b/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