13 lines
405 B
Python
13 lines
405 B
Python
from typing import Optional
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class RerankerConfig(BaseModel):
|
|
"""Configuration for rerankers."""
|
|
|
|
provider: str = Field(description="Reranker provider (e.g., 'cohere', 'sentence_transformer')", default="cohere")
|
|
config: Optional[dict] = Field(description="Provider-specific reranker configuration", default=None)
|
|
|
|
model_config = {"extra": "forbid"}
|