From 7f50d3698fdcb9f17ffd4b142a62116b5738e8e9 Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Mon, 29 Jun 2026 15:04:04 +0300 Subject: [PATCH] Validate reranker config before importing the provider package --- haiku_rag_slim/haiku/rag/reranking/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/haiku_rag_slim/haiku/rag/reranking/__init__.py b/haiku_rag_slim/haiku/rag/reranking/__init__.py index 876df372..e9e11913 100644 --- a/haiku_rag_slim/haiku/rag/reranking/__init__.py +++ b/haiku_rag_slim/haiku/rag/reranking/__init__.py @@ -24,10 +24,10 @@ def get_reranker(config: AppConfig = Config) -> RerankerBase | None: return CohereReranker() if model.provider == "vllm": - from haiku.rag.reranking.vllm import VLLMReranker - if not model.base_url: raise ValueError("vLLM reranker requires base_url in reranking.model") + from haiku.rag.reranking.vllm import VLLMReranker + return VLLMReranker(model.name, model.base_url) if model.provider == "zeroentropy": @@ -46,12 +46,12 @@ def get_reranker(config: AppConfig = Config) -> RerankerBase | None: return JinaLocalReranker(model.name or "jinaai/jina-reranker-v3") if model.provider == "cross-encoder": - from haiku.rag.reranking.cross_encoder import CrossEncoderReranker - if not model.name: raise ValueError( "cross-encoder reranker requires name in reranking.model" ) + from haiku.rag.reranking.cross_encoder import CrossEncoderReranker + return CrossEncoderReranker(model.name) except ImportError: # pragma: no cover return None