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

add name args to override name

Browse source
This commit is contained in:
Xi Yan 2024-09-14 14:42:02 -07:00
parent 1fec9930aa
commit 3802d5acdc
2 changed files with 8 additions and 283 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -30,6 +30,12 @@ class StackBuild(Subcommand):
help="Path to a config file to use for the build",
)
self.parser.add_argument(
"--name",
type=str,
help="Override the name of the llama stack build",
)
def _run_stack_build_command_from_build_config(
self, build_config: BuildConfig
) -> None:
@ -68,8 +74,9 @@ class StackBuild(Subcommand):
except Exception as e:
self.parser.error(f"Could not parse config file {args.config}: {e}")
return
build_config.name = args.name if args.name else build_config.name
self._run_stack_build_command_from_build_config(build_config)
return
build_config = prompt_for_config(BuildConfig, None)
build_config = prompt_for_config(BuildConfig, build_config_default)
self._run_stack_build_command_from_build_config(build_config)