1import os
2import qianfan
3from qianfan.resources import Reranker
4
5# 使用安全认证AK/SK鉴权，通过环境变量方式初始化；替换下列示例中参数，安全认证Access Key替换your_iam_ak，Secret Key替换your_iam_sk
6os.environ["QIANFAN_ACCESS_KEY"] = "your_iam_ak"
7os.environ["QIANFAN_SECRET_KEY"] = "your_iam_sk"
8
9# 默认模型
10r = Reranker()
11
12# 使用指定reranker模型，固定值为bce-reranker-base_v1
13# r = Reranker(model="bce-reranker-base_v1") 
14
15res = r.do("北京的天气", ["北京今天12.5度，北风，阴天", "北京美食很多"])
16print(res.body)

1{
2    'id': 'as-hkck49jsue', 
3    'object': 'reranker_list', 
4    'created': 1715067228, 
5    'results': 
6        [
7          {
8             'document': '北京今天12.5度，北风，阴天', 
9              'relevance_score': 0.5633178949356079, 
10              'index': 0
11            }, 
12            {
13              'document': '北京美食很多', 
14              'relevance_score': 0.4250681698322296, 
15              'index': 1
16            }
17        ], 
18     'usage': 
19        {
20           'prompt_tokens': 22, 
21           'total_tokens': 22
22        }
23}