env/tasks.py

TASKS = {
    "easy": {
        "query": "I forgot my password",
        "expected_category": "account",
        "expected_action": "respond",
        "difficulty": "easy"
    },
    "medium": {
        "query": "I was charged twice",
        "expected_category": "billing",
        "expected_action": "resolve",
        "difficulty": "medium"
    },
    "hard": {
        "query": "My account was hacked",
        "expected_category": "security",
        "expected_action": "escalate",
        "difficulty": "hard"
    }
}