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

fixed embedding dimensions to get stripped model

Browse source
This commit is contained in:
Lucas Oliveira 2025-09-30 16:20:59 -03:00
parent d6b100459f
commit d018a96923
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/utils/embeddings.py
View file

@ -10,6 +10,8 @@ def get_embedding_dimensions(model_name: str) -> int:
# Check all model dictionaries
all_models = {**OPENAI_EMBEDDING_DIMENSIONS, **OLLAMA_EMBEDDING_DIMENSIONS, **WATSONX_EMBEDDING_DIMENSIONS}
model_name = model_name.lower().strip().split(":")[0]
if model_name in all_models:
dimensions = all_models[model_name]
logger.info(f"Found dimensions for model '{model_name}': {dimensions}")