test_hf_monitor.py

import sys
from pathlib import Path

# Add project root to path
project_root = Path(__file__).parent
sys.path.append(str(project_root))

from services.hf_endpoint_monitor import hf_monitor

def test_hf_monitor():
    """Test the HF endpoint monitor"""
    print("=== HF Endpoint Monitor Test ===")
    print()
    
    # Show current status
    print("Current HF Endpoint Status:")
    status = hf_monitor.check_endpoint_status()
    print(f"  Available: {status['available']}")
    print(f"  Status Code: {status['status_code']}")
    print(f"  Initialized: {status.get('initialized', 'Unknown')}")
    if 'error' in status:
        print(f"  Error: {status['error']}")
    print()
    
    # Show human-readable status
    print("Human-Readable Status:")
    print(hf_monitor.get_status_summary())
    print()
    
    # Try to warm up endpoint if not available
    if not status['available']:
        print("Attempting to warm up endpoint...")
        success = hf_monitor.warm_up_endpoint()
        print(f"Warm-up result: {'Success' if success else 'Failed'}")
        print()
        
        # Check status again
        print("Status after warm-up attempt:")
        print(hf_monitor.get_status_summary())

if __name__ == "__main__":
    test_hf_monitor()