mirror of
https://github.com/Wx-2025/ST-Amily2-Chat-Optimisation.git
synced 2026-06-06 12:45:51 +00:00
Update bindings.js
This commit is contained in:
GitHub
320
ui/bindings.js
320
ui/bindings.js
@@ -4,6 +4,7 @@ import { defaultSettings, extensionName, saveSettings } from "../utils/settings.
|
||||
import { pluginAuthStatus, activatePluginAuthorization, getPasswordForDate } from "../utils/auth.js";
|
||||
import { fetchModels, testApiConnection } from "../core/api.js";
|
||||
import { getJqyhApiSettings, testJqyhApiConnection, fetchJqyhModels } from '../core/api/JqyhApi.js';
|
||||
import { testConcurrentApiConnection, fetchConcurrentModels } from '../core/api/ConcurrentApi.js';
|
||||
import { safeLorebooks, safeCharLorebooks, safeLorebookEntries, isTavernHelperAvailable } from "../core/tavernhelper-compatibility.js";
|
||||
|
||||
import { setAvailableModels, populateModelDropdown, getLatestUpdateInfo } from "./state.js";
|
||||
@@ -1591,11 +1592,14 @@ function opt_saveCurrentPromptsAsPreset(panel) {
|
||||
const currentEditorPromptKey = panel.find('#amily2_opt_prompt_selector').val();
|
||||
promptCache[currentEditorPromptKey] = panel.find('#amily2_opt_prompt_editor').val();
|
||||
|
||||
const currentSettings = extension_settings[extensionName] || {};
|
||||
const newPresetData = {
|
||||
name: presetName,
|
||||
mainPrompt: promptCache.main,
|
||||
systemPrompt: promptCache.system,
|
||||
finalSystemDirective: promptCache.final_system,
|
||||
concurrentMainPrompt: currentSettings.plotOpt_concurrentMainPrompt || '',
|
||||
concurrentSystemPrompt: currentSettings.plotOpt_concurrentSystemPrompt || '',
|
||||
rateMain: parseFloat(panel.find('#amily2_opt_rate_main').val()),
|
||||
ratePersonal: parseFloat(panel.find('#amily2_opt_rate_personal').val()),
|
||||
rateErotic: parseFloat(panel.find('#amily2_opt_rate_erotic').val()),
|
||||
@@ -1702,6 +1706,8 @@ function opt_importPromptPresets(file, panel) {
|
||||
mainPrompt: preset.mainPrompt || '',
|
||||
systemPrompt: preset.systemPrompt || '',
|
||||
finalSystemDirective: preset.finalSystemDirective || '',
|
||||
concurrentMainPrompt: preset.concurrentMainPrompt || '',
|
||||
concurrentSystemPrompt: preset.concurrentSystemPrompt || '',
|
||||
rateMain: preset.rateMain ?? 1.0,
|
||||
ratePersonal: preset.ratePersonal ?? 1.0,
|
||||
rateErotic: preset.rateErotic ?? 1.0,
|
||||
@@ -1827,6 +1833,301 @@ const promptCache = {
|
||||
final_system: ''
|
||||
};
|
||||
|
||||
function bindConcurrentApiEvents() {
|
||||
const concurrentToggle = document.getElementById('amily2_plotOpt_concurrentEnabled');
|
||||
const concurrentContent = document.getElementById('amily2_concurrent_content');
|
||||
|
||||
if (!concurrentToggle || !concurrentContent) return;
|
||||
|
||||
const settings = extension_settings[extensionName] || {};
|
||||
|
||||
// Initial Load
|
||||
concurrentToggle.checked = settings.plotOpt_concurrentEnabled ?? false;
|
||||
concurrentContent.style.display = concurrentToggle.checked ? 'grid' : 'none';
|
||||
|
||||
const fields = [
|
||||
{ id: 'amily2_plotOpt_concurrentApiProvider', key: 'plotOpt_concurrentApiProvider' },
|
||||
{ id: 'amily2_plotOpt_concurrentApiUrl', key: 'plotOpt_concurrentApiUrl' },
|
||||
{ id: 'amily2_plotOpt_concurrentApiKey', key: 'plotOpt_concurrentApiKey' },
|
||||
{ id: 'amily2_plotOpt_concurrentModel', key: 'plotOpt_concurrentModel' }
|
||||
];
|
||||
|
||||
fields.forEach(field => {
|
||||
const element = document.getElementById(field.id);
|
||||
if (element) {
|
||||
element.value = settings[field.key] || '';
|
||||
}
|
||||
});
|
||||
|
||||
// Button Listeners
|
||||
const testButton = document.getElementById('amily2_plotOpt_concurrent_test_connection');
|
||||
if (testButton) {
|
||||
testButton.addEventListener('click', async () => {
|
||||
const button = $(testButton);
|
||||
const originalHtml = button.html();
|
||||
button.prop('disabled', true).html('<i class="fas fa-spinner fa-spin"></i> 测试中');
|
||||
try {
|
||||
await testConcurrentApiConnection();
|
||||
} finally {
|
||||
button.prop('disabled', false).html(originalHtml);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
const fetchButton = document.getElementById('amily2_plotOpt_concurrent_fetch_models');
|
||||
const modelInput = document.getElementById('amily2_plotOpt_concurrentModel');
|
||||
const modelSelect = document.getElementById('amily2_plotOpt_concurrentModel_select');
|
||||
|
||||
if (fetchButton && modelInput && modelSelect) {
|
||||
fetchButton.addEventListener('click', async () => {
|
||||
const button = $(fetchButton);
|
||||
const originalHtml = button.html();
|
||||
button.prop('disabled', true).html('<i class="fas fa-spinner fa-spin"></i> 获取中');
|
||||
try {
|
||||
const models = await fetchConcurrentModels();
|
||||
if (models && models.length > 0) {
|
||||
modelSelect.innerHTML = '<option value="">-- 选择一个模型 --</option>';
|
||||
models.forEach(model => {
|
||||
const option = document.createElement('option');
|
||||
option.value = model.id;
|
||||
option.textContent = model.name;
|
||||
if (model.id === modelInput.value) {
|
||||
option.selected = true;
|
||||
}
|
||||
modelSelect.appendChild(option);
|
||||
});
|
||||
modelSelect.style.display = 'block';
|
||||
modelInput.style.display = 'none';
|
||||
toastr.success(`成功获取 ${models.length} 个并发模型`, '获取模型成功');
|
||||
} else {
|
||||
toastr.warning('未获取到任何并发模型。', '获取模型');
|
||||
}
|
||||
} catch (error) {
|
||||
toastr.error(`获取并发模型失败: ${error.message}`, '获取模型失败');
|
||||
} finally {
|
||||
button.prop('disabled', false).html(originalHtml);
|
||||
}
|
||||
});
|
||||
|
||||
modelSelect.addEventListener('change', function() {
|
||||
const selectedModel = this.value;
|
||||
if (selectedModel) {
|
||||
modelInput.value = selectedModel;
|
||||
if (!extension_settings[extensionName]) extension_settings[extensionName] = {};
|
||||
extension_settings[extensionName].plotOpt_concurrentModel = selectedModel;
|
||||
saveSettingsDebounced();
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
|
||||
// Event Listeners
|
||||
concurrentToggle.addEventListener('change', function() {
|
||||
const isEnabled = this.checked;
|
||||
if (!extension_settings[extensionName]) extension_settings[extensionName] = {};
|
||||
extension_settings[extensionName].plotOpt_concurrentEnabled = isEnabled;
|
||||
saveSettingsDebounced();
|
||||
concurrentContent.style.display = isEnabled ? 'grid' : 'none';
|
||||
});
|
||||
|
||||
fields.forEach(field => {
|
||||
const element = document.getElementById(field.id);
|
||||
if (element) {
|
||||
element.addEventListener('change', function() {
|
||||
if (!extension_settings[extensionName]) extension_settings[extensionName] = {};
|
||||
extension_settings[extensionName][field.key] = this.value;
|
||||
saveSettingsDebounced();
|
||||
});
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
function bindConcurrentPromptEvents() {
|
||||
const panel = $('#sinan-prompt-settings-tab');
|
||||
if (panel.length === 0) return;
|
||||
|
||||
const selector = panel.find('#amily2_concurrent_prompt_selector');
|
||||
const editor = panel.find('#amily2_concurrent_prompt_editor');
|
||||
const resetButton = panel.find('#amily2_opt_reset_concurrent_prompt');
|
||||
|
||||
const promptMap = {
|
||||
main: 'plotOpt_concurrentMainPrompt',
|
||||
system: 'plotOpt_concurrentSystemPrompt'
|
||||
};
|
||||
|
||||
function updateConcurrentEditor() {
|
||||
const settings = extension_settings[extensionName] || {};
|
||||
const selectedKey = selector.val();
|
||||
const settingKey = promptMap[selectedKey];
|
||||
editor.val(settings[settingKey] || '');
|
||||
}
|
||||
|
||||
// Initial load
|
||||
updateConcurrentEditor();
|
||||
|
||||
// Event Listeners
|
||||
selector.on('change', updateConcurrentEditor);
|
||||
|
||||
editor.on('input', function() {
|
||||
const selectedKey = selector.val();
|
||||
const settingKey = promptMap[selectedKey];
|
||||
if (!extension_settings[extensionName]) extension_settings[extensionName] = {};
|
||||
extension_settings[extensionName][settingKey] = $(this).val();
|
||||
saveSettingsDebounced();
|
||||
});
|
||||
|
||||
resetButton.on('click', function() {
|
||||
const selectedKey = selector.val();
|
||||
const settingKey = promptMap[selectedKey];
|
||||
const defaultValue = defaultSettings[settingKey] || '';
|
||||
|
||||
if (confirm(`您确定要将 "${selector.find('option:selected').text()}" 恢复为默认值吗？`)) {
|
||||
editor.val(defaultValue);
|
||||
if (!extension_settings[extensionName]) extension_settings[extensionName] = {};
|
||||
extension_settings[extensionName][settingKey] = defaultValue;
|
||||
saveSettingsDebounced();
|
||||
toastr.success('并发提示词已成功恢复为默认值。');
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
function opt_loadConcurrentWorldbookSettings() {
|
||||
const panel = $('#amily2_plot_optimization_panel');
|
||||
if (panel.length === 0) return;
|
||||
|
||||
const settings = extension_settings[extensionName] || {};
|
||||
const enabledCheckbox = panel.find('#amily2_plotOpt_concurrentWorldbookEnabled');
|
||||
const sourceRadios = panel.find('input[name="amily2_plotOpt_concurrentWorldbook_source"]');
|
||||
const charLimitSlider = panel.find('#amily2_plotOpt_concurrentWorldbookCharLimit');
|
||||
const charLimitValue = panel.find('#amily2_plotOpt_concurrentWorldbookCharLimit_value');
|
||||
|
||||
enabledCheckbox.prop('checked', settings.plotOpt_concurrentWorldbookEnabled ?? true);
|
||||
const currentSource = settings.plotOpt_concurrentWorldbookSource || 'character';
|
||||
panel.find(`input[name="amily2_plotOpt_concurrentWorldbook_source"][value="${currentSource}"]`).prop('checked', true);
|
||||
charLimitSlider.val(settings.plotOpt_concurrentWorldbookCharLimit || 60000);
|
||||
charLimitValue.text(charLimitSlider.val());
|
||||
|
||||
// This will also trigger the visibility update
|
||||
enabledCheckbox.trigger('change');
|
||||
}
|
||||
|
||||
function bindConcurrentWorldbookEvents() {
|
||||
const panel = $('#amily2_plot_optimization_panel');
|
||||
if (panel.length === 0) return;
|
||||
|
||||
const settings = extension_settings[extensionName] || {};
|
||||
const enabledCheckbox = panel.find('#amily2_plotOpt_concurrentWorldbookEnabled');
|
||||
const contentDiv = panel.find('#amily2_concurrent_worldbook_content');
|
||||
const sourceRadios = panel.find('input[name="amily2_plotOpt_concurrentWorldbook_source"]');
|
||||
const manualSelectWrapper = panel.find('#amily2_plotOpt_concurrent_worldbook_select_wrapper');
|
||||
const refreshButton = panel.find('#amily2_plotOpt_concurrent_refresh_worldbooks');
|
||||
const bookListContainer = panel.find('#amily2_plotOpt_concurrent_worldbook_checkbox_list');
|
||||
const charLimitSlider = panel.find('#amily2_plotOpt_concurrentWorldbookCharLimit');
|
||||
const charLimitValue = panel.find('#amily2_plotOpt_concurrentWorldbookCharLimit_value');
|
||||
|
||||
function updateVisibility() {
|
||||
const isEnabled = enabledCheckbox.is(':checked');
|
||||
contentDiv.css('display', isEnabled ? 'block' : 'none');
|
||||
if (isEnabled) {
|
||||
const source = panel.find('input[name="amily2_plotOpt_concurrentWorldbook_source"]:checked').val();
|
||||
manualSelectWrapper.css('display', source === 'manual' ? 'block' : 'none');
|
||||
}
|
||||
}
|
||||
|
||||
async function loadConcurrentWorldbooks() {
|
||||
bookListContainer.html('<p class="notes">加载中...</p>');
|
||||
try {
|
||||
const lorebooks = await safeLorebooks();
|
||||
bookListContainer.empty();
|
||||
if (!lorebooks || lorebooks.length === 0) {
|
||||
bookListContainer.html('<p class="notes">未找到世界书。</p>');
|
||||
return;
|
||||
}
|
||||
const selectedBooks = settings.plotOpt_concurrentSelectedWorldbooks || [];
|
||||
const autoSelectedBooks = settings.plotOpt_concurrentAutoSelectWorldbooks || [];
|
||||
lorebooks.forEach(name => {
|
||||
const bookId = `amily2-opt-concurrent-wb-check-${name.replace(/[^a-zA-Z0-9]/g, '-')}`;
|
||||
const autoId = `amily2-opt-concurrent-wb-auto-${name.replace(/[^a-zA-Z0-9]/g, '-')}`;
|
||||
const isChecked = selectedBooks.includes(name);
|
||||
const isAuto = autoSelectedBooks.includes(name);
|
||||
const item = $(`
|
||||
<div class="amily2_opt_worldbook_list_item" style="display: flex; align-items: center; justify-content: space-between; padding-right: 5px;">
|
||||
<div style="display: flex; align-items: center;">
|
||||
<input type="checkbox" id="${bookId}" value="${name}" ${isChecked ? 'checked' : ''} style="margin-right: 5px;">
|
||||
<label for="${bookId}" style="margin-bottom: 0;">${name}</label>
|
||||
</div>
|
||||
<div style="display: flex; align-items: center;" title="开启后自动加载该世界书所有条目（包括新增）">
|
||||
<input type="checkbox" class="amily2_opt_concurrent_wb_auto_check" id="${autoId}" data-book="${name}" ${isAuto ? 'checked' : ''} style="margin-right: 5px;">
|
||||
<label for="${autoId}" style="margin-bottom: 0; font-size: 0.9em; opacity: 0.8; cursor: pointer;">全选</label>
|
||||
</div>
|
||||
</div>
|
||||
`);
|
||||
bookListContainer.append(item);
|
||||
});
|
||||
} catch (error) {
|
||||
console.error(`[${extensionName}] 加载并发世界书失败:`, error);
|
||||
bookListContainer.html('<p class="notes" style="color:red;">加载世界书列表失败。</p>');
|
||||
}
|
||||
}
|
||||
|
||||
// Initial State is now handled by opt_loadConcurrentWorldbookSettings
|
||||
updateVisibility();
|
||||
if (panel.find('input[name="amily2_plotOpt_concurrentWorldbook_source"]:checked').val() === 'manual') {
|
||||
loadConcurrentWorldbooks();
|
||||
}
|
||||
|
||||
// Event Listeners
|
||||
enabledCheckbox.on('change', function() {
|
||||
if (!extension_settings[extensionName]) extension_settings[extensionName] = {};
|
||||
extension_settings[extensionName].plotOpt_concurrentWorldbookEnabled = this.checked;
|
||||
saveSettingsDebounced();
|
||||
updateVisibility();
|
||||
});
|
||||
|
||||
sourceRadios.on('change', function() {
|
||||
if (this.checked) {
|
||||
const source = $(this).val();
|
||||
if (!extension_settings[extensionName]) extension_settings[extensionName] = {};
|
||||
extension_settings[extensionName].plotOpt_concurrentWorldbookSource = source;
|
||||
saveSettingsDebounced();
|
||||
updateVisibility();
|
||||
if (source === 'manual') {
|
||||
loadConcurrentWorldbooks();
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
refreshButton.on('click', loadConcurrentWorldbooks);
|
||||
|
||||
bookListContainer.on('change', 'input[type="checkbox"]:not(.amily2_opt_concurrent_wb_auto_check)', function() {
|
||||
const selected = [];
|
||||
bookListContainer.find('input[type="checkbox"]:not(.amily2_opt_concurrent_wb_auto_check):checked').each(function() {
|
||||
selected.push($(this).val());
|
||||
});
|
||||
if (!extension_settings[extensionName]) extension_settings[extensionName] = {};
|
||||
extension_settings[extensionName].plotOpt_concurrentSelectedWorldbooks = selected;
|
||||
saveSettingsDebounced();
|
||||
});
|
||||
|
||||
bookListContainer.on('change', '.amily2_opt_concurrent_wb_auto_check', function() {
|
||||
const autoSelected = [];
|
||||
bookListContainer.find('.amily2_opt_concurrent_wb_auto_check:checked').each(function() {
|
||||
autoSelected.push($(this).data('book'));
|
||||
});
|
||||
if (!extension_settings[extensionName]) extension_settings[extensionName] = {};
|
||||
extension_settings[extensionName].plotOpt_concurrentAutoSelectWorldbooks = autoSelected;
|
||||
saveSettingsDebounced();
|
||||
});
|
||||
|
||||
charLimitSlider.on('input', function() {
|
||||
const value = $(this).val();
|
||||
charLimitValue.text(value);
|
||||
if (!extension_settings[extensionName]) extension_settings[extensionName] = {};
|
||||
extension_settings[extensionName].plotOpt_concurrentWorldbookCharLimit = parseInt(value, 10);
|
||||
saveSettingsDebounced();
|
||||
});
|
||||
}
|
||||
|
||||
export function initializePlotOptimizationBindings() {
|
||||
const panel = $('#amily2_plot_optimization_panel');
|
||||
if (panel.length === 0 || panel.data('events-bound')) {
|
||||
@@ -1936,6 +2237,10 @@ export function initializePlotOptimizationBindings() {
|
||||
|
||||
opt_loadSettings(panel);
|
||||
bindJqyhApiEvents();
|
||||
bindConcurrentApiEvents();
|
||||
bindConcurrentPromptEvents();
|
||||
opt_loadConcurrentWorldbookSettings(); // Load settings
|
||||
bindConcurrentWorldbookEvents(); // Then bind events
|
||||
|
||||
eventSource.on(event_types.CHAT_CHANGED, () => {
|
||||
console.log(`[${extensionName}] 检测到角色/聊天切换，正在刷新剧情优化设置UI...`);
|
||||
@@ -2053,6 +2358,21 @@ export function initializePlotOptimizationBindings() {
|
||||
opt_saveSetting('plotOpt_systemPrompt', promptCache.system);
|
||||
opt_saveSetting('plotOpt_finalSystemDirective', promptCache.final_system);
|
||||
|
||||
// Also load and save concurrent prompts
|
||||
const concurrentMain = selectedPreset.concurrentMainPrompt || defaultSettings.plotOpt_concurrentMainPrompt;
|
||||
const concurrentSystem = selectedPreset.concurrentSystemPrompt || defaultSettings.plotOpt_concurrentSystemPrompt;
|
||||
opt_saveSetting('plotOpt_concurrentMainPrompt', concurrentMain);
|
||||
opt_saveSetting('plotOpt_concurrentSystemPrompt', concurrentSystem);
|
||||
|
||||
// Trigger UI update for concurrent editor
|
||||
const concurrentEditor = panel.find('#amily2_concurrent_prompt_editor');
|
||||
const concurrentSelector = panel.find('#amily2_concurrent_prompt_selector');
|
||||
if (concurrentSelector.val() === 'main') {
|
||||
concurrentEditor.val(concurrentMain);
|
||||
} else {
|
||||
concurrentEditor.val(concurrentSystem);
|
||||
}
|
||||
|
||||
panel.find('#amily2_opt_rate_main').val(selectedPreset.rateMain ?? 1.0).trigger('change');
|
||||
panel.find('#amily2_opt_rate_personal').val(selectedPreset.ratePersonal ?? 1.0).trigger('change');
|
||||
panel.find('#amily2_opt_rate_erotic').val(selectedPreset.rateErotic ?? 1.0).trigger('change');
|
||||
|
||||
Reference in New Issue
Block a user