1 files changed, 5 insertions, 2 deletions
diff --git a/packages/cli/src/ui/hooks/useToolScheduler.ts b/packages/cli/src/ui/hooks/useToolScheduler.ts
index f1eee9fd..7d8cfbe4 100644
--- a/packages/cli/src/ui/hooks/useToolScheduler.ts
+++ b/packages/cli/src/ui/hooks/useToolScheduler.ts
@@ -142,7 +142,10 @@ export function useToolScheduler(
 
         const { request: r, tool } = initialCall;
         try {
-          const userApproval = await tool.shouldConfirmExecute(r.args);
+          const userApproval = await tool.shouldConfirmExecute(
+            r.args,
+            abortController.signal,
+          );
           if (userApproval) {
             // Confirmation is needed. Update status to 'awaiting_approval'.
             setToolCalls(
@@ -183,7 +186,7 @@ export function useToolScheduler(
         }
       });
     },
-    [isRunning, setToolCalls, toolRegistry],
+    [isRunning, setToolCalls, toolRegistry, abortController.signal],
   );
 
   const cancel = useCallback(