Update app.py

app.py

@@ -1,7 +1,41 @@
 from fastapi import FastAPI
+from pydantic import BaseModel
+from typing import Tuple, List
 
+class RankingExecInfo:
+    def __init__(
+        self, prompt, response: str, input_token_count: int, output_token_count: int
+    ):
+        self.prompt = prompt
+        self.response = response
+        self.input_token_count = input_token_count
+        self.output_token_count = output_token_count
+
+    def __repr__(self):
+        return str(self.__dict__)
+
+class Result:
+    def __init__(
+        self,
+        query: str,
+        hits: List[Dict[str, Any]],
+        ranking_exec_summary: Optional[List[RankingExecInfo]] = None,
+    ):
+        self.query = query
+        self.hits = hits
+        self.ranking_exec_summary = ranking_exec_summary
+
+    def __repr__(self):
+        return str(self.__dict__)
+
+class RerankRequest(BaseModel):
+    query: str
+    hits: List(Tuple(int, str))
+    
 app = FastAPI()
 
 @app.post("/")
-def test():
-    return {"data": "this is a success"}
+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)]}