def retrieve_level_two()

in skills/retrieval_augmented_generation/evaluation/prompts.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

def retrieve_level_two(query):
    results = db_summary.search(query, k=3)
    context = ""
    for result in results:
        chunk = result['metadata']
        context += f"\n <document> \n {chunk['chunk_heading']}\n\nText\n {chunk['text']} \n\nSummary: \n {chunk['summary']} \n </document> \n" #show model all 3 items
    return results, context