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
No known key found for this signature in database
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:',