diff --git a/crates/op-host-web/src/widget_host/press.rs b/crates/op-host-web/src/widget_host/press.rs
index ba8ddac6b..f9331eb9a 100644
--- a/crates/op-host-web/src/widget_host/press.rs
+++ b/crates/op-host-web/src/widget_host/press.rs
@@ -608,6 +608,11 @@ impl WidgetHost {
                         self.mark_dirty();
                         return true;
                     }
+                    AIChatHit::Stop => {
+                        self.editor_state.chat.stop_streaming();
+                        self.mark_dirty();
+                        return true;
+                    }
                     AIChatHit::Example(text) => {
                         self.editor_state.chat.input = text;
                         self.editor_state.chat.focused = true;