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 <api>` works as expected.

---------

Signed-off-by: Yuan Tang <terrytangyuan@gmail.com>
This commit is contained in:
Yuan Tang 2025-02-07 12:02:15 -05:00 committed by GitHub
parent 840344975d
commit 3f9764d50c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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,