From 987e651755f97d68b05d2997fcff3cdaffaf6522 Mon Sep 17 00:00:00 2001 From: Yuan Tang Date: Sun, 22 Dec 2024 00:10:13 -0500 Subject: [PATCH] Add missing venv option in --image-type (#677) "venv" option is supported but not mentioned in the prompt. Signed-off-by: Yuan Tang --- llama_stack/cli/stack/build.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llama_stack/cli/stack/build.py b/llama_stack/cli/stack/build.py index 0cb873b57..f18d262c0 100644 --- a/llama_stack/cli/stack/build.py +++ b/llama_stack/cli/stack/build.py @@ -100,7 +100,7 @@ class StackBuild(Subcommand): build_config.image_type = args.image_type else: self.parser.error( - f"Please specify a image-type (docker | conda) for {args.template}" + f"Please specify a image-type (docker | conda | venv) for {args.template}" ) self._run_stack_build_command_from_build_config( build_config, template_name=args.template @@ -122,7 +122,7 @@ class StackBuild(Subcommand): ) image_type = prompt( - "> Enter the image type you want your Llama Stack to be built as (docker or conda): ", + "> Enter the image type you want your Llama Stack to be built as (docker or conda or venv): ", validator=Validator.from_callable( lambda x: x in ["docker", "conda", "venv"], error_message="Invalid image type, please enter conda or docker or venv",