From 502268be80b55ca9c68b90401cc5eca3650da679 Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Wed, 21 Jun 2023 10:39:22 +0800
Subject: [PATCH] Fix bug

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 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