Add missing update functions to rust and python sdks.

Author: calavera

crates/cloud-sdk/src/sandboxes/mod.rs

@@ -14,7 +14,7 @@ use models::{
     DaemonInfo, HealthResponse, ListDirectoryResponse, ListProcessesResponse,
     ListSandboxPoolsResponse, ListSandboxesResponse, ListSnapshotsResponse, OutputEvent,
     OutputResponse, ProcessInfo, SandboxInfo, SandboxPoolInfo, SandboxPoolRequest,
-    SendSignalResponse, SnapshotInfo,
+    SendSignalResponse, SnapshotInfo, UpdateSandboxRequest,
 };
 
 /// A client for managing sandbox lifecycle, pool, and snapshot APIs.
@@ -91,6 +91,19 @@ impl SandboxesClient {
         Ok(list.sandboxes)
     }
 
+    pub async fn update(
+        &self,
+        sandbox_id: &str,
+        request: &UpdateSandboxRequest,
+    ) -> Result<SandboxInfo, SdkError> {
+        let uri = self.endpoint(&format!("sandboxes/{sandbox_id}"));
+        let req = self
+            .client
+            .build_post_json_request(Method::PATCH, &uri, request)?;
+        let resp = self.client.execute(req).await?;
+        Self::parse_json(resp).await
+    }
+
     pub async fn delete(&self, sandbox_id: &str) -> Result<(), SdkError> {
         let uri = self.endpoint(&format!("sandboxes/{sandbox_id}"));
         let req = self.client.request(Method::DELETE, &uri).build()?;

crates/cloud-sdk/src/sandboxes/models.rs

@@ -42,6 +42,12 @@ pub struct CreateSandboxRequest {
     pub name: Option<String>,
 }
 
+#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
+pub struct UpdateSandboxRequest {
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub name: Option<String>,
+}
+
 #[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
 pub struct SandboxPoolRequest {
     #[serde(skip_serializing_if = "Option::is_none")]

crates/rust-cloud-sdk-py/src/lib.rs

@@ -22,7 +22,9 @@ use tensorlake_cloud_sdk::images::ImagesClient;
 use tensorlake_cloud_sdk::images::models::{
     ApplicationBuildContext, CreateApplicationBuildRequest,
 };
-use tensorlake_cloud_sdk::sandboxes::models::{CreateSandboxRequest, SandboxPoolRequest};
+use tensorlake_cloud_sdk::sandboxes::models::{
+    CreateSandboxRequest, SandboxPoolRequest, UpdateSandboxRequest,
+};
 use tensorlake_cloud_sdk::sandboxes::{SandboxProxyClient, SandboxesClient};
 use tensorlake_cloud_sdk::{Client, ClientBuilder, error::SdkError};
 use tokio::runtime::Runtime;
@@ -611,6 +613,18 @@ impl CloudSandboxClient {
         })
     }
 
+    fn update_sandbox(&self, sandbox_id: String, request_json: String) -> PyResult<String> {
+        let request: UpdateSandboxRequest = parse_json_payload(&request_json)?;
+        self.run_with_retry(5, move |client| {
+            let sandbox_id = sandbox_id.clone();
+            let request = request.clone();
+            async move {
+                let response = client.update(&sandbox_id, &request).await?;
+                Ok(serde_json::to_string(&response)?)
+            }
+        })
+    }
+
     fn delete_sandbox(&self, sandbox_id: String) -> PyResult<()> {
         self.run_with_retry(5, move |client| {
             let sandbox_id = sandbox_id.clone();

src/tensorlake/sandbox/client.py

@@ -31,6 +31,7 @@
     SandboxStatus,
     SnapshotInfo,
     SnapshotStatus,
+    UpdateSandboxRequest,
 )
 
 try:
@@ -364,6 +365,36 @@ def list(self) -> list[SandboxInfo]:
         except Exception as e:
             _raise_as_sandbox_error(e)
 
+    def update_sandbox(self, sandbox_id: str, name: str) -> SandboxInfo:
+        """Update a sandbox's properties.
+
+        Currently supports updating the sandbox name. Naming an ephemeral sandbox
+        makes it non-ephemeral and enables suspend/resume.
+
+        Args:
+            sandbox_id: ID of the sandbox to update
+            name: New name for the sandbox
+
+        Returns:
+            SandboxInfo with updated sandbox details
+
+        Raises:
+            SandboxNotFoundError: If sandbox doesn't exist
+            RemoteAPIError: If the API request fails
+            SandboxConnectionError: If the server is unreachable
+        """
+        request = UpdateSandboxRequest(name=name)
+        try:
+            response_json = self._rust_client.update_sandbox(
+                sandbox_id=sandbox_id,
+                request_json=request.model_dump_json(exclude_none=True),
+            )
+            return SandboxInfo.model_validate_json(response_json)
+        except Exception as e:
+            if _rust_status_code(e) == 404:
+                raise SandboxNotFoundError(sandbox_id) from None
+            _raise_as_sandbox_error(e)
+
     def delete(self, sandbox_id: str) -> None:
         """Terminate a sandbox.
 

src/tensorlake/sandbox/models.py

@@ -108,6 +108,12 @@ class CreateSandboxRequest(BaseModel):
     name: str | None = None
 
 
+class UpdateSandboxRequest(BaseModel):
+    """Request payload for updating a sandbox."""
+
+    name: str | None = None
+
+
 class SandboxPoolRequest(BaseModel):
     """Request payload for creating or updating a sandbox pool."""