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,