From 35bc05d71bb11a9a8dc2074f09379dacddc3f4fc Mon Sep 17 00:00:00 2001 From: Yuan Tang Date: Wed, 5 Feb 2025 10:26:37 -0500 Subject: [PATCH] List providers command prints out non-existing APIs from registry. Fixes #966 Signed-off-by: Yuan Tang --- llama_stack/apis/datatypes.py | 14 ++++++++++++++ llama_stack/cli/stack/list_providers.py | 4 ++-- 2 files changed, 16 insertions(+), 2 deletions(-) diff --git a/llama_stack/apis/datatypes.py b/llama_stack/apis/datatypes.py index ccc395b80..23e8a4834 100644 --- a/llama_stack/apis/datatypes.py +++ b/llama_stack/apis/datatypes.py @@ -9,6 +9,20 @@ from enum import Enum from llama_models.schema_utils import json_schema_type +@json_schema_type +class RegisteredApi(Enum): + agents = "agents" + datasetio = "datasetio" + eval = "eval" + inference = "inference" + post_training = "post_training" + safety = "safety" + scoring = "scoring" + telemetry = "telemetry" + tool_runtime = "tool_runtime" + vector_io = "vector_io" + + @json_schema_type class Api(Enum): inference = "inference" diff --git a/llama_stack/cli/stack/list_providers.py b/llama_stack/cli/stack/list_providers.py index 96e978826..a19dc3898 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.apis.datatypes import RegisteredApi - api_values = [a.value for a in Api] + api_values = [a.value for a in RegisteredApi] self.parser.add_argument( "api", type=str,