File size: 1,264 Bytes
e900a8d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
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()
|