mirror of
https://github.com/Youzini-afk/ST-Bionic-Memory-Ecology.git
synced 2026-06-14 02:40:45 +08:00
refactor(vector): extract syncVectorState controller, migrate mobile-status off index.js slicing
This commit is contained in:
@@ -29,7 +29,6 @@ const SELF_RELATIVE = "tests/index-slicing-ratchet.mjs";
|
||||
const ALLOWLIST = Object.freeze({
|
||||
"tests/graph-persistence.mjs": { maxMarkerCalls: 7, stage: "Phase 5" },
|
||||
"tests/p0-regressions.mjs": { maxMarkerCalls: 13, stage: "Phase 3" },
|
||||
"tests/mobile-status-regressions.mjs": { maxMarkerCalls: 7, stage: "Phase 1" },
|
||||
"tests/helpers/generation-recall-harness.mjs": { maxMarkerCalls: 3, stage: "Phase 4" },
|
||||
"tests/message-render-limit.mjs": { maxMarkerCalls: 4, stage: "Phase 2" },
|
||||
"tests/index-esm-entry-smoke.mjs": { maxMarkerCalls: 4, stage: "Phase 5" },
|
||||
|
||||
@@ -1,43 +1,13 @@
|
||||
import assert from "node:assert/strict";
|
||||
import fs from "node:fs/promises";
|
||||
import path from "node:path";
|
||||
import { fileURLToPath } from "node:url";
|
||||
import vm from "node:vm";
|
||||
import { onManualExtractController } from "../maintenance/extraction-controller.js";
|
||||
import { onRebuildController } from "../ui/ui-actions-controller.js";
|
||||
import { syncVectorStateController } from "../vector/vector-sync-controller.js";
|
||||
|
||||
const __dirname = path.dirname(fileURLToPath(import.meta.url));
|
||||
const indexPath = path.resolve(__dirname, "../index.js");
|
||||
const indexSource = await fs.readFile(indexPath, "utf8");
|
||||
|
||||
function extractSnippet(startMarker, endMarker) {
|
||||
const start = indexSource.indexOf(startMarker);
|
||||
const end = indexSource.indexOf(endMarker);
|
||||
if (start < 0 || end < 0 || end <= start) {
|
||||
throw new Error(`无法提取 index.js 片段: ${startMarker} -> ${endMarker}`);
|
||||
}
|
||||
return indexSource.slice(start, end).replace(/^export\s+/gm, "");
|
||||
}
|
||||
|
||||
const statusSnippet = extractSnippet(
|
||||
"function setRuntimeStatus(",
|
||||
"function notifyExtractionIssue(",
|
||||
);
|
||||
const vectorSnippet = extractSnippet(
|
||||
"async function syncVectorState({",
|
||||
"async function ensureVectorReadyIfNeeded(",
|
||||
);
|
||||
const manualExtractSnippet = extractSnippet(
|
||||
"async function onManualExtract(options = {}) {",
|
||||
"async function onReroll(",
|
||||
);
|
||||
const rebuildSnippet = extractSnippet(
|
||||
"async function onRebuild() {",
|
||||
"async function onManualCompress() {",
|
||||
);
|
||||
|
||||
// Shared status facade that mirrors index.js setRuntimeStatus / setLastXStatus
|
||||
// semantics the way controllers consume them (status setters are injected
|
||||
// dependencies). No index.js slicing — controllers are imported directly.
|
||||
function createBaseStatusContext() {
|
||||
return {
|
||||
const context = {
|
||||
console,
|
||||
Date,
|
||||
createUiStatus(text = "待命", meta = "", level = "idle") {
|
||||
@@ -58,8 +28,6 @@ function createBaseStatusContext() {
|
||||
return {};
|
||||
},
|
||||
resolveOperationalChatId(context, graph, explicitChatId = "") {
|
||||
// VM snippet calls this as a free function (no `this`); derive only from
|
||||
// arguments so it never depends on per-test getCurrentChatId closures.
|
||||
return (
|
||||
String(explicitChatId || "").trim() ||
|
||||
String(graph?.historyState?.chatId || "").trim() ||
|
||||
@@ -79,13 +47,39 @@ function createBaseStatusContext() {
|
||||
error() {},
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
function testIndexDefinesLastProcessedAssistantFloorHelper() {
|
||||
assert.match(
|
||||
indexSource,
|
||||
/function\s+getLastProcessedAssistantFloor\s*\(/,
|
||||
);
|
||||
context.setRuntimeStatus = function (text, meta, level = "info") {
|
||||
this.runtimeStatus = this.createUiStatus(text, meta, level);
|
||||
};
|
||||
context.setLastExtractionStatus = function (
|
||||
text,
|
||||
meta,
|
||||
level = "info",
|
||||
{ syncRuntime = true } = {},
|
||||
) {
|
||||
this.lastExtractionStatus = this.createUiStatus(text, meta, level);
|
||||
if (syncRuntime) this.setRuntimeStatus(text, meta, level);
|
||||
};
|
||||
context.setLastVectorStatus = function (
|
||||
text,
|
||||
meta,
|
||||
level = "info",
|
||||
{ syncRuntime = false } = {},
|
||||
) {
|
||||
this.lastVectorStatus = this.createUiStatus(text, meta, level);
|
||||
if (syncRuntime) this.setRuntimeStatus(text, meta, level);
|
||||
};
|
||||
context.setLastRecallStatus = function (
|
||||
text,
|
||||
meta,
|
||||
level = "info",
|
||||
{ syncRuntime = true } = {},
|
||||
) {
|
||||
this.lastRecallStatus = this.createUiStatus(text, meta, level);
|
||||
if (syncRuntime) this.setRuntimeStatus(text, meta, level);
|
||||
};
|
||||
|
||||
return context;
|
||||
}
|
||||
|
||||
async function testVectorSyncTerminalStateUpdatesRuntime() {
|
||||
@@ -100,6 +94,9 @@ async function testVectorSyncTerminalStateUpdatesRuntime() {
|
||||
ensureCurrentGraphRuntimeState() {
|
||||
return context.currentGraph;
|
||||
},
|
||||
getCurrentGraph() {
|
||||
return context.currentGraph;
|
||||
},
|
||||
getEmbeddingConfig() {
|
||||
return { mode: "direct" };
|
||||
},
|
||||
@@ -125,16 +122,9 @@ async function testVectorSyncTerminalStateUpdatesRuntime() {
|
||||
return false;
|
||||
},
|
||||
markVectorStateDirty() {},
|
||||
result: null,
|
||||
};
|
||||
vm.createContext(context);
|
||||
vm.runInContext(
|
||||
`${statusSnippet}\n${vectorSnippet}\nresult = { syncVectorState };`,
|
||||
context,
|
||||
{ filename: indexPath },
|
||||
);
|
||||
|
||||
const result = await context.result.syncVectorState({ force: true });
|
||||
const result = await syncVectorStateController(context, { force: true });
|
||||
assert.equal(result.stats.indexed, 12);
|
||||
assert.equal(context.lastVectorStatus.text, "向量完成");
|
||||
assert.equal(context.runtimeStatus.text, "向量完成");
|
||||
@@ -157,6 +147,15 @@ async function testManualExtractNoBatchesDoesNotStayRunning() {
|
||||
getGraphPersistenceState() {
|
||||
return { pendingPersist: false };
|
||||
},
|
||||
getCurrentGraph() {
|
||||
return context.currentGraph;
|
||||
},
|
||||
getIsExtracting() {
|
||||
return context.isExtracting;
|
||||
},
|
||||
setIsExtracting(value) {
|
||||
context.isExtracting = value;
|
||||
},
|
||||
ensureGraphMutationReady() {
|
||||
return true;
|
||||
},
|
||||
@@ -166,6 +165,9 @@ async function testManualExtractNoBatchesDoesNotStayRunning() {
|
||||
normalizeGraphRuntimeState(graph) {
|
||||
return graph;
|
||||
},
|
||||
setCurrentGraph(graph) {
|
||||
context.currentGraph = graph;
|
||||
},
|
||||
createEmptyGraph() {
|
||||
return {};
|
||||
},
|
||||
@@ -202,16 +204,9 @@ async function testManualExtractNoBatchesDoesNotStayRunning() {
|
||||
},
|
||||
onManualExtractController,
|
||||
finishStageAbortController() {},
|
||||
result: null,
|
||||
};
|
||||
vm.createContext(context);
|
||||
vm.runInContext(
|
||||
`${statusSnippet}\n${manualExtractSnippet}\nresult = { onManualExtract };`,
|
||||
context,
|
||||
{ filename: indexPath },
|
||||
);
|
||||
|
||||
await context.result.onManualExtract();
|
||||
await onManualExtractController(context, { drainAll: false });
|
||||
assert.equal(context.isExtracting, false);
|
||||
assert.equal(context.lastExtractionStatus.text, "无待提取内容");
|
||||
assert.equal(context.runtimeStatus.text, "无待提取内容");
|
||||
@@ -617,6 +612,12 @@ async function testManualRebuildSetsTerminalRuntimeStatus() {
|
||||
assert.equal(this?.__confirmHost, true);
|
||||
return true;
|
||||
},
|
||||
getCurrentGraph() {
|
||||
return context.currentGraph;
|
||||
},
|
||||
setCurrentGraph(graph) {
|
||||
context.currentGraph = graph;
|
||||
},
|
||||
ensureGraphMutationReady() {
|
||||
return true;
|
||||
},
|
||||
@@ -684,16 +685,9 @@ async function testManualRebuildSetsTerminalRuntimeStatus() {
|
||||
return await task();
|
||||
},
|
||||
onRebuildController,
|
||||
result: null,
|
||||
};
|
||||
vm.createContext(context);
|
||||
vm.runInContext(
|
||||
`${statusSnippet}\n${rebuildSnippet}\nresult = { onRebuild };`,
|
||||
context,
|
||||
{ filename: indexPath },
|
||||
);
|
||||
|
||||
await context.result.onRebuild();
|
||||
await onRebuildController(context);
|
||||
assert.equal(context.lastExtractionStatus.text, "图谱重建完成");
|
||||
assert.equal(context.runtimeStatus.text, "图谱重建完成");
|
||||
assert.equal(context.runtimeStatus.level, "success");
|
||||
@@ -704,7 +698,6 @@ async function testManualRebuildSetsTerminalRuntimeStatus() {
|
||||
assert.equal(savedNeedRefresh, false);
|
||||
}
|
||||
|
||||
testIndexDefinesLastProcessedAssistantFloorHelper();
|
||||
await testVectorSyncTerminalStateUpdatesRuntime();
|
||||
await testManualExtractNoBatchesDoesNotStayRunning();
|
||||
await testManualExtractIgnoresSupersededPendingPersistence();
|
||||
|
||||
Reference in New Issue
Block a user