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

comments, get started guide

Browse source
This commit is contained in:
Xi Yan 2024-09-18 11:31:46 -07:00
parent e9fbe85dc5
commit 0fa7f1d1aa
3 changed files with 41 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

10
llama_stack/cli/stack/build.py
View file

@ -11,8 +11,6 @@ from llama_stack.distribution.datatypes import * # noqa: F403
from pathlib import Path
import yaml
from prompt_toolkit.validation import Validator
from termcolor import cprint
class StackBuild(Subcommand):
@ -84,16 +82,18 @@ class StackBuild(Subcommand):
from llama_stack.distribution.distribution import Api, api_providers
from llama_stack.distribution.utils.dynamic import instantiate_class_type
from prompt_toolkit import prompt
from prompt_toolkit.validation import Validator
from termcolor import cprint
if not args.config:
name = prompt(
"> Enter an unique name for identifying your Llama Stack build distribution (e.g. my-local-stack): "
"> Enter a unique name for identifying your Llama Stack build distribution (e.g. my-local-stack): "
)
image_type = prompt(
"> Enter the image type you want your distribution to be built with (docker or conda): ",
"> Enter the image type you want your distribution to be built as (docker or conda): ",
validator=Validator.from_callable(
lambda x: x in ["docker", "conda"],
error_message="Invalid image type, please enter (conda|docker)",
error_message="Invalid image type, please enter conda or docker",
),
default="conda",
)

2
llama_stack/cli/stack/configure.py
View file

@ -14,7 +14,6 @@ from llama_stack.cli.subcommand import Subcommand
from llama_stack.distribution.utils.config_dirs import BUILDS_BASE_DIR
from llama_stack.distribution.utils.exec import run_with_pty
from termcolor import cprint
from llama_stack.distribution.datatypes import * # noqa: F403
import os
@ -50,6 +49,7 @@ class StackConfigure(Subcommand):
import pkg_resources
from llama_stack.distribution.build import ImageType
from termcolor import cprint
docker_image = None
conda_dir = Path(os.getenv("CONDA_PREFIX")).parent / f"llamastack-{args.config}"