fix prompt with name args (meta-llama#80)

togethercomputer · Sep 19, 2024 · 6302a1e · 6302a1e
1 parent c63d6cb
commit 6302a1e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/llama_stack/cli/stack/build.py b/llama_stack/cli/stack/build.py
@@ -191,7 +191,13 @@ def _run_stack_build_command(self, args: argparse.Namespace) -> None:
             return
 
         if not args.config and not args.template:
-            name = prompt("> Enter a name for your Llama Stack (e.g. my-local-stack): ")
+            if not args.name:
+                name = prompt(
+                    "> Enter a name for your Llama Stack (e.g. my-local-stack): "
+                )
+            else:
+                name = args.name
+
             image_type = prompt(
                 "> Enter the image type you want your Llama Stack to be built as (docker or conda): ",
                 validator=Validator.from_callable(