import sys
from pathlib import Path
from dotenv import load_dotenv

sys.path.insert(0, str(Path(__file__).resolve().parents[0]))
load_dotenv(override=True)

from src.retrieval import retrieve

def run_test():
    try:
        print("Executing Hybrid Search for 'Pemetrexed'...")
        results = retrieve("What is the standard recommended dosage for Pemetrexed in patients with non-small cell lung cancer?", top_k=5)
        
        if not results:
            print("❌ HYBRID SEARCH RETURNED 0 RESULTS")
            return
            
        print(f"\n✅ HYBRID SEARCH SUCCESSFUL! Retrieved {len(results)} citations.")
        for i, (text, score) in enumerate(results):
            print(f"[{i+1}] Score: {score:.4f} | {text[:100]}...")
            
    except Exception as e:
        print(f"❌ HYBRID SEARCH FAILED: {str(e)}")
        import traceback
        traceback.print_exc()

if __name__ == "__main__":
    run_test()