agent.py

class Agent:
    """
    Core reasoning unit inside the society system.
    Holds routing + identity + policy mode.
    """

    def __init__(self, name, model_router, llm_router, mode="honest"):

        self.name = name
        self.model_router = model_router
        self.llm_router = llm_router
        self.mode = mode

        # internal state (important for controller stability)
        self.memory = []
        self.last_output = None
        from goal_system import GoalSystem

        self.state["goals"] = GoalSystem()

    # =====================================================
    # BASIC RESPONSE INTERFACE
    # =====================================================
    def respond(self, prompt, task_type="ir_generation"):

        models = list(self.llm_router.models.keys())

        model_name = self.model_router.select(
            task_type,
            models
        )

        model = self.llm_router.models[model_name]

        output = model.call(prompt)

        self.last_output = output
        self.memory.append(output)

        return output

    # =====================================================
    # LIGHTWEIGHT STATE HOOK (for controller/debugging)
    # =====================================================
    def snapshot(self):

        return {
            "name": self.name,
            "mode": self.mode,
            "last_output": self.last_output
        }