From 8dec33f5f9a972a38978b5600fde5bc292fb6401 Mon Sep 17 00:00:00 2001
From: Michelle Tilley <michelle@michelletilley.net>
Date: Tue, 21 Apr 2026 15:30:46 -0700
Subject: feat: shell tool execution timeouts (#3437)

---
 crates/atuin-ai/src/fsm/effects.rs | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

(limited to 'crates/atuin-ai/src/fsm/effects.rs')

diff --git a/crates/atuin-ai/src/fsm/effects.rs b/crates/atuin-ai/src/fsm/effects.rs
index ede72a42..306f1401 100644
--- a/crates/atuin-ai/src/fsm/effects.rs
+++ b/crates/atuin-ai/src/fsm/effects.rs
@@ -61,14 +61,27 @@ pub(crate) enum Effect {
     ArchiveSession,
 
     // ─── Timers ─────────────────────────────────────────────────
-    /// Schedule a timer that will fire ConfirmationTimeout after delay.
-    ScheduleTimeout { timeout_id: u64, duration: Duration },
+    /// Schedule a timer that fires an event after the given delay.
+    ScheduleTimeout {
+        timeout_id: u64,
+        duration: Duration,
+        kind: TimeoutKind,
+    },
 
     // ─── Exit ───────────────────────────────────────────────────
     /// Exit the application with the given action.
     ExitApp(ExitAction),
 }
 
+/// What kind of timeout was scheduled.
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub(crate) enum TimeoutKind {
+    /// Dangerous command confirmation dialog auto-dismiss.
+    Confirmation,
+    /// Shell tool execution timeout — abort the tool if it's still running.
+    ToolExecution { tool_id: String },
+}
+
 /// What to do when exiting the TUI.
 #[derive(Debug, Clone, PartialEq, Eq)]
 pub(crate) enum ExitAction {
-- 
cgit v1.3.1