phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix configure script to work with config file

Browse source
This commit is contained in:
Xi Yan 2024-09-10 11:15:38 -07:00
parent 9d5582245d
commit f05347be8f
1 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

9
llama_toolchain/cli/stack/configure.py
View file

@ -63,17 +63,18 @@ class StackConfigure(Subcommand):
from llama_toolchain.core.package import BuildType
if args.config_file:
with open(args.config_file, "r") as f:
build_config = BuildConfig(**yaml.safe_load(f))
build_type = BuildType(build_config.package_type)
distribution = build_config.distribution
name = build_config.name
else:
build_type = BuildType(args.package_type)
name = args.name
distribution = args.distribution
config_file = (
BUILDS_BASE_DIR
/ args.distribution
/ build_type.descriptor()
/ f"{name}.yaml"
BUILDS_BASE_DIR / distribution / build_type.descriptor() / f"{name}.yaml"
)
if not config_file.exists():
self.parser.error(