update
Browse files
app.py
CHANGED
|
@@ -68,7 +68,12 @@ async def rerank(
|
|
| 68 |
# 返回 top_k 结果
|
| 69 |
if request.top_k is not None:
|
| 70 |
return sorted_results[:request.top_k]
|
| 71 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 72 |
|
| 73 |
except Exception as e:
|
| 74 |
logger.error(f"API Error: {str(e)}", exc_info=True)
|
|
|
|
| 68 |
# 返回 top_k 结果
|
| 69 |
if request.top_k is not None:
|
| 70 |
return sorted_results[:request.top_k]
|
| 71 |
+
# 返回符合 OpenAI 风格的格式
|
| 72 |
+
return {
|
| 73 |
+
"object": "list",
|
| 74 |
+
"data": sorted_results, # 结果包装在 data 字段中
|
| 75 |
+
"model": MODEL_NAME
|
| 76 |
+
}
|
| 77 |
|
| 78 |
except Exception as e:
|
| 79 |
logger.error(f"API Error: {str(e)}", exc_info=True)
|