mirror of
https://github.com/SilenceLurker/ST-Amily2-Chat-Optimisation.git
synced 2026-06-06 15:55:50 +00:00
327 lines
13 KiB
JavaScript
327 lines
13 KiB
JavaScript
import { extension_settings, getContext } from "/scripts/extensions.js";
|
||
import { characters } from "/script.js";
|
||
import { world_names } from "/scripts/world-info.js";
|
||
import { extensionName } from "../utils/settings.js";
|
||
import { extractContentByTag, replaceContentByTag, extractFullTagBlock } from '../utils/tagProcessor.js';
|
||
import {
|
||
getCombinedWorldbookContent,
|
||
findLatestSummaryLore,
|
||
DEDICATED_LOREBOOK_NAME,
|
||
getChatIdentifier,
|
||
} from "./lore.js";
|
||
import { checkAndFixWithAPI as summarizerCheckAndFix } from './summarizer.js';
|
||
|
||
// 导入 Google 适配器和轮询管理器
|
||
import {
|
||
isGoogleEndpoint,
|
||
convertToGoogleRequest,
|
||
parseGoogleResponse,
|
||
buildGoogleApiUrl
|
||
} from '../core/utils/googleAdapter.js';
|
||
|
||
import {
|
||
intelligentPoll,
|
||
createGooglePollingTask,
|
||
progressTracker
|
||
} from '../core/utils/pollingManager.js';
|
||
|
||
|
||
let ChatCompletionService = undefined;
|
||
try {
|
||
const module = await import('/scripts/custom-request.js');
|
||
ChatCompletionService = module.ChatCompletionService;
|
||
console.log('[Amily2号-外交部] 已成功召唤“皇家信使”(ChatCompletionService)。');
|
||
} catch (e) {
|
||
console.warn("[Amily2号-外交部] 未能召唤“皇家信使”，部分高级功能（如Claw代理）将受限。请考虑更新SillyTavern版本。", e);
|
||
}
|
||
|
||
const UPDATE_CHECK_URL =
|
||
"https://raw.githubusercontent.com/Wx-2025/ST-Amily2-Chat-Optimisation/refs/heads/main/amily2_update_info.json";
|
||
|
||
const MESSAGE_BOARD_URL =
|
||
"https://raw.githubusercontent.com/Wx-2025/ST-Amily2-Chat-Optimisation/refs/heads/main/amily2_message_board.json";
|
||
|
||
export async function fetchMessageBoardContent() {
|
||
if (!MESSAGE_BOARD_URL) {
|
||
console.log('[Amily2号-内务府] 任务取消：陛下尚未配置留言板URL。');
|
||
return null;
|
||
}
|
||
try {
|
||
const response = await fetch(MESSAGE_BOARD_URL, { cache: 'no-store' });
|
||
if (!response.ok) {
|
||
throw new Error(`服务器响应异常: ${response.status}`);
|
||
}
|
||
const data = await response.json();
|
||
return data;
|
||
} catch (error) {
|
||
console.error('[Amily2号-内务府] 获取留言板内容失败:', error);
|
||
return null;
|
||
}
|
||
}
|
||
|
||
export async function checkForUpdates() {
|
||
if (!UPDATE_CHECK_URL || UPDATE_CHECK_URL.includes('YourUsername')) {
|
||
console.log('[Amily2号-外交部] 任务取消：陛下尚未配置情报来源URL。');
|
||
return null;
|
||
}
|
||
|
||
|
||
try {
|
||
console.log('[Amily2号-外交部] 已派遣使者前往云端获取最新情报...');
|
||
const response = await fetch(UPDATE_CHECK_URL, {
|
||
method: 'GET',
|
||
cache: 'no-store',
|
||
mode: 'cors'
|
||
});
|
||
|
||
|
||
|
||
if (!response.ok) {
|
||
throw new Error(`远方服务器响应异常，状态: ${response.status}`);
|
||
}
|
||
|
||
const data = await response.json();
|
||
console.log('[Amily2号-外交部] 情报已成功获取并解析。');
|
||
return data;
|
||
|
||
} catch (error) {
|
||
console.error('[Amily2号-外交部] 紧急军情：外交任务失败！', error);
|
||
return null;
|
||
}
|
||
}
|
||
|
||
let isFetchingModels = false;
|
||
export async function fetchSupportedModels() {
|
||
const settings = extension_settings[extensionName];
|
||
if (settings && settings.forceProxyForCustomApi) {
|
||
console.log('[Amily2号-使节团] 已启用“皇家密道”模式，跳过模型列表获取。请手动输入模型ID并保存。');
|
||
toastr.info("已启用手动模式，请直接输入模型ID。", "模式切换");
|
||
|
||
const $selector = $("#amily2_model");
|
||
$selector.empty().append($('<option>', { value: '', text: '手动模式已启用' }));
|
||
return []; // 直接结束任务，不再执行后续的网络请求
|
||
}
|
||
if (window.AMILY2_LOCK_MODEL_FETCHING) {
|
||
console.warn("[Amily2号-使节团] 上次任务尚未完成，本次任务取消。");
|
||
toastr.info("上次任务尚未完成，请稍后再试。", "任务排队中");
|
||
return [];
|
||
}
|
||
|
||
window.AMILY2_LOCK_MODEL_FETCHING = true;
|
||
|
||
try {
|
||
const apiUrl = $("#amily2_api_url").val().trim();
|
||
const apiKeysString = $("#amily2_api_key").val().trim();
|
||
const $button = $("#amily2_refresh_models");
|
||
const $selector = $("#amily2_model");
|
||
|
||
if (!apiUrl || !apiKeysString) {
|
||
toastr.error("陛下，请先赐予 API 地址与至少一枚 API Key。", "配置缺失");
|
||
return [];
|
||
}
|
||
|
||
$button.prop("disabled", true).html('<i class="fas fa-spinner fa-spin"></i> 加载中');
|
||
$selector.empty().append($('<option>', { value: '', text: '正在轮换使节团获取模型...' }));
|
||
|
||
const apiKeys = apiKeysString.split(',').map(k => k.trim()).filter(Boolean);
|
||
if (apiKeys.length === 0) {
|
||
toastr.error("陛下，您提供的 API Key 无效或为空。", "API Key无效");
|
||
$selector.empty().append($('<option>', { value: '', text: 'API Key无效' }));
|
||
return [];
|
||
}
|
||
|
||
const errorLog = [];
|
||
let successResults = [];
|
||
|
||
for (let i = 0; i < apiKeys.length; i++) {
|
||
const currentApiKey = apiKeys[i];
|
||
console.log(`[Amily2号-使节团] 派遣第 ${i + 1}/${apiKeys.length} 位使节 (Key: ...${currentApiKey.slice(-4)}) ...`);
|
||
|
||
try {
|
||
let modelsApiUrl;
|
||
const baseUrlObject = new URL(apiUrl);
|
||
const isGoogleAPI = isGoogleEndpoint(apiUrl);
|
||
|
||
// 处理 Google API 端点
|
||
if (isGoogleAPI) {
|
||
if (baseUrlObject.hostname.includes("generativelanguage.googleapis.com") ||
|
||
baseUrlObject.hostname.includes("ai.google.dev")) {
|
||
baseUrlObject.pathname = '/v1beta/models';
|
||
} else if (baseUrlObject.hostname.includes("aiplatform.googleapis.com")) {
|
||
baseUrlObject.pathname = '/v1beta/projects/locations/global/models';
|
||
} else {
|
||
throw new Error("无法识别的 Google API 端点");
|
||
}
|
||
modelsApiUrl = baseUrlObject.href;
|
||
}
|
||
// 处理 OpenAI 兼容端点
|
||
else {
|
||
let path = baseUrlObject.pathname;
|
||
/*
|
||
// 旧的逻辑 - 过于宽泛，会导致对Google兼容层URL错误地附加/v1
|
||
if (path.endsWith('/v1/chat/completions')) {
|
||
path = path.substring(0, path.length - '/chat/completions'.length);
|
||
} else if (path.endsWith('/v1/')) {
|
||
path = path.slice(0, -1);
|
||
} else if (!path.endsWith('/v1')) {
|
||
path = path.replace(/\/$/, '') + '/v1';
|
||
}
|
||
*/
|
||
|
||
// 新逻辑：区分处理 Google OpenAI 兼容层和通用 OpenAI 接口
|
||
if (path.toLowerCase().includes('/openai')) {
|
||
// 对于 Google 的兼容层 (e.g., /v1beta/openai)，我们假定它就是根路径
|
||
// 直接附加 /models，并确保没有尾部斜杠
|
||
path = path.replace(/\/$/, '');
|
||
} else {
|
||
// 原有的、通用的 OpenAI 兼容端点逻辑
|
||
if (path.endsWith('/v1/chat/completions')) {
|
||
path = path.substring(0, path.length - '/chat/completions'.length);
|
||
} else if (path.endsWith('/v1/')) {
|
||
path = path.slice(0, -1);
|
||
} else if (!path.endsWith('/v1')) {
|
||
path = path.replace(/\/$/, '') + '/v1';
|
||
}
|
||
}
|
||
baseUrlObject.pathname = path.replace(/\/$/, '') + '/models';
|
||
modelsApiUrl = baseUrlObject.href;
|
||
}
|
||
|
||
console.log(`[Amily2号-外交部] 使节团尝试使用地址: ${modelsApiUrl}`);
|
||
console.log(`[Amily2号-外交部] API 类型: ${isGoogleAPI ? 'Google' : 'OpenAI 兼容'}`);
|
||
|
||
const headers = {
|
||
"Content-Type": "application/json",
|
||
Accept: "application/json"
|
||
};
|
||
|
||
// Google API 认证头处理
|
||
if (isGoogleAPI) {
|
||
console.log(`[Amily2号-使节团] 使用 Google API Key: ...${currentApiKey.slice(-4)}`);
|
||
|
||
if (baseUrlObject.hostname.includes("generativelanguage.googleapis.com") ||
|
||
baseUrlObject.hostname.includes("ai.google.dev")) {
|
||
headers["X-goog-api-key"] = currentApiKey;
|
||
} else if (baseUrlObject.hostname.includes("aiplatform.googleapis.com")) {
|
||
headers["Authorization"] = `Bearer ${currentApiKey}`;
|
||
}
|
||
}
|
||
// OpenAI 兼容 API 认证头处理
|
||
else {
|
||
headers["Authorization"] = `Bearer ${currentApiKey}`;
|
||
}
|
||
|
||
// 自定义代理头
|
||
if (modelsApiUrl.includes("love.qinyan.xyz")) {
|
||
headers["X-Custom-Proxy"] = "Amily2-ChatPlugin";
|
||
headers["Origin"] = window.location.origin;
|
||
}
|
||
|
||
const response = await fetch(modelsApiUrl, {
|
||
method: "GET",
|
||
headers: headers,
|
||
mode: "cors",
|
||
credentials: "omit",
|
||
});
|
||
|
||
if (!response.ok) {
|
||
let errorBody = "";
|
||
try {
|
||
const jsonError = await response.json();
|
||
errorBody = JSON.stringify(jsonError, null, 2);
|
||
} catch {
|
||
try { errorBody = await response.text(); }
|
||
catch (e) { errorBody = "<无法提取错误正文>"; }
|
||
}
|
||
throw new Error(`API返回错误: ${response.status} ${response.statusText}\n${errorBody}`);
|
||
}
|
||
|
||
const data = await response.json();
|
||
let models = [];
|
||
|
||
// Google API 模型解析
|
||
if (isGoogleAPI) {
|
||
if (data.models && Array.isArray(data.models)) {
|
||
models = data.models.map(m => m.name);
|
||
} else if (data.data && Array.isArray(data.data)) {
|
||
models = data.data.map(m => m.name || m.id);
|
||
} else if (Array.isArray(data)) {
|
||
models = data.map(m => m.name);
|
||
} else {
|
||
throw new Error("未知的 Google 模型列表格式");
|
||
}
|
||
}
|
||
// OpenAI 兼容 API 模型解析
|
||
else {
|
||
if (Array.isArray(data)) {
|
||
models = data.map((m) => m.id || m);
|
||
} else if (data.data && Array.isArray(data.data)) {
|
||
models = data.data.map((m) => m.id);
|
||
} else if (data.models && Array.isArray(data.models)) {
|
||
models = data.models.map(m => m.id);
|
||
} else {
|
||
throw new Error("未知的模型列表格式");
|
||
}
|
||
}
|
||
|
||
// 过滤不需要的模型（嵌入/音频等）
|
||
const availableModels = models
|
||
.filter(m => typeof m === 'string')
|
||
.filter(m => !m.toLowerCase().includes("embed"))
|
||
.filter(m => !m.toLowerCase().includes("search"))
|
||
.filter(m => !m.toLowerCase().includes("similarity"))
|
||
.filter(m => !m.toLowerCase().includes("audio"))
|
||
.filter(m => !m.toLowerCase().includes("code"))
|
||
.filter(m => !m.toLowerCase().includes("whisper"));
|
||
|
||
availableModels.sort((a, b) => a.localeCompare(b));
|
||
console.log(`[Amily2号-使节团] 第 ${i + 1} 位使节成功带回 ${availableModels.length} 个模型！`);
|
||
|
||
// 合并结果并过滤重复项
|
||
successResults = [...new Set([...successResults, ...availableModels])];
|
||
successResults.sort();
|
||
|
||
if (isGoogleAPI) {
|
||
toastr.success(`成功获取 ${availableModels.length} 个 Google 模型 (使用第 ${i + 1} 个Key)`, "任务成功");
|
||
} else {
|
||
toastr.success(`成功获取 ${availableModels.length} 个模型 (使用第 ${i + 1} 个Key)`, "任务成功");
|
||
}
|
||
|
||
// 不再继续尝试，返回成功结果
|
||
break;
|
||
} catch (error) {
|
||
const errorMessage = `Key ...${currentApiKey.slice(-4)} 失败: ${error.message}`;
|
||
console.error(`[Amily2号-使节团] 第 ${i + 1} 位使节任务失败:`, error);
|
||
errorLog.push(errorMessage);
|
||
}
|
||
}
|
||
|
||
if (successResults.length > 0) {
|
||
console.log(`[Amily2号-使节团] 最终带回 ${successResults.length} 个可用模型`);
|
||
toastr.info(`所有使节团任务完成，找到 ${successResults.length} 个可用模型`, "任务总结");
|
||
return successResults;
|
||
}
|
||
|
||
toastr.error("所有使节均未能完成任务。详情请见控制台(F12)。", "外交任务失败");
|
||
console.error("[Amily2号-使节团] 失败详情汇总:\n" + errorLog.join("\n"));
|
||
return [];
|
||
|
||
} catch (error) {
|
||
console.error("[Amily2号-使节团] 全局错误:", error);
|
||
toastr.error("模型获取失败: " + error.message, "系统错误");
|
||
return [];
|
||
} finally {
|
||
window.AMILY2_LOCK_MODEL_FETCHING = false;
|
||
const $button = $("#amily2_refresh_models");
|
||
$button.prop("disabled", false).html('<i class="fas fa-sync-alt"></i> 刷新模型');
|
||
}
|
||
}
|
||
|
||
export async function checkAndFixWithAPI(latestMessage, previousMessages) {
|
||
// 将实现委托给 summarizer 模块
|
||
return await summarizerCheckAndFix(latestMessage, previousMessages);
|
||
}
|
||
|
||
//以此标记
|
||
//以此标记
|