fix: send delete-forever request in client

lablup · Mar 7, 2024 · c17abc8 · c17abc8
1 parent c9e4c4f
commit c17abc8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/ai/backend/client/cli/vfolder.py b/src/ai/backend/client/cli/vfolder.py
@@ -148,9 +148,9 @@ def create(name, host, group, host_path, usage_mode, permission, cloneable):
             sys.exit(ExitCode.FAILURE)
 
 
-@vfolder.command()
+@vfolder.command(aliases=["move-to-trash"])
 @click.argument("name", type=str)
-def move_to_trash(name):
+def trash(name):
     """Move the given virtual folder to trash-bin. The virtual folder will be under `delete-pending` status.
     This operation can be retracted by
     calling `restore()`.

diff --git a/src/ai/backend/client/func/vfolder.py b/src/ai/backend/client/func/vfolder.py
@@ -292,7 +292,8 @@ async def delete_forever(self) -> Mapping[str, Any]:
         rqst.set_json({
             "id": self.id.hex,
         })
-        return {}
+        async with rqst.fetch():
+            return {}
 
     @api_function
     async def rename(self, new_name):