From d60c916feb4e0627dd69ed918a2ffaee89059acd Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Mon, 29 May 2023 15:16:06 +0900
Subject: [PATCH] [feat] added novelai

---
 src/lib/Setting/Pages/BotSettings.svelte | 2 +-
 src/lib/UI/ModelList.svelte              | 6 ++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/lib/Setting/Pages/BotSettings.svelte b/src/lib/Setting/Pages/BotSettings.svelte
index 1c7548a2..35bc78a5 100644
--- a/src/lib/Setting/Pages/BotSettings.svelte
+++ b/src/lib/Setting/Pages/BotSettings.svelte
@@ -9,7 +9,7 @@
     import ModelList from "src/lib/UI/ModelList.svelte";
     import DropList from "src/lib/SideBars/DropList.svelte";
     import { PlusIcon, TrashIcon } from "lucide-svelte";
-  import { onDestroy } from "svelte";
+    import { onDestroy } from "svelte";
     let tokens = {
         mainPrompt: 0,
         jailbreak: 0,
diff --git a/src/lib/UI/ModelList.svelte b/src/lib/UI/ModelList.svelte
index 371b2ffd..fb89d73a 100644
--- a/src/lib/UI/ModelList.svelte
+++ b/src/lib/UI/ModelList.svelte
@@ -3,6 +3,7 @@
     import { getHordeModels } from "src/ts/horde/getModels";
     import Arcodion from "./Arcodion.svelte";
     import { language } from "src/lang";
+  import { isNodeServer, isTauri } from "src/ts/storage/globalApi";
 
     export let value = ""
     export let onChange: (v:string) => void = (v) => {}
@@ -24,6 +25,8 @@
                 return "Kobold"
             case "custom":
                 return "Plugin"
+            case "novelai":
+                return "NovelAI"
             default:
                 if(name.startsWith("horde:::")){
                     return name.replace(":::", " ")
@@ -55,6 +58,9 @@
             <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('textgen_webui')}}>Oobabooga WebUI</button>
             <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('palm2')}}>Google PaLM2</button>
             <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('kobold')}}>Kobold</button>
+            {#if isTauri ||isNodeServer}
+                <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('novelai')}}>NovelAI Clio</button>
+            {/if}
             <Arcodion name="Horde">
                 {#await getHordeModels()}
                     <button class="p-2">Loading...</button>