File size: 807 Bytes
0280c89
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
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