from retriever import retrieve

test_queries = [

    "Hiring a Java developer with communication skills",

    "Looking for a personality assessment for managers",

    "Need coding assessments for Python developers",

    "Hiring customer service employees",

    "Need cognitive ability tests"

]

for q in test_queries:

    print("\n" + "="*70)
    print("QUERY:", q)
    print("="*70)

    results = retrieve(q, k=5)

    for i, r in enumerate(results,1):

        print(f"\n{i}. {r['name']}")
        print(f"URL: {r['url']}")
        print(f"Type: {r['test_type']}")