From aa5061602f3232f2d940680e3d0b6951ce68e030 Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Wed, 11 Dec 2024 08:45:26 -0800 Subject: [PATCH] remove _model suffix --- llama_stack/apis/models/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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