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

remove distribution types!

Browse source
This commit is contained in:
Xi Yan 2024-09-16 10:12:45 -07:00
parent e466ec389b
commit c0c5839361
17 changed files with 17 additions and 98 deletions
Download patch file Download diff file Expand all files Collapse all files

29
llama_toolchain/cli/stack/build.py
View file

@ -27,7 +27,7 @@ class StackBuild(Subcommand):
def _add_arguments(self):
self.parser.add_argument(
"--config",
"config",
type=str,
help="Path to a config file to use for the build",
)
@ -77,17 +77,20 @@ class StackBuild(Subcommand):
from llama_toolchain.common.prompt_for_config import prompt_for_config
from llama_toolchain.core.dynamic import instantiate_class_type
if args.config:
with open(args.config, "r") as f:
try:
build_config = BuildConfig(**yaml.safe_load(f))
except Exception as e:
self.parser.error(f"Could not parse config file {args.config}: {e}")
return
if args.name:
build_config.name = args.name
self._run_stack_build_command_from_build_config(build_config)
if not args.config:
self.parser.error(
"No config file specified. Please use `llama stack build /path/to/*-build.yaml`. Example config files can be found in llama_toolchain/configs/distributions"
)
return
build_config = prompt_for_config(BuildConfig, None)
self._run_stack_build_command_from_build_config(build_config)
with open(args.config, "r") as f:
try:
build_config = BuildConfig(**yaml.safe_load(f))
except Exception as e:
self.parser.error(f"Could not parse config file {args.config}: {e}")
return
if args.name:
build_config.name = args.name
self._run_stack_build_command_from_build_config(build_config)
return