test_setup.py

#!/usr/bin/env python3
"""
Script de testing básico para verificar componentes de Fase -1
"""

import sys
import os
from pathlib import Path

# Añadir src al path
sys.path.insert(0, str(Path(__file__).parent / "src"))

print("=" * 60)
print("Test de Componentes - Fase -1")
print("=" * 60)
print()

# Test 1: Verificar imports
print("Test 1: Verificando imports...")
try:
    from utils.config import Config
    from utils.logging import setup_logging, get_logger
    print("  ✓ Utils importados correctamente")
except Exception as e:
    print(f"  ✗ Error importando utils: {e}")
    sys.exit(1)

# Test 2: Configuración
print("\nTest 2: Verificando configuración...")
try:
    config = Config()
    hotword = config.get('audio.hotword')
    print(f"  ✓ Configuración cargada")
    print(f"    - Hotword: {hotword}")
    print(f"    - Modelo STT: {config.get('audio.stt.model')}")
    print(f"    - Modelo LLM rápido: {config.get('llm.rapid.model')}")
except Exception as e:
    print(f"  ✗ Error en configuración: {e}")
    sys.exit(1)

# Test 3: API Key de Claude
print("\nTest 3: Verificando API key de Claude...")
try:
    from dotenv import load_dotenv
    load_dotenv()
    
    api_key = os.getenv('ANTHROPIC_API_KEY')
    if not api_key:
        print("  ✗ ANTHROPIC_API_KEY no encontrada en .env")
        print("    Edita .env y añade tu API key")
        sys.exit(1)
    
    if api_key == "tu_api_key_aqui":
        print("  ✗ API key no configurada (aún tiene valor por defecto)")
        print("    Edita .env con tu API key real")
        sys.exit(1)
    
    print("  ✓ API key encontrada")
except Exception as e:
    print(f"  ✗ Error verificando API key: {e}")
    sys.exit(1)

# Test 4: Ollama
print("\nTest 4: Verificando Ollama...")
try:
    import aiohttp
    import asyncio
    
    async def test_ollama():
        async with aiohttp.ClientSession() as session:
            try:
                async with session.get('http://localhost:11434/api/tags', timeout=5) as resp:
                    if resp.status == 200:
                        data = await resp.json()
                        models = [m['name'] for m in data.get('models', [])]
                        
                        if 'llama3.2:3b' in models:
                            print("  ✓ Ollama corriendo con llama3.2:3b")
                            return True
                        else:
                            print("  ✗ Modelo llama3.2:3b no encontrado")
                            print("    Ejecuta: ollama pull llama3.2:3b")
                            return False
                    else:
                        print("  ✗ Ollama respondió con error")
                        return False
            except asyncio.TimeoutError:
                print("  ✗ Timeout conectando a Ollama")
                print("    ¿Está Ollama corriendo? Ejecuta: ollama serve")
                return False
            except aiohttp.ClientConnectorError:
                print("  ✗ No se puede conectar a Ollama")
                print("    Ejecuta: ollama serve")
                return False
    
    result = asyncio.run(test_ollama())
    if not result:
        sys.exit(1)
        
except Exception as e:
    print(f"  ✗ Error verificando Ollama: {e}")
    sys.exit(1)

# Test 5: Whisper
print("\nTest 5: Verificando Whisper...")
try:
    import whisper
    print("  ✓ Whisper importado correctamente")
    print("    (Modelo se descargará en primer uso)")
except Exception as e:
    print(f"  ✗ Error importando Whisper: {e}")
    print("    Instala con: pip install openai-whisper")
    sys.exit(1)

# Test 6: Torch (para VAD)
print("\nTest 6: Verificando PyTorch...")
try:
    import torch
    print(f"  ✓ PyTorch {torch.__version__} instalado")
except Exception as e:
    print(f"  ✗ Error importando PyTorch: {e}")
    sys.exit(1)

# Test 7: Sounddevice
print("\nTest 7: Verificando sounddevice...")
try:
    import sounddevice as sd
    devices = sd.query_devices()
    print("  ✓ sounddevice funcionando")
    
    # Mostrar dispositivos de entrada
    input_devices = [d for d in devices if d['max_input_channels'] > 0]
    if input_devices:
        print(f"    Dispositivos de entrada encontrados: {len(input_devices)}")
    else:
        print("  ⚠ No se encontraron dispositivos de entrada")
        
except Exception as e:
    print(f"  ✗ Error con sounddevice: {e}")
    sys.exit(1)

# Test 8: Anthropic
print("\nTest 8: Verificando conexión con Claude API...")
try:
    from anthropic import Anthropic
    
    client = Anthropic(api_key=os.getenv('ANTHROPIC_API_KEY'))
    
    # Test simple
    try:
        message = client.messages.create(
            model="claude-sonnet-4-20250514",
            max_tokens=10,
            messages=[{"role": "user", "content": "test"}]
        )
        print("  ✓ Claude API funcionando")
    except Exception as e:
        print(f"  ✗ Error conectando con Claude API: {e}")
        print("    Verifica que tu API key es válida")
        sys.exit(1)
        
except Exception as e:
    print(f"  ✗ Error con Anthropic: {e}")
    sys.exit(1)

# Test 9: Directorios
print("\nTest 9: Verificando directorios...")
try:
    bandeja = Path("/Users/victorgarzon/Desktop/bandeja_entrada")
    data_dir = Path.home() / ".asistente_data"
    logs_dir = data_dir / "logs"
    
    if not bandeja.exists():
        print(f"  ⚠ Bandeja de entrada no existe, creando...")
        bandeja.mkdir(parents=True, exist_ok=True)
    
    if not data_dir.exists():
        print(f"  ⚠ Directorio de datos no existe, creando...")
        data_dir.mkdir(parents=True, exist_ok=True)
    
    if not logs_dir.exists():
        print(f"  ⚠ Directorio de logs no existe, creando...")
        logs_dir.mkdir(parents=True, exist_ok=True)
    
    print("  ✓ Todos los directorios existen")
    
except Exception as e:
    print(f"  ✗ Error con directorios: {e}")
    sys.exit(1)

# Resumen
print("\n" + "=" * 60)
print("✓ Todos los tests pasaron correctamente")
print("=" * 60)
print()
print("Sistema listo para ejecutar:")
print("  python src/main.py")
print()