Source code for rerank_service.models

"""Request/response models for the rerank service."""

from pydantic import BaseModel



[docs]
class RerankDocument(BaseModel):
    """Document candidate accepted by the rerank endpoint."""

    id: str
    content: str




[docs]
class RerankRequest(BaseModel):
    """Request payload for reranking a list of document candidates."""

    query: str
    documents: list[RerankDocument]
    top_k: int | None = None




[docs]
class RerankResult(BaseModel):
    """Single reranked document with its score and rank."""

    id: str
    score: float
    rank: int




[docs]
class RerankResponse(BaseModel):
    """Response payload returned by the rerank endpoint."""

    results: list[RerankResult]
    model: str
    duration_ms: float