From 0a279aa12aa19a9aa11c09c92c97a8bc6717eba1 Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Tue, 9 Sep 2025 10:58:41 +0300 Subject: [PATCH] Fix missing vllm embeddings from embedder factory --- src/haiku/rag/embeddings/__init__.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/haiku/rag/embeddings/__init__.py b/src/haiku/rag/embeddings/__init__.py index c463e2ee..0964ea9e 100644 --- a/src/haiku/rag/embeddings/__init__.py +++ b/src/haiku/rag/embeddings/__init__.py @@ -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}")