From c2a4e2e52a207b076524fc073c2eea888b4ebb3f Mon Sep 17 00:00:00 2001
From: based <phrog@riseup.net>
Date: Sun, 20 Aug 2023 04:06:43 +1000
Subject: [PATCH] fix example dialogue nudge

---
 public/scripts/openai.js | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/public/scripts/openai.js b/public/scripts/openai.js
index 41a2d1d8b..fde0ab075 100644
--- a/public/scripts/openai.js
+++ b/public/scripts/openai.js
@@ -518,9 +518,9 @@ function populateDialogueExamples(prompts, chatCompletion) {
     chatCompletion.add(new MessageCollection('dialogueExamples'), prompts.index('dialogueExamples'));
     if (openai_msgs_example.length) {
         const newExampleChat = new Message('system', oai_settings.new_example_chat_prompt, 'newChat');
-        chatCompletion.reserveBudget(newExampleChat);
-
         [...openai_msgs_example].forEach((dialogue, dialogueIndex) => {
+            chatCompletion.reserveBudget(newExampleChat);
+            chatCompletion.insert(newExampleChat, 'dialogueExamples');
             dialogue.forEach((prompt, promptIndex) => {
                 const role = 'system';
                 const content = prompt.content || '';
@@ -533,11 +533,6 @@ function populateDialogueExamples(prompts, chatCompletion) {
                 }
             });
         });
-
-        chatCompletion.freeBudget(newExampleChat);
-
-        const chatExamples = chatCompletion.getMessages().getItemByIdentifier('dialogueExamples').getCollection();
-        if (chatExamples.length) chatCompletion.insertAtStart(newExampleChat, 'dialogueExamples');
     }
 }