Fix issue #5661: [Bug]: SocketIO event regression, AgentStateChangeObservation is double emitted and emitted out of sequence

openhands/server/listen_socket.py

@@ -1,5 +1,6 @@
 from openhands.core.logger import openhands_logger as logger
 from openhands.core.schema.action import ActionType
+from openhands.core.schema.agent import AgentState
 from openhands.events.action import (
     NullAction,
 )
@@ -76,11 +77,9 @@ async def init_connection(
         ):
             continue
         elif isinstance(event, AgentStateChangedObservation):
-            if event.agent_state == 'init':
+            if event.agent_state == AgentState.INIT:
                 await sio.emit('oh_event', event_to_dict(event), to=connection_id)
-            else:
-                agent_state_changed = event
-                continue
+            agent_state_changed = event
         await sio.emit('oh_event', event_to_dict(event), to=connection_id)
     if agent_state_changed:
         await sio.emit('oh_event', event_to_dict(agent_state_changed), to=connection_id)

pyproject.toml

@@ -100,6 +100,7 @@ reportlab = "*"
 [tool.coverage.run]
 concurrency = ["gevent"]
 
+
 [tool.poetry.group.runtime.dependencies]
 jupyterlab = "*"
 notebook = "*"
@@ -129,6 +130,7 @@ ignore = ["D1"]
 [tool.ruff.lint.pydocstyle]
 convention = "google"
 
+
 [tool.poetry.group.evaluation.dependencies]
 streamlit = "*"
 whatthepatch = "*"