from pydantic import BaseModel class RerankDocument(BaseModel): """ Model class for rerank document. """ index: int text: str score: float class RerankResult(BaseModel): """ Model class for rerank result. """ model: str docs: list[RerankDocument]
