From a370b9c03301fd04311a746bc1cffb353f91869d Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Sat, 29 Jul 2023 08:07:01 +0900
Subject: [PATCH] [feat] add kayra

---
 src/lib/Setting/Pages/BotSettings.svelte | 2 +-
 src/lib/UI/ModelList.svelte              | 7 ++++++-
 src/ts/process/request.ts                | 9 ++-------
 3 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/src/lib/Setting/Pages/BotSettings.svelte b/src/lib/Setting/Pages/BotSettings.svelte
index 7e669414..757a1f58 100644
--- a/src/lib/Setting/Pages/BotSettings.svelte
+++ b/src/lib/Setting/Pages/BotSettings.svelte
@@ -191,7 +191,7 @@
         {/each}
     </SelectInput>
 {/if}
-{#if $DataBase.aiModel === "novelai" || $DataBase.subModel === "novelai"}
+{#if $DataBase.aiModel === "novelai" || $DataBase.subModel === "novelai" || $DataBase.aiModel === 'novelai_kayra' || $DataBase.subModel === 'novelai_kayra'}
     <span class="text-neutral-200">NovelAI Bearer Token</span>
     <TextInput marginBottom={true} bind:value={$DataBase.novelai.token}/>
 
diff --git a/src/lib/UI/ModelList.svelte b/src/lib/UI/ModelList.svelte
index 86a40115..cb8894ba 100644
--- a/src/lib/UI/ModelList.svelte
+++ b/src/lib/UI/ModelList.svelte
@@ -42,6 +42,8 @@
                 return "Plugin"
             case "novelai":
                 return "NovelAI"
+            case "novelai_kayra":
+                return "NovelAI_kayra"
             case "novellist":
                 return "NovelList SuperTrin"
             case "novellist damsel":
@@ -115,7 +117,10 @@
                 <button class="p-2 hover:text-green-500" on:click={() => {changeModel('novellist')}}>SuperTrin</button>
                 <button class="p-2 hover:text-green-500" on:click={() => {changeModel('novellist_damsel')}}>Damsel</button>
             </Arcodion>
-            <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('novelai')}}>NovelAI Clio</button>
+            <Arcodion name="NovelAI">
+                <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('novelai')}}>NovelAI Clio</button>
+                <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('novelai_kayra')}}>NovelAI Kayra</button>
+            </Arcodion>
             {#if isTauri ||isNodeServer}
                 <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('deepai')}}>DeepAI</button>
             {/if}
diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index 352b1f06..9377ec89 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -315,13 +315,8 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
 
             break
         }
-        case 'novelai':{
-            if(!isTauri){
-                return{
-                    type: 'fail',
-                    result: "NovelAI doesn't work in web version."
-                }
-            }
+        case 'novelai':
+        case 'novelai_kayra':{
             const proompt = stringlizeChat(formated, currentChar?.name ?? '')
             const params = {
                 "input": proompt,