workaround list templates command

This commit is contained in:
Xi Yan 2024-10-24 23:05:26 -07:00
parent cb43caa2c3
commit 58c3c45f19
2 changed files with 11 additions and 1 deletions

View file

@ -16,6 +16,12 @@ TEMPLATES_PATH = (
Path(os.path.relpath(__file__)).parent.parent.parent.parent / "distributions"
)
# build.yaml templates exist in the llama-stack/distributions while wheel installs llama-stack/llama_stack
# we copied the distributions folder to llama-stack/llama_stack/cli/distributions for wheel builds,
# so we need to check both locations
if not TEMPLATES_PATH.exists():
TEMPLATES_PATH = Path(os.path.relpath(__file__)).parent.parent / "distributions"
@lru_cache()
def available_templates_specs() -> List[BuildConfig]: