diff --git a/llama_stack/apis/models/models.py b/llama_stack/apis/models/models.py index 713dfa377..ed9549d63 100644 --- a/llama_stack/apis/models/models.py +++ b/llama_stack/apis/models/models.py @@ -23,7 +23,7 @@ class CommonModelFields(BaseModel): class ModelType(Enum): llm = "llm" - embedding_model = "embedding_model" + embedding_model = "embedding" @json_schema_type