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

Merge pull request #54 from ggozad/fix/vllm-embeddings-factory

Browse source 
Fix missing vllm embeddings from embedder factory
This commit is contained in:
Yiorgis Gozadinos 2025-09-09 10:59:09 +03:00 committed by GitHub
commit c5efa94076
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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)
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}")