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

feat: --image-type argument overrides value in --config build.yaml (#2179)

Browse source 
closes #2162

# test plan

run `llama stack build --image-name ollama --image-type
<venv/conda/container> --config llama_stack/templates/ollama/build.yaml`
and verify venv | conda | container are built.
This commit is contained in:
Matthew Farrellee 2025-05-16 17:45:41 -04:00 • committed by GitHub
parent f02f7b28c1
commit f40693e720
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -206,6 +206,8 @@ def run_stack_build_command(args: argparse.Namespace) -> None:
contents = yaml.safe_load(f) contents = yaml.safe_load(f)
contents = replace_env_vars(contents) contents = replace_env_vars(contents)
build_config = BuildConfig(**contents) build_config = BuildConfig(**contents)
if args.image_type:
build_config.image_type = args.image_type
except Exception as e: except Exception as e:
cprint( cprint(
f"Could not parse config file {args.config}: {e}", f"Could not parse config file {args.config}: {e}",

2
llama_stack/cli/stack/build.py
View file

@ -49,7 +49,7 @@ class StackBuild(Subcommand):
type=str, type=str,
help="Image Type to use for the build. If not specified, will use the image type from the template config.", help="Image Type to use for the build. If not specified, will use the image type from the template config.",
choices=[e.value for e in ImageType], choices=[e.value for e in ImageType],
default=ImageType.CONDA.value, default=None, # no default so we can detect if a user specified --image-type and override image_type in the config
) )
self.parser.add_argument( self.parser.add_argument(