import sys
import os
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
from naive_rag import NaiveRAG
import json

def test():
    print("Initializing NaiveRAG (FastEmbed + Qdrant)...")
    try:
        rag = NaiveRAG()
        query = "Genesis System"
        
        print(f"Embedding query: '{query}'...")
        # This uses local FastEmbed, so no API key needed
        
        print("Retrieving from Qdrant...")
        results = rag.retrieve(query, limit=3)
        
        print(f"\nSuccess! Found {len(results)} results:")
        for i, res in enumerate(results):
            print(f"[{i+1}] Score: {res['score']:.4f}")
            print(f"    Source: {res['metadata'].get('source', 'Unknown')}")
            print(f"    Content: {res['content'][:100]}...")
            
    except Exception as e:
        print(f"\nFAILED: {e}")
        import traceback
        traceback.print_exc()

if __name__ == "__main__":
    test()