from fastapi import FastAPI
from pydantic import BaseModel
from typing import Tuple, List

class RerankRequest(BaseModel):
    query: str
    hits: List[Tuple[int, str]]
    
app = FastAPI()

@app.post("/")
def test(request: RerankRequest):
    hits = request.hits
    reranked = sorted(hits, key=lambda x: x[0])
    return {"data": [(i + 1, item[1]) for i, item in enumerate(reranked)]}