From 98aa9cb9519ea2742b0b56ff64b5ab44323061a9 Mon Sep 17 00:00:00 2001
From: kwaroran <amasio1234@proton.me>
Date: Sat, 20 Apr 2024 19:49:23 +0900
Subject: [PATCH] Fix oobamode chat formating

---
 src/lib/Setting/Pages/OobaSettings.svelte | 22 ++++++++++++++--------
 1 file changed, 14 insertions(+), 8 deletions(-)

diff --git a/src/lib/Setting/Pages/OobaSettings.svelte b/src/lib/Setting/Pages/OobaSettings.svelte
index 5093291b..f9327fcd 100644
--- a/src/lib/Setting/Pages/OobaSettings.svelte
+++ b/src/lib/Setting/Pages/OobaSettings.svelte
@@ -18,15 +18,21 @@
 
 <Arcodion name="Ooba Settings" styled>
     {#if instructionMode}
-        <span class="text-textcolor">System Prefix</span>
-        <TextAreaInput fullwidth autocomplete="off" height={"24"} bind:value={$DataBase.ooba.formating.systemPrefix} />
-        <span class="text-textcolor">User Prefix</span>
-        <TextAreaInput fullwidth autocomplete="off" height={"24"} bind:value={$DataBase.ooba.formating.userPrefix} />
-        <span class="text-textcolor">Assistant Prefix</span>
-        <TextAreaInput fullwidth autocomplete="off" height={"24"} bind:value={$DataBase.ooba.formating.assistantPrefix} />
-        <span class="text-textcolor">Seperator</span>
-        <TextAreaInput fullwidth autocomplete="off" height={"24"} bind:value={$DataBase.ooba.formating.seperator} />
+        <span class="text-textcolor">Chat Formating</span>
+        <SelectInput bind:value={$DataBase.instructChatTemplate}>
+            <OptionInput value="chatml">ChatML</OptionInput>
+            <OptionInput value="llama3">Llama3</OptionInput>
+            <OptionInput value="gpt2">GPT2</OptionInput>
+            <OptionInput value="gemma">Gemma</OptionInput>
+            <OptionInput value="mistral">Mistral</OptionInput>
+            <OptionInput value="llama2">Llama2</OptionInput>
+            <OptionInput value="jinja">Custom (Jinja)</OptionInput>
+        </SelectInput>
 
+        {#if $DataBase.instructChatTemplate === 'jinja'}
+            <span class="text-textcolor">Jinja Template</span>
+            <TextAreaInput fullwidth autocomplete="off" height={"24"} bind:value={$DataBase.JinjaTemplate} />
+        {/if}
     {:else}
         <span class="text-textcolor">Ooba Mode</span>
         <SelectInput className="mt-2 mb-4" bind:value={$DataBase.reverseProxyOobaArgs.mode}>