[feat] add novellist damsel

This commit is contained in:
kwaroran
2023-06-26 00:25:25 +09:00
parent 90b60d71b5
commit 0bbd5dbdac
2 changed files with 11 additions and 3 deletions

View File

@@ -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>

View File

@@ -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', {