perf: reduce memory startup overhead

This commit is contained in:
Peter Steinberger
2026-03-21 23:30:15 +00:00
parent 80441baa15
commit cf4d301a69
12 changed files with 334 additions and 163 deletions

View File

@@ -33,8 +33,17 @@ const stubManager = {
close: vi.fn(),
};
const getMemorySearchManagerMock = vi.fn(async () => ({ manager: stubManager }));
const readAgentMemoryFileMock = vi.fn(
async (params: MemoryReadParams) => await readFileImpl(params),
);
vi.mock("../../src/memory/index.js", () => ({
getMemorySearchManager: async () => ({ manager: stubManager }),
getMemorySearchManager: getMemorySearchManagerMock,
}));
vi.mock("../../src/memory/read-file.js", () => ({
readAgentMemoryFile: readAgentMemoryFileMock,
}));
export function setMemoryBackend(next: MemoryBackend): void {
@@ -63,3 +72,11 @@ export function resetMemoryToolMockState(overrides?: {
(async (params: MemoryReadParams) => ({ text: "", path: params.relPath }));
vi.clearAllMocks();
}
export function getMemorySearchManagerMockCalls(): number {
return getMemorySearchManagerMock.mock.calls.length;
}
export function getReadAgentMemoryFileMockCalls(): number {
return readAgentMemoryFileMock.mock.calls.length;
}