chore: format tool guard files
This commit is contained in:
@@ -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);
|
||||||
|
|||||||
Reference in New Issue
Block a user