Add seed input field for OpenAI settings #1412

This commit is contained in:
Cohee
2023-11-30 02:54:52 +02:00
parent 6abd0b4b75
commit e9ad55aef2
3 changed files with 26 additions and 0 deletions

View File

@ -229,6 +229,7 @@ const default_settings = {
squash_system_messages: false,
image_inlining: false,
bypass_status_check: false,
seed: -1,
};
const oai_settings = {
@ -282,6 +283,7 @@ const oai_settings = {
squash_system_messages: false,
image_inlining: false,
bypass_status_check: false,
seed: -1,
};
let openai_setting_names;
@ -1557,6 +1559,10 @@ async function sendOpenAIRequest(type, messages, signal) {
generate_data['stop_tokens'] = [name1 + ':', oai_settings.new_chat_prompt, oai_settings.new_group_chat_prompt];
}
if ((isOAI || isOpenRouter) && oai_settings.seed >= 0) {
generate_data['seed'] = oai_settings.seed;
}
const generate_url = '/generate_openai';
const response = await fetch(generate_url, {
method: 'POST',
@ -2415,6 +2421,7 @@ function loadOpenAISettings(data, settings) {
$('#top_k_openai').val(oai_settings.top_k_openai);
$('#top_k_counter_openai').val(Number(oai_settings.top_k_openai).toFixed(0));
$('#seed_openai').val(oai_settings.seed);
if (settings.reverse_proxy !== undefined) oai_settings.reverse_proxy = settings.reverse_proxy;
$('#openai_reverse_proxy').val(oai_settings.reverse_proxy);
@ -2594,6 +2601,7 @@ async function saveOpenAIPreset(name, settings, triggerUi = true) {
use_alt_scale: settings.use_alt_scale,
squash_system_messages: settings.squash_system_messages,
image_inlining: settings.image_inlining,
seed: settings.seed,
};
const savePresetSettings = await fetch(`/api/presets/save-openai?name=${name}`, {
@ -2953,6 +2961,7 @@ function onSettingsPresetChange() {
use_alt_scale: ['#use_alt_scale', 'use_alt_scale', true],
squash_system_messages: ['#squash_system_messages', 'squash_system_messages', true],
image_inlining: ['#openai_image_inlining', 'image_inlining', true],
seed: ['#seed_openai', 'seed', false],
};
const presetName = $('#settings_preset_openai').find(":selected").text();
@ -3768,6 +3777,11 @@ $(document).ready(async function () {
saveSettingsDebounced();
});
$('#seed_openai').on('input', function () {
oai_settings.seed = Number($(this).val());
saveSettingsDebounced();
});
$(document).on('input', '#openai_settings .autoSetHeight', function () {
resetScrollHeight($(this));
});