refactor: remove redundant DockerDeploymentConfig imports and usage. resolved [FEATURE]

rock/admin/entrypoints/sandbox_api.py

@@ -23,7 +23,6 @@
     SandboxWriteFileRequest,
 )
 from rock.admin.proto.response import SandboxStartResponse
-from rock.deployments.config import DockerDeploymentConfig
 from rock.sandbox.sandbox_manager import SandboxManager
 from rock.utils import handle_exceptions
 
@@ -39,7 +38,7 @@ def set_sandbox_manager(service: SandboxManager):
 @sandbox_router.post("/start")
 @handle_exceptions(error_message="start sandbox failed")
 async def start(config: SandboxStartRequest) -> RockResponse[SandboxStartResponse]:
-    sandbox_start_response = await sandbox_manager.start(DockerDeploymentConfig(**config.transform().model_dump()))
+    sandbox_start_response = await sandbox_manager.start(config.transform())
     return RockResponse(result=sandbox_start_response)
 
 
@@ -51,7 +50,7 @@ async def start_async(
     x_experiment_id: str | None = Header(default="default", alias="X-Experiment-Id"),
 ) -> RockResponse[SandboxStartResponse]:
     sandbox_start_response = await sandbox_manager.start_async(
-        DockerDeploymentConfig(**config.transform().model_dump()),
+        config.transform(),
         user_info={"user_id": x_user_id, "experiment_id": x_experiment_id},
     )
     return RockResponse(result=sandbox_start_response)

rock/admin/proto/request.py

@@ -26,9 +26,9 @@ class SandboxStartRequest(BaseModel):
     """The amount of CPUs to allocate for the container."""
 
     def transform(self):
-        from rock.rocklet.proto.request import InitDockerEnvRequest
+        from rock.deployments.config import DockerDeploymentConfig
 
-        res = InitDockerEnvRequest(**self.model_dump())
+        res = DockerDeploymentConfig(**self.model_dump())
         return res
 
 

rock/rocklet/proto/request.py

@@ -9,19 +9,10 @@
     SandboxCommand,
     SandboxCreateBashSessionRequest,
     SandboxReadFileRequest,
-    SandboxStartRequest,
     SandboxWriteFileRequest,
 )
 
 
-class InitDockerEnvRequest(SandboxStartRequest):
-    python_standalone_dir: str | None = None
-    """Directory path for the Python standalone installation."""
-
-    container_name: str | None = None
-    """Custom name for the container. If None, a random name will be generated."""
-
-
 class InternalCommand(SandboxCommand):
     container_name: str | None = None