phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-17 18:42:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

change to llm type

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-12-10 16:57:00 -08:00
parent 62890b3171
commit 2b6aa71a21
3 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/distribution/routers/routers.py
View file

@ -168,7 +168,7 @@ class InferenceRouter(Inference):
model = await self.routing_table.get_model(model_id)
if model is None:
raise ValueError(f"Model '{model_id}' not found")
if model.model_type == ModelType.llm_model:
if model.model_type == ModelType.llm:
raise ValueError(
f"Model '{model_id}' is an LLM model and does not support embeddings"
)

2
llama_stack/distribution/routers/routing_tables.py
View file

@ -224,7 +224,7 @@ class ModelsRoutingTable(CommonRoutingTableImpl, Models):
if metadata is None:
metadata = {}
if model_type is None:
model_type = ModelType.llm_model
model_type = ModelType.llm
if (
"embedding_dimension" not in metadata
and model_type == ModelType.embedding_model