File size: 387 Bytes
5ec9280 37098e4 5ec9280 37098e4 d5a47d8 37098e4 5ec9280 37098e4 b4208f0 37098e4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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)]} |