taozuqin/ST-Bionic-Memory-Ecology
1
0
Fork 0
mirror of https://github.com/Youzini-afk/ST-Bionic-Memory-Ecology.git synced 2026-06-13 18:31:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(recall): resolve reroll parent user floor

Browse Source
This commit is contained in:
youzini
2026-05-31 20:14:32 +00:00
parent 01291acb2d
commit 5b5930e442
3 changed files with 109 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
runtime/generation-context.js
View File

@@ -23,6 +23,66 @@ export function classifyGenerationKind(type = "normal", params = {}) {
return "fresh";
}
export function isVisibleUserGenerationMessage(message, { index = null, chat = null, isSystemMessage = null } = {}) {
if (!message?.is_user) return false;
if (message?.extra?.isSmallSys) return false;
if (typeof isSystemMessage === "function" && isSystemMessage(message, { index, chat })) {
return false;
}
if (message?.is_system) return false;
return true;
}
export function resolveGenerationParentUserFloor(
chat,
context = {},
{ phase = "", isSystemMessage = null } = {},
) {
if (!Array.isArray(chat) || chat.length === 0) return null;
const generationType = normalizeGenerationType(context?.type || context?.generationType || "normal");
const findVisibleUserBefore = (startIndex) => {
for (let index = Math.min(chat.length - 1, Math.floor(Number(startIndex))); index >= 0; index--) {
if (isVisibleUserGenerationMessage(chat[index], { index, chat, isSystemMessage })) return index;
}
return null;
};
const findLastVisibleNonSystemIndex = () => {
for (let index = chat.length - 1; index >= 0; index--) {
const message = chat[index];
if (!message) continue;
if (message?.extra?.isSmallSys) continue;
if (typeof isSystemMessage === "function" && isSystemMessage(message, { index, chat })) continue;
if (message?.is_system) continue;
return index;
}
return null;
};
if (generationType === "swipe") {
const swipedFloor = Number(context?.swipedAssistantFloor);
if (Number.isFinite(swipedFloor)) return findVisibleUserBefore(swipedFloor - 1);
const lastVisible = findLastVisibleNonSystemIndex();
return Number.isFinite(lastVisible) ? findVisibleUserBefore(lastVisible - 1) : null;
}
if (generationType === "regenerate") {
const lastVisible = findLastVisibleNonSystemIndex();
if (!Number.isFinite(lastVisible)) return null;
if (isVisibleUserGenerationMessage(chat[lastVisible], { index: lastVisible, chat, isSystemMessage })) {
return lastVisible;
}
return findVisibleUserBefore(lastVisible - 1);
}
if (generationType === "continue") {
const lastVisible = findLastVisibleNonSystemIndex();
if (!Number.isFinite(lastVisible)) return null;
return findVisibleUserBefore(lastVisible - (chat[lastVisible]?.is_user ? 0 : 1));
}
return findVisibleUserBefore(chat.length - 1);
}
function clonePlain(value, fallback = null) {
if (!value || typeof value !== "object") return fallback;
try {