fix(agent): restore write/shell tools on follow-up confirmation turns

is_generation_query() only checked the current user input, so short
follow-up messages like "sure go ahead fix all then" didn't match any
keyword and caused the agent to be built without WriteFileTool,
WriteFilesTool, and ShellTool. The model then tried to call those tools
(they were in conversation history) and got ToolNotFoundError.

Two fixes:
1. Add commonly missing generation verbs to is_generation_query:
   fix, update, add, change, modify, edit, configure, setup, patch,
   install - these clearly imply file modification intent.

2. Track last_was_generation in ChatSession. For short inputs (< 60
   chars) that aren't planning mode, inherit generation mode from the
   previous turn so confirmations like "sure", "yes", "go ahead" keep
   write/shell tools active.

Co-Authored-By: Claude <noreply@anthropic.com>

Author: 2 people

src/agent/mod.rs

@@ -690,8 +690,13 @@ pub async fn run_interactive(
                 Some(&current_input),
                 session.plan_mode,
             );
-            let is_generation = prompts::is_generation_query(&current_input);
             let is_planning = session.plan_mode.is_planning();
+            // Inherit generation mode for short follow-up messages ("sure", "yes", "go ahead",
+            // etc.) so the write/shell tool set is not lost between turns.
+            let is_generation = prompts::is_generation_query(&current_input)
+                || (!is_planning
+                    && session.last_was_generation
+                    && current_input.trim().len() < 60);
 
             // Note: using raw_chat_history directly which preserves Reasoning blocks
             // This is needed for extended thinking to work with multi-turn conversations
@@ -1138,6 +1143,10 @@ pub async fn run_interactive(
                     // Add to conversation history with tool call records
                     conversation_history.add_turn(input.clone(), text.clone(), tool_calls.clone());
 
+                    // Remember whether this turn had generation tools active so short follow-up
+                    // messages ("sure", "go ahead", etc.) don't lose write/shell access.
+                    session.last_was_generation = is_generation;
+
                     // Check if this heavy turn requires immediate compaction
                     // This helps prevent context overflow in subsequent requests
                     if conversation_history.needs_compaction() {

src/agent/prompts/mod.rs

@@ -705,6 +705,18 @@ pub fn is_generation_query(query: &str) -> bool {
         "new feature",
         "develop",
         "code",
+        // Common modification verbs (previously missing)
+        "fix",
+        "update",
+        "add",
+        "change",
+        "modify",
+        "edit",
+        "configure",
+        "setup",
+        "set up",
+        "patch",
+        "install",
         // Plan execution keywords - needed for plan continuation
         "plan",
         "continue",

src/agent/session/mod.rs

@@ -34,6 +34,10 @@ pub struct ChatSession {
     pub token_usage: TokenUsage,
     /// Current planning mode state
     pub plan_mode: PlanMode,
+    /// Whether the previous turn used generation mode (write/shell tools active).
+    /// Used so short follow-up messages ("sure", "go ahead", "yes") inherit the
+    /// tool set from the previous turn instead of losing write/shell access.
+    pub last_was_generation: bool,
     /// Session loaded via /resume command, to be processed by main loop
     pub pending_resume: Option<crate::agent::persistence::ConversationRecord>,
     /// Platform session state (selected project/org context)
@@ -58,6 +62,7 @@ impl ChatSession {
             history: Vec::new(),
             token_usage: TokenUsage::new(),
             plan_mode: PlanMode::default(),
+            last_was_generation: false,
             pending_resume: None,
             platform_session,
         }