From 6663a4a18c4451e62b82ef9cb8618c2ef7a84f67 Mon Sep 17 00:00:00 2001
From: Alex Hancock <alexhancock@block.xyz>
Date: Wed, 5 Nov 2025 11:40:02 -0500
Subject: [PATCH] fix: make ACP cancellation work

---
 crates/goose-cli/src/commands/acp.rs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/crates/goose-cli/src/commands/acp.rs b/crates/goose-cli/src/commands/acp.rs
index dfdbe269b8da..5694fe87ab5c 100644
--- a/crates/goose-cli/src/commands/acp.rs
+++ b/crates/goose-cli/src/commands/acp.rs
@@ -542,6 +542,13 @@ impl acp::Agent for GooseAcpAgent {
 
         // Create and store cancellation token for this prompt
         let cancel_token = CancellationToken::new();
+        {
+            let mut sessions = self.sessions.lock().await;
+            let session = sessions
+                .get_mut(&session_id)
+                .ok_or_else(acp::Error::invalid_params)?;
+            session.cancel_token = Some(cancel_token.clone());
+        }
 
         let user_message = self.convert_acp_prompt_to_message(args.prompt);