phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-12 13:00:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add a graceful quit for build process

Browse source 
Signed-off-by: reidliu <reid201711@gmail.com>
This commit is contained in:
reidliu 2025-02-19 22:24:55 +08:00
parent a66b4c4c81
commit a1d25035a8
1 changed files with 25 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

26
llama_stack/cli/stack/_build.py
View file

@ -55,6 +55,13 @@ def available_templates_specs() -> Dict[str, BuildConfig]:
return template_specs return template_specs
def _check_quit_input(user_input: str) -> bool:
if user_input.lower() == "quit" or user_input.lower() == "q":
cprint("Quit the build process.", color="yellow")
return True
return False
def run_stack_build_command(args: argparse.Namespace) -> None: def run_stack_build_command(args: argparse.Namespace) -> None:
if args.list_templates: if args.list_templates:
return _run_template_list_cmd() return _run_template_list_cmd()
@ -80,6 +87,10 @@ def run_stack_build_command(args: argparse.Namespace) -> None:
) )
return return
elif not args.config and not args.template: elif not args.config and not args.template:
cprint("=" * 34, color="cyan")
cprint("Welcome to Llama Stack Build Setup", color="cyan", attrs=["bold"])
cprint("=" * 34, color="cyan")
cprint("Type 'quit' or 'q' to cancel at any time.", color="yellow")
name = prompt( name = prompt(
"> Enter a name for your Llama Stack (e.g. my-local-stack): ", "> Enter a name for your Llama Stack (e.g. my-local-stack): ",
validator=Validator.from_callable( validator=Validator.from_callable(
@ -88,15 +99,21 @@ def run_stack_build_command(args: argparse.Namespace) -> None:
), ),
) )
if _check_quit_input(name):
return
image_type = prompt( image_type = prompt(
"> Enter the image type you want your Llama Stack to be built as (container or conda or venv): ", "> Enter the image type you want your Llama Stack to be built as (container or conda or venv): ",
validator=Validator.from_callable( validator=Validator.from_callable(
lambda x: x in ["container", "conda", "venv"], lambda x: x in ["container", "conda", "venv", "q", "quit"],
error_message="Invalid image type, please enter conda or container or venv", error_message="Invalid image type, please enter conda or container or venv",
), ),
default="conda", default="conda",
) )
if _check_quit_input(image_type):
return
if image_type == "conda": if image_type == "conda":
if not image_name: if not image_name:
cprint( cprint(
@ -127,6 +144,7 @@ def run_stack_build_command(args: argparse.Namespace) -> None:
providers = dict() providers = dict()
for api, providers_for_api in get_provider_registry().items(): for api, providers_for_api in get_provider_registry().items():
available_providers = [x for x in providers_for_api.keys() if x not in ("remote", "remote::sample")] available_providers = [x for x in providers_for_api.keys() if x not in ("remote", "remote::sample")]
available_providers += ["q", "quit"]
api_provider = prompt( api_provider = prompt(
"> Enter provider for API {}: ".format(api.value), "> Enter provider for API {}: ".format(api.value),
completer=WordCompleter(available_providers), completer=WordCompleter(available_providers),
@ -137,6 +155,9 @@ def run_stack_build_command(args: argparse.Namespace) -> None:
), ),
) )
if _check_quit_input(api_provider):
return
providers[api.value] = api_provider providers[api.value] = api_provider
description = prompt( description = prompt(
@ -144,6 +165,9 @@ def run_stack_build_command(args: argparse.Namespace) -> None:
default="", default="",
) )
if _check_quit_input(description):
return
distribution_spec = DistributionSpec( distribution_spec = DistributionSpec(
providers=providers, providers=providers,
description=description, description=description,