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

Add pragma nocover to optional dependency imports

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-01-15 10:32:35 +02:00
parent fae274a04d
commit bb60ad127a
No known key found for this signature in database
4 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
haiku_rag_slim/haiku/rag/embeddings/__init__.py
View file

@ -128,7 +128,7 @@ def get_embedder(config: AppConfig = Config) -> EmbedderWrapper:
if provider == "voyageai": if provider == "voyageai":
try: try:
from haiku.rag.embeddings.voyageai import VoyageAIEmbeddingModel from haiku.rag.embeddings.voyageai import VoyageAIEmbeddingModel
except ImportError: except ImportError: # pragma: no cover
raise ImportError( raise ImportError(
"VoyageAI embedder requires the 'voyageai' package. " "VoyageAI embedder requires the 'voyageai' package. "
"Please install haiku.rag with the 'voyageai' extra: " "Please install haiku.rag with the 'voyageai' extra: "

2
haiku_rag_slim/haiku/rag/inspector/__init__.py
View file

@ -1,6 +1,6 @@
try: try:
from haiku.rag.inspector.app import run_inspector from haiku.rag.inspector.app import run_inspector
except ImportError as e: except ImportError as e: # pragma: no cover
raise ImportError( raise ImportError(
"textual is not installed. Please install it with `pip install 'haiku.rag-slim[tui]'` or use the full haiku.rag package." "textual is not installed. Please install it with `pip install 'haiku.rag-slim[tui]'` or use the full haiku.rag package."
) from e ) from e

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

@ -30,7 +30,7 @@ def get_reranker(config: AppConfig = Config) -> RerankerBase | None:
os.environ["TOKENIZERS_PARALLELISM"] = "true" os.environ["TOKENIZERS_PARALLELISM"] = "true"
reranker = MxBAIReranker() reranker = MxBAIReranker()
except ImportError: except ImportError: # pragma: no cover
reranker = None reranker = None
elif config.reranking.model and config.reranking.model.provider == "cohere": elif config.reranking.model and config.reranking.model.provider == "cohere":
@ -38,7 +38,7 @@ def get_reranker(config: AppConfig = Config) -> RerankerBase | None:
from haiku.rag.reranking.cohere import CohereReranker from haiku.rag.reranking.cohere import CohereReranker
reranker = CohereReranker() reranker = CohereReranker()
except ImportError: except ImportError: # pragma: no cover
reranker = None reranker = None
elif config.reranking.model and config.reranking.model.provider == "vllm": elif config.reranking.model and config.reranking.model.provider == "vllm":
@ -49,7 +49,7 @@ def get_reranker(config: AppConfig = Config) -> RerankerBase | None:
if not base_url: if not base_url:
raise ValueError("vLLM reranker requires base_url in reranking.model") raise ValueError("vLLM reranker requires base_url in reranking.model")
reranker = VLLMReranker(config.reranking.model.name, base_url) reranker = VLLMReranker(config.reranking.model.name, base_url)
except ImportError: except ImportError: # pragma: no cover
reranker = None reranker = None
elif config.reranking.model and config.reranking.model.provider == "zeroentropy": elif config.reranking.model and config.reranking.model.provider == "zeroentropy":
@ -59,7 +59,7 @@ def get_reranker(config: AppConfig = Config) -> RerankerBase | None:
# Use configured model or default to zerank-1 # Use configured model or default to zerank-1
model = config.reranking.model.name or "zerank-1" model = config.reranking.model.name or "zerank-1"
reranker = ZeroEntropyReranker(model) reranker = ZeroEntropyReranker(model)
except ImportError: except ImportError: # pragma: no cover
reranker = None reranker = None
_reranker_cache[config_id] = reranker _reranker_cache[config_id] = reranker

2
haiku_rag_slim/haiku/rag/reranking/cohere.py
View file

@ -3,7 +3,7 @@ from haiku.rag.store.models.chunk import Chunk
try: try:
import cohere import cohere
except ImportError as e: except ImportError as e: # pragma: no cover
raise ImportError( raise ImportError(
"cohere is not installed. Please install it with `pip install cohere` or use the cohere optional dependency." "cohere is not installed. Please install it with `pip install cohere` or use the cohere optional dependency."
) from e ) from e