def retrieve_base()

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

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

def retrieve_base(query, options, context):
    input_query = context['vars']['query']
    results = db.search(input_query, k=3)
    outputs = []
    for result in results:
        outputs.append(result['metadata']['chunk_link'])
    print(outputs)
    result = {"output": outputs}
    return result