agent.py

from transformers import pipeline

classifier = pipeline("text-classification")

def llm_agent(observation):
    result = classifier(observation)[0]['label'].lower()

    if "spam" in result:
        return "spam"
    elif "important" in result:
        return "important"
    else:
        return "normal"