Fixes #2326, KeyError: 'assistant' (#2382)

This commit is contained in:
matatonic 2023-05-29 13:19:57 -04:00 committed by GitHub
parent 6de727c524
commit 2b7ba9586f
WARNING! Although there is a key with this ID in the database it does not verify this commit! This commit is SUSPICIOUS.
GPG key ID: 4AEE18F83AFDEB23

View file

@ -259,7 +259,7 @@ class Handler(BaseHTTPRequestHandler):
role_formats = {
'user': 'user: {message}\n',
'bot': 'assistant: {message}\n',
'assistant': 'assistant: {message}\n',
'system': '{message}',
'context': 'You are a helpful assistant. Answer as concisely as possible.',
'prompt': 'assistant:',