From 0bbd5dbdac61b8f723713b5ea3d125416167719d Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Mon, 26 Jun 2023 00:25:25 +0900
Subject: [PATCH] [feat] add novellist damsel

---
 src/lib/UI/ModelList.svelte | 10 ++++++++--
 src/ts/process/request.ts   |  4 +++-
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/src/lib/UI/ModelList.svelte b/src/lib/UI/ModelList.svelte
index ce1fc50e..e43150af 100644
--- a/src/lib/UI/ModelList.svelte
+++ b/src/lib/UI/ModelList.svelte
@@ -37,6 +37,10 @@
                 return "Plugin"
             case "novelai":
                 return "NovelAI"
+            case "novellist":
+                return "NovelList SuperTrin"
+            case "novellist damsel":
+                return "NovelList Damsel"
             default:
                 if(name.startsWith("horde:::")){
                     return name.replace(":::", " ")
@@ -79,8 +83,10 @@
             <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>
-            <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('novellist')}}>Novellist</button>
-
+            <Arcodion name="Novellist">
+                <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>
             {#if isTauri ||isNodeServer}
                 <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('deepai')}}>DeepAI</button>
diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index 397aec16..9abf0e9b 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -556,7 +556,8 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
                 result: data.results[0].text
             }
         }
-        case "novellist":{
+        case "novellist":
+        case "novellist_damsel":{
             const auth_key = db.novellistAPI;
             const api_server_url = 'https://api.tringpt.com/';
 
@@ -572,6 +573,7 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
                 top_p: 0.7,
                 tailfree: 1.0,
                 rep_pen: arg.frequencyPenalty ?? (db.frequencyPenalty / 100) + 1,
+                model: aiModel === 'novellist_damsel' ? 'damsel' : 'supertrin'
             };
 
             const response = await globalFetch(api_server_url + '/api', {