From 0fc0b39967199dd4d0e28c35b7fc1c945492c4e1 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Wed, 18 Sep 2024 11:38:16 -0700 Subject: [PATCH] fix imports --- llama_stack/cli/stack/build.py | 4 ++++ llama_stack/cli/stack/configure.py | 1 + 2 files changed, 5 insertions(+) diff --git a/llama_stack/cli/stack/build.py b/llama_stack/cli/stack/build.py index fa4967190..e49da58f0 100644 --- a/llama_stack/cli/stack/build.py +++ b/llama_stack/cli/stack/build.py @@ -76,6 +76,10 @@ class StackBuild(Subcommand): cprint( f"Build spec configuration saved at {str(build_file_path)}", + color="blue", + ) + cprint( + f"You may now run `llama stack configure {build_config.name}` or `llama stack configure {str(build_file_path)}`", color="green", ) diff --git a/llama_stack/cli/stack/configure.py b/llama_stack/cli/stack/configure.py index 82ef57760..6b3b53f3e 100644 --- a/llama_stack/cli/stack/configure.py +++ b/llama_stack/cli/stack/configure.py @@ -100,6 +100,7 @@ class StackConfigure(Subcommand): ): from llama_stack.distribution.configure import configure_api_providers from llama_stack.distribution.utils.serialize import EnumEncoder + from termcolor import cprint builds_dir = BUILDS_BASE_DIR / build_config.image_type if output_dir: