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

Remove the unnecessary message after llama stack build

Browse source
This commit is contained in:
Ashwin Bharambe 2024-12-10 09:46:37 -08:00
parent 885bb0900b
commit fa68ded07c
1 changed files with 1 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -261,7 +261,6 @@ class StackBuild(Subcommand):
) -> None: ) -> None:
import json import json
import os import os
import re
import yaml import yaml
from termcolor import cprint from termcolor import cprint
@ -291,20 +290,8 @@ class StackBuild(Subcommand):
run_config_file = build_dir / f"{build_config.name}-run.yaml" run_config_file = build_dir / f"{build_config.name}-run.yaml"
shutil.copy(template_path, run_config_file) shutil.copy(template_path, run_config_file)
with open(template_path, "r") as f:
yaml_content = f.read()
# Find all ${env.VARIABLE} patterns # Find all ${env.VARIABLE} patterns
env_vars = set(re.findall(r"\${env\.([A-Za-z0-9_]+)}", yaml_content)) cprint("Build Successful!", color="green")
cprint("Build Successful! Next steps: ", color="green")
cprint(
f" 1. Set the environment variables: {list(env_vars)}",
color="green",
)
cprint(
f" 2. Run: `llama stack run {template_name}`",
color="green",
)
else: else:
self._generate_run_config(build_config, build_dir) self._generate_run_config(build_config, build_dir)