Skip to content

文本排序

本页说明文本排序(rerank)接口,采用 models-model.md 风格:概览、认证、参数表、请求示例、返回示例。


概览

  • 请求方法:POST
  • 路径:/v1/rerank
  • 内容类型:application/json

认证

  • 使用 HTTP Bearer Token,示例:Authorization: Bearer sk-xxxxx

请求参数

参数类型必填描述
modelstring排序模型名称,例如 gte-rerank-v2
querystring查询提示词,用于与候选文档匹配并排序
documentsarray[string]候选文档列表(字符串数组),将按与 query 的相关性排序
top_ninteger返回的前 N 条文档,未指定则返回全部候选文档
return_documentsboolean返回结果中是否包含原始文档文本,默认 false

请求示例

json
{
  "model": "gte-rerank-v2",
  "query": "什么是文本排序模型",
  "documents": [
    "文本排序模型广泛用于搜索引擎和推荐系统中,它们根据文本相关性对候选文本进行排序",
    "量子计算是计算科学的一个前沿领域",
    "预训练语言模型的发展给文本排序模型带来了新的进展"
  ],
  "return_documents": true,
  "top_n": 5
}

curl

bash
curl -X POST "https://api.gpt.ge/v1/rerank" \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer sk-xxxx" \
  -d '{"model":"gte-rerank-v2","query":"什么是文本排序模型","documents":["文本排序模型广泛用于搜索引擎和推荐系统中,它们根据文本相关性对候选文本进行排序","量子计算是计算科学的一个前沿领域","预训练语言模型的发展给文本排序模型带来了新的进展"],"return_documents":true,"top_n":5}'

JavaScript (fetch)

javascript
fetch('https://api.gpt.ge/v1/rerank', {
  method: 'POST',
  headers: { 'Content-Type': 'application/json', 'Authorization': 'Bearer sk-xxxx' },
  body: JSON.stringify({ model: 'gte-rerank-v2', query: '什么是文本排序模型', documents: ['文本排序模型广泛用于搜索引擎和推荐系统中,它们根据文本相关性对候选文本进行排序','量子计算是计算科学的一个前沿领域','预训练语言模型的发展给文本排序模型带来了新的进展'], return_documents: true, top_n: 5 })
}).then(r => r.json()).then(console.log)

Python (requests)

python
import requests

payload = {
  'model': 'gte-rerank-v2',
  'query': '什么是文本排序模型',
  'documents': [
    '文本排序模型广泛用于搜索引擎和推荐系统中,它们根据文本相关性对候选文本进行排序',
    '量子计算是计算科学的一个前沿领域',
    '预训练语言模型的发展给文本排序模型带来了新的进展'
  ],
  'return_documents': True,
  'top_n': 5
}

resp = requests.post('https://api.gpt.ge/v1/rerank', headers={'Content-Type':'application/json','Authorization':'Bearer sk-xxxx'}, json=payload)
print(resp.json())

返回示例(200)

json
{
  "results": [
    {
      "document": { "text": "文本排序模型广泛用于搜索引擎和推荐系统中,它们根据文本相关性对候选文本进行排序" },
      "index": 0,
      "relevance_score": 0.9334521178273196
    },
    {
      "document": { "text": "预训练语言模型的发展给文本排序模型带来了新的进展" },
      "index": 2,
      "relevance_score": 0.34100082626411193
    },
    {
      "document": { "text": "量子计算是计算科学的一个前沿领域" },
      "index": 1,
      "relevance_score": 0.005814161578735119
    }
  ],
  "usage": { "prompt_tokens": 79, "total_tokens": 79 }
}