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

Don't depend on templates.py when print llama stack build messages (#496)

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-20 15:44:49 -08:00 • committed by GitHub
parent 00816cc8ef
commit 068ac00a3b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 9 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -8,7 +8,6 @@ import argparse
from llama_stack.cli.subcommand import Subcommand from llama_stack.cli.subcommand import Subcommand
from llama_stack.distribution.datatypes import * # noqa: F403 from llama_stack.distribution.datatypes import * # noqa: F403
import importlib
import os import os
import shutil import shutil
from functools import lru_cache from functools import lru_cache
@ -258,6 +257,7 @@ 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
@ -286,17 +286,19 @@ class StackBuild(Subcommand):
os.makedirs(build_dir, exist_ok=True) os.makedirs(build_dir, exist_ok=True)
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)
module_name = f"llama_stack.templates.{template_name}"
module = importlib.import_module(module_name) with open(template_path, "r") as f:
distribution_template = module.get_distribution_template() yaml_content = f.read()
# Find all ${env.VARIABLE} patterns
env_vars = set(re.findall(r"\${env\.([A-Za-z0-9_]+)}", yaml_content))
cprint("Build Successful! Next steps: ", color="green") cprint("Build Successful! Next steps: ", color="green")
env_vars = ", ".join(distribution_template.run_config_env_vars.keys())
cprint( cprint(
f" 1. Set the environment variables: {env_vars}", f" 1. Set the environment variables: {list(env_vars)}",
color="green", color="green",
) )
cprint( cprint(
f" 2. `llama stack run {run_config_file}`", f" 2. Run: `llama stack run {template_name}`",
color="green", color="green",
) )
else: else: