Add missing venv option in --image-type (#677)

"venv" option is supported but not mentioned in the prompt.

Signed-off-by: Yuan Tang <terrytangyuan@gmail.com>
This commit is contained in:
Yuan Tang 2024-12-22 00:10:13 -05:00 committed by GitHub
parent bae197c37e
commit 987e651755
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -100,7 +100,7 @@ class StackBuild(Subcommand):
build_config.image_type = args.image_type build_config.image_type = args.image_type
else: else:
self.parser.error( 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( self._run_stack_build_command_from_build_config(
build_config, template_name=args.template build_config, template_name=args.template
@ -122,7 +122,7 @@ class StackBuild(Subcommand):
) )
image_type = prompt( 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( validator=Validator.from_callable(
lambda x: x in ["docker", "conda", "venv"], lambda x: x in ["docker", "conda", "venv"],
error_message="Invalid image type, please enter conda or docker or venv", error_message="Invalid image type, please enter conda or docker or venv",