Реранкинг

POST/api/v1/rerank упорядочивает документы по тому, насколько они отвечают на запрос. Обычно его ставят после поиска, чтобы наверх попали самые подходящие фрагменты.

Запрос

Поля: model — модель реранкинга, query — запрос, documents — строки или объекты {text, image} (image нужен мультимодальным моделям) и необязательный top_n — сколько лучших документов вернуть.

curl https://infergate.ru/api/v1/rerank \
  -H "Authorization: Bearer $INFERGATE_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "cohere/rerank-v3.5",
    "query": "Столица Франции",
    "documents": ["Париж — столица Франции", "Берлин — столица Германии"],
    "top_n": 1
  }'

Ответ

results — документы по убыванию релевантности, у каждого index (номер в вашем списке), relevance_score и document. В usage — расход и стоимость запроса.

{
  "model": "cohere/rerank-v3.5",
  "results": [
    { "index": 0, "relevance_score": 0.98, "document": { "text": "Париж — столица Франции" } }
  ],
  "usage": { "search_units": 1, "total_tokens": 24, "cost_rub": 0.1035, "cost": 0.001, "fx_rate": 103.5 }
}

Модели реранкинга тарифицируются за поиск или за входные токены — это указано на странице модели. Их список — GET/api/v1/rerank/models.