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

fix llama stack build

Browse source
This commit is contained in:
Xi Yan 2024-09-22 21:04:06 -07:00
parent 484dc2e4f5
commit 97feb92136
1 changed files with 13 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -160,7 +160,11 @@ class StackBuild(Subcommand):
def _run_stack_build_command(self, args: argparse.Namespace) -> None: def _run_stack_build_command(self, args: argparse.Namespace) -> None:
import yaml import yaml
from llama_stack.distribution.distribution import Api, api_providers from llama_stack.distribution.distribution import (
Api,
api_providers,
builtin_automatically_routed_apis,
)
from llama_stack.distribution.utils.dynamic import instantiate_class_type from llama_stack.distribution.utils.dynamic import instantiate_class_type
from prompt_toolkit import prompt from prompt_toolkit import prompt
from prompt_toolkit.validation import Validator from prompt_toolkit.validation import Validator
@ -213,8 +217,15 @@ class StackBuild(Subcommand):
) )
providers = dict() providers = dict()
all_providers = api_providers()
routing_table_apis = set(
x.routing_table_api for x in builtin_automatically_routed_apis()
)
for api in Api: for api in Api:
all_providers = api_providers() if api in routing_table_apis:
continue
providers_for_api = all_providers[api] providers_for_api = all_providers[api]
api_provider = prompt( api_provider = prompt(