Update apis/chat_api.py

apis/chat_api.py

@@ -384,17 +384,21 @@ class ChatAPIApp:
     class RerankRequest(BaseModel):
         model: str
         input: str
-        document: list
-        top_k: int
+        documents: list
         return_documents: bool
+        top_k: Optional[int]=None
 
+    def _score_to_list(self, x):
+        x['score'] = x['score'].tolist()
+        return x
     def get_rerank(self, request: RerankRequest, api_key: str = Depends(extract_api_key)):
-        return self.rerank[request.model].rank(
+        ranks = self.rerank[request.model].rank(
             request.input, 
-            request.document, 
+            request.documents, 
             top_k=request.top_k, 
             return_documents=request.return_documents
         )
+        return [self._score_to_list(x) for x in ranks]
         
 
     def get_readme(self):