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

2023-05-29 13:19:57 -04:00 · 2023-05-29 13:19:57 -04:00 · 2b7ba9586f
commit 2b7ba9586f
parent 6de727c524
1 changed files with 1 additions and 1 deletions
--- a/extensions/openai/script.py
+++ b/extensions/openai/script.py
@ -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:',