in skills/retrieval_augmented_generation/evaluation/provider_retrieval.py [0:0]
def retrieve_level_three(query, options, context):
# Step 1: Get initial results from the summary db
initial_results = db_rerank.search(query, k=20)
# Step 2: Re-rank results
reranked_results = _rerank_results(query, initial_results, k=3)
# Step 3: Generate new context string from re-ranked results
new_context = ""
for result in reranked_results:
chunk = result['metadata']
new_context += f"\n <document> \n {chunk['chunk_heading']}\n\n{chunk['text']} \n </document> \n"
outputs = []
for result in reranked_results:
outputs.append(result['metadata']['chunk_link'])
print(outputs)
result = {"output": outputs}
return result