diff --git a/llama_stack/cli/stack/list_providers.py b/llama_stack/cli/stack/list_providers.py index 96e978826..909fea030 100644 --- a/llama_stack/cli/stack/list_providers.py +++ b/llama_stack/cli/stack/list_providers.py @@ -22,9 +22,9 @@ class StackListProviders(Subcommand): self.parser.set_defaults(func=self._run_providers_list_cmd) def _add_arguments(self): - from llama_stack.distribution.datatypes import Api + from llama_stack.distribution.distribution import providable_apis - api_values = [a.value for a in Api] + api_values = [api.value for api in providable_apis()] self.parser.add_argument( "api", type=str,