Reranking with Cohere

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

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

1
# Add the user query
2
query = "Are there fitness-related perks?"
3
4
# Rerank the documents
5
results = co.rerank(query=query,
6
                    documents=faqs_short,
7
                    top_n=2,
8
                    model='rerank-english-v3.0')
9
10
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
return_results(results, faqs_short)

1
# Define the documents
2
emails = [
3
    {"from": "hr@co1t.com", "to": "david@co1t.com", "date": "2024-06-24", "subject": "A Warm Welcome to Co1t!", "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."},
4
    {"from": "it@co1t.com", "to": "david@co1t.com", "date": "2024-06-24", "subject": "Setting Up Your IT Needs", "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."},
5
    {"from": "john@co1t.com", "to": "david@co1t.com", "date": "2024-06-24", "subject": "First Week Check-In", "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!"}
6
]

1
# Add the user query
2
query = "Any email about check ins?"
3
4
# Rerank the documents
5
results = co.rerank(query=query,
6
                    documents=emails,
7
                    top_n=2,
8
                    model='rerank-english-v3.0',
9
                    rank_fields=["from", "to", "date", "subject", "body"])
10
11
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 and rank fields
2
employees = df.to_dict('records')
3
rank_fields = df.columns.tolist()
4
5
# Add the user query
6
query = "Any full-time product designers who joined recently?"
7
8
# Rerank the documents
9
results = co.rerank(query=query,
10
                    documents=employees,
11
                    top_n=1,
12
                    model='rerank-english-v3.0',
13
                    rank_fields=rank_fields)
14
15
return_results(results, employees)

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