app/tasks.py

class TaskConfig:
    def __init__(self, level: str, max_steps: int):
        self.level = level
        self.max_steps = max_steps

EASY_TASK = TaskConfig(level="easy", max_steps=1)
MEDIUM_TASK = TaskConfig(level="medium", max_steps=1)
HARD_TASK = TaskConfig(level="hard", max_steps=5)

def get_task(level: str) -> TaskConfig:
    tasks = {"easy": EASY_TASK, "medium": MEDIUM_TASK, "hard": HARD_TASK}
    return tasks.get(level.lower(), EASY_TASK)