src/core/theory_of_mind.py

class TheoryOfMind:
    def __init__(self):
        self.mental_state_modeler = MentalStateModeler()
        self.belief_system = BeliefSystem()
        self.perspective_engine = PerspectiveEngine()
        
    def model_agent_mind(self, 
                        agent_data: Dict[str, Any],
                        interaction_history: List[Dict[str, Any]]) -> Dict[str, Any]:
        mental_state = self.mental_state_modeler.infer_state(agent_data)
        belief_model = self.belief_system.construct_model(agent_data, interaction_history)
        perspective = self.perspective_engine.simulate_perspective(mental_state, belief_model)
        
        return {
            'mental_state': mental_state,
            'beliefs': belief_model,
            'perspective': perspective,
            'prediction': self._predict_behavior(mental_state, belief_model)
        }