rerank_text5.py

# Iterate through the rank response and reorder the kb_responses and add relevance_score
# We're also filtering just for the most relevant results according to rerank_number_of_results
ranked_kb_responses = [
    {
        # Use the index value in rank_response to find the correct kb_response
        "text": kb_responses[rank_response['index']]["text"],
        "url": kb_responses[rank_response['index']]["url"],
        "relevance_score": rank_response['relevance_score']
    } for rank_response in rank_response_body['results']
]

return ranked_kb_responses