fix(agents): initialize context engines before CLI compaction

This commit is contained in:
Andy Ye
2026-05-08 11:12:01 -07:00
committed by Peter Steinberger
parent 8fdbac62f9
commit f9ecbef08f
2 changed files with 60 additions and 0 deletions

View File

@@ -3,6 +3,7 @@ import { SessionManager } from "@mariozechner/pi-coding-agent";
import type { SessionEntry } from "../../config/sessions/types.js";
import type { AgentCompactionMode } from "../../config/types.agent-defaults.js";
import type { OpenClawConfig } from "../../config/types.openclaw.js";
import { ensureContextEnginesInitialized as ensureContextEnginesInitializedImpl } from "../../context-engine/init.js";
import { resolveContextEngine as resolveContextEngineImpl } from "../../context-engine/registry.js";
import type { ContextEngine } from "../../context-engine/types.js";
import { createSubsystemLogger } from "../../logging/subsystem.js";
@@ -32,6 +33,7 @@ type SettingsManagerLike = {
};
type CliCompactionDeps = {
openSessionManager: (sessionFile: string) => SessionManagerLike;
ensureContextEnginesInitialized: () => void;
resolveContextEngine: (cfg: OpenClawConfig) => Promise<ContextEngine>;
createPreparedEmbeddedPiSettingsManager: (params: {
cwd: string;
@@ -54,6 +56,7 @@ const log = createSubsystemLogger("agents/cli-compaction");
const cliCompactionDeps: CliCompactionDeps = {
openSessionManager: (sessionFile: string) => SessionManager.open(sessionFile),
ensureContextEnginesInitialized: ensureContextEnginesInitializedImpl,
resolveContextEngine: resolveContextEngineImpl,
createPreparedEmbeddedPiSettingsManager: createPreparedEmbeddedPiSettingsManagerImpl,
applyPiAutoCompactionGuard: applyPiAutoCompactionGuardImpl,
@@ -70,6 +73,7 @@ export function setCliCompactionTestDeps(overrides: Partial<typeof cliCompaction
export function resetCliCompactionTestDeps(): void {
Object.assign(cliCompactionDeps, {
openSessionManager: (sessionFile: string) => SessionManager.open(sessionFile),
ensureContextEnginesInitialized: ensureContextEnginesInitializedImpl,
resolveContextEngine: resolveContextEngineImpl,
createPreparedEmbeddedPiSettingsManager: createPreparedEmbeddedPiSettingsManagerImpl,
applyPiAutoCompactionGuard: applyPiAutoCompactionGuardImpl,
@@ -201,6 +205,7 @@ export async function runCliTurnCompactionLifecycle(params: {
return params.sessionEntry;
}
cliCompactionDeps.ensureContextEnginesInitialized();
const contextEngine = await cliCompactionDeps.resolveContextEngine(params.cfg);
const sessionManager = cliCompactionDeps.openSessionManager(sessionFile);
const settingsManager = await cliCompactionDeps.createPreparedEmbeddedPiSettingsManager({