Browse Source
remove ernie system message because it dont support it now.
Signed-off-by: ChengZi <chen.zhang@zilliz.com>
main
ChengZi
1 year ago
1 changed files with
5 additions and
3 deletions
-
ernie_chat.py
|
|
|
@ -60,14 +60,16 @@ class ErnieChat(PyOperator): |
|
|
|
for k, v in m.items(): |
|
|
|
if k == 'question': |
|
|
|
new_m = {'role': 'user', 'content': v} |
|
|
|
new_messages.append(new_m) |
|
|
|
elif k == 'answer': |
|
|
|
new_m = {'role': 'assistant', 'content': v} |
|
|
|
new_messages.append(new_m) |
|
|
|
elif k == 'system': |
|
|
|
new_m = {'role': 'system', 'content': v} |
|
|
|
pass |
|
|
|
# new_m = {'role': 'system', 'content': v} |
|
|
|
else: |
|
|
|
raise KeyError( |
|
|
|
'Invalid message key: only accept key value from ["system", "question", "answer"].') |
|
|
|
new_messages.append(new_m) |
|
|
|
'Invalid message key: only accept key value from ["question", "answer"].') |
|
|
|
return new_messages |
|
|
|
|
|
|
|
def stream_output(self, response): |
|
|
|
|
