mirror of
https://github.com/Wx-2025/ST-Amily2-Chat-Optimisation.git
synced 2026-06-06 16:15:50 +00:00
744 lines
32 KiB
JavaScript
744 lines
32 KiB
JavaScript
/**
|
||
* api-config-bindings.js — API 连接配置面板 UI 事件绑定
|
||
*
|
||
* 依赖：
|
||
* ApiProfileManager（数据层）
|
||
* ApiKeyStore（密钥存储）
|
||
*/
|
||
|
||
import { apiProfileManager, PROFILE_TYPES, SLOTS } from '../utils/config/ApiProfileManager.js';
|
||
import { apiKeyStore } from '../utils/config/api-key-store/ApiKeyStore.js';
|
||
import { configManager } from '../utils/config/ConfigManager.js';
|
||
import { getRequestHeaders, saveSettingsDebounced } from '/script.js';
|
||
import { extension_settings } from '/scripts/extensions.js';
|
||
import { extensionName } from '../utils/settings.js';
|
||
import { testApiConnection } from '../core/api.js';
|
||
import { testJqyhApiConnection } from '../core/api/JqyhApi.js';
|
||
import { testConcurrentApiConnection } from '../core/api/ConcurrentApi.js';
|
||
import { testNgmsApiConnection } from '../core/api/Ngms_api.js';
|
||
import { testNccsApiConnection } from '../core/api/NccsApi.js';
|
||
|
||
// 槽位 → 真实测试函数映射（发送聊天请求验证连接）
|
||
// plotOpt 槽位同时服务剧情优化和 JQYH（互斥），根据启用状态选择测试函数
|
||
const SLOT_TEST_FNS = {
|
||
main: testApiConnection,
|
||
plotOpt: () => {
|
||
const s = extension_settings[extensionName] || {};
|
||
return s.jqyhEnabled ? testJqyhApiConnection() : testApiConnection();
|
||
},
|
||
plotOptConc: testConcurrentApiConnection,
|
||
ngms: testNgmsApiConnection,
|
||
nccs: testNccsApiConnection,
|
||
};
|
||
|
||
// 槽位 → 功能总开关映射
|
||
// key : extension_settings[extensionName] 中的设置键
|
||
// checkbox : 原面板中对应 checkbox 的 DOM 选择器（用于双向同步）
|
||
const SLOT_TOGGLES = {
|
||
plotOptConc: { key: 'plotOpt_concurrentEnabled', checkbox: '#amily2_plotOpt_concurrentEnabled' },
|
||
ngms: { key: 'ngmsEnabled', checkbox: '#amily2_ngms_enabled' },
|
||
nccs: { key: 'nccsEnabled', checkbox: '#nccs-api-enabled' },
|
||
cwb: { key: 'cwb_master_enabled', checkbox: '#cwb_master_enabled-checkbox' },
|
||
};
|
||
|
||
// ── 状态 ─────────────────────────────────────────────────────────────────────
|
||
|
||
let _editingId = null; // 当前编辑的 Profile ID（null = 新建）
|
||
let _currentFilter = 'all'; // 当前类型筛选
|
||
|
||
// ── 入口：绑定整个面板 ────────────────────────────────────────────────────────
|
||
|
||
export function bindApiConfigPanel(container) {
|
||
const $c = $(container);
|
||
|
||
// 存储模式
|
||
_bindStorageMode($c);
|
||
|
||
// 类型筛选
|
||
$c.on('click', '.amily2_profile_type_filter', function () {
|
||
$c.find('.amily2_profile_type_filter').removeClass('active');
|
||
$(this).addClass('active');
|
||
_currentFilter = $(this).data('type');
|
||
renderProfileList($c);
|
||
});
|
||
|
||
// 新建 Profile
|
||
$c.find('#amily2_add_profile').on('click', () => openModal($c, null));
|
||
|
||
// 弹窗：类型切换时显示/隐藏专有参数
|
||
$c.find('#amily2_pf_type').on('change', function () {
|
||
_switchParamSections($c, $(this).val());
|
||
});
|
||
|
||
// 弹窗：接口类型切换（Google 自动填 URL）
|
||
$c.find('#amily2_pf_provider').on('change', function () {
|
||
_handleProviderChange($c, $(this).val());
|
||
});
|
||
|
||
// 弹窗：获取模型列表
|
||
$c.find('#amily2_pf_fetch_models').on('click', () => _fetchModels($c));
|
||
|
||
// 弹窗：测试连接
|
||
$c.find('#amily2_pf_test_conn').on('click', () => _testConnection($c));
|
||
|
||
// 表单：取消
|
||
$c.find('#amily2_profile_modal_cancel').on('click', () => closeModal($c));
|
||
|
||
// 弹窗：保存
|
||
$c.find('#amily2_profile_modal_save').on('click', () => saveProfile($c));
|
||
|
||
// 初始渲染
|
||
renderProfileList($c);
|
||
renderSlotAssignments($c);
|
||
}
|
||
|
||
// ── 存储模式 ──────────────────────────────────────────────────────────────────
|
||
|
||
function _bindStorageMode($c) {
|
||
const $select = $c.find('#amily2_keystore_mode');
|
||
const $cloud = $c.find('#amily2_cloud_key_section');
|
||
const $note = $c.find('#amily2_keystore_mode_note');
|
||
const $importInput = $c.find('#amily2_import_key_bundle_input');
|
||
|
||
const MODE_NOTES = {
|
||
local: '本地存储：API Key 仅存于本设备浏览器，绝不上传服务端。换设备需重新填写。',
|
||
cloud: '加密云同步：API Key 经 RSA+AES 混合加密后随设置同步。私钥仅留在本设备，服务商只能看到密文。',
|
||
};
|
||
|
||
// 初始状态
|
||
const currentMode = apiKeyStore.getMode();
|
||
$select.val(currentMode);
|
||
$cloud.toggle(currentMode === 'cloud');
|
||
$note.text(MODE_NOTES[currentMode]);
|
||
if (currentMode === 'cloud') _refreshFingerprint($c);
|
||
|
||
// 切换模式
|
||
$select.on('change', async function () {
|
||
const newMode = $(this).val();
|
||
const confirmed = newMode === 'cloud'
|
||
? confirm('切换到加密云同步模式：\n将自动为本设备生成 RSA 密钥对，现有 Key 会重新加密存储。\n\n确认切换？')
|
||
: confirm('切换回本地存储模式：\n已加密的 Key 将解密迁移至本地，云端密文会被清除。\n\n确认切换？');
|
||
|
||
if (!confirmed) {
|
||
$select.val(apiKeyStore.getMode());
|
||
return;
|
||
}
|
||
|
||
try {
|
||
await apiKeyStore.setMode(newMode);
|
||
if (newMode === 'cloud') {
|
||
await configManager.syncSensitiveCache({ force: true });
|
||
}
|
||
$cloud.toggle(newMode === 'cloud');
|
||
$note.text(MODE_NOTES[newMode]);
|
||
if (newMode === 'cloud') _refreshFingerprint($c);
|
||
toastr.success(`已切换为${newMode === 'cloud' ? '加密云同步' : '本地存储'}模式。`);
|
||
} catch (e) {
|
||
console.error('[ApiConfig] 模式切换失败:', e);
|
||
toastr.error('模式切换失败，请查看控制台。');
|
||
$select.val(apiKeyStore.getMode());
|
||
}
|
||
});
|
||
|
||
// 重新生成密钥对
|
||
$c.find('#amily2_generate_keypair').on('click', async () => {
|
||
if (!confirm('重新生成密钥对后，所有已加密的 API Key 将失效，需要逐一重新输入。\n\n确认重新生成？')) return;
|
||
await apiKeyStore.generateKeyPair();
|
||
_refreshFingerprint($c);
|
||
toastr.warning('新密钥对已生成，请重新输入各 Profile 的 API Key。');
|
||
});
|
||
|
||
$c.find('#amily2_export_key_bundle').on('click', async () => {
|
||
try {
|
||
const bundle = await apiKeyStore.exportPrivateKeyBundle();
|
||
_downloadJson(
|
||
`amily2-keystore-${_timestampForFilename()}.json`,
|
||
bundle
|
||
);
|
||
toastr.success('私钥包已导出，请妥善保管。');
|
||
} catch (e) {
|
||
console.error('[ApiConfig] 导出私钥包失败:', e);
|
||
toastr.error(e.message || '导出私钥包失败。');
|
||
}
|
||
});
|
||
|
||
$c.find('#amily2_import_key_bundle').on('click', () => {
|
||
$importInput.val('');
|
||
$importInput.trigger('click');
|
||
});
|
||
|
||
$importInput.on('change', async function () {
|
||
const file = this.files?.[0];
|
||
if (!file) return;
|
||
|
||
try {
|
||
const text = await file.text();
|
||
await apiKeyStore.importPrivateKeyBundle(text);
|
||
await configManager.syncSensitiveCache({ force: true });
|
||
await _refreshFingerprint($c);
|
||
toastr.success('私钥包导入成功，已尝试恢复云同步的 API Key 缓存。');
|
||
} catch (e) {
|
||
console.error('[ApiConfig] 导入私钥包失败:', e);
|
||
toastr.error(e.message || '导入私钥包失败。');
|
||
} finally {
|
||
$importInput.val('');
|
||
}
|
||
});
|
||
}
|
||
|
||
async function _refreshFingerprint($c) {
|
||
const fp = await apiKeyStore.getPublicKeyInfo();
|
||
$c.find('#amily2_keypair_fingerprint').text(fp);
|
||
}
|
||
|
||
function _downloadJson(filename, data) {
|
||
const blob = new Blob([JSON.stringify(data, null, 2)], { type: 'application/json' });
|
||
const url = URL.createObjectURL(blob);
|
||
const a = document.createElement('a');
|
||
a.href = url;
|
||
a.download = filename;
|
||
document.body.appendChild(a);
|
||
a.click();
|
||
a.remove();
|
||
URL.revokeObjectURL(url);
|
||
}
|
||
|
||
function _timestampForFilename() {
|
||
const now = new Date();
|
||
const pad = n => String(n).padStart(2, '0');
|
||
return `${now.getFullYear()}${pad(now.getMonth() + 1)}${pad(now.getDate())}-${pad(now.getHours())}${pad(now.getMinutes())}${pad(now.getSeconds())}`;
|
||
}
|
||
|
||
// ── Profile 列表渲染 ──────────────────────────────────────────────────────────
|
||
|
||
export function renderProfileList($c) {
|
||
const $list = $c.find('#amily2_profile_list');
|
||
const profiles = apiProfileManager.getProfiles(
|
||
_currentFilter === 'all' ? undefined : _currentFilter
|
||
);
|
||
|
||
if (profiles.length === 0) {
|
||
$list.html('<div class="amily2_profile_empty" style="color:var(--SmartThemeQuoteColor);text-align:center;padding:20px;">暂无连接配置，点击「新建配置」添加。</div>');
|
||
return;
|
||
}
|
||
|
||
const TYPE_BADGE_COLOR = {
|
||
chat: 'var(--SmartThemeBodyColor)',
|
||
embedding: '#7eb8f7',
|
||
rerank: '#f7b07e',
|
||
};
|
||
|
||
const html = profiles.map(p => {
|
||
const typeInfo = PROFILE_TYPES[p.type];
|
||
const badgeStyle = `background:${TYPE_BADGE_COLOR[p.type]}22; color:${TYPE_BADGE_COLOR[p.type]}; border:1px solid ${TYPE_BADGE_COLOR[p.type]}55; border-radius:4px; padding:1px 6px; font-size:0.78em;`;
|
||
return `
|
||
<div class="amily2_profile_card" data-id="${p.id}" style="
|
||
display:flex; align-items:center; gap:10px;
|
||
padding:8px 12px;
|
||
background:var(--black10a);
|
||
border:1px solid var(--SmartThemeBorderColor);
|
||
border-radius:6px;">
|
||
<i class="fas ${typeInfo.icon}" style="width:16px; color:var(--SmartThemeQuoteColor);"></i>
|
||
<div style="flex:1; min-width:0;">
|
||
<div style="font-weight:600; white-space:nowrap; overflow:hidden; text-overflow:ellipsis;">${_escapeHtml(p.name)}</div>
|
||
<div style="font-size:0.82em; color:var(--SmartThemeQuoteColor); margin-top:2px;">
|
||
<span style="${badgeStyle}"><i class="fas ${typeInfo.icon}"></i> ${typeInfo.label}</span>
|
||
<span style="margin-left:6px;">${_escapeHtml(p.model || '（未设置模型）')}</span>
|
||
${p.apiUrl ? `<span style="margin-left:6px; opacity:0.7;">${_escapeHtml(_truncateUrl(p.apiUrl))}</span>` : ''}
|
||
</div>
|
||
</div>
|
||
<div style="display:flex; gap:4px; flex-shrink:0;">
|
||
<button class="menu_button small_button interactable amily2_edit_profile" data-id="${p.id}" title="编辑">
|
||
<i class="fas fa-edit"></i>
|
||
</button>
|
||
<button class="menu_button small_button secondary interactable amily2_delete_profile" data-id="${p.id}" title="删除">
|
||
<i class="fas fa-trash"></i>
|
||
</button>
|
||
</div>
|
||
</div>`;
|
||
}).join('');
|
||
|
||
$list.html(html);
|
||
|
||
// 编辑 / 删除事件
|
||
$list.find('.amily2_edit_profile').on('click', function () {
|
||
openModal($c, $(this).data('id'));
|
||
});
|
||
$list.find('.amily2_delete_profile').on('click', function () {
|
||
const id = $(this).data('id');
|
||
const name = apiProfileManager.getProfile(id)?.name || id;
|
||
if (!confirm(`确认删除连接配置「${name}」？\n此操作不可撤销，存储的 API Key 将同时清除。`)) return;
|
||
apiProfileManager.deleteProfile(id);
|
||
renderProfileList($c);
|
||
renderSlotAssignments($c);
|
||
toastr.success(`已删除配置「${name}」。`);
|
||
});
|
||
}
|
||
|
||
// ── 功能槽分配渲染 ────────────────────────────────────────────────────────────
|
||
|
||
export function renderSlotAssignments($c) {
|
||
const $slots = $c.find('#amily2_slot_assignments');
|
||
|
||
const settings = extension_settings[extensionName] || {};
|
||
|
||
const rows = Object.entries(SLOTS).map(([slot, slotInfo]) => {
|
||
const profiles = apiProfileManager.getProfiles(slotInfo.type);
|
||
const assigned = apiProfileManager.getAssignment(slot) || '';
|
||
const typeInfo = PROFILE_TYPES[slotInfo.type];
|
||
const toggle = SLOT_TOGGLES[slot];
|
||
|
||
const options = [
|
||
`<option value="">— 未分配 —</option>`,
|
||
...profiles.map(p =>
|
||
`<option value="${p.id}" ${p.id === assigned ? 'selected' : ''}>${_escapeHtml(p.name)}</option>`
|
||
),
|
||
].join('');
|
||
|
||
// 功能开关（仅有映射的槽位显示）
|
||
const toggleHtml = toggle
|
||
? `<label class="toggle-switch" style="flex-shrink:0;" title="启用/禁用此功能">
|
||
<input type="checkbox" class="amily2_slot_toggle" data-slot="${slot}" ${settings[toggle.key] ? 'checked' : ''} />
|
||
<span class="slider"></span>
|
||
</label>`
|
||
: '';
|
||
|
||
return `
|
||
<div style="display:flex; align-items:center; gap:8px; padding:4px 0;">
|
||
${toggleHtml}
|
||
<span style="width:140px; flex-shrink:0; font-size:0.9em;">${slotInfo.label}</span>
|
||
<span style="color:var(--SmartThemeQuoteColor); font-size:0.78em; width:70px; flex-shrink:0;">
|
||
<i class="fas ${typeInfo.icon}"></i> ${typeInfo.label}
|
||
</span>
|
||
<select class="text_pole amily2_slot_select" data-slot="${slot}" style="flex:1;">
|
||
${options}
|
||
</select>
|
||
<button class="menu_button small_button interactable amily2_slot_test" data-slot="${slot}"
|
||
title="测试此槽位的连接" style="flex-shrink:0; ${assigned ? '' : 'opacity:0.4; pointer-events:none;'}">
|
||
<i class="fas fa-plug"></i>
|
||
</button>
|
||
</div>`;
|
||
}).join('');
|
||
|
||
$slots.html(rows);
|
||
|
||
$slots.find('.amily2_slot_select').on('change', function () {
|
||
const slot = $(this).data('slot');
|
||
const id = $(this).val() || null;
|
||
if (!apiProfileManager.setAssignment(slot, id)) {
|
||
toastr.error('类型不匹配，分配失败。');
|
||
renderSlotAssignments($c);
|
||
return;
|
||
}
|
||
document.dispatchEvent(new CustomEvent('amily2:slotAssigned', { detail: { slot } }));
|
||
// 刷新行以更新测试按钮状态
|
||
renderSlotAssignments($c);
|
||
});
|
||
|
||
// 槽位快捷测试按钮（调用各模块真实测试函数，发送聊天请求验证连接）
|
||
$slots.find('.amily2_slot_test').on('click', async function () {
|
||
const slot = $(this).data('slot');
|
||
const $btn = $(this).prop('disabled', true);
|
||
$btn.html('<i class="fas fa-spinner fa-spin"></i>');
|
||
|
||
try {
|
||
const testFn = SLOT_TEST_FNS[slot];
|
||
if (!testFn) {
|
||
toastr.warning('该槽位暂不支持快捷测试。', slot);
|
||
return;
|
||
}
|
||
const profile = await apiProfileManager.getAssignedProfile(slot);
|
||
if (!profile) {
|
||
toastr.warning('该槽位未分配配置。', slot);
|
||
return;
|
||
}
|
||
// 测试函数内部会显示 toastr 结果
|
||
await testFn();
|
||
} catch (e) {
|
||
toastr.error(`测试失败：${e.message}`, slot);
|
||
} finally {
|
||
$btn.prop('disabled', false).html('<i class="fas fa-plug"></i>');
|
||
}
|
||
});
|
||
|
||
// 功能总开关：同步 extension_settings + 原面板 checkbox
|
||
$slots.find('.amily2_slot_toggle').on('change', function () {
|
||
const slot = $(this).data('slot');
|
||
const toggle = SLOT_TOGGLES[slot];
|
||
if (!toggle) return;
|
||
|
||
const checked = this.checked;
|
||
const s = extension_settings[extensionName];
|
||
if (s) s[toggle.key] = checked;
|
||
|
||
// 同步原面板的 checkbox（保持一致）
|
||
const origCb = document.querySelector(toggle.checkbox);
|
||
if (origCb && origCb.checked !== checked) {
|
||
origCb.checked = checked;
|
||
origCb.dispatchEvent(new Event('change', { bubbles: true }));
|
||
}
|
||
|
||
saveSettingsDebounced();
|
||
});
|
||
}
|
||
|
||
// ── 弹窗操作 ──────────────────────────────────────────────────────────────────
|
||
|
||
async function openModal($c, id) {
|
||
_editingId = id;
|
||
|
||
if (id) {
|
||
// 编辑模式
|
||
const p = apiProfileManager.getProfile(id);
|
||
if (!p) return;
|
||
$c.find('#amily2_profile_modal_title').text('编辑连接配置');
|
||
$c.find('#amily2_profile_form_icon').attr('class', 'fas fa-edit');
|
||
$c.find('#amily2_pf_type').val(p.type).prop('disabled', true); // 不允许修改类型
|
||
$c.find('#amily2_pf_name').val(p.name);
|
||
$c.find('#amily2_pf_provider').val(p.provider);
|
||
$c.find('#amily2_pf_url').val(p.apiUrl);
|
||
$c.find('#amily2_pf_key').val(''); // Key 不回显
|
||
$c.find('#amily2_pf_model').val(p.model);
|
||
|
||
if (p.type === 'chat') {
|
||
$c.find('#amily2_pf_max_tokens').val(p.maxTokens);
|
||
$c.find('#amily2_pf_temperature').val(p.temperature);
|
||
$c.find('#amily2_pf_fake_stream').prop('checked', p.fakeStream ?? false);
|
||
} else if (p.type === 'embedding') {
|
||
$c.find('#amily2_pf_dimensions').val(p.dimensions ?? '');
|
||
$c.find('#amily2_pf_encoding_format').val(p.encodingFormat);
|
||
} else if (p.type === 'rerank') {
|
||
$c.find('#amily2_pf_top_n').val(p.topN);
|
||
$c.find('#amily2_pf_return_documents').prop('checked', p.returnDocuments);
|
||
}
|
||
_switchParamSections($c, p.type);
|
||
_handleProviderChange($c, p.provider);
|
||
} else {
|
||
// 新建模式
|
||
$c.find('#amily2_profile_modal_title').text('新建连接配置');
|
||
$c.find('#amily2_profile_form_icon').attr('class', 'fas fa-plus');
|
||
$c.find('#amily2_pf_type').val('chat').prop('disabled', false);
|
||
$c.find('#amily2_pf_name, #amily2_pf_url, #amily2_pf_key, #amily2_pf_model').val('');
|
||
$c.find('#amily2_pf_provider').val('openai');
|
||
_handleProviderChange($c, 'openai');
|
||
$c.find('#amily2_pf_max_tokens').val(65500);
|
||
$c.find('#amily2_pf_temperature').val(1.0);
|
||
$c.find('#amily2_pf_fake_stream').prop('checked', false);
|
||
$c.find('#amily2_pf_dimensions').val('');
|
||
$c.find('#amily2_pf_encoding_format').val('float');
|
||
$c.find('#amily2_pf_top_n').val(5);
|
||
$c.find('#amily2_pf_return_documents').prop('checked', false);
|
||
_switchParamSections($c, 'chat');
|
||
}
|
||
|
||
// 清空上次测试结果，重置模型选择器为手动输入状态
|
||
$c.find('#amily2_pf_test_result').text('');
|
||
$c.find('#amily2_pf_model_select').hide().empty();
|
||
$c.find('#amily2_pf_model').show();
|
||
|
||
const $details = $c.find('#amily2_profile_form_details');
|
||
$details.prop('open', true);
|
||
$details[0].scrollIntoView({ behavior: 'smooth', block: 'nearest' });
|
||
}
|
||
|
||
function closeModal($c) {
|
||
$c.find('#amily2_profile_form_details').prop('open', false);
|
||
$c.find('#amily2_pf_type').prop('disabled', false);
|
||
_editingId = null;
|
||
}
|
||
|
||
async function saveProfile($c) {
|
||
const type = $c.find('#amily2_pf_type').val();
|
||
const name = $c.find('#amily2_pf_name').val().trim();
|
||
const provider = $c.find('#amily2_pf_provider').val();
|
||
const apiUrl = $c.find('#amily2_pf_url').val().trim();
|
||
const apiKey = $c.find('#amily2_pf_key').val();
|
||
const $sel = $c.find('#amily2_pf_model_select');
|
||
const model = ($sel.is(':visible') ? $sel.val() : $c.find('#amily2_pf_model').val()).trim();
|
||
|
||
if (!name) { toastr.warning('请填写配置名称。'); return; }
|
||
|
||
const data = { type, name, provider, apiUrl, model };
|
||
|
||
if (type === 'chat') {
|
||
data.maxTokens = parseInt($c.find('#amily2_pf_max_tokens').val(), 10) || 65500;
|
||
data.temperature = parseFloat($c.find('#amily2_pf_temperature').val()) || 1.0;
|
||
data.fakeStream = $c.find('#amily2_pf_fake_stream').prop('checked');
|
||
} else if (type === 'embedding') {
|
||
const dim = $c.find('#amily2_pf_dimensions').val();
|
||
data.dimensions = dim ? parseInt(dim, 10) : null;
|
||
data.encodingFormat = $c.find('#amily2_pf_encoding_format').val();
|
||
} else if (type === 'rerank') {
|
||
data.topN = parseInt($c.find('#amily2_pf_top_n').val(), 10) || 5;
|
||
data.returnDocuments = $c.find('#amily2_pf_return_documents').is(':checked');
|
||
}
|
||
|
||
const $btn = $c.find('#amily2_profile_modal_save').prop('disabled', true);
|
||
|
||
try {
|
||
let profileId;
|
||
if (_editingId) {
|
||
apiProfileManager.updateProfile(_editingId, data);
|
||
profileId = _editingId;
|
||
} else {
|
||
profileId = apiProfileManager.createProfile(data);
|
||
}
|
||
|
||
// 保存 Key（非空才写入）
|
||
if (apiKey) {
|
||
await apiProfileManager.setKey(profileId, apiKey);
|
||
}
|
||
|
||
closeModal($c);
|
||
renderProfileList($c);
|
||
renderSlotAssignments($c);
|
||
toastr.success(`配置「${name}」已保存。`);
|
||
} catch (e) {
|
||
console.error('[ApiConfig] 保存 Profile 失败:', e);
|
||
toastr.error('保存失败，请查看控制台。');
|
||
} finally {
|
||
$btn.prop('disabled', false);
|
||
}
|
||
}
|
||
|
||
// ── 获取模型 / 测试连接 ───────────────────────────────────────────────────────
|
||
|
||
async function _fetchModels($c) {
|
||
const apiUrl = $c.find('#amily2_pf_url').val().trim();
|
||
const provider = $c.find('#amily2_pf_provider').val();
|
||
|
||
// 编辑模式下 Key 不回显，字段为空时从 ApiKeyStore 读取已存储的 Key
|
||
let apiKey = $c.find('#amily2_pf_key').val().trim();
|
||
if (!apiKey && _editingId) {
|
||
apiKey = await apiProfileManager.getKey(_editingId) ?? '';
|
||
}
|
||
|
||
if (!apiUrl) { toastr.warning('请先填写 API 地址。'); return; }
|
||
|
||
const $btn = $c.find('#amily2_pf_fetch_models').prop('disabled', true);
|
||
$btn.html('<i class="fas fa-spinner fa-spin"></i> 获取中...');
|
||
|
||
try {
|
||
let models;
|
||
|
||
if (provider === 'google') {
|
||
// Google 用原生 API，Key 通过 x-goog-api-key 头传递避免 URL 泄露
|
||
if (!apiKey) { toastr.warning('请先填写 Google API Key。'); return; }
|
||
const resp = await fetch(
|
||
'https://generativelanguage.googleapis.com/v1beta/models',
|
||
{ headers: { 'x-goog-api-key': apiKey } }
|
||
);
|
||
if (!resp.ok) {
|
||
const status = resp.status;
|
||
toastr.error(status === 400 ? '获取失败：API Key 格式错误。'
|
||
: status === 403 ? '获取失败：API Key 无效或无权限。'
|
||
: `获取失败：HTTP ${status}`);
|
||
return;
|
||
}
|
||
const data = await resp.json();
|
||
// 只保留支持文本生成的模型
|
||
models = (data.models ?? [])
|
||
.filter(m => m.supportedGenerationMethods?.some(
|
||
method => ['generateContent', 'embedContent'].includes(method)
|
||
))
|
||
.map(m => m.name.replace(/^models\//, ''));
|
||
} else {
|
||
// OpenAI 兼容接口 — 通过 ST 后端代理，规避 CORS
|
||
const resp = await fetch('/api/backends/chat-completions/status', {
|
||
method: 'POST',
|
||
headers: { ...getRequestHeaders(), 'Content-Type': 'application/json' },
|
||
body: JSON.stringify({
|
||
reverse_proxy: apiUrl,
|
||
proxy_password: apiKey,
|
||
chat_completion_source: 'openai',
|
||
}),
|
||
});
|
||
if (!resp.ok) {
|
||
const status = resp.status;
|
||
if (status === 401 || status === 403) {
|
||
toastr.error('获取失败：API Key 无效或无权限。');
|
||
} else if (status === 404) {
|
||
toastr.warning('该接口不支持模型列表查询，请手动填写模型 ID。');
|
||
} else {
|
||
toastr.error(`获取失败：HTTP ${status}`);
|
||
}
|
||
return;
|
||
}
|
||
const rawData = await resp.json();
|
||
// ST 返回原始数组或包含 data/models 字段的对象
|
||
const rawList = Array.isArray(rawData) ? rawData : (rawData.data ?? rawData.models ?? []);
|
||
const list = Array.isArray(rawList) ? rawList : [];
|
||
models = list.map(m => m.id ?? m.name ?? m).filter(m => typeof m === 'string' && m);
|
||
}
|
||
|
||
if (models.length === 0) {
|
||
toastr.warning('未获取到模型列表，请手动填写。');
|
||
return;
|
||
}
|
||
|
||
models.sort((a, b) => a.localeCompare(b));
|
||
|
||
const currentVal = $c.find('#amily2_pf_model').val().trim();
|
||
const $sel = $c.find('#amily2_pf_model_select');
|
||
$sel.html(models.map(m => `<option value="${_escapeHtml(m)}">${_escapeHtml(m)}</option>`).join(''));
|
||
if (currentVal && models.includes(currentVal)) $sel.val(currentVal);
|
||
$c.find('#amily2_pf_model').hide();
|
||
$sel.show();
|
||
|
||
toastr.success(`已获取 ${models.length} 个可用模型。`);
|
||
} catch (e) {
|
||
toastr.error(`获取失败：${e.message}`);
|
||
} finally {
|
||
$btn.prop('disabled', false).html('<i class="fas fa-list"></i> 获取');
|
||
}
|
||
}
|
||
|
||
async function _testConnection($c) {
|
||
const apiUrl = $c.find('#amily2_pf_url').val().trim();
|
||
const provider = $c.find('#amily2_pf_provider').val();
|
||
|
||
// 编辑模式下 Key 不回显，字段为空时从 ApiKeyStore 读取已存储的 Key
|
||
let apiKey = $c.find('#amily2_pf_key').val().trim();
|
||
if (!apiKey && _editingId) {
|
||
apiKey = await apiProfileManager.getKey(_editingId) ?? '';
|
||
}
|
||
|
||
if (!apiUrl) { toastr.warning('请先填写 API 地址。'); return; }
|
||
|
||
const $btn = $c.find('#amily2_pf_test_conn').prop('disabled', true);
|
||
const $result = $c.find('#amily2_pf_test_result').text('测试中…').css('color', 'var(--SmartThemeQuoteColor)');
|
||
$btn.html('<i class="fas fa-spinner fa-spin"></i> 测试中...');
|
||
|
||
try {
|
||
if (provider === 'google') {
|
||
// Google 用原生 models 端点测试
|
||
if (!apiKey) {
|
||
$result.text('请填写 API Key').css('color', 'var(--warning-color)');
|
||
return;
|
||
}
|
||
const resp = await fetch(
|
||
'https://generativelanguage.googleapis.com/v1beta/models',
|
||
{ headers: { 'x-goog-api-key': apiKey } }
|
||
);
|
||
if (resp.ok) {
|
||
const data = await resp.json();
|
||
const count = (data.models ?? []).length;
|
||
$result.text(`连接成功${count ? `，${count} 个可用模型` : ''}`).css('color', 'var(--green)');
|
||
toastr.success('Google AI Studio 连接测试通过！');
|
||
} else {
|
||
const status = resp.status;
|
||
const msg = status === 400 ? 'API Key 格式错误'
|
||
: status === 403 ? 'API Key 无效或无权限'
|
||
: `HTTP ${status}`;
|
||
$result.text(`失败：${msg}`).css('color', 'var(--warning-color)');
|
||
toastr.error(`测试失败：${msg}`);
|
||
}
|
||
return;
|
||
}
|
||
|
||
// OpenAI 兼容接口 — 通过 ST 后端代理，规避 CORS
|
||
const modelsResp = await fetch('/api/backends/chat-completions/status', {
|
||
method: 'POST',
|
||
headers: { ...getRequestHeaders(), 'Content-Type': 'application/json' },
|
||
body: JSON.stringify({
|
||
reverse_proxy: apiUrl,
|
||
proxy_password: apiKey,
|
||
chat_completion_source: 'openai',
|
||
}),
|
||
});
|
||
|
||
if (modelsResp.ok) {
|
||
const rawData = await modelsResp.json();
|
||
const rawList = Array.isArray(rawData) ? rawData : (rawData.data ?? rawData.models ?? []);
|
||
const list = Array.isArray(rawList) ? rawList : [];
|
||
const count = list.length;
|
||
|
||
// chat 类型额外发一次假补全，验证 completion 端点也能正常鉴权
|
||
const type = $c.find('#amily2_pf_type').val();
|
||
const $sel = $c.find('#amily2_pf_model_select');
|
||
const model = ($sel.is(':visible') ? $sel.val() : $c.find('#amily2_pf_model').val()).trim();
|
||
|
||
if (type === 'chat' && model) {
|
||
$result.text('模型列表 ✓，正在验证补全端点…').css('color', 'var(--SmartThemeQuoteColor)');
|
||
const genResp = await fetch('/api/backends/chat-completions/generate', {
|
||
method: 'POST',
|
||
headers: { ...getRequestHeaders(), 'Content-Type': 'application/json' },
|
||
body: JSON.stringify({
|
||
reverse_proxy: apiUrl,
|
||
proxy_password: apiKey,
|
||
chat_completion_source: 'openai',
|
||
model,
|
||
messages: [{ role: 'user', content: 'Hi' }],
|
||
max_tokens: 1,
|
||
stream: false,
|
||
}),
|
||
});
|
||
if (!genResp.ok) {
|
||
const genErr = await genResp.json().catch(() => ({}));
|
||
const genMsg = genErr?.error?.message || `补全端点返回 HTTP ${genResp.status}`;
|
||
$result.text(`模型列表 ✓，补全失败：${genMsg}`).css('color', 'var(--warning-color)');
|
||
toastr.warning(`补全端点测试失败：${genMsg}`);
|
||
return;
|
||
}
|
||
}
|
||
|
||
$result.text(`连接成功${count ? `，${count} 个可用模型` : ''}`).css('color', 'var(--green)');
|
||
toastr.success('连接测试通过！');
|
||
return;
|
||
}
|
||
|
||
const status = modelsResp.status;
|
||
const errBody = await modelsResp.json().catch(() => ({}));
|
||
const msg = errBody?.error?.message
|
||
|| (status === 401 || status === 403 ? 'API Key 无效或无权限'
|
||
: status === 404 ? '接口地址不存在'
|
||
: `HTTP ${status}`);
|
||
$result.text(`失败：${msg}`).css('color', 'var(--warning-color)');
|
||
toastr.error(`测试失败：${msg}`);
|
||
} catch (e) {
|
||
$result.text(`无法连接：${e.message}`).css('color', 'var(--warning-color)');
|
||
toastr.error(`连接失败：${e.message}`);
|
||
} finally {
|
||
$btn.prop('disabled', false).html('<i class="fas fa-plug"></i> 测试连接');
|
||
}
|
||
}
|
||
|
||
// ── Provider 切换 ─────────────────────────────────────────────────────────────
|
||
|
||
const GOOGLE_API_BASE = 'https://generativelanguage.googleapis.com/v1beta/openai';
|
||
|
||
function _handleProviderChange($c, provider) {
|
||
const isGoogle = provider === 'google';
|
||
$c.find('#amily2_pf_url_row').toggle(!isGoogle);
|
||
$c.find('#amily2_pf_google_note').toggle(isGoogle);
|
||
|
||
if (isGoogle) {
|
||
$c.find('#amily2_pf_url').val(GOOGLE_API_BASE);
|
||
}
|
||
}
|
||
|
||
// ── 内部工具 ──────────────────────────────────────────────────────────────────
|
||
|
||
function _switchParamSections($c, type) {
|
||
$c.find('#amily2_pf_chat_params').toggle(type === 'chat');
|
||
$c.find('#amily2_pf_embedding_params').toggle(type === 'embedding');
|
||
$c.find('#amily2_pf_rerank_params').toggle(type === 'rerank');
|
||
}
|
||
|
||
function _truncateUrl(url) {
|
||
try {
|
||
const u = new URL(url);
|
||
return u.host + (u.pathname.length > 1 ? u.pathname : '');
|
||
} catch {
|
||
return url.slice(0, 30);
|
||
}
|
||
}
|
||
|
||
function _escapeHtml(str) {
|
||
return String(str)
|
||
.replace(/&/g, '&')
|
||
.replace(/</g, '<')
|
||
.replace(/>/g, '>')
|
||
.replace(/"/g, '"');
|
||
}
|