FIX: cancel functionality regressed

lib/ai_bot/playground.rb

@@ -461,6 +461,7 @@ def reply_to(post, custom_instructions: nil, &blk)
             if stream_reply && !Discourse.redis.get(redis_stream_key)
               cancel&.call
               reply_post.update!(raw: reply, cooked: PrettyText.cook(reply))
+              break
             end
 
             if post_streamer

lib/completions/endpoints/base.rb

@@ -139,15 +139,20 @@ def perform_completion!(
 
               begin
                 cancelled = false
-                cancel = -> { cancelled = true }
-                if cancelled
+                cancel = -> do
+                  cancelled = true
                   http.finish
-                  break
                 end
 
+                break if cancelled
+
                 response.read_body do |chunk|
+                  break if cancelled
+
                   response_raw << chunk
+
                   decode_chunk(chunk).each do |partial|
+                    break if cancelled
                     partials_raw << partial.to_s
                     response_data << partial if partial.is_a?(String)
                     partials = [partial]