mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

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

Browse source
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
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
extensions/openai/script.py
View file

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