Add pragma nocover to optional dependency imports

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

View file

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

View file

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

View file

@ -30,7 +30,7 @@ def get_reranker(config: AppConfig = Config) -> RerankerBase | None:
os.environ["TOKENIZERS_PARALLELISM"] = "true"
reranker = MxBAIReranker()
except ImportError:
except ImportError: # pragma: no cover
reranker = None
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
reranker = CohereReranker()
except ImportError:
except ImportError: # pragma: no cover
reranker = None
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:
raise ValueError("vLLM reranker requires base_url in reranking.model")
reranker = VLLMReranker(config.reranking.model.name, base_url)
except ImportError:
except ImportError: # pragma: no cover
reranker = None
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
model = config.reranking.model.name or "zerank-1"
reranker = ZeroEntropyReranker(model)
except ImportError:
except ImportError: # pragma: no cover
reranker = None
_reranker_cache[config_id] = reranker

View file

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