cprint in print_pip_install_help

This commit is contained in:
dltn 2024-12-05 13:21:33 -08:00
parent a2d9a983de
commit 703a20c3bc

View file

@ -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()