Coverage for src / qdrant_loader_mcp_server / config_reranking.py: 100%

1from pydantic import BaseModel, ConfigDict, Field, PositiveInt 

2 

3 

4class MCPReranking(BaseModel): 

5 

6 model_config = ConfigDict(extra="forbid") 

7 

8 enabled: bool = Field(default=True, description="Enable or disable reranking") 

9 model: str = Field( 

10 default="cross-encoder/ms-marco-MiniLM-L-12-v2", 

11 description="Reranking model to use", 

12 ) 

13 device: str | None = Field( 

14 default=None, 

15 description="Device to run the reranking model (auto-detects if not specified)", 

16 ) 

17 batch_size: PositiveInt = Field( 

18 default=32, 

19 description="Batch size for reranking model inference (must be >= 1)", 

20 )