From a9ddf660018f3961bcf93d9ea5338a89263904b6 Mon Sep 17 00:00:00 2001
From: kwaroran <amasio1234@proton.me>
Date: Fri, 13 Sep 2024 08:54:29 +0900
Subject: [PATCH] Add Jamba Support

---
 src/lib/Setting/Pages/BotSettings.svelte | 4 ++++
 src/lib/UI/ModelList.svelte              | 4 ++++
 src/ts/process/request.ts                | 6 ++++++
 src/ts/storage/database.ts               | 1 +
 4 files changed, 15 insertions(+)

diff --git a/src/lib/Setting/Pages/BotSettings.svelte b/src/lib/Setting/Pages/BotSettings.svelte
index f0ab8bbf..2626bbf3 100644
--- a/src/lib/Setting/Pages/BotSettings.svelte
+++ b/src/lib/Setting/Pages/BotSettings.svelte
@@ -96,6 +96,10 @@
             <TextInput marginBottom={true} size={"sm"} placeholder="..." hideText={$DataBase.hideApiKey} bind:value={$DataBase.google.projectId}/>
         {/if}
     {/if}
+    {#if $DataBase.aiModel.startsWith('jamba') || $DataBase.subModel.startsWith('jamba')}
+        <span class="text-textcolor">AI21 {language.apiKey}</span>
+        <TextInput hideText={$DataBase.hideApiKey} marginBottom={true} size={"sm"} placeholder="..." bind:value={$DataBase.ai21Key}/>
+    {/if}
     {#if $DataBase.aiModel.startsWith('novellist') || $DataBase.subModel.startsWith('novellist')}
         <span class="text-textcolor">NovelList {language.apiKey}</span>
         <TextInput hideText={$DataBase.hideApiKey} marginBottom={true} size={"sm"} placeholder="..." bind:value={$DataBase.novellistAPI}/>
diff --git a/src/lib/UI/ModelList.svelte b/src/lib/UI/ModelList.svelte
index 6d9c31c8..005f8337 100644
--- a/src/lib/UI/ModelList.svelte
+++ b/src/lib/UI/ModelList.svelte
@@ -137,6 +137,10 @@
                 <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>
+            <Arcodion name="AI21">
+                <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('jamba-1.5-large')}}>Jamba 1.5 Large</button>
+                <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('jamba-1.5-medium')}}>Jamba 1.5 Medium</button>
+            </Arcodion>
             <Arcodion name="Horde">
                 {#await getHordeModels()}
                     <button class="p-2">Loading...</button>
diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index f2e5815b..44513621 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -238,6 +238,8 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
         case 'gpt4o-chatgpt':
         case 'gpt4o1-preview':
         case 'gpt4o1-mini':
+        case 'jamba-1.5-large':
+        case 'jamba-1.5-medium':
         case 'reverse_proxy':{
             let formatedChat:OpenAIChatExtra[] = []
             for(let i=0;i<formated.length;i++){
@@ -613,6 +615,10 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
             if(risuIdentify){
                 headers["X-Proxy-Risu"] = 'RisuAI'
             }
+            if(aiModel.startsWith('jamba')){
+                headers['Authorization'] = 'Bearer ' + db.ai21Key
+                replacerURL = 'https://api.ai21.com/studio/v1/chat/completions'
+            }
             if(multiGen){
                 // @ts-ignore
                 body.n = db.genTime
diff --git a/src/ts/storage/database.ts b/src/ts/storage/database.ts
index 7feda433..c667953d 100644
--- a/src/ts/storage/database.ts
+++ b/src/ts/storage/database.ts
@@ -738,6 +738,7 @@ export interface Database{
     jsonSchema:string
     strictJsonSchema:boolean
     extractJson:string
+    ai21Key:string
     statics: {
         messages: number
         imports: number