update

app.py

@@ -42,7 +42,7 @@ class RerankRequest(BaseModel):
 
 class RerankResult(BaseModel):
     index: int
-    score: float
+    relevance_score: float
     document: str
 
 @app.post("/rerank")
@@ -55,14 +55,10 @@ async def rerank(
         scores = model.predict(pairs)
         
         results = [
-        {
-            "index": idx,
-            "relevance_score": float(score),  # 同步修改键名
-            "document": doc
-        }
-        for idx, (doc, score) in enumerate(zip(request.documents, scores))
-    ]
-        sorted_results = sorted(results, key=lambda x: x["score"], reverse=True)
+            {"index": idx, "relevance_score": float(relevance_score), "document": doc}
+            for idx, (doc, relevance_score) in enumerate(zip(request.documents, scores))
+        ]
+        sorted_results = sorted(results, key=lambda x: x["relevance_score"], reverse=True)
         
         if request.top_k is not None:
             sorted_results = sorted_results[:request.top_k]