geqintan commited on
Commit
cd16cca
·
1 Parent(s): 90e7888
Files changed (1) hide show
  1. app.py +6 -1
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
- return sorted_results
 
 
 
 
 
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)