让贪吃蛇小游戏适配大语言模型（含代码）

---

直接上代码

import numpy as np
from snake_game import SnakeGame
from snake_game import Direction

class SnakeEnv:
    def __init__(self):
        """初始化蛇游戏环境"""
        self.game = SnakeGame()
        self.action_space = 4  # 上、下、左、右4个动作
        self.observation_space = None  # 状态空间为字符串描述
    
    def reset(self):
        """重置游戏状态"""
        self.game.reset()
        return self._get_state()
    
    def step(self, action):
        """
        执行一个动作
        参数:
            action: 0=上, 1=下, 2=左, 3=右
        返回:
            state: 当前状态字符串
            reward: 奖励值
            done: 是否结束
            info: 额外信息
        """
        # 执行动作