Re-add Together as a text completion source

public/scripts/textgen-models.js

@@ -0,0 +1,120 @@
+import { setGenerationParamsFromPreset } from '../script.js';
+import { isMobile } from './RossAscends-mods.js';
+import { textgenerationwebui_settings as textgen_settings } from './textgen-settings.js';
+
+let mancerModels = [];
+let togetherModels = [];
+
+export async function loadTogetherAIModels(data) {
+    if (!Array.isArray(data)) {
+        console.error('Invalid Together AI models data', data);
+        return;
+    }
+
+    togetherModels = data;
+
+    $('#model_togetherai_select').empty();
+    for (const model of data) {
+        // Hey buddy, I think you've got the wrong door.
+        if (model.display_type === 'image') {
+            continue;
+        }
+
+        const option = document.createElement('option');
+        option.value = model.name;
+        option.text = model.display_name;
+        option.selected = model.name === textgen_settings.togetherai_model;
+        $('#model_togetherai_select').append(option);
+    }
+}
+
+export async function loadMancerModels(data) {
+    if (!Array.isArray(data)) {
+        console.error('Invalid Mancer models data', data);
+        return;
+    }
+
+    mancerModels = data;
+
+    $('#mancer_model').empty();
+    for (const model of data) {
+        const option = document.createElement('option');
+        option.value = model.id;
+        option.text = model.name;
+        option.selected = model.id === textgen_settings.mancer_model;
+        $('#mancer_model').append(option);
+    }
+}
+
+function onMancerModelSelect() {
+    const modelId = String($('#mancer_model').val());
+    textgen_settings.mancer_model = modelId;
+    $('#api_button_textgenerationwebui').trigger('click');
+
+    const limits = mancerModels.find(x => x.id === modelId)?.limits;
+    setGenerationParamsFromPreset({ max_length: limits.context, genamt: limits.completion });
+}
+
+
+function onTogetherModelSelect() {
+    const modelName = String($('#model_togetherai_select').val());
+    textgen_settings.togetherai_model = modelName;
+    $('#api_button_textgenerationwebui').trigger('click');
+    const model = togetherModels.find(x => x.name === modelName);
+    setGenerationParamsFromPreset({ max_length: model.context_length });
+}
+
+function getMancerModelTemplate(option) {
+    const model = mancerModels.find(x => x.id === option?.element?.value);
+
+    if (!option.id || !model) {
+        return option.text;
+    }
+
+    const creditsPerPrompt = (model.limits?.context - model.limits?.completion) * model.pricing?.prompt;
+    const creditsPerCompletion = model.limits?.completion * model.pricing?.completion;
+    const creditsTotal = Math.round(creditsPerPrompt + creditsPerCompletion).toFixed(0);
+
+    return $((`
+        <div class="flex-container flexFlowColumn">
+            <div><strong>${DOMPurify.sanitize(model.name)}</strong> | <span>${model.limits?.context} ctx</span> / <span>${model.limits?.completion} res</span> | <small>Credits per request (max): ${creditsTotal}</small></div>
+        </div>
+    `));
+}
+
+function getTogetherModelTemplate(option) {
+    const model = togetherModels.find(x => x.name === option?.element?.value);
+
+    if (!option.id || !model) {
+        return option.text;
+    }
+
+    return $((`
+        <div class="flex-container flexFlowColumn">
+            <div><strong>${DOMPurify.sanitize(model.name)}</strong> | <span>${model.context_length || '???'} tokens</span></div>
+            <div><small>${DOMPurify.sanitize(model.description)}</small></div>
+        </div>
+    `));
+}
+
+jQuery(function () {
+    $('#mancer_model').on('change', onMancerModelSelect);
+    $('#model_togetherai_select').on('change', onTogetherModelSelect);
+
+    if (!isMobile()) {
+        $('#mancer_model').select2({
+            placeholder: 'Select a model',
+            searchInputPlaceholder: 'Search models...',
+            searchInputCssClass: 'text_pole',
+            width: '100%',
+            templateResult: getMancerModelTemplate,
+        });
+        $('#model_togetherai_select').select2({
+            placeholder: 'Select a model',
+            searchInputPlaceholder: 'Search models...',
+            searchInputCssClass: 'text_pole',
+            width: '100%',
+            templateResult: getTogetherModelTemplate,
+        });
+    }
+});