feat: add llama stack rm and llama stack list commands

llama_stack/cli/stack/list_stacks.py

@@ -0,0 +1,70 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the terms described in the LICENSE file in
+# the root directory of this source tree.
+
+import argparse
+from pathlib import Path
+from typing import Dict
+from llama_stack.cli.subcommand import Subcommand
+from llama_stack.cli.table import print_table
+
+
+class StackListBuilds(Subcommand):
+     """List built stacks in .llama/distributions directory"""
+
+     def __init__(self, subparsers: argparse._SubParsersAction):
+        super().__init__()
+        self.parser = subparsers.add_parser(
+            "list",
+            prog="llama stack list",
+            description="list the build stacks",
+            formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+        )
+        self._add_arguments()
+        self.parser.set_defaults(func=self._list_stack_command)
+        
+     def _add_arguments(self):
+        self.parser.add_argument(
+            "--verbose",
+            "-v",
+            action="store_true",
+            help="Show additional details about each stack",
+        )
+
+     def _get_distribution_dirs(self) -> Dict[str, Path]:
+        """Return a dictionary of distribution names and their paths"""
+        distributions = {}
+        dist_dir = Path.home() / ".llama" / "distributions"
+        
+        if dist_dir.exists():
+            for stack_dir in dist_dir.iterdir():
+                if stack_dir.is_dir():
+                    distributions[stack_dir.name] = stack_dir
+        return distributions
+
+     def _list_stack_command(self, args: argparse.Namespace) -> None:
+        distributions = self._get_distribution_dirs()
+        
+        if not distributions:
+            print("No stacks found in ~/.llama/distributions")
+            return
+
+        headers = ["Stack Name", "Path"]
+        if args.verbose:
+            headers.extend(["Build Config", "Run Config"])
+        
+        rows = []
+        for name, path in distributions.items():
+            row = [name, str(path)]
+            
+            if args.verbose:
+                # Check for build and run config files
+                build_config = "Yes" if (path / f"{name}-build.yaml").exists() else "No"
+                run_config = "Yes" if (path / f"{name}-run.yaml").exists() else "No"
+                row.extend([build_config, run_config])
+            
+            rows.append(row)
+
+        print_table(rows, headers, separate_rows=True)

llama_stack/cli/stack/remove.py

@@ -0,0 +1,118 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the terms described in the LICENSE file in
+# the root directory of this source tree.
+
+import argparse
+from pathlib import Path
+import shutil
+from typing import Dict
+
+from llama_stack.cli.subcommand import Subcommand
+from llama_stack.cli.table import print_table
+
+
+class StackRemove(Subcommand):
+     """Remove the build stack"""
+
+     def __init__(self, subparsers: argparse._SubParsersAction):
+        super().__init__()
+        self.parser = subparsers.add_parser(
+            "rm",
+            prog="llama stack rm",
+            description="Remove the build stack",
+            formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+        )
+        self._add_arguments()
+        self.parser.set_defaults(func=self._remove_stack_build_command)
+
+     def _add_arguments(self) -> None:
+        self.parser.add_argument(
+            "name",
+            type=str,
+            nargs="?",
+            help="Name of the stack to delete",
+        )
+        self.parser.add_argument(
+            "--list",
+            "-l",
+            action="store_true",
+            help="List available stacks before deletion",
+        )
+        self.parser.add_argument(
+            "--force",
+            "-f",
+            action="store_true",
+            help="Force deletion without confirmation",
+        )
+        self.parser.add_argument(
+            "--all",
+            "-a",
+            action="store_true",
+            help="Delete all stacks (use with caution)",
+        )
+
+     def _get_distribution_dirs(self) -> Dict[str, Path]:
+        """Return a dictionary of distribution names and their paths"""
+        distributions = {}
+        dist_dir = Path.home() / ".llama" / "distributions"
+        
+        if dist_dir.exists():
+            for stack_dir in dist_dir.iterdir():
+                if stack_dir.is_dir():
+                    distributions[stack_dir.name] = stack_dir
+        return distributions
+
+     def _list_stacks(self) -> None:
+        """Display available stacks in a table"""
+        distributions = self._get_distribution_dirs()
+        if not distributions:
+            print("No stacks found in ~/.llama/distributions")
+            return
+
+        headers = ["Stack Name", "Path"]
+        rows = [[name, str(path)] for name, path in distributions.items()]
+        print_table(rows, headers, separate_rows=True)
+
+     def _remove_stack_build_command(self, args: argparse.Namespace) -> None:
+        distributions = self._get_distribution_dirs()
+        
+        if args.all:
+            if not args.force:
+                confirm = input("Are you sure you want to delete ALL stacks? [y/N] ").lower()
+                if confirm != 'y':
+                    print("Deletion cancelled.")
+                    return
+            
+            for name, path in distributions.items():
+                try:
+                    shutil.rmtree(path)
+                    print(f"Deleted stack: {name}")
+                except Exception as e:
+                    print(f"Failed to delete stack {name}: {e}")
+            return
+
+        if args.list or not args.name:
+            self._list_stacks()
+            if not args.name:
+                return
+
+        if args.name not in distributions:
+            self._list_stacks()
+            print(f"Stack not found: {args.name}")
+            return
+
+        stack_path = distributions[args.name]
+        
+        if not args.force:
+            confirm = input(f"Are you sure you want to delete stack '{args.name}'? [y/N] ").lower()
+            if confirm != 'y':
+                print("Deletion cancelled.")
+                return
+
+        try:
+            shutil.rmtree(stack_path)
+            print(f"Successfully deleted stack: {args.name}")
+        except Exception as e:
+            print(f"Failed to delete stack {args.name}: {e}")

llama_stack/cli/stack/stack.py

@@ -7,6 +7,7 @@
 import argparse
 from importlib.metadata import version
 
+from llama_stack.cli.stack.list_stacks import StackListBuilds
 from llama_stack.cli.stack.utils import print_subcommand_description
 from llama_stack.cli.subcommand import Subcommand
 
@@ -14,6 +15,7 @@ from .build import StackBuild
 from .list_apis import StackListApis
 from .list_providers import StackListProviders
 from .run import StackRun
+from .remove import StackRemove
 
 
 class StackParser(Subcommand):
@@ -41,5 +43,6 @@ class StackParser(Subcommand):
         StackListApis.create(subparsers)
         StackListProviders.create(subparsers)
         StackRun.create(subparsers)
-
+        StackRemove.create(subparsers)
+        StackListBuilds.create(subparsers)
         print_subcommand_description(self.parser, subparsers)