fix(context-engine): mark CLI heartbeat finalization

This commit is contained in:
Josh Lehman
2026-06-05 09:19:39 -07:00
parent 599580cd96
commit 2f6da84c4b
2 changed files with 3 additions and 0 deletions

View File

@@ -175,6 +175,7 @@ describe("runPreparedCliAgent context engine lifecycle", () => {
const dispose = vi.fn(async () => {});
const contextEngine = createContextEngine({ bootstrap, afterTurn, maintain, dispose });
const context = buildPreparedContext(contextEngine);
context.params.bootstrapContextRunKind = "heartbeat";
const result = await runPreparedCliAgent(context);
expect(result.meta.agentMeta?.sessionId).toBe("external-cli-session-1");
@@ -198,6 +199,7 @@ describe("runPreparedCliAgent context engine lifecycle", () => {
sessionKey: "agent:main:main",
sessionFile: "session.jsonl",
prePromptMessageCount: 2,
isHeartbeat: true,
tokenBudget: undefined,
runtimeContext: undefined,
});

View File

@@ -270,6 +270,7 @@ async function finalizeCliContextEngineTurn(params: {
sessionIdUsed: runParams.sessionId,
sessionKey: runParams.sessionKey,
sessionFile: runParams.sessionFile,
isHeartbeat: runParams.bootstrapContextRunKind === "heartbeat",
messagesSnapshot: [...prePromptMessages, ...turnMessages],
prePromptMessageCount: prePromptMessages.length,
config: context.contextEngineConfig,