chore: format tool guard files

This commit is contained in:
Peter Steinberger
2026-01-12 17:45:19 +00:00
parent b88ea39b83
commit 29807119d5

View File

@@ -1649,7 +1649,9 @@ export async function runEmbeddedPiAgent(params: {
}); });
// Pre-warm session file to bring it into OS page cache // Pre-warm session file to bring it into OS page cache
await prewarmSessionFile(params.sessionFile); await prewarmSessionFile(params.sessionFile);
const sessionManager = SessionManager.open(params.sessionFile); const sessionManager = guardSessionManager(
SessionManager.open(params.sessionFile),
);
trackSessionManagerAccess(params.sessionFile); trackSessionManagerAccess(params.sessionFile);
const settingsManager = SettingsManager.create( const settingsManager = SettingsManager.create(
effectiveWorkspace, effectiveWorkspace,
@@ -1669,8 +1671,6 @@ export async function runEmbeddedPiAgent(params: {
model, model,
}); });
const toolResultGuard = guardSessionManager(sessionManager);
const { builtInTools, customTools } = splitSdkTools({ const { builtInTools, customTools } = splitSdkTools({
tools, tools,
sandboxEnabled: !!sandbox?.enabled, sandboxEnabled: !!sandbox?.enabled,
@@ -1723,7 +1723,7 @@ export async function runEmbeddedPiAgent(params: {
session.agent.replaceMessages(limited); session.agent.replaceMessages(limited);
} }
} catch (err) { } catch (err) {
toolResultGuard.flushPendingToolResults?.(); sessionManager.flushPendingToolResults?.();
session.dispose(); session.dispose();
await sessionLock.release(); await sessionLock.release();
throw err; throw err;
@@ -1755,7 +1755,7 @@ export async function runEmbeddedPiAgent(params: {
enforceFinalTag: params.enforceFinalTag, enforceFinalTag: params.enforceFinalTag,
}); });
} catch (err) { } catch (err) {
toolResultGuard.flushPendingToolResults?.(); sessionManager.flushPendingToolResults?.();
session.dispose(); session.dispose();
await sessionLock.release(); await sessionLock.release();
throw err; throw err;
@@ -1853,7 +1853,7 @@ export async function runEmbeddedPiAgent(params: {
ACTIVE_EMBEDDED_RUNS.delete(params.sessionId); ACTIVE_EMBEDDED_RUNS.delete(params.sessionId);
notifyEmbeddedRunEnded(params.sessionId); notifyEmbeddedRunEnded(params.sessionId);
} }
toolResultGuard.flushPendingToolResults?.(); sessionManager.flushPendingToolResults?.();
session.dispose(); session.dispose();
await sessionLock.release(); await sessionLock.release();
params.abortSignal?.removeEventListener?.("abort", onAbort); params.abortSignal?.removeEventListener?.("abort", onAbort);