SillyTavern/public/scripts/kai-settings.js

379 lines
13 KiB
JavaScript
Raw Normal View History

2023-07-20 19:32:15 +02:00
import {
getRequestHeaders,
saveSettingsDebounced,
getStoppingStrings,
} from "../script.js";
2023-08-04 22:49:55 +02:00
import {
power_user,
} from "./power-user.js";
import { getSortableDelay } from "./utils.js";
2023-08-04 22:49:55 +02:00
export const kai_settings = {
2023-07-20 19:32:15 +02:00
temp: 1,
rep_pen: 1,
rep_pen_range: 0,
top_p: 1,
top_a: 1,
top_k: 0,
typical: 1,
tfs: 1,
rep_pen_slope: 0.9,
single_line: false,
streaming_kobold: false,
sampler_order: [0, 1, 2, 3, 4, 5, 6],
2023-09-01 00:07:04 +02:00
mirostat: 0,
mirostat_tau: 5.0,
mirostat_eta: 0.1,
use_default_badwordsids: true,
2023-09-01 00:07:04 +02:00
};
export const kai_flags = {
can_use_tokenization: false,
can_use_stop_sequence: false,
can_use_streaming: false,
can_use_default_badwordsids: false,
2023-09-01 00:07:04 +02:00
can_use_mirostat: false,
2023-07-20 19:32:15 +02:00
};
2023-09-01 00:07:04 +02:00
const defaultValues = Object.freeze(structuredClone(kai_settings));
2023-07-20 19:32:15 +02:00
const MIN_STOP_SEQUENCE_VERSION = '1.2.2';
2023-09-01 00:07:04 +02:00
const MIN_UNBAN_VERSION = '1.2.4';
2023-07-20 19:32:15 +02:00
const MIN_STREAMING_KCPPVERSION = '1.30';
const MIN_TOKENIZATION_KCPPVERSION = '1.41';
2023-09-01 00:07:04 +02:00
const MIN_MIROSTAT_KCPPVERSION = '1.35';
const KOBOLDCPP_ORDER = [6, 0, 1, 3, 4, 2, 5];
2023-07-20 19:32:15 +02:00
export function formatKoboldUrl(value) {
2023-07-20 19:32:15 +02:00
try {
const url = new URL(value);
if (!power_user.relaxed_api_urls) {
2023-08-04 22:49:55 +02:00
url.pathname = '/api';
}
return url.toString();
2023-08-07 23:10:05 +02:00
} catch { } // Just using URL as a validation check
return null;
2023-07-20 19:32:15 +02:00
}
export function loadKoboldSettings(preset) {
2023-07-20 19:32:15 +02:00
for (const name of Object.keys(kai_settings)) {
2023-09-01 00:07:04 +02:00
const value = preset[name] ?? defaultValues[name];
2023-07-20 19:32:15 +02:00
const slider = sliders.find(x => x.name === name);
2023-09-01 00:07:04 +02:00
if (!slider) {
2023-07-20 19:32:15 +02:00
continue;
}
const formattedValue = slider.format(value);
2023-09-01 00:07:04 +02:00
slider.setValue(value);
$(slider.sliderId).val(value);
2023-07-20 19:32:15 +02:00
$(slider.counterId).text(formattedValue);
}
// TODO: refactor checkboxes (if adding any more)
if (preset.hasOwnProperty('single_line')) {
kai_settings.single_line = preset.single_line;
$('#single_line').prop('checked', kai_settings.single_line);
}
if (preset.hasOwnProperty('streaming_kobold')) {
kai_settings.streaming_kobold = preset.streaming_kobold;
$('#streaming_kobold').prop('checked', kai_settings.streaming_kobold);
}
2023-09-04 00:51:14 +02:00
if (preset.hasOwnProperty('use_default_badwordsids')) {
kai_settings.use_default_badwordsids = preset.use_default_badwordsids;
$('#use_default_badwordsids').prop('checked', kai_settings.use_default_badwordsids);
2023-09-01 00:07:04 +02:00
}
2023-07-20 19:32:15 +02:00
}
export function getKoboldGenerationData(finalPrompt, this_settings, this_amount_gen, this_max_context, isImpersonate, type) {
2023-07-20 19:32:15 +02:00
const sampler_order = kai_settings.sampler_order || this_settings.sampler_order;
let generate_data = {
2023-08-22 13:30:49 +02:00
prompt: finalPrompt,
2023-07-20 19:32:15 +02:00
gui_settings: false,
sampler_order: sampler_order,
2023-08-22 13:30:49 +02:00
max_context_length: Number(this_max_context),
2023-07-20 19:32:15 +02:00
max_length: this_amount_gen,
2023-08-22 13:30:49 +02:00
rep_pen: Number(kai_settings.rep_pen),
rep_pen_range: Number(kai_settings.rep_pen_range),
2023-07-20 19:32:15 +02:00
rep_pen_slope: kai_settings.rep_pen_slope,
2023-08-22 13:30:49 +02:00
temperature: Number(kai_settings.temp),
2023-07-20 19:32:15 +02:00
tfs: kai_settings.tfs,
top_a: kai_settings.top_a,
top_k: kai_settings.top_k,
top_p: kai_settings.top_p,
typical: kai_settings.typical,
s1: sampler_order[0],
s2: sampler_order[1],
s3: sampler_order[2],
s4: sampler_order[3],
s5: sampler_order[4],
s6: sampler_order[5],
s7: sampler_order[6],
use_world_info: false,
singleline: kai_settings.single_line,
stop_sequence: kai_flags.can_use_stop_sequence ? getStoppingStrings(isImpersonate) : undefined,
2023-09-01 00:07:04 +02:00
streaming: kai_settings.streaming_kobold && kai_flags.can_use_streaming && type !== 'quiet',
can_abort: kai_flags.can_use_streaming,
mirostat: kai_flags.can_use_mirostat ? kai_settings.mirostat : undefined,
mirostat_tau: kai_flags.can_use_mirostat ? kai_settings.mirostat_tau : undefined,
mirostat_eta: kai_flags.can_use_mirostat ? kai_settings.mirostat_eta : undefined,
use_default_badwordsids: kai_flags.can_use_default_badwordsids ? kai_settings.use_default_badwordsids : undefined,
2023-07-20 19:32:15 +02:00
};
return generate_data;
}
export async function generateKoboldWithStreaming(generate_data, signal) {
const response = await fetch('/generate', {
headers: getRequestHeaders(),
body: JSON.stringify(generate_data),
method: 'POST',
signal: signal,
});
return async function* streamData() {
const decoder = new TextDecoder();
const reader = response.body.getReader();
let getMessage = '';
let messageBuffer = "";
while (true) {
const { done, value } = await reader.read();
let response = decoder.decode(value);
let eventList = [];
// ReadableStream's buffer is not guaranteed to contain full SSE messages as they arrive in chunks
// We need to buffer chunks until we have one or more full messages (separated by double newlines)
messageBuffer += response;
eventList = messageBuffer.split("\n\n");
// Last element will be an empty string or a leftover partial message
messageBuffer = eventList.pop();
for (let event of eventList) {
for (let subEvent of event.split('\n')) {
if (subEvent.startsWith("data")) {
let data = JSON.parse(subEvent.substring(5));
getMessage += (data?.token || '');
yield getMessage;
}
}
}
if (done) {
return;
}
2023-07-20 19:32:15 +02:00
}
}
}
const sliders = [
{
name: "temp",
sliderId: "#temp",
counterId: "#temp_counter",
format: (val) => Number(val).toFixed(2),
setValue: (val) => { kai_settings.temp = Number(val); },
},
{
name: "rep_pen",
sliderId: "#rep_pen",
counterId: "#rep_pen_counter",
format: (val) => Number(val).toFixed(2),
setValue: (val) => { kai_settings.rep_pen = Number(val); },
},
{
name: "rep_pen_range",
sliderId: "#rep_pen_range",
counterId: "#rep_pen_range_counter",
format: (val) => val,
setValue: (val) => { kai_settings.rep_pen_range = Number(val); },
},
{
name: "top_p",
sliderId: "#top_p",
counterId: "#top_p_counter",
format: (val) => val,
setValue: (val) => { kai_settings.top_p = Number(val); },
},
{
name: "top_a",
sliderId: "#top_a",
counterId: "#top_a_counter",
format: (val) => val,
setValue: (val) => { kai_settings.top_a = Number(val); },
},
{
name: "top_k",
sliderId: "#top_k",
counterId: "#top_k_counter",
format: (val) => val,
setValue: (val) => { kai_settings.top_k = Number(val); },
},
{
name: "typical",
sliderId: "#typical",
counterId: "#typical_counter",
format: (val) => val,
setValue: (val) => { kai_settings.typical = Number(val); },
},
{
name: "tfs",
sliderId: "#tfs",
counterId: "#tfs_counter",
format: (val) => val,
setValue: (val) => { kai_settings.tfs = Number(val); },
},
{
name: "rep_pen_slope",
sliderId: "#rep_pen_slope",
counterId: "#rep_pen_slope_counter",
format: (val) => val,
setValue: (val) => { kai_settings.rep_pen_slope = Number(val); },
},
{
name: "sampler_order",
sliderId: "#no_op_selector",
counterId: "#no_op_selector",
format: (val) => val,
setValue: (val) => { sortItemsByOrder(val); kai_settings.sampler_order = val; },
2023-09-01 00:07:04 +02:00
},
{
name: "mirostat",
sliderId: "#mirostat_mode_kobold",
counterId: "#mirostat_mode_counter_kobold",
format: (val) => val,
setValue: (val) => { kai_settings.mirostat = Number(val); },
},
{
name: "mirostat_tau",
sliderId: "#mirostat_tau_kobold",
counterId: "#mirostat_tau_counter_kobold",
format: (val) => val,
setValue: (val) => { kai_settings.mirostat_tau = Number(val); },
},
{
name: "mirostat_eta",
sliderId: "#mirostat_eta_kobold",
counterId: "#mirostat_eta_counter_kobold",
format: (val) => val,
setValue: (val) => { kai_settings.mirostat_eta = Number(val); },
},
2023-07-20 19:32:15 +02:00
];
2023-09-01 00:07:04 +02:00
export function setKoboldFlags(version, koboldVersion) {
kai_flags.can_use_stop_sequence = canUseKoboldStopSequence(version);
kai_flags.can_use_streaming = canUseKoboldStreaming(koboldVersion);
kai_flags.can_use_tokenization = canUseKoboldTokenization(koboldVersion);
kai_flags.can_use_default_badwordsids = canUseDefaultBadwordIds(version);
2023-09-01 00:07:04 +02:00
kai_flags.can_use_mirostat = canUseMirostat(koboldVersion);
}
2023-08-22 13:30:49 +02:00
/**
* Determines if the Kobold stop sequence can be used with the given version.
* @param {string} version KoboldAI version to check.
* @returns {boolean} True if the Kobold stop sequence can be used, false otherwise.
*/
2023-09-01 00:07:04 +02:00
function canUseKoboldStopSequence(version) {
2023-07-20 19:32:15 +02:00
return (version || '0.0.0').localeCompare(MIN_STOP_SEQUENCE_VERSION, undefined, { numeric: true, sensitivity: 'base' }) > -1;
}
2023-09-01 00:07:04 +02:00
/**
* Determines if the Kobold default badword ids can be used with the given version.
* @param {string} version KoboldAI version to check.
* @returns {boolean} True if the Kobold default badword ids can be used, false otherwise.
*/
function canUseDefaultBadwordIds(version) {
return (version || '0.0.0').localeCompare(MIN_UNBAN_VERSION, undefined, { numeric: true, sensitivity: 'base' }) > -1;
}
2023-08-22 13:30:49 +02:00
/**
* Determines if the Kobold streaming API can be used with the given version.
* @param {{ result: string; version: string; }} koboldVersion KoboldAI version object.
* @returns {boolean} True if the Kobold streaming API can be used, false otherwise.
*/
2023-09-01 00:07:04 +02:00
function canUseKoboldStreaming(koboldVersion) {
2023-07-20 19:32:15 +02:00
if (koboldVersion && koboldVersion.result == 'KoboldCpp') {
return (koboldVersion.version || '0.0').localeCompare(MIN_STREAMING_KCPPVERSION, undefined, { numeric: true, sensitivity: 'base' }) > -1;
} else return false;
}
/**
* Determines if the Kobold tokenization API can be used with the given version.
* @param {{ result: string; version: string; }} koboldVersion KoboldAI version object.
* @returns {boolean} True if the Kobold tokenization API can be used, false otherwise.
*/
2023-09-01 00:07:04 +02:00
function canUseKoboldTokenization(koboldVersion) {
if (koboldVersion && koboldVersion.result == 'KoboldCpp') {
return (koboldVersion.version || '0.0').localeCompare(MIN_TOKENIZATION_KCPPVERSION, undefined, { numeric: true, sensitivity: 'base' }) > -1;
} else return false;
}
2023-09-01 00:07:04 +02:00
function canUseMirostat(koboldVersion) {
if (koboldVersion && koboldVersion.result == 'KoboldCpp') {
return (koboldVersion.version || '0.0').localeCompare(MIN_MIROSTAT_KCPPVERSION, undefined, { numeric: true, sensitivity: 'base' }) > -1;
} else return false;
}
2023-08-22 13:30:49 +02:00
/**
* Sorts the sampler items by the given order.
* @param {any[]} orderArray Sampler order array.
*/
2023-07-20 19:32:15 +02:00
function sortItemsByOrder(orderArray) {
console.debug('Preset samplers order: ' + orderArray);
const $draggableItems = $("#kobold_order");
for (let i = 0; i < orderArray.length; i++) {
const index = orderArray[i];
const $item = $draggableItems.find(`[data-id="${index}"]`).detach();
$draggableItems.append($item);
}
}
jQuery(function () {
2023-07-20 19:32:15 +02:00
sliders.forEach(slider => {
$(document).on("input", slider.sliderId, function () {
const value = $(this).val();
const formattedValue = slider.format(value);
slider.setValue(value);
$(slider.counterId).html(formattedValue);
saveSettingsDebounced();
});
});
$('#single_line').on("input", function () {
2023-09-01 00:07:04 +02:00
const value = !!$(this).prop('checked');
2023-07-20 19:32:15 +02:00
kai_settings.single_line = value;
saveSettingsDebounced();
});
$('#streaming_kobold').on("input", function () {
2023-09-01 00:07:04 +02:00
const value = !!$(this).prop('checked');
2023-07-20 19:32:15 +02:00
kai_settings.streaming_kobold = value;
saveSettingsDebounced();
});
2023-09-04 00:51:14 +02:00
$('#use_default_badwordsids').on("input", function () {
2023-09-01 00:07:04 +02:00
const value = !!$(this).prop('checked');
2023-09-04 00:51:14 +02:00
kai_settings.use_default_badwordsids = value;
2023-09-01 00:07:04 +02:00
saveSettingsDebounced();
});
2023-07-20 19:32:15 +02:00
$('#kobold_order').sortable({
delay: getSortableDelay(),
2023-07-20 19:32:15 +02:00
stop: function () {
const order = [];
$('#kobold_order').children().each(function () {
order.push($(this).data('id'));
});
kai_settings.sampler_order = order;
console.log('Samplers reordered:', kai_settings.sampler_order);
saveSettingsDebounced();
},
});
$('#samplers_order_recommended').on('click', function () {
kai_settings.sampler_order = KOBOLDCPP_ORDER;
sortItemsByOrder(kai_settings.sampler_order);
saveSettingsDebounced();
});
2023-07-20 19:32:15 +02:00
});