From da4645a27ab83b49c15703a3a7838ed95aadc523 Mon Sep 17 00:00:00 2001
From: Dalton Flanagan <6599399+dltn@users.noreply.github.com>
Date: Wed, 7 Aug 2024 23:31:30 -0400
Subject: [PATCH] hide non-featured (older) models from model list command
 without show-all flag (#23)

---
 llama_toolchain/cli/model/list.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/llama_toolchain/cli/model/list.py b/llama_toolchain/cli/model/list.py
index c6d4b24ac..e7e1e4054 100644
--- a/llama_toolchain/cli/model/list.py
+++ b/llama_toolchain/cli/model/list.py
@@ -27,7 +27,7 @@ class ModelList(Subcommand):
         self.parser.set_defaults(func=self._run_model_list_cmd)
 
     def _add_arguments(self):
-        pass
+        self.parser.add_argument('--show-all', action='store_true', help='Show all models (not just defaults)')
 
     def _run_model_list_cmd(self, args: argparse.Namespace) -> None:
         headers = [
@@ -39,6 +39,9 @@ class ModelList(Subcommand):
 
         rows = []
         for model in all_registered_models():
+            if not args.show_all and not model.is_featured:
+                continue
+
             req = model.hardware_requirements
 
             descriptor = model.descriptor()