danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix missing vllm embeddings from embedder factory

Browse source
This commit is contained in:
Yiorgis Gozadinos 2025-09-09 10:58:41 +03:00
parent a1efc5a9a3
commit 0a279aa12a
No known key found for this signature in database
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/haiku/rag/embeddings/__init__.py
View file

@ -27,4 +27,9 @@ def get_embedder() -> EmbedderBase:
return OpenAIEmbedder(Config.EMBEDDINGS_MODEL, Config.EMBEDDINGS_VECTOR_DIM) return OpenAIEmbedder(Config.EMBEDDINGS_MODEL, Config.EMBEDDINGS_VECTOR_DIM)
if Config.EMBEDDINGS_PROVIDER == "vllm":
from haiku.rag.embeddings.vllm import Embedder as VllmEmbedder
return VllmEmbedder(Config.EMBEDDINGS_MODEL, Config.EMBEDDINGS_VECTOR_DIM)
raise ValueError(f"Unsupported embedding provider: {Config.EMBEDDINGS_PROVIDER}") raise ValueError(f"Unsupported embedding provider: {Config.EMBEDDINGS_PROVIDER}")