fix agent state

Author: enyst

opendevin/controller/agent_controller.py

@@ -221,13 +221,6 @@ async def set_agent_state_to(self, new_state: AgentState):
 
     def get_agent_state(self):
         """Returns the current state of the agent task."""
-        if self.delegate is not None:
-            agent_state = self.delegate.get_agent_state()
-            if agent_state not in (AgentState.FINISHED, AgentState.STOPPED):
-                return agent_state
-            else:
-                # then we return parent state
-                return self.state.agent_state
         return self.state.agent_state
 
     async def start_delegate(self, action: AgentDelegateAction):

opendevin/core/main.py

@@ -123,7 +123,7 @@ async def on_event(event: Event):
                 event_stream.add_event(action, EventSource.USER)
 
     event_stream.subscribe(EventStreamSubscriber.MAIN, on_event)
-    while controller.get_agent_state() not in [
+    while controller.state.agent_state not in [
         AgentState.FINISHED,
         AgentState.REJECTED,
         AgentState.ERROR,