phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-07 02:58:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

List providers command prints out non-existing APIs from registry. Fixes #966

Browse source 
Signed-off-by: Yuan Tang <terrytangyuan@gmail.com>
This commit is contained in:
Yuan Tang 2025-02-05 10:26:37 -05:00
parent 529708215c
commit 35bc05d71b
No known key found for this signature in database
2 changed files with 16 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

14
llama_stack/apis/datatypes.py
View file

@ -9,6 +9,20 @@ from enum import Enum
from llama_models.schema_utils import json_schema_type 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 @json_schema_type
class Api(Enum): class Api(Enum):
inference = "inference" inference = "inference"

4
llama_stack/cli/stack/list_providers.py
View file

@ -22,9 +22,9 @@ class StackListProviders(Subcommand):
self.parser.set_defaults(func=self._run_providers_list_cmd) self.parser.set_defaults(func=self._run_providers_list_cmd)
def _add_arguments(self): 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( self.parser.add_argument(
"api", "api",
type=str, type=str,