Wrap custom tools with hooks (#248)

This commit is contained in:
Nico Bailon 2025-12-19 10:05:20 -08:00 committed by GitHub
parent 95dcd04e7c
commit 74f7e6c9d5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 10 additions and 4 deletions

View file

@ -315,9 +315,6 @@ export async function main(args: string[]) {
if (hooks.length > 0) {
const timeout = settingsManager.getHookTimeout();
hookRunner = new HookRunner(hooks, cwd, timeout);
// Wrap tools with hook callbacks
selectedTools = wrapToolsWithHooks(selectedTools, hookRunner);
}
// Discover and load custom tools from:
@ -344,6 +341,11 @@ export async function main(args: string[]) {
selectedTools = [...selectedTools, ...customToolInstances] as typeof selectedTools;
}
// Wrap tools with hook callbacks (built-in and custom)
if (hookRunner) {
selectedTools = wrapToolsWithHooks(selectedTools, hookRunner);
}
// Create agent
const agent = new Agent({
initialState: {