handle rocklet timeout (#375)

Author: zhongwen666

pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
 authors = [{ name = "chatos@alibaba" }]
 requires-python = "<4.0,>=3.10"
 name = "rl-rock"
-version = "1.0.2"
+version = "1.0.3"
 description = "ROCK-Reinforcement Open Construction Kit"
 readme = "README.md"
 dependencies = [

rock/sandbox/service/sandbox_proxy_service.py

@@ -8,6 +8,7 @@
 from aliyunsdkcore import client
 from aliyunsdkcore.request import CommonRequest
 from fastapi import UploadFile
+from starlette.status import HTTP_504_GATEWAY_TIMEOUT
 
 from rock import env_vars
 from rock.actions import (
@@ -269,6 +270,8 @@ async def _send_request(
             )
             if response.status_code == 511:
                 return {"exit_code": -1, "failure_reason": response.json()["rockletexception"]["message"]}
+            if response.status_code == HTTP_504_GATEWAY_TIMEOUT:
+                return {"exit_code": -1, "failure_reason": response.json()["detail"]}
             return response.json()
         except httpx.RequestError as e:
             # Handle network-level errors, such as DNS resolution failure, connection timeout, etc.