diff --git a/llama_stack/distribution/distribution.py b/llama_stack/distribution/distribution.py index f09b13530..db7ddfa52 100644 --- a/llama_stack/distribution/distribution.py +++ b/llama_stack/distribution/distribution.py @@ -66,7 +66,7 @@ def builtin_automatically_routed_apis() -> list[AutoRoutedApiInfo]: ] -def apis_provided_by_stack() -> set[Api]: +def all_builtin_apis() -> set[Api]: return { Api.models, Api.shields, @@ -82,7 +82,7 @@ def apis_provided_by_stack() -> set[Api]: def providable_apis() -> list[Api]: - return [api for api in Api if api not in apis_provided_by_stack()] + return [api for api in Api if api not in all_builtin_apis()] def _load_remote_provider_spec(spec_data: dict[str, Any], api: Api) -> ProviderSpec: