From 36fa549fecd2195e897bd6f1fe6b8dd653e23e6e Mon Sep 17 00:00:00 2001
From: Dmytro Bogovych <dmytro.bogovych@gmail.com>
Date: Tue, 5 Sep 2023 16:35:25 +0300
Subject: [PATCH] - answerer follows the task limit

---
 src/agent_gsm.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/agent_gsm.py b/src/agent_gsm.py
index 1536a23..aa35c69 100644
--- a/src/agent_gsm.py
+++ b/src/agent_gsm.py
@@ -263,7 +263,8 @@ def perform_answerer():
     
     # Setup analyzer script
     # Run answering script
-    while True:
+    attempt_idx = 0
+    while attempt_idx < CONFIG.TaskLimit or CONFIG.TaskLimit is None or CONFIG.TaskLimit == 0:
         # Remove old recording
         if CONFIG.RecordFile.exists():
             os.remove(CONFIG.RecordFile)