From c6161149b934f6276bd1afc59792651d651d4f4b Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Wed, 18 Sep 2024 11:21:12 -0700 Subject: [PATCH] move imports --- llama_stack/cli/stack/build.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/llama_stack/cli/stack/build.py b/llama_stack/cli/stack/build.py index 0ca05cac1..55fa66807 100644 --- a/llama_stack/cli/stack/build.py +++ b/llama_stack/cli/stack/build.py @@ -11,7 +11,6 @@ from llama_stack.distribution.datatypes import * # noqa: F403 from pathlib import Path import yaml -from prompt_toolkit import prompt from prompt_toolkit.validation import Validator from termcolor import cprint @@ -84,7 +83,7 @@ class StackBuild(Subcommand): def _run_stack_build_command(self, args: argparse.Namespace) -> None: from llama_stack.distribution.distribution import Api, api_providers from llama_stack.distribution.utils.dynamic import instantiate_class_type - from llama_stack.distribution.utils.prompt_for_config import prompt_for_config + from prompt_toolkit import prompt if not args.config: name = prompt(