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": "What are the must-see attractions in Hangzhou?",
    "documents": [
      "West Lake is Hangzhou'\''s most famous attraction, known for Broken Bridge and Leifeng Pagoda.",
      "The Bund in Shanghai sits along the Huangpu River and is the city'\''s signature landmark.",
      "Lingyin Temple, in Hangzhou'\''s West Lake district, is a well-known Buddhist temple."
    ],
    "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": "What are the must-see attractions in Hangzhou?",
        "documents": [
            "West Lake is Hangzhou's most famous attraction, known for Broken Bridge and Leifeng Pagoda.",
            "The Bund in Shanghai sits along the Huangpu River and is the city's signature landmark.",
            "Lingyin Temple, in Hangzhou's West Lake district, is a well-known Buddhist temple.",
        ],
        "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: 'What are the must-see attractions in Hangzhou?',
    documents: [
      "West Lake is Hangzhou's most famous attraction, known for Broken Bridge and Leifeng Pagoda.",
      "The Bund in Shanghai sits along the Huangpu River and is the city's signature landmark.",
      "Lingyin Temple, in Hangzhou's West Lake district, is a well-known Buddhist temple.",
    ],
    top_n: 2,
  }),
});

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