action_dict = {
    "MiniHack-LavaCross-Levitate-Potion-Inv-v0":{0: ["north"],
                                                 1: ["south"],
                                                 2: ["east"],
                                                 3: ["west"],
                                                 4: ["northwest"],
                                                 5: ["northeast"],
                                                 6: ["southwest"],
                                                 7: ["southeast"],
                                                 8: ["q", "f"]
    },
    "MiniHack-WoD-Hard-v0" : {
        0: ["north"],
        1: ["south"],
        2: ["east"],
        3: ["west"],
        4: ["northwest"],
        5: ["northeast"],
        6: ["southwest"],
        7: ["southeast"],
        8: [","],
        9: ['z', 'f', 'north'],
        10: ['z', 'f', 'south'],
        11: ['z', 'f', 'east'],
        12: ['z', 'f', 'west'],
        13: ['z', 'f', 'northwest'],
        14: ['z', 'f', 'northeast'],
        15: ['z', 'f', 'southwest'],
        16: ['z', 'f', 'southeast']
    },
    "MiniHack-Room-Monster-5x5-v0" : {
        0: ["north"],
        1: ["south"],
        2: ["east"],
        3: ["west"],
        4: ["northwest"],
        5: ["northeast"],
        6: ["southwest"],
        7: ["southeast"]
    },
    "MiniHack-Room-Monster-15x15-v0" : {
        0: ["north"],
        1: ["south"],
        2: ["east"],
        3: ["west"],
        4: ["northwest"],
        5: ["northeast"],
        6: ["southwest"],
        7: ["southeast"]
    }
}

GAME_DESCRIPTION = {
    "MiniHack-LavaCross-Levitate-Potion-Inv-v0": {
        "description": "You are a game agent in the Nethack environment. Your goal is to drink a poison and cross lava. First, you need to drink the poison which is already in your inventory. Next, you need to cross lava lake and enter the exit.",
        "options": [
            "drink posion",
            "go to goal"
        ]
    },
    
    "MiniHack-WoD-Hard-v0": {
        "description": "You are a game agent in the Nethack environment. Your goal is to pick up a wand, use deadly magic by the wand to kill a monster and finally enter the exit.",
        "options": [
            "go to wand",
            "pick up wand",
            "use magic",
            "go to goal"
        ]
    },
    "MiniHack-Room-Monster-5x5-v0": {
        "description": "You are a game agent in the Nethack environment. Your goal is to leave the room by killing monsters.",
        "options": [
            "attack monster",
            "go to goal"
        ]
    },
    "MiniHack-Room-Monster-15x15-v0": {
        "description": "You are a game agent in the Nethack environment. Your goal is to leave the room by killing monsters.",
        "options": [
            "attack monster",
            "go to goal"
        ]
    }
}