restore logging of user messages when using cli (OpenHands#3145)

enyst · web-flow · commit f07280153aaf · 2024-07-27T12:58:23.000Z
diff --git a/opendevin/controller/agent_controller.py b/opendevin/controller/agent_controller.py
@@ -161,6 +161,10 @@ async def on_event(self, event: Event):
             await self.set_agent_state_to(event.agent_state)  # type: ignore
         elif isinstance(event, MessageAction):
             if event.source == EventSource.USER:
+                logger.info(
+                    event,
+                    extra={'msg_type': 'ACTION', 'event_source': EventSource.USER},
+                )
                 if self.get_agent_state() != AgentState.RUNNING:
                     await self.set_agent_state_to(AgentState.RUNNING)
             elif event.source == EventSource.AGENT and event.wait_for_response:
diff --git a/opendevin/server/session/session.py b/opendevin/server/session/session.py
@@ -10,7 +10,7 @@
 from opendevin.core.schema import AgentState
 from opendevin.core.schema.action import ActionType
 from opendevin.core.schema.config import ConfigType
-from opendevin.events.action import Action, ChangeAgentStateAction, NullAction
+from opendevin.events.action import ChangeAgentStateAction, NullAction
 from opendevin.events.event import Event, EventSource
 from opendevin.events.observation import (
     AgentStateChangedObservation,
@@ -146,10 +146,6 @@ async def dispatch(self, data: dict):
             return
         event = event_from_dict(data.copy())
         self.agent_session.event_stream.add_event(event, EventSource.USER)
-        if isinstance(event, Action):
-            logger.info(
-                event, extra={'msg_type': 'ACTION', 'event_source': EventSource.USER}
-            )
 
     async def send(self, data: dict[str, object]) -> bool:
         try:
