diff --git a/public/scripts/openai.js b/public/scripts/openai.js index 0ac751555..108f4ba56 100644 --- a/public/scripts/openai.js +++ b/public/scripts/openai.js @@ -517,11 +517,12 @@ function populateDialogueExamples(prompts, chatCompletion) { [...openai_msgs_example].forEach((dialogue, dialogueIndex) => { dialogue.forEach((prompt, promptIndex) => { - const role = prompt.name === 'example_assistant' ? 'assistant' : 'user'; + const role = 'system'; const content = prompt.content || ''; const identifier = `dialogueExamples ${dialogueIndex}-${promptIndex}`; const chatMessage = new Message(role, content, identifier); + chatMessage.setName(prompt.name); if (chatCompletion.canAfford(chatMessage)) { chatCompletion.insert(chatMessage, 'dialogueExamples'); } @@ -1402,6 +1403,10 @@ class Message { } } + setName(name) { + this.name = name; + } + /** * Create a new Message instance from a prompt. * @static