server/scripts/check_index.py

import os
import sys
import asyncio
import pickle

# Add server directory to path
sys.path.append(os.path.join(os.getcwd(), 'server'))

from app.services.medical_retriever import get_medical_retriever

async def check_index():
    retriever = get_medical_retriever()
    print(f"Total documents in index: {len(retriever.documents)}")
    
    sources = {}
    for meta in retriever.doc_metadata:
        src = meta.get('source', 'Unknown')
        sources[src] = sources.get(src, 0) + 1
    
    print("Sources breakdown:")
    for src, count in sources.items():
        print(f"  - {src}: {count}")

if __name__ == "__main__":
    asyncio.run(check_index())