agent_mind.py

class AgentMind:
    """
    Adds goal + memory + planning to agents.
    """

    def __init__(self):

        self.memory = []  # compressed experiences
        self.goals = ["survive", "explore"]
        self.goal_weights = {
            "survive": 1.0,
            "explore": 0.5
        }

    # =====================================================
    # MEMORY UPDATE
    # =====================================================
    def remember(self, perception, action, reward):

        self.memory.append({
            "perception": perception,
            "action": action,
            "reward": reward
        })

        if len(self.memory) > 100:
            self.memory.pop(0)

    # =====================================================
    # GOAL SELECTION
    # =====================================================
    def select_goal(self):

        # simple adaptive weighting
        return max(self.goal_weights, key=self.goal_weights.get)

    # =====================================================
    # ACTION PLANNING (simple heuristic)
    # =====================================================
    def plan(self, perception):

        goal = self.select_goal()

        if goal == "survive":
            if perception["energy"] < 0.4:
                return "move"
            return "idle"

        if goal == "explore":
            return "move"

        return "idle"