diff --git a/llama_stack/cli/stack/build.py b/llama_stack/cli/stack/build.py index 112a5d74e..76988b058 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 llama_stack.distribution.datatypes import Api from prompt_toolkit import prompt from prompt_toolkit.validation import Validator from termcolor import cprint @@ -35,7 +34,7 @@ class StackBuild(Subcommand): type=str, default=None, nargs="*", - help="Path to a config file to use for the build. You may find example configs in llama_stack/distribution/example_configs. If not defined, you will be prompted for entering wizard", + help="Path to a config file to use for the build. You can find example configs in llama_stack/distribution/example_configs. If this argument is not provided, you will be prompted to enter information interactively", ) self.parser.add_argument( @@ -87,7 +86,6 @@ class StackBuild(Subcommand): from llama_stack.distribution.utils.prompt_for_config import prompt_for_config if not args.config: - # build_config = prompt_for_config(BuildConfig, None) name = prompt( "> Please enter an unique name for identifying your Llama Stack build distribution (e.g. my-local-stack): " ) @@ -101,7 +99,7 @@ class StackBuild(Subcommand): ) cprint( - f"\n Now, let's configure your Llama Stack distribution specs with API providers", + f"\n Llama Stack is composed of several APIs working together. Let's configure the providers (implementations) you want to use for these APIs.", color="green", ) diff --git a/llama_stack/distribution/datatypes.py b/llama_stack/distribution/datatypes.py index 8328acd82..e57617016 100644 --- a/llama_stack/distribution/datatypes.py +++ b/llama_stack/distribution/datatypes.py @@ -184,13 +184,7 @@ class DistributionSpec(BaseModel): ) docker_image: Optional[str] = None providers: Dict[str, Union[str, List[str]]] = Field( - default={ - "inference": "meta-reference", - "memory": "meta-reference", - "safety": "meta-reference", - "agents": "meta-reference", - "telemetry": "meta-reference", - }, + default_factory=dict, description=""" Provider Types for each of the APIs provided by this distribution. If you select multiple providers, you should provide an appropriate 'routing_map'