vibe.rerank_providers.berget¶
Berget AI rerank provider.
Uses Berget's API for reranking via BAAI/bge-reranker-v2-m3.
BergetRerankError ¶
Error from Berget rerank API.
BergetRerankProvider ¶
Rerank provider using Berget AI API.
Uses BAAI/bge-reranker-v2-m3 by default.
Configuration
api_key: Berget API key (or BERGET_API_KEY env var) base_url: API base URL (default: https://api.berget.ai/v1) model: Model to use (default: BAAI/bge-reranker-v2-m3) timeout: Request timeout in seconds (default: 60) max_retries: Max retries on transient errors (default: 6)
rerank ¶
rerank(query: str, documents: list[str], top_n: int | None = None) -> list[RerankResult]
Rerank documents by relevance to query.
Includes retry logic with exponential backoff for transient errors.