bge-reranker-v2-m3 文本重排序

curl https://api.apiyi.com/v1/rerank \
  -H "Authorization: Bearer $APIYI_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "bge-reranker-v2-m3",
    "query": "杭州有哪些适合旅游的景点？",
    "documents": [
      "西湖是杭州著名的旅游景点，拥有断桥、苏堤和雷峰塔等景观。",
      "上海外滩位于黄浦江畔，是上海的标志性景点。",
      "灵隐寺位于杭州西湖区，是中国著名的佛教寺院。"
    ],
    "top_n": 2
  }'

import os, requests

resp = requests.post(
    "https://api.apiyi.com/v1/rerank",
    headers={"Authorization": f"Bearer {os.environ['APIYI_API_KEY']}"},
    json={
        "model": "bge-reranker-v2-m3",
        "query": "杭州有哪些适合旅游的景点？",
        "documents": [
            "西湖是杭州著名的旅游景点，拥有断桥、苏堤和雷峰塔等景观。",
            "上海外滩位于黄浦江畔，是上海的标志性景点。",
            "灵隐寺位于杭州西湖区，是中国著名的佛教寺院。",
        ],
        "top_n": 2,
    },
    timeout=60,
).json()

for r in resp["results"]:
    print(f"{r['relevance_score']:.4f}  {r['document']['text']}")

const resp = await fetch('https://api.apiyi.com/v1/rerank', {
  method: 'POST',
  headers: {
    Authorization: `Bearer ${process.env.APIYI_API_KEY}`,
    'Content-Type': 'application/json',
  },
  body: JSON.stringify({
    model: 'bge-reranker-v2-m3',
    query: '杭州有哪些适合旅游的景点？',
    documents: [
      '西湖是杭州著名的旅游景点，拥有断桥、苏堤和雷峰塔等景观。',
      '上海外滩位于黄浦江畔，是上海的标志性景点。',
      '灵隐寺位于杭州西湖区，是中国著名的佛教寺院。',
    ],
    top_n: 2,
  }),
});

const data = await resp.json();
data.results.forEach((r) => console.log(r.relevance_score, r.document.text));