@xopcai/xopc 0.0.82 → 0.0.84

package/dist/src/agent/embedded/session-runner.js

@@ -1,8 +1,8 @@
 import { createLogger } from "../../utils/logger/index.js";
 import { init_logger } from "../../utils/logger.js";
-import { guardSessionManager } from "./session-tool-result-guard-wrapper.js";
+import { guardSessionManager } from "./session-tool-result-guard.js";
 import { prepareSessionManagerForRun } from "./session-manager-init.js";
-import { prewarmSessionFile } from "./session-manager-cache.js";
+import { defaultSessionManagerCache } from "./session-manager-cache.js";
 import { applyXopcProviderApiKey, createEmbeddedAuthStorage } from "./xopc-auth-storage.js";
 import { wrapStreamFnForXopcExtensions } from "./xopc-stream-bridge.js";
 import { xopcToolsToDefinitions } from "./xopc-tools-bridge.js";
@@ -25,13 +25,6 @@ function buildEmbeddedRunnerFingerprint(input) {
 		input.thinkingLevel
 	].join("");
 }
-const pool = /* @__PURE__ */ new Map();
-let stats = {
-	acquires: 0,
-	reuses: 0,
-	creates: 0,
-	evictions: 0
-};
 function isEmbeddedSessionRunnerEnabled() {
 	const raw = process.env.XOPC_SESSION_RUNNER?.trim().toLowerCase();
 	if (raw === "0" || raw === "false" || raw === "off") return false;
@@ -43,163 +36,201 @@ function getEmbeddedSessionRunnerIdleTtlMs() {
 	const parsed = Number.parseInt(raw, 10);
 	return Number.isFinite(parsed) && parsed > 0 ? parsed : DEFAULT_IDLE_TTL_MS;
 }
-function getEmbeddedSessionRunnerStats() {
-	return {
-		...stats,
-		pooled: pool.size
-	};
+function createEmbeddedSettingsManager(cwd) {
+	const sm = SettingsManager.inMemory({ compaction: { enabled: false } });
+	sm.setCompactionEnabled(false);
+	return sm;
 }
-function resetEmbeddedSessionRunnerForTest() {
-	for (const entry of pool.values()) clearIdleTimer(entry);
-	pool.clear();
+/**
+* Owns the per-session pool of pi `AgentSession` runners. The class is the supported,
+* injectable owner; {@link defaultEmbeddedSessionRunnerPool} keeps the historic
+* module-level free functions working until every caller is migrated to DI.
+*/
+var EmbeddedSessionRunnerPool = class {
+	pool = /* @__PURE__ */ new Map();
+	cache;
+	isEnabledFn;
+	getIdleTtlMsFn;
 	stats = {
 		acquires: 0,
 		reuses: 0,
 		creates: 0,
 		evictions: 0
 	};
-}
-function clearIdleTimer(entry) {
-	if (entry.idleTimer) {
-		clearTimeout(entry.idleTimer);
-		entry.idleTimer = null;
+	constructor(opts = {}) {
+		this.cache = opts.sessionManagerCache ?? defaultSessionManagerCache;
+		this.isEnabledFn = opts.isEnabled ?? isEmbeddedSessionRunnerEnabled;
+		this.getIdleTtlMsFn = opts.getIdleTtlMs ?? getEmbeddedSessionRunnerIdleTtlMs;
 	}
+	getStats() {
+		return {
+			...this.stats,
+			pooled: this.pool.size
+		};
+	}
+	resetForTest() {
+		for (const entry of this.pool.values()) this.clearIdleTimer(entry);
+		this.pool.clear();
+		this.stats = {
+			acquires: 0,
+			reuses: 0,
+			creates: 0,
+			evictions: 0
+		};
+	}
+	evict(sessionKey, reason = "explicit") {
+		const entry = this.pool.get(sessionKey);
+		if (!entry) return;
+		this.disposePooledRunner(sessionKey, entry, reason);
+	}
+	evictAll(reason = "dispose_all") {
+		for (const sessionKey of [...this.pool.keys()]) this.evict(sessionKey, reason);
+	}
+	async acquire(params) {
+		this.stats.acquires += 1;
+		const fingerprint = buildEmbeddedRunnerFingerprint({
+			sessionFile: params.sessionFile,
+			workspaceDir: params.workspaceDir,
+			modelRef: params.modelRef,
+			toolNames: params.tools.map((t) => t.name),
+			systemPrompt: params.systemPrompt,
+			thinkingLevel: params.thinkingLevel ?? "medium"
+		});
+		const reuseEnabled = this.isEnabledFn();
+		const existing = this.pool.get(params.sessionKey);
+		let entry;
+		let reused = false;
+		if (reuseEnabled && existing && existing.fingerprint === fingerprint) {
+			this.clearIdleTimer(existing);
+			entry = existing;
+			entry.lastUsedAt = Date.now();
+			reused = true;
+			this.stats.reuses += 1;
+			applySystemPromptOverrideToSession(entry.session, params.systemPrompt);
+			entry.session.agent.streamFn = entry.baseStreamFn;
+			log.debug({ sessionKey: params.sessionKey }, "Reusing pooled embedded session runner");
+		} else {
+			if (existing) this.disposePooledRunner(params.sessionKey, existing, "fingerprint_mismatch");
+			entry = await this.createPooledRunner(params);
+			this.pool.set(params.sessionKey, entry);
+			this.stats.creates += 1;
+			log.debug({ sessionKey: params.sessionKey }, "Created embedded session runner");
+		}
+		return {
+			session: entry.session,
+			piSm: entry.piSm,
+			reused,
+			release: () => {
+				if (!this.isEnabledFn()) {
+					this.disposePooledRunner(params.sessionKey, entry, "runner_disabled");
+					return;
+				}
+				entry.lastUsedAt = Date.now();
+				this.scheduleIdleEviction(params.sessionKey, entry);
+			}
+		};
+	}
+	clearIdleTimer(entry) {
+		if (entry.idleTimer) {
+			clearTimeout(entry.idleTimer);
+			entry.idleTimer = null;
+		}
+	}
+	scheduleIdleEviction(sessionKey, entry) {
+		this.clearIdleTimer(entry);
+		const ttlMs = this.getIdleTtlMsFn();
+		entry.idleTimer = setTimeout(() => {
+			if (this.pool.get(sessionKey) === entry) this.disposePooledRunner(sessionKey, entry, "idle_ttl");
+		}, ttlMs);
+		entry.idleTimer.unref?.();
+	}
+	disposePooledRunner(sessionKey, entry, reason) {
+		this.clearIdleTimer(entry);
+		this.pool.delete(sessionKey);
+		this.stats.evictions += 1;
+		try {
+			entry.piSm.flushPendingToolResults?.();
+		} catch {}
+		log.debug({
+			sessionKey,
+			reason
+		}, "Embedded session runner evicted");
+	}
+	async createPooledRunner(params) {
+		const { sessionKey, sessionId, sessionFile, sessionsDir, hadSessionFile, workspaceDir, model, thinkingLevel, tools, systemPrompt } = params;
+		await this.cache.prewarm(sessionFile);
+		const settingsManager = createEmbeddedSettingsManager(workspaceDir);
+		const piSm = guardSessionManager(SessionManager.open(sessionFile, sessionsDir, workspaceDir), {
+			sessionKey,
+			contextWindowTokens: model.contextWindow ?? 128e3
+		});
+		await prepareSessionManagerForRun({
+			sessionManager: piSm,
+			sessionFile,
+			hadSessionFile,
+			sessionId,
+			cwd: workspaceDir
+		});
+		const toolDefs = xopcToolsToDefinitions(tools);
+		const toolNames = tools.map((t) => t.name);
+		const authStorage = createEmbeddedAuthStorage();
+		applyXopcProviderApiKey(authStorage, model.provider);
+		const resourceLoader = new DefaultResourceLoader({
+			cwd: workspaceDir,
+			agentDir: getAgentDir(),
+			settingsManager,
+			noContextFiles: true
+		});
+		await resourceLoader.reload();
+		const { session } = await createAgentSession({
+			cwd: workspaceDir,
+			model,
+			thinkingLevel: thinkingLevel ?? "medium",
+			sessionManager: piSm,
+			settingsManager,
+			authStorage,
+			resourceLoader,
+			noTools: "builtin",
+			customTools: toolDefs,
+			tools: toolNames
+		});
+		applySystemPromptOverrideToSession(session, systemPrompt);
+		const baseStreamFn = wrapStreamFnForXopcExtensions(session.agent.streamFn);
+		session.agent.streamFn = baseStreamFn;
+		return {
+			sessionKey,
+			fingerprint: buildEmbeddedRunnerFingerprint({
+				sessionFile,
+				workspaceDir,
+				modelRef: params.modelRef,
+				toolNames,
+				systemPrompt,
+				thinkingLevel: thinkingLevel ?? "medium"
+			}),
+			session,
+			piSm,
+			settingsManager,
+			baseStreamFn,
+			lastUsedAt: Date.now(),
+			idleTimer: null
+		};
+	}
+};
+const defaultEmbeddedSessionRunnerPool = new EmbeddedSessionRunnerPool();
+function getEmbeddedSessionRunnerStats() {
+	return defaultEmbeddedSessionRunnerPool.getStats();
 }
-function scheduleIdleEviction(sessionKey, entry) {
-	clearIdleTimer(entry);
-	const ttlMs = getEmbeddedSessionRunnerIdleTtlMs();
-	entry.idleTimer = setTimeout(() => {
-		if (pool.get(sessionKey) === entry) disposePooledRunner(sessionKey, entry, "idle_ttl");
-	}, ttlMs);
-	entry.idleTimer.unref?.();
-}
-function disposePooledRunner(sessionKey, entry, reason) {
-	clearIdleTimer(entry);
-	pool.delete(sessionKey);
-	stats.evictions += 1;
-	try {
-		entry.piSm.flushPendingToolResults?.();
-	} catch {}
-	log.debug({
-		sessionKey,
-		reason
-	}, "Embedded session runner evicted");
+function resetEmbeddedSessionRunnerForTest() {
+	defaultEmbeddedSessionRunnerPool.resetForTest();
 }
 function evictEmbeddedSessionRunner(sessionKey, reason = "explicit") {
-	const entry = pool.get(sessionKey);
-	if (!entry) return;
-	disposePooledRunner(sessionKey, entry, reason);
+	defaultEmbeddedSessionRunnerPool.evict(sessionKey, reason);
 }
 function evictAllEmbeddedSessionRunners(reason = "dispose_all") {
-	for (const sessionKey of [...pool.keys()]) evictEmbeddedSessionRunner(sessionKey, reason);
-}
-function createEmbeddedSettingsManager(cwd) {
-	const sm = SettingsManager.inMemory({ compaction: { enabled: false } });
-	sm.setCompactionEnabled(false);
-	return sm;
+	defaultEmbeddedSessionRunnerPool.evictAll(reason);
 }
-async function createPooledRunner(params) {
-	const { sessionKey, sessionId, sessionFile, sessionsDir, hadSessionFile, workspaceDir, model, thinkingLevel, tools, systemPrompt } = params;
-	await prewarmSessionFile(sessionFile);
-	const settingsManager = createEmbeddedSettingsManager(workspaceDir);
-	const piSm = guardSessionManager(SessionManager.open(sessionFile, sessionsDir, workspaceDir), {
-		sessionKey,
-		contextWindowTokens: model.contextWindow ?? 128e3
-	});
-	await prepareSessionManagerForRun({
-		sessionManager: piSm,
-		sessionFile,
-		hadSessionFile,
-		sessionId,
-		cwd: workspaceDir
-	});
-	const toolDefs = xopcToolsToDefinitions(tools);
-	const toolNames = tools.map((t) => t.name);
-	const authStorage = createEmbeddedAuthStorage();
-	applyXopcProviderApiKey(authStorage, model.provider);
-	const resourceLoader = new DefaultResourceLoader({
-		cwd: workspaceDir,
-		agentDir: getAgentDir(),
-		settingsManager,
-		noContextFiles: true
-	});
-	await resourceLoader.reload();
-	const { session } = await createAgentSession({
-		cwd: workspaceDir,
-		model,
-		thinkingLevel: thinkingLevel ?? "medium",
-		sessionManager: piSm,
-		settingsManager,
-		authStorage,
-		resourceLoader,
-		noTools: "builtin",
-		customTools: toolDefs,
-		tools: toolNames
-	});
-	applySystemPromptOverrideToSession(session, systemPrompt);
-	const baseStreamFn = wrapStreamFnForXopcExtensions(session.agent.streamFn);
-	session.agent.streamFn = baseStreamFn;
-	return {
-		sessionKey,
-		fingerprint: buildEmbeddedRunnerFingerprint({
-			sessionFile,
-			workspaceDir,
-			modelRef: params.modelRef,
-			toolNames,
-			systemPrompt,
-			thinkingLevel: thinkingLevel ?? "medium"
-		}),
-		session,
-		piSm,
-		settingsManager,
-		baseStreamFn,
-		lastUsedAt: Date.now(),
-		idleTimer: null
-	};
-}
-async function acquireEmbeddedSessionRunner(params) {
-	stats.acquires += 1;
-	const fingerprint = buildEmbeddedRunnerFingerprint({
-		sessionFile: params.sessionFile,
-		workspaceDir: params.workspaceDir,
-		modelRef: params.modelRef,
-		toolNames: params.tools.map((t) => t.name),
-		systemPrompt: params.systemPrompt,
-		thinkingLevel: params.thinkingLevel ?? "medium"
-	});
-	const reuseEnabled = isEmbeddedSessionRunnerEnabled();
-	const existing = pool.get(params.sessionKey);
-	let entry;
-	let reused = false;
-	if (reuseEnabled && existing && existing.fingerprint === fingerprint) {
-		clearIdleTimer(existing);
-		entry = existing;
-		entry.lastUsedAt = Date.now();
-		reused = true;
-		stats.reuses += 1;
-		applySystemPromptOverrideToSession(entry.session, params.systemPrompt);
-		entry.session.agent.streamFn = entry.baseStreamFn;
-		log.debug({ sessionKey: params.sessionKey }, "Reusing pooled embedded session runner");
-	} else {
-		if (existing) disposePooledRunner(params.sessionKey, existing, "fingerprint_mismatch");
-		entry = await createPooledRunner(params);
-		pool.set(params.sessionKey, entry);
-		stats.creates += 1;
-		log.debug({ sessionKey: params.sessionKey }, "Created embedded session runner");
-	}
-	return {
-		session: entry.session,
-		piSm: entry.piSm,
-		reused,
-		release: () => {
-			if (!isEmbeddedSessionRunnerEnabled()) {
-				disposePooledRunner(params.sessionKey, entry, "runner_disabled");
-				return;
-			}
-			entry.lastUsedAt = Date.now();
-			scheduleIdleEviction(params.sessionKey, entry);
-		}
-	};
+function acquireEmbeddedSessionRunner(params) {
+	return defaultEmbeddedSessionRunnerPool.acquire(params);
 }
 /** Resolve transcript path inputs used by both runner acquire and turn execution. */
 async function resolveEmbeddedTranscriptInputs(sessionStore, sessionKey) {
@@ -212,6 +243,6 @@ async function resolveEmbeddedTranscriptInputs(sessionStore, sessionKey) {
 	};
 }
 //#endregion
-export { acquireEmbeddedSessionRunner, buildEmbeddedRunnerFingerprint, evictAllEmbeddedSessionRunners, evictEmbeddedSessionRunner, getEmbeddedSessionRunnerIdleTtlMs, getEmbeddedSessionRunnerStats, isEmbeddedSessionRunnerEnabled, resetEmbeddedSessionRunnerForTest, resolveEmbeddedTranscriptInputs };
+export { EmbeddedSessionRunnerPool, acquireEmbeddedSessionRunner, buildEmbeddedRunnerFingerprint, defaultEmbeddedSessionRunnerPool, evictAllEmbeddedSessionRunners, evictEmbeddedSessionRunner, getEmbeddedSessionRunnerIdleTtlMs, getEmbeddedSessionRunnerStats, isEmbeddedSessionRunnerEnabled, resetEmbeddedSessionRunnerForTest, resolveEmbeddedTranscriptInputs };
 
 //# sourceMappingURL=session-runner.js.map

package/dist/src/agent/embedded/session-runner.js.map

@@ -1 +1 @@
-{"version":3,"file":"session-runner.js","names":[],"sources":["../../../../src/agent/embedded/session-runner.ts"],"sourcesContent":["import { existsSync } from 'node:fs';\nimport type { AgentTool } from '@earendil-works/pi-agent-core';\nimport type { ThinkingLevel } from '@earendil-works/pi-agent-core';\nimport {\n  createAgentSession,\n  DefaultResourceLoader,\n  getAgentDir,\n  SessionManager,\n  SettingsManager,\n  type AgentSession,\n} from '@earendil-works/pi-coding-agent';\nimport type { Model, Api } from '@earendil-works/pi-ai';\n\nimport { createLogger } from '../../utils/logger.js';\nimport { guardSessionManager, type GuardedPiTranscriptManager } from './session-tool-result-guard-wrapper.js';\nimport { prepareSessionManagerForRun } from './session-manager-init.js';\nimport { prewarmSessionFile } from './session-manager-cache.js';\nimport { applyXopcProviderApiKey, createEmbeddedAuthStorage } from './xopc-auth-storage.js';\nimport { wrapStreamFnForXopcExtensions } from './xopc-stream-bridge.js';\nimport { xopcToolsToDefinitions } from './xopc-tools-bridge.js';\nimport { applySystemPromptOverrideToSession } from './system-prompt-override.js';\n\nconst log = createLogger('EmbeddedSessionRunner');\n\nconst DEFAULT_IDLE_TTL_MS = 5 * 60_000;\n\nexport type EmbeddedRunnerFingerprintInput = {\n  sessionFile: string;\n  workspaceDir: string;\n  modelRef: string;\n  toolNames: readonly string[];\n  systemPrompt: string;\n  thinkingLevel: string;\n};\n\nexport function buildEmbeddedRunnerFingerprint(input: EmbeddedRunnerFingerprintInput): string {\n  const tools = [...input.toolNames].sort().join('\\0');\n  const promptMarker = `${input.systemPrompt.length}:${input.systemPrompt.slice(0, 128)}`;\n  return [\n    input.sessionFile,\n    input.workspaceDir,\n    input.modelRef,\n    tools,\n    promptMarker,\n    input.thinkingLevel,\n  ].join('\\u001f');\n}\n\ntype PooledRunner = {\n  sessionKey: string;\n  fingerprint: string;\n  session: AgentSession;\n  piSm: GuardedPiTranscriptManager;\n  settingsManager: SettingsManager;\n  baseStreamFn: AgentSession['agent']['streamFn'];\n  lastUsedAt: number;\n  idleTimer: ReturnType<typeof setTimeout> | null;\n};\n\nexport type AcquireEmbeddedSessionRunnerParams = {\n  sessionKey: string;\n  sessionId: string;\n  sessionFile: string;\n  sessionsDir: string;\n  hadSessionFile: boolean;\n  workspaceDir: string;\n  model: Model<Api>;\n  modelRef: string;\n  tools: AgentTool[];\n  systemPrompt: string;\n  thinkingLevel: ThinkingLevel;\n};\n\nexport type AcquiredEmbeddedSessionRunner = {\n  session: AgentSession;\n  piSm: GuardedPiTranscriptManager;\n  reused: boolean;\n  release: () => void;\n};\n\nconst pool = new Map<string, PooledRunner>();\n\nlet stats = {\n  acquires: 0,\n  reuses: 0,\n  creates: 0,\n  evictions: 0,\n};\n\nexport function isEmbeddedSessionRunnerEnabled(): boolean {\n  const raw = process.env.XOPC_SESSION_RUNNER?.trim().toLowerCase();\n  if (raw === '0' || raw === 'false' || raw === 'off') {\n    return false;\n  }\n  return true;\n}\n\nexport function getEmbeddedSessionRunnerIdleTtlMs(): number {\n  const raw = process.env.XOPC_SESSION_RUNNER_TTL_MS?.trim();\n  if (!raw) {\n    return DEFAULT_IDLE_TTL_MS;\n  }\n  const parsed = Number.parseInt(raw, 10);\n  return Number.isFinite(parsed) && parsed > 0 ? parsed : DEFAULT_IDLE_TTL_MS;\n}\n\nexport function getEmbeddedSessionRunnerStats(): Readonly<typeof stats> & { pooled: number } {\n  return { ...stats, pooled: pool.size };\n}\n\nexport function resetEmbeddedSessionRunnerForTest(): void {\n  for (const entry of pool.values()) {\n    clearIdleTimer(entry);\n  }\n  pool.clear();\n  stats = { acquires: 0, reuses: 0, creates: 0, evictions: 0 };\n}\n\nfunction clearIdleTimer(entry: PooledRunner): void {\n  if (entry.idleTimer) {\n    clearTimeout(entry.idleTimer);\n    entry.idleTimer = null;\n  }\n}\n\nfunction scheduleIdleEviction(sessionKey: string, entry: PooledRunner): void {\n  clearIdleTimer(entry);\n  const ttlMs = getEmbeddedSessionRunnerIdleTtlMs();\n  entry.idleTimer = setTimeout(() => {\n    const current = pool.get(sessionKey);\n    if (current === entry) {\n      disposePooledRunner(sessionKey, entry, 'idle_ttl');\n    }\n  }, ttlMs);\n  entry.idleTimer.unref?.();\n}\n\nfunction disposePooledRunner(sessionKey: string, entry: PooledRunner, reason: string): void {\n  clearIdleTimer(entry);\n  pool.delete(sessionKey);\n  stats.evictions += 1;\n  try {\n    entry.piSm.flushPendingToolResults?.();\n  } catch {\n    /* ignore */\n  }\n  log.debug({ sessionKey, reason }, 'Embedded session runner evicted');\n}\n\nexport function evictEmbeddedSessionRunner(sessionKey: string, reason = 'explicit'): void {\n  const entry = pool.get(sessionKey);\n  if (!entry) {\n    return;\n  }\n  disposePooledRunner(sessionKey, entry, reason);\n}\n\nexport function evictAllEmbeddedSessionRunners(reason = 'dispose_all'): void {\n  for (const sessionKey of [...pool.keys()]) {\n    evictEmbeddedSessionRunner(sessionKey, reason);\n  }\n}\n\nfunction createEmbeddedSettingsManager(cwd: string): SettingsManager {\n  const sm = SettingsManager.inMemory({ compaction: { enabled: false } });\n  sm.setCompactionEnabled(false);\n  void cwd;\n  return sm;\n}\n\nasync function createPooledRunner(params: AcquireEmbeddedSessionRunnerParams): Promise<PooledRunner> {\n  const {\n    sessionKey,\n    sessionId,\n    sessionFile,\n    sessionsDir,\n    hadSessionFile,\n    workspaceDir,\n    model,\n    thinkingLevel,\n    tools,\n    systemPrompt,\n  } = params;\n\n  await prewarmSessionFile(sessionFile);\n  const settingsManager = createEmbeddedSettingsManager(workspaceDir);\n\n  const piSm = guardSessionManager(SessionManager.open(sessionFile, sessionsDir, workspaceDir), {\n    sessionKey,\n    contextWindowTokens: model.contextWindow ?? 128_000,\n  });\n\n  await prepareSessionManagerForRun({\n    sessionManager: piSm,\n    sessionFile,\n    hadSessionFile,\n    sessionId,\n    cwd: workspaceDir,\n  });\n\n  const toolDefs = xopcToolsToDefinitions(tools);\n  const toolNames = tools.map((t) => t.name);\n\n  const authStorage = createEmbeddedAuthStorage();\n  applyXopcProviderApiKey(authStorage, model.provider);\n\n  const resourceLoader = new DefaultResourceLoader({\n    cwd: workspaceDir,\n    agentDir: getAgentDir(),\n    settingsManager,\n    noContextFiles: true,\n  });\n  await resourceLoader.reload();\n\n  const { session } = await createAgentSession({\n    cwd: workspaceDir,\n    model,\n    thinkingLevel: thinkingLevel ?? 'medium',\n    sessionManager: piSm,\n    settingsManager,\n    authStorage,\n    resourceLoader,\n    noTools: 'builtin',\n    customTools: toolDefs,\n    tools: toolNames,\n  });\n\n  applySystemPromptOverrideToSession(session, systemPrompt);\n  const baseStreamFn = wrapStreamFnForXopcExtensions(session.agent.streamFn);\n  session.agent.streamFn = baseStreamFn;\n\n  const fingerprint = buildEmbeddedRunnerFingerprint({\n    sessionFile,\n    workspaceDir,\n    modelRef: params.modelRef,\n    toolNames,\n    systemPrompt,\n    thinkingLevel: thinkingLevel ?? 'medium',\n  });\n\n  return {\n    sessionKey,\n    fingerprint,\n    session,\n    piSm,\n    settingsManager,\n    baseStreamFn,\n    lastUsedAt: Date.now(),\n    idleTimer: null,\n  };\n}\n\nexport async function acquireEmbeddedSessionRunner(\n  params: AcquireEmbeddedSessionRunnerParams,\n): Promise<AcquiredEmbeddedSessionRunner> {\n  stats.acquires += 1;\n\n  const fingerprint = buildEmbeddedRunnerFingerprint({\n    sessionFile: params.sessionFile,\n    workspaceDir: params.workspaceDir,\n    modelRef: params.modelRef,\n    toolNames: params.tools.map((t) => t.name),\n    systemPrompt: params.systemPrompt,\n    thinkingLevel: params.thinkingLevel ?? 'medium',\n  });\n\n  const reuseEnabled = isEmbeddedSessionRunnerEnabled();\n  const existing = pool.get(params.sessionKey);\n\n  let entry: PooledRunner;\n  let reused = false;\n\n  if (reuseEnabled && existing && existing.fingerprint === fingerprint) {\n    clearIdleTimer(existing);\n    entry = existing;\n    entry.lastUsedAt = Date.now();\n    reused = true;\n    stats.reuses += 1;\n    applySystemPromptOverrideToSession(entry.session, params.systemPrompt);\n    entry.session.agent.streamFn = entry.baseStreamFn;\n    log.debug({ sessionKey: params.sessionKey }, 'Reusing pooled embedded session runner');\n  } else {\n    if (existing) {\n      disposePooledRunner(params.sessionKey, existing, 'fingerprint_mismatch');\n    }\n    entry = await createPooledRunner(params);\n    pool.set(params.sessionKey, entry);\n    stats.creates += 1;\n    log.debug({ sessionKey: params.sessionKey }, 'Created embedded session runner');\n  }\n\n  return {\n    session: entry.session,\n    piSm: entry.piSm,\n    reused,\n    release: () => {\n      if (!isEmbeddedSessionRunnerEnabled()) {\n        disposePooledRunner(params.sessionKey, entry, 'runner_disabled');\n        return;\n      }\n      entry.lastUsedAt = Date.now();\n      scheduleIdleEviction(params.sessionKey, entry);\n    },\n  };\n}\n\n/** Resolve transcript path inputs used by both runner acquire and turn execution. */\nexport async function resolveEmbeddedTranscriptInputs(\n  sessionStore: import('../../session/store.js').SessionStore,\n  sessionKey: string,\n): Promise<{\n  sessionId: string;\n  sessionFile: string;\n  sessionsDir: string;\n  hadSessionFile: boolean;\n}> {\n  const { sessionId, absPath: sessionFile, sessionsDir } = await sessionStore.resolveTranscriptPath(sessionKey);\n  return {\n    sessionId,\n    sessionFile,\n    sessionsDir,\n    hadSessionFile: existsSync(sessionFile),\n  };\n}\n"],"mappings":";;;;;;;;;;;;aAaqD;AASrD,MAAM,MAAM,aAAa,wBAAwB;AAEjD,MAAM,sBAAsB,IAAI;AAWhC,SAAgB,+BAA+B,OAA+C;CAC5F,MAAM,QAAQ,CAAC,GAAG,MAAM,UAAU,CAAC,MAAM,CAAC,KAAK,KAAK;CACpD,MAAM,eAAe,GAAG,MAAM,aAAa,OAAO,GAAG,MAAM,aAAa,MAAM,GAAG,IAAI;AACrF,QAAO;EACL,MAAM;EACN,MAAM;EACN,MAAM;EACN;EACA;EACA,MAAM;EACP,CAAC,KAAK,IAAS;;AAmClB,MAAM,uBAAO,IAAI,KAA2B;AAE5C,IAAI,QAAQ;CACV,UAAU;CACV,QAAQ;CACR,SAAS;CACT,WAAW;CACZ;AAED,SAAgB,iCAA0C;CACxD,MAAM,MAAM,QAAQ,IAAI,qBAAqB,MAAM,CAAC,aAAa;AACjE,KAAI,QAAQ,OAAO,QAAQ,WAAW,QAAQ,MAC5C,QAAO;AAET,QAAO;;AAGT,SAAgB,oCAA4C;CAC1D,MAAM,MAAM,QAAQ,IAAI,4BAA4B,MAAM;AAC1D,KAAI,CAAC,IACH,QAAO;CAET,MAAM,SAAS,OAAO,SAAS,KAAK,GAAG;AACvC,QAAO,OAAO,SAAS,OAAO,IAAI,SAAS,IAAI,SAAS;;AAG1D,SAAgB,gCAA6E;AAC3F,QAAO;EAAE,GAAG;EAAO,QAAQ,KAAK;EAAM;;AAGxC,SAAgB,oCAA0C;AACxD,MAAK,MAAM,SAAS,KAAK,QAAQ,CAC/B,gBAAe,MAAM;AAEvB,MAAK,OAAO;AACZ,SAAQ;EAAE,UAAU;EAAG,QAAQ;EAAG,SAAS;EAAG,WAAW;EAAG;;AAG9D,SAAS,eAAe,OAA2B;AACjD,KAAI,MAAM,WAAW;AACnB,eAAa,MAAM,UAAU;AAC7B,QAAM,YAAY;;;AAItB,SAAS,qBAAqB,YAAoB,OAA2B;AAC3E,gBAAe,MAAM;CACrB,MAAM,QAAQ,mCAAmC;AACjD,OAAM,YAAY,iBAAiB;AAEjC,MADgB,KAAK,IAAI,WACd,KAAK,MACd,qBAAoB,YAAY,OAAO,WAAW;IAEnD,MAAM;AACT,OAAM,UAAU,SAAS;;AAG3B,SAAS,oBAAoB,YAAoB,OAAqB,QAAsB;AAC1F,gBAAe,MAAM;AACrB,MAAK,OAAO,WAAW;AACvB,OAAM,aAAa;AACnB,KAAI;AACF,QAAM,KAAK,2BAA2B;SAChC;AAGR,KAAI,MAAM;EAAE;EAAY;EAAQ,EAAE,kCAAkC;;AAGtE,SAAgB,2BAA2B,YAAoB,SAAS,YAAkB;CACxF,MAAM,QAAQ,KAAK,IAAI,WAAW;AAClC,KAAI,CAAC,MACH;AAEF,qBAAoB,YAAY,OAAO,OAAO;;AAGhD,SAAgB,+BAA+B,SAAS,eAAqB;AAC3E,MAAK,MAAM,cAAc,CAAC,GAAG,KAAK,MAAM,CAAC,CACvC,4BAA2B,YAAY,OAAO;;AAIlD,SAAS,8BAA8B,KAA8B;CACnE,MAAM,KAAK,gBAAgB,SAAS,EAAE,YAAY,EAAE,SAAS,OAAO,EAAE,CAAC;AACvE,IAAG,qBAAqB,MAAM;AAE9B,QAAO;;AAGT,eAAe,mBAAmB,QAAmE;CACnG,MAAM,EACJ,YACA,WACA,aACA,aACA,gBACA,cACA,OACA,eACA,OACA,iBACE;AAEJ,OAAM,mBAAmB,YAAY;CACrC,MAAM,kBAAkB,8BAA8B,aAAa;CAEnE,MAAM,OAAO,oBAAoB,eAAe,KAAK,aAAa,aAAa,aAAa,EAAE;EAC5F;EACA,qBAAqB,MAAM,iBAAiB;EAC7C,CAAC;AAEF,OAAM,4BAA4B;EAChC,gBAAgB;EAChB;EACA;EACA;EACA,KAAK;EACN,CAAC;CAEF,MAAM,WAAW,uBAAuB,MAAM;CAC9C,MAAM,YAAY,MAAM,KAAK,MAAM,EAAE,KAAK;CAE1C,MAAM,cAAc,2BAA2B;AAC/C,yBAAwB,aAAa,MAAM,SAAS;CAEpD,MAAM,iBAAiB,IAAI,sBAAsB;EAC/C,KAAK;EACL,UAAU,aAAa;EACvB;EACA,gBAAgB;EACjB,CAAC;AACF,OAAM,eAAe,QAAQ;CAE7B,MAAM,EAAE,YAAY,MAAM,mBAAmB;EAC3C,KAAK;EACL;EACA,eAAe,iBAAiB;EAChC,gBAAgB;EAChB;EACA;EACA;EACA,SAAS;EACT,aAAa;EACb,OAAO;EACR,CAAC;AAEF,oCAAmC,SAAS,aAAa;CACzD,MAAM,eAAe,8BAA8B,QAAQ,MAAM,SAAS;AAC1E,SAAQ,MAAM,WAAW;AAWzB,QAAO;EACL;EACA,aAXkB,+BAA+B;GACjD;GACA;GACA,UAAU,OAAO;GACjB;GACA;GACA,eAAe,iBAAiB;GACjC,CAIY;EACX;EACA;EACA;EACA;EACA,YAAY,KAAK,KAAK;EACtB,WAAW;EACZ;;AAGH,eAAsB,6BACpB,QACwC;AACxC,OAAM,YAAY;CAElB,MAAM,cAAc,+BAA+B;EACjD,aAAa,OAAO;EACpB,cAAc,OAAO;EACrB,UAAU,OAAO;EACjB,WAAW,OAAO,MAAM,KAAK,MAAM,EAAE,KAAK;EAC1C,cAAc,OAAO;EACrB,eAAe,OAAO,iBAAiB;EACxC,CAAC;CAEF,MAAM,eAAe,gCAAgC;CACrD,MAAM,WAAW,KAAK,IAAI,OAAO,WAAW;CAE5C,IAAI;CACJ,IAAI,SAAS;AAEb,KAAI,gBAAgB,YAAY,SAAS,gBAAgB,aAAa;AACpE,iBAAe,SAAS;AACxB,UAAQ;AACR,QAAM,aAAa,KAAK,KAAK;AAC7B,WAAS;AACT,QAAM,UAAU;AAChB,qCAAmC,MAAM,SAAS,OAAO,aAAa;AACtE,QAAM,QAAQ,MAAM,WAAW,MAAM;AACrC,MAAI,MAAM,EAAE,YAAY,OAAO,YAAY,EAAE,yCAAyC;QACjF;AACL,MAAI,SACF,qBAAoB,OAAO,YAAY,UAAU,uBAAuB;AAE1E,UAAQ,MAAM,mBAAmB,OAAO;AACxC,OAAK,IAAI,OAAO,YAAY,MAAM;AAClC,QAAM,WAAW;AACjB,MAAI,MAAM,EAAE,YAAY,OAAO,YAAY,EAAE,kCAAkC;;AAGjF,QAAO;EACL,SAAS,MAAM;EACf,MAAM,MAAM;EACZ;EACA,eAAe;AACb,OAAI,CAAC,gCAAgC,EAAE;AACrC,wBAAoB,OAAO,YAAY,OAAO,kBAAkB;AAChE;;AAEF,SAAM,aAAa,KAAK,KAAK;AAC7B,wBAAqB,OAAO,YAAY,MAAM;;EAEjD;;;AAIH,eAAsB,gCACpB,cACA,YAMC;CACD,MAAM,EAAE,WAAW,SAAS,aAAa,gBAAgB,MAAM,aAAa,sBAAsB,WAAW;AAC7G,QAAO;EACL;EACA;EACA;EACA,gBAAgB,WAAW,YAAY;EACxC"}
+{"version":3,"file":"session-runner.js","names":[],"sources":["../../../../src/agent/embedded/session-runner.ts"],"sourcesContent":["import { existsSync } from 'node:fs';\nimport type { AgentTool } from '@earendil-works/pi-agent-core';\nimport type { ThinkingLevel } from '@earendil-works/pi-agent-core';\nimport {\n  createAgentSession,\n  DefaultResourceLoader,\n  getAgentDir,\n  SessionManager,\n  SettingsManager,\n  type AgentSession,\n} from '@earendil-works/pi-coding-agent';\nimport type { Model, Api } from '@earendil-works/pi-ai';\n\nimport { createLogger } from '../../utils/logger.js';\nimport { guardSessionManager, type GuardedPiTranscriptManager } from './session-tool-result-guard.js';\nimport { prepareSessionManagerForRun } from './session-manager-init.js';\nimport { defaultSessionManagerCache, type SessionManagerCache } from './session-manager-cache.js';\nimport { applyXopcProviderApiKey, createEmbeddedAuthStorage } from './xopc-auth-storage.js';\nimport { wrapStreamFnForXopcExtensions } from './xopc-stream-bridge.js';\nimport { xopcToolsToDefinitions } from './xopc-tools-bridge.js';\nimport { applySystemPromptOverrideToSession } from './system-prompt-override.js';\n\nconst log = createLogger('EmbeddedSessionRunner');\n\nconst DEFAULT_IDLE_TTL_MS = 5 * 60_000;\n\nexport type EmbeddedRunnerFingerprintInput = {\n  sessionFile: string;\n  workspaceDir: string;\n  modelRef: string;\n  toolNames: readonly string[];\n  systemPrompt: string;\n  thinkingLevel: string;\n};\n\nexport function buildEmbeddedRunnerFingerprint(input: EmbeddedRunnerFingerprintInput): string {\n  const tools = [...input.toolNames].sort().join('\\0');\n  const promptMarker = `${input.systemPrompt.length}:${input.systemPrompt.slice(0, 128)}`;\n  return [\n    input.sessionFile,\n    input.workspaceDir,\n    input.modelRef,\n    tools,\n    promptMarker,\n    input.thinkingLevel,\n  ].join('\u001f');\n}\n\ntype PooledRunner = {\n  sessionKey: string;\n  fingerprint: string;\n  session: AgentSession;\n  piSm: GuardedPiTranscriptManager;\n  settingsManager: SettingsManager;\n  baseStreamFn: AgentSession['agent']['streamFn'];\n  lastUsedAt: number;\n  idleTimer: ReturnType<typeof setTimeout> | null;\n};\n\nexport type AcquireEmbeddedSessionRunnerParams = {\n  sessionKey: string;\n  sessionId: string;\n  sessionFile: string;\n  sessionsDir: string;\n  hadSessionFile: boolean;\n  workspaceDir: string;\n  model: Model<Api>;\n  modelRef: string;\n  tools: AgentTool[];\n  systemPrompt: string;\n  thinkingLevel: ThinkingLevel;\n};\n\nexport type AcquiredEmbeddedSessionRunner = {\n  session: AgentSession;\n  piSm: GuardedPiTranscriptManager;\n  reused: boolean;\n  release: () => void;\n};\n\nexport interface EmbeddedSessionRunnerPoolStats {\n  acquires: number;\n  reuses: number;\n  creates: number;\n  evictions: number;\n  pooled: number;\n}\n\nexport function isEmbeddedSessionRunnerEnabled(): boolean {\n  const raw = process.env.XOPC_SESSION_RUNNER?.trim().toLowerCase();\n  if (raw === '0' || raw === 'false' || raw === 'off') {\n    return false;\n  }\n  return true;\n}\n\nexport function getEmbeddedSessionRunnerIdleTtlMs(): number {\n  const raw = process.env.XOPC_SESSION_RUNNER_TTL_MS?.trim();\n  if (!raw) {\n    return DEFAULT_IDLE_TTL_MS;\n  }\n  const parsed = Number.parseInt(raw, 10);\n  return Number.isFinite(parsed) && parsed > 0 ? parsed : DEFAULT_IDLE_TTL_MS;\n}\n\nfunction createEmbeddedSettingsManager(cwd: string): SettingsManager {\n  const sm = SettingsManager.inMemory({ compaction: { enabled: false } });\n  sm.setCompactionEnabled(false);\n  void cwd;\n  return sm;\n}\n\nexport interface EmbeddedSessionRunnerPoolOptions {\n  /** File-exists cache shared with prewarmSessionFile callers. */\n  sessionManagerCache?: SessionManagerCache;\n  /** Override for the env-driven enable flag (testing). */\n  isEnabled?: () => boolean;\n  /** Override for the env-driven idle TTL (testing). */\n  getIdleTtlMs?: () => number;\n}\n\n/**\n * Owns the per-session pool of pi `AgentSession` runners. The class is the supported,\n * injectable owner; {@link defaultEmbeddedSessionRunnerPool} keeps the historic\n * module-level free functions working until every caller is migrated to DI.\n */\nexport class EmbeddedSessionRunnerPool {\n  private readonly pool = new Map<string, PooledRunner>();\n  private readonly cache: SessionManagerCache;\n  private readonly isEnabledFn: () => boolean;\n  private readonly getIdleTtlMsFn: () => number;\n\n  private stats: Omit<EmbeddedSessionRunnerPoolStats, 'pooled'> = {\n    acquires: 0,\n    reuses: 0,\n    creates: 0,\n    evictions: 0,\n  };\n\n  constructor(opts: EmbeddedSessionRunnerPoolOptions = {}) {\n    this.cache = opts.sessionManagerCache ?? defaultSessionManagerCache;\n    this.isEnabledFn = opts.isEnabled ?? isEmbeddedSessionRunnerEnabled;\n    this.getIdleTtlMsFn = opts.getIdleTtlMs ?? getEmbeddedSessionRunnerIdleTtlMs;\n  }\n\n  getStats(): Readonly<EmbeddedSessionRunnerPoolStats> {\n    return { ...this.stats, pooled: this.pool.size };\n  }\n\n  resetForTest(): void {\n    for (const entry of this.pool.values()) {\n      this.clearIdleTimer(entry);\n    }\n    this.pool.clear();\n    this.stats = { acquires: 0, reuses: 0, creates: 0, evictions: 0 };\n  }\n\n  evict(sessionKey: string, reason = 'explicit'): void {\n    const entry = this.pool.get(sessionKey);\n    if (!entry) {\n      return;\n    }\n    this.disposePooledRunner(sessionKey, entry, reason);\n  }\n\n  evictAll(reason = 'dispose_all'): void {\n    for (const sessionKey of [...this.pool.keys()]) {\n      this.evict(sessionKey, reason);\n    }\n  }\n\n  async acquire(params: AcquireEmbeddedSessionRunnerParams): Promise<AcquiredEmbeddedSessionRunner> {\n    this.stats.acquires += 1;\n\n    const fingerprint = buildEmbeddedRunnerFingerprint({\n      sessionFile: params.sessionFile,\n      workspaceDir: params.workspaceDir,\n      modelRef: params.modelRef,\n      toolNames: params.tools.map((t) => t.name),\n      systemPrompt: params.systemPrompt,\n      thinkingLevel: params.thinkingLevel ?? 'medium',\n    });\n\n    const reuseEnabled = this.isEnabledFn();\n    const existing = this.pool.get(params.sessionKey);\n\n    let entry: PooledRunner;\n    let reused = false;\n\n    if (reuseEnabled && existing && existing.fingerprint === fingerprint) {\n      this.clearIdleTimer(existing);\n      entry = existing;\n      entry.lastUsedAt = Date.now();\n      reused = true;\n      this.stats.reuses += 1;\n      applySystemPromptOverrideToSession(entry.session, params.systemPrompt);\n      entry.session.agent.streamFn = entry.baseStreamFn;\n      log.debug({ sessionKey: params.sessionKey }, 'Reusing pooled embedded session runner');\n    } else {\n      if (existing) {\n        this.disposePooledRunner(params.sessionKey, existing, 'fingerprint_mismatch');\n      }\n      entry = await this.createPooledRunner(params);\n      this.pool.set(params.sessionKey, entry);\n      this.stats.creates += 1;\n      log.debug({ sessionKey: params.sessionKey }, 'Created embedded session runner');\n    }\n\n    return {\n      session: entry.session,\n      piSm: entry.piSm,\n      reused,\n      release: () => {\n        if (!this.isEnabledFn()) {\n          this.disposePooledRunner(params.sessionKey, entry, 'runner_disabled');\n          return;\n        }\n        entry.lastUsedAt = Date.now();\n        this.scheduleIdleEviction(params.sessionKey, entry);\n      },\n    };\n  }\n\n  private clearIdleTimer(entry: PooledRunner): void {\n    if (entry.idleTimer) {\n      clearTimeout(entry.idleTimer);\n      entry.idleTimer = null;\n    }\n  }\n\n  private scheduleIdleEviction(sessionKey: string, entry: PooledRunner): void {\n    this.clearIdleTimer(entry);\n    const ttlMs = this.getIdleTtlMsFn();\n    entry.idleTimer = setTimeout(() => {\n      const current = this.pool.get(sessionKey);\n      if (current === entry) {\n        this.disposePooledRunner(sessionKey, entry, 'idle_ttl');\n      }\n    }, ttlMs);\n    entry.idleTimer.unref?.();\n  }\n\n  private disposePooledRunner(sessionKey: string, entry: PooledRunner, reason: string): void {\n    this.clearIdleTimer(entry);\n    this.pool.delete(sessionKey);\n    this.stats.evictions += 1;\n    try {\n      entry.piSm.flushPendingToolResults?.();\n    } catch {\n      /* ignore */\n    }\n    log.debug({ sessionKey, reason }, 'Embedded session runner evicted');\n  }\n\n  private async createPooledRunner(params: AcquireEmbeddedSessionRunnerParams): Promise<PooledRunner> {\n    const {\n      sessionKey,\n      sessionId,\n      sessionFile,\n      sessionsDir,\n      hadSessionFile,\n      workspaceDir,\n      model,\n      thinkingLevel,\n      tools,\n      systemPrompt,\n    } = params;\n\n    await this.cache.prewarm(sessionFile);\n    const settingsManager = createEmbeddedSettingsManager(workspaceDir);\n\n    const piSm = guardSessionManager(SessionManager.open(sessionFile, sessionsDir, workspaceDir), {\n      sessionKey,\n      contextWindowTokens: model.contextWindow ?? 128_000,\n    });\n\n    await prepareSessionManagerForRun({\n      sessionManager: piSm,\n      sessionFile,\n      hadSessionFile,\n      sessionId,\n      cwd: workspaceDir,\n    });\n\n    const toolDefs = xopcToolsToDefinitions(tools);\n    const toolNames = tools.map((t) => t.name);\n\n    const authStorage = createEmbeddedAuthStorage();\n    applyXopcProviderApiKey(authStorage, model.provider);\n\n    const resourceLoader = new DefaultResourceLoader({\n      cwd: workspaceDir,\n      agentDir: getAgentDir(),\n      settingsManager,\n      noContextFiles: true,\n    });\n    await resourceLoader.reload();\n\n    const { session } = await createAgentSession({\n      cwd: workspaceDir,\n      model,\n      thinkingLevel: thinkingLevel ?? 'medium',\n      sessionManager: piSm,\n      settingsManager,\n      authStorage,\n      resourceLoader,\n      noTools: 'builtin',\n      customTools: toolDefs,\n      tools: toolNames,\n    });\n\n    applySystemPromptOverrideToSession(session, systemPrompt);\n    const baseStreamFn = wrapStreamFnForXopcExtensions(session.agent.streamFn);\n    session.agent.streamFn = baseStreamFn;\n\n    const fingerprint = buildEmbeddedRunnerFingerprint({\n      sessionFile,\n      workspaceDir,\n      modelRef: params.modelRef,\n      toolNames,\n      systemPrompt,\n      thinkingLevel: thinkingLevel ?? 'medium',\n    });\n\n    return {\n      sessionKey,\n      fingerprint,\n      session,\n      piSm,\n      settingsManager,\n      baseStreamFn,\n      lastUsedAt: Date.now(),\n      idleTimer: null,\n    };\n  }\n}\n\nexport const defaultEmbeddedSessionRunnerPool = new EmbeddedSessionRunnerPool();\n\nexport function getEmbeddedSessionRunnerStats(): Readonly<EmbeddedSessionRunnerPoolStats> {\n  return defaultEmbeddedSessionRunnerPool.getStats();\n}\n\nexport function resetEmbeddedSessionRunnerForTest(): void {\n  defaultEmbeddedSessionRunnerPool.resetForTest();\n}\n\nexport function evictEmbeddedSessionRunner(sessionKey: string, reason = 'explicit'): void {\n  defaultEmbeddedSessionRunnerPool.evict(sessionKey, reason);\n}\n\nexport function evictAllEmbeddedSessionRunners(reason = 'dispose_all'): void {\n  defaultEmbeddedSessionRunnerPool.evictAll(reason);\n}\n\nexport function acquireEmbeddedSessionRunner(\n  params: AcquireEmbeddedSessionRunnerParams,\n): Promise<AcquiredEmbeddedSessionRunner> {\n  return defaultEmbeddedSessionRunnerPool.acquire(params);\n}\n\n/** Resolve transcript path inputs used by both runner acquire and turn execution. */\nexport async function resolveEmbeddedTranscriptInputs(\n  sessionStore: import('../../session/store.js').SessionStore,\n  sessionKey: string,\n): Promise<{\n  sessionId: string;\n  sessionFile: string;\n  sessionsDir: string;\n  hadSessionFile: boolean;\n}> {\n  const { sessionId, absPath: sessionFile, sessionsDir } = await sessionStore.resolveTranscriptPath(sessionKey);\n  return {\n    sessionId,\n    sessionFile,\n    sessionsDir,\n    hadSessionFile: existsSync(sessionFile),\n  };\n}\n"],"mappings":";;;;;;;;;;;;aAaqD;AASrD,MAAM,MAAM,aAAa,wBAAwB;AAEjD,MAAM,sBAAsB,IAAI;AAWhC,SAAgB,+BAA+B,OAA+C;CAC5F,MAAM,QAAQ,CAAC,GAAG,MAAM,UAAU,CAAC,MAAM,CAAC,KAAK,KAAK;CACpD,MAAM,eAAe,GAAG,MAAM,aAAa,OAAO,GAAG,MAAM,aAAa,MAAM,GAAG,IAAI;AACrF,QAAO;EACL,MAAM;EACN,MAAM;EACN,MAAM;EACN;EACA;EACA,MAAM;EACP,CAAC,KAAK,IAAI;;AA2Cb,SAAgB,iCAA0C;CACxD,MAAM,MAAM,QAAQ,IAAI,qBAAqB,MAAM,CAAC,aAAa;AACjE,KAAI,QAAQ,OAAO,QAAQ,WAAW,QAAQ,MAC5C,QAAO;AAET,QAAO;;AAGT,SAAgB,oCAA4C;CAC1D,MAAM,MAAM,QAAQ,IAAI,4BAA4B,MAAM;AAC1D,KAAI,CAAC,IACH,QAAO;CAET,MAAM,SAAS,OAAO,SAAS,KAAK,GAAG;AACvC,QAAO,OAAO,SAAS,OAAO,IAAI,SAAS,IAAI,SAAS;;AAG1D,SAAS,8BAA8B,KAA8B;CACnE,MAAM,KAAK,gBAAgB,SAAS,EAAE,YAAY,EAAE,SAAS,OAAO,EAAE,CAAC;AACvE,IAAG,qBAAqB,MAAM;AAE9B,QAAO;;;;;;;AAiBT,IAAa,4BAAb,MAAuC;CACrC,uBAAwB,IAAI,KAA2B;CACvD;CACA;CACA;CAEA,QAAgE;EAC9D,UAAU;EACV,QAAQ;EACR,SAAS;EACT,WAAW;EACZ;CAED,YAAY,OAAyC,EAAE,EAAE;AACvD,OAAK,QAAQ,KAAK,uBAAuB;AACzC,OAAK,cAAc,KAAK,aAAa;AACrC,OAAK,iBAAiB,KAAK,gBAAgB;;CAG7C,WAAqD;AACnD,SAAO;GAAE,GAAG,KAAK;GAAO,QAAQ,KAAK,KAAK;GAAM;;CAGlD,eAAqB;AACnB,OAAK,MAAM,SAAS,KAAK,KAAK,QAAQ,CACpC,MAAK,eAAe,MAAM;AAE5B,OAAK,KAAK,OAAO;AACjB,OAAK,QAAQ;GAAE,UAAU;GAAG,QAAQ;GAAG,SAAS;GAAG,WAAW;GAAG;;CAGnE,MAAM,YAAoB,SAAS,YAAkB;EACnD,MAAM,QAAQ,KAAK,KAAK,IAAI,WAAW;AACvC,MAAI,CAAC,MACH;AAEF,OAAK,oBAAoB,YAAY,OAAO,OAAO;;CAGrD,SAAS,SAAS,eAAqB;AACrC,OAAK,MAAM,cAAc,CAAC,GAAG,KAAK,KAAK,MAAM,CAAC,CAC5C,MAAK,MAAM,YAAY,OAAO;;CAIlC,MAAM,QAAQ,QAAoF;AAChG,OAAK,MAAM,YAAY;EAEvB,MAAM,cAAc,+BAA+B;GACjD,aAAa,OAAO;GACpB,cAAc,OAAO;GACrB,UAAU,OAAO;GACjB,WAAW,OAAO,MAAM,KAAK,MAAM,EAAE,KAAK;GAC1C,cAAc,OAAO;GACrB,eAAe,OAAO,iBAAiB;GACxC,CAAC;EAEF,MAAM,eAAe,KAAK,aAAa;EACvC,MAAM,WAAW,KAAK,KAAK,IAAI,OAAO,WAAW;EAEjD,IAAI;EACJ,IAAI,SAAS;AAEb,MAAI,gBAAgB,YAAY,SAAS,gBAAgB,aAAa;AACpE,QAAK,eAAe,SAAS;AAC7B,WAAQ;AACR,SAAM,aAAa,KAAK,KAAK;AAC7B,YAAS;AACT,QAAK,MAAM,UAAU;AACrB,sCAAmC,MAAM,SAAS,OAAO,aAAa;AACtE,SAAM,QAAQ,MAAM,WAAW,MAAM;AACrC,OAAI,MAAM,EAAE,YAAY,OAAO,YAAY,EAAE,yCAAyC;SACjF;AACL,OAAI,SACF,MAAK,oBAAoB,OAAO,YAAY,UAAU,uBAAuB;AAE/E,WAAQ,MAAM,KAAK,mBAAmB,OAAO;AAC7C,QAAK,KAAK,IAAI,OAAO,YAAY,MAAM;AACvC,QAAK,MAAM,WAAW;AACtB,OAAI,MAAM,EAAE,YAAY,OAAO,YAAY,EAAE,kCAAkC;;AAGjF,SAAO;GACL,SAAS,MAAM;GACf,MAAM,MAAM;GACZ;GACA,eAAe;AACb,QAAI,CAAC,KAAK,aAAa,EAAE;AACvB,UAAK,oBAAoB,OAAO,YAAY,OAAO,kBAAkB;AACrE;;AAEF,UAAM,aAAa,KAAK,KAAK;AAC7B,SAAK,qBAAqB,OAAO,YAAY,MAAM;;GAEtD;;CAGH,eAAuB,OAA2B;AAChD,MAAI,MAAM,WAAW;AACnB,gBAAa,MAAM,UAAU;AAC7B,SAAM,YAAY;;;CAItB,qBAA6B,YAAoB,OAA2B;AAC1E,OAAK,eAAe,MAAM;EAC1B,MAAM,QAAQ,KAAK,gBAAgB;AACnC,QAAM,YAAY,iBAAiB;AAEjC,OADgB,KAAK,KAAK,IAAI,WACnB,KAAK,MACd,MAAK,oBAAoB,YAAY,OAAO,WAAW;KAExD,MAAM;AACT,QAAM,UAAU,SAAS;;CAG3B,oBAA4B,YAAoB,OAAqB,QAAsB;AACzF,OAAK,eAAe,MAAM;AAC1B,OAAK,KAAK,OAAO,WAAW;AAC5B,OAAK,MAAM,aAAa;AACxB,MAAI;AACF,SAAM,KAAK,2BAA2B;UAChC;AAGR,MAAI,MAAM;GAAE;GAAY;GAAQ,EAAE,kCAAkC;;CAGtE,MAAc,mBAAmB,QAAmE;EAClG,MAAM,EACJ,YACA,WACA,aACA,aACA,gBACA,cACA,OACA,eACA,OACA,iBACE;AAEJ,QAAM,KAAK,MAAM,QAAQ,YAAY;EACrC,MAAM,kBAAkB,8BAA8B,aAAa;EAEnE,MAAM,OAAO,oBAAoB,eAAe,KAAK,aAAa,aAAa,aAAa,EAAE;GAC5F;GACA,qBAAqB,MAAM,iBAAiB;GAC7C,CAAC;AAEF,QAAM,4BAA4B;GAChC,gBAAgB;GAChB;GACA;GACA;GACA,KAAK;GACN,CAAC;EAEF,MAAM,WAAW,uBAAuB,MAAM;EAC9C,MAAM,YAAY,MAAM,KAAK,MAAM,EAAE,KAAK;EAE1C,MAAM,cAAc,2BAA2B;AAC/C,0BAAwB,aAAa,MAAM,SAAS;EAEpD,MAAM,iBAAiB,IAAI,sBAAsB;GAC/C,KAAK;GACL,UAAU,aAAa;GACvB;GACA,gBAAgB;GACjB,CAAC;AACF,QAAM,eAAe,QAAQ;EAE7B,MAAM,EAAE,YAAY,MAAM,mBAAmB;GAC3C,KAAK;GACL;GACA,eAAe,iBAAiB;GAChC,gBAAgB;GAChB;GACA;GACA;GACA,SAAS;GACT,aAAa;GACb,OAAO;GACR,CAAC;AAEF,qCAAmC,SAAS,aAAa;EACzD,MAAM,eAAe,8BAA8B,QAAQ,MAAM,SAAS;AAC1E,UAAQ,MAAM,WAAW;AAWzB,SAAO;GACL;GACA,aAXkB,+BAA+B;IACjD;IACA;IACA,UAAU,OAAO;IACjB;IACA;IACA,eAAe,iBAAiB;IACjC,CAIY;GACX;GACA;GACA;GACA;GACA,YAAY,KAAK,KAAK;GACtB,WAAW;GACZ;;;AAIL,MAAa,mCAAmC,IAAI,2BAA2B;AAE/E,SAAgB,gCAA0E;AACxF,QAAO,iCAAiC,UAAU;;AAGpD,SAAgB,oCAA0C;AACxD,kCAAiC,cAAc;;AAGjD,SAAgB,2BAA2B,YAAoB,SAAS,YAAkB;AACxF,kCAAiC,MAAM,YAAY,OAAO;;AAG5D,SAAgB,+BAA+B,SAAS,eAAqB;AAC3E,kCAAiC,SAAS,OAAO;;AAGnD,SAAgB,6BACd,QACwC;AACxC,QAAO,iCAAiC,QAAQ,OAAO;;;AAIzD,eAAsB,gCACpB,cACA,YAMC;CACD,MAAM,EAAE,WAAW,SAAS,aAAa,gBAAgB,MAAM,aAAa,sBAAsB,WAAW;AAC7G,QAAO;EACL;EACA;EACA;EACA,gBAAgB,WAAW,YAAY;EACxC"}

package/dist/src/agent/embedded/session-tool-result-guard.d.ts

@@ -1,5 +1,25 @@
-import type { AgentMessage } from "@earendil-works/pi-agent-core";
-import type { SessionManager } from "@earendil-works/pi-coding-agent";
+/**
+ * Session tool-result guard — wraps a pi `SessionManager.appendMessage` to:
+ *   1. cap oversized tool-result text + details so they cannot blow up the next
+ *      LLM request (size + persistence limits).
+ *   2. track pending tool-call IDs so missing tool results can be synthesised
+ *      (some providers refuse a turn that has an orphan tool_use block).
+ *   3. drop assistant `toolCall` blocks whose tool name is not in the allowlist
+ *      (these would also trigger provider 400s).
+ *   4. broadcast `xopc:transcript-row` updates so the gateway UI can stream them.
+ *
+ * Previously this module shipped with three sibling files
+ * (`session-tool-result-state.ts`, `session-raw-append-message.ts`,
+ * `session-tool-result-guard-wrapper.ts`). They are now consolidated here as
+ * private constructs around the `ToolResultGuard` class. Pi-coding-agent owns
+ * the `SessionManager` instance and calls `appendMessage` from inside the
+ * runtime, so we still need to monkey-patch that method — but the patched
+ * implementation is just `guard.guardedAppend.bind(guard)` and all state lives
+ * on the class.
+ */
+import type { AgentMessage } from '@earendil-works/pi-agent-core';
+import type { SessionManager } from '@earendil-works/pi-coding-agent';
+import type { Config } from '../../config/schema.js';
 export type BeforeMessageWriteHookEvent = {
     message: AgentMessage;
 };
@@ -7,14 +27,10 @@ export type BeforeMessageWriteHookResult = {
     block?: boolean;
     message?: AgentMessage;
 } | undefined;
-import { getRawSessionAppendMessage } from "./session-raw-append-message.js";
-export { getRawSessionAppendMessage };
-export declare function installSessionToolResultGuard(sessionManager: SessionManager, opts?: {
+export interface ToolResultGuardOptions {
     /** Optional session key for transcript update broadcasts. */
     sessionKey?: string;
-    /**
-     * Optional transform applied to any message before persistence.
-     */
+    /** Optional transform applied to any message before persistence. */
     transformMessageForPersistence?: (message: AgentMessage) => AgentMessage;
     /**
      * Optional, synchronous transform applied to toolResult messages *before* they are
@@ -43,8 +59,40 @@ export declare function installSessionToolResultGuard(sessionManager: SessionMan
      */
     beforeMessageWriteHook?: (event: BeforeMessageWriteHookEvent) => BeforeMessageWriteHookResult;
     maxToolResultChars?: number;
-}): {
+}
+export interface InstallSessionToolResultGuardResult {
     flushPendingToolResults: () => void;
     clearPendingToolResults: () => void;
     getPendingIds: () => string[];
+}
+/** Idempotent wrapper that also adds the helper methods consumers expect. */
+export type GuardedPiTranscriptManager = SessionManager & {
+    flushPendingToolResults?: () => void;
+    clearPendingToolResults?: () => void;
 };
+/**
+ * Install the guard on a SessionManager and return its control API.
+ * Subsequent assistant/toolResult writes by pi-coding-agent flow through the
+ * guard transparently.
+ */
+export declare function installSessionToolResultGuard(sessionManager: SessionManager, opts?: ToolResultGuardOptions): InstallSessionToolResultGuardResult;
+/**
+ * Convenience wrapper used by the embedded runner pool: install the guard
+ * (idempotent), pin the size cap from the model context window, and expose
+ * `flushPendingToolResults` / `clearPendingToolResults` directly on the
+ * SessionManager instance so callers do not need to keep the install result.
+ */
+export declare function guardSessionManager(sessionManager: SessionManager, opts?: {
+    agentId?: string;
+    sessionKey?: string;
+    config?: Config;
+    contextWindowTokens?: number;
+    allowSyntheticToolResults?: boolean;
+    missingToolResultText?: string;
+    allowedToolNames?: Iterable<string>;
+}): GuardedPiTranscriptManager;
+/**
+ * Recover the original (un-guarded) appendMessage for a session manager.
+ * Useful for callers that need a low-level "bypass the guard" write path.
+ */
+export declare function getRawSessionAppendMessage(sessionManager: SessionManager): SessionManager['appendMessage'];