From eeefddf6bfe25ccfbd1b521b03f50e068dd7fb5d Mon Sep 17 00:00:00 2001
From: liangxinbing <1580466765@qq.com>
Date: Tue, 11 Mar 2025 13:13:24 +0800
Subject: [PATCH] Bug Fixing based on basy.py Repair from Bob Feng8 Xu

---
 app/agent/base.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/app/agent/base.py b/app/agent/base.py
index 74d6845..43132cf 100644
--- a/app/agent/base.py
+++ b/app/agent/base.py
@@ -144,6 +144,8 @@ class BaseAgent(BaseModel, ABC):
                 results.append(f"Step {self.current_step}: {step_result}")
 
             if self.current_step >= self.max_steps:
+                self.current_step = 0  # setting back to 0 when reached max steps
+                self.state = AgentState.IDLE  # setting the status
                 results.append(f"Terminated: Reached max steps ({self.max_steps})")
 
         return "\n".join(results) if results else "No steps executed"