From 4cbeb59bec17a71bc8bffc1d677477e2bb19499a Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Sat, 8 Jun 2024 13:38:18 -0700
Subject: [PATCH] browser ver should stop after 3 reflections

---
 aider/gui.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/aider/gui.py b/aider/gui.py
index 2776a477d..16a13a004 100755
--- a/aider/gui.py
+++ b/aider/gui.py
@@ -411,14 +411,22 @@ class GUI:
         prompt = self.state.prompt
         self.state.prompt = None
 
+        # This duplicates logic from within Coder
+        self.num_reflections = 0
+        self.max_reflections = 3
+
         while prompt:
             with self.messages.chat_message("assistant"):
                 res = st.write_stream(self.coder.run_stream(prompt))
                 self.state.messages.append({"role": "assistant", "content": res})
                 # self.cost()
+
+            prompt = None
             if self.coder.reflected_message:
-                self.info(self.coder.reflected_message)
-            prompt = self.coder.reflected_message
+                if self.num_reflections < self.max_reflections:
+                    self.num_reflections += 1
+                    self.info(self.coder.reflected_message)
+                    prompt = self.coder.reflected_message
 
         with self.messages:
             edit = dict(