AI重排序API：优化搜索相关性

小小管写大大码2025-10-24 13:37

目录

1.前期准备

[2. 资源准备](#2. 资源准备)

[3. 📝 简介](#3. 📝 简介)

[4. 💡 请求示例](#4. 💡 请求示例)

[4.1 基础重排序请求（shell）](#4.1 基础重排序请求（shell）)

[4.2 使用结构化数据（shell）](#4.2 使用结构化数据（shell）)

[4.3 Python代码实现(共三种方式，可选择任意一种方式)](#4.3 Python代码实现(共三种方式，可选择任意一种方式))

1.前期准备

1.1准备APi官网

1.2准备软件

2. 资源准备

API Key：此项配置填写在艾可API官网创建API令牌，一键直达API令牌创建页面

创建API令牌步骤请参考API Key的获取和使用

API Host：此项配置填写https://aicanapi.com/v1

查看支持的模型请参在这里复制模型在线查询

3. 📝 简介

给定查询和文本列表，重排序API将根据与查询的相关性对文本进行排序。每个文本都会被分配一个相关性分数，从而产生一个有序的数组结果。此功能特别适用于搜索和检索应用，可以优化文档的排序，帮助用户更快找到相关信息。

4. 💡 请求示例

4.1 基础重排序请求（shell）

复制代码 
curl https://aicanapi.com/v1/rerank \
  -H "Authorization: Bearer $NEWAPI_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "BAAI/bge-reranker-v2-m3",
    "query": "什么是美国的首都？",
    "documents": [
      "内华达州的首府是卡森城。",
      "北马里亚纳群岛是太平洋上的一组岛屿，其首都是塞班岛。",
      "华盛顿特区（也称为华盛顿或特区，正式名称为哥伦比亚特区）是美国的首都。",
      "英语语法中的大写是在单词开头使用大写字母。英语用法与其他语言的大写不同。",
      "自美国成为一个国家之前，美国就存在死刑。截至2017年，在50个州中有30个州死刑合法。"
    ],
    "top_n": 3
  }'

响应示例：

复制代码 
{
  "results": [
    {
      "index": 2,
      "relevance_score": 0.999071
    },
    {
      "index": 0,
      "relevance_score": 0.32713068
    },
    {
      "index": 1,
      "relevance_score": 0.1867867
    }
  ],
  "id": "07734bd2-2473-4f07-94e1-0d9f0e6843cf",
  "meta": {
    "api_version": {
      "version": "2",
      "is_experimental": false
    },
    "billed_units": {
      "search_units": 1
    }
  }
}

4.2 使用结构化数据（shell）

复制代码 
curl https://aicanapi.com/v1/rerank \
  -H "Authorization: Bearer $NEWAPI_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "BAAI/bge-reranker-v2-m3",
    "query": "寻找适合初学者的高性价比单反相机",
    "documents": [
      "型号: Canon EOS 800D\n价格: 4299元\n功能: 2410万像素, 光学取景器, Wi-Fi\n适合人群: 初学者, 爱好者",
      "型号: Nikon D3500\n价格: 3099元\n功能: 2416万像素, 光学取景器, 长达1550张的电池续航\n适合人群: 新手, 学生",
      "型号: Sony A7III\n价格: 12999元\n功能: 2420万像素, 全画幅, 4K视频\n适合人群: 专业摄影师, 视频创作者"
    ],
    "max_tokens_per_doc": 512
  }'

响应示例：

复制代码 
curl https:/aicanapi.com/v1/rerank \
  -H "Authorization: Bearer $NEWAPI_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "rerank-v3.5",
    "query": "寻找适合初学者的高性价比单反相机",
    "documents": [
      "型号: Canon EOS 800D\n价格: 4299元\n功能: 2410万像素, 光学取景器, Wi-Fi\n适合人群: 初学者, 爱好者",
      "型号: Nikon D3500\n价格: 3099元\n功能: 2416万像素, 光学取景器, 长达1550张的电池续航\n适合人群: 新手, 学生",
      "型号: Sony A7III\n价格: 12999元\n功能: 2420万像素, 全画幅, 4K视频\n适合人群: 专业摄影师, 视频创作者"
    ],
    "max_tokens_per_doc": 512
  }'

4.3 Python代码实现(共三种方式，可选择任意一种方式)

复制代码 
import json

import requests


# 方式一：使用 requests 库（推荐）
def call_rerank_api_requests():
    """使用 requests 库调用排序模型API"""

    # API配置
    url = "https://aicanapi.com/v1/rerank"
    headers = {
        "Authorization": "Bearer sk-xxxxx",
        "Content-Type": "application/json"
    }

    # 请求数据
    data = {
        "model": "BAAI/bge-reranker-v2-m3",
        "query": "什么是美国的首都？",
        "documents": [
            "内华达州的首府是卡森城。",
            "北马里亚纳群岛是太平洋上的一组岛屿，其首都是塞班岛。",
            "华盛顿特区（也称为华盛顿或特区，正式名称为哥伦比亚特区）是美国的首都。",
            "英语语法中的大写是在单词开头使用大写字母。英语用法与其他语言的大写不同。",
            "自美国成为一个国家之前，美国就存在死刑。截至2017年，在50个州中有30个州死刑合法。"
        ],
        "top_n": 3
    }

    try:
        # 发送POST请求
        response = requests.post(url, headers=headers, json=data)

        # 检查响应状态
        if response.status_code == 200:
            result = response.json()
            print("API调用成功！")
            print(f"响应数据: {json.dumps(result, ensure_ascii=False, indent=2)}")
            return result
        else:
            print(f"API调用失败，状态码: {response.status_code}")
            print(f"错误信息: {response.text}")
            return None

    except requests.exceptions.RequestException as e:
        print(f"请求异常: {e}")
        return None


# 方式二：使用 urllib（标准库，无需额外安装）
def call_rerank_api_urllib():
    """使用 urllib 库调用排序模型API"""

    import urllib.request
    import urllib.parse

    # API配置
    url = "https://aicanapi.com/v1/rerank"
    headers = {
        "Authorization": "Bearer sk-xxxxx",
        "Content-Type": "application/json"
    }

    # 请求数据
    data = {
        "model": "BAAI/bge-reranker-v2-m3",
        "query": "什么是美国的首都？",
        "documents": [
            "内华达州的首府是卡森城。",
            "北马里亚纳群岛是太平洋上的一组岛屿，其首都是塞班岛。",
            "华盛顿特区（也称为华盛顿或特区，正式名称为哥伦比亚特区）是美国的首都。",
            "英语语法中的大写是在单词开头使用大写字母。英语用法与其他语言的大写不同。",
            "自美国成为一个国家之前，美国就存在死刑。截至2017年，在50个州中有30个州死刑合法。"
        ],
        "top_n": 3
    }

    try:
        # 将数据转换为JSON字符串并编码
        json_data = json.dumps(data, ensure_ascii=False).encode('utf-8')

        # 创建请求对象
        req = urllib.request.Request(url, data=json_data, headers=headers)

        # 发送请求
        with urllib.request.urlopen(req) as response:
            result_data = response.read().decode('utf-8')
            result = json.loads(result_data)

            print("API调用成功！")
            print(f"响应数据: {json.dumps(result, ensure_ascii=False, indent=2)}")
            return result

    except Exception as e:
        print(f"请求异常: {e}")
        return None


# 封装成类的形式（更适合集成到项目中）
class RerankAPI:
    def __init__(self, api_key, base_url="https://aicanapi.com/v1"):
        self.api_key = api_key
        self.base_url = base_url

    def rerank(self, query, documents, model="BAAI/bge-reranker-v2-m3", top_n=3):
        """
        调用排序模型API

        Args:
            query (str): 查询字符串
            documents (list): 文档列表
            model (str): 模型名称
            top_n (int): 返回前N个结果

        Returns:
            dict: API响应结果
        """
        url = f"{self.base_url}/rerank"
        headers = {
            "Authorization": f"Bearer {self.api_key}",
            "Content-Type": "application/json"
        }

        data = {
            "model": model,
            "query": query,
            "documents": documents,
            "top_n": top_n
        }

        try:
            response = requests.post(url, headers=headers, json=data)

            if response.status_code == 200:
                return response.json()
            else:
                raise Exception(f"API调用失败，状态码: {response.status_code}, 错误信息: {response.text}")

        except requests.exceptions.RequestException as e:
            raise Exception(f"请求异常: {e}")


# 主函数 - 运行示例
if __name__ == "__main__":
    print("=== 方式一：使用 requests 库 ===")
    result1 = call_rerank_api_requests()

    print("\n=== 方式二：使用 urllib 库 ===")
    result2 = call_rerank_api_urllib()

    print("\n=== 方式三：使用类封装 ===")
    # 创建API实例
    api = RerankAPI("sk-xxxxx")

    # 调用API
    try:
        result3 = api.rerank(
            query="什么是美国的首都？",
            documents=[
                "内华达州的首府是卡森城。",
                "北马里亚纳群岛是太平洋上的一组岛屿，其首都是塞班岛。",
                "华盛顿特区（也称为华盛顿或特区，正式名称为哥伦比亚特区）是美国的首都。",
                "英语语法中的大写是在单词开头使用大写字母。英语用法与其他语言的大写不同。",
                "自美国成为一个国家之前，美国就存在死刑。截至2017年，在50个州中有30个州死刑合法。"
            ],
            top_n=3
        )
        print("API调用成功！")
        print(f"响应数据: {json.dumps(result3, ensure_ascii=False, indent=2)}")
    except Exception as e:
        print(f"API调用失败: {e}")
相关推荐
OG one.Z2 小时前
07_朴素贝叶斯
人工智能·机器学习
智能相对论3 小时前
把AI装进OS、批量落地智慧服务，智能手机革命2.0来了
人工智能·智能手机
flying_13143 小时前
图神经网络分享系列-GAT(GRAPH ATTENTION NETWORKS) (一)
人工智能·神经网络·图神经网络·注意力机制·gnn·gat·图注意力网络
周末程序猿3 小时前
谈谈 `Claude Skills`
人工智能·ai编程
IT_陈寒4 小时前
5个Vue3性能优化技巧，让你的应用提速50% 🚀（附实测对比）
前端·人工智能·后端
kalvin_y_liu4 小时前
微软Agent Framework
人工智能·microsoft
ximy13354 小时前
AI服务器工作之显卡测试
人工智能
孤独野指针*P4 小时前
深度学习之美》读书笔记 - 第一章 & 第二章
人工智能·深度学习
努力学习的少女4 小时前
SpaekSql函数
前端·数据库
热门推荐
01BongoCat - 跨平台键盘猫动画工具02GitHub 镜像站点03UV安装并设置国内源04GitLab 零基础入门指南：从安装到项目管理全流程05Linux下V2Ray安装配置指南06Labelme从安装到标注：零基础完整指南072025软件测试面试八股文（含答案+文档）08NVIDIA显卡驱动、CUDA、cuDNN 和 TensorRT 版本匹配指南09在VSCode配置Java开发环境的保姆级教程（适配各类AI编程IDE）10【vue篇】Vue 项目中的静态资源管理：assets vs static 终极指南