mirror of
https://github.com/SilenceLurker/ST-Amily2-Chat-Optimisation.git
synced 2026-06-06 13:35:51 +00:00
226 lines
7.5 KiB
JavaScript
226 lines
7.5 KiB
JavaScript
import { getContext, extension_settings } from "/scripts/extensions.js";
|
||
import { saveChatConditional, reloadCurrentChat } from "/script.js";
|
||
import { extensionName } from "../utils/settings.js";
|
||
import { SlashCommand } from "/scripts/slash-commands/SlashCommand.js";
|
||
import { SlashCommandParser } from "/scripts/slash-commands/SlashCommandParser.js";
|
||
import { checkAndFixWithAPI } from "./api.js";
|
||
import { amilyHelper } from './tavern-helper/main.js';
|
||
|
||
async function checkLatestMessage() {
|
||
const context = getContext();
|
||
const chat = context.chat || [];
|
||
|
||
if (!chat || chat.length === 0) {
|
||
console.log("[Amily2-命令检查器] 没有聊天记录。");
|
||
return { message: null, previousMessages: [] };
|
||
}
|
||
|
||
const latestMessage = chat[chat.length - 1];
|
||
|
||
console.log("[Amily2-命令检查器] 正在侦测消息:", {
|
||
isUser: latestMessage.is_user,
|
||
messagePreview: latestMessage.mes?.substring(0, 50) + "...",
|
||
});
|
||
|
||
if (latestMessage.is_user) {
|
||
console.log("[Amily2-命令检查器] 目标为用户消息，跳过。");
|
||
return { message: latestMessage, previousMessages: [] };
|
||
}
|
||
|
||
const settings = extension_settings[extensionName];
|
||
const contextCount = settings.contextMessages || 2;
|
||
const startIndex = Math.max(0, chat.length - contextCount - 1);
|
||
const previousMessages = chat.slice(startIndex, chat.length - 1);
|
||
|
||
console.log("[Amily2-命令检查器] 已获取上下文消息:", {
|
||
count: previousMessages.length,
|
||
});
|
||
|
||
return { message: latestMessage, previousMessages };
|
||
}
|
||
|
||
async function checkCommand() {
|
||
const settings = extension_settings[extensionName];
|
||
if (!settings.apiUrl) {
|
||
toastr.error("请先配置API URL", "命令检查器");
|
||
return "";
|
||
}
|
||
const checkResult = await checkLatestMessage();
|
||
if (!checkResult.message || checkResult.message.is_user) {
|
||
toastr.info("最新消息是用户消息，无需检查", "命令检查器");
|
||
return "";
|
||
}
|
||
toastr.info("正在使用API检查回复...", "命令检查器");
|
||
const result = await checkAndFixWithAPI(
|
||
checkResult.message,
|
||
checkResult.previousMessages,
|
||
);
|
||
if (
|
||
result &&
|
||
result.optimizedContent &&
|
||
result.optimizedContent !== checkResult.message.mes
|
||
) {
|
||
toastr.warning("检测到问题，建议使用修复功能", "命令检查器");
|
||
} else {
|
||
toastr.success("未检测到问题", "命令检查器");
|
||
}
|
||
return "";
|
||
}
|
||
|
||
|
||
export async function fixCommand() {
|
||
const settings = extension_settings[extensionName];
|
||
if (!settings.apiUrl) {
|
||
toastr.error("请先配置API URL", "命令检查器");
|
||
return "";
|
||
}
|
||
const context = getContext();
|
||
const chat = context.chat;
|
||
if (!chat || chat.length === 0) {
|
||
toastr.info("没有可修复的消息", "命令检查器");
|
||
return "";
|
||
}
|
||
const latestMessage = chat[chat.length - 1];
|
||
if (latestMessage.is_user) {
|
||
toastr.info("最新消息是用户消息，无需修复", "命令检查器");
|
||
return "";
|
||
}
|
||
const contextCount = settings.contextMessages || 2;
|
||
const startIndex = Math.max(0, chat.length - 1 - contextCount);
|
||
const previousMessages = chat.slice(startIndex, chat.length - 1);
|
||
toastr.info("正在检查并修复回复...", "命令检查器");
|
||
const result = await checkAndFixWithAPI(latestMessage, previousMessages);
|
||
if (
|
||
result &&
|
||
result.optimizedContent &&
|
||
result.optimizedContent !== latestMessage.mes
|
||
) {
|
||
const messageId = chat.length - 1;
|
||
await amilyHelper.setChatMessage(
|
||
{ message: result.optimizedContent },
|
||
messageId,
|
||
{ refresh: 'display_and_render_current' }
|
||
);
|
||
toastr.success("回复已修复", "命令检查器");
|
||
} else {
|
||
toastr.info("未检测到需要修复的问题", "命令检查器");
|
||
}
|
||
return "";
|
||
}
|
||
|
||
export async function testReplyChecker() {
|
||
const settings = extension_settings[extensionName];
|
||
if (!settings.apiUrl) {
|
||
toastr.error("请先配置API URL", "命令检查器");
|
||
return "";
|
||
}
|
||
const context = getContext();
|
||
const chat = context.chat;
|
||
if (!chat || chat.length < 2) {
|
||
toastr.warning("需要至少2条消息才能测试", "命令检查器");
|
||
return "";
|
||
}
|
||
let testMessage = null;
|
||
for (let i = chat.length - 2; i >= 0; i--) {
|
||
if (!chat[i].is_user) {
|
||
testMessage = chat[i].mes;
|
||
break;
|
||
}
|
||
}
|
||
if (!testMessage) {
|
||
toastr.warning("没有找到可用于测试的AI消息", "命令检查器");
|
||
return "";
|
||
}
|
||
const lastMessage = chat[chat.length - 1];
|
||
if (lastMessage.is_user) {
|
||
toastr.warning("最后一条消息是用户消息，无法测试", "命令检查器");
|
||
return "";
|
||
}
|
||
const originalMessage = lastMessage.mes;
|
||
lastMessage.mes = testMessage + "\n\n" + testMessage;
|
||
toastr.info("正在使用API测试检测功能...", "命令检查器");
|
||
const contextCount = settings.contextMessages || 2;
|
||
const startIndex = Math.max(0, chat.length - contextCount - 1);
|
||
const previousMessages = chat.slice(startIndex, chat.length - 1);
|
||
const result = await checkAndFixWithAPI(lastMessage, previousMessages);
|
||
lastMessage.mes = originalMessage;
|
||
if (
|
||
result &&
|
||
result.optimizedContent &&
|
||
result.optimizedContent !== testMessage + "\n\n" + testMessage
|
||
) {
|
||
toastr.success("测试成功！API检测到重复内容并提供了修复建议", "命令检查器");
|
||
} else {
|
||
toastr.warning(
|
||
"测试结果：API未检测到问题，请检查API配置或提示词",
|
||
"命令检查器",
|
||
);
|
||
}
|
||
return "";
|
||
}
|
||
|
||
async function triggerSendButton() {
|
||
// 模拟点击发送按钮
|
||
const sendButton = document.getElementById('send_but');
|
||
if (sendButton) {
|
||
sendButton.click();
|
||
console.log("[Amily2-触发器] 已触发发送按钮");
|
||
return "";
|
||
} else {
|
||
console.warn("[Amily2-触发器] 未找到发送按钮");
|
||
toastr.warning("未找到发送按钮", "触发器");
|
||
return "";
|
||
}
|
||
}
|
||
|
||
export async function registerSlashCommands() {
|
||
try {
|
||
if (
|
||
typeof SlashCommand === "undefined" ||
|
||
typeof SlashCommandParser === "undefined"
|
||
) {
|
||
console.error(
|
||
"[Amily2] 致命错误：SlashCommand 或 SlashCommandParser 模块未能加载。",
|
||
);
|
||
return;
|
||
}
|
||
SlashCommandParser.addCommandObject(
|
||
SlashCommand.fromProps({
|
||
name: "check-reply",
|
||
callback: checkCommand,
|
||
helpString: "检查最新的AI回复是否有问题",
|
||
}),
|
||
);
|
||
console.log("[Amily2-新诏] /check-reply 命令已成功颁布。");
|
||
|
||
SlashCommandParser.addCommandObject(
|
||
SlashCommand.fromProps({
|
||
name: "fix-reply",
|
||
callback: fixCommand,
|
||
helpString: "修复最新的AI回复中的问题",
|
||
}),
|
||
);
|
||
console.log("[Amily2-新诏] /fix-reply 命令已成功颁布。");
|
||
|
||
SlashCommandParser.addCommandObject(
|
||
SlashCommand.fromProps({
|
||
name: "test-reply-checker",
|
||
callback: testReplyChecker,
|
||
helpString: "测试聊天回复检查器功能",
|
||
}),
|
||
);
|
||
console.log("[Amily2-新诏] /test-reply-checker 命令已成功颁布。");
|
||
|
||
SlashCommandParser.addCommandObject(
|
||
SlashCommand.fromProps({
|
||
name: "trigger",
|
||
callback: triggerSendButton,
|
||
helpString: "触发发送按钮 (用于自动发送消息)",
|
||
}),
|
||
);
|
||
console.log("[Amily2-新诏] /trigger 命令已成功颁布。");
|
||
} catch (e) {
|
||
console.error("[Amily2] 命令注册过程中发生意外错误:", e);
|
||
}
|
||
}
|