def _retrieve_base()

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

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

def _retrieve_base(query, db):
    results = db.search(query, k=3)
    context = ""
    for result in results:
        chunk = result['metadata']
        context += f"\n{chunk['text']}\n"
    return results, context