From 9f2d32524c5a4c1ccf0065604ab5f5552cb45199 Mon Sep 17 00:00:00 2001
From: Fayiron <59962292+Fayiron@users.noreply.github.com>
Date: Sat, 16 Dec 2023 14:39:30 +0100
Subject: [PATCH] Add TogetherAI as a chat completion source, basic

---
 public/index.html                               | 7 ++++---
 public/script.js                                | 8 +++++++-
 public/scripts/RossAscends-mods.js              | 1 +
 public/scripts/extensions/vectors/index.js      | 5 +++--
 public/scripts/extensions/vectors/settings.html | 1 +
 5 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/public/index.html b/public/index.html
index 58e42f700..8b79b2455 100644
--- a/public/index.html
+++ b/public/index.html
@@ -509,7 +509,7 @@
                                         </div>
                                     </div>
                                 </div>
-                                <div data-newbie-hidden class="range-block" data-source="openai,claude,openrouter,ai21,scale,palm">
+                                <div data-newbie-hidden class="range-block" data-source="openai,claude,openrouter,ai21,scale,palm,togetherai">
                                     <div class="range-block-title" data-i18n="Top-p">
                                         Top P
                                     </div>
@@ -749,7 +749,7 @@
                                         <div id="openai_proxy_password_show" title="Peek a password" class="menu_button fa-solid fa-eye-slash fa-fw"></div>
                                     </div>
                                 </div>
-                                <div data-newbie-hidden class="range-block" data-source="openai,openrouter">
+                                <div data-newbie-hidden class="range-block" data-source="openai,openrouter,togetherai">
                                     <div class="range-block-title justifyLeft" data-i18n="Seed">
                                         Seed
                                     </div>
@@ -1598,7 +1598,7 @@
                             <option value="koboldhorde"><span data-i18n="KoboldAI Horde">KoboldAI Horde</span></option>
                             <option value="novel"><span data-i18n="NovelAI">NovelAI</span></option>
                             <option value="textgenerationwebui"><span data-i18n="Text Completion">Text Completion (ooba, Mancer, Aphrodite, TabbyAPI, KoboldCpp)</span></option>
-                            <option value="openai"><span data-i18n="Chat Completion (OpenAI, Claude, Window/OpenRouter, Scale, AI21)">Chat Completion (OpenAI, Claude, Window, OpenRouter, Scale, AI21, PaLM)</span></option>
+                            <option value="openai"><span data-i18n="Chat Completion (OpenAI, Claude, Window/OpenRouter, Scale, AI21, TogetherAI)">Chat Completion (OpenAI, Claude, Window, OpenRouter, Scale, AI21, PaLM, TogetherAI)</span></option>
                         </select>
                     </div>
                     <div id="kobold_horde" style="position: relative;"> <!-- shows the kobold settings -->
@@ -1847,6 +1847,7 @@
                             <option value="scale">Scale</option>
                             <option value="ai21">AI21</option>
                             <option value="palm">Google PaLM 2</option>
+                            <option value="togetherai">TogetherAI</option>
                         </select>
                         <form id="openai_form" data-source="openai" action="javascript:void(null);" method="post" enctype="multipart/form-data">
                             <h4><span data-i18n="OpenAI API key">OpenAI API key</span></h4>
diff --git a/public/script.js b/public/script.js
index 64cb41752..5a173d7f8 100644
--- a/public/script.js
+++ b/public/script.js
@@ -5326,6 +5326,7 @@ function changeMainAPI() {
         case chat_completion_sources.OPENAI:
         case chat_completion_sources.AI21:
         case chat_completion_sources.PALM:
+        case chat_completion_sources.TOGETHERAI:
         default:
             setupChatCompletionPromptManager(oai_settings);
             break;
@@ -7460,6 +7461,11 @@ function connectAPISlash(_, text) {
             source: 'palm',
             button: '#api_button_openai',
         },
+        'togetherai': {
+            selected: 'openai',
+            source: 'togetherai',
+            button: '#api_button_openai',
+        },
     };
 
     const apiConfig = apiMap[text];
@@ -7734,7 +7740,7 @@ jQuery(async function () {
     }
 
     registerSlashCommand('dupe', DupeChar, [], '– duplicates the currently selected character', true, true);
-    registerSlashCommand('api', connectAPISlash, [], '<span class="monospace">(kobold, horde, novel, ooba, oai, claude, windowai, openrouter, scale, ai21, palm)</span> – connect to an API', true, true);
+    registerSlashCommand('api', connectAPISlash, [], '<span class="monospace">(kobold, horde, novel, ooba, oai, claude, windowai, openrouter, scale, ai21, palm, togetherai)</span> – connect to an API', true, true);
     registerSlashCommand('impersonate', doImpersonate, ['imp'], '– calls an impersonation response', true, true);
     registerSlashCommand('delchat', doDeleteChat, [], '– deletes the current chat', true, true);
     registerSlashCommand('closechat', doCloseChat, [], '– closes the current chat', true, true);
diff --git a/public/scripts/RossAscends-mods.js b/public/scripts/RossAscends-mods.js
index a565502eb..875dd571d 100644
--- a/public/scripts/RossAscends-mods.js
+++ b/public/scripts/RossAscends-mods.js
@@ -416,6 +416,7 @@ function RA_autoconnect(PrevApi) {
                     || (secret_state[SECRET_KEYS.OPENROUTER] && oai_settings.chat_completion_source == chat_completion_sources.OPENROUTER)
                     || (secret_state[SECRET_KEYS.AI21] && oai_settings.chat_completion_source == chat_completion_sources.AI21)
                     || (secret_state[SECRET_KEYS.PALM] && oai_settings.chat_completion_source == chat_completion_sources.PALM)
+                    || (secret_state[SECRET_KEYS.TOGETHERAI] && oai_settings.chat_completion_source == chat_completion_sources.TOGETHERAI)
                 ) {
                     $('#api_button_openai').trigger('click');
                 }
diff --git a/public/scripts/extensions/vectors/index.js b/public/scripts/extensions/vectors/index.js
index dd5744c72..89137d939 100644
--- a/public/scripts/extensions/vectors/index.js
+++ b/public/scripts/extensions/vectors/index.js
@@ -392,8 +392,9 @@ async function getSavedHashes(collectionId) {
  * @returns {Promise<void>}
  */
 async function insertVectorItems(collectionId, items) {
-    if (settings.source === 'openai' && !secret_state[SECRET_KEYS.OPENAI] ||
-        settings.source === 'palm' && !secret_state[SECRET_KEYS.PALM]) {
+    if ((settings.source === 'openai' && !secret_state[SECRET_KEYS.OPENAI]) ||
+        (settings.source === 'palm' && !secret_state[SECRET_KEYS.PALM]) ||
+        (settings.source === 'togetherai' && !secret_state[SECRET_KEYS.TOGETHERAI])) {
         throw new Error('Vectors: API key missing', { cause: 'api_key_missing' });
     }
 
diff --git a/public/scripts/extensions/vectors/settings.html b/public/scripts/extensions/vectors/settings.html
index fa6c6f4c7..c7fbbeab4 100644
--- a/public/scripts/extensions/vectors/settings.html
+++ b/public/scripts/extensions/vectors/settings.html
@@ -13,6 +13,7 @@
                     <option value="transformers">Local (Transformers)</option>
                     <option value="openai">OpenAI</option>
                     <option value="palm">Google MakerSuite (PaLM)</option>
+                    <option value="togetherai">Together AI</option>
                 </select>
             </div>