import json

class AgenticTraitMapper:
    def __init__(self, config_path='config/trait_bindings.json'):
        with open(config_path, 'r') as f:
            self.trait_map = json.load(f)

    def get_traits(self, segment_id):
        return self.trait_map.get(segment_id, {}).get("traits", [])

    def get_agent(self, segment_id):
        return self.trait_map.get(segment_id, {}).get("agent", None)