diff --git a/llama_stack/distribution/build.py b/llama_stack/distribution/build.py index fb4b6a161..526815038 100644 --- a/llama_stack/distribution/build.py +++ b/llama_stack/distribution/build.py @@ -9,9 +9,9 @@ from enum import Enum from typing import List import pkg_resources -from pydantic import BaseModel - from llama_stack.distribution.utils.exec import run_with_pty +from pydantic import BaseModel +from termcolor import cprint from llama_stack.distribution.datatypes import * # noqa: F403 from pathlib import Path @@ -90,11 +90,12 @@ def get_provider_dependencies( def print_pip_install_help(providers: Dict[str, List[Provider]]): normal_deps, special_deps = get_provider_dependencies(providers) - print( - f"Please install needed dependencies using the following commands:\n\n\tpip install {' '.join(normal_deps)}" + cprint( + f"Please install needed dependencies using the following commands:\n\n\tpip install {' '.join(normal_deps)}", + "yellow", ) for special_dep in special_deps: - log.info(f"\tpip install {special_dep}") + cprint(f"\tpip install {special_dep}", "yellow") print()