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

re-build from name

Browse source
This commit is contained in:
Xi Yan 2024-09-30 16:22:52 -07:00
parent 4897bf2f85
commit 73decb3781
1 changed files with 21 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -29,26 +29,6 @@ def available_templates_specs() -> List[BuildConfig]:
return template_specs
def get_build_config_from_name(name: str) -> Optional[Path]:
if os.getenv("CONDA_PREFIX", ""):
conda_dir = (
Path(os.getenv("CONDA_PREFIX")).parent / f"llamastack-{args.config}"
)
else:
cprint(
"Cannot find CONDA_PREFIX. Trying default conda path ~/.conda/envs...",
color="green",
)
conda_dir = (
Path(os.path.expanduser("~/.conda/envs")) / f"llamastack-{args.config}"
)
build_config_file = Path(conda_dir) / f"{args.config}-build.yaml"
if build_config_file.exists():
return build_config_file
return None
class StackBuild(Subcommand):
def __init__(self, subparsers: argparse._SubParsersAction):
@ -98,6 +78,25 @@ class StackBuild(Subcommand):
choices=["conda", "docker"],
)
def _get_build_config_from_name(self, args: argparse.Namespace) -> Optional[Path]:
if os.getenv("CONDA_PREFIX", ""):
conda_dir = (
Path(os.getenv("CONDA_PREFIX")).parent / f"llamastack-{args.name}"
)
else:
cprint(
"Cannot find CONDA_PREFIX. Trying default conda path ~/.conda/envs...",
color="green",
)
conda_dir = (
Path(os.path.expanduser("~/.conda/envs")) / f"llamastack-{args.name}"
)
build_config_file = Path(conda_dir) / f"{args.name}-build.yaml"
if build_config_file.exists():
return build_config_file
return None
def _run_stack_build_command_from_build_config(
self, build_config: BuildConfig
) -> None:
@ -211,9 +210,9 @@ class StackBuild(Subcommand):
# try to see if we can find a pre-existing build config file through name
if args.name:
maybe_build_config = get_build_config_from_name(args.name)
maybe_build_config = self._get_build_config_from_name(args)
if maybe_build_config:
print(f"Building from existing build config for {args.name} in {str(maybe_build_config)}")
cprint(f"Building from existing build config for {args.name} in {str(maybe_build_config)}...", "green")
with open(maybe_build_config, "r") as f:
build_config = BuildConfig(**yaml.safe_load(f))
self._run_stack_build_command_from_build_config(build_config)