(fix) Update logs after run_action (EventStreamRuntime) (#3870)

All-Hands-AI · Sep 15, 2024 · ecf4aed · ecf4aed
1 parent a97ef34
commit ecf4aed
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/openhands/runtime/client/runtime.py b/openhands/runtime/client/runtime.py
@@ -426,7 +426,6 @@ def run_action(self, action: Action) -> Observation:
                     output = response.json()
                     obs = observation_from_dict(output)
                     obs._cause = action.id  # type: ignore[attr-defined]
-                    return obs
                 else:
                     error_message = response.text
                     logger.error(f'Error from server: {error_message}')
@@ -437,6 +436,8 @@ def run_action(self, action: Action) -> Observation:
             except Exception as e:
                 logger.error(f'Error during command execution: {e}')
                 obs = ErrorObservation(f'Command execution failed: {str(e)}')
+            # Refresh docker logs
+            self._wait_until_alive()
             return obs
 
     def run(self, action: CmdRunAction) -> Observation: