first commit
This commit is contained in:
12
configs/rerankers/config.py
Normal file
12
configs/rerankers/config.py
Normal file
@@ -0,0 +1,12 @@
|
||||
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"}
|
||||
Reference in New Issue
Block a user