gmakstutis/openrag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

changed models service to not remove : on ollama

Browse source
This commit is contained in:
Lucas Oliveira 2025-09-22 17:39:19 -03:00
parent 745c00cad0
commit a3a1a17231
1 changed files with 7 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/services/models_service.py
View file

@ -124,10 +124,8 @@ class ModelsService:
for model in models:
model_name = model.get(JSON_NAME_KEY, "")
# Remove tag if present (e.g., "llama3:latest" -> "llama3")
clean_model_name = model_name.split(":")[0] if model_name else ""
if not clean_model_name:
if not model_name:
continue
logger.debug(f"Checking model: {model_name}")
@ -152,7 +150,7 @@ class ModelsService:
# Check if it's an embedding model
is_embedding = any(
embed_model in clean_model_name.lower()
embed_model in model_name.lower()
for embed_model in self.OLLAMA_EMBEDDING_MODELS
)
@ -160,8 +158,8 @@ class ModelsService:
# Embedding models only need completion capability
embedding_models.append(
{
"value": clean_model_name,
"label": clean_model_name,
"value": model_name,
"label": model_name,
"default": False,
}
)
@ -169,9 +167,9 @@ class ModelsService:
# Language models need both completion and tool calling
language_models.append(
{
"value": clean_model_name,
"label": clean_model_name,
"default": "llama3" in clean_model_name.lower(),
"value": model_name,
"label": model_name,
"default": "llama3" in model_name.lower(),
}
)
except Exception as e: