diff --git a/src/llama_stack/core/build.py b/src/llama_stack/core/build.py index ef64f2036..630916db4 100644 --- a/src/llama_stack/core/build.py +++ b/src/llama_stack/core/build.py @@ -9,7 +9,7 @@ import sys from pydantic import BaseModel from termcolor import cprint -from llama_stack.core.datatypes import BuildConfig, StackRunConfig +from llama_stack.core.datatypes import StackRunConfig from llama_stack.core.distribution import get_provider_registry from llama_stack.distributions.template import DistributionTemplate from llama_stack.log import get_logger @@ -83,7 +83,7 @@ def get_provider_dependencies( return list(set(normal_deps)), list(set(special_deps)), list(set(external_provider_deps)) -def print_pip_install_help(config: BuildConfig): +def print_pip_install_help(config: StackRunConfig): normal_deps, special_deps, _ = get_provider_dependencies(config) cprint( diff --git a/src/llama_stack/core/library_client.py b/src/llama_stack/core/library_client.py index d6be7aeca..06d23e3d1 100644 --- a/src/llama_stack/core/library_client.py +++ b/src/llama_stack/core/library_client.py @@ -41,7 +41,6 @@ from termcolor import cprint from llama_stack.core.build import print_pip_install_help from llama_stack.core.configure import parse_and_maybe_upgrade_config -from llama_stack.core.datatypes import BuildConfig, BuildProvider, DistributionSpec from llama_stack.core.request_headers import PROVIDER_DATA_VAR, request_provider_data_context from llama_stack.core.resolver import ProviderRegistry from llama_stack.core.server.routes import RouteImpls, find_matching_route, initialize_route_impls @@ -266,20 +265,7 @@ class AsyncLlamaStackAsLibraryClient(AsyncLlamaStackClient): file=sys.stderr, ) if self.config_path_or_distro_name.endswith(".yaml"): - providers: dict[str, list[BuildProvider]] = {} - for api, run_providers in self.config.providers.items(): - for provider in run_providers: - providers.setdefault(api, []).append( - BuildProvider(provider_type=provider.provider_type, module=provider.module) - ) - providers = dict(providers) - build_config = BuildConfig( - distribution_spec=DistributionSpec( - providers=providers, - ), - external_providers_dir=self.config.external_providers_dir, - ) - print_pip_install_help(build_config) + print_pip_install_help(self.config) else: prefix = "!" if in_notebook() else "" cprint(