tests/test_basic.py

"""
MediGuard AI RAG-Helper - SIMPLIFIED TEST
Tests the multi-agent workflow with a diabetes patient case
"""

import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).parent.parent))

# Test if we can at least import everything
print("Testing imports...")

try:
    from src.state import PatientInput
    print("PatientInput imported")
    
    from src.config import BASELINE_SOP
    print("BASELINE_SOP imported")
    
    from src.pdf_processor import get_all_retrievers
    print("get_all_retrievers imported")
    
    from src.llm_config import llm_config
    print("llm_config imported")
    
    from src.biomarker_validator import BiomarkerValidator
    print("BiomarkerValidator imported")
    
    print("\n" + "="*70)
    print("ALL IMPORTS SUCCESSFUL")
    print("="*70)
    
    # Test retrievers
    print("\nTesting retrievers...")
    retrievers = get_all_retrievers(force_rebuild=False)
    print(f"Retrieved {len(retrievers)} retrievers")
    print(f"  Available: {list(retrievers.keys())}")
    
    # Test patient input creation
    print("\nTesting PatientInput creation...")
    patient = PatientInput(
        biomarkers={"Glucose": 185.0, "HbA1c": 8.2},
        model_prediction={"disease": "Type 2 Diabetes", "confidence": 0.87, "probabilities": {}},
        patient_context={"age": 52, "gender": "male", "bmi": 31.2}
    )
    print("PatientInput created")
    print(f"  Disease: {patient.model_prediction['disease']}")
    print(f"  Confidence: {patient.model_prediction['confidence']:.1%}")
    
    # Test biomarker validator
    print("\nTesting BiomarkerValidator...")
    validator = BiomarkerValidator()
    flags, alerts = validator.validate_all(patient.biomarkers, patient.patient_context.get('gender', 'male'))
    print("Validator working")
    print(f"  Flags: {len(flags)}")
    print(f"  Alerts: {len(alerts)}")
    
    print("\n" + "="*70)
    print("BASIC SYSTEM TEST PASSED!")
    print("="*70)
    print("\nNote: Full workflow integration requires state refactoring.")
    print("All core components are functional and ready.")
    
except Exception as e:
    print(f"\nERROR: {e}")
    import traceback
    traceback.print_exc()