train.py

import logging
import groq
from config import GROQ_API_KEY, GROQ_MODEL

logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(levelname)s - %(message)s',
    datefmt='%Y-%m-%d %H:%M:%S'
)

def verificar_conexion_groq():
    try:
        cliente = groq.Client(api_key=GROQ_API_KEY)
        cliente.chat.completions.create(
            model=GROQ_MODEL,
            messages=[{"role": "user", "content": "¿Estás disponible?"}]
        )
        logging.info("✅ Conexión con Groq verificada exitosamente")
        return True
    except Exception as e:
        logging.error(f"❌ Error al conectar con Groq: {e}")
        return False