environment.py

import random

class Environment:
    def execute(self, action):
        trace = {
            "move": [random.choice([-1, 0, 1]), random.choice([-1, 0, 1])],
            "event": random.choice(["ping", "ack", "idle"]),
            "mode": action.get("mode")
        }
        return 1.0, trace