reranking.py
1 """Reranking configuration.""" 2 3 4 from pydantic import Field 5 from pydantic_settings import BaseSettings 6 7 from src.rerankers.types import RerankerType 8 9 10 class RerankingConfig(BaseSettings): 11 """Reranking configuration.""" 12 13 reranker_name: RerankerType = Field( 14 default=RerankerType.GEMINI, 15 description="Reranker backend type", 16 ) 17 reranker_config: dict | None = Field( 18 default=None, 19 description="Additional reranker-specific keyword arguments", 20 )