Rerank API؛ مرتب‌سازی اسناد بر اساس ارتباط با پرسش برای RAG

curl https://1ai.ir/v1/rerank \
  -H "Authorization: Bearer $ONEAI_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "cohere/rerank-v4.0-fast",
    "query": "چطور رمز عبورم را عوض کنم؟",
    "documents": [
      "برای تغییر رمز به تنظیمات، بخش امنیت بروید.",
      "ساعت کاری پشتیبانی ۹ تا ۱۷ است.",
      "اگر رمز را فراموش کرده‌اید از گزینه‌ی بازیابی استفاده کنید."
    ],
    "top_n": 2
  }'

# pip install requests
import os, requests

def rerank(query, docs, top_n=5):
    res = requests.post(
        "https://1ai.ir/v1/rerank",
        headers={"Authorization": f"Bearer {os.environ['ONEAI_API_KEY']}"},
        json={"model": "qwen/qwen3-rerank", "query": query, "documents": docs, "top_n": top_n, "return_documents": False},
        timeout=60,
    )
    res.raise_for_status()
    return [docs[r["index"]] for r in res.json()["results"]]

# Vector search first (fast, approximate), rerank the candidates (slow, precise).
candidates = ["...", "...", "..."]  # top 50 hits from your vector store
context = rerank("How do I reset my password?", candidates)

const res = await fetch("https://1ai.ir/v1/rerank", {
  method: "POST",
  headers: { Authorization: `Bearer ${process.env.ONEAI_API_KEY}`, "Content-Type": "application/json" },
  body: JSON.stringify({ model: "cohere/rerank-v4.0-fast", query, documents: hits.map((h) => h.text), top_n: 5 }),
});
const { results } = await res.json();
const best = results.map((r) => hits[r.index]);

{
  "id": "rerank-…",
  "object": "list",
  "model": "cohere/rerank-v4.0-fast",
  "results": [
    { "index": 0, "relevance_score": 0.91, "document": { "text": "برای تغییر رمز به تنظیمات، بخش امنیت بروید." } },
    { "index": 2, "relevance_score": 0.64, "document": { "text": "اگر رمز را فراموش کرده‌اید از گزینه‌ی بازیابی استفاده کنید." } }
  ],
  "usage": { "search_units": 1 }
}