File size: 2,937 Bytes
03f412b |
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 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
import sys
from pathlib import Path
import asyncio
# Add project root to path
project_root = Path(__file__).parent
sys.path.append(str(project_root))
from core.coordinator import coordinator
from core.session import session_manager
from services.hf_endpoint_monitor import hf_monitor
from utils.config import config
def test_hf_space_readiness():
"""Test if system is ready for HF Space deployment"""
print("=== HF Space Deployment Readiness Test ===")
print()
readiness_score = 0
total_tests = 5
# Test 1: Core infrastructure
print("1. Core Infrastructure:")
try:
from core.redis_client import redis_client
from core.llm_factory import llm_factory
print("✅ Core modules import successfully")
readiness_score += 1
except Exception as e:
print(f"❌ Core modules failed: {e}")
print()
# Test 2: Coordinator system
print("2. Coordinator System:")
try:
# Test coordinator instantiation
test_coordinator = coordinator
print("✅ Coordinator system ready")
readiness_score += 1
except Exception as e:
print(f"❌ Coordinator system failed: {e}")
print()
# Test 3: Service monitors
print("3. Service Monitors:")
try:
# Test HF monitor
hf_monitor_instance = hf_monitor
print("✅ Service monitors ready")
readiness_score += 1
except Exception as e:
print(f"❌ Service monitors failed: {e}")
print()
# Test 4: Session management
print("4. Session Management:")
try:
# Test session system
session = session_manager.get_session("test_user_hf")
session_manager.clear_session("test_user_hf")
print("✅ Session management ready")
readiness_score += 1
except Exception as e:
print(f"❌ Session management failed: {e}")
print()
# Test 5: Configuration awareness
print("5. Configuration Awareness:")
print(f" HF Space Detected: {'✅' if config.is_hf_space else 'ℹ️ Local'}")
print(f" Ollama Configured: {'✅' if config.ollama_host else '❌'}")
print(f" HF Token Available: {'✅' if config.hf_token else '❌'}")
readiness_score += 1 # Always count this as ready
print()
# Final assessment
print("=== Readiness Assessment ===")
percentage = (readiness_score / total_tests) * 100
print(f"Readiness Score: {readiness_score}/{total_tests} ({percentage:.0f}%)")
if percentage >= 80:
print("🎉 System is ready for HF Space deployment!")
print("✅ All core systems operational")
print("✅ Coordinator architecture implemented")
print("✅ Service monitoring ready")
else:
print("⚠️ System needs attention before HF Space deployment")
return readiness_score >= 4
if __name__ == "__main__":
test_hf_space_readiness()
|