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

address comments

Browse source
This commit is contained in:
Xi Yan 2024-09-18 11:09:52 -07:00
parent 3b4351062f
commit 714a1703c4
2 changed files with 3 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -11,7 +11,6 @@ from llama_stack.distribution.datatypes import * # noqa: F403
from pathlib import Path from pathlib import Path
import yaml import yaml
from llama_stack.distribution.datatypes import Api
from prompt_toolkit import prompt from prompt_toolkit import prompt
from prompt_toolkit.validation import Validator from prompt_toolkit.validation import Validator
from termcolor import cprint from termcolor import cprint
@ -35,7 +34,7 @@ class StackBuild(Subcommand):
type=str, type=str,
default=None, default=None,
nargs="*", 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( self.parser.add_argument(
@ -87,7 +86,6 @@ class StackBuild(Subcommand):
from llama_stack.distribution.utils.prompt_for_config import prompt_for_config from llama_stack.distribution.utils.prompt_for_config import prompt_for_config
if not args.config: if not args.config:
# build_config = prompt_for_config(BuildConfig, None)
name = prompt( name = prompt(
"> Please enter an unique name for identifying your Llama Stack build distribution (e.g. my-local-stack): " "> 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( 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", color="green",
) )

8
llama_stack/distribution/datatypes.py
View file

@ -184,13 +184,7 @@ class DistributionSpec(BaseModel):
) )
docker_image: Optional[str] = None docker_image: Optional[str] = None
providers: Dict[str, Union[str, List[str]]] = Field( providers: Dict[str, Union[str, List[str]]] = Field(
default={ default_factory=dict,
"inference": "meta-reference",
"memory": "meta-reference",
"safety": "meta-reference",
"agents": "meta-reference",
"telemetry": "meta-reference",
},
description=""" description="""
Provider Types for each of the APIs provided by this distribution. If you Provider Types for each of the APIs provided by this distribution. If you
select multiple providers, you should provide an appropriate 'routing_map' select multiple providers, you should provide an appropriate 'routing_map'