An Overview of Cohere’s Rerank Model

1
import cohere
2
3
co = cohere.Client(api_key="<YOUR API KEY>")
4
5
query = "What is the capital of the United States?"
6
docs = [
7
    "Carson City is the capital city of the American state of Nevada. At the 2010 United States Census, Carson City had a population of 55,274.",
8
    "The Commonwealth of the Northern Mariana Islands is a group of islands in the Pacific Ocean that are a political division controlled by the United States. Its capital is Saipan.",
9
    "Charlotte Amalie is the capital and largest city of the United States Virgin Islands. It has about 20,000 people. The city is on the island of Saint Thomas.",
10
    "Washington, D.C. (also known as simply Washington or D.C., and officially as the District of Columbia) is the capital of the United States. It is a federal district. The President of the USA and many major national government offices are in the territory. This makes it the political center of the United States of America.",
11
    "Capital punishment has existed in the United States since before the United States was a country. As of 2017, capital punishment is legal in 30 of the 50 states. The federal government (including the United States military) also uses capital punishment.",
12
]
13
results = co.rerank(
14
    model="rerank-v3.5",
15
    query=query,
16
    documents=docs,
17
    top_n=5,
18
    return_documents=True,
19
)

1
query = "What is the capital of the United States?"
2
docs = [
3
    {
4
        "Title": "Facts about Carson City",
5
        "Content": "Carson City is the capital city of the American state of Nevada. At the 2010 United States Census, Carson City had a population of 55,274.",
6
    },
7
    {
8
        "Title": "The Commonwealth of Northern Mariana Islands",
9
        "Content": "The Commonwealth of the Northern Mariana Islands is a group of islands in the Pacific Ocean that are a political division controlled by the United States. Its capital is Saipan.",
10
    },
11
    {
12
        "Title": "The Capital of United States Virgin Islands",
13
        "Content": "Charlotte Amalie is the capital and largest city of the United States Virgin Islands. It has about 20,000 people. The city is on the island of Saint Thomas.",
14
    },
15
    {
16
        "Title": "Washington D.C.",
17
        "Content": "Washington, D.C. (also known as simply Washington or D.C., and officially as the District of Columbia) is the capital of the United States. It is a federal district. The President of the USA and many major national government offices are in the territory. This makes it the political center of the United States of America.",
18
    },
19
    {
20
        "Title": "Capital Punishment in the US",
21
        "Content": "Capital punishment has existed in the United States since before the United States was a country. As of 2017, capital punishment is legal in 30 of the 50 states. The federal government (including the United States military) also uses capital punishment.",
22
    },
23
]
24
results = co.rerank(
25
    model="rerank-v3.5",
26
    query=query,
27
    documents=docs,
28
    rank_fields=["Title", "Content"],
29
    top_n=5,
30
    return_documents=True,
31
)

1
RerankResponse(
2
    id="e8f55f3f-d86e-47d7-9b24-7feb18286505",
3
    results=[
4
        RerankResponseResultsItem(
5
            document=RerankResponseResultsItemDocument(
6
                text=None,
7
                Content=(
8
                    "Washington, D.C. (also known as simply Washington or D.C., and officially as the District of Columbia) "
9
                    "is the capital of the United States. It is a federal district. The President of the USA and many major "
10
                    "national government offices are in the territory. This makes it the political center of the United States of America."
11
                ),
12
                Title="Washington D.C.",
13
            ),
14
            index=3,
15
            relevance_score=0.8914433,
16
        ),
17
        RerankResponseResultsItem(
18
            document=RerankResponseResultsItemDocument(
19
                text=None,
20
                Content=(
21
                    "Charlotte Amalie is the capital and largest city of the United States Virgin Islands. "
22
                    "It has about 20,000 people. The city is on the island of Saint Thomas."
23
                ),
24
                Title="The Capital of United States Virgin Islands",
25
            ),
26
            index=2,
27
            relevance_score=0.40344992,
28
        ),
29
        RerankResponseResultsItem(
30
            document=RerankResponseResultsItemDocument(
31
                text=None,
32
                Content=(
33
                    "Carson City is the capital city of the American state of Nevada. At the 2010 United States Census, "
34
                    "Carson City had a population of 55,274."
35
                ),
36
                Title="Facts about Carson City",
37
            ),
38
            index=0,
39
            relevance_score=0.23343581,
40
        ),
41
        RerankResponseResultsItem(
42
            document=RerankResponseResultsItemDocument(
43
                text=None,
44
                Content=(
45
                    "The Commonwealth of the Northern Mariana Islands is a group of islands in the Pacific Ocean that "
46
                    "are a political division controlled by the United States. Its capital is Saipan."
47
                ),
48
                Title="The Commonwealth of Northern Mariana Islands",
49
            ),
50
            index=1,
51
            relevance_score=0.15964958,
52
        ),
53
        RerankResponseResultsItem(
54
            document=RerankResponseResultsItemDocument(
55
                text=None,
56
                Content=(
57
                    "Capital punishment has existed in the United States since before the United States was a country. "
58
                    "As of 2017, capital punishment is legal in 30 of the 50 states. The federal government (including the United States military) "
59
                    "also uses capital punishment."
60
                ),
61
                Title="Capital Punishment in the US",
62
            ),
63
            index=4,
64
            relevance_score=0.10465127,
65
        ),
66
    ],
67
    meta=ApiMeta(
68
        api_version=ApiMetaApiVersion(
69
            version="1", is_deprecated=None, is_experimental=None
70
        ),
71
        billed_units=ApiMetaBilledUnits(
72
            images=None,
73
            input_tokens=None,
74
            output_tokens=None,
75
            search_units=1.0,
76
            classifications=None,
77
        ),
78
        tokens=None,
79
        warnings=None,
80
    ),
81
)