src/retrieval/test.py

from query import retrieve

if __name__ == "__main__":
    q = "how to install transformers"
    results, debug = retrieve(q)

    for i, r in enumerate(results):
        print(f"\n--- Result {i+1} ---")
        print("Score:", r["rerank_score"])
        print(r["text"][:300])

    print("\n--- DEBUG INFO ---")
    print(debug)