Reranking with Cohere

1
# pip install cohere
2
3
import cohere
4
5
# Get your free API key: https://dashboard.cohere.com/api-keys
6
co = cohere.ClientV2(api_key="COHERE_API_KEY")

1
# Define the documents
2
faqs = [
3
    {
4
        "text": "Reimbursing Travel Expenses: Easily manage your travel expenses by submitting them through our finance tool. Approvals are prompt and straightforward."
5
    },
6
    {
7
        "text": "Working from Abroad: Working remotely from another country is possible. Simply coordinate with your manager and ensure your availability during core hours."
8
    },
9
    {
10
        "text": "Health and Wellness Benefits: We care about your well-being and offer gym memberships, on-site yoga classes, and comprehensive health insurance."
11
    },
12
    {
13
        "text": "Performance Reviews Frequency: We conduct informal check-ins every quarter and formal performance reviews twice a year."
14
    },
15
]

1
# Add the user query
2
query = "Are there fitness-related perks?"
3
4
# Rerank the documents
5
results = co.rerank(
6
    model="rerank-v3.5",
7
    query=query,
8
    documents=faqs,
9
    top_n=1,
10
)
11
12
print(results)

1
# Display the reranking results
2
def return_results(results, documents):
3
    for idx, result in enumerate(results.results):
4
        print(f"Rank: {idx+1}")
5
        print(f"Score: {result.relevance_score}")
6
        print(f"Document: {documents[result.index]}\n")
7
8
9
return_results(results, faqs_short)

1
# Define the documents
2
emails = [
3
    {
4
        "from": "hr@co1t.com",
5
        "to": "david@co1t.com",
6
        "date": "2024-06-24",
7
        "subject": "A Warm Welcome to Co1t!",
8
        "text": "We are delighted to welcome you to the team! As you embark on your journey with us, you'll find attached an agenda to guide you through your first week.",
9
    },
10
    {
11
        "from": "it@co1t.com",
12
        "to": "david@co1t.com",
13
        "date": "2024-06-24",
14
        "subject": "Setting Up Your IT Needs",
15
        "text": "Greetings! To ensure a seamless start, please refer to the attached comprehensive guide, which will assist you in setting up all your work accounts.",
16
    },
17
    {
18
        "from": "john@co1t.com",
19
        "to": "david@co1t.com",
20
        "date": "2024-06-24",
21
        "subject": "First Week Check-In",
22
        "text": "Hello! I hope you're settling in well. Let's connect briefly tomorrow to discuss how your first week has been going. Also, make sure to join us for a welcoming lunch this Thursday at noon—it's a great opportunity to get to know your colleagues!",
23
    },
24
]

1
# Convert the documents to YAML format
2
yaml_docs = [yaml.dump(doc, sort_keys=False) for doc in emails]
3
4
# Add the user query
5
query = "Any email about check ins?"
6
7
# Rerank the documents
8
results = co.rerank(
9
    model="rerank-v3.5",
10
    query=query,
11
    documents=yaml_docs,
12
    top_n=2,
13
)
14
15
return_results(results, emails)

1
import pandas as pd
2
from io import StringIO
3
4
# Create a demo CSV file
5
data = """name,role,join_date,email,status
6
Rebecca Lee,Senior Software Engineer,2024-07-01,rebecca@co1t.com,Full-time
7
Emma Williams,Product Designer,2024-06-15,emma@co1t.com,Full-time
8
Michael Jones,Marketing Manager,2024-05-20,michael@co1t.com,Full-time
9
Amelia Thompson,Sales Representative,2024-05-20,amelia@co1t.com,Part-time
10
Ethan Davis,Product Designer,2024-05-25,ethan@co1t.com,Contractor"""
11
data_csv = StringIO(data)
12
13
# Load the CSV file
14
df = pd.read_csv(data_csv)
15
df.head(1)

1
# Define the documents
2
employees = df.to_dict("records")
3
4
# Convert the documents to YAML format
5
yaml_docs = [yaml.dump(doc, sort_keys=False) for doc in employees]
6
7
# Add the user query
8
query = "Any full-time product designers who joined recently?"
9
10
# Rerank the documents
11
results = co.rerank(
12
    model="rerank-v3.5",
13
    query=query,
14
    documents=yaml_docs,
15
    top_n=1,
16
)
17
return_results(results, employees)

1
# Define the query
2
query = "هل هناك مزايا تتعلق باللياقة البدنية؟"  # Are there fitness benefits?
3
4
# Rerank the documents
5
results = co.rerank(
6
    model="rerank-v3.5",
7
    query=query,
8
    documents=faqs,
9
    top_n=1,
10
)
11
12
return_results(results, faqs)