agent.py

from transformers import pipeline

# Usaremos un modelo ligero para mejor compatibilidad
modelo = "google/gemma-2b-it"

generador = pipeline("text-generation", model=modelo)

def ejecutar_agente(texto):
    """
    Procesa una pregunta usando el modelo LLM.
    """
    resultado = generador(texto, max_new_tokens=300, do_sample=True, temperature=0.7)
    return resultado[0]["generated_text"]