From 7b9e04258928ef80547cbcdca593fec773fa7fe9 Mon Sep 17 00:00:00 2001
From: SillyLossy <titan0599@gmail.com>
Date: Mon, 1 May 2023 17:15:16 +0300
Subject: [PATCH] #210 [OAI] Don't insert example dialogues without bots

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

diff --git a/public/scripts/openai.js b/public/scripts/openai.js
index d636ca36a..852994f5a 100644
--- a/public/scripts/openai.js
+++ b/public/scripts/openai.js
@@ -183,6 +183,13 @@ function setOpenAIMessageExamples(mesExamplesArray) {
         // remove <START> {Example Dialogue:} and replace \r\n with just \n
         let replaced = item.replace(/<START>/i, "{Example Dialogue:}").replace(/\r/gm, '');
         let parsed = parseExampleIntoIndividual(replaced);
+
+        // Discard example dialogue if there's not a single bot message in it 
+        if (parsed.findIndex(x => x.name == "example_assistant") === -1) {
+            console.warn('Example dialogue without bot reply discarded:', replaced);
+            parsed = [];
+        }
+
         // add to the example message blocks array
         openai_msgs_example.push(parsed);
     }