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

fix: limit reroll re-extract to single latest assistant batch

Browse Source
This commit is contained in:
Youzini-afk
2026-03-29 19:13:46 +08:00
parent ab37e2111c
commit fc2e9abc72
3 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
extraction-controller.js
View File

@@ -223,7 +223,7 @@ export async function runExtractionController(runtime) {
}
}
export async function onManualExtractController(runtime) {
export async function onManualExtractController(runtime, options = {}) {
if (runtime.getIsExtracting()) {
runtime.toastr.info("记忆提取正在进行中，请稍候");
return;
@@ -271,6 +271,7 @@ export async function onManualExtractController(runtime) {
"手动提取中",
`待处理 assistant 楼层 ${pendingAssistantTurns.length} 条`,
"running",
{ syncRuntime: true, toastKind: "info", toastTitle: "ST-BME 手动提取" },
);
try {
@@ -307,6 +308,10 @@ export async function onManualExtractController(runtime) {
if (Array.isArray(batchResult.effects?.warnings)) {
warnings.push(...batchResult.effects.warnings);
}
if (options?.drainAll === false) {
break;
}
}
if (totals.batches === 0) {
@@ -521,7 +526,7 @@ export async function onRerollController(runtime, { fromFloor } = {}) {
timeOut: 2500,
});
await runtime.onManualExtract();
await runtime.onManualExtract({ drainAll: false });
runtime.refreshPanelLiveState();
return {
...rollbackResult,