rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-03-02 02:47:52 +01:00
Code Issues Packages Projects Releases Wiki Activity

Enum provider for /model

Browse Source
This commit is contained in:
Wolfsblvt 2024-06-21 21:24:37 +02:00
parent 824d0a9b63
commit da6d77cffd
1 changed files with 27 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
public/scripts/slash-commands.js
View File

@ -1326,9 +1326,11 @@ SlashCommandParser.addCommandObject(SlashCommand.fromProps({
callback: modelCallback, callback: modelCallback,
returns: 'current model', returns: 'current model',
unnamedArgumentList: [ unnamedArgumentList: [
new SlashCommandArgument( SlashCommandArgument.fromProps({
'model name', [ARGUMENT_TYPE.STRING], false, description: 'model name',
), typeList: [ARGUMENT_TYPE.STRING],
enumProvider: () => getModelOptions()?.options.map(option => new SlashCommandEnumValue(option.value, option.value !== option.text ? option.text : null)),
}),
], ],
helpString: 'Sets the model for the current API. Gets the current model name if no argument is provided.', helpString: 'Sets the model for the current API. Gets the current model name if no argument is provided.',
})); }));
@ -2905,12 +2907,11 @@ function setBackgroundCallback(_, bg) {
} }
/** /**
* Sets a model for the current API. * Retrieves the available model options based on the currently selected main API and its subtype
* @param {object} _ Unused *
* @param {string} model New model name * @returns {{control: HTMLSelectElement, options: HTMLOptionElement[]}?} An array of objects representing the available model options, or null if not supported
* @returns {string} New or existing model name
*/ */
function modelCallback(_, model) { function getModelOptions() {
const modelSelectMap = [ const modelSelectMap = [
{ id: 'model_togetherai_select', api: 'textgenerationwebui', type: textgen_types.TOGETHERAI }, { id: 'model_togetherai_select', api: 'textgenerationwebui', type: textgen_types.TOGETHERAI },
{ id: 'openrouter_model', api: 'textgenerationwebui', type: textgen_types.OPENROUTER }, { id: 'openrouter_model', api: 'textgenerationwebui', type: textgen_types.OPENROUTER },
@ -2951,17 +2952,33 @@ function modelCallback(_, model) {
if (!modelSelectItem) { if (!modelSelectItem) {
toastr.info('Setting a model for your API is not supported or not implemented yet.'); toastr.info('Setting a model for your API is not supported or not implemented yet.');
return ''; return null;
} }
const modelSelectControl = document.getElementById(modelSelectItem); const modelSelectControl = document.getElementById(modelSelectItem);
if (!(modelSelectControl instanceof HTMLSelectElement)) { if (!(modelSelectControl instanceof HTMLSelectElement)) {
toastr.error(`Model select control not found: ${main_api}[${apiSubType}]`); toastr.error(`Model select control not found: ${main_api}[${apiSubType}]`);
return ''; return null;
} }
const options = Array.from(modelSelectControl.options); const options = Array.from(modelSelectControl.options);
return { control: modelSelectControl, options };
}
/**
* Sets a model for the current API.
* @param {object} _ Unused
* @param {string} model New model name
* @returns {string} New or existing model name
*/
function modelCallback(_, model) {
const { control: modelSelectControl, options } = getModelOptions();
// If no model was found, the reason was already logged, we just return here
if (options === null) {
return '';
}
if (!options.length) { if (!options.length) {
toastr.warning('No model options found. Check your API settings.'); toastr.warning('No model options found. Check your API settings.');