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

View File

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