alexion
4def357d09
Imported as a linear vendor commit so forge rebases do not replay the upstream root tree. git-subtree-dir: third_party/pi git-subtree-split: d981de1229ef899957bbe968bc8dcda02a21f477
92 lines
2.8 KiB
TypeScript
92 lines
2.8 KiB
TypeScript
import { fauxAssistantMessage } from "@earendil-works/pi-ai";
|
|
import { afterEach, describe, expect, it } from "vitest";
|
|
import { createHarness, getMessageText, getUserTexts, type Harness } from "../harness.ts";
|
|
|
|
describe("issue #7150: RPC prompt during manual compaction", () => {
|
|
const harnesses: Harness[] = [];
|
|
|
|
afterEach(() => {
|
|
while (harnesses.length > 0) {
|
|
harnesses.pop()?.cleanup();
|
|
}
|
|
});
|
|
|
|
it("rejects an RPC prompt while manual compaction is in progress", async () => {
|
|
let markCompactionStarted = () => {};
|
|
const compactionStarted = new Promise<void>((resolve) => {
|
|
markCompactionStarted = resolve;
|
|
});
|
|
let releaseCompaction = () => {};
|
|
const compactionReleased = new Promise<void>((resolve) => {
|
|
releaseCompaction = resolve;
|
|
});
|
|
|
|
const harness = await createHarness({
|
|
settings: { compaction: { keepRecentTokens: 1 } },
|
|
extensionFactories: [
|
|
(pi) => {
|
|
pi.on("session_before_compact", async (event) => {
|
|
markCompactionStarted();
|
|
await compactionReleased;
|
|
return {
|
|
compaction: {
|
|
summary: "manual compacted",
|
|
firstKeptEntryId: event.preparation.firstKeptEntryId,
|
|
tokensBefore: event.preparation.tokensBefore,
|
|
details: {},
|
|
},
|
|
};
|
|
});
|
|
},
|
|
],
|
|
});
|
|
harnesses.push(harness);
|
|
|
|
const timestamp = Date.now();
|
|
harness.sessionManager.appendMessage({
|
|
role: "user",
|
|
content: [{ type: "text", text: "old user message" }],
|
|
timestamp: timestamp - 1000,
|
|
});
|
|
harness.sessionManager.appendMessage(
|
|
fauxAssistantMessage("old assistant response", { timestamp: timestamp - 500 }),
|
|
);
|
|
harness.session.agent.state.messages = harness.sessionManager.buildSessionContext().messages;
|
|
harness.setResponses([fauxAssistantMessage("probe response")]);
|
|
|
|
const compactPromise = harness.session.compact();
|
|
await compactionStarted;
|
|
|
|
let preflightResult: boolean | undefined;
|
|
let promptError: unknown;
|
|
try {
|
|
await harness.session.prompt("PROBE-7150", {
|
|
source: "rpc",
|
|
preflightResult: (success) => {
|
|
preflightResult = success;
|
|
},
|
|
});
|
|
} catch (error) {
|
|
promptError = error;
|
|
} finally {
|
|
releaseCompaction();
|
|
await compactPromise;
|
|
}
|
|
|
|
const persistedUserTexts = harness.sessionManager
|
|
.getEntries()
|
|
.flatMap((entry) =>
|
|
entry.type === "message" && entry.message.role === "user" ? [getMessageText(entry.message)] : [],
|
|
);
|
|
|
|
expect(preflightResult).toBe(false);
|
|
expect(promptError).toEqual(
|
|
expect.objectContaining({ message: expect.stringContaining("compaction is in progress") }),
|
|
);
|
|
expect(getUserTexts(harness)).not.toContain("PROBE-7150");
|
|
expect(persistedUserTexts).not.toContain("PROBE-7150");
|
|
expect(harness.eventsOfType("agent_start")).toHaveLength(0);
|
|
expect(harness.eventsOfType("agent_settled")).toHaveLength(0);
|
|
});
|
|
});
|