From 74a5167a0e5bfa3fdaef333c7a23256ba0e333c6 Mon Sep 17 00:00:00 2001
From: maver <kentucky@posteo.de>
Date: Thu, 15 Jun 2023 18:10:25 +0200
Subject: [PATCH] Add nsfw avoidance prompt, quiet prompt and bias to chat
 completion

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

diff --git a/public/scripts/openai.js b/public/scripts/openai.js
index 71ff9bc12..12af16517 100644
--- a/public/scripts/openai.js
+++ b/public/scripts/openai.js
@@ -423,6 +423,22 @@ async function prepareOpenAIMessages({
         }
     });
 
+    // Insert nsfw avoidance prompt into main, if no nsfw prompt is present
+    if (false === chatCompletion.has('nsfw') && oai_settings.nsfw_avoidance_prompt) {
+        const nsfwAvoidanceMessage = new Message('system', oai_settings.nsfw_avoidance_prompt, 'nsfwAvoidance');
+        chatCompletion.insert(nsfwAvoidanceMessage, 'main');
+    }
+
+    // Insert quiet prompt into main
+    if (quietPrompt) {
+        const quietPromptMessage = new Message('system', quietPrompt, 'quietPrompt');
+        chatCompletion.insert(quietPromptMessage, 'main')
+    }
+
+    if (bias && bias.trim().length) {
+        addMessageToChatCompletion('system', bias, 'main');
+    }
+
     // Chat History
     chatCompletion.add(new MessageCollection('chatHistory'), prompts.index('chatHistory'));
     const mainChat = selected_group ? '[Start a new group chat. Group members: ${names}]' : '[Start a new Chat]';