test(agents): share overflow retry compaction fixture

This commit is contained in:
Peter Steinberger
2026-02-19 08:53:56 +00:00
parent b41fd20741
commit 745068a597
3 changed files with 51 additions and 30 deletions

View File

@@ -21,3 +21,46 @@ export function makeAttemptResult(
...overrides,
};
}
type MockRunEmbeddedAttempt = {
mockResolvedValueOnce: (value: EmbeddedRunAttemptResult) => unknown;
};
type MockCompactDirect = {
mockResolvedValueOnce: (value: {
ok: true;
compacted: true;
result: {
summary: string;
firstKeptEntryId: string;
tokensBefore: number;
};
}) => unknown;
};
export function mockOverflowRetrySuccess(params: {
runEmbeddedAttempt: MockRunEmbeddedAttempt;
compactDirect: MockCompactDirect;
overflowMessage?: string;
}) {
const overflowError = new Error(
params.overflowMessage ?? "request_too_large: Request size exceeds model context window",
);
params.runEmbeddedAttempt.mockResolvedValueOnce(
makeAttemptResult({ promptError: overflowError }),
);
params.runEmbeddedAttempt.mockResolvedValueOnce(makeAttemptResult({ promptError: null }));
params.compactDirect.mockResolvedValueOnce({
ok: true,
compacted: true,
result: {
summary: "Compacted session",
firstKeptEntryId: "entry-5",
tokensBefore: 150000,
},
});
return overflowError;
}