phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 19:04:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix validator of "container" image type (#901)

Browse source 
This was missed in https://github.com/meta-llama/llama-stack/pull/802
somehow.
This commit is contained in:
Yuan Tang 2025-01-29 12:36:52 -05:00 • committed by GitHub
parent 11b1cdf31d
commit 53721e91ad
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
llama_stack/cli/stack/_build.py
View file

@ -74,7 +74,7 @@ def run_stack_build_command(
build_config.image_type = args.image_type build_config.image_type = args.image_type
else: else:
cprint( cprint(
f"Please specify a image-type (docker | conda | venv) for {args.template}", f"Please specify a image-type (container | conda | venv) for {args.template}",
color="red", color="red",
) )
return return
@ -95,10 +95,10 @@ def run_stack_build_command(
) )
image_type = prompt( image_type = prompt(
"> Enter the image type you want your Llama Stack to be built as (docker or conda or venv): ", "> Enter the image type you want your Llama Stack to be built as (container or conda or venv): ",
validator=Validator.from_callable( validator=Validator.from_callable(
lambda x: x in ["docker", "conda", "venv"], lambda x: x in ["container", "conda", "venv"],
error_message="Invalid image type, please enter conda or docker or venv", error_message="Invalid image type, please enter conda or container or venv",
), ),
default="conda", default="conda",
) )
@ -259,7 +259,7 @@ def _run_stack_build_command_from_build_config(
else: else:
if not image_name: if not image_name:
raise ValueError( raise ValueError(
"Please specify an image name when building a docker image without a template" "Please specify an image name when building a container image without a template"
) )
elif build_config.image_type == ImageType.conda.value: elif build_config.image_type == ImageType.conda.value:
if not image_name: if not image_name: