From 3f9764d50ca90789517f16a6f29621f4a893afe9 Mon Sep 17 00:00:00 2001 From: Yuan Tang Date: Fri, 7 Feb 2025 12:02:15 -0500 Subject: [PATCH] fix: List providers command prints out non-existing APIs from registry. Fixes #966 (#969) Fixes #966. Verified that: 1. Correct list of APIs are printed out when running `llama stack list-providers` 2. `llama stack list-providers ` works as expected. --------- Signed-off-by: Yuan Tang --- llama_stack/cli/stack/list_providers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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,