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

Validate reranker config before importing the provider package

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-06-29 15:04:04 +03:00
parent ac64aa9c17
commit 7f50d3698f
No known key found for this signature in database
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
haiku_rag_slim/haiku/rag/reranking/__init__.py
View file

@ -24,10 +24,10 @@ def get_reranker(config: AppConfig = Config) -> RerankerBase | None:
return CohereReranker() return CohereReranker()
if model.provider == "vllm": if model.provider == "vllm":
from haiku.rag.reranking.vllm import VLLMReranker
if not model.base_url: if not model.base_url:
raise ValueError("vLLM reranker requires base_url in reranking.model") raise ValueError("vLLM reranker requires base_url in reranking.model")
from haiku.rag.reranking.vllm import VLLMReranker
return VLLMReranker(model.name, model.base_url) return VLLMReranker(model.name, model.base_url)
if model.provider == "zeroentropy": 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") return JinaLocalReranker(model.name or "jinaai/jina-reranker-v3")
if model.provider == "cross-encoder": if model.provider == "cross-encoder":
from haiku.rag.reranking.cross_encoder import CrossEncoderReranker
if not model.name: if not model.name:
raise ValueError( raise ValueError(
"cross-encoder reranker requires name in reranking.model" "cross-encoder reranker requires name in reranking.model"
) )
from haiku.rag.reranking.cross_encoder import CrossEncoderReranker
return CrossEncoderReranker(model.name) return CrossEncoderReranker(model.name)
except ImportError: # pragma: no cover except ImportError: # pragma: no cover
return None return None