121 lines
4.1 KiB
JavaScript
121 lines
4.1 KiB
JavaScript
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,
|
|
});
|
|
}
|
|
});
|