loop fix

ControllerExecutorFlow.py

@@ -105,7 +105,6 @@ class ControllerExecutorFlow(CompositeFlow):
     def _on_reach_max_round(self):
         """ This method is called when the flow reaches the maximum amount of rounds. It updates the state of the flow and starts the process of terminating the flow."""
         self._state_update_dict({
-            "EARLY_EXIT": False,
             "answer": "The maximum amount of rounds was reached before the model found an answer.",
             "status": "unfinished"
         })
@@ -180,12 +179,12 @@ class ControllerExecutorFlow(CompositeFlow):
                 
                 self._state_update_dict(
                     {
-                        "EARLY_EXIT": True,
+                        "early_exit_flag": True,
                         "answer": self.flow_state["command_args"]["answer"],
                         "status": "finished"
                     }
                 )
-                self.flow_state["early_exit_flag"] = True
+              
                 
             
         
@@ -214,5 +213,9 @@ class ControllerExecutorFlow(CompositeFlow):
             self._on_reach_max_round()
             self.generate_reply()
         
-        self.flow_state["last_state"] = current_state
+        
+        if self.flow_state.get("early_exit_flag",False) or current_state is None:
+            self.flow_state["last_state"] = None
+        else:
+            self.flow_state["last_state"] = current_state