From cdb31699d4020d18d3273e48d749bfc52b0f1f23 Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Sun, 29 Dec 2024 21:20:15 +0200
Subject: [PATCH] Expose new post-processing as "Semi-strict"

---
 public/index.html                          | 5 +++--
 public/scripts/openai.js                   | 1 +
 src/endpoints/backends/chat-completions.js | 4 +++-
 3 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/public/index.html b/public/index.html
index 7d2a1587f..3c25cec87 100644
--- a/public/index.html
+++ b/public/index.html
@@ -3329,8 +3329,9 @@
                             <h4 data-i18n="Prompt Post-Processing">Prompt Post-Processing</h4>
                             <select id="custom_prompt_post_processing" class="text_pole" title="Applies additional processing to the prompt before sending it to the API." data-i18n="[title]Applies additional processing to the prompt before sending it to the API.">
                                 <option data-i18n="prompt_post_processing_none" value="">None</option>
-                                <option value="merge">Merge consecutive roles</option>
-                                <option value="strict">Strict (user first, alternating roles)</option>
+                                <option data-i18n="prompt_post_processing_merge" value="merge">Merge consecutive roles</option>
+                                <option data-i18n="prompt_post_processing_semi" value="semi">Semi-strict (alternating roles)</option>
+                                <option data-i18n="prompt_post_processing_strict" value="strict">Strict (user first, alternating roles)</option>
                             </select>
                         </form>
                         <div id="01ai_form" data-source="01ai">
diff --git a/public/scripts/openai.js b/public/scripts/openai.js
index 9c0b06189..278ddb222 100644
--- a/public/scripts/openai.js
+++ b/public/scripts/openai.js
@@ -205,6 +205,7 @@ const custom_prompt_post_processing_types = {
     /** @deprecated Use MERGE instead. */
     CLAUDE: 'claude',
     MERGE: 'merge',
+    SEMI: 'semi',
     STRICT: 'strict',
 };
 
diff --git a/src/endpoints/backends/chat-completions.js b/src/endpoints/backends/chat-completions.js
index 352360ff0..82cdab5fa 100644
--- a/src/endpoints/backends/chat-completions.js
+++ b/src/endpoints/backends/chat-completions.js
@@ -64,7 +64,9 @@ function postProcessPrompt(messages, type, names) {
     switch (type) {
         case 'merge':
         case 'claude':
-            return mergeMessages(messages, names, false, true);
+            return mergeMessages(messages, names, false, false);
+        case 'semi':
+            return mergeMessages(messages, names, true, false);
         case 'strict':
             return mergeMessages(messages, names, true, true);
         case 'deepseek':