/ src / config / reranking.py
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      )