feat: limit to 3 retries in a row

PeronGH · Jan 7, 2024 · 537d3bb · 537d3bb
1 parent 0ac734a
commit 537d3bb
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/controller.dart b/lib/controller.dart
@@ -87,8 +87,12 @@ class Controller extends GetxController {
     await _submit(userPrompt);
 
     // handle message revoke
-    while (messages.last.type == Message.typeError &&
-        messages.last.content == Message.messageRevoke) {
+    for (var retries = 0;
+        retries < 3 &&
+            messages.isNotEmpty &&
+            messages.last.type == Message.typeError &&
+            messages.last.content == Message.messageRevoke;
+        retries++) {
       await _submit(Message.continueFromRevokeMessage);
     }