Update nested coding env server for sandbox pool sockets

envs/coding_env/server/app.py

@@ -21,6 +21,9 @@ Usage:
     python -m envs.coding_env.server.app
 """
 
+import os
+from contextlib import suppress
+
 from coding_env.models import CodeAction, CodeObservation
 from coding_env.server.python_codeact_env import PythonCodeActEnv
 from openenv.core.env_server import create_app
@@ -30,17 +33,19 @@ from openenv.core.env_server import create_app
 app = create_app(PythonCodeActEnv, CodeAction, CodeObservation, env_name="coding_env")
 
 
-if __name__ == "__main__":
-    import uvicorn
-
-    uvicorn.run(app, host="0.0.0.0", port=8000)
-
-
 def main():
     """Main entry point for running the server."""
     import uvicorn
 
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    port = int(os.environ.get("SBX_SERVICE_PORT", "8000"))
+    if proxy_dir := os.environ.get("SBX_PROXY_DIR"):
+        socket_path = os.path.join(proxy_dir, f"{port}.sock")
+        os.makedirs(proxy_dir, exist_ok=True)
+        with suppress(FileNotFoundError):
+            os.unlink(socket_path)
+        uvicorn.run(app, uds=socket_path)
+    else:
+        uvicorn.run(app, host="0.0.0.0", port=port)
 
 
 if __name__ == "__main__":