fix test case: add rock auth to sandbox info

Author: FangwenDave

rock/sandbox/sandbox_manager.py

@@ -149,13 +149,6 @@ async def create_session(self, request: CreateSessionRequest) -> CreateBashSessi
         await self._update_expire_time(request.sandbox_id)
         return await self._deployment_service.async_ray_get(sandbox_actor.create_session.remote(request))
 
-    async def execute(self, command: Command) -> CommandResponse:
-        sandbox_actor = await self._deployment_service.async_ray_get_actor(command.sandbox_id)
-        if sandbox_actor is None:
-            raise Exception(f"sandbox {command.sandbox_id} not found to execute")
-        await self._update_expire_time(command.sandbox_id)
-        return await self._deployment_service.async_ray_get(sandbox_actor.execute.remote(command))
-
     @monitor_sandbox_operation()
     async def run_in_session(self, action: Action) -> BashObservation:
         sandbox_actor = await self._deployment_service.async_ray_get_actor(action.sandbox_id)
@@ -164,6 +157,13 @@ async def run_in_session(self, action: Action) -> BashObservation:
         await self._update_expire_time(action.sandbox_id)
         return await self._deployment_service.async_ray_get(sandbox_actor.run_in_session.remote(action))
 
+    async def execute(self, command: Command) -> CommandResponse:
+        sandbox_actor = await self._deployment_service.async_ray_get_actor(command.sandbox_id)
+        if sandbox_actor is None:
+            raise Exception(f"sandbox {command.sandbox_id} not found to execute")
+        await self._update_expire_time(command.sandbox_id)
+        return await self._deployment_service.async_ray_get(sandbox_actor.execute.remote(command))
+
     async def _is_expired(self, sandbox_id):
         timeout_dict = await self._redis_provider.json_get(timeout_sandbox_key(sandbox_id), "$")
         if timeout_dict is None or len(timeout_dict) == 0:

rock/sandbox/service/deployment_service.py

@@ -92,6 +92,7 @@ async def submit(self, config: DockerDeploymentConfig, user_info: dict) -> Sandb
         user_id = user_info.get("user_id", "default")
         experiment_id = user_info.get("experiment_id", "default")
         namespace = user_info.get("namespace", "default")
+        rock_authorization = user_info.get("rock_authorization", "default")
         sandbox_actor.start.remote()
         sandbox_actor.set_user_id.remote(user_id)
         sandbox_actor.set_experiment_id.remote(experiment_id)
@@ -101,6 +102,7 @@ async def submit(self, config: DockerDeploymentConfig, user_info: dict) -> Sandb
         sandbox_info["experiment_id"] = experiment_id
         sandbox_info["namespace"] = namespace
         sandbox_info["state"] = State.PENDING
+        sandbox_info["rock_authorization"] = rock_authorization
         return sandbox_info
 
     async def creator_actor(self, config: DockerDeploymentConfig):