model.py

import logging
import groq
from config import GROQ_API_KEY, GROQ_MODEL, GROQ_TEMPERATURE, GROQ_MAX_TOKENS

logger = logging.getLogger(__name__)

def conectar_con_groq(mensaje):
    try:
        cliente = groq.Client(api_key=GROQ_API_KEY)
        respuesta = cliente.chat.completions.create(
            model=GROQ_MODEL,
            messages=[{"role": "user", "content": mensaje}],
            temperature=GROQ_TEMPERATURE,
            max_tokens=GROQ_MAX_TOKENS
        )
        return respuesta.choices[0].message.content
    except Exception as e:
        logger.error(f"Error al conectar con Groq: {e}")
        return None

def obtener_respuesta_de_groq(mensaje):
    respuesta = conectar_con_groq(mensaje)
    if respuesta is None:
        return "❌ Error al conectar con Groq"
    return respuesta