fix(Runtime): Wait for container to start up (All-Hands-AI#7548)

Co-authored-by: openhands <[email protected]>
Co-authored-by: tofarr <[email protected]>

Author: 3 people

openhands/runtime/impl/docker/docker_runtime.py

@@ -37,6 +37,16 @@
 APP_PORT_RANGE_2 = (55000, 59999)
 
 
+def _is_retryable_wait_until_alive_error(exception):
+    if isinstance(exception, tenacity.RetryError):
+        cause = exception.last_attempt.exception()
+        return _is_retryable_wait_until_alive_error(cause)
+
+    return isinstance(
+        exception, (ConnectionError, httpx.NetworkError, httpx.RemoteProtocolError)
+    )
+
+
 class DockerRuntime(ActionExecutionClient):
     """This runtime will subscribe the event stream.
 
@@ -355,7 +365,7 @@ def _attach_to_container(self):
 
     @tenacity.retry(
         stop=tenacity.stop_after_delay(120) | stop_if_should_exit(),
-        retry=tenacity.retry_if_exception_type((ConnectionError, httpx.NetworkError)),
+        retry=tenacity.retry_if_exception(_is_retryable_wait_until_alive_error),
         reraise=True,
         wait=tenacity.wait_fixed(2),
     )